Fix default port for ssh

wzshiming · Jul 12, 2021 · c7d728f · c7d728f
1 parent 337bea3
commit c7d728f
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/protocols/ssh/ssh.go b/protocols/ssh/ssh.go
@@ -173,14 +173,6 @@ func config(addr string) (host string, config *ssh.ClientConfig, err error) {
 		pwd, isPwd = ur.User.Password()
 	}
 
-	host, port, err := net.SplitHostPort(ur.Host)
-	if err != nil {
-		return "", nil, err
-	}
-	if port == "" {
-		port = "22"
-	}
-
 	config = &ssh.ClientConfig{
 		User:            user,
 		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
@@ -212,5 +204,11 @@ func config(addr string) (host string, config *ssh.ClientConfig, err error) {
 		}
 		config.Auth = append(config.Auth, ssh.PublicKeys(signer))
 	}
+
+	host = ur.Hostname()
+	port := ur.Port()
+	if port == "" {
+		port = "22"
+	}
 	return net.JoinHostPort(host, port), config, nil
 }