dial with timeout

1 files changed, 4 insertions, 1 deletions

diff --git a/pkg/server/socket/dial.go b/pkg/server/socket/dial.go
index 18dee8f..7623084 100644
--- a/pkg/server/socket/dial.go
+++ b/pkg/server/socket/dial.go
@@ -5,9 +5,12 @@ import (
 	"log"
 	"net"
 	"strings"
+	"time"
 	"tunnel/pkg/server/env"
 )
 
+const defaultTimeout = 5 * time.Second
+
 type dialSocket struct {
 	proto, addr string
 }
@@ -30,7 +33,7 @@ func (s *dialSocket) Open(e env.Env) (Conn, error) {
 		}
 	}
 
-	conn, err := net.Dial(s.proto, addr)
+	conn, err := net.DialTimeout(s.proto, addr, defaultTimeout)
 	if err != nil {
 		return nil, err
 	}