From 76012e39e9201b9dfcbbad546c7c2b38512903cb Mon Sep 17 00:00:00 2001
From: Mikhail Osipov <mike.osipov@gmail.com>
Date: Wed, 11 Mar 2020 10:49:15 +0300
Subject: auto dial, tunnel limit

---
 pkg/server/socket/listen.go | 49 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 pkg/server/socket/listen.go

(limited to 'pkg/server/socket/listen.go')

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()
+}
-- 
cgit v1.2.3-70-g09d2