Refactor(tunnel): using parseAddr

This commit is contained in:
xjasonlyu
2021-07-19 14:04:17 +08:00
parent e77c12156a
commit ccc35c3c1b
4 changed files with 22 additions and 17 deletions

View File

@@ -4,7 +4,6 @@ import (
"errors"
"net"
"os"
"strconv"
"time"
"github.com/xjasonlyu/tun2socks/common/pool"
@@ -88,10 +87,7 @@ func handleUDP(packet core.UDPPacket) {
metadata.MidIP = dialerAddr.IP
metadata.MidPort = uint16(dialerAddr.Port)
} else { /* fallback */
ip, p, _ := net.SplitHostPort(pc.LocalAddr().String())
port, _ := strconv.ParseUint(p, 10, 16)
metadata.MidIP = net.ParseIP(ip)
metadata.MidPort = uint16(port)
metadata.MidIP, metadata.MidPort = parseAddr(pc.LocalAddr().String())
}
pc = newUDPTracker(pc, metadata)