58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"golang.org/x/net/proxy"
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
CannotResolveLocalTCPAddressError = Error("CannotResolveLocalTCPAddressError")
|
|
CannotDialLocalTCPAddressError = Error("CannotDialLocalTCPAddressError")
|
|
CannotDialRicochetAddressError = Error("CannotDialRicochetAddressError")
|
|
)
|
|
|
|
// NetworkResolver allows a client to resolve various hostnames to connections
|
|
// The supported types are onions address are:
|
|
// * ricochet:jlq67qzo6s4yp3sp
|
|
// * jlq67qzo6s4yp3sp
|
|
// * 127.0.0.1:55555|jlq67qzo6s4yp3sp - Localhost Connection
|
|
type NetworkResolver struct {
|
|
}
|
|
|
|
// Resolve takes a hostname and returns a net.Conn to the derived endpoint
|
|
func (nr *NetworkResolver) Resolve(hostname string) (net.Conn, string, error) {
|
|
if strings.HasPrefix(hostname, "127.0.0.1") {
|
|
addrParts := strings.Split(hostname, "|")
|
|
tcpAddr, err := net.ResolveTCPAddr("tcp", addrParts[0])
|
|
if err != nil {
|
|
return nil, "", CannotResolveLocalTCPAddressError
|
|
}
|
|
conn, err := net.DialTCP("tcp", nil, tcpAddr)
|
|
if err != nil {
|
|
return nil, "", CannotDialLocalTCPAddressError
|
|
}
|
|
|
|
// return just the onion address, not the local override for the hostname
|
|
return conn, addrParts[1], nil
|
|
}
|
|
|
|
resolvedHostname := hostname
|
|
if strings.HasPrefix(hostname, "ricochet:") {
|
|
addrParts := strings.Split(hostname, ":")
|
|
resolvedHostname = addrParts[1]
|
|
}
|
|
|
|
torDialer, err := proxy.SOCKS5("tcp", "127.0.0.1:9050", nil, proxy.Direct)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
conn, err := torDialer.Dial("tcp", resolvedHostname+".onion:9878")
|
|
if err != nil {
|
|
return nil, "", CannotDialRicochetAddressError
|
|
}
|
|
|
|
return conn, resolvedHostname, nil
|
|
}
|