diff --git a/cmd/zrok/testWebsocket.go b/cmd/zrok/testWebsocket.go index 57789e95f..0dd3e6b53 100644 --- a/cmd/zrok/testWebsocket.go +++ b/cmd/zrok/testWebsocket.go @@ -76,7 +76,7 @@ func (cmd *testWebsocketCommand) run(_ *cobra.Command, args []string) { } dial := func(_ context.Context, _, addr string) (net.Conn, error) { service := strings.Split(addr, ":")[0] - return zitiContext.Dial(service) + return zitiContext.DialWithOptions(service, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}) } zitiTransport := http.DefaultTransport.(*http.Transport).Clone() diff --git a/endpoints/proxy/frontend.go b/endpoints/proxy/frontend.go index f868f0aa5..ce5494c54 100644 --- a/endpoints/proxy/frontend.go +++ b/endpoints/proxy/frontend.go @@ -91,7 +91,7 @@ type zitiDialContext struct { } func (zdc *zitiDialContext) Dial(_ context.Context, _ string, addr string) (net.Conn, error) { - conn, err := zdc.ctx.Dial(zdc.shrToken) + conn, err := zdc.ctx.DialWithOptions(zdc.shrToken, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}) if err != nil { return conn, err } diff --git a/endpoints/publicProxy/http.go b/endpoints/publicProxy/http.go index 2cd52ebad..f40093d28 100644 --- a/endpoints/publicProxy/http.go +++ b/endpoints/publicProxy/http.go @@ -95,7 +95,7 @@ type zitiDialContext struct { func (c *zitiDialContext) Dial(_ context.Context, _ string, addr string) (net.Conn, error) { shrToken := strings.Split(addr, ":")[0] // ignore :port (we get passed 'host:port') - conn, err := c.ctx.Dial(shrToken) + conn, err := c.ctx.DialWithOptions(shrToken, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}) if err != nil { return conn, err } diff --git a/endpoints/tcpTunnel/frontend.go b/endpoints/tcpTunnel/frontend.go index 662f74fb8..8695a20f6 100644 --- a/endpoints/tcpTunnel/frontend.go +++ b/endpoints/tcpTunnel/frontend.go @@ -69,7 +69,7 @@ func (f *Frontend) Run() error { } func (f *Frontend) accept(conn net.Conn) { - if zConn, err := f.zCtx.Dial(f.cfg.ShrToken); err == nil { + if zConn, err := f.zCtx.DialWithOptions(f.cfg.ShrToken, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}); err == nil { go endpoints.TXer(conn, zConn) go endpoints.TXer(zConn, conn) if f.cfg.RequestsChan != nil { diff --git a/endpoints/udpTunnel/frontend.go b/endpoints/udpTunnel/frontend.go index aa44d0980..9df61b640 100644 --- a/endpoints/udpTunnel/frontend.go +++ b/endpoints/udpTunnel/frontend.go @@ -148,7 +148,7 @@ func (f *Frontend) Run() error { _ = clt.zitiConn.Close() } } else { - zitiConn, err := f.zCtx.Dial(f.cfg.ShrToken) + zitiConn, err := f.zCtx.DialWithOptions(f.cfg.ShrToken, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}) if err != nil { logrus.Errorf("error dialing '%v': %v", f.cfg.ShrToken, err) continue diff --git a/endpoints/vpn/frontend.go b/endpoints/vpn/frontend.go index cab11df29..3ad26436d 100644 --- a/endpoints/vpn/frontend.go +++ b/endpoints/vpn/frontend.go @@ -45,7 +45,7 @@ func NewFrontend(cfg *FrontendConfig) (*Frontend, error) { return nil, errors.Wrap(err, "error loading ziti context") } - zConn, err := zCtx.Dial(cfg.ShrToken) + zConn, err := zCtx.DialWithOptions(cfg.ShrToken, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}) if err != nil { zCtx.Close() return nil, errors.Wrap(err, "error connecting to ziti") diff --git a/sdk/golang/sdk/dialer.go b/sdk/golang/sdk/dialer.go index 2283ea6c1..b0e2f9e12 100644 --- a/sdk/golang/sdk/dialer.go +++ b/sdk/golang/sdk/dialer.go @@ -5,6 +5,7 @@ import ( "github.com/openziti/sdk-golang/ziti/edge" "github.com/openziti/zrok/environment/env_core" "github.com/pkg/errors" + "time" ) func NewDialer(shrToken string, root env_core.Root) (edge.Conn, error) { @@ -23,7 +24,7 @@ func NewDialer(shrToken string, root env_core.Root) (edge.Conn, error) { return nil, errors.Wrap(err, "error getting ziti context") } - conn, err := zctx.Dial(shrToken) + conn, err := zctx.DialWithOptions(shrToken, &ziti.DialOptions{ConnectTimeout: 30 * time.Second}) if err != nil { return nil, errors.Wrapf(err, "error dialing '%v'", shrToken) }