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         "errors"
  10         "fmt"
  11         "internal/testenv"
  12         "io"
  13         "net"
  14         "strings"
  15         "testing"
  16         "time"
  17 )
  18
  19 var rsaCertPEM = `-----BEGIN CERTIFICATE-----
  20 MIIB0zCCAX2gAwIBAgIJAI/M7BYjwB+uMA0GCSqGSIb3DQEBBQUAMEUxCzAJBgNV
  21 BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYDVQQKDBhJbnRlcm5ldCBX
  22 aWRnaXRzIFB0eSBMdGQwHhcNMTIwOTEyMjE1MjAyWhcNMTUwOTEyMjE1MjAyWjBF
  23 MQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1TdGF0ZTEhMB8GA1UECgwYSW50
  24 ZXJuZXQgV2lkZ2l0cyBQdHkgTHRkMFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBANLJ
  25 hPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wok/4xIA+ui35/MmNa
  26 rtNuC+BdZ1tMuVCPFZcCAwEAAaNQME4wHQYDVR0OBBYEFJvKs8RfJaXTH08W+SGv
  27 zQyKn0H8MB8GA1UdIwQYMBaAFJvKs8RfJaXTH08W+SGvzQyKn0H8MAwGA1UdEwQF
  28 MAMBAf8wDQYJKoZIhvcNAQEFBQADQQBJlffJHybjDGxRMqaRmDhX0+6v02TUKZsW
  29 r5QuVbpQhH6u+0UgcW0jp9QwpxoPTLTWGXEWBBBurxFwiCBhkQ+V
  30 -----END CERTIFICATE-----
  31 `
  32
  33 var rsaKeyPEM = `-----BEGIN RSA PRIVATE KEY-----
  34 MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo
  35 k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G
  36 6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N
  37 MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW
  38 SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T
  39 xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi
  40 D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g==
  41 -----END RSA PRIVATE KEY-----
  42 `
  43
  44 // keyPEM is the same as rsaKeyPEM, but declares itself as just
  45 // "PRIVATE KEY", not "RSA PRIVATE KEY".  https://golang.org/issue/4477
  46 var keyPEM = `-----BEGIN PRIVATE KEY-----
  47 MIIBOwIBAAJBANLJhPHhITqQbPklG3ibCVxwGMRfp/v4XqhfdQHdcVfHap6NQ5Wo
  48 k/4xIA+ui35/MmNartNuC+BdZ1tMuVCPFZcCAwEAAQJAEJ2N+zsR0Xn8/Q6twa4G
  49 6OB1M1WO+k+ztnX/1SvNeWu8D6GImtupLTYgjZcHufykj09jiHmjHx8u8ZZB/o1N
  50 MQIhAPW+eyZo7ay3lMz1V01WVjNKK9QSn1MJlb06h/LuYv9FAiEA25WPedKgVyCW
  51 SmUwbPw8fnTcpqDWE3yTO3vKcebqMSsCIBF3UmVue8YU3jybC3NxuXq3wNm34R8T
  52 xVLHwDXh/6NJAiEAl2oHGGLz64BuAfjKrqwz7qMYr9HCLIe/YsoWq/olzScCIQDi
  53 D2lWusoe2/nEqfDVVWGWlyJ7yOmqaVm/iNUN9B2N2g==
  54 -----END PRIVATE KEY-----
  55 `
  56
  57 var ecdsaCertPEM = `-----BEGIN CERTIFICATE-----
  58 MIIB/jCCAWICCQDscdUxw16XFDAJBgcqhkjOPQQBMEUxCzAJBgNVBAYTAkFVMRMw
  59 EQYDVQQIEwpTb21lLVN0YXRlMSEwHwYDVQQKExhJbnRlcm5ldCBXaWRnaXRzIFB0
  60 eSBMdGQwHhcNMTIxMTE0MTI0MDQ4WhcNMTUxMTE0MTI0MDQ4WjBFMQswCQYDVQQG
  61 EwJBVTETMBEGA1UECBMKU29tZS1TdGF0ZTEhMB8GA1UEChMYSW50ZXJuZXQgV2lk
  62 Z2l0cyBQdHkgTHRkMIGbMBAGByqGSM49AgEGBSuBBAAjA4GGAAQBY9+my9OoeSUR
  63 lDQdV/x8LsOuLilthhiS1Tz4aGDHIPwC1mlvnf7fg5lecYpMCrLLhauAc1UJXcgl
  64 01xoLuzgtAEAgv2P/jgytzRSpUYvgLBt1UA0leLYBy6mQQbrNEuqT3INapKIcUv8
  65 XxYP0xMEUksLPq6Ca+CRSqTtrd/23uTnapkwCQYHKoZIzj0EAQOBigAwgYYCQXJo
  66 A7Sl2nLVf+4Iu/tAX/IF4MavARKC4PPHK3zfuGfPR3oCCcsAoz3kAzOeijvd0iXb
  67 H5jBImIxPL4WxQNiBTexAkF8D1EtpYuWdlVQ80/h/f4pBcGiXPqX5h2PQSQY7hP1
  68 +jwM1FGS4fREIOvlBYr/SzzQRtwrvrzGYxDEDbsC0ZGRnA==
  69 -----END CERTIFICATE-----
  70 `
  71
  72 var ecdsaKeyPEM = `-----BEGIN EC PARAMETERS-----
  73 BgUrgQQAIw==
  74 -----END EC PARAMETERS-----
  75 -----BEGIN EC PRIVATE KEY-----
  76 MIHcAgEBBEIBrsoKp0oqcv6/JovJJDoDVSGWdirrkgCWxrprGlzB9o0X8fV675X0
  77 NwuBenXFfeZvVcwluO7/Q9wkYoPd/t3jGImgBwYFK4EEACOhgYkDgYYABAFj36bL
  78 06h5JRGUNB1X/Hwuw64uKW2GGJLVPPhoYMcg/ALWaW+d/t+DmV5xikwKssuFq4Bz
  79 VQldyCXTXGgu7OC0AQCC/Y/+ODK3NFKlRi+AsG3VQDSV4tgHLqZBBus0S6pPcg1q
  80 kohxS/xfFg/TEwRSSws+roJr4JFKpO2t3/be5OdqmQ==
  81 -----END EC PRIVATE KEY-----
  82 `
  83
  84 var keyPairTests = []struct {
  85         algo string
  86         cert string
  87         key  string
  88 }{
  89         {"ECDSA", ecdsaCertPEM, ecdsaKeyPEM},
  90         {"RSA", rsaCertPEM, rsaKeyPEM},
  91         {"RSA-untyped", rsaCertPEM, keyPEM}, // golang.org/issue/4477
  92 }
  93
  94 func TestX509KeyPair(t *testing.T) {
  95         var pem []byte
  96         for _, test := range keyPairTests {
  97                 pem = []byte(test.cert + test.key)
  98                 if _, err := X509KeyPair(pem, pem); err != nil {
  99                         t.Errorf("Failed to load %s cert followed by %s key: %s", test.algo, test.algo, err)
 100                 }
 101                 pem = []byte(test.key + test.cert)
 102                 if _, err := X509KeyPair(pem, pem); err != nil {
 103                         t.Errorf("Failed to load %s key followed by %s cert: %s", test.algo, test.algo, err)
 104                 }
 105         }
 106 }
 107
 108 func TestX509KeyPairErrors(t *testing.T) {
 109         _, err := X509KeyPair([]byte(rsaKeyPEM), []byte(rsaCertPEM))
 110         if err == nil {
 111                 t.Fatalf("X509KeyPair didn't return an error when arguments were switched")
 112         }
 113         if subStr := "been switched"; !strings.Contains(err.Error(), subStr) {
 114                 t.Fatalf("Expected %q in the error when switching arguments to X509KeyPair, but the error was %q", subStr, err)
 115         }
 116
 117         _, err = X509KeyPair([]byte(rsaCertPEM), []byte(rsaCertPEM))
 118         if err == nil {
 119                 t.Fatalf("X509KeyPair didn't return an error when both arguments were certificates")
 120         }
 121         if subStr := "certificate"; !strings.Contains(err.Error(), subStr) {
 122                 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were certificates, but the error was %q", subStr, err)
 123         }
 124
 125         const nonsensePEM = `
 126 -----BEGIN NONSENSE-----
 127 Zm9vZm9vZm9v
 128 -----END NONSENSE-----
 129 `
 130
 131         _, err = X509KeyPair([]byte(nonsensePEM), []byte(nonsensePEM))
 132         if err == nil {
 133                 t.Fatalf("X509KeyPair didn't return an error when both arguments were nonsense")
 134         }
 135         if subStr := "NONSENSE"; !strings.Contains(err.Error(), subStr) {
 136                 t.Fatalf("Expected %q in the error when both arguments to X509KeyPair were nonsense, but the error was %q", subStr, err)
 137         }
 138 }
 139
 140 func TestX509MixedKeyPair(t *testing.T) {
 141         if _, err := X509KeyPair([]byte(rsaCertPEM), []byte(ecdsaKeyPEM)); err == nil {
 142                 t.Error("Load of RSA certificate succeeded with ECDSA private key")
 143         }
 144         if _, err := X509KeyPair([]byte(ecdsaCertPEM), []byte(rsaKeyPEM)); err == nil {
 145                 t.Error("Load of ECDSA certificate succeeded with RSA private key")
 146         }
 147 }
 148
 149 func newLocalListener(t *testing.T) net.Listener {
 150         ln, err := net.Listen("tcp", "127.0.0.1:0")
 151         if err != nil {
 152                 ln, err = net.Listen("tcp6", "[::1]:0")
 153         }
 154         if err != nil {
 155                 t.Fatal(err)
 156         }
 157         return ln
 158 }
 159
 160 func TestDialTimeout(t *testing.T) {
 161         if testing.Short() {
 162                 t.Skip("skipping in short mode")
 163         }
 164         listener := newLocalListener(t)
 165
 166         addr := listener.Addr().String()
 167         defer listener.Close()
 168
 169         complete := make(chan bool)
 170         defer close(complete)
 171
 172         go func() {
 173                 conn, err := listener.Accept()
 174                 if err != nil {
 175                         t.Error(err)
 176                         return
 177                 }
 178                 <-complete
 179                 conn.Close()
 180         }()
 181
 182         dialer := &net.Dialer{
 183                 Timeout: 10 * time.Millisecond,
 184         }
 185
 186         var err error
 187         if _, err = DialWithDialer(dialer, "tcp", addr, nil); err == nil {
 188                 t.Fatal("DialWithTimeout completed successfully")
 189         }
 190
 191         if !isTimeoutError(err) {
 192                 t.Errorf("resulting error not a timeout: %v\nType %T: %#v", err, err, err)
 193         }
 194 }
 195
 196 func isTimeoutError(err error) bool {
 197         if ne, ok := err.(net.Error); ok {
 198                 return ne.Timeout()
 199         }
 200         return false
 201 }
 202
 203 // tests that Conn.Read returns (non-zero, io.EOF) instead of
 204 // (non-zero, nil) when a Close (alertCloseNotify) is sitting right
 205 // behind the application data in the buffer.
 206 func TestConnReadNonzeroAndEOF(t *testing.T) {
 207         // This test is racy: it assumes that after a write to a
 208         // localhost TCP connection, the peer TCP connection can
 209         // immediately read it. Because it's racy, we skip this test
 210         // in short mode, and then retry it several times with an
 211         // increasing sleep in between our final write (via srv.Close
 212         // below) and the following read.
 213         if testing.Short() {
 214                 t.Skip("skipping in short mode")
 215         }
 216         var err error
 217         for delay := time.Millisecond; delay <= 64*time.Millisecond; delay *= 2 {
 218                 if err = testConnReadNonzeroAndEOF(t, delay); err == nil {
 219                         return
 220                 }
 221         }
 222         t.Error(err)
 223 }
 224
 225 func testConnReadNonzeroAndEOF(t *testing.T, delay time.Duration) error {
 226         ln := newLocalListener(t)
 227         defer ln.Close()
 228
 229         srvCh := make(chan *Conn, 1)
 230         var serr error
 231         go func() {
 232                 sconn, err := ln.Accept()
 233                 if err != nil {
 234                         serr = err
 235                         srvCh <- nil
 236                         return
 237                 }
 238                 serverConfig := *testConfig
 239                 srv := Server(sconn, &serverConfig)
 240                 if err := srv.Handshake(); err != nil {
 241                         serr = fmt.Errorf("handshake: %v", err)
 242                         srvCh <- nil
 243                         return
 244                 }
 245                 srvCh <- srv
 246         }()
 247
 248         clientConfig := *testConfig
 249         conn, err := Dial("tcp", ln.Addr().String(), &clientConfig)
 250         if err != nil {
 251                 t.Fatal(err)
 252         }
 253         defer conn.Close()
 254
 255         srv := <-srvCh
 256         if srv == nil {
 257                 return serr
 258         }
 259
 260         buf := make([]byte, 6)
 261
 262         srv.Write([]byte("foobar"))
 263         n, err := conn.Read(buf)
 264         if n != 6 || err != nil || string(buf) != "foobar" {
 265                 return fmt.Errorf("Read = %d, %v, data %q; want 6, nil, foobar", n, err, buf)
 266         }
 267
 268         srv.Write([]byte("abcdef"))
 269         srv.Close()
 270         time.Sleep(delay)
 271         n, err = conn.Read(buf)
 272         if n != 6 || string(buf) != "abcdef" {
 273                 return fmt.Errorf("Read = %d, buf= %q; want 6, abcdef", n, buf)
 274         }
 275         if err != io.EOF {
 276                 return fmt.Errorf("Second Read error = %v; want io.EOF", err)
 277         }
 278         return nil
 279 }
 280
 281 func TestTLSUniqueMatches(t *testing.T) {
 282         ln := newLocalListener(t)
 283         defer ln.Close()
 284
 285         serverTLSUniques := make(chan []byte)
 286         go func() {
 287                 for i := 0; i < 2; i++ {
 288                         sconn, err := ln.Accept()
 289                         if err != nil {
 290                                 t.Fatal(err)
 291                         }
 292                         serverConfig := *testConfig
 293                         srv := Server(sconn, &serverConfig)
 294                         if err := srv.Handshake(); err != nil {
 295                                 t.Fatal(err)
 296                         }
 297                         serverTLSUniques <- srv.ConnectionState().TLSUnique
 298                 }
 299         }()
 300
 301         clientConfig := *testConfig
 302         clientConfig.ClientSessionCache = NewLRUClientSessionCache(1)
 303         conn, err := Dial("tcp", ln.Addr().String(), &clientConfig)
 304         if err != nil {
 305                 t.Fatal(err)
 306         }
 307         if !bytes.Equal(conn.ConnectionState().TLSUnique, <-serverTLSUniques) {
 308                 t.Error("client and server channel bindings differ")
 309         }
 310         conn.Close()
 311
 312         conn, err = Dial("tcp", ln.Addr().String(), &clientConfig)
 313         if err != nil {
 314                 t.Fatal(err)
 315         }
 316         defer conn.Close()
 317         if !conn.ConnectionState().DidResume {
 318                 t.Error("second session did not use resumption")
 319         }
 320         if !bytes.Equal(conn.ConnectionState().TLSUnique, <-serverTLSUniques) {
 321                 t.Error("client and server channel bindings differ when session resumption is used")
 322         }
 323 }
 324
 325 func TestVerifyHostname(t *testing.T) {
 326         testenv.MustHaveExternalNetwork(t)
 327
 328         c, err := Dial("tcp", "www.google.com:https", nil)
 329         if err != nil {
 330                 t.Fatal(err)
 331         }
 332         if err := c.VerifyHostname("www.google.com"); err != nil {
 333                 t.Fatalf("verify www.google.com: %v", err)
 334         }
 335         if err := c.VerifyHostname("www.yahoo.com"); err == nil {
 336                 t.Fatalf("verify www.yahoo.com succeeded")
 337         }
 338
 339         c, err = Dial("tcp", "www.google.com:https", &Config{InsecureSkipVerify: true})
 340         if err != nil {
 341                 t.Fatal(err)
 342         }
 343         if err := c.VerifyHostname("www.google.com"); err == nil {
 344                 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true")
 345         }
 346         if err := c.VerifyHostname("www.yahoo.com"); err == nil {
 347                 t.Fatalf("verify www.google.com succeeded with InsecureSkipVerify=true")
 348         }
 349 }
 350
 351 func TestVerifyHostnameResumed(t *testing.T) {
 352         testenv.MustHaveExternalNetwork(t)
 353
 354         config := &Config{
 355                 ClientSessionCache: NewLRUClientSessionCache(32),
 356         }
 357         for i := 0; i < 2; i++ {
 358                 c, err := Dial("tcp", "www.google.com:https", config)
 359                 if err != nil {
 360                         t.Fatalf("Dial #%d: %v", i, err)
 361                 }
 362                 cs := c.ConnectionState()
 363                 if i > 0 && !cs.DidResume {
 364                         t.Fatalf("Subsequent connection unexpectedly didn't resume")
 365                 }
 366                 if cs.VerifiedChains == nil {
 367                         t.Fatalf("Dial #%d: cs.VerifiedChains == nil", i)
 368                 }
 369                 if err := c.VerifyHostname("www.google.com"); err != nil {
 370                         t.Fatalf("verify www.google.com #%d: %v", i, err)
 371                 }
 372                 c.Close()
 373         }
 374 }
 375
 376 func TestConnCloseBreakingWrite(t *testing.T) {
 377         ln := newLocalListener(t)
 378         defer ln.Close()
 379
 380         srvCh := make(chan *Conn, 1)
 381         var serr error
 382         var sconn net.Conn
 383         go func() {
 384                 var err error
 385                 sconn, err = ln.Accept()
 386                 if err != nil {
 387                         serr = err
 388                         srvCh <- nil
 389                         return
 390                 }
 391                 serverConfig := *testConfig
 392                 srv := Server(sconn, &serverConfig)
 393                 if err := srv.Handshake(); err != nil {
 394                         serr = fmt.Errorf("handshake: %v", err)
 395                         srvCh <- nil
 396                         return
 397                 }
 398                 srvCh <- srv
 399         }()
 400
 401         cconn, err := net.Dial("tcp", ln.Addr().String())
 402         if err != nil {
 403                 t.Fatal(err)
 404         }
 405         defer cconn.Close()
 406
 407         conn := &changeImplConn{
 408                 Conn: cconn,
 409         }
 410
 411         clientConfig := *testConfig
 412         tconn := Client(conn, &clientConfig)
 413         if err := tconn.Handshake(); err != nil {
 414                 t.Fatal(err)
 415         }
 416
 417         srv := <-srvCh
 418         if srv == nil {
 419                 t.Fatal(serr)
 420         }
 421         defer sconn.Close()
 422
 423         connClosed := make(chan struct{})
 424         conn.closeFunc = func() error {
 425                 close(connClosed)
 426                 return nil
 427         }
 428
 429         inWrite := make(chan bool, 1)
 430         var errConnClosed = errors.New("conn closed for test")
 431         conn.writeFunc = func(p []byte) (n int, err error) {
 432                 inWrite <- true
 433                 <-connClosed
 434                 return 0, errConnClosed
 435         }
 436
 437         closeReturned := make(chan bool, 1)
 438         go func() {
 439                 <-inWrite
 440                 tconn.Close() // test that this doesn't block forever.
 441                 closeReturned <- true
 442         }()
 443
 444         _, err = tconn.Write([]byte("foo"))
 445         if err != errConnClosed {
 446                 t.Errorf("Write error = %v; want errConnClosed", err)
 447         }
 448
 449         <-closeReturned
 450         if err := tconn.Close(); err != errClosed {
 451                 t.Errorf("Close error = %v; want errClosed", err)
 452         }
 453 }
 454
 455 // changeImplConn is a net.Conn which can change its Write and Close
 456 // methods.
 457 type changeImplConn struct {
 458         net.Conn
 459         writeFunc func([]byte) (int, error)
 460         closeFunc func() error
 461 }
 462
 463 func (w *changeImplConn) Write(p []byte) (n int, err error) {
 464         if w.writeFunc != nil {
 465                 return w.writeFunc(p)
 466         }
 467         return w.Conn.Write(p)
 468 }
 469
 470 func (w *changeImplConn) Close() error {
 471         if w.closeFunc != nil {
 472                 return w.closeFunc()
 473         }
 474         return w.Conn.Close()
 475 }