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