Close session on io.Copy error in interactive shells

Author: Fahrj

ssh_session_unix.go

@@ -47,8 +47,14 @@ func createPty(s ssh.Session, shell string) {
 		}
 	}()
 
-	go io.Copy(f, s)
-	go io.Copy(s, f)
+	go func() {
+		io.Copy(f, s)
+		s.Close()
+	}()
+	go func() {
+		io.Copy(s, f)
+		s.Close()
+	}()
 
 	done := make(chan error, 1)
 	go func() { done <- cmd.Wait() }()

ssh_session_windows.go

@@ -127,8 +127,14 @@ func createPty(s ssh.Session, shell string) {
 		defer process.Kill()
 
 		// Link data streams of ssh session and conpty
-		go io.Copy(s, cpty.OutPipe())
-		go io.Copy(cpty.InPipe(), s)
+		go func() {
+			io.Copy(s, cpty.OutPipe())
+			s.Close()
+		}()
+		go func() {
+			io.Copy(cpty.InPipe(), s)
+			s.Close()
+		}()
 
 		done := make(chan struct {
 			*os.ProcessState