1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-19 02:46:33 -05:00
v2fly/transport/internet/udp/hub_linux.go

40 lines
961 B
Go
Raw Normal View History

2016-08-15 11:44:46 -04:00
// +build linux
package udp
import (
2016-08-19 06:58:26 -04:00
"net"
2016-08-15 11:44:46 -04:00
"syscall"
2016-08-20 14:55:45 -04:00
v2net "v2ray.com/core/common/net"
2016-08-15 11:44:46 -04:00
)
func SetOriginalDestOptions(fd int) error {
if err := syscall.SetsockoptInt(fd, syscall.SOL_IP, syscall.IP_TRANSPARENT, 1); err != nil {
return err
}
if err := syscall.SetsockoptInt(fd, syscall.SOL_IP, syscall.IP_RECVORIGDSTADDR, 1); err != nil {
return err
}
return nil
}
func RetrieveOriginalDest(oob []byte) v2net.Destination {
msgs, err := syscall.ParseSocketControlMessage(oob)
if err != nil {
2016-09-20 05:53:05 -04:00
return v2net.Destination{}
2016-08-15 11:44:46 -04:00
}
for _, msg := range msgs {
if msg.Header.Level == syscall.SOL_IP && msg.Header.Type == syscall.IP_ORIGDSTADDR {
ip := v2net.IPAddress(msg.Data[4:8])
port := v2net.PortFromBytes(msg.Data[2:4])
return v2net.UDPDestination(ip, port)
}
}
2016-09-20 05:53:05 -04:00
return v2net.Destination{}
2016-08-15 11:44:46 -04:00
}
2016-08-19 06:58:26 -04:00
func ReadUDPMsg(conn *net.UDPConn, payload []byte, oob []byte) (int, int, int, *net.UDPAddr, error) {
return conn.ReadMsgUDP(payload, oob)
}