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)
 }