clash/listener/socks/udp.go

77 lines
1.5 KiB
Go
Raw Normal View History

package socks
import (
"net"
2021-06-10 06:05:56 +00:00
"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"
2021-05-13 14:18:49 +00:00
"github.com/Dreamacro/clash/transport/socks5"
)
2021-06-13 09:23:10 +00:00
type UDPListener struct {
packetConn net.PacketConn
closed bool
}
2021-06-13 09:23:10 +00:00
func NewUDP(addr string, in chan<- *inbound.PacketAdapter) (*UDPListener, error) {
l, err := net.ListenPacket("udp", addr)
if err != nil {
return nil, err
}
2021-06-13 09:23:10 +00:00
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 {
2020-04-24 16:30:40 +00:00
buf := pool.Get(pool.RelayBufferSize)
n, remoteAddr, err := l.ReadFrom(buf)
if err != nil {
2020-04-24 16:30:40 +00:00
pool.Put(buf)
if sl.closed {
break
}
continue
}
2021-06-13 09:23:10 +00:00
handleSocksUDP(l, in, buf[:n], remoteAddr)
}
}()
return sl, nil
}
2021-06-13 09:23:10 +00:00
func (l *UDPListener) Close() error {
l.closed = true
return l.packetConn.Close()
}
2021-06-13 09:23:10 +00:00
func (l *UDPListener) Address() string {
return l.packetConn.LocalAddr().String()
}
2021-06-13 09:23:10 +00:00
func handleSocksUDP(pc net.PacketConn, in chan<- *inbound.PacketAdapter, buf []byte, addr net.Addr) {
2019-10-11 12:11:18 +00:00
target, payload, err := socks5.DecodeUDPPacket(buf)
if err != nil {
2019-10-11 12:11:18 +00:00
// Unresolved UDP packet, return buffer to the pool
2020-04-24 16:30:40 +00:00
pool.Put(buf)
return
}
packet := &packet{
pc: pc,
rAddr: addr,
payload: payload,
bufRef: buf,
2019-10-11 12:11:18 +00:00
}
2021-06-13 09:23:10 +00:00
select {
2021-07-21 15:08:52 +00:00
case in <- inbound.NewPacket(target, packet, C.SOCKS):
2021-06-13 09:23:10 +00:00
default:
}
}