diff options
author | Jeff <jeffreyh192@gmail.com> | 2017-10-05 10:11:17 -0700 |
---|---|---|
committer | Russ Cox <rsc@golang.org> | 2017-10-25 18:57:03 +0000 |
commit | 8093678ae101efe17103b2b64260a3051253d852 (patch) | |
tree | 4035e9cfaaeb13d9a6d4146d399b66beda5bbf84 | |
parent | 0bf8909ff86777cebbb58b4e1497cfc7b8721fc1 (diff) | |
download | go-8093678ae101efe17103b2b64260a3051253d852.tar.gz go-8093678ae101efe17103b2b64260a3051253d852.zip |
[release-branch.go1.8] net/smtp: NewClient: set tls field to true when already using a TLS connection
Change-Id: I34008f56c191df0edcaafc20d569bbc6184f89fc
Reviewed-on: https://go-review.googlesource.com/68470
Run-TryBot: Ian Lance Taylor <iant@golang.org>
TryBot-Result: Gobot Gobot <gobot@golang.org>
Reviewed-by: Ian Lance Taylor <iant@golang.org>
Reviewed-on: https://go-review.googlesource.com/70843
Run-TryBot: Russ Cox <rsc@golang.org>
-rw-r--r-- | src/net/smtp/smtp.go | 1 | ||||
-rw-r--r-- | src/net/smtp/smtp_test.go | 47 |
2 files changed, 48 insertions, 0 deletions
diff --git a/src/net/smtp/smtp.go b/src/net/smtp/smtp.go index a408fa5336..cf91851de9 100644 --- a/src/net/smtp/smtp.go +++ b/src/net/smtp/smtp.go @@ -67,6 +67,7 @@ func NewClient(conn net.Conn, host string) (*Client, error) { return nil, err } c := &Client{Text: text, conn: conn, serverName: host, localName: "localhost"} + _, c.tls = conn.(*tls.Conn) return c, nil } diff --git a/src/net/smtp/smtp_test.go b/src/net/smtp/smtp_test.go index 15eaca524b..35e37a19dc 100644 --- a/src/net/smtp/smtp_test.go +++ b/src/net/smtp/smtp_test.go @@ -362,6 +362,53 @@ HELO localhost QUIT ` +func TestNewClientWithTLS(t *testing.T) { + cert, err := tls.X509KeyPair(localhostCert, localhostKey) + if err != nil { + t.Fatalf("loadcert: %v", err) + } + + config := tls.Config{Certificates: []tls.Certificate{cert}} + + ln, err := tls.Listen("tcp", "127.0.0.1:0", &config) + if err != nil { + ln, err = tls.Listen("tcp", "[::1]:0", &config) + if err != nil { + t.Fatalf("server: listen: %s", err) + } + } + + go func() { + conn, err := ln.Accept() + if err != nil { + t.Fatalf("server: accept: %s", err) + return + } + defer conn.Close() + + _, err = conn.Write([]byte("220 SIGNS\r\n")) + if err != nil { + t.Fatalf("server: write: %s", err) + return + } + }() + + config.InsecureSkipVerify = true + conn, err := tls.Dial("tcp", ln.Addr().String(), &config) + if err != nil { + t.Fatalf("client: dial: %s", err) + } + defer conn.Close() + + client, err := NewClient(conn, ln.Addr().String()) + if err != nil { + t.Fatalf("smtp: newclient: %s", err) + } + if !client.tls { + t.Errorf("client.tls Got: %t Expected: %t", client.tls, true) + } +} + func TestHello(t *testing.T) { if len(helloServer) != len(helloClient) { |