mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-17 01:53:36 -05:00
34 lines
903 B
Go
34 lines
903 B
Go
// +build linux
|
|
|
|
package udp
|
|
|
|
import (
|
|
"syscall"
|
|
|
|
"golang.org/x/sys/unix"
|
|
"v2ray.com/core/common/net"
|
|
)
|
|
|
|
func RetrieveOriginalDest(oob []byte) net.Destination {
|
|
msgs, err := syscall.ParseSocketControlMessage(oob)
|
|
if err != nil {
|
|
return net.Destination{}
|
|
}
|
|
for _, msg := range msgs {
|
|
if msg.Header.Level == syscall.SOL_IP && msg.Header.Type == syscall.IP_RECVORIGDSTADDR {
|
|
ip := net.IPAddress(msg.Data[4:8])
|
|
port := net.PortFromBytes(msg.Data[2:4])
|
|
return net.UDPDestination(ip, port)
|
|
} else if msg.Header.Level == syscall.SOL_IPV6 && msg.Header.Type == unix.IPV6_RECVORIGDSTADDR {
|
|
ip := net.IPAddress(msg.Data[8:24])
|
|
port := net.PortFromBytes(msg.Data[2:4])
|
|
return net.UDPDestination(ip, port)
|
|
}
|
|
}
|
|
return net.Destination{}
|
|
}
|
|
|
|
func ReadUDPMsg(conn *net.UDPConn, payload []byte, oob []byte) (int, int, int, *net.UDPAddr, error) {
|
|
return conn.ReadMsgUDP(payload, oob)
|
|
}
|