update. add StdoutMutex

Author: blacknon

connect.go

@@ -11,6 +11,7 @@ import (
 	"net"
 	"os"
 	"os/signal"
+	"sync"
 	"syscall"
 	"time"
 
@@ -53,6 +54,9 @@ type Connect struct {
 	// Set the TTY to be used as the input and output for the Session/Cmd.
 	PtyRelayTty *os.File
 
+	// StdoutMutex is a mutex for use Stdout.
+	StdoutMutex *sync.Mutex
+
 	// CheckKnownHosts if true, check knownhosts.
 	// Ignored if HostKeyCallback is set.
 	// Set it before CraeteClient.

knownhosts.go

@@ -79,6 +79,12 @@ func (c *Connect) VerifyAndAppendNew(hostname string, remote net.Addr, key ssh.P
 	filepath := knownHostsFiles[0]
 	var line int
 
+	// check mutex
+	if c.StdoutMutex != nil {
+		c.StdoutMutex.Lock()
+		defer c.StdoutMutex.Unlock()
+	}
+
 	// check error
 	keyErr, ok := err.(*knownhosts.KeyError)
 	if !ok || len(keyErr.Want) > 0 {
@@ -107,7 +113,7 @@ func (c *Connect) VerifyAndAppendNew(hostname string, remote net.Addr, key ssh.P
 
 	err = writeKnownHostsKey(filepath, line, hostname, remote, key)
 
-	return nil
+	return err
 }
 
 // askAddingUnknownHostKey