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         "errors"
  11         "fmt"
  12         "internal/testenv"
  13         "io"
  14         "io/ioutil"
  15         "math"
  16         "math/rand"
  17         "net"
  18         "os"
  19         "reflect"
  20         "strings"
  21         "testing"
  22         "testing/quick"
  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 = `-----BEGIN RSA PRIVATE 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 PRIVATE 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 = `-----BEGIN PRIVATE 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 PRIVATE 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 = `-----BEGIN EC PARAMETERS-----
  80 BgUrgQQAIw==
  81 -----END EC PARAMETERS-----
  82 -----BEGIN EC PRIVATE 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 PRIVATE 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         var pem []byte
 103         for _, test := range keyPairTests {
 104                 pem = []byte(test.cert + test.key)
 105                 if _, err := X509KeyPair(pem, pem); err != nil {
 106                         t.Errorf("Failed to load %s cert followed by %s key: %s", test.algo, test.algo, err)
 107                 }
 108                 pem = []byte(test.key + test.cert)
 109                 if _, err := X509KeyPair(pem, pem); err != nil {
 110                         t.Errorf("Failed to load %s key followed by %s cert: %s", test.algo, test.algo, err)
 111                 }
 112         }
 113 }
 114
 115 func TestX509KeyPairErrors(t *testing.T) {
 116         _, err := X509KeyPair([]byte(rsaKeyPEM), []byte(rsaCertPEM))
 117         if err == nil {
 118                 t.Fatalf("X509KeyPair didn't return an error when arguments were switched")
 119         }
 120         if subStr := "been switched"; !strings.Contains(err.Error(), subStr) {
 121                 t.Fatalf("Expected %q in the error when switching arguments to X509KeyPair, but the error was %q", subStr, err)
 122         }
 123
 124         _, err = X509KeyPair([]byte(rsaCertPEM), []byte(rsaCertPEM))
 125         if err == nil {
 126                 t.Fatalf("X509KeyPair didn't return an error when both arguments were certificates")
 127         }
 128         if subStr := "certificate"; !strings.Contains(err.Error(), subStr) {
 129                 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were certificates, but the error was %q", subStr, err)
 130         }
 131
 132         const nonsensePEM = `
 133 -----BEGIN NONSENSE-----
 134 Zm9vZm9vZm9v
 135 -----END NONSENSE-----
 136 `
 137
 138         _, err = X509KeyPair([]byte(nonsensePEM), []byte(nonsensePEM))
 139         if err == nil {
 140                 t.Fatalf("X509KeyPair didn't return an error when both arguments were nonsense")
 141         }
 142         if subStr := "NONSENSE"; !strings.Contains(err.Error(), subStr) {
 143                 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were nonsense, but the error was %q", subStr, err)
 144         }
 145 }
 146
 147 func TestX509MixedKeyPair(t *testing.T) {
 148         if _, err := X509KeyPair([]byte(rsaCertPEM), []byte(ecdsaKeyPEM)); err == nil {
 149                 t.Error("Load of RSA certificate succeeded with ECDSA private key")
 150         }
 151         if _, err := X509KeyPair([]byte(ecdsaCertPEM), []byte(rsaKeyPEM)); err == nil {
 152                 t.Error("Load of ECDSA certificate succeeded with RSA private key")
 153         }
 154 }
 155
 156 func newLocalListener(t testing.TB) net.Listener {
 157         ln, err := net.Listen("tcp", "127.0.0.1:0")
 158         if err != nil {
 159                 ln, err = net.Listen("tcp6", "[::1]:0")
 160         }
 161         if err != nil {
 162                 t.Fatal(err)
 163         }
 164         return ln
 165 }
 166
 167 func TestDialTimeout(t *testing.T) {
 168         if testing.Short() {
 169                 t.Skip("skipping in short mode")
 170         }
 171         listener := newLocalListener(t)
 172
 173         addr := listener.Addr().String()
 174         defer listener.Close()
 175
 176         complete := make(chan bool)
 177         defer close(complete)
 178
 179         go func() {
 180                 conn, err := listener.Accept()
 181                 if err != nil {
 182                         t.Error(err)
 183                         return
 184                 }
 185                 <-complete
 186                 conn.Close()
 187         }()
 188
 189         dialer := &net.Dialer{
 190                 Timeout: 10 * time.Millisecond,
 191         }
 192
 193         var err error
 194         if _, err = DialWithDialer(dialer, "tcp", addr, nil); err == nil {
 195                 t.Fatal("DialWithTimeout completed successfully")
 196         }
 197
 198         if !isTimeoutError(err) {
 199                 t.Errorf("resulting error not a timeout: %v\nType %T: %#v", err, err, err)
 200         }
 201 }
 202
 203 func isTimeoutError(err error) bool {
 204         if ne, ok := err.(net.Error); ok {
 205                 return ne.Timeout()
 206         }
 207         return false
 208 }
 209
 210 // tests that Conn.Read returns (non-zero, io.EOF) instead of
 211 // (non-zero, nil) when a Close (alertCloseNotify) is sitting right
 212 // behind the application data in the buffer.
 213 func TestConnReadNonzeroAndEOF(t *testing.T) {
 214         // This test is racy: it assumes that after a write to a
 215         // localhost TCP connection, the peer TCP connection can
 216         // immediately read it. Because it's racy, we skip this test
 217         // in short mode, and then retry it several times with an
 218         // increasing sleep in between our final write (via srv.Close
 219         // below) and the following read.
 220         if testing.Short() {
 221                 t.Skip("skipping in short mode")
 222         }
 223         var err error
 224         for delay := time.Millisecond; delay <= 64*time.Millisecond; delay *= 2 {
 225                 if err = testConnReadNonzeroAndEOF(t, delay); err == nil {
 226                         return
 227                 }
 228         }
 229         t.Error(err)
 230 }
 231
 232 func testConnReadNonzeroAndEOF(t *testing.T, delay time.Duration) error {
 233         ln := newLocalListener(t)
 234         defer ln.Close()
 235
 236         srvCh := make(chan *Conn, 1)
 237         var serr error
 238         go func() {
 239                 sconn, err := ln.Accept()
 240                 if err != nil {
 241                         serr = err
 242                         srvCh <- nil
 243                         return
 244                 }
 245                 serverConfig := testConfig.Clone()
 246                 srv := Server(sconn, serverConfig)
 247                 if err := srv.Handshake(); err != nil {
 248                         serr = fmt.Errorf("handshake: %v", err)
 249                         srvCh <- nil
 250                         return
 251                 }
 252                 srvCh <- srv
 253         }()
 254
 255         clientConfig := testConfig.Clone()
 256         conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
 257         if err != nil {
 258                 t.Fatal(err)
 259         }
 260         defer conn.Close()
 261
 262         srv := <-srvCh
 263         if srv == nil {
 264                 return serr
 265         }
 266
 267         buf := make([]byte, 6)
 268
 269         srv.Write([]byte("foobar"))
 270         n, err := conn.Read(buf)
 271         if n != 6 || err != nil || string(buf) != "foobar" {
 272                 return fmt.Errorf("Read = %d, %v, data %q; want 6, nil, foobar", n, err, buf)
 273         }
 274
 275         srv.Write([]byte("abcdef"))
 276         srv.Close()
 277         time.Sleep(delay)
 278         n, err = conn.Read(buf)
 279         if n != 6 || string(buf) != "abcdef" {
 280                 return fmt.Errorf("Read = %d, buf= %q; want 6, abcdef", n, buf)
 281         }
 282         if err != io.EOF {
 283                 return fmt.Errorf("Second Read error = %v; want io.EOF", err)
 284         }
 285         return nil
 286 }
 287
 288 func TestTLSUniqueMatches(t *testing.T) {
 289         ln := newLocalListener(t)
 290         defer ln.Close()
 291
 292         serverTLSUniques := make(chan []byte)
 293         go func() {
 294                 for i := 0; i < 2; i++ {
 295                         sconn, err := ln.Accept()
 296                         if err != nil {
 297                                 t.Fatal(err)
 298                         }
 299                         serverConfig := testConfig.Clone()
 300                         srv := Server(sconn, serverConfig)
 301                         if err := srv.Handshake(); err != nil {
 302                                 t.Fatal(err)
 303                         }
 304                         serverTLSUniques <- srv.ConnectionState().TLSUnique
 305                 }
 306         }()
 307
 308         clientConfig := testConfig.Clone()
 309         clientConfig.ClientSessionCache = NewLRUClientSessionCache(1)
 310         conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
 311         if err != nil {
 312                 t.Fatal(err)
 313         }
 314         if !bytes.Equal(conn.ConnectionState().TLSUnique, <-serverTLSUniques) {
 315                 t.Error("client and server channel bindings differ")
 316         }
 317         conn.Close()
 318
 319         conn, err = Dial("tcp", ln.Addr().String(), clientConfig)
 320         if err != nil {
 321                 t.Fatal(err)
 322         }
 323         defer conn.Close()
 324         if !conn.ConnectionState().DidResume {
 325                 t.Error("second session did not use resumption")
 326         }
 327         if !bytes.Equal(conn.ConnectionState().TLSUnique, <-serverTLSUniques) {
 328                 t.Error("client and server channel bindings differ when session resumption is used")
 329         }
 330 }
 331
 332 func TestVerifyHostname(t *testing.T) {
 333         testenv.MustHaveExternalNetwork(t)
 334
 335         c, err := Dial("tcp", "www.google.com:https", nil)
 336         if err != nil {
 337                 t.Fatal(err)
 338         }
 339         if err := c.VerifyHostname("www.google.com"); err != nil {
 340                 t.Fatalf("verify www.google.com: %v", err)
 341         }
 342         if err := c.VerifyHostname("www.yahoo.com"); err == nil {
 343                 t.Fatalf("verify www.yahoo.com succeeded")
 344         }
 345
 346         c, err = Dial("tcp", "www.google.com:https", &Config{InsecureSkipVerify: true})
 347         if err != nil {
 348                 t.Fatal(err)
 349         }
 350         if err := c.VerifyHostname("www.google.com"); err == nil {
 351                 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true")
 352         }
 353         if err := c.VerifyHostname("www.yahoo.com"); err == nil {
 354                 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true")
 355         }
 356 }
 357
 358 func TestVerifyHostnameResumed(t *testing.T) {
 359         testenv.MustHaveExternalNetwork(t)
 360
 361         config := &Config{
 362                 ClientSessionCache: NewLRUClientSessionCache(32),
 363         }
 364         for i := 0; i < 2; i++ {
 365                 c, err := Dial("tcp", "www.google.com:https", config)
 366                 if err != nil {
 367                         t.Fatalf("Dial #%d: %v", i, err)
 368                 }
 369                 cs := c.ConnectionState()
 370                 if i > 0 && !cs.DidResume {
 371                         t.Fatalf("Subsequent connection unexpectedly didn't resume")
 372                 }
 373                 if cs.VerifiedChains == nil {
 374                         t.Fatalf("Dial #%d: cs.VerifiedChains == nil", i)
 375                 }
 376                 if err := c.VerifyHostname("www.google.com"); err != nil {
 377                         t.Fatalf("verify www.google.com #%d: %v", i, err)
 378                 }
 379                 c.Close()
 380         }
 381 }
 382
 383 func TestConnCloseBreakingWrite(t *testing.T) {
 384         ln := newLocalListener(t)
 385         defer ln.Close()
 386
 387         srvCh := make(chan *Conn, 1)
 388         var serr error
 389         var sconn net.Conn
 390         go func() {
 391                 var err error
 392                 sconn, err = ln.Accept()
 393                 if err != nil {
 394                         serr = err
 395                         srvCh <- nil
 396                         return
 397                 }
 398                 serverConfig := testConfig.Clone()
 399                 srv := Server(sconn, serverConfig)
 400                 if err := srv.Handshake(); err != nil {
 401                         serr = fmt.Errorf("handshake: %v", err)
 402                         srvCh <- nil
 403                         return
 404                 }
 405                 srvCh <- srv
 406         }()
 407
 408         cconn, err := net.Dial("tcp", ln.Addr().String())
 409         if err != nil {
 410                 t.Fatal(err)
 411         }
 412         defer cconn.Close()
 413
 414         conn := &changeImplConn{
 415                 Conn: cconn,
 416         }
 417
 418         clientConfig := testConfig.Clone()
 419         tconn := Client(conn, clientConfig)
 420         if err := tconn.Handshake(); err != nil {
 421                 t.Fatal(err)
 422         }
 423
 424         srv := <-srvCh
 425         if srv == nil {
 426                 t.Fatal(serr)
 427         }
 428         defer sconn.Close()
 429
 430         connClosed := make(chan struct{})
 431         conn.closeFunc = func() error {
 432                 close(connClosed)
 433                 return nil
 434         }
 435
 436         inWrite := make(chan bool, 1)
 437         var errConnClosed = errors.New("conn closed for test")
 438         conn.writeFunc = func(p []byte) (n int, err error) {
 439                 inWrite <- true
 440                 <-connClosed
 441                 return 0, errConnClosed
 442         }
 443
 444         closeReturned := make(chan bool, 1)
 445         go func() {
 446                 <-inWrite
 447                 tconn.Close() // test that this doesn't block forever.
 448                 closeReturned <- true
 449         }()
 450
 451         _, err = tconn.Write([]byte("foo"))
 452         if err != errConnClosed {
 453                 t.Errorf("Write error = %v; want errConnClosed", err)
 454         }
 455
 456         <-closeReturned
 457         if err := tconn.Close(); err != errClosed {
 458                 t.Errorf("Close error = %v; want errClosed", err)
 459         }
 460 }
 461
 462 func TestConnCloseWrite(t *testing.T) {
 463         ln := newLocalListener(t)
 464         defer ln.Close()
 465
 466         clientDoneChan := make(chan struct{})
 467
 468         serverCloseWrite := func() error {
 469                 sconn, err := ln.Accept()
 470                 if err != nil {
 471                         return fmt.Errorf("accept: %v", err)
 472                 }
 473                 defer sconn.Close()
 474
 475                 serverConfig := testConfig.Clone()
 476                 srv := Server(sconn, serverConfig)
 477                 if err := srv.Handshake(); err != nil {
 478                         return fmt.Errorf("handshake: %v", err)
 479                 }
 480                 defer srv.Close()
 481
 482                 data, err := ioutil.ReadAll(srv)
 483                 if err != nil {
 484                         return err
 485                 }
 486                 if len(data) > 0 {
 487                         return fmt.Errorf("Read data = %q; want nothing", data)
 488                 }
 489
 490                 if err := srv.CloseWrite(); err != nil {
 491                         return fmt.Errorf("server CloseWrite: %v", err)
 492                 }
 493
 494                 // Wait for clientCloseWrite to finish, so we know we
 495                 // tested the CloseWrite before we defer the
 496                 // sconn.Close above, which would also cause the
 497                 // client to unblock like CloseWrite.
 498                 <-clientDoneChan
 499                 return nil
 500         }
 501
 502         clientCloseWrite := func() error {
 503                 defer close(clientDoneChan)
 504
 505                 clientConfig := testConfig.Clone()
 506                 conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
 507                 if err != nil {
 508                         return err
 509                 }
 510                 if err := conn.Handshake(); err != nil {
 511                         return err
 512                 }
 513                 defer conn.Close()
 514
 515                 if err := conn.CloseWrite(); err != nil {
 516                         return fmt.Errorf("client CloseWrite: %v", err)
 517                 }
 518
 519                 if _, err := conn.Write([]byte{0}); err != errShutdown {
 520                         return fmt.Errorf("CloseWrite error = %v; want errShutdown", err)
 521                 }
 522
 523                 data, err := ioutil.ReadAll(conn)
 524                 if err != nil {
 525                         return err
 526                 }
 527                 if len(data) > 0 {
 528                         return fmt.Errorf("Read data = %q; want nothing", data)
 529                 }
 530                 return nil
 531         }
 532
 533         errChan := make(chan error, 2)
 534
 535         go func() { errChan <- serverCloseWrite() }()
 536         go func() { errChan <- clientCloseWrite() }()
 537
 538         for i := 0; i < 2; i++ {
 539                 select {
 540                 case err := <-errChan:
 541                         if err != nil {
 542                                 t.Fatal(err)
 543                         }
 544                 case <-time.After(10 * time.Second):
 545                         t.Fatal("deadlock")
 546                 }
 547         }
 548
 549         // Also test CloseWrite being called before the handshake is
 550         // finished:
 551         {
 552                 ln2 := newLocalListener(t)
 553                 defer ln2.Close()
 554
 555                 netConn, err := net.Dial("tcp", ln2.Addr().String())
 556                 if err != nil {
 557                         t.Fatal(err)
 558                 }
 559                 defer netConn.Close()
 560                 conn := Client(netConn, testConfig.Clone())
 561
 562                 if err := conn.CloseWrite(); err != errEarlyCloseWrite {
 563                         t.Errorf("CloseWrite error = %v; want errEarlyCloseWrite", err)
 564                 }
 565         }
 566 }
 567
 568 func TestClone(t *testing.T) {
 569         var c1 Config
 570         v := reflect.ValueOf(&c1).Elem()
 571
 572         rnd := rand.New(rand.NewSource(time.Now().Unix()))
 573         typ := v.Type()
 574         for i := 0; i < typ.NumField(); i++ {
 575                 f := v.Field(i)
 576                 if !f.CanSet() {
 577                         // unexported field; not cloned.
 578                         continue
 579                 }
 580
 581                 // testing/quick can't handle functions or interfaces.
 582                 fn := typ.Field(i).Name
 583                 switch fn {
 584                 case "Rand":
 585                         f.Set(reflect.ValueOf(io.Reader(os.Stdin)))
 586                         continue
 587                 case "Time", "GetCertificate", "GetConfigForClient", "VerifyPeerCertificate", "GetClientCertificate":
 588                         // DeepEqual can't compare functions.
 589                         continue
 590                 case "Certificates":
 591                         f.Set(reflect.ValueOf([]Certificate{
 592                                 {Certificate: [][]byte{{'b'}}},
 593                         }))
 594                         continue
 595                 case "NameToCertificate":
 596                         f.Set(reflect.ValueOf(map[string]*Certificate{"a": nil}))
 597                         continue
 598                 case "RootCAs", "ClientCAs":
 599                         f.Set(reflect.ValueOf(x509.NewCertPool()))
 600                         continue
 601                 case "ClientSessionCache":
 602                         f.Set(reflect.ValueOf(NewLRUClientSessionCache(10)))
 603                         continue
 604                 case "KeyLogWriter":
 605                         f.Set(reflect.ValueOf(io.Writer(os.Stdout)))
 606                         continue
 607
 608                 }
 609
 610                 q, ok := quick.Value(f.Type(), rnd)
 611                 if !ok {
 612                         t.Fatalf("quick.Value failed on field %s", fn)
 613                 }
 614                 f.Set(q)
 615         }
 616
 617         c2 := c1.Clone()
 618         // DeepEqual also compares unexported fields, thus c2 needs to have run
 619         // serverInit in order to be DeepEqual to c1. Cloning it and discarding
 620         // the result is sufficient.
 621         c2.Clone()
 622
 623         if !reflect.DeepEqual(&c1, c2) {
 624                 t.Errorf("clone failed to copy a field")
 625         }
 626 }
 627
 628 // changeImplConn is a net.Conn which can change its Write and Close
 629 // methods.
 630 type changeImplConn struct {
 631         net.Conn
 632         writeFunc func([]byte) (int, error)
 633         closeFunc func() error
 634 }
 635
 636 func (w *changeImplConn) Write(p []byte) (n int, err error) {
 637         if w.writeFunc != nil {
 638                 return w.writeFunc(p)
 639         }
 640         return w.Conn.Write(p)
 641 }
 642
 643 func (w *changeImplConn) Close() error {
 644         if w.closeFunc != nil {
 645                 return w.closeFunc()
 646         }
 647         return w.Conn.Close()
 648 }
 649
 650 func throughput(b *testing.B, totalBytes int64, dynamicRecordSizingDisabled bool) {
 651         ln := newLocalListener(b)
 652         defer ln.Close()
 653
 654         N := b.N
 655
 656         // Less than 64KB because Windows appears to use a TCP rwin < 64KB.
 657         // See Issue #15899.
 658         const bufsize = 32 << 10
 659
 660         go func() {
 661                 buf := make([]byte, bufsize)
 662                 for i := 0; i < N; i++ {
 663                         sconn, err := ln.Accept()
 664                         if err != nil {
 665                                 // panic rather than synchronize to avoid benchmark overhead
 666                                 // (cannot call b.Fatal in goroutine)
 667                                 panic(fmt.Errorf("accept: %v", err))
 668                         }
 669                         serverConfig := testConfig.Clone()
 670                         serverConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
 671                         srv := Server(sconn, serverConfig)
 672                         if err := srv.Handshake(); err != nil {
 673                                 panic(fmt.Errorf("handshake: %v", err))
 674                         }
 675                         if _, err := io.CopyBuffer(srv, srv, buf); err != nil {
 676                                 panic(fmt.Errorf("copy buffer: %v", err))
 677                         }
 678                 }
 679         }()
 680
 681         b.SetBytes(totalBytes)
 682         clientConfig := testConfig.Clone()
 683         clientConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
 684
 685         buf := make([]byte, bufsize)
 686         chunks := int(math.Ceil(float64(totalBytes) / float64(len(buf))))
 687         for i := 0; i < N; i++ {
 688                 conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
 689                 if err != nil {
 690                         b.Fatal(err)
 691                 }
 692                 for j := 0; j < chunks; j++ {
 693                         _, err := conn.Write(buf)
 694                         if err != nil {
 695                                 b.Fatal(err)
 696                         }
 697                         _, err = io.ReadFull(conn, buf)
 698                         if err != nil {
 699                                 b.Fatal(err)
 700                         }
 701                 }
 702                 conn.Close()
 703         }
 704 }
 705
 706 func BenchmarkThroughput(b *testing.B) {
 707         for _, mode := range []string{"Max", "Dynamic"} {
 708                 for size := 1; size <= 64; size <<= 1 {
 709                         name := fmt.Sprintf("%sPacket/%dMB", mode, size)
 710                         b.Run(name, func(b *testing.B) {
 711                                 throughput(b, int64(size<<20), mode == "Max")
 712                         })
 713                 }
 714         }
 715 }
 716
 717 type slowConn struct {
 718         net.Conn
 719         bps int
 720 }
 721
 722 func (c *slowConn) Write(p []byte) (int, error) {
 723         if c.bps == 0 {
 724                 panic("too slow")
 725         }
 726         t0 := time.Now()
 727         wrote := 0
 728         for wrote < len(p) {
 729                 time.Sleep(100 * time.Microsecond)
 730                 allowed := int(time.Since(t0).Seconds()*float64(c.bps)) / 8
 731                 if allowed > len(p) {
 732                         allowed = len(p)
 733                 }
 734                 if wrote < allowed {
 735                         n, err := c.Conn.Write(p[wrote:allowed])
 736                         wrote += n
 737                         if err != nil {
 738                                 return wrote, err
 739                         }
 740                 }
 741         }
 742         return len(p), nil
 743 }
 744
 745 func latency(b *testing.B, bps int, dynamicRecordSizingDisabled bool) {
 746         ln := newLocalListener(b)
 747         defer ln.Close()
 748
 749         N := b.N
 750
 751         go func() {
 752                 for i := 0; i < N; i++ {
 753                         sconn, err := ln.Accept()
 754                         if err != nil {
 755                                 // panic rather than synchronize to avoid benchmark overhead
 756                                 // (cannot call b.Fatal in goroutine)
 757                                 panic(fmt.Errorf("accept: %v", err))
 758                         }
 759                         serverConfig := testConfig.Clone()
 760                         serverConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
 761                         srv := Server(&slowConn{sconn, bps}, serverConfig)
 762                         if err := srv.Handshake(); err != nil {
 763                                 panic(fmt.Errorf("handshake: %v", err))
 764                         }
 765                         io.Copy(srv, srv)
 766                 }
 767         }()
 768
 769         clientConfig := testConfig.Clone()
 770         clientConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled
 771
 772         buf := make([]byte, 16384)
 773         peek := make([]byte, 1)
 774
 775         for i := 0; i < N; i++ {
 776                 conn, err := Dial("tcp", ln.Addr().String(), clientConfig)
 777                 if err != nil {
 778                         b.Fatal(err)
 779                 }
 780                 // make sure we're connected and previous connection has stopped
 781                 if _, err := conn.Write(buf[:1]); err != nil {
 782                         b.Fatal(err)
 783                 }
 784                 if _, err := io.ReadFull(conn, peek); err != nil {
 785                         b.Fatal(err)
 786                 }
 787                 if _, err := conn.Write(buf); err != nil {
 788                         b.Fatal(err)
 789                 }
 790                 if _, err = io.ReadFull(conn, peek); err != nil {
 791                         b.Fatal(err)
 792                 }
 793                 conn.Close()
 794         }
 795 }
 796
 797 func BenchmarkLatency(b *testing.B) {
 798         for _, mode := range []string{"Max", "Dynamic"} {
 799                 for _, kbps := range []int{200, 500, 1000, 2000, 5000} {
 800                         name := fmt.Sprintf("%sPacket/%dkbps", mode, kbps)
 801                         b.Run(name, func(b *testing.B) {
 802                                 latency(b, kbps*1000, mode == "Max")
 803                         })
 804                 }
 805         }
 806 }