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

@@ -3,7 +3,6 @@ package tunnel
import (
"io"
"net"
"strconv"
"sync"
"time"
@@ -45,10 +44,7 @@ func handleTCP(localConn core.TCPConn) {
metadata.MidIP = dialerAddr.IP
metadata.MidPort = uint16(dialerAddr.Port)
} else { /* fallback */
ip, p, _ := net.SplitHostPort(targetConn.LocalAddr().String())
port, _ := strconv.ParseUint(p, 10, 16)
metadata.MidIP = net.ParseIP(ip)
metadata.MidPort = uint16(port)
metadata.MidIP, metadata.MidPort = parseAddr(targetConn.LocalAddr().String())
}
targetConn = newTCPTracker(targetConn, metadata)