1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-09-19 02:16:11 -04:00
v2fly/common/net/destination.go

117 lines
2.5 KiB
Go
Raw Normal View History

package net
2015-09-21 06:15:25 -04:00
// Destination represents a network destination including address and protocol (tcp / udp).
2015-09-20 12:22:29 -04:00
type Destination interface {
2015-09-21 06:15:25 -04:00
Network() string // Protocol of communication (tcp / udp)
Address() Address // Address of destination
2015-12-16 17:53:38 -05:00
Port() Port
String() string // String representation of the destination
NetAddr() string
Equals(Destination) bool
2015-09-21 06:15:25 -04:00
IsTCP() bool // True if destination is reachable via TCP
IsUDP() bool // True if destination is reachable via UDP
2015-09-20 12:22:29 -04:00
}
2015-12-16 17:53:38 -05:00
// TCPDestination creates a TCP destination with given address
func TCPDestination(address Address, port Port) Destination {
return &tcpDestination{address: address, port: port}
2015-09-20 12:22:29 -04:00
}
2015-12-16 17:53:38 -05:00
// UDPDestination creates a UDP destination with given address
func UDPDestination(address Address, port Port) Destination {
return &udpDestination{address: address, port: port}
2015-09-20 12:22:29 -04:00
}
2015-12-16 17:53:38 -05:00
type tcpDestination struct {
address Address
2015-12-16 17:53:38 -05:00
port Port
}
2015-12-16 17:53:38 -05:00
func (dest *tcpDestination) Network() string {
2015-09-20 12:22:29 -04:00
return "tcp"
}
2015-12-16 17:53:38 -05:00
func (dest *tcpDestination) Address() Address {
2015-09-20 12:22:29 -04:00
return dest.address
}
2015-12-16 17:53:38 -05:00
func (dest *tcpDestination) NetAddr() string {
return dest.address.String() + ":" + dest.port.String()
}
2015-12-16 17:53:38 -05:00
func (dest *tcpDestination) String() string {
return "tcp:" + dest.NetAddr()
}
func (dest *tcpDestination) IsTCP() bool {
2015-09-20 12:22:29 -04:00
return true
}
2015-12-16 17:53:38 -05:00
func (dest *tcpDestination) IsUDP() bool {
2015-09-20 12:22:29 -04:00
return false
}
2015-12-16 17:53:38 -05:00
func (dest *tcpDestination) Port() Port {
return dest.port
}
func (dest *tcpDestination) Equals(another Destination) bool {
2016-01-21 07:05:28 -05:00
if dest == nil && another == nil {
return true
}
if dest == nil || another == nil {
return false
}
if !another.IsTCP() {
return false
}
return dest.Port() == another.Port() && dest.Address().Equals(another.Address())
}
2015-12-16 17:53:38 -05:00
type udpDestination struct {
2015-09-20 12:22:29 -04:00
address Address
2015-12-16 17:53:38 -05:00
port Port
2015-09-20 12:22:29 -04:00
}
2015-12-16 17:53:38 -05:00
func (dest *udpDestination) Network() string {
2015-09-20 12:22:29 -04:00
return "udp"
}
2015-12-16 17:53:38 -05:00
func (dest *udpDestination) Address() Address {
return dest.address
}
2015-12-16 17:53:38 -05:00
func (dest *udpDestination) NetAddr() string {
return dest.address.String() + ":" + dest.port.String()
2015-09-20 12:22:29 -04:00
}
2015-12-16 17:53:38 -05:00
func (dest *udpDestination) String() string {
return "udp:" + dest.NetAddr()
}
func (dest *udpDestination) IsTCP() bool {
2015-09-20 12:22:29 -04:00
return false
}
2015-12-16 17:53:38 -05:00
func (dest *udpDestination) IsUDP() bool {
2015-09-20 12:22:29 -04:00
return true
}
2015-12-16 17:53:38 -05:00
func (dest *udpDestination) Port() Port {
return dest.port
}
func (dest *udpDestination) Equals(another Destination) bool {
2016-01-21 07:05:28 -05:00
if dest == nil && another == nil {
return true
}
if dest == nil || another == nil {
return false
}
if !another.IsUDP() {
return false
}
return dest.Port() == another.Port() && dest.Address().Equals(another.Address())
}