add pkgs from core

listener/rtcp/listener.go

@@ -0,0 +1,102 @@
+package rtcp
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/core/chain"
+	"github.com/go-gost/core/connector"
+	"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("rtcp", NewListener)
+}
+
+type rtcpListener struct {
+	laddr   net.Addr
+	ln      net.Listener
+	md      metadata
+	router  *chain.Router
+	logger  logger.Logger
+	closed  chan struct{}
+	options listener.Options
+}
+
+func NewListener(opts ...listener.Option) listener.Listener {
+	options := listener.Options{}
+	for _, opt := range opts {
+		opt(&options)
+	}
+	return &rtcpListener{
+		closed:  make(chan struct{}),
+		logger:  options.Logger,
+		options: options,
+	}
+}
+
+func (l *rtcpListener) Init(md md.Metadata) (err error) {
+	if err = l.parseMetadata(md); err != nil {
+		return
+	}
+
+	laddr, err := net.ResolveTCPAddr("tcp", l.options.Addr)
+	if err != nil {
+		return
+	}
+
+	l.laddr = laddr
+	l.router = (&chain.Router{}).
+		WithChain(l.options.Chain).
+		WithLogger(l.logger)
+
+	return
+}
+
+func (l *rtcpListener) Accept() (conn net.Conn, err error) {
+	select {
+	case <-l.closed:
+		return nil, net.ErrClosed
+	default:
+	}
+
+	if l.ln == nil {
+		l.ln, err = l.router.Bind(
+			context.Background(), "tcp", l.laddr.String(),
+			connector.MuxBindOption(true),
+		)
+		if err != nil {
+			return nil, listener.NewAcceptError(err)
+		}
+		l.ln = metrics.WrapListener(l.options.Service, l.ln)
+	}
+	conn, err = l.ln.Accept()
+	if err != nil {
+		l.ln.Close()
+		l.ln = nil
+		return nil, listener.NewAcceptError(err)
+	}
+	return
+}
+
+func (l *rtcpListener) Addr() net.Addr {
+	return l.laddr
+}
+
+func (l *rtcpListener) Close() error {
+	select {
+	case <-l.closed:
+	default:
+		close(l.closed)
+		if l.ln != nil {
+			l.ln.Close()
+			l.ln = nil
+		}
+	}
+
+	return nil
+}

listener/rtcp/metadata.go

@@ -0,0 +1,11 @@
+package rtcp
+
+import (
+	mdata "github.com/go-gost/core/metadata"
+)
+
+type metadata struct{}
+
+func (l *rtcpListener) parseMetadata(md mdata.Metadata) (err error) {
+	return
+}