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  }