fix(v1.0): fix md5

cuisongliu · Mar 17, 2020 · bfb1f5a · bfb1f5a
1 parent 81d16b6
commit bfb1f5a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/sshutil/connect.go b/pkg/sshutil/connect.go
@@ -6,6 +6,7 @@ import (
 	"golang.org/x/crypto/ssh"
 	"io/ioutil"
 	"net"
+	"os"
 	"strings"
 	"time"
 )
@@ -59,7 +60,11 @@ func (ss *SSH) sshAuthMethod(passwd, pkFile string) ssh.AuthMethod {
 		am = ssh.Password(passwd)
 	} else {
 		pkData := ss.readFile(pkFile)
-		pk, _ := ssh.ParsePrivateKey([]byte(pkData))
+		pk, err := ssh.ParsePrivateKey([]byte(pkData))
+		if err != nil {
+			logger.Error(err)
+			os.Exit(1)
+		}
 		am = ssh.PublicKeys(pk)
 	}
 	return am
@@ -69,7 +74,7 @@ func (ss *SSH) readFile(name string) string {
 	content, err := ioutil.ReadFile(name)
 	if err != nil {
 		logger.Error("[globals] read file err is : %s", err)
-		return ""
+		os.Exit(1)
 	}
 	return string(content)
 }