add pkgs from core

listener/udp/listener.go

@@ -0,0 +1,74 @@
+package udp
+
+import (
+	"net"
+
+	"github.com/go-gost/core/common/net/udp"
+	"github.com/go-gost/core/listener"
+	"github.com/go-gost/core/logger"
+	md "github.com/go-gost/core/metadata"
+	metrics "github.com/go-gost/core/metrics/wrapper"
+	"github.com/go-gost/x/registry"
+)
+
+func init() {
+	registry.ListenerRegistry().Register("udp", NewListener)
+}
+
+type udpListener struct {
+	ln      net.Listener
+	logger  logger.Logger
+	md      metadata
+	options listener.Options
+}
+
+func NewListener(opts ...listener.Option) listener.Listener {
+	options := listener.Options{}
+	for _, opt := range opts {
+		opt(&options)
+	}
+	return &udpListener{
+		logger:  options.Logger,
+		options: options,
+	}
+}
+
+func (l *udpListener) Init(md md.Metadata) (err error) {
+	if err = l.parseMetadata(md); err != nil {
+		return
+	}
+
+	laddr, err := net.ResolveUDPAddr("udp", l.options.Addr)
+	if err != nil {
+		return
+	}
+
+	var conn net.PacketConn
+	conn, err = net.ListenUDP("udp", laddr)
+	if err != nil {
+		return
+	}
+	conn = metrics.WrapPacketConn(l.options.Service, conn)
+
+	l.ln = udp.NewListener(conn, &udp.ListenConfig{
+		Backlog:        l.md.backlog,
+		ReadQueueSize:  l.md.readQueueSize,
+		ReadBufferSize: l.md.readBufferSize,
+		KeepAlive:      l.md.keepalive,
+		TTL:            l.md.ttl,
+		Logger:         l.logger,
+	})
+	return
+}
+
+func (l *udpListener) Accept() (conn net.Conn, err error) {
+	return l.ln.Accept()
+}
+
+func (l *udpListener) Addr() net.Addr {
+	return l.ln.Addr()
+}
+
+func (l *udpListener) Close() error {
+	return l.ln.Close()
+}