From 76012e39e9201b9dfcbbad546c7c2b38512903cb Mon Sep 17 00:00:00 2001 From: Mikhail Osipov Date: Wed, 11 Mar 2020 10:49:15 +0300 Subject: auto dial, tunnel limit --- pkg/server/socket/dial.go | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 pkg/server/socket/dial.go (limited to 'pkg/server/socket/dial.go') 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() { +} -- cgit v1.2.3-70-g09d2