balki/tss
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add auth support for unix socks5 proxy

Browse Source
This commit is contained in:
Balakrishnan Balasubramanian 2022-05-23 12:45:25 -04:00
parent 472b306dbf
commit 67e479c865
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
proxy/proxy.go
View File

@ -20,7 +20,7 @@ func GetTransport(proxy string) (http.RoundTripper, error) {
} }
if proxyUrl.Host == "unix" && proxyUrl.Scheme == "socks5" && if proxyUrl.Host == "unix" && proxyUrl.Scheme == "socks5" &&
len(proxyUrl.Path) > 1 /* Path cannot be empty or just / */ { len(proxyUrl.Path) > 1 /* Path cannot be empty or just / */ {
return unixSocks5Proxy(proxyUrl.Path) return unixSocks5Proxy(proxyUrl)
} }
return proxyTcp(proxyUrl) return proxyTcp(proxyUrl)
} }
@ -38,14 +38,24 @@ func (d *forwardDialer) Dial(network, address string) (net.Conn, error) {
//return defaultTransport().Dial(network, address) //return defaultTransport().Dial(network, address)
} }
func unixSocks5Proxy(path string) (http.RoundTripper, error) { func unixSocks5Proxy(proxyUrl *url.URL) (http.RoundTripper, error) {
trans := defaultTransport().Clone() trans := defaultTransport().Clone()
if trans.DialContext == nil { if trans.DialContext == nil {
panic("DefaultTransport has nil DialContext") panic("DefaultTransport has nil DialContext")
} }
dialer, err := proxy.SOCKS5("unix", path, nil /*auth*/, &forwardDialer{trans.DialContext}) var auth *proxy.Auth
username := proxyUrl.User.Username()
password, _ := proxyUrl.User.Password()
// Both username and password should have atleast one char
if username != "" && password != "" {
auth = &proxy.Auth{
User: username,
Password: password,
}
}
dialer, err := proxy.SOCKS5("unix", proxyUrl.Path, auth, &forwardDialer{trans.DialContext})
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to make socks proxy, path: %s, err: %w", path, err) return nil, fmt.Errorf("failed to make socks proxy, url: %s, err: %w", proxyUrl, err)
} }
ctxDialer, ok := dialer.(proxy.ContextDialer) ctxDialer, ok := dialer.(proxy.ContextDialer)
if !ok { if !ok {