1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-26 09:25:23 +00:00

fix test break

This commit is contained in:
Darien Raymond 2017-04-18 12:35:54 +02:00
parent 14c75e5035
commit cee2fe4b6f
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
3 changed files with 10 additions and 10 deletions

View File

@ -42,7 +42,10 @@ type tcpWorker struct {
func (w *tcpWorker) callback(conn internet.Connection) {
ctx, cancel := context.WithCancel(w.ctx)
if w.recvOrigDest {
dest := tcp.GetOriginalDestination(conn)
dest, err := tcp.GetOriginalDestination(conn)
if err != nil {
log.Trace(newError("failed to get original destination").Base(err))
}
if dest.IsValid() {
ctx = proxy.ContextWithOriginalTarget(ctx, dest)
}

View File

@ -6,7 +6,6 @@ import (
"net"
"syscall"
"v2ray.com/core/app/log"
v2net "v2ray.com/core/common/net"
"v2ray.com/core/transport/internet"
"v2ray.com/core/transport/internet/internal"
@ -14,19 +13,17 @@ import (
const SO_ORIGINAL_DST = 80
func GetOriginalDestination(conn internet.Connection) v2net.Destination {
func GetOriginalDestination(conn internet.Connection) (v2net.Destination, error) {
fd, err := internal.GetSysFd(conn.(net.Conn))
if err != nil {
log.Trace(newError("failed to get original destination").Base(err))
return v2net.Destination{}
return v2net.Destination{}, newError("failed to get original destination").Base(err)
}
addr, err := syscall.GetsockoptIPv6Mreq(fd, syscall.IPPROTO_IP, SO_ORIGINAL_DST)
if err != nil {
log.Trace(newError("failed to call getsockopt").Base(err))
return v2net.Destination{}
return v2net.Destination{}, newError("failed to call getsockopt").Base(err)
}
ip := v2net.IPAddress(addr.Multiaddr[4:8])
port := uint16(addr.Multiaddr[2])<<8 + uint16(addr.Multiaddr[3])
return v2net.TCPDestination(ip, v2net.Port(port))
return v2net.TCPDestination(ip, v2net.Port(port)), nil
}

View File

@ -7,6 +7,6 @@ import (
"v2ray.com/core/transport/internet"
)
func GetOriginalDestination(conn internet.Connection) net.Destination {
return net.Destination{}
func GetOriginalDestination(conn internet.Connection) (net.Destination, error) {
return net.Destination{}, nil
}