tofu

connect.go

@@ -54,6 +54,8 @@ type Connection struct {
 	keypair             keypair.KeyPair
 	encryptedPassword   string
 	spinner             *spinner.Spinner
+	knownKeys           map[string]struct{}
+	partnerKey          string
 }
 
 type FileMetaData struct {
@@ -85,6 +87,7 @@ func NewConnection(config *AppConfig) (*Connection, error) {
 	c.rate = config.Rate
 	c.Local = config.Local
 	c.NoLocal = config.Local
+	c.partnerKey = ""
 
 	// make or load keypair
 	homedir, err := homedir.Dir()
@@ -101,6 +104,19 @@ func NewConnection(config *AppConfig) (*Connection, error) {
 			return c, err
 		}
 	}
+	pathToCrocKnownKeys := path.Join(homedir, ".config", "croc", "known")
+	if _, errExists := os.Stat(pathToCrocKnownKeys); os.IsNotExist(errExists) {
+		// path/to/whatever does not exist
+		knownKeys := make(map[string]struct{})
+		bKnownKeys, err := json.MarshalIndent(knownKeys, "", " ")
+		if err != nil {
+			return c, err
+		}
+		err = ioutil.WriteFile(pathToCrocKnownKeys, bKnownKeys, 0644)
+		if err != nil {
+			return c, err
+		}
+	}
 	keypairBytes, err := ioutil.ReadFile(pathToCrocConfig)
 	if err != nil {
 		return c, err
@@ -109,6 +125,14 @@ func NewConnection(config *AppConfig) (*Connection, error) {
 	if err != nil {
 		return c, err
 	}
+	knownKeyBytes, err := ioutil.ReadFile(pathToCrocKnownKeys)
+	if err != nil {
+		return c, err
+	}
+	err = json.Unmarshal(knownKeyBytes, &c.knownKeys)
+	if err != nil {
+		return c, err
+	}
 
 	if c.Debug {
 		SetLogLevel("debug")
@@ -414,9 +438,12 @@ func (c *Connection) runClient(serverName string) error {
 					// message is IP address, lets check next message
 					log.Debugf("[%d] got ok from relay: %s", id, message)
 					publicKeyRecipient := receiveMessage(connection)
+					c.partnerKey = publicKeyRecipient
 					// check if okay again
 					if id == 0 {
 						c.spinner.Stop()
+						getOK := "y"
+						if _, knownKey := c.knownKeys[publicKeyRecipient]; !knownKey {
 							fmt.Fprintf(os.Stderr, "Your public key: ")
 							if runtime.GOOS == "windows" {
 								color.New(color.FgHiRed).Fprintln(color.Output, HashWords(c.keypair.Public))
@@ -430,11 +457,10 @@ func (c *Connection) runClient(serverName string) error {
 							} else {
 								color.New(color.FgHiYellow).Fprintln(os.Stderr, HashWords(publicKeyRecipient))
 							}
-
-						getOK := "y"
 							if !c.Yes {
 								getOK = getInput("ok? (y/n): ")
 							}
+						}
 						responses.Lock()
 						responses.gotOK = true
 						responses.Unlock()
@@ -515,8 +541,14 @@ func (c *Connection) runClient(serverName string) error {
 					log.Debugf("sending encrypted passphrase: %s", c.encryptedPassword)
 					sendMessage(c.encryptedPassword, connection)
 					// wait for relay go
+					if id == 0 {
+						c.spinner = spinner.New(spinner.CharSets[24], 100*time.Millisecond) // Build our new spinner
+						c.spinner.Suffix = " Waiting on recipient..."
+						c.spinner.Start()
+					}
 					receiveMessage(connection)
 					if id == 0 {
+						c.spinner.Stop()
 						fmt.Fprintf(os.Stderr, "\nSending (->@%s)..\n", message)
 					}
 					// wait for pipe to be made
@@ -560,6 +592,7 @@ func (c *Connection) runClient(serverName string) error {
 					}
 					// now get public key
 					publicKeySender := receiveMessage(connection)
+					c.partnerKey = publicKeySender
 
 					// have the main thread ask for the okay
 					if id == 0 {
@@ -599,6 +632,8 @@ func (c *Connection) runClient(serverName string) error {
 							fmt.Fprintf(os.Stderr, "Will not overwrite file!")
 							os.Exit(1)
 						}
+						getOK := "y"
+						if _, knownKey := c.knownKeys[publicKeySender]; !knownKey {
 							fmt.Fprintf(os.Stderr, "Your public key: ")
 							if runtime.GOOS == "windows" {
 								color.New(color.FgHiRed).Fprintln(color.Output, HashWords(c.keypair.Public))
@@ -612,10 +647,10 @@ func (c *Connection) runClient(serverName string) error {
 							} else {
 								color.New(color.FgHiYellow).Fprintln(os.Stderr, HashWords(publicKeySender))
 							}
-						getOK := "y"
 							if !c.Yes {
 								getOK = getInput("ok? (y/n): ")
 							}
+						}
 						if getOK == "y" {
 							responses.Lock()
 							responses.gotOK = true
@@ -800,6 +835,23 @@ func (c *Connection) runClient(serverName string) error {
 			}
 		}
 	}
+	if c.partnerKey != "" {
+		// save the partner for next time
+		homedir, _ := homedir.Dir()
+		pathToCrocKnownKeys := path.Join(homedir, ".config", "croc", "known")
+		_, haveKey := c.knownKeys[c.partnerKey]
+		if !haveKey {
+			c.knownKeys[c.partnerKey] = struct{}{}
+			bKnownKeys, err := json.MarshalIndent(c.knownKeys, "", " ")
+			if err != nil {
+				return err
+			}
+			err = ioutil.WriteFile(pathToCrocKnownKeys, bKnownKeys, 0644)
+			if err != nil {
+				return err
+			}
+		}
+	}
 	fmt.Fprintf(os.Stderr, " (%s/s)\n", humanize.Bytes(uint64(float64(1000000000)*float64(c.File.Size)/float64(timeSinceStart))))
 	return nil
 }