fix(platform): validate socks5 proxy

tkestack · Dec 15, 2022 · f47613b · f47613b
1 parent 966b72e
commit f47613b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/util/ssh/socks5_proxy.go b/pkg/util/ssh/socks5_proxy.go
@@ -71,6 +71,14 @@ func (sk SOCKS5) ProxyConn(targetAddr string) (net.Conn, func(), error) {
 }
 
 func (sk SOCKS5) CheckTunnel() error {
-	_, err := proxy.SOCKS5("tcp", net.JoinHostPort(sk.Host, fmt.Sprintf("%d", sk.Port)), nil, proxy.Direct)
-	return err
+	tunnelAddr := net.JoinHostPort(sk.Host, fmt.Sprintf("%d", sk.Port))
+	dialer, err := proxy.SOCKS5("tcp", tunnelAddr, nil, nil)
+	if err != nil {
+		return fmt.Errorf("get proxy dialler failed, tunnel is unavailable： %v", err)
+	}
+	_, err = dialer.Dial("tcp", "ccr.ccs.tencentyun.com:443")
+	if err != nil {
+		return fmt.Errorf("dial tunnel failed, tunnel is unavailable： %v", err)
+	}
+	return nil
 }