mirror of
https://github.com/refraction-networking/utls.git
synced 2025-04-04 20:47:36 +03:00
a2ca1d5330
As part of #42026, these helpers from io/ioutil were moved to os. (ioutil.TempFile and TempDir became os.CreateTemp and MkdirTemp.) Update the Go tree to use the preferred names. As usual, code compiled with the Go 1.4 bootstrap toolchain and code vendored from other sources is excluded. ReadDir changes are in a separate CL, because they are not a simple search and replace. For #42026. Change-Id: If318df0216d57e95ea0c4093b89f65e5b0ababb3 Reviewed-on: https://go-review.googlesource.com/c/go/+/266365 Trust: Russ Cox <rsc@golang.org> Run-TryBot: Russ Cox <rsc@golang.org> TryBot-Result: Go Bot <gobot@golang.org> Reviewed-by: Ian Lance Taylor <iant@golang.org>
107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
// Copyright 2020 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package tls
|
|
|
|
import (
|
|
"bytes"
|
|
"internal/testenv"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
// Tests that the linker is able to remove references to the Client or Server if unused.
|
|
func TestLinkerGC(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping in short mode")
|
|
}
|
|
t.Parallel()
|
|
goBin := testenv.GoToolPath(t)
|
|
testenv.MustHaveGoBuild(t)
|
|
|
|
tests := []struct {
|
|
name string
|
|
program string
|
|
want []string
|
|
bad []string
|
|
}{
|
|
{
|
|
name: "empty_import",
|
|
program: `package main
|
|
import _ "crypto/tls"
|
|
func main() {}
|
|
`,
|
|
bad: []string{
|
|
"tls.(*Conn)",
|
|
"type.crypto/tls.clientHandshakeState",
|
|
"type.crypto/tls.serverHandshakeState",
|
|
},
|
|
},
|
|
{
|
|
name: "client_and_server",
|
|
program: `package main
|
|
import "crypto/tls"
|
|
func main() {
|
|
tls.Dial("", "", nil)
|
|
tls.Server(nil, nil)
|
|
}
|
|
`,
|
|
want: []string{
|
|
"crypto/tls.(*Conn).clientHandshake",
|
|
"crypto/tls.(*Conn).serverHandshake",
|
|
},
|
|
},
|
|
{
|
|
name: "only_client",
|
|
program: `package main
|
|
import "crypto/tls"
|
|
func main() { tls.Dial("", "", nil) }
|
|
`,
|
|
want: []string{
|
|
"crypto/tls.(*Conn).clientHandshake",
|
|
},
|
|
bad: []string{
|
|
"crypto/tls.(*Conn).serverHandshake",
|
|
},
|
|
},
|
|
// TODO: add only_server like func main() { tls.Server(nil, nil) }
|
|
// That currently brings in the client via Conn.handleRenegotiation.
|
|
|
|
}
|
|
tmpDir := t.TempDir()
|
|
goFile := filepath.Join(tmpDir, "x.go")
|
|
exeFile := filepath.Join(tmpDir, "x.exe")
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if err := os.WriteFile(goFile, []byte(tt.program), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
os.Remove(exeFile)
|
|
cmd := exec.Command(goBin, "build", "-o", "x.exe", "x.go")
|
|
cmd.Dir = tmpDir
|
|
if out, err := cmd.CombinedOutput(); err != nil {
|
|
t.Fatalf("compile: %v, %s", err, out)
|
|
}
|
|
|
|
cmd = exec.Command(goBin, "tool", "nm", "x.exe")
|
|
cmd.Dir = tmpDir
|
|
nm, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Fatalf("nm: %v, %s", err, nm)
|
|
}
|
|
for _, sym := range tt.want {
|
|
if !bytes.Contains(nm, []byte(sym)) {
|
|
t.Errorf("expected symbol %q not found", sym)
|
|
}
|
|
}
|
|
for _, sym := range tt.bad {
|
|
if bytes.Contains(nm, []byte(sym)) {
|
|
t.Errorf("unexpected symbol %q found", sym)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|