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()
+}

listener/udp/metadata.go

@@ -0,0 +1,55 @@
+package udp
+
+import (
+	"time"
+
+	mdata "github.com/go-gost/core/metadata"
+	mdx "github.com/go-gost/x/metadata"
+)
+
+const (
+	defaultTTL            = 5 * time.Second
+	defaultReadBufferSize = 1500
+	defaultReadQueueSize  = 128
+	defaultBacklog        = 128
+)
+
+type metadata struct {
+	readBufferSize int
+	readQueueSize  int
+	backlog        int
+	keepalive      bool
+	ttl            time.Duration
+}
+
+func (l *udpListener) parseMetadata(md mdata.Metadata) (err error) {
+	const (
+		readBufferSize = "readBufferSize"
+		readQueueSize  = "readQueueSize"
+		backlog        = "backlog"
+		keepAlive      = "keepAlive"
+		ttl            = "ttl"
+	)
+
+	l.md.ttl = mdx.GetDuration(md, ttl)
+	if l.md.ttl <= 0 {
+		l.md.ttl = defaultTTL
+	}
+	l.md.readBufferSize = mdx.GetInt(md, readBufferSize)
+	if l.md.readBufferSize <= 0 {
+		l.md.readBufferSize = defaultReadBufferSize
+	}
+
+	l.md.readQueueSize = mdx.GetInt(md, readQueueSize)
+	if l.md.readQueueSize <= 0 {
+		l.md.readQueueSize = defaultReadQueueSize
+	}
+
+	l.md.backlog = mdx.GetInt(md, backlog)
+	if l.md.backlog <= 0 {
+		l.md.backlog = defaultBacklog
+	}
+	l.md.keepalive = mdx.GetBool(md, keepAlive)
+
+	return
+}