package socks import ( "net" "github.com/Dreamacro/clash/adapter/inbound" "github.com/Dreamacro/clash/common/pool" "github.com/Dreamacro/clash/common/sockopt" C "github.com/Dreamacro/clash/constant" "github.com/Dreamacro/clash/log" "github.com/Dreamacro/clash/transport/socks5" ) type UDPListener struct { packetConn net.PacketConn closed bool } func NewUDP(addr string, in chan<- *inbound.PacketAdapter) (*UDPListener, error) { l, err := net.ListenPacket("udp", addr) if err != nil { return nil, err } if err := sockopt.UDPReuseaddr(l.(*net.UDPConn)); err != nil { log.Warnln("Failed to Reuse UDP Address: %s", err) } sl := &UDPListener{ packetConn: l, } go func() { for { buf := pool.Get(pool.RelayBufferSize) n, remoteAddr, err := l.ReadFrom(buf) if err != nil { pool.Put(buf) if sl.closed { break } continue } handleSocksUDP(l, in, buf[:n], remoteAddr) } }() return sl, nil } func (l *UDPListener) Close() error { l.closed = true return l.packetConn.Close() } func (l *UDPListener) Address() string { return l.packetConn.LocalAddr().String() } func handleSocksUDP(pc net.PacketConn, in chan<- *inbound.PacketAdapter, buf []byte, addr net.Addr) { target, payload, err := socks5.DecodeUDPPacket(buf) if err != nil { // Unresolved UDP packet, return buffer to the pool pool.Put(buf) return } packet := &packet{ pc: pc, rAddr: addr, payload: payload, bufRef: buf, } select { case in <- inbound.NewPacket(target, packet, C.SOCKS): default: } }