git add

Hunter0x7c7

2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
//go:build linux 
// +build linux 
 
package udp 
 
import ( 
    "syscall" 
 
    "golang.org/x/sys/unix" 
 
    "github.com/v2fly/v2ray-core/v5/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) 
}