From 6202d0d9322e63f934d0f1b15952213f5aedeb21 Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Fri, 20 Sep 2019 09:32:32 -0700
Subject: [PATCH] reduce complexity

---
 src/croc/croc.go | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/croc/croc.go b/src/croc/croc.go
index ab15e78d..9e66663e 100644
--- a/src/croc/croc.go
+++ b/src/croc/croc.go
@@ -653,7 +653,7 @@ func (c *Client) processMessage(payload []byte) (done bool, err error) {
 	return
 }
 
-func (c *Client) updateState() (err error) {
+func (c *Client) updateIfSenderChannelSecured() (err error) {
 	if c.Options.IsSender && c.Step1ChannelSecured && !c.Step2FileInfoTransfered {
 		var b []byte
 		b, err = json.Marshal(SenderInfo{
@@ -673,6 +673,10 @@ func (c *Client) updateState() (err error) {
 
 		c.Step2FileInfoTransfered = true
 	}
+	return
+}
+
+func (c *Client) updateIfRecipientHasFileInfo() (err error) {
 	if !c.Options.IsSender && c.Step2FileInfoTransfered && !c.Step3RecipientRequestFile {
 		// find the next file to transfer and send that number
 		// if the files are the same size, then look for missing chunks
@@ -821,6 +825,20 @@ func (c *Client) updateState() (err error) {
 		}
 		c.Step3RecipientRequestFile = true
 	}
+	return
+}
+
+func (c *Client) updateState() (err error) {
+	err = c.updateIfSenderChannelSecured()
+	if err != nil {
+		return
+	}
+
+	err = c.updateIfRecipientHasFileInfo()
+	if err != nil {
+		return
+	}
+
 	if c.Options.IsSender && c.Step3RecipientRequestFile && !c.Step4FileTransfer {
 		log.Debug("start sending data!")
 		if !c.firstSend {