github/tun2socks
4
0
Fork 0
mirror of https://github.com/xjasonlyu/tun2socks.git synced 2024-12-30 02:37:01 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Minor: UDP unidirectionalPacketStream

Browse Source
This commit is contained in:
xjasonlyu
2023-04-03 19:13:03 +08:00
parent 4cc02c822c
commit 61a9d26815
1 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tunnel/udp.go
+9 -13
View File
@@ -60,23 +60,19 @@ func pipePacket(origin, remote net.PacketConn, to net.Addr) {
wg := sync.WaitGroup{}
wg.Add(2)
go func() {
defer wg.Done()
if err := copyPacketData(remote, origin, to, _udpSessionTimeout); err != nil {
log.Debugf("[UDP] copy data for origin->remote: %v", err)
}
}()
go func() {
defer wg.Done()
if err := copyPacketData(origin, remote, nil, _udpSessionTimeout); err != nil {
log.Debugf("[UDP] copy data for remote->origin: %v", err)
}
}()
go unidirectionalPacketStream(remote, origin, to, "origin->remote", &wg)
go unidirectionalPacketStream(origin, remote, nil, "remote->origin", &wg)
wg.Wait()
}
func unidirectionalPacketStream(dst, src net.PacketConn, to net.Addr, dir string, wg *sync.WaitGroup) {
defer wg.Done()
if err := copyPacketData(dst, src, to, _udpSessionTimeout); err != nil {
log.Debugf("[UDP] copy data for %s: %v", dir, err)
}
}
func copyPacketData(dst, src net.PacketConn, to net.Addr, timeout time.Duration) error {
buf := pool.Get(pool.MaxSegmentSize)
defer pool.Put(buf)