forked from coyim/coyim
-
Notifications
You must be signed in to change notification settings - Fork 0
/
xmpp.go
73 lines (60 loc) · 1.35 KB
/
xmpp.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package config
import (
"net/url"
ournet "github.com/twstrike/coyim/net"
"golang.org/x/net/proxy"
)
func socks5UnixProxy(u *url.URL, d proxy.Dialer) (proxy.Dialer, error) {
var auth *proxy.Auth
if u.User != nil {
auth = &proxy.Auth{
User: u.User.Username(),
}
if p, ok := u.User.Password(); ok {
auth.Password = p
}
}
return proxy.SOCKS5("unix", u.Path, auth, d)
}
func genTorAutoString() string {
s := [10]byte{}
randomString(s[:])
return "randomTorAuto-" + string(s[:])
}
func genTorAutoUsername() string {
return genTorAutoString()
}
func genTorAutoPassword() string {
return genTorAutoString()
}
func genTorAutoAuth(u *url.URL) *proxy.Auth {
auth := &proxy.Auth{}
if u.User != nil {
auth.User = u.User.Username()
if p, ok := u.User.Password(); ok {
auth.Password = p
}
} else {
auth.User = genTorAutoUsername()
auth.Password = genTorAutoPassword()
}
return auth
}
func genTorAutoAddr(u *url.URL) string {
if u.Host == "" {
return ournet.Tor.Address()
}
return u.Host
}
func torAutoProxy(u *url.URL, d proxy.Dialer) (proxy.Dialer, error) {
auth := genTorAutoAuth(u)
addr := genTorAutoAddr(u)
if addr == "" {
return nil, ErrTorNotRunning
}
return proxy.SOCKS5("tcp", addr, auth, d)
}
func init() {
proxy.RegisterDialerType("socks5+unix", socks5UnixProxy)
proxy.RegisterDialerType("tor-auto", torAutoProxy)
}