fix sftp client hang on exit

ferama · Nov 26, 2023 · bf76fe4 · bf76fe4
1 parent 8c617b1
commit bf76fe4
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pkg/sshd/channel_handler.go b/pkg/sshd/channel_handler.go
@@ -282,12 +282,13 @@ func (s *channelHandler) handleSftpRequest(channel ssh.Channel) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	if err := server.Serve(); err == io.EOF {
-		server.Close()
-		log.Print("sftp client exited session.")
-	} else if err != nil {
-		log.Printf("sftp server completed with error: %s", err)
+	if err := server.Serve(); err != nil {
+		if err != io.EOF {
+			log.Fatal("sftp server completed with error:", err)
+		}
 	}
+	server.Close()
+	log.Print("sftp client exited session.")
 }
 
 func (s *channelHandler) sendSignal(channel ssh.Channel, signal string) {