From 87e54acd8a3ef10b1dbb60067cb2a0d8789d1c5a Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Thu, 3 Sep 2020 15:09:23 -0700
Subject: [PATCH] add option to disable compression Fixes #239

---
 src/cli/cli.go   |  2 ++
 src/croc/croc.go | 32 +++++++++++++++++++++++++-------
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/src/cli/cli.go b/src/cli/cli.go
index dac45e5..e774173 100644
--- a/src/cli/cli.go
+++ b/src/cli/cli.go
@@ -78,6 +78,7 @@ func Run() (err error) {
 		cli.BoolFlag{Name: "debug", Usage: "toggle debug mode"},
 		cli.BoolFlag{Name: "yes", Usage: "automatically agree to all prompts"},
 		cli.BoolFlag{Name: "stdout", Usage: "redirect file to stdout"},
+		cli.BoolFlag{Name: "no-compress", Usage: "disable compression"},
 		cli.BoolFlag{Name: "ask", Usage: "make sure sender and recipient are prompted"},
 		cli.StringFlag{Name: "relay", Value: models.DEFAULT_RELAY, Usage: "address of the relay"},
 		cli.StringFlag{Name: "relay6", Value: models.DEFAULT_RELAY6, Usage: "ipv6 address of the relay"},
@@ -162,6 +163,7 @@ func send(c *cli.Context) (err error) {
 		NoMultiplexing: c.Bool("no-multi"),
 		RelayPassword:  determinePass(c),
 		SendingText:    c.String("text") != "",
+		NoCompress:     c.GlobalBool("no-compress"),
 	}
 	if crocOptions.RelayAddress != models.DEFAULT_RELAY {
 		crocOptions.RelayAddress6 = ""
diff --git a/src/croc/croc.go b/src/croc/croc.go
index e68e0d6..0307632 100644
--- a/src/croc/croc.go
+++ b/src/croc/croc.go
@@ -62,6 +62,7 @@ type Options struct {
 	DisableLocal   bool
 	Ask            bool
 	SendingText    bool
+	NoCompress     bool
 }
 
 // Client holds the state of the croc transfer
@@ -140,6 +141,7 @@ type SenderInfo struct {
 	MachineID       string
 	Ask             bool
 	SendingText     bool
+	NoCompress      bool
 }
 
 // New establishes a new connection for transferring files between two instances.
@@ -700,6 +702,10 @@ func (c *Client) processMessageFileInfo(m message.Message) (done bool, err error
 		return
 	}
 	c.Options.SendingText = senderInfo.SendingText
+	c.Options.NoCompress = senderInfo.NoCompress
+	if c.Options.NoCompress {
+		log.Debug("disabling compression")
+	}
 	if c.Options.SendingText {
 		c.Options.Stdout = true
 	}
@@ -960,6 +966,7 @@ func (c *Client) updateIfSenderChannelSecured() (err error) {
 			MachineID:       machID,
 			Ask:             c.Options.Ask,
 			SendingText:     c.Options.SendingText,
+			NoCompress:      c.Options.NoCompress,
 		})
 		if err != nil {
 			log.Error(err)
@@ -1281,7 +1288,9 @@ func (c *Client) receiveData(i int) {
 		if err != nil {
 			panic(err)
 		}
-		data = compress.Decompress(data)
+		if !c.Options.NoCompress {
+			data = compress.Decompress(data)
+		}
 
 		// get position
 		var position uint64
@@ -1366,13 +1375,22 @@ func (c *Client) sendData(i int) {
 				// log.Debugf("sending chunk %d", pos)
 				posByte := make([]byte, 8)
 				binary.LittleEndian.PutUint64(posByte, pos)
-
-				dataToSend, err := crypt.Encrypt(
-					compress.Compress(
+				var err error
+				var dataToSend []byte
+				if c.Options.NoCompress {
+					dataToSend, err = crypt.Encrypt(
 						append(posByte, data[:n]...),
-					),
-					c.Key,
-				)
+						c.Key,
+					)
+				} else {
+
+					dataToSend, err = crypt.Encrypt(
+						compress.Compress(
+							append(posByte, data[:n]...),
+						),
+						c.Key,
+					)
+				}
 				if err != nil {
 					panic(err)
 				}