mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-06 03:54:22 -04:00
132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package net
|
|
|
|
import (
|
|
"net"
|
|
)
|
|
|
|
// Destination represents a network destination including address and protocol (tcp / udp).
|
|
type Destination interface {
|
|
Network() Network // Protocol of communication (tcp / udp)
|
|
Address() Address // Address of destination
|
|
Port() Port
|
|
String() string // String representation of the destination
|
|
NetAddr() string
|
|
Equals(Destination) bool
|
|
|
|
IsTCP() bool // True if destination is reachable via TCP
|
|
IsUDP() bool // True if destination is reachable via UDP
|
|
}
|
|
|
|
func DestinationFromAddr(addr net.Addr) Destination {
|
|
switch addr := addr.(type) {
|
|
case *net.TCPAddr:
|
|
return TCPDestination(IPAddress(addr.IP), Port(addr.Port))
|
|
case *net.UDPAddr:
|
|
return UDPDestination(IPAddress(addr.IP), Port(addr.Port))
|
|
default:
|
|
panic("Unknown address type.")
|
|
}
|
|
}
|
|
|
|
// TCPDestination creates a TCP destination with given address
|
|
func TCPDestination(address Address, port Port) Destination {
|
|
return &tcpDestination{address: address, port: port}
|
|
}
|
|
|
|
// UDPDestination creates a UDP destination with given address
|
|
func UDPDestination(address Address, port Port) Destination {
|
|
return &udpDestination{address: address, port: port}
|
|
}
|
|
|
|
type tcpDestination struct {
|
|
address Address
|
|
port Port
|
|
}
|
|
|
|
func (dest *tcpDestination) Network() Network {
|
|
return TCPNetwork
|
|
}
|
|
|
|
func (dest *tcpDestination) Address() Address {
|
|
return dest.address
|
|
}
|
|
|
|
func (dest *tcpDestination) NetAddr() string {
|
|
return dest.address.String() + ":" + dest.port.String()
|
|
}
|
|
|
|
func (dest *tcpDestination) String() string {
|
|
return "tcp:" + dest.NetAddr()
|
|
}
|
|
|
|
func (dest *tcpDestination) IsTCP() bool {
|
|
return true
|
|
}
|
|
|
|
func (dest *tcpDestination) IsUDP() bool {
|
|
return false
|
|
}
|
|
|
|
func (dest *tcpDestination) Port() Port {
|
|
return dest.port
|
|
}
|
|
|
|
func (dest *tcpDestination) Equals(another Destination) bool {
|
|
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())
|
|
}
|
|
|
|
type udpDestination struct {
|
|
address Address
|
|
port Port
|
|
}
|
|
|
|
func (dest *udpDestination) Network() Network {
|
|
return UDPNetwork
|
|
}
|
|
|
|
func (dest *udpDestination) Address() Address {
|
|
return dest.address
|
|
}
|
|
|
|
func (dest *udpDestination) NetAddr() string {
|
|
return dest.address.String() + ":" + dest.port.String()
|
|
}
|
|
|
|
func (dest *udpDestination) String() string {
|
|
return "udp:" + dest.NetAddr()
|
|
}
|
|
|
|
func (dest *udpDestination) IsTCP() bool {
|
|
return false
|
|
}
|
|
|
|
func (dest *udpDestination) IsUDP() bool {
|
|
return true
|
|
}
|
|
|
|
func (dest *udpDestination) Port() Port {
|
|
return dest.port
|
|
}
|
|
|
|
func (dest *udpDestination) Equals(another Destination) bool {
|
|
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())
|
|
}
|