add new env DIAL_TIMEOUT to set dial timeout

.gitignore

@@ -0,0 +1,2 @@
+go.sum
+/*.exe

Dockerfile

@@ -1,11 +1,11 @@
-ARG GOLANG_VERSION="1.12"
+ARG GOLANG_VERSION="1.14.2"
 
 FROM golang:$GOLANG_VERSION-alpine as builder
-WORKDIR /go/src/github.com/olebedev/socks5
+WORKDIR /go/src/github.com/serjs/socks5-server
 COPY . .
-RUN CGO_ENABLED=0 GOOS=linux go build -a -installsuffix cgo -ldflags '-s' -o ./socks5
+RUN CGO_ENABLED=0 GOOS=linux go build --ldflags "-s -w" -a -installsuffix cgo -o socks5 .
 
 FROM scratch
 COPY --from=builder /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/
-COPY --from=builder /go/src/github.com/olebedev/socks5/socks5 /
+COPY --from=builder /go/src/github.com/serjs/socks5/socks5 /
 ENTRYPOINT ["/socks5"]

README.md

@@ -18,6 +18,7 @@ Leave `PROXY_USER` and `PROXY_PASSWORD` empty for skip authentication options wh
 |PROXY_USER|String|EMPTY|Set proxy user (also required existed PROXY_PASS)|
 |PROXY_PASSWORD|String|EMPTY|Set proxy password for auth, used with PROXY_USER|
 |PROXY_PORT|String|1080|Set listen port for application inside docker container|
+|DIAL_TIMEOUT|String|3s|Set dial connect timeout,default 3s|
 
 ## Test running service
 

go.mod

@@ -0,0 +1,9 @@
+module github.com/serjs/socks5-server
+
+go 1.13
+
+require (
+	github.com/armon/go-socks5 v0.0.0-20160902184237-e75332964ef5
+	github.com/caarlos0/env v3.3.1-0.20180521112546-3e0f30cbf50b+incompatible
+	golang.org/x/net v0.0.0-20200506145744-7e3656a0809f // indirect
+)

server.go

@@ -1,17 +1,21 @@
 package main
 
 import (
+	"context"
 	"log"
+	"net"
 	"os"
+	"time"
 
 	"github.com/armon/go-socks5"
 	"github.com/caarlos0/env"
 )
 
 type params struct {
-	User     string `env:"PROXY_USER" envDefault:""`
-	Password string `env:"PROXY_PASSWORD" envDefault:""`
-	Port     string `env:"PROXY_PORT" envDefault:"1080"`
+	User     string        `env:"PROXY_USER" envDefault:""`
+	Password string        `env:"PROXY_PASSWORD" envDefault:""`
+	Port     string        `env:"PROXY_PORT" envDefault:"1080"`
+	Timeout  time.Duration `env:"DIAL_TIMEOUT" envDefault:"3s"`
 }
 
 func main() {
@@ -25,6 +29,12 @@ func main() {
 	//Initialize socks5 config
 	socsk5conf := &socks5.Config{
 		Logger: log.New(os.Stdout, "", log.LstdFlags),
+		Dial: func(ctx context.Context, network, addr string) (net.Conn, error) {
+			d := net.Dialer{
+				Timeout: cfg.Timeout,
+			}
+			return d.DialContext(ctx, network, addr)
+		},
 	}
 
 	if cfg.User+cfg.Password != "" {