mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
778 B
45 lines
778 B
9 years ago
|
package dialer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"math/rand"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
ErrInvalidHost = errors.New("Invalid Host.")
|
||
|
|
)
|
||
|
|
|
||
|
|
func Dial(dest v2net.Destination) (net.Conn, error) {
|
||
|
|
var ip net.IP
|
||
|
|
if dest.Address().IsIPv4() || dest.Address().IsIPv6() {
|
||
|
|
ip = dest.Address().IP()
|
||
|
|
} else {
|
||
|
|
ips, err := net.LookupIP(dest.Address().Domain())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if len(ips) == 0 {
|
||
|
|
return nil, ErrInvalidHost
|
||
|
|
}
|
||
|
|
if len(ips) == 1 {
|
||
|
|
ip = ips[0]
|
||
|
|
} else {
|
||
|
|
ip = ips[rand.Intn(len(ips))]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if dest.IsTCP() {
|
||
|
|
return net.DialTCP("tcp", nil, &net.TCPAddr{
|
||
|
|
IP: ip,
|
||
|
|
Port: int(dest.Port()),
|
||
|
|
})
|
||
|
|
} else {
|
||
|
|
return net.DialUDP("udp", nil, &net.UDPAddr{
|
||
|
|
IP: ip,
|
||
|
|
Port: int(dest.Port()),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|