auto dial, tunnel limit

1 files changed, 49 insertions, 0 deletions

diff --git a/pkg/server/socket/listen.go b/pkg/server/socket/listen.go
new file mode 100644
index 0000000..c328945
--- /dev/null
+++ b/pkg/server/socket/listen.go
@@ -0,0 +1,49 @@
+package socket
+
+import (
+	"fmt"
+	"net"
+	"strings"
+	"tunnel/pkg/server/env"
+)
+
+func newListenSocket(proto, addr string) (S, error) {
+	if proto == "tcp" {
+		if !strings.Contains(addr, ":") {
+			addr = ":" + addr
+		}
+	}
+
+	listen, err := net.Listen(proto, addr)
+	if err != nil {
+		return nil, err
+	}
+
+	s := &listenSocket{
+		proto:  proto,
+		addr:   addr,
+		listen: listen,
+	}
+
+	return s, nil
+}
+
+func (s *listenSocket) Open(env.Env) (Channel, error) {
+	conn, err := s.listen.Accept()
+	if err != nil {
+		return nil, err
+	}
+
+	addr := conn.RemoteAddr()
+	info := fmt.Sprintf("<%s/%s", addr.Network(), addr)
+
+	return exported{info, newConn(conn)}, nil
+}
+
+func (s *listenSocket) String() string {
+	return fmt.Sprintf("%s/%s,listen", s.proto, s.addr)
+}
+
+func (s *listenSocket) Close() {
+	s.listen.Close()
+}