Better error management

1 files changed, 4 insertions, 1 deletions

diff --git a/main.go b/main.go
index c70dd4d..a76690e 100644
--- a/main.go
+++ b/main.go
@@ -97,6 +97,7 @@ func handleSSHConn(nConn net.Conn, dconfig *Config, config *ssh.ServerConfig) {
 	serverConn, chans, reqs, err := ssh.NewServerConn(nConn, config)
 	if err != nil {
 		log.Printf("failed to handshake: ", err)
+		return
 	}
 	defer serverConn.Conn.Close()
 	user := serverConn.Conn.User()
@@ -120,6 +121,7 @@ func handleSSHConn(nConn net.Conn, dconfig *Config, config *ssh.ServerConfig) {
 		channel, requests, err := newChannel.Accept()
 		if err != nil {
 			log.Print("could not accept channel.", err)
+			return
 		}
 		log.Printf("Channel accepted\n")
 
@@ -155,7 +157,8 @@ func handleSSHConn(nConn net.Conn, dconfig *Config, config *ssh.ServerConfig) {
 		server, err := sftp.NewServer(ctx, channel, &fs)
 
 		if err != nil {
-			log.Fatal(err)
+			log.Println(err)
+			return
 		}
 
 		if err := server.Serve(); err == io.EOF {