v2ray-core/transport/internet/tcp/sockopt_linux.go

36 lines
863 B
Go
Raw Normal View History

2016-06-12 01:35:40 +02:00
// +build linux
package tcp
2016-06-12 01:35:40 +02:00
import (
"syscall"
2017-02-01 21:35:40 +01:00
"v2ray.com/core/app/log"
2017-01-14 00:27:45 +01:00
"v2ray.com/core/common/net"
2016-08-20 20:55:45 +02:00
"v2ray.com/core/transport/internet"
2016-06-12 01:35:40 +02:00
)
const SO_ORIGINAL_DST = 80
2017-01-14 00:27:45 +01:00
func GetOriginalDestination(conn internet.Connection) net.Destination {
2016-06-30 00:08:20 +02:00
tcpConn, ok := conn.(internet.SysFd)
if !ok {
2017-04-09 01:43:25 +02:00
log.Trace(newError("failed to get sys fd"))
2017-01-14 00:27:45 +01:00
return net.Destination{}
2016-06-30 00:08:20 +02:00
}
2016-06-14 22:54:08 +02:00
fd, err := tcpConn.SysFd()
2016-06-12 01:35:40 +02:00
if err != nil {
2017-04-09 01:43:25 +02:00
log.Trace(newError("failed to get original destination").Base(err))
2017-01-14 00:27:45 +01:00
return net.Destination{}
2016-06-12 01:35:40 +02:00
}
addr, err := syscall.GetsockoptIPv6Mreq(fd, syscall.IPPROTO_IP, SO_ORIGINAL_DST)
if err != nil {
2017-04-09 01:43:25 +02:00
log.Trace(newError("failed to call getsockopt").Base(err))
2017-01-14 00:27:45 +01:00
return net.Destination{}
2016-06-12 01:35:40 +02:00
}
2017-01-14 00:27:45 +01:00
ip := net.IPAddress(addr.Multiaddr[4:8])
2016-06-12 01:35:40 +02:00
port := uint16(addr.Multiaddr[2])<<8 + uint16(addr.Multiaddr[3])
2017-01-14 00:27:45 +01:00
return net.TCPDestination(ip, net.Port(port))
2016-06-12 01:35:40 +02:00
}