auto dial, tunnel limit

1 files changed, 42 insertions, 0 deletions

diff --git a/pkg/server/socket/dial.go b/pkg/server/socket/dial.go
new file mode 100644
index 0000000..818fbc6
--- /dev/null
+++ b/pkg/server/socket/dial.go
@@ -0,0 +1,42 @@
+package socket
+
+import (
+	"fmt"
+	"net"
+	"strings"
+	"tunnel/pkg/server/env"
+)
+
+type dialSocket struct {
+	proto, addr string
+}
+
+func newDialSocket(proto, addr string) (S, error) {
+	switch proto {
+	case "tcp", "udp":
+		if !strings.Contains(addr, ":") {
+			addr = "localhost:" + addr
+		}
+	}
+
+	return &dialSocket{proto: proto, addr: addr}, nil
+}
+
+func (s *dialSocket) String() string {
+	return fmt.Sprintf("%s/%s", s.proto, s.addr)
+}
+
+func (s *dialSocket) Open(env.Env) (Channel, error) {
+	conn, err := net.Dial(s.proto, s.addr)
+	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 *dialSocket) Close() {
+}