mirror of
https://github.com/golang/go
synced 2024-11-12 02:50:25 -07:00
net/http: Add TLS Connection State to Responses.
Fixes #7289. LGTM=bradfitz R=golang-codereviews, r, bradfitz, rsc CC=golang-codereviews https://golang.org/cl/52660047
This commit is contained in:
parent
efe381c889
commit
4816986ff5
@ -709,6 +709,34 @@ func TestTransportUsesTLSConfigServerName(t *testing.T) {
|
||||
res.Body.Close()
|
||||
}
|
||||
|
||||
func TestResponseSetsTLSConnectionState(t *testing.T) {
|
||||
defer afterTest(t)
|
||||
ts := httptest.NewTLSServer(HandlerFunc(func(w ResponseWriter, r *Request) {
|
||||
w.Write([]byte("Hello"))
|
||||
}))
|
||||
defer ts.Close()
|
||||
|
||||
tr := newTLSTransport(t, ts)
|
||||
tr.TLSClientConfig.CipherSuites = []uint16{tls.TLS_RSA_WITH_3DES_EDE_CBC_SHA}
|
||||
tr.Dial = func(netw, addr string) (net.Conn, error) {
|
||||
return net.Dial(netw, ts.Listener.Addr().String())
|
||||
}
|
||||
defer tr.CloseIdleConnections()
|
||||
c := &Client{Transport: tr}
|
||||
res, err := c.Get("https://example.com/")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if res.TLS == nil {
|
||||
t.Fatal("Response didn't set TLS Connection State.")
|
||||
}
|
||||
if res.TLS.CipherSuite != tls.TLS_RSA_WITH_3DES_EDE_CBC_SHA {
|
||||
t.Errorf("Unexpected TLS Cipher Suite: %d != %d",
|
||||
res.TLS.CipherSuite, tls.TLS_RSA_WITH_3DES_EDE_CBC_SHA)
|
||||
}
|
||||
res.Body.Close()
|
||||
}
|
||||
|
||||
// Verify Response.ContentLength is populated. http://golang.org/issue/4126
|
||||
func TestClientHeadContentLength(t *testing.T) {
|
||||
defer afterTest(t)
|
||||
|
@ -8,6 +8,7 @@ package http
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"crypto/tls"
|
||||
"errors"
|
||||
"io"
|
||||
"net/textproto"
|
||||
@ -74,6 +75,12 @@ type Response struct {
|
||||
// Request's Body is nil (having already been consumed).
|
||||
// This is only populated for Client requests.
|
||||
Request *Request
|
||||
|
||||
// TLS allows information about the TLS connection on which the
|
||||
// response was received. The Transport in this package sets the field
|
||||
// for TLS-enabled connections before returning the Response otherwise
|
||||
// it leaves the field nil.
|
||||
TLS *tls.ConnectionState
|
||||
}
|
||||
|
||||
// Cookies parses and returns the cookies set in the Set-Cookie headers.
|
||||
|
@ -791,6 +791,12 @@ func (pc *persistConn) readLoop() {
|
||||
resp, err = ReadResponse(pc.br, rc.req)
|
||||
}
|
||||
}
|
||||
|
||||
if tlsConn, ok := pc.conn.(*tls.Conn); resp != nil && ok {
|
||||
resp.TLS = new(tls.ConnectionState)
|
||||
*resp.TLS = tlsConn.ConnectionState()
|
||||
}
|
||||
|
||||
hasBody := resp != nil && rc.req.Method != "HEAD" && resp.ContentLength != 0
|
||||
|
||||
if err != nil {
|
||||
|
Loading…
Reference in New Issue
Block a user