]> Cypherpunks.ru repositories - gostls13.git/blob - src/crypto/tls/tls_test.go
[dev.boringcrypto] all: merge master into dev.boringcrypto
[gostls13.git] / src / crypto / tls / tls_test.go
1 // Copyright 2012 The Go Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
4
5 package tls
6
7 import (
8         "bytes"
9         "crypto/x509"
10         "encoding/json"
11         "errors"
12         "fmt"
13         "internal/testenv"
14         "io"
15         "io/ioutil"
16         "math"
17         "net"
18         "os"
19         "reflect"
20         "strings"
21         "sync"
22         "testing"
23         "time"
24 )
25
26 func init() {
27         // TLS 1.3 is opt-in for Go 1.12, but we want to run most tests with it enabled.
28         // TestTLS13Switch below tests the disabled behavior. See Issue 30055.
29         tls13Support.Do(func() {}) // defuse the sync.Once
30         tls13Support.cached = true
31 }
32
33 var rsaCertPEM = `-----BEGIN CERTIFICATE-----
34 MIIB0zCCAX2gAwIBAgIJAI/M7BYjwB+uMA0GCSqGSIb3DQEBBQUAMEUxCzAJBgNV
35 BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYDVQQKDBhJbnRlcm5ldCBX
36 aWRnaXRzIFB0eSBMdGQwHhcNMTIwOTEyMjE1MjAyWhcNMTUwOTEyMjE1MjAyWjBF
37 MQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1TdGF0ZTEhMB8GA1UECgwYSW50
38 ZXJuZXQgV2lkZ2l0cyBQdHkgTHRkMFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBANLJ
39 hPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wok/4xIA+ui35/MmNa
40 rtNuC+BdZ1tMuVCPFZcCAwEAAaNQME4wHQYDVR0OBBYEFJvKs8RfJaXTH08W+SGv
41 zQyKn0H8MB8GA1UdIwQYMBaAFJvKs8RfJaXTH08W+SGvzQyKn0H8MAwGA1UdEwQF
42 MAMBAf8wDQYJKoZIhvcNAQEFBQADQQBJlffJHybjDGxRMqaRmDhX0+6v02TUKZsW
43 r5QuVbpQhH6u+0UgcW0jp9QwpxoPTLTWGXEWBBBurxFwiCBhkQ+V
44 -----END CERTIFICATE-----
45 `
46
47 var rsaKeyPEM = `-----BEGIN RSA PRIVATE KEY-----
48 MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo
49 k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G
50 6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N
51 MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW
52 SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T
53 xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi
54 D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g==
55 -----END RSA PRIVATE KEY-----
56 `
57
58 // keyPEM is the same as rsaKeyPEM, but declares itself as just
59 // "PRIVATE KEY", not "RSA PRIVATE KEY".  https://golang.org/issue/4477
60 var keyPEM = `-----BEGIN PRIVATE KEY-----
61 MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo
62 k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G
63 6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N
64 MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW
65 SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T
66 xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi
67 D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g==
68 -----END PRIVATE KEY-----
69 `
70
71 var ecdsaCertPEM = `-----BEGIN CERTIFICATE-----
72 MIIB/jCCAWICCQDscdUxw16XFDAJBgcqhkjOPQQBMEUxCzAJBgNVBAYTAkFVMRMw
73 EQYDVQQIEwpTb21lLVN0YXRlMSEwHwYDVQQKExhJbnRlcm5ldCBXaWRnaXRzIFB0
74 eSBMdGQwHhcNMTIxMTE0MTI0MDQ4WhcNMTUxMTE0MTI0MDQ4WjBFMQswCQYDVQQG
75 EwJBVTETMBEGA1UECBMKU29tZS1TdGF0ZTEhMB8GA1UEChMYSW50ZXJuZXQgV2lk
76 Z2l0cyBQdHkgTHRkMIGbMBAGByqGSM49AgEGBSuBBAAjA4GGAAQBY9+my9OoeSUR
77 lDQdV/x8LsOuLilthhiS1Tz4aGDHIPwC1mlvnf7fg5lecYpMCrLLhauAc1UJXcgl
78 01xoLuzgtAEAgv2P/jgytzRSpUYvgLBt1UA0leLYBy6mQQbrNEuqT3INapKIcUv8
79 XxYP0xMEUksLPq6Ca+CRSqTtrd/23uTnapkwCQYHKoZIzj0EAQOBigAwgYYCQXJo
80 A7Sl2nLVf+4Iu/tAX/IF4MavARKC4PPHK3zfuGfPR3oCCcsAoz3kAzOeijvd0iXb
81 H5jBImIxPL4WxQNiBTexAkF8D1EtpYuWdlVQ80/h/f4pBcGiXPqX5h2PQSQY7hP1
82 +jwM1FGS4fREIOvlBYr/SzzQRtwrvrzGYxDEDbsC0ZGRnA==
83 -----END CERTIFICATE-----
84 `
85
86 var ecdsaKeyPEM = `-----BEGIN EC PARAMETERS-----
87 BgUrgQQAIw==
88 -----END EC PARAMETERS-----
89 -----BEGIN EC PRIVATE KEY-----
90 MIHcAgEBBEIBrsoKp0oqcv6/JovJJDoDVSGWdirrkgCWxrprGlzB9o0X8fV675X0
91 NwuBenXFfeZvVcwluO7/Q9wkYoPd/t3jGImgBwYFK4EEACOhgYkDgYYABAFj36bL
92 06h5JRGUNB1X/Hwuw64uKW2GGJLVPPhoYMcg/ALWaW+d/t+DmV5xikwKssuFq4Bz
93 VQldyCXTXGgu7OC0AQCC/Y/+ODK3NFKlRi+AsG3VQDSV4tgHLqZBBus0S6pPcg1q
94 kohxS/xfFg/TEwRSSws+roJr4JFKpO2t3/be5OdqmQ==
95 -----END EC PRIVATE KEY-----
96 `
97
98 var keyPairTests = []struct {
99         algo string
100         cert string
101         key  string
102 }{
103         {"ECDSA", ecdsaCertPEM, ecdsaKeyPEM},
104         {"RSA", rsaCertPEM, rsaKeyPEM},
105         {"RSA-untyped", rsaCertPEM, keyPEM}, // golang.org/issue/4477
106 }
107
108 func TestX509KeyPair(t *testing.T) {
109         t.Parallel()
110         var pem []byte
111         for _, test := range keyPairTests {
112                 pem = []byte(test.cert + test.key)
113                 if _, err := X509KeyPair(pem, pem); err != nil {
114                         t.Errorf("Failed to load %s cert followed by %s key: %s", test.algo, test.algo, err)
115                 }
116                 pem = []byte(test.key + test.cert)
117                 if _, err := X509KeyPair(pem, pem); err != nil {
118                         t.Errorf("Failed to load %s key followed by %s cert: %s", test.algo, test.algo, err)
119                 }
120         }
121 }
122
123 func TestX509KeyPairErrors(t *testing.T) {
124         _, err := X509KeyPair([]byte(rsaKeyPEM), []byte(rsaCertPEM))
125         if err == nil {
126                 t.Fatalf("X509KeyPair didn't return an error when arguments were switched")
127         }
128         if subStr := "been switched"; !strings.Contains(err.Error(), subStr) {
129                 t.Fatalf("Expected %q in the error when switching arguments to X509KeyPair, but the error was %q", subStr, err)
130         }
131
132         _, err = X509KeyPair([]byte(rsaCertPEM), []byte(rsaCertPEM))
133         if err == nil {
134                 t.Fatalf("X509KeyPair didn't return an error when both arguments were certificates")
135         }
136         if subStr := "certificate"; !strings.Contains(err.Error(), subStr) {
137                 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were certificates, but the error was %q", subStr, err)
138         }
139
140         const nonsensePEM = `
141 -----BEGIN NONSENSE-----
142 Zm9vZm9vZm9v
143 -----END NONSENSE-----
144 `
145
146         _, err = X509KeyPair([]byte(nonsensePEM), []byte(nonsensePEM))
147         if err == nil {
148                 t.Fatalf("X509KeyPair didn't return an error when both arguments were nonsense")
149         }
150         if subStr := "NONSENSE"; !strings.Contains(err.Error(), subStr) {
151                 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were nonsense, but the error was %q", subStr, err)
152         }
153 }
154
155 func TestX509MixedKeyPair(t *testing.T) {
156         if _, err := X509KeyPair([]byte(rsaCertPEM), []byte(ecdsaKeyPEM)); err == nil {
157                 t.Error("Load of RSA certificate succeeded with ECDSA private key")
158         }
159         if _, err := X509KeyPair([]byte(ecdsaCertPEM), []byte(rsaKeyPEM)); err == nil {
160                 t.Error("Load of ECDSA certificate succeeded with RSA private key")
161         }
162 }
163
164 func newLocalListener(t testing.TB) net.Listener {
165         ln, err := net.Listen("tcp", "127.0.0.1:0")
166         if err != nil {
167                 ln, err = net.Listen("tcp6", "[::1]:0")
168         }
169         if err != nil {
170                 t.Fatal(err)
171         }
172         return ln
173 }
174
175 func TestDialTimeout(t *testing.T) {
176         if testing.Short() {
177                 t.Skip("skipping in short mode")
178         }
179         listener := newLocalListener(t)
180
181         addr := listener.Addr().String()
182         defer listener.Close()
183
184         complete := make(chan bool)
185         defer close(complete)
186
187         go func() {
188                 conn, err := listener.Accept()
189                 if err != nil {
190                         t.Error(err)
191                         return
192                 }
193                 <-complete
194                 conn.Close()
195         }()
196
197         dialer := &net.Dialer{
198                 Timeout: 10 * time.Millisecond,
199         }
200
201         var err error
202         if _, err = DialWithDialer(dialer, "tcp", addr, nil); err == nil {
203                 t.Fatal("DialWithTimeout completed successfully")
204         }
205
206         if !isTimeoutError(err) {
207                 t.Errorf("resulting error not a timeout: %v\nType %T: %#v", err, err, err)
208         }
209 }
210
211 func isTimeoutError(err error) bool {
212         if ne, ok := err.(net.Error); ok {
213                 return ne.Timeout()
214         }
215         return false
216 }
217
218 // tests that Conn.Read returns (non-zero, io.EOF) instead of
219 // (non-zero, nil) when a Close (alertCloseNotify) is sitting right
220 // behind the application data in the buffer.
221 func TestConnReadNonzeroAndEOF(t *testing.T) {
222         // This test is racy: it assumes that after a write to a
223         // localhost TCP connection, the peer TCP connection can
224         // immediately read it. Because it's racy, we skip this test
225         // in short mode, and then retry it several times with an
226         // increasing sleep in between our final write (via srv.Close
227         // below) and the following read.
228         if testing.Short() {
229                 t.Skip("skipping in short mode")
230         }
231         var err error
232         for delay := time.Millisecond; delay <= 64*time.Millisecond; delay *= 2 {
233                 if err = testConnReadNonzeroAndEOF(t, delay); err == nil {
234                         return
235                 }
236         }
237         t.Error(err)
238 }
239
240 func testConnReadNonzeroAndEOF(t *testing.T, delay time.Duration) error {
241         ln := newLocalListener(t)
242         defer ln.Close()
243
244         srvCh := make(chan *Conn, 1)
245         var serr error
246         go func() {
247                 sconn, err := ln.Accept()
248                 if err != nil {
249                         serr = err
250                         srvCh <- nil
251                         return
252                 }
253                 serverConfig := testConfig.Clone()
254                 srv := Server(sconn, serverConfig)
255                 if err := srv.Handshake(); err != nil {
256                         serr = fmt.Errorf("handshake: %v", err)
257                         srvCh <- nil
258                         return
259                 }
260                 srvCh <- srv
261         }()
262
263         clientConfig := testConfig.Clone()
264         // In TLS 1.3, alerts are encrypted and disguised as application data, so
265         // the opportunistic peek won't work.
266         clientConfig.MaxVersion = VersionTLS12
267         conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
268         if err != nil {
269                 t.Fatal(err)
270         }
271         defer conn.Close()
272
273         srv := <-srvCh
274         if srv == nil {
275                 return serr
276         }
277
278         buf := make([]byte, 6)
279
280         srv.Write([]byte("foobar"))
281         n, err := conn.Read(buf)
282         if n != 6 || err != nil || string(buf) != "foobar" {
283                 return fmt.Errorf("Read = %d, %v, data %q; want 6, nil, foobar", n, err, buf)
284         }
285
286         srv.Write([]byte("abcdef"))
287         srv.Close()
288         time.Sleep(delay)
289         n, err = conn.Read(buf)
290         if n != 6 || string(buf) != "abcdef" {
291                 return fmt.Errorf("Read = %d, buf= %q; want 6, abcdef", n, buf)
292         }
293         if err != io.EOF {
294                 return fmt.Errorf("Second Read error = %v; want io.EOF", err)
295         }
296         return nil
297 }
298
299 func TestTLSUniqueMatches(t *testing.T) {
300         ln := newLocalListener(t)
301         defer ln.Close()
302
303         serverTLSUniques := make(chan []byte)
304         go func() {
305                 for i := 0; i < 2; i++ {
306                         sconn, err := ln.Accept()
307                         if err != nil {
308                                 t.Error(err)
309                                 return
310                         }
311                         serverConfig := testConfig.Clone()
312                         serverConfig.MaxVersion = VersionTLS12 // TLSUnique is not defined in TLS 1.3
313                         srv := Server(sconn, serverConfig)
314                         if err := srv.Handshake(); err != nil {
315                                 t.Error(err)
316                                 return
317                         }
318                         serverTLSUniques <- srv.ConnectionState().TLSUnique
319                 }
320         }()
321
322         clientConfig := testConfig.Clone()
323         clientConfig.ClientSessionCache = NewLRUClientSessionCache(1)
324         conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
325         if err != nil {
326                 t.Fatal(err)
327         }
328         if !bytes.Equal(conn.ConnectionState().TLSUnique, <-serverTLSUniques) {
329                 t.Error("client and server channel bindings differ")
330         }
331         conn.Close()
332
333         conn, err = Dial("tcp", ln.Addr().String(), clientConfig)
334         if err != nil {
335                 t.Fatal(err)
336         }
337         defer conn.Close()
338         if !conn.ConnectionState().DidResume {
339                 t.Error("second session did not use resumption")
340         }
341         if !bytes.Equal(conn.ConnectionState().TLSUnique, <-serverTLSUniques) {
342                 t.Error("client and server channel bindings differ when session resumption is used")
343         }
344 }
345
346 func TestVerifyHostname(t *testing.T) {
347         testenv.MustHaveExternalNetwork(t)
348
349         c, err := Dial("tcp", "www.google.com:https", nil)
350         if err != nil {
351                 t.Fatal(err)
352         }
353         if err := c.VerifyHostname("www.google.com"); err != nil {
354                 t.Fatalf("verify www.google.com: %v", err)
355         }
356         if err := c.VerifyHostname("www.yahoo.com"); err == nil {
357                 t.Fatalf("verify www.yahoo.com succeeded")
358         }
359
360         c, err = Dial("tcp", "www.google.com:https", &Config{InsecureSkipVerify: true})
361         if err != nil {
362                 t.Fatal(err)
363         }
364         if err := c.VerifyHostname("www.google.com"); err == nil {
365                 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true")
366         }
367 }
368
369 func TestVerifyHostnameResumed(t *testing.T) {
370         t.Run("TLSv12", func(t *testing.T) { testVerifyHostnameResumed(t, VersionTLS12) })
371         t.Run("TLSv13", func(t *testing.T) { testVerifyHostnameResumed(t, VersionTLS13) })
372 }
373
374 func testVerifyHostnameResumed(t *testing.T, version uint16) {
375         testenv.MustHaveExternalNetwork(t)
376
377         config := &Config{
378                 MaxVersion:         version,
379                 ClientSessionCache: NewLRUClientSessionCache(32),
380         }
381         for i := 0; i < 2; i++ {
382                 c, err := Dial("tcp", "mail.google.com:https", config)
383                 if err != nil {
384                         t.Fatalf("Dial #%d: %v", i, err)
385                 }
386                 cs := c.ConnectionState()
387                 if i > 0 && !cs.DidResume {
388                         t.Fatalf("Subsequent connection unexpectedly didn't resume")
389                 }
390                 if cs.Version != version {
391                         t.Fatalf("Unexpectedly negotiated version %x", cs.Version)
392                 }
393                 if cs.VerifiedChains == nil {
394                         t.Fatalf("Dial #%d: cs.VerifiedChains == nil", i)
395                 }
396                 if err := c.VerifyHostname("mail.google.com"); err != nil {
397                         t.Fatalf("verify mail.google.com #%d: %v", i, err)
398                 }
399                 // Give the client a chance to read the server session tickets.
400                 c.SetReadDeadline(time.Now().Add(500 * time.Millisecond))
401                 if _, err := c.Read(make([]byte, 1)); err != nil {
402                         if err, ok := err.(net.Error); !ok || !err.Timeout() {
403                                 t.Fatal(err)
404                         }
405                 }
406                 c.Close()
407         }
408 }
409
410 func TestConnCloseBreakingWrite(t *testing.T) {
411         ln := newLocalListener(t)
412         defer ln.Close()
413
414         srvCh := make(chan *Conn, 1)
415         var serr error
416         var sconn net.Conn
417         go func() {
418                 var err error
419                 sconn, err = ln.Accept()
420                 if err != nil {
421                         serr = err
422                         srvCh <- nil
423                         return
424                 }
425                 serverConfig := testConfig.Clone()
426                 srv := Server(sconn, serverConfig)
427                 if err := srv.Handshake(); err != nil {
428                         serr = fmt.Errorf("handshake: %v", err)
429                         srvCh <- nil
430                         return
431                 }
432                 srvCh <- srv
433         }()
434
435         cconn, err := net.Dial("tcp", ln.Addr().String())
436         if err != nil {
437                 t.Fatal(err)
438         }
439         defer cconn.Close()
440
441         conn := &changeImplConn{
442                 Conn: cconn,
443         }
444
445         clientConfig := testConfig.Clone()
446         tconn := Client(conn, clientConfig)
447         if err := tconn.Handshake(); err != nil {
448                 t.Fatal(err)
449         }
450
451         srv := <-srvCh
452         if srv == nil {
453                 t.Fatal(serr)
454         }
455         defer sconn.Close()
456
457         connClosed := make(chan struct{})
458         conn.closeFunc = func() error {
459                 close(connClosed)
460                 return nil
461         }
462
463         inWrite := make(chan bool, 1)
464         var errConnClosed = errors.New("conn closed for test")
465         conn.writeFunc = func(p []byte) (n int, err error) {
466                 inWrite <- true
467                 <-connClosed
468                 return 0, errConnClosed
469         }
470
471         closeReturned := make(chan bool, 1)
472         go func() {
473                 <-inWrite
474                 tconn.Close() // test that this doesn't block forever.
475                 closeReturned <- true
476         }()
477
478         _, err = tconn.Write([]byte("foo"))
479         if err != errConnClosed {
480                 t.Errorf("Write error = %v; want errConnClosed", err)
481         }
482
483         <-closeReturned
484         if err := tconn.Close(); err != errClosed {
485                 t.Errorf("Close error = %v; want errClosed", err)
486         }
487 }
488
489 func TestConnCloseWrite(t *testing.T) {
490         ln := newLocalListener(t)
491         defer ln.Close()
492
493         clientDoneChan := make(chan struct{})
494
495         serverCloseWrite := func() error {
496                 sconn, err := ln.Accept()
497                 if err != nil {
498                         return fmt.Errorf("accept: %v", err)
499                 }
500                 defer sconn.Close()
501
502                 serverConfig := testConfig.Clone()
503                 srv := Server(sconn, serverConfig)
504                 if err := srv.Handshake(); err != nil {
505                         return fmt.Errorf("handshake: %v", err)
506                 }
507                 defer srv.Close()
508
509                 data, err := ioutil.ReadAll(srv)
510                 if err != nil {
511                         return err
512                 }
513                 if len(data) > 0 {
514                         return fmt.Errorf("Read data = %q; want nothing", data)
515                 }
516
517                 if err := srv.CloseWrite(); err != nil {
518                         return fmt.Errorf("server CloseWrite: %v", err)
519                 }
520
521                 // Wait for clientCloseWrite to finish, so we know we
522                 // tested the CloseWrite before we defer the
523                 // sconn.Close above, which would also cause the
524                 // client to unblock like CloseWrite.
525                 <-clientDoneChan
526                 return nil
527         }
528
529         clientCloseWrite := func() error {
530                 defer close(clientDoneChan)
531
532                 clientConfig := testConfig.Clone()
533                 conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
534                 if err != nil {
535                         return err
536                 }
537                 if err := conn.Handshake(); err != nil {
538                         return err
539                 }
540                 defer conn.Close()
541
542                 if err := conn.CloseWrite(); err != nil {
543                         return fmt.Errorf("client CloseWrite: %v", err)
544                 }
545
546                 if _, err := conn.Write([]byte{0}); err != errShutdown {
547                         return fmt.Errorf("CloseWrite error = %v; want errShutdown", err)
548                 }
549
550                 data, err := ioutil.ReadAll(conn)
551                 if err != nil {
552                         return err
553                 }
554                 if len(data) > 0 {
555                         return fmt.Errorf("Read data = %q; want nothing", data)
556                 }
557                 return nil
558         }
559
560         errChan := make(chan error, 2)
561
562         go func() { errChan <- serverCloseWrite() }()
563         go func() { errChan <- clientCloseWrite() }()
564
565         for i := 0; i < 2; i++ {
566                 select {
567                 case err := <-errChan:
568                         if err != nil {
569                                 t.Fatal(err)
570                         }
571                 case <-time.After(10 * time.Second):
572                         t.Fatal("deadlock")
573                 }
574         }
575
576         // Also test CloseWrite being called before the handshake is
577         // finished:
578         {
579                 ln2 := newLocalListener(t)
580                 defer ln2.Close()
581
582                 netConn, err := net.Dial("tcp", ln2.Addr().String())
583                 if err != nil {
584                         t.Fatal(err)
585                 }
586                 defer netConn.Close()
587                 conn := Client(netConn, testConfig.Clone())
588
589                 if err := conn.CloseWrite(); err != errEarlyCloseWrite {
590                         t.Errorf("CloseWrite error = %v; want errEarlyCloseWrite", err)
591                 }
592         }
593 }
594
595 func TestWarningAlertFlood(t *testing.T) {
596         ln := newLocalListener(t)
597         defer ln.Close()
598
599         server := func() error {
600                 sconn, err := ln.Accept()
601                 if err != nil {
602                         return fmt.Errorf("accept: %v", err)
603                 }
604                 defer sconn.Close()
605
606                 serverConfig := testConfig.Clone()
607                 srv := Server(sconn, serverConfig)
608                 if err := srv.Handshake(); err != nil {
609                         return fmt.Errorf("handshake: %v", err)
610                 }
611                 defer srv.Close()
612
613                 _, err = ioutil.ReadAll(srv)
614                 if err == nil {
615                         return errors.New("unexpected lack of error from server")
616                 }
617                 const expected = "too many ignored"
618                 if str := err.Error(); !strings.Contains(str, expected) {
619                         return fmt.Errorf("expected error containing %q, but saw: %s", expected, str)
620                 }
621
622                 return nil
623         }
624
625         errChan := make(chan error, 1)
626         go func() { errChan <- server() }()
627
628         clientConfig := testConfig.Clone()
629         clientConfig.MaxVersion = VersionTLS12 // there are no warning alerts in TLS 1.3
630         conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
631         if err != nil {
632                 t.Fatal(err)
633         }
634         defer conn.Close()
635         if err := conn.Handshake(); err != nil {
636                 t.Fatal(err)
637         }
638
639         for i := 0; i < maxUselessRecords+1; i++ {
640                 conn.sendAlert(alertNoRenegotiation)
641         }
642
643         if err := <-errChan; err != nil {
644                 t.Fatal(err)
645         }
646 }
647
648 func TestCloneFuncFields(t *testing.T) {
649         const expectedCount = 5
650         called := 0
651
652         c1 := Config{
653                 Time: func() time.Time {
654                         called |= 1 << 0
655                         return time.Time{}
656                 },
657                 GetCertificate: func(*ClientHelloInfo) (*Certificate, error) {
658                         called |= 1 << 1
659                         return nil, nil
660                 },
661                 GetClientCertificate: func(*CertificateRequestInfo) (*Certificate, error) {
662                         called |= 1 << 2
663                         return nil, nil
664                 },
665                 GetConfigForClient: func(*ClientHelloInfo) (*Config, error) {
666                         called |= 1 << 3
667                         return nil, nil
668                 },
669                 VerifyPeerCertificate: func(rawCerts [][]byte, verifiedChains [][]*x509.Certificate) error {
670                         called |= 1 << 4
671                         return nil
672                 },
673         }
674
675         c2 := c1.Clone()
676
677         c2.Time()
678         c2.GetCertificate(nil)
679         c2.GetClientCertificate(nil)
680         c2.GetConfigForClient(nil)
681         c2.VerifyPeerCertificate(nil, nil)
682
683         if called != (1<<expectedCount)-1 {
684                 t.Fatalf("expected %d calls but saw calls %b", expectedCount, called)
685         }
686 }
687
688 func TestCloneNonFuncFields(t *testing.T) {
689         var c1 Config
690         v := reflect.ValueOf(&c1).Elem()
691
692         typ := v.Type()
693         for i := 0; i < typ.NumField(); i++ {
694                 f := v.Field(i)
695                 if !f.CanSet() {
696                         // unexported field; not cloned.
697                         continue
698                 }
699
700                 // testing/quick can't handle functions or interfaces and so
701                 // isn't used here.
702                 switch fn := typ.Field(i).Name; fn {
703                 case "Rand":
704                         f.Set(reflect.ValueOf(io.Reader(os.Stdin)))
705                 case "Time", "GetCertificate", "GetConfigForClient", "VerifyPeerCertificate", "GetClientCertificate":
706                         // DeepEqual can't compare functions. If you add a
707                         // function field to this list, you must also change
708                         // TestCloneFuncFields to ensure that the func field is
709                         // cloned.
710                 case "Certificates":
711                         f.Set(reflect.ValueOf([]Certificate{
712                                 {Certificate: [][]byte{{'b'}}},
713                         }))
714                 case "NameToCertificate":
715                         f.Set(reflect.ValueOf(map[string]*Certificate{"a": nil}))
716                 case "RootCAs", "ClientCAs":
717                         f.Set(reflect.ValueOf(x509.NewCertPool()))
718                 case "ClientSessionCache":
719                         f.Set(reflect.ValueOf(NewLRUClientSessionCache(10)))
720                 case "KeyLogWriter":
721                         f.Set(reflect.ValueOf(io.Writer(os.Stdout)))
722                 case "NextProtos":
723                         f.Set(reflect.ValueOf([]string{"a", "b"}))
724                 case "ServerName":
725                         f.Set(reflect.ValueOf("b"))
726                 case "ClientAuth":
727                         f.Set(reflect.ValueOf(VerifyClientCertIfGiven))
728                 case "InsecureSkipVerify", "SessionTicketsDisabled", "DynamicRecordSizingDisabled", "PreferServerCipherSuites":
729                         f.Set(reflect.ValueOf(true))
730                 case "MinVersion", "MaxVersion":
731                         f.Set(reflect.ValueOf(uint16(VersionTLS12)))
732                 case "SessionTicketKey":
733                         f.Set(reflect.ValueOf([32]byte{}))
734                 case "CipherSuites":
735                         f.Set(reflect.ValueOf([]uint16{1, 2}))
736                 case "CurvePreferences":
737                         f.Set(reflect.ValueOf([]CurveID{CurveP256}))
738                 case "Renegotiation":
739                         f.Set(reflect.ValueOf(RenegotiateOnceAsClient))
740                 default:
741                         t.Errorf("all fields must be accounted for, but saw unknown field %q", fn)
742                 }
743         }
744
745         c2 := c1.Clone()
746         // DeepEqual also compares unexported fields, thus c2 needs to have run
747         // serverInit in order to be DeepEqual to c1. Cloning it and discarding
748         // the result is sufficient.
749         c2.Clone()
750
751         if !reflect.DeepEqual(&c1, c2) {
752                 t.Errorf("clone failed to copy a field")
753         }
754 }
755
756 // changeImplConn is a net.Conn which can change its Write and Close
757 // methods.
758 type changeImplConn struct {
759         net.Conn
760         writeFunc func([]byte) (int, error)
761         closeFunc func() error
762 }
763
764 func (w *changeImplConn) Write(p []byte) (n int, err error) {
765         if w.writeFunc != nil {
766                 return w.writeFunc(p)
767         }
768         return w.Conn.Write(p)
769 }
770
771 func (w *changeImplConn) Close() error {
772         if w.closeFunc != nil {
773                 return w.closeFunc()
774         }
775         return w.Conn.Close()
776 }
777
778 func throughput(b *testing.B, version uint16, totalBytes int64, dynamicRecordSizingDisabled bool) {
779         ln := newLocalListener(b)
780         defer ln.Close()
781
782         N := b.N
783
784         // Less than 64KB because Windows appears to use a TCP rwin < 64KB.
785         // See Issue #15899.
786         const bufsize = 32 << 10
787
788         go func() {
789                 buf := make([]byte, bufsize)
790                 for i := 0; i < N; i++ {
791                         sconn, err := ln.Accept()
792                         if err != nil {
793                                 // panic rather than synchronize to avoid benchmark overhead
794                                 // (cannot call b.Fatal in goroutine)
795                                 panic(fmt.Errorf("accept: %v", err))
796                         }
797                         serverConfig := testConfig.Clone()
798                         serverConfig.CipherSuites = nil // the defaults may prefer faster ciphers
799                         serverConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
800                         srv := Server(sconn, serverConfig)
801                         if err := srv.Handshake(); err != nil {
802                                 panic(fmt.Errorf("handshake: %v", err))
803                         }
804                         if _, err := io.CopyBuffer(srv, srv, buf); err != nil {
805                                 panic(fmt.Errorf("copy buffer: %v", err))
806                         }
807                 }
808         }()
809
810         b.SetBytes(totalBytes)
811         clientConfig := testConfig.Clone()
812         clientConfig.CipherSuites = nil // the defaults may prefer faster ciphers
813         clientConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
814         clientConfig.MaxVersion = version
815
816         buf := make([]byte, bufsize)
817         chunks := int(math.Ceil(float64(totalBytes) / float64(len(buf))))
818         for i := 0; i < N; i++ {
819                 conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
820                 if err != nil {
821                         b.Fatal(err)
822                 }
823                 for j := 0; j < chunks; j++ {
824                         _, err := conn.Write(buf)
825                         if err != nil {
826                                 b.Fatal(err)
827                         }
828                         _, err = io.ReadFull(conn, buf)
829                         if err != nil {
830                                 b.Fatal(err)
831                         }
832                 }
833                 conn.Close()
834         }
835 }
836
837 func BenchmarkThroughput(b *testing.B) {
838         for _, mode := range []string{"Max", "Dynamic"} {
839                 for size := 1; size <= 64; size <<= 1 {
840                         name := fmt.Sprintf("%sPacket/%dMB", mode, size)
841                         b.Run(name, func(b *testing.B) {
842                                 b.Run("TLSv12", func(b *testing.B) {
843                                         throughput(b, VersionTLS12, int64(size<<20), mode == "Max")
844                                 })
845                                 b.Run("TLSv13", func(b *testing.B) {
846                                         throughput(b, VersionTLS13, int64(size<<20), mode == "Max")
847                                 })
848                         })
849                 }
850         }
851 }
852
853 type slowConn struct {
854         net.Conn
855         bps int
856 }
857
858 func (c *slowConn) Write(p []byte) (int, error) {
859         if c.bps == 0 {
860                 panic("too slow")
861         }
862         t0 := time.Now()
863         wrote := 0
864         for wrote < len(p) {
865                 time.Sleep(100 * time.Microsecond)
866                 allowed := int(time.Since(t0).Seconds()*float64(c.bps)) / 8
867                 if allowed > len(p) {
868                         allowed = len(p)
869                 }
870                 if wrote < allowed {
871                         n, err := c.Conn.Write(p[wrote:allowed])
872                         wrote += n
873                         if err != nil {
874                                 return wrote, err
875                         }
876                 }
877         }
878         return len(p), nil
879 }
880
881 func latency(b *testing.B, version uint16, bps int, dynamicRecordSizingDisabled bool) {
882         ln := newLocalListener(b)
883         defer ln.Close()
884
885         N := b.N
886
887         go func() {
888                 for i := 0; i < N; i++ {
889                         sconn, err := ln.Accept()
890                         if err != nil {
891                                 // panic rather than synchronize to avoid benchmark overhead
892                                 // (cannot call b.Fatal in goroutine)
893                                 panic(fmt.Errorf("accept: %v", err))
894                         }
895                         serverConfig := testConfig.Clone()
896                         serverConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
897                         srv := Server(&slowConn{sconn, bps}, serverConfig)
898                         if err := srv.Handshake(); err != nil {
899                                 panic(fmt.Errorf("handshake: %v", err))
900                         }
901                         io.Copy(srv, srv)
902                 }
903         }()
904
905         clientConfig := testConfig.Clone()
906         clientConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
907         clientConfig.MaxVersion = version
908
909         buf := make([]byte, 16384)
910         peek := make([]byte, 1)
911
912         for i := 0; i < N; i++ {
913                 conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
914                 if err != nil {
915                         b.Fatal(err)
916                 }
917                 // make sure we're connected and previous connection has stopped
918                 if _, err := conn.Write(buf[:1]); err != nil {
919                         b.Fatal(err)
920                 }
921                 if _, err := io.ReadFull(conn, peek); err != nil {
922                         b.Fatal(err)
923                 }
924                 if _, err := conn.Write(buf); err != nil {
925                         b.Fatal(err)
926                 }
927                 if _, err = io.ReadFull(conn, peek); err != nil {
928                         b.Fatal(err)
929                 }
930                 conn.Close()
931         }
932 }
933
934 func BenchmarkLatency(b *testing.B) {
935         for _, mode := range []string{"Max", "Dynamic"} {
936                 for _, kbps := range []int{200, 500, 1000, 2000, 5000} {
937                         name := fmt.Sprintf("%sPacket/%dkbps", mode, kbps)
938                         b.Run(name, func(b *testing.B) {
939                                 b.Run("TLSv12", func(b *testing.B) {
940                                         latency(b, VersionTLS12, kbps*1000, mode == "Max")
941                                 })
942                                 b.Run("TLSv13", func(b *testing.B) {
943                                         latency(b, VersionTLS13, kbps*1000, mode == "Max")
944                                 })
945                         })
946                 }
947         }
948 }
949
950 func TestConnectionStateMarshal(t *testing.T) {
951         cs := &ConnectionState{}
952         _, err := json.Marshal(cs)
953         if err != nil {
954                 t.Errorf("json.Marshal failed on ConnectionState: %v", err)
955         }
956 }
957
958 func TestConnectionState(t *testing.T) {
959         issuer, err := x509.ParseCertificate(testRSACertificateIssuer)
960         if err != nil {
961                 panic(err)
962         }
963         rootCAs := x509.NewCertPool()
964         rootCAs.AddCert(issuer)
965
966         now := func() time.Time { return time.Unix(1476984729, 0) }
967
968         const alpnProtocol = "golang"
969         const serverName = "example.golang"
970         var scts = [][]byte{[]byte("dummy sct 1"), []byte("dummy sct 2")}
971         var ocsp = []byte("dummy ocsp")
972
973         for _, v := range []uint16{VersionTLS12, VersionTLS13} {
974                 var name string
975                 switch v {
976                 case VersionTLS12:
977                         name = "TLSv12"
978                 case VersionTLS13:
979                         name = "TLSv13"
980                 }
981                 t.Run(name, func(t *testing.T) {
982                         config := &Config{
983                                 Time:         now,
984                                 Rand:         zeroSource{},
985                                 Certificates: make([]Certificate, 1),
986                                 MaxVersion:   v,
987                                 RootCAs:      rootCAs,
988                                 ClientCAs:    rootCAs,
989                                 ClientAuth:   RequireAndVerifyClientCert,
990                                 NextProtos:   []string{alpnProtocol},
991                                 ServerName:   serverName,
992                         }
993                         config.Certificates[0].Certificate = [][]byte{testRSACertificate}
994                         config.Certificates[0].PrivateKey = testRSAPrivateKey
995                         config.Certificates[0].SignedCertificateTimestamps = scts
996                         config.Certificates[0].OCSPStaple = ocsp
997
998                         ss, cs, err := testHandshake(t, config, config)
999                         if err != nil {
1000                                 t.Fatalf("Handshake failed: %v", err)
1001                         }
1002
1003                         if ss.Version != v || cs.Version != v {
1004                                 t.Errorf("Got versions %x (server) and %x (client), expected %x", ss.Version, cs.Version, v)
1005                         }
1006
1007                         if !ss.HandshakeComplete || !cs.HandshakeComplete {
1008                                 t.Errorf("Got HandshakeComplete %v (server) and %v (client), expected true", ss.HandshakeComplete, cs.HandshakeComplete)
1009                         }
1010
1011                         if ss.DidResume || cs.DidResume {
1012                                 t.Errorf("Got DidResume %v (server) and %v (client), expected false", ss.DidResume, cs.DidResume)
1013                         }
1014
1015                         if ss.CipherSuite == 0 || cs.CipherSuite == 0 {
1016                                 t.Errorf("Got invalid cipher suite: %v (server) and %v (client)", ss.CipherSuite, cs.CipherSuite)
1017                         }
1018
1019                         if ss.NegotiatedProtocol != alpnProtocol || cs.NegotiatedProtocol != alpnProtocol {
1020                                 t.Errorf("Got negotiated protocol %q (server) and %q (client), expected %q", ss.NegotiatedProtocol, cs.NegotiatedProtocol, alpnProtocol)
1021                         }
1022
1023                         if !cs.NegotiatedProtocolIsMutual {
1024                                 t.Errorf("Got false NegotiatedProtocolIsMutual on the client side")
1025                         }
1026                         // NegotiatedProtocolIsMutual on the server side is unspecified.
1027
1028                         if ss.ServerName != serverName {
1029                                 t.Errorf("Got server name %q, expected %q", ss.ServerName, serverName)
1030                         }
1031                         if cs.ServerName != "" {
1032                                 t.Errorf("Got unexpected server name on the client side")
1033                         }
1034
1035                         if len(ss.PeerCertificates) != 1 || len(cs.PeerCertificates) != 1 {
1036                                 t.Errorf("Got %d (server) and %d (client) peer certificates, expected %d", len(ss.PeerCertificates), len(cs.PeerCertificates), 1)
1037                         }
1038
1039                         if len(ss.VerifiedChains) != 1 || len(cs.VerifiedChains) != 1 {
1040                                 t.Errorf("Got %d (server) and %d (client) verified chains, expected %d", len(ss.VerifiedChains), len(cs.VerifiedChains), 1)
1041                         } else if len(ss.VerifiedChains[0]) != 2 || len(cs.VerifiedChains[0]) != 2 {
1042                                 t.Errorf("Got %d (server) and %d (client) long verified chain, expected %d", len(ss.VerifiedChains[0]), len(cs.VerifiedChains[0]), 2)
1043                         }
1044
1045                         if len(cs.SignedCertificateTimestamps) != 2 {
1046                                 t.Errorf("Got %d SCTs, expected %d", len(cs.SignedCertificateTimestamps), 2)
1047                         }
1048                         if !bytes.Equal(cs.OCSPResponse, ocsp) {
1049                                 t.Errorf("Got OCSPs %x, expected %x", cs.OCSPResponse, ocsp)
1050                         }
1051                         // Only TLS 1.3 supports OCSP and SCTs on client certs.
1052                         if v == VersionTLS13 {
1053                                 if len(ss.SignedCertificateTimestamps) != 2 {
1054                                         t.Errorf("Got %d client SCTs, expected %d", len(ss.SignedCertificateTimestamps), 2)
1055                                 }
1056                                 if !bytes.Equal(ss.OCSPResponse, ocsp) {
1057                                         t.Errorf("Got client OCSPs %x, expected %x", ss.OCSPResponse, ocsp)
1058                                 }
1059                         }
1060
1061                         if v == VersionTLS13 {
1062                                 if ss.TLSUnique != nil || cs.TLSUnique != nil {
1063                                         t.Errorf("Got TLSUnique %x (server) and %x (client), expected nil in TLS 1.3", ss.TLSUnique, cs.TLSUnique)
1064                                 }
1065                         } else {
1066                                 if ss.TLSUnique == nil || cs.TLSUnique == nil {
1067                                         t.Errorf("Got TLSUnique %x (server) and %x (client), expected non-nil", ss.TLSUnique, cs.TLSUnique)
1068                                 }
1069                         }
1070                 })
1071         }
1072 }
1073
1074 // TestEscapeRoute tests that the library will still work if support for TLS 1.3
1075 // is dropped later in the Go 1.12 cycle.
1076 func TestEscapeRoute(t *testing.T) {
1077         defer func(savedSupportedVersions []uint16) {
1078                 supportedVersions = savedSupportedVersions
1079         }(supportedVersions)
1080         supportedVersions = []uint16{
1081                 VersionTLS12,
1082                 VersionTLS11,
1083                 VersionTLS10,
1084                 VersionSSL30,
1085         }
1086
1087         expectVersion(t, testConfig, testConfig, VersionTLS12)
1088 }
1089
1090 func expectVersion(t *testing.T, clientConfig, serverConfig *Config, v uint16) {
1091         ss, cs, err := testHandshake(t, clientConfig, serverConfig)
1092         if err != nil {
1093                 t.Fatalf("Handshake failed: %v", err)
1094         }
1095         if ss.Version != v {
1096                 t.Errorf("Server negotiated version %x, expected %x", cs.Version, v)
1097         }
1098         if cs.Version != v {
1099                 t.Errorf("Client negotiated version %x, expected %x", cs.Version, v)
1100         }
1101 }
1102
1103 // TestTLS13Switch checks the behavior of GODEBUG=tls13=[0|1]. See Issue 30055.
1104 func TestTLS13Switch(t *testing.T) {
1105         defer func(savedGODEBUG string) {
1106                 os.Setenv("GODEBUG", savedGODEBUG)
1107         }(os.Getenv("GODEBUG"))
1108
1109         os.Setenv("GODEBUG", "tls13=0")
1110         tls13Support.Once = sync.Once{} // reset the cache
1111
1112         tls12Config := testConfig.Clone()
1113         tls12Config.MaxVersion = VersionTLS12
1114         expectVersion(t, testConfig, testConfig, VersionTLS12)
1115         expectVersion(t, tls12Config, testConfig, VersionTLS12)
1116         expectVersion(t, testConfig, tls12Config, VersionTLS12)
1117         expectVersion(t, tls12Config, tls12Config, VersionTLS12)
1118
1119         os.Setenv("GODEBUG", "tls13=1")
1120         tls13Support.Once = sync.Once{} // reset the cache
1121
1122         expectVersion(t, testConfig, testConfig, VersionTLS13)
1123         expectVersion(t, tls12Config, testConfig, VersionTLS12)
1124         expectVersion(t, testConfig, tls12Config, VersionTLS12)
1125         expectVersion(t, tls12Config, tls12Config, VersionTLS12)
1126 }
1127
1128 // Issue 28744: Ensure that we don't modify memory
1129 // that Config doesn't own such as Certificates.
1130 func TestBuildNameToCertificate_doesntModifyCertificates(t *testing.T) {
1131         c0 := Certificate{
1132                 Certificate: [][]byte{testRSACertificate},
1133                 PrivateKey:  testRSAPrivateKey,
1134         }
1135         c1 := Certificate{
1136                 Certificate: [][]byte{testSNICertificate},
1137                 PrivateKey:  testRSAPrivateKey,
1138         }
1139         config := testConfig.Clone()
1140         config.Certificates = []Certificate{c0, c1}
1141
1142         config.BuildNameToCertificate()
1143         got := config.Certificates
1144         want := []Certificate{c0, c1}
1145         if !reflect.DeepEqual(got, want) {
1146                 t.Fatalf("Certificates were mutated by BuildNameToCertificate\nGot: %#v\nWant: %#v\n", got, want)
1147         }
1148 }