Function loadSSHConnByProxy

core/utils/ssh/ssh.go:271–321

(network, addr string, timeout time.Duration)

}

func loadSSHConnByProxy(network, addr string, timeout time.Duration) (net.Conn, error) {
	settingRepo := repo.NewISettingRepo()
	proxyType, err := settingRepo.Get(repo.WithByKey("ProxyType"))
	if err != nil {
		return nil, fmt.Errorf("get proxy type from db failed, err: %v", err)
	}
	if len(proxyType.Value) == 0 {
		return nil, fmt.Errorf("get proxy type from db failed, err: %v", err)
	}
	proxyUrl, _ := settingRepo.Get(repo.WithByKey("ProxyUrl"))
	port, _ := settingRepo.Get(repo.WithByKey("ProxyPort"))
	user, _ := settingRepo.Get(repo.WithByKey("ProxyUser"))
	passwd, _ := settingRepo.Get(repo.WithByKey("ProxyPasswd"))

	pass, _ := encrypt.StringDecrypt(passwd.Value)
	proxyItem := fmt.Sprintf("%s:%s", proxyUrl.Value, port.Value)
	switch proxyType.Value {
	case "http", "https":
		item := HTTPProxyDialer{
			Type:     proxyType.Value,
			URL:      proxyItem,
			User:     user.Value,
			Password: pass,
		}
		return HTTPDial(item, network, addr)
	case "socks5":
		var auth *proxy.Auth
		if len(user.Value) == 0 {
			auth = nil
		} else {
			auth = &proxy.Auth{
				User:     user.Value,
				Password: pass,
			}
		}
		dialer, err := proxy.SOCKS5("tcp", proxyItem, auth, &net.Dialer{
			Timeout:   30 * time.Second,
			KeepAlive: 30 * time.Second,
		})
		if err != nil {
			return nil, fmt.Errorf("new socks5 proxy failed, err: %v", err)
		}
		return dialer.Dial(network, addr)
	default:
		conn, err := net.DialTimeout(network, addr, timeout)
		if err != nil {
			return nil, err
		}
		return conn, nil
	}
}