ssh/sftp: add method to close client connection

The Close method close the client sftp session and release all its
resources.

2 files changed, 20 insertions, 1 deletions

diff --git a/lib/ssh/sftp/client.go b/lib/ssh/sftp/client.go
index e6462ce3..ea8c4ae8 100644
--- a/lib/ssh/sftp/client.go
+++ b/lib/ssh/sftp/client.go
@@ -79,6 +79,18 @@ func NewClient(sshc *ssh.Client) (cl *Client, err error) {
 	return cl, nil
 }
 
+// Close the client sftp session and release all resources.
+func (cl *Client) Close() (err error) {
+	err = cl.sess.Close()
+
+	cl.requestId = 0
+	cl.pipeErr = nil
+	cl.pipeOut = nil
+	cl.pipeIn = nil
+
+	return fmt.Errorf(`Close: %w`, err)
+}
+
 // CloseFile close the remote file handle.
 func (cl *Client) CloseFile(fh *FileHandle) (err error) {
 	if fh == nil {
diff --git a/lib/ssh/sftp/sftp_test.go b/lib/ssh/sftp/sftp_test.go
index 38523b59..5aba892e 100644
--- a/lib/ssh/sftp/sftp_test.go
+++ b/lib/ssh/sftp/sftp_test.go
@@ -36,7 +36,7 @@ func TestMain(m *testing.M) {
 	cfg := &config.Section{
 		Field: map[string]string{
 			config.KeyUser:     `ms`,
-			config.KeyHostname: `127.0.0.1`,
+			config.KeyHostname: `localhost`,
 			config.KeyPort:     `22`,
 		},
 		IdentityFile: []string{
@@ -57,6 +57,13 @@ func TestMain(m *testing.M) {
 	fmt.Printf("Server version: %d\n", testClient.version)
 	fmt.Printf("Server extensions: %v\n", testClient.exts)
 
+	defer func() {
+		var errClose = testClient.Close()
+		if errClose != nil {
+			log.Printf(`TestMain: %s`, errClose)
+		}
+	}()
+
 	os.Exit(m.Run())
 }