package/tun2socks
1
0
Fork 0
mirror of https://github.com/xjasonlyu/tun2socks.git synced 2025-10-22 00:09:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore cleanup

Browse Source
This commit is contained in:
Jason
2019-08-12 15:06:14 +08:00
parent e3b1ff47bd
commit 2254731d3b
3 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
proxy/direct/udp.go
View File

@@ -40,7 +40,7 @@ func (h *udpHandler) fetchUDPInput(conn core.UDPConn, pc *net.UDPConn) {
return
}
if _, err = conn.WriteFrom(buf[:n], addr); err != nil {
if _, err := conn.WriteFrom(buf[:n], addr); err != nil {
log.Warnf("failed to write UDP data to TUN: %v", err)
return
}
@@ -54,8 +54,11 @@ func (h *udpHandler) Connect(conn core.UDPConn, target *net.UDPAddr) error {
log.Errorf("failed to bind udp address")
return err
}
h.remoteUDPConnMap.Store(conn, pc)
go h.fetchUDPInput(conn, pc)
log.Infof("new proxy connection for target: %s:%s", target.Network(), target.String())
return nil
}
@@ -80,9 +83,10 @@ func (h *udpHandler) ReceiveTo(conn core.UDPConn, data []byte, addr *net.UDPAddr
}
func (h *udpHandler) Close(conn core.UDPConn) {
conn.Close()
if pc, ok := h.remoteUDPConnMap.Load(conn); ok {
pc.(*net.UDPConn).Close()
h.remoteUDPConnMap.Delete(conn)
}
conn.Close()
}