mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-21 01:27:03 -05:00
69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
package net
|
|
|
|
import (
|
|
"net"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
AddrTypeIP = byte(0x01)
|
|
AddrTypeDomain = byte(0x03)
|
|
)
|
|
|
|
type Address struct {
|
|
Type byte
|
|
IP net.IP
|
|
Domain string
|
|
Port uint16
|
|
}
|
|
|
|
func IPAddress(ip []byte, port uint16) Address {
|
|
ipCopy := make([]byte, len(ip))
|
|
copy(ipCopy, ip)
|
|
// TODO: check IP length
|
|
return Address{
|
|
Type: AddrTypeIP,
|
|
IP: net.IP(ipCopy),
|
|
Domain: "",
|
|
Port: port,
|
|
}
|
|
}
|
|
|
|
func DomainAddress(domain string, port uint16) Address {
|
|
return Address{
|
|
Type: AddrTypeDomain,
|
|
IP: nil,
|
|
Domain: domain,
|
|
Port: port,
|
|
}
|
|
}
|
|
|
|
func (addr Address) IsIPv4() bool {
|
|
return addr.Type == AddrTypeIP && len(addr.IP) == net.IPv4len
|
|
}
|
|
|
|
func (addr Address) IsIPv6() bool {
|
|
return addr.Type == AddrTypeIP && len(addr.IP) == net.IPv6len
|
|
}
|
|
|
|
func (addr Address) IsDomain() bool {
|
|
return addr.Type == AddrTypeDomain
|
|
}
|
|
|
|
func (addr Address) String() string {
|
|
var host string
|
|
switch addr.Type {
|
|
case AddrTypeIP:
|
|
host = addr.IP.String()
|
|
if len(addr.IP) == net.IPv6len {
|
|
host = "[" + host + "]"
|
|
}
|
|
|
|
case AddrTypeDomain:
|
|
host = addr.Domain
|
|
default:
|
|
panic("Unknown Address Type " + strconv.Itoa(int(addr.Type)))
|
|
}
|
|
return host + ":" + strconv.Itoa(int(addr.Port))
|
|
}
|