Skip to content

Commit f881d30

Browse files
DerekBumDerekBum
committed
dial: add DialContext function
In order to replace timeouts with contexts in `Connect` instance creation (go-tarantool), I need a `DialContext` function. It accepts context, and cancels, if context is canceled by user. Part of tarantool/go-tarantool#136
1 parent b452431 commit f881d30

File tree

3 files changed

+196
-33
lines changed

3 files changed

+196
-33
lines changed

‎net.go

Lines changed: 95 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
package openssl
1616

1717
import (
18+
"context"
1819
"errors"
1920
"net"
2021
"time"
@@ -89,8 +90,55 @@ func Dial(network, addr string, ctx *Ctx, flags DialFlags) (*Conn, error) {
8990
// parameters.
9091
func DialTimeout(network, addr string, timeout time.Duration, ctx *Ctx,
9192
flags DialFlags) (*Conn, error) {
92-
d := net.Dialer{Timeout: timeout}
93-
return dialSession(d, network, addr, ctx, flags, nil)
93+
host, err := parseHost(addr)
94+
if err != nil {
95+
return nil, err
96+
}
97+
98+
conn, err := net.DialTimeout(network, addr, timeout)
99+
if err != nil {
100+
return nil, err
101+
}
102+
ctx, err = prepareCtx(ctx)
103+
if err != nil {
104+
conn.Close()
105+
return nil, err
106+
}
107+
client, err := createSession(conn, flags, host, ctx, nil)
108+
if err != nil {
109+
conn.Close()
110+
}
111+
return client, err
112+
}
113+
114+
// DialContext acts like Dial but takes a context for network dial.
115+
//
116+
// The context includes only network dial. It does not include OpenSSL calls.
117+
//
118+
// See func Dial for a description of the network, addr, ctx and flags
119+
// parameters.
120+
func DialContext(context context.Context, network, addr string,
121+
ctx *Ctx, flags DialFlags) (*Conn, error) {
122+
host, err := parseHost(addr)
123+
if err != nil {
124+
return nil, err
125+
}
126+
127+
dialer := net.Dialer{}
128+
conn, err := dialer.DialContext(context, network, addr)
129+
if err != nil {
130+
return nil, err
131+
}
132+
ctx, err = prepareCtx(ctx)
133+
if err != nil {
134+
conn.Close()
135+
return nil, err
136+
}
137+
client, err := createSession(conn, flags, host, ctx, nil)
138+
if err != nil {
139+
conn.Close()
140+
}
141+
return client, err
94142
}
95143

96144
// DialSession will connect to network/address and then wrap the corresponding
@@ -108,59 +156,76 @@ func DialTimeout(network, addr string, timeout time.Duration, ctx *Ctx,
108156
// can be retrieved from the GetSession method on the Conn.
109157
func DialSession(network, addr string, ctx *Ctx, flags DialFlags,
110158
session []byte) (*Conn, error) {
111-
var d net.Dialer
112-
return dialSession(d, network, addr, ctx, flags, session)
113-
}
114-
115-
func dialSession(d net.Dialer, network, addr string, ctx *Ctx, flags DialFlags,
116-
session []byte) (*Conn, error) {
117-
host, _, err := net.SplitHostPort(addr)
159+
host, err := parseHost(addr)
118160
if err != nil {
119161
return nil, err
120162
}
121-
if ctx == nil {
122-
var err error
123-
ctx, err = NewCtx()
124-
if err != nil {
125-
return nil, err
126-
}
127-
// TODO: use operating system default certificate chain?
128-
}
129163

130-
c, err := d.Dial(network, addr)
164+
conn, err := net.Dial(network, addr)
131165
if err != nil {
132166
return nil, err
133167
}
134-
conn, err := Client(c, ctx)
168+
ctx, err = prepareCtx(ctx)
135169
if err != nil {
136-
c.Close()
170+
conn.Close()
137171
return nil, err
138172
}
139-
if session != nil {
140-
err := conn.setSession(session)
141-
if err != nil {
142-
c.Close()
143-
return nil, err
144-
}
173+
client, err := createSession(conn, flags, host, ctx, session)
174+
if err != nil {
175+
conn.Close()
176+
}
177+
return client, err
178+
}
179+
180+
func prepareCtx(ctx *Ctx) (*Ctx, error) {
181+
if ctx == nil {
182+
return NewCtx()
145183
}
184+
return ctx, nil
185+
}
186+
187+
func parseHost(addr string) (string, error) {
188+
host, _, err := net.SplitHostPort(addr)
189+
return host, err
190+
}
191+
192+
func handshake(conn *Conn, host string, flags DialFlags) error {
193+
var err error
146194
if flags&DisableSNI == 0 {
147195
err = conn.SetTlsExtHostName(host)
148196
if err != nil {
149-
conn.Close()
150-
return nil, err
197+
return err
151198
}
152199
}
153200
err = conn.Handshake()
154201
if err != nil {
155-
conn.Close()
156-
return nil, err
202+
return err
157203
}
158204
if flags&InsecureSkipHostVerification == 0 {
159205
err = conn.VerifyHostname(host)
206+
if err != nil {
207+
return err
208+
}
209+
}
210+
return nil
211+
}
212+
213+
func createSession(c net.Conn, flags DialFlags, host string, ctx *Ctx,
214+
session []byte) (*Conn, error) {
215+
conn, err := Client(c, ctx)
216+
if err != nil {
217+
return nil, err
218+
}
219+
if session != nil {
220+
err := conn.setSession(session)
160221
if err != nil {
161222
conn.Close()
162223
return nil, err
163224
}
164225
}
226+
if err := handshake(conn, host, flags); err != nil {
227+
conn.Close()
228+
return nil, err
229+
}
165230
return conn, nil
166231
}

‎net_test.go

Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,98 @@
1+
package openssl_test
2+
3+
import (
4+
"context"
5+
"crypto/rand"
6+
"io"
7+
"net"
8+
"sync"
9+
"testing"
10+
"time"
11+
12+
"github.com/tarantool/go-openssl"
13+
)
14+
15+
func sslConnect(t *testing.T, ssl_listener net.Listener) {
16+
for {
17+
var err error
18+
conn, err := ssl_listener.Accept()
19+
if err != nil {
20+
t.Errorf("failed accept: %s", err)
21+
continue
22+
}
23+
io.Copy(conn, io.LimitReader(rand.Reader, 1024))
24+
break
25+
}
26+
}
27+
28+
func TestDial(t *testing.T) {
29+
ctx := openssl.GetCtx(t)
30+
if err := ctx.SetCipherList("AES128-SHA"); err != nil {
31+
t.Fatal(err)
32+
}
33+
ssl_listener, err := openssl.Listen("tcp", "localhost:0", ctx)
34+
if err != nil {
35+
t.Fatal(err)
36+
}
37+
38+
wg := sync.WaitGroup{}
39+
wg.Add(1)
40+
go func() {
41+
sslConnect(t, ssl_listener)
42+
wg.Done()
43+
}()
44+
45+
client, err := openssl.Dial(ssl_listener.Addr().Network(),
46+
ssl_listener.Addr().String(), ctx, openssl.InsecureSkipHostVerification)
47+
48+
wg.Wait()
49+
50+
if err != nil {
51+
t.Fatalf("unexpected err: %v", err)
52+
}
53+
n, err := io.Copy(io.Discard, io.LimitReader(client, 1024))
54+
if n != 1024 {
55+
if n == 0 {
56+
t.Fatal("client is closed after creation")
57+
}
58+
t.Fatalf("client lost some bytes, expected %d, got %d", 1024, n)
59+
}
60+
}
61+
62+
func TestDialTimeout(t *testing.T) {
63+
ctx := openssl.GetCtx(t)
64+
if err := ctx.SetCipherList("AES128-SHA"); err != nil {
65+
t.Fatal(err)
66+
}
67+
ssl_listener, err := openssl.Listen("tcp", "localhost:0", ctx)
68+
if err != nil {
69+
t.Fatal(err)
70+
}
71+
72+
client, err := openssl.DialTimeout(ssl_listener.Addr().Network(),
73+
ssl_listener.Addr().String(), time.Nanosecond, ctx, 0)
74+
75+
if client != nil || err == nil {
76+
t.Fatalf("expected error")
77+
}
78+
}
79+
80+
func TestDialContext(t *testing.T) {
81+
ctx := openssl.GetCtx(t)
82+
if err := ctx.SetCipherList("AES128-SHA"); err != nil {
83+
t.Fatal(err)
84+
}
85+
ssl_listener, err := openssl.Listen("tcp", "localhost:0", ctx)
86+
if err != nil {
87+
t.Fatal(err)
88+
}
89+
90+
cancelCtx, cancel := context.WithCancel(context.Background())
91+
cancel()
92+
client, err := openssl.DialContext(cancelCtx, ssl_listener.Addr().Network(),
93+
ssl_listener.Addr().String(), ctx, 0)
94+
95+
if client != nil || err == nil {
96+
t.Fatalf("expected error")
97+
}
98+
}

‎ssl_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -738,7 +738,7 @@ func TestStdlibLotsOfConns(t *testing.T) {
738738
})
739739
}
740740

741-
func getCtx(t *testing.T) *Ctx {
741+
func GetCtx(t *testing.T) *Ctx {
742742
ctx, err := NewCtx()
743743
if err != nil {
744744
t.Fatal(err)
@@ -761,7 +761,7 @@ func getCtx(t *testing.T) *Ctx {
761761
}
762762

763763
func TestOpenSSLLotsOfConns(t *testing.T) {
764-
ctx := getCtx(t)
764+
ctx := GetCtx(t)
765765
if err := ctx.SetCipherList("AES128-SHA"); err != nil {
766766
t.Fatal(err)
767767
}
@@ -928,7 +928,7 @@ func TestOpenSSLLotsOfConnsWithFail(t *testing.T) {
928928
t.Run(name, func(t *testing.T) {
929929
LotsOfConns(t, 1024*64, 10, 100, 0*time.Second,
930930
func(l net.Listener) net.Listener {
931-
return NewListener(l, getCtx(t))
931+
return NewListener(l, GetCtx(t))
932932
}, func(c net.Conn) (net.Conn, error) {
933933
return Client(c, getClientCtx(t))
934934
})

0 commit comments

Comments
 (0)