github.com/goproxy0/go@v0.0.0-20171111080102-49cc0c489d2c/src/crypto/tls/tls_test.go (about) 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 "io" 13 "io/ioutil" 14 "math" 15 "net" 16 "os" 17 "reflect" 18 "strings" 19 "testing" 20 "time" 21 ) 22 23 var rsaCertPEM = `-----BEGIN CERTIFICATE----- 24 MIIB0zCCAX2gAwIBAgIJAI/M7BYjwB+uMA0GCSqGSIb3DQEBBQUAMEUxCzAJBgNV 25 BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYDVQQKDBhJbnRlcm5ldCBX 26 aWRnaXRzIFB0eSBMdGQwHhcNMTIwOTEyMjE1MjAyWhcNMTUwOTEyMjE1MjAyWjBF 27 MQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1TdGF0ZTEhMB8GA1UECgwYSW50 28 ZXJuZXQgV2lkZ2l0cyBQdHkgTHRkMFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBANLJ 29 hPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wok/4xIA+ui35/MmNa 30 rtNuC+BdZ1tMuVCPFZcCAwEAAaNQME4wHQYDVR0OBBYEFJvKs8RfJaXTH08W+SGv 31 zQyKn0H8MB8GA1UdIwQYMBaAFJvKs8RfJaXTH08W+SGvzQyKn0H8MAwGA1UdEwQF 32 MAMBAf8wDQYJKoZIhvcNAQEFBQADQQBJlffJHybjDGxRMqaRmDhX0+6v02TUKZsW 33 r5QuVbpQhH6u+0UgcW0jp9QwpxoPTLTWGXEWBBBurxFwiCBhkQ+V 34 -----END CERTIFICATE----- 35 ` 36 37 var rsaKeyPEM = `-----BEGIN RSA PRIVATE KEY----- 38 MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo 39 k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G 40 6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N 41 MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW 42 SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T 43 xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi 44 D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g== 45 -----END RSA PRIVATE KEY----- 46 ` 47 48 // keyPEM is the same as rsaKeyPEM, but declares itself as just 49 // "PRIVATE KEY", not "RSA PRIVATE KEY". https://golang.org/issue/4477 50 var keyPEM = `-----BEGIN PRIVATE KEY----- 51 MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo 52 k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G 53 6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N 54 MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW 55 SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T 56 xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi 57 D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g== 58 -----END PRIVATE KEY----- 59 ` 60 61 var ecdsaCertPEM = `-----BEGIN CERTIFICATE----- 62 MIIB/jCCAWICCQDscdUxw16XFDAJBgcqhkjOPQQBMEUxCzAJBgNVBAYTAkFVMRMw 63 EQYDVQQIEwpTb21lLVN0YXRlMSEwHwYDVQQKExhJbnRlcm5ldCBXaWRnaXRzIFB0 64 eSBMdGQwHhcNMTIxMTE0MTI0MDQ4WhcNMTUxMTE0MTI0MDQ4WjBFMQswCQYDVQQG 65 EwJBVTETMBEGA1UECBMKU29tZS1TdGF0ZTEhMB8GA1UEChMYSW50ZXJuZXQgV2lk 66 Z2l0cyBQdHkgTHRkMIGbMBAGByqGSM49AgEGBSuBBAAjA4GGAAQBY9+my9OoeSUR 67 lDQdV/x8LsOuLilthhiS1Tz4aGDHIPwC1mlvnf7fg5lecYpMCrLLhauAc1UJXcgl 68 01xoLuzgtAEAgv2P/jgytzRSpUYvgLBt1UA0leLYBy6mQQbrNEuqT3INapKIcUv8 69 XxYP0xMEUksLPq6Ca+CRSqTtrd/23uTnapkwCQYHKoZIzj0EAQOBigAwgYYCQXJo 70 A7Sl2nLVf+4Iu/tAX/IF4MavARKC4PPHK3zfuGfPR3oCCcsAoz3kAzOeijvd0iXb 71 H5jBImIxPL4WxQNiBTexAkF8D1EtpYuWdlVQ80/h/f4pBcGiXPqX5h2PQSQY7hP1 72 +jwM1FGS4fREIOvlBYr/SzzQRtwrvrzGYxDEDbsC0ZGRnA== 73 -----END CERTIFICATE----- 74 ` 75 76 var ecdsaKeyPEM = `-----BEGIN EC PARAMETERS----- 77 BgUrgQQAIw== 78 -----END EC PARAMETERS----- 79 -----BEGIN EC PRIVATE KEY----- 80 MIHcAgEBBEIBrsoKp0oqcv6/JovJJDoDVSGWdirrkgCWxrprGlzB9o0X8fV675X0 81 NwuBenXFfeZvVcwluO7/Q9wkYoPd/t3jGImgBwYFK4EEACOhgYkDgYYABAFj36bL 82 06h5JRGUNB1X/Hwuw64uKW2GGJLVPPhoYMcg/ALWaW+d/t+DmV5xikwKssuFq4Bz 83 VQldyCXTXGgu7OC0AQCC/Y/+ODK3NFKlRi+AsG3VQDSV4tgHLqZBBus0S6pPcg1q 84 kohxS/xfFg/TEwRSSws+roJr4JFKpO2t3/be5OdqmQ== 85 -----END EC PRIVATE KEY----- 86 ` 87 88 var keyPairTests = []struct { 89 algo string 90 cert string 91 key string 92 }{ 93 {"ECDSA", ecdsaCertPEM, ecdsaKeyPEM}, 94 {"RSA", rsaCertPEM, rsaKeyPEM}, 95 {"RSA-untyped", rsaCertPEM, keyPEM}, // golang.org/issue/4477 96 } 97 98 func TestX509KeyPair(t *testing.T) { 99 t.Parallel() 100 var pem []byte 101 for _, test := range keyPairTests { 102 pem = []byte(test.cert + test.key) 103 if _, err := X509KeyPair(pem, pem); err != nil { 104 t.Errorf("Failed to load %s cert followed by %s key: %s", test.algo, test.algo, err) 105 } 106 pem = []byte(test.key + test.cert) 107 if _, err := X509KeyPair(pem, pem); err != nil { 108 t.Errorf("Failed to load %s key followed by %s cert: %s", test.algo, test.algo, err) 109 } 110 } 111 } 112 113 func TestX509KeyPairErrors(t *testing.T) { 114 _, err := X509KeyPair([]byte(rsaKeyPEM), []byte(rsaCertPEM)) 115 if err == nil { 116 t.Fatalf("X509KeyPair didn't return an error when arguments were switched") 117 } 118 if subStr := "been switched"; !strings.Contains(err.Error(), subStr) { 119 t.Fatalf("Expected %q in the error when switching arguments to X509KeyPair, but the error was %q", subStr, err) 120 } 121 122 _, err = X509KeyPair([]byte(rsaCertPEM), []byte(rsaCertPEM)) 123 if err == nil { 124 t.Fatalf("X509KeyPair didn't return an error when both arguments were certificates") 125 } 126 if subStr := "certificate"; !strings.Contains(err.Error(), subStr) { 127 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were certificates, but the error was %q", subStr, err) 128 } 129 130 const nonsensePEM = ` 131 -----BEGIN NONSENSE----- 132 Zm9vZm9vZm9v 133 -----END NONSENSE----- 134 ` 135 136 _, err = X509KeyPair([]byte(nonsensePEM), []byte(nonsensePEM)) 137 if err == nil { 138 t.Fatalf("X509KeyPair didn't return an error when both arguments were nonsense") 139 } 140 if subStr := "NONSENSE"; !strings.Contains(err.Error(), subStr) { 141 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were nonsense, but the error was %q", subStr, err) 142 } 143 } 144 145 func TestX509MixedKeyPair(t *testing.T) { 146 if _, err := X509KeyPair([]byte(rsaCertPEM), []byte(ecdsaKeyPEM)); err == nil { 147 t.Error("Load of RSA certificate succeeded with ECDSA private key") 148 } 149 if _, err := X509KeyPair([]byte(ecdsaCertPEM), []byte(rsaKeyPEM)); err == nil { 150 t.Error("Load of ECDSA certificate succeeded with RSA private key") 151 } 152 } 153 154 func newLocalListener(t testing.TB) net.Listener { 155 ln, err := net.Listen("tcp", "127.0.0.1:0") 156 if err != nil { 157 ln, err = net.Listen("tcp6", "[::1]:0") 158 } 159 if err != nil { 160 t.Fatal(err) 161 } 162 return ln 163 } 164 165 func TestDialTimeout(t *testing.T) { 166 if testing.Short() { 167 t.Skip("skipping in short mode") 168 } 169 listener := newLocalListener(t) 170 171 addr := listener.Addr().String() 172 defer listener.Close() 173 174 complete := make(chan bool) 175 defer close(complete) 176 177 go func() { 178 conn, err := listener.Accept() 179 if err != nil { 180 t.Error(err) 181 return 182 } 183 <-complete 184 conn.Close() 185 }() 186 187 dialer := &net.Dialer{ 188 Timeout: 10 * time.Millisecond, 189 } 190 191 var err error 192 if _, err = DialWithDialer(dialer, "tcp", addr, nil); err == nil { 193 t.Fatal("DialWithTimeout completed successfully") 194 } 195 196 if !isTimeoutError(err) { 197 t.Errorf("resulting error not a timeout: %v\nType %T: %#v", err, err, err) 198 } 199 } 200 201 func isTimeoutError(err error) bool { 202 if ne, ok := err.(net.Error); ok { 203 return ne.Timeout() 204 } 205 return false 206 } 207 208 // tests that Conn.Read returns (non-zero, io.EOF) instead of 209 // (non-zero, nil) when a Close (alertCloseNotify) is sitting right 210 // behind the application data in the buffer. 211 func TestConnReadNonzeroAndEOF(t *testing.T) { 212 // This test is racy: it assumes that after a write to a 213 // localhost TCP connection, the peer TCP connection can 214 // immediately read it. Because it's racy, we skip this test 215 // in short mode, and then retry it several times with an 216 // increasing sleep in between our final write (via srv.Close 217 // below) and the following read. 218 if testing.Short() { 219 t.Skip("skipping in short mode") 220 } 221 var err error 222 for delay := time.Millisecond; delay <= 64*time.Millisecond; delay *= 2 { 223 if err = testConnReadNonzeroAndEOF(t, delay); err == nil { 224 return 225 } 226 } 227 t.Error(err) 228 } 229 230 func testConnReadNonzeroAndEOF(t *testing.T, delay time.Duration) error { 231 ln := newLocalListener(t) 232 defer ln.Close() 233 234 srvCh := make(chan *Conn, 1) 235 var serr error 236 go func() { 237 sconn, err := ln.Accept() 238 if err != nil { 239 serr = err 240 srvCh <- nil 241 return 242 } 243 serverConfig := testConfig.Clone() 244 srv := Server(sconn, serverConfig) 245 if err := srv.Handshake(); err != nil { 246 serr = fmt.Errorf("handshake: %v", err) 247 srvCh <- nil 248 return 249 } 250 srvCh <- srv 251 }() 252 253 clientConfig := testConfig.Clone() 254 conn, err := Dial("tcp", ln.Addr().String(), clientConfig) 255 if err != nil { 256 t.Fatal(err) 257 } 258 defer conn.Close() 259 260 srv := <-srvCh 261 if srv == nil { 262 return serr 263 } 264 265 buf := make([]byte, 6) 266 267 srv.Write([]byte("foobar")) 268 n, err := conn.Read(buf) 269 if n != 6 || err != nil || string(buf) != "foobar" { 270 return fmt.Errorf("Read = %d, %v, data %q; want 6, nil, foobar", n, err, buf) 271 } 272 273 srv.Write([]byte("abcdef")) 274 srv.Close() 275 time.Sleep(delay) 276 n, err = conn.Read(buf) 277 if n != 6 || string(buf) != "abcdef" { 278 return fmt.Errorf("Read = %d, buf= %q; want 6, abcdef", n, buf) 279 } 280 if err != io.EOF { 281 return fmt.Errorf("Second Read error = %v; want io.EOF", err) 282 } 283 return nil 284 } 285 286 func TestTLSUniqueMatches(t *testing.T) { 287 ln := newLocalListener(t) 288 defer ln.Close() 289 290 serverTLSUniques := make(chan []byte) 291 go func() { 292 for i := 0; i < 2; i++ { 293 sconn, err := ln.Accept() 294 if err != nil { 295 t.Error(err) 296 return 297 } 298 serverConfig := testConfig.Clone() 299 srv := Server(sconn, serverConfig) 300 if err := srv.Handshake(); err != nil { 301 t.Error(err) 302 return 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 c, err := Dial("tcp", "www.google.com:https", nil) 334 if err != nil { 335 t.Fatal(err) 336 } 337 if err := c.VerifyHostname("www.google.com"); err != nil { 338 t.Fatalf("verify www.google.com: %v", err) 339 } 340 if err := c.VerifyHostname("www.yahoo.com"); err == nil { 341 t.Fatalf("verify www.yahoo.com succeeded") 342 } 343 344 c, err = Dial("tcp", "www.google.com:https", &Config{InsecureSkipVerify: true}) 345 if err != nil { 346 t.Fatal(err) 347 } 348 if err := c.VerifyHostname("www.google.com"); err == nil { 349 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true") 350 } 351 if err := c.VerifyHostname("www.yahoo.com"); err == nil { 352 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true") 353 } 354 } 355 356 func TestVerifyHostnameResumed(t *testing.T) { 357 config := &Config{ 358 ClientSessionCache: NewLRUClientSessionCache(32), 359 } 360 for i := 0; i < 2; i++ { 361 c, err := Dial("tcp", "www.google.com:https", config) 362 if err != nil { 363 t.Fatalf("Dial #%d: %v", i, err) 364 } 365 cs := c.ConnectionState() 366 if i > 0 && !cs.DidResume { 367 t.Fatalf("Subsequent connection unexpectedly didn't resume") 368 } 369 if cs.VerifiedChains == nil { 370 t.Fatalf("Dial #%d: cs.VerifiedChains == nil", i) 371 } 372 if err := c.VerifyHostname("www.google.com"); err != nil { 373 t.Fatalf("verify www.google.com #%d: %v", i, err) 374 } 375 c.Close() 376 } 377 } 378 379 func TestConnCloseBreakingWrite(t *testing.T) { 380 ln := newLocalListener(t) 381 defer ln.Close() 382 383 srvCh := make(chan *Conn, 1) 384 var serr error 385 var sconn net.Conn 386 go func() { 387 var err error 388 sconn, err = ln.Accept() 389 if err != nil { 390 serr = err 391 srvCh <- nil 392 return 393 } 394 serverConfig := testConfig.Clone() 395 srv := Server(sconn, serverConfig) 396 if err := srv.Handshake(); err != nil { 397 serr = fmt.Errorf("handshake: %v", err) 398 srvCh <- nil 399 return 400 } 401 srvCh <- srv 402 }() 403 404 cconn, err := net.Dial("tcp", ln.Addr().String()) 405 if err != nil { 406 t.Fatal(err) 407 } 408 defer cconn.Close() 409 410 conn := &changeImplConn{ 411 Conn: cconn, 412 } 413 414 clientConfig := testConfig.Clone() 415 tconn := Client(conn, clientConfig) 416 if err := tconn.Handshake(); err != nil { 417 t.Fatal(err) 418 } 419 420 srv := <-srvCh 421 if srv == nil { 422 t.Fatal(serr) 423 } 424 defer sconn.Close() 425 426 connClosed := make(chan struct{}) 427 conn.closeFunc = func() error { 428 close(connClosed) 429 return nil 430 } 431 432 inWrite := make(chan bool, 1) 433 var errConnClosed = errors.New("conn closed for test") 434 conn.writeFunc = func(p []byte) (n int, err error) { 435 inWrite <- true 436 <-connClosed 437 return 0, errConnClosed 438 } 439 440 closeReturned := make(chan bool, 1) 441 go func() { 442 <-inWrite 443 tconn.Close() // test that this doesn't block forever. 444 closeReturned <- true 445 }() 446 447 _, err = tconn.Write([]byte("foo")) 448 if err != errConnClosed { 449 t.Errorf("Write error = %v; want errConnClosed", err) 450 } 451 452 <-closeReturned 453 if err := tconn.Close(); err != errClosed { 454 t.Errorf("Close error = %v; want errClosed", err) 455 } 456 } 457 458 func TestConnCloseWrite(t *testing.T) { 459 ln := newLocalListener(t) 460 defer ln.Close() 461 462 clientDoneChan := make(chan struct{}) 463 464 serverCloseWrite := func() error { 465 sconn, err := ln.Accept() 466 if err != nil { 467 return fmt.Errorf("accept: %v", err) 468 } 469 defer sconn.Close() 470 471 serverConfig := testConfig.Clone() 472 srv := Server(sconn, serverConfig) 473 if err := srv.Handshake(); err != nil { 474 return fmt.Errorf("handshake: %v", err) 475 } 476 defer srv.Close() 477 478 data, err := ioutil.ReadAll(srv) 479 if err != nil { 480 return err 481 } 482 if len(data) > 0 { 483 return fmt.Errorf("Read data = %q; want nothing", data) 484 } 485 486 if err := srv.CloseWrite(); err != nil { 487 return fmt.Errorf("server CloseWrite: %v", err) 488 } 489 490 // Wait for clientCloseWrite to finish, so we know we 491 // tested the CloseWrite before we defer the 492 // sconn.Close above, which would also cause the 493 // client to unblock like CloseWrite. 494 <-clientDoneChan 495 return nil 496 } 497 498 clientCloseWrite := func() error { 499 defer close(clientDoneChan) 500 501 clientConfig := testConfig.Clone() 502 conn, err := Dial("tcp", ln.Addr().String(), clientConfig) 503 if err != nil { 504 return err 505 } 506 if err := conn.Handshake(); err != nil { 507 return err 508 } 509 defer conn.Close() 510 511 if err := conn.CloseWrite(); err != nil { 512 return fmt.Errorf("client CloseWrite: %v", err) 513 } 514 515 if _, err := conn.Write([]byte{0}); err != errShutdown { 516 return fmt.Errorf("CloseWrite error = %v; want errShutdown", err) 517 } 518 519 data, err := ioutil.ReadAll(conn) 520 if err != nil { 521 return err 522 } 523 if len(data) > 0 { 524 return fmt.Errorf("Read data = %q; want nothing", data) 525 } 526 return nil 527 } 528 529 errChan := make(chan error, 2) 530 531 go func() { errChan <- serverCloseWrite() }() 532 go func() { errChan <- clientCloseWrite() }() 533 534 for i := 0; i < 2; i++ { 535 select { 536 case err := <-errChan: 537 if err != nil { 538 t.Fatal(err) 539 } 540 case <-time.After(10 * time.Second): 541 t.Fatal("deadlock") 542 } 543 } 544 545 // Also test CloseWrite being called before the handshake is 546 // finished: 547 { 548 ln2 := newLocalListener(t) 549 defer ln2.Close() 550 551 netConn, err := net.Dial("tcp", ln2.Addr().String()) 552 if err != nil { 553 t.Fatal(err) 554 } 555 defer netConn.Close() 556 conn := Client(netConn, testConfig.Clone()) 557 558 if err := conn.CloseWrite(); err != errEarlyCloseWrite { 559 t.Errorf("CloseWrite error = %v; want errEarlyCloseWrite", err) 560 } 561 } 562 } 563 564 func TestWarningAlertFlood(t *testing.T) { 565 ln := newLocalListener(t) 566 defer ln.Close() 567 568 server := func() error { 569 sconn, err := ln.Accept() 570 if err != nil { 571 return fmt.Errorf("accept: %v", err) 572 } 573 defer sconn.Close() 574 575 serverConfig := testConfig.Clone() 576 srv := Server(sconn, serverConfig) 577 if err := srv.Handshake(); err != nil { 578 return fmt.Errorf("handshake: %v", err) 579 } 580 defer srv.Close() 581 582 _, err = ioutil.ReadAll(srv) 583 if err == nil { 584 return errors.New("unexpected lack of error from server") 585 } 586 const expected = "too many warn" 587 if str := err.Error(); !strings.Contains(str, expected) { 588 return fmt.Errorf("expected error containing %q, but saw: %s", expected, str) 589 } 590 591 return nil 592 } 593 594 errChan := make(chan error, 1) 595 go func() { errChan <- server() }() 596 597 clientConfig := testConfig.Clone() 598 conn, err := Dial("tcp", ln.Addr().String(), clientConfig) 599 if err != nil { 600 t.Fatal(err) 601 } 602 defer conn.Close() 603 if err := conn.Handshake(); err != nil { 604 t.Fatal(err) 605 } 606 607 for i := 0; i < maxWarnAlertCount+1; i++ { 608 conn.sendAlert(alertNoRenegotiation) 609 } 610 611 if err := <-errChan; err != nil { 612 t.Fatal(err) 613 } 614 } 615 616 func TestCloneFuncFields(t *testing.T) { 617 const expectedCount = 5 618 called := 0 619 620 c1 := Config{ 621 Time: func() time.Time { 622 called |= 1 << 0 623 return time.Time{} 624 }, 625 GetCertificate: func(*ClientHelloInfo) (*Certificate, error) { 626 called |= 1 << 1 627 return nil, nil 628 }, 629 GetClientCertificate: func(*CertificateRequestInfo) (*Certificate, error) { 630 called |= 1 << 2 631 return nil, nil 632 }, 633 GetConfigForClient: func(*ClientHelloInfo) (*Config, error) { 634 called |= 1 << 3 635 return nil, nil 636 }, 637 VerifyPeerCertificate: func(rawCerts [][]byte, verifiedChains [][]*x509.Certificate) error { 638 called |= 1 << 4 639 return nil 640 }, 641 } 642 643 c2 := c1.Clone() 644 645 c2.Time() 646 c2.GetCertificate(nil) 647 c2.GetClientCertificate(nil) 648 c2.GetConfigForClient(nil) 649 c2.VerifyPeerCertificate(nil, nil) 650 651 if called != (1<<expectedCount)-1 { 652 t.Fatalf("expected %d calls but saw calls %b", expectedCount, called) 653 } 654 } 655 656 func TestCloneNonFuncFields(t *testing.T) { 657 var c1 Config 658 v := reflect.ValueOf(&c1).Elem() 659 660 typ := v.Type() 661 for i := 0; i < typ.NumField(); i++ { 662 f := v.Field(i) 663 if !f.CanSet() { 664 // unexported field; not cloned. 665 continue 666 } 667 668 // testing/quick can't handle functions or interfaces and so 669 // isn't used here. 670 switch fn := typ.Field(i).Name; fn { 671 case "Rand": 672 f.Set(reflect.ValueOf(io.Reader(os.Stdin))) 673 case "Time", "GetCertificate", "GetConfigForClient", "VerifyPeerCertificate", "GetClientCertificate": 674 // DeepEqual can't compare functions. If you add a 675 // function field to this list, you must also change 676 // TestCloneFuncFields to ensure that the func field is 677 // cloned. 678 case "Certificates": 679 f.Set(reflect.ValueOf([]Certificate{ 680 {Certificate: [][]byte{{'b'}}}, 681 })) 682 case "NameToCertificate": 683 f.Set(reflect.ValueOf(map[string]*Certificate{"a": nil})) 684 case "RootCAs", "ClientCAs": 685 f.Set(reflect.ValueOf(x509.NewCertPool())) 686 case "ClientSessionCache": 687 f.Set(reflect.ValueOf(NewLRUClientSessionCache(10))) 688 case "KeyLogWriter": 689 f.Set(reflect.ValueOf(io.Writer(os.Stdout))) 690 case "NextProtos": 691 f.Set(reflect.ValueOf([]string{"a", "b"})) 692 case "ServerName": 693 f.Set(reflect.ValueOf("b")) 694 case "ClientAuth": 695 f.Set(reflect.ValueOf(VerifyClientCertIfGiven)) 696 case "InsecureSkipVerify", "SessionTicketsDisabled", "DynamicRecordSizingDisabled", "PreferServerCipherSuites", "Accept0RTTData": 697 f.Set(reflect.ValueOf(true)) 698 case "MinVersion", "MaxVersion": 699 f.Set(reflect.ValueOf(uint16(VersionTLS12))) 700 case "SessionTicketKey": 701 f.Set(reflect.ValueOf([32]byte{})) 702 case "CipherSuites": 703 f.Set(reflect.ValueOf([]uint16{1, 2})) 704 case "CurvePreferences": 705 f.Set(reflect.ValueOf([]CurveID{CurveP256})) 706 case "Renegotiation": 707 f.Set(reflect.ValueOf(RenegotiateOnceAsClient)) 708 case "Max0RTTDataSize": 709 f.Set(reflect.ValueOf(uint32(0))) 710 case "SessionTicketSealer": 711 // TODO 712 default: 713 t.Errorf("all fields must be accounted for, but saw unknown field %q", fn) 714 } 715 } 716 717 c2 := c1.Clone() 718 // DeepEqual also compares unexported fields, thus c2 needs to have run 719 // serverInit in order to be DeepEqual to c1. Cloning it and discarding 720 // the result is sufficient. 721 c2.Clone() 722 723 if !reflect.DeepEqual(&c1, c2) { 724 t.Errorf("clone failed to copy a field") 725 } 726 } 727 728 // changeImplConn is a net.Conn which can change its Write and Close 729 // methods. 730 type changeImplConn struct { 731 net.Conn 732 writeFunc func([]byte) (int, error) 733 closeFunc func() error 734 } 735 736 func (w *changeImplConn) Write(p []byte) (n int, err error) { 737 if w.writeFunc != nil { 738 return w.writeFunc(p) 739 } 740 return w.Conn.Write(p) 741 } 742 743 func (w *changeImplConn) Close() error { 744 if w.closeFunc != nil { 745 return w.closeFunc() 746 } 747 return w.Conn.Close() 748 } 749 750 func throughput(b *testing.B, totalBytes int64, dynamicRecordSizingDisabled bool) { 751 ln := newLocalListener(b) 752 defer ln.Close() 753 754 N := b.N 755 756 // Less than 64KB because Windows appears to use a TCP rwin < 64KB. 757 // See Issue #15899. 758 const bufsize = 32 << 10 759 760 go func() { 761 buf := make([]byte, bufsize) 762 for i := 0; i < N; i++ { 763 sconn, err := ln.Accept() 764 if err != nil { 765 // panic rather than synchronize to avoid benchmark overhead 766 // (cannot call b.Fatal in goroutine) 767 panic(fmt.Errorf("accept: %v", err)) 768 } 769 serverConfig := testConfig.Clone() 770 serverConfig.CipherSuites = nil // the defaults may prefer faster ciphers 771 serverConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled 772 srv := Server(sconn, serverConfig) 773 if err := srv.Handshake(); err != nil { 774 panic(fmt.Errorf("handshake: %v", err)) 775 } 776 if _, err := io.CopyBuffer(srv, srv, buf); err != nil { 777 panic(fmt.Errorf("copy buffer: %v", err)) 778 } 779 } 780 }() 781 782 b.SetBytes(totalBytes) 783 clientConfig := testConfig.Clone() 784 clientConfig.CipherSuites = nil // the defaults may prefer faster ciphers 785 clientConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled 786 787 buf := make([]byte, bufsize) 788 chunks := int(math.Ceil(float64(totalBytes) / float64(len(buf)))) 789 for i := 0; i < N; i++ { 790 conn, err := Dial("tcp", ln.Addr().String(), clientConfig) 791 if err != nil { 792 b.Fatal(err) 793 } 794 for j := 0; j < chunks; j++ { 795 _, err := conn.Write(buf) 796 if err != nil { 797 b.Fatal(err) 798 } 799 _, err = io.ReadFull(conn, buf) 800 if err != nil { 801 b.Fatal(err) 802 } 803 } 804 conn.Close() 805 } 806 } 807 808 func BenchmarkThroughput(b *testing.B) { 809 for _, mode := range []string{"Max", "Dynamic"} { 810 for size := 1; size <= 64; size <<= 1 { 811 name := fmt.Sprintf("%sPacket/%dMB", mode, size) 812 b.Run(name, func(b *testing.B) { 813 throughput(b, int64(size<<20), mode == "Max") 814 }) 815 } 816 } 817 } 818 819 type slowConn struct { 820 net.Conn 821 bps int 822 } 823 824 func (c *slowConn) Write(p []byte) (int, error) { 825 if c.bps == 0 { 826 panic("too slow") 827 } 828 t0 := time.Now() 829 wrote := 0 830 for wrote < len(p) { 831 time.Sleep(100 * time.Microsecond) 832 allowed := int(time.Since(t0).Seconds()*float64(c.bps)) / 8 833 if allowed > len(p) { 834 allowed = len(p) 835 } 836 if wrote < allowed { 837 n, err := c.Conn.Write(p[wrote:allowed]) 838 wrote += n 839 if err != nil { 840 return wrote, err 841 } 842 } 843 } 844 return len(p), nil 845 } 846 847 func latency(b *testing.B, bps int, dynamicRecordSizingDisabled bool) { 848 ln := newLocalListener(b) 849 defer ln.Close() 850 851 N := b.N 852 853 go func() { 854 for i := 0; i < N; i++ { 855 sconn, err := ln.Accept() 856 if err != nil { 857 // panic rather than synchronize to avoid benchmark overhead 858 // (cannot call b.Fatal in goroutine) 859 panic(fmt.Errorf("accept: %v", err)) 860 } 861 serverConfig := testConfig.Clone() 862 serverConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled 863 srv := Server(&slowConn{sconn, bps}, serverConfig) 864 if err := srv.Handshake(); err != nil { 865 panic(fmt.Errorf("handshake: %v", err)) 866 } 867 io.Copy(srv, srv) 868 } 869 }() 870 871 clientConfig := testConfig.Clone() 872 clientConfig.DynamicRecordSizingDisabled = dynamicRecordSizingDisabled 873 874 buf := make([]byte, 16384) 875 peek := make([]byte, 1) 876 877 for i := 0; i < N; i++ { 878 conn, err := Dial("tcp", ln.Addr().String(), clientConfig) 879 if err != nil { 880 b.Fatal(err) 881 } 882 // make sure we're connected and previous connection has stopped 883 if _, err := conn.Write(buf[:1]); err != nil { 884 b.Fatal(err) 885 } 886 if _, err := io.ReadFull(conn, peek); err != nil { 887 b.Fatal(err) 888 } 889 if _, err := conn.Write(buf); err != nil { 890 b.Fatal(err) 891 } 892 if _, err = io.ReadFull(conn, peek); err != nil { 893 b.Fatal(err) 894 } 895 conn.Close() 896 } 897 } 898 899 func BenchmarkLatency(b *testing.B) { 900 for _, mode := range []string{"Max", "Dynamic"} { 901 for _, kbps := range []int{200, 500, 1000, 2000, 5000} { 902 name := fmt.Sprintf("%sPacket/%dkbps", mode, kbps) 903 b.Run(name, func(b *testing.B) { 904 latency(b, kbps*1000, mode == "Max") 905 }) 906 } 907 } 908 }