From d31cedf9db419f6a7e92b768578147d6aa4c0f4c Mon Sep 17 00:00:00 2001
From: Mikhail Osipov <mike.osipov@gmail.com>
Date: Thu, 1 Oct 2020 13:04:02 +0300
Subject: dial with timeout

---
 pkg/server/socket/dial.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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

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