mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 01:57:12 -05:00
65 lines
1.1 KiB
Go
65 lines
1.1 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 {
|
|
// TODO: check IP length
|
|
return Address{
|
|
AddrTypeIP,
|
|
net.IP(ip),
|
|
"",
|
|
port}
|
|
}
|
|
|
|
func DomainAddress(domain string, port uint16) Address {
|
|
return Address{
|
|
AddrTypeDomain,
|
|
nil,
|
|
domain,
|
|
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))
|
|
}
|