From 09a16cacd51ab8e6a06cf2d922d39b405aec71a8 Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Wed, 4 Jul 2018 08:43:30 -0700
Subject: [PATCH] remove partially sent files

---
 main.go       | 2 +-
 src/api.go    | 1 +
 src/client.go | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index e3348f3..d7a3998 100644
--- a/main.go
+++ b/main.go
@@ -114,7 +114,7 @@ func main() {
 		fmt.Fprintf(c.App.Writer, "lipstick\nkiss\nme\nlipstick\nringo\n")
 	}
 	app.Action = func(c *cli.Context) error {
-		return cr.Receive(c.Args().First())
+		return receive(c)
 	}
 	app.Before = func(c *cli.Context) error {
 		cr = croc.Init()
diff --git a/src/api.go b/src/api.go
index 0fb379f..a9b8b6d 100644
--- a/src/api.go
+++ b/src/api.go
@@ -133,6 +133,7 @@ func (c *Croc) Send(fname string, codePhrase string) (err error) {
 // Receive will receive something through the croc relay
 func (c *Croc) Receive(codePhrase string) (err error) {
 	defer c.cleanup()
+	log.Debugf("receiving with code phrase: %s", codePhrase)
 	if !c.NoLocal {
 		// try to discovery codephrase and server through peer network
 		discovered, errDiscover := peerdiscovery.Discover(peerdiscovery.Settings{
diff --git a/src/client.go b/src/client.go
index 3fa10c1..a042c24 100644
--- a/src/client.go
+++ b/src/client.go
@@ -505,6 +505,7 @@ func (c *Croc) sendFile(filename string, id int, connection net.Conn) error {
 
 	// open encrypted file chunk, if it exists
 	log.Debug("opening encrypted file chunk: " + filename)
+	defer os.Remove(filename)
 	file, err := os.Open(filename)
 	if err != nil {
 		log.Error(err)