github.com/jonasi/go@v0.0.0-20150930005915-e78e654c1de0/src/cmd/dist/test.go (about)

     1  // Copyright 2015 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 main
     6  
     7  import (
     8  	"bytes"
     9  	"errors"
    10  	"flag"
    11  	"fmt"
    12  	"io/ioutil"
    13  	"log"
    14  	"os"
    15  	"os/exec"
    16  	"path/filepath"
    17  	"regexp"
    18  	"strconv"
    19  	"strings"
    20  	"time"
    21  )
    22  
    23  func cmdtest() {
    24  	var t tester
    25  	flag.BoolVar(&t.listMode, "list", false, "list available tests")
    26  	flag.BoolVar(&t.noRebuild, "no-rebuild", false, "don't rebuild std and cmd packages")
    27  	flag.BoolVar(&t.keepGoing, "k", false, "keep going even when error occurred")
    28  	flag.BoolVar(&t.race, "race", false, "run in race builder mode (different set of tests)")
    29  	flag.StringVar(&t.banner, "banner", "##### ", "banner prefix; blank means no section banners")
    30  	flag.StringVar(&t.runRxStr, "run", os.Getenv("GOTESTONLY"),
    31  		"run only those tests matching the regular expression; empty means to run all. "+
    32  			"Special exception: if the string begins with '!', the match is inverted.")
    33  	xflagparse(-1) // any number of args
    34  	t.run()
    35  }
    36  
    37  // tester executes cmdtest.
    38  type tester struct {
    39  	race      bool
    40  	listMode  bool
    41  	noRebuild bool
    42  	keepGoing bool
    43  	runRxStr  string
    44  	runRx     *regexp.Regexp
    45  	runRxWant bool     // want runRx to match (true) or not match (false)
    46  	runNames  []string // tests to run, exclusive with runRx; empty means all
    47  	banner    string   // prefix, or "" for none
    48  
    49  	goroot     string
    50  	goarch     string
    51  	gohostarch string
    52  	goos       string
    53  	gohostos   string
    54  	cgoEnabled bool
    55  	partial    bool
    56  	haveTime   bool // the 'time' binary is available
    57  
    58  	tests        []distTest
    59  	timeoutScale int
    60  }
    61  
    62  // A distTest is a test run by dist test.
    63  // Each test has a unique name and belongs to a group (heading)
    64  type distTest struct {
    65  	name    string // unique test name; may be filtered with -run flag
    66  	heading string // group section; this header is printed before the test is run.
    67  	fn      func() error
    68  }
    69  
    70  func mustEnv(k string) string {
    71  	v := os.Getenv(k)
    72  	if v == "" {
    73  		log.Fatalf("Unset environment variable %v", k)
    74  	}
    75  	return v
    76  }
    77  
    78  func (t *tester) run() {
    79  	t.goroot = mustEnv("GOROOT")
    80  	t.goos = mustEnv("GOOS")
    81  	t.gohostos = mustEnv("GOHOSTOS")
    82  	t.goarch = mustEnv("GOARCH")
    83  	t.gohostarch = mustEnv("GOHOSTARCH")
    84  	slurp, err := exec.Command("go", "env", "CGO_ENABLED").Output()
    85  	if err != nil {
    86  		log.Fatalf("Error running go env CGO_ENABLED: %v", err)
    87  	}
    88  	t.cgoEnabled, _ = strconv.ParseBool(strings.TrimSpace(string(slurp)))
    89  	if flag.NArg() > 0 && t.runRxStr != "" {
    90  		log.Fatalf("the -run regular expression flag is mutually exclusive with test name arguments")
    91  	}
    92  	t.runNames = flag.Args()
    93  
    94  	if t.hasBash() {
    95  		if _, err := exec.LookPath("time"); err == nil {
    96  			t.haveTime = true
    97  		}
    98  	}
    99  
   100  	if !t.noRebuild {
   101  		t.out("Building packages and commands.")
   102  		cmd := exec.Command("go", "install", "-a", "-v", "std", "cmd")
   103  		cmd.Stdout = os.Stdout
   104  		cmd.Stderr = os.Stderr
   105  		if err := cmd.Run(); err != nil {
   106  			log.Fatalf("building packages and commands: %v", err)
   107  		}
   108  	}
   109  
   110  	if t.iOS() {
   111  		// Install the Mach exception handler used to intercept
   112  		// EXC_BAD_ACCESS and convert it into a Go panic. This is
   113  		// necessary for a Go program running under lldb (the way
   114  		// we run tests). It is disabled by default because iOS
   115  		// apps are not allowed to access the exc_server symbol.
   116  		cmd := exec.Command("go", "install", "-a", "-tags", "lldb", "runtime/cgo")
   117  		cmd.Stdout = os.Stdout
   118  		cmd.Stderr = os.Stderr
   119  		if err := cmd.Run(); err != nil {
   120  			log.Fatalf("building mach exception handler: %v", err)
   121  		}
   122  
   123  		defer func() {
   124  			cmd := exec.Command("go", "install", "-a", "runtime/cgo")
   125  			cmd.Stdout = os.Stdout
   126  			cmd.Stderr = os.Stderr
   127  			if err := cmd.Run(); err != nil {
   128  				log.Fatalf("reverting mach exception handler: %v", err)
   129  			}
   130  		}()
   131  	}
   132  
   133  	t.timeoutScale = 1
   134  	if t.goarch == "arm" || t.goos == "windows" {
   135  		t.timeoutScale = 2
   136  	}
   137  	if s := os.Getenv("GO_TEST_TIMEOUT_SCALE"); s != "" {
   138  		t.timeoutScale, err = strconv.Atoi(s)
   139  		if err != nil {
   140  			log.Fatalf("failed to parse $GO_TEST_TIMEOUT_SCALE = %q as integer: %v", s, err)
   141  		}
   142  	}
   143  
   144  	if t.runRxStr != "" {
   145  		if t.runRxStr[0] == '!' {
   146  			t.runRxWant = false
   147  			t.runRxStr = t.runRxStr[1:]
   148  		} else {
   149  			t.runRxWant = true
   150  		}
   151  		t.runRx = regexp.MustCompile(t.runRxStr)
   152  	}
   153  
   154  	t.registerTests()
   155  	if t.listMode {
   156  		for _, tt := range t.tests {
   157  			fmt.Println(tt.name)
   158  		}
   159  		return
   160  	}
   161  
   162  	// we must unset GOROOT_FINAL before tests, because runtime/debug requires
   163  	// correct access to source code, so if we have GOROOT_FINAL in effect,
   164  	// at least runtime/debug test will fail.
   165  	os.Unsetenv("GOROOT_FINAL")
   166  
   167  	for _, name := range t.runNames {
   168  		if !t.isRegisteredTestName(name) {
   169  			log.Fatalf("unknown test %q", name)
   170  		}
   171  	}
   172  
   173  	var lastHeading string
   174  	ok := true
   175  	for _, dt := range t.tests {
   176  		if !t.shouldRunTest(dt.name) {
   177  			t.partial = true
   178  			continue
   179  		}
   180  		if dt.heading != "" && lastHeading != dt.heading {
   181  			lastHeading = dt.heading
   182  			t.out(dt.heading)
   183  		}
   184  		if vflag > 0 {
   185  			fmt.Printf("# go tool dist test -run=^%s$\n", dt.name)
   186  		}
   187  		if err := dt.fn(); err != nil {
   188  			ok = false
   189  			if t.keepGoing {
   190  				log.Printf("Failed: %v", err)
   191  			} else {
   192  				log.Fatalf("Failed: %v", err)
   193  			}
   194  		}
   195  	}
   196  	if !ok {
   197  		fmt.Println("\nFAILED")
   198  		os.Exit(1)
   199  	} else if t.partial {
   200  		fmt.Println("\nALL TESTS PASSED (some were excluded)")
   201  	} else {
   202  		fmt.Println("\nALL TESTS PASSED")
   203  	}
   204  }
   205  
   206  func (t *tester) shouldRunTest(name string) bool {
   207  	if t.runRx != nil {
   208  		return t.runRx.MatchString(name) == t.runRxWant
   209  	}
   210  	if len(t.runNames) == 0 {
   211  		return true
   212  	}
   213  	for _, runName := range t.runNames {
   214  		if runName == name {
   215  			return true
   216  		}
   217  	}
   218  	return false
   219  }
   220  
   221  func (t *tester) tags() string {
   222  	if t.iOS() {
   223  		return "-tags=lldb"
   224  	}
   225  	return "-tags="
   226  }
   227  
   228  func (t *tester) timeout(sec int) string {
   229  	return "-timeout=" + fmt.Sprint(time.Duration(sec)*time.Second*time.Duration(t.timeoutScale))
   230  }
   231  
   232  // ranGoTest and stdMatches are state closed over by the stdlib
   233  // testing func in registerStdTest below. The tests are run
   234  // sequentially, so there's no need for locks.
   235  //
   236  // ranGoBench and benchMatches are the same, but are only used
   237  // in -race mode.
   238  var (
   239  	ranGoTest  bool
   240  	stdMatches []string
   241  
   242  	ranGoBench   bool
   243  	benchMatches []string
   244  )
   245  
   246  func (t *tester) registerStdTest(pkg string) {
   247  	testName := "go_test:" + pkg
   248  	if t.runRx == nil || t.runRx.MatchString(testName) {
   249  		stdMatches = append(stdMatches, pkg)
   250  	}
   251  	t.tests = append(t.tests, distTest{
   252  		name:    testName,
   253  		heading: "Testing packages.",
   254  		fn: func() error {
   255  			if ranGoTest {
   256  				return nil
   257  			}
   258  			ranGoTest = true
   259  			args := []string{
   260  				"test",
   261  				"-short",
   262  				t.tags(),
   263  				t.timeout(180),
   264  				"-gcflags=" + os.Getenv("GO_GCFLAGS"),
   265  			}
   266  			if t.race {
   267  				args = append(args, "-race")
   268  			}
   269  			args = append(args, stdMatches...)
   270  			cmd := exec.Command("go", args...)
   271  			cmd.Stdout = os.Stdout
   272  			cmd.Stderr = os.Stderr
   273  			return cmd.Run()
   274  		},
   275  	})
   276  }
   277  
   278  func (t *tester) registerRaceBenchTest(pkg string) {
   279  	testName := "go_test_bench:" + pkg
   280  	if t.runRx == nil || t.runRx.MatchString(testName) {
   281  		benchMatches = append(benchMatches, pkg)
   282  	}
   283  	t.tests = append(t.tests, distTest{
   284  		name:    testName,
   285  		heading: "Running benchmarks briefly.",
   286  		fn: func() error {
   287  			if ranGoBench {
   288  				return nil
   289  			}
   290  			ranGoBench = true
   291  			args := []string{
   292  				"test",
   293  				"-short",
   294  				"-race",
   295  				"-run=^$", // nothing. only benchmarks.
   296  				"-bench=.*",
   297  				"-benchtime=.1s",
   298  				"-cpu=4",
   299  			}
   300  			args = append(args, benchMatches...)
   301  			cmd := exec.Command("go", args...)
   302  			cmd.Stdout = os.Stdout
   303  			cmd.Stderr = os.Stderr
   304  			return cmd.Run()
   305  		},
   306  	})
   307  }
   308  
   309  func (t *tester) registerTests() {
   310  	// Fast path to avoid the ~1 second of `go list std cmd` when
   311  	// the caller lists specific tests to run. (as the continuous
   312  	// build coordinator does).
   313  	if len(t.runNames) > 0 {
   314  		for _, name := range t.runNames {
   315  			if strings.HasPrefix(name, "go_test:") {
   316  				t.registerStdTest(strings.TrimPrefix(name, "go_test:"))
   317  			}
   318  			if strings.HasPrefix(name, "go_test_bench:") {
   319  				t.registerRaceBenchTest(strings.TrimPrefix(name, "go_test_bench:"))
   320  			}
   321  		}
   322  	} else {
   323  		// Use a format string to only list packages and commands that have tests.
   324  		const format = "{{if (or .TestGoFiles .XTestGoFiles)}}{{.ImportPath}}{{end}}"
   325  		cmd := exec.Command("go", "list", "-f", format, "std")
   326  		if !t.race {
   327  			cmd.Args = append(cmd.Args, "cmd")
   328  		}
   329  		all, err := cmd.CombinedOutput()
   330  		if err != nil {
   331  			log.Fatalf("Error running go list std cmd: %v, %s", err, all)
   332  		}
   333  		pkgs := strings.Fields(string(all))
   334  		for _, pkg := range pkgs {
   335  			t.registerStdTest(pkg)
   336  		}
   337  		if t.race {
   338  			for _, pkg := range pkgs {
   339  				t.registerRaceBenchTest(pkg)
   340  			}
   341  		}
   342  	}
   343  
   344  	if t.race {
   345  		return
   346  	}
   347  
   348  	// Runtime CPU tests.
   349  	testName := "runtime:cpu124"
   350  	t.tests = append(t.tests, distTest{
   351  		name:    testName,
   352  		heading: "GOMAXPROCS=2 runtime -cpu=1,2,4",
   353  		fn: func() error {
   354  			cmd := t.dirCmd("src", "go", "test", "-short", t.timeout(300), t.tags(), "-cpu=1,2,4", "runtime")
   355  			// We set GOMAXPROCS=2 in addition to -cpu=1,2,4 in order to test runtime bootstrap code,
   356  			// creation of first goroutines and first garbage collections in the parallel setting.
   357  			cmd.Env = mergeEnvLists([]string{"GOMAXPROCS=2"}, os.Environ())
   358  			return cmd.Run()
   359  		},
   360  	})
   361  
   362  	// sync tests
   363  	t.tests = append(t.tests, distTest{
   364  		name:    "sync_cpu",
   365  		heading: "sync -cpu=10",
   366  		fn: func() error {
   367  			return t.dirCmd("src", "go", "test", "-short", t.timeout(120), t.tags(), "-cpu=10", "sync").Run()
   368  		},
   369  	})
   370  
   371  	if t.cgoEnabled && t.goos != "android" && !t.iOS() {
   372  		// Disabled on android and iOS. golang.org/issue/8345
   373  		t.tests = append(t.tests, distTest{
   374  			name:    "cgo_stdio",
   375  			heading: "../misc/cgo/stdio",
   376  			fn: func() error {
   377  				return t.dirCmd("misc/cgo/stdio",
   378  					"go", "run", filepath.Join(os.Getenv("GOROOT"), "test/run.go"), "-", ".").Run()
   379  			},
   380  		})
   381  		t.tests = append(t.tests, distTest{
   382  			name:    "cgo_life",
   383  			heading: "../misc/cgo/life",
   384  			fn: func() error {
   385  				return t.dirCmd("misc/cgo/life",
   386  					"go", "run", filepath.Join(os.Getenv("GOROOT"), "test/run.go"), "-", ".").Run()
   387  			},
   388  		})
   389  	}
   390  	if t.cgoEnabled && t.goos != "android" && !t.iOS() {
   391  		// TODO(crawshaw): reenable on android and iOS
   392  		// golang.org/issue/8345
   393  		//
   394  		// These tests are not designed to run off the host.
   395  		t.tests = append(t.tests, distTest{
   396  			name:    "cgo_test",
   397  			heading: "../misc/cgo/test",
   398  			fn:      t.cgoTest,
   399  		})
   400  	}
   401  
   402  	if t.raceDetectorSupported() {
   403  		t.tests = append(t.tests, distTest{
   404  			name:    "race",
   405  			heading: "Testing race detector",
   406  			fn:      t.raceTest,
   407  		})
   408  	}
   409  
   410  	if t.hasBash() && t.cgoEnabled && t.goos != "android" && t.goos != "darwin" {
   411  		t.registerTest("testgodefs", "../misc/cgo/testgodefs", "./test.bash")
   412  	}
   413  	if t.cgoEnabled {
   414  		if t.cgoTestSOSupported() {
   415  			t.tests = append(t.tests, distTest{
   416  				name:    "testso",
   417  				heading: "../misc/cgo/testso",
   418  				fn: func() error {
   419  					return t.cgoTestSO("misc/cgo/testso")
   420  				},
   421  			})
   422  			t.tests = append(t.tests, distTest{
   423  				name:    "testsovar",
   424  				heading: "../misc/cgo/testsovar",
   425  				fn: func() error {
   426  					return t.cgoTestSO("misc/cgo/testsovar")
   427  				},
   428  			})
   429  		}
   430  		if t.supportedBuildmode("c-archive") {
   431  			t.registerTest("testcarchive", "../misc/cgo/testcarchive", "./test.bash")
   432  		}
   433  		if t.supportedBuildmode("c-shared") {
   434  			t.registerTest("testcshared", "../misc/cgo/testcshared", "./test.bash")
   435  		}
   436  		if t.supportedBuildmode("shared") {
   437  			t.registerTest("testshared", "../misc/cgo/testshared", "go", "test")
   438  		}
   439  		if t.gohostos == "linux" && t.goarch == "amd64" {
   440  			t.registerTest("testasan", "../misc/cgo/testasan", "go", "run", "main.go")
   441  		}
   442  		if t.hasBash() && t.goos != "android" && !t.iOS() && t.gohostos != "windows" {
   443  			t.registerTest("cgo_errors", "../misc/cgo/errors", "./test.bash")
   444  		}
   445  		if t.gohostos == "linux" && t.extLink() {
   446  			t.registerTest("testsigfwd", "../misc/cgo/testsigfwd", "go", "run", "main.go")
   447  		}
   448  	}
   449  	if t.hasBash() && t.goos != "nacl" && t.goos != "android" && !t.iOS() {
   450  		t.registerTest("doc_progs", "../doc/progs", "time", "go", "run", "run.go")
   451  		t.registerTest("wiki", "../doc/articles/wiki", "./test.bash")
   452  		t.registerTest("codewalk", "../doc/codewalk", "time", "./run")
   453  		for _, name := range t.shootoutTests() {
   454  			if name == "spectralnorm" && os.Getenv("GO_BUILDER_NAME") == "linux-arm-arm5" {
   455  				// Heavy on floating point and takes over 20 minutes with softfloat.
   456  				// Disabled per Issue 12688.
   457  				continue
   458  			}
   459  			t.registerTest("shootout:"+name, "../test/bench/shootout", "time", "./timing.sh", "-test", name)
   460  		}
   461  	}
   462  	if t.goos != "android" && !t.iOS() {
   463  		t.registerTest("bench_go1", "../test/bench/go1", "go", "test")
   464  	}
   465  	if t.goos != "android" && !t.iOS() {
   466  		const nShards = 5
   467  		for shard := 0; shard < nShards; shard++ {
   468  			shard := shard
   469  			t.tests = append(t.tests, distTest{
   470  				name:    fmt.Sprintf("test:%d_%d", shard, nShards),
   471  				heading: "../test",
   472  				fn:      func() error { return t.testDirTest(shard, nShards) },
   473  			})
   474  		}
   475  	}
   476  	if t.goos != "nacl" && t.goos != "android" && !t.iOS() {
   477  		t.tests = append(t.tests, distTest{
   478  			name:    "api",
   479  			heading: "API check",
   480  			fn: func() error {
   481  				return t.dirCmd("src", "go", "run", filepath.Join(t.goroot, "src/cmd/api/run.go")).Run()
   482  			},
   483  		})
   484  	}
   485  }
   486  
   487  // isRegisteredTestName reports whether a test named testName has already
   488  // been registered.
   489  func (t *tester) isRegisteredTestName(testName string) bool {
   490  	for _, tt := range t.tests {
   491  		if tt.name == testName {
   492  			return true
   493  		}
   494  	}
   495  	return false
   496  }
   497  
   498  func (t *tester) registerTest(name, dirBanner, bin string, args ...string) {
   499  	if bin == "time" && !t.haveTime {
   500  		bin, args = args[0], args[1:]
   501  	}
   502  	if t.isRegisteredTestName(name) {
   503  		panic("duplicate registered test name " + name)
   504  	}
   505  	t.tests = append(t.tests, distTest{
   506  		name:    name,
   507  		heading: dirBanner,
   508  		fn: func() error {
   509  			return t.dirCmd(filepath.Join(t.goroot, "src", dirBanner), bin, args...).Run()
   510  		},
   511  	})
   512  }
   513  
   514  func (t *tester) dirCmd(dir string, bin string, args ...string) *exec.Cmd {
   515  	cmd := exec.Command(bin, args...)
   516  	if filepath.IsAbs(dir) {
   517  		cmd.Dir = dir
   518  	} else {
   519  		cmd.Dir = filepath.Join(t.goroot, dir)
   520  	}
   521  	cmd.Stdout = os.Stdout
   522  	cmd.Stderr = os.Stderr
   523  	if vflag > 1 {
   524  		errprintf("%s\n", strings.Join(cmd.Args, " "))
   525  	}
   526  	return cmd
   527  }
   528  
   529  func (t *tester) iOS() bool {
   530  	return t.goos == "darwin" && (t.goarch == "arm" || t.goarch == "arm64")
   531  }
   532  
   533  func (t *tester) out(v string) {
   534  	if t.banner == "" {
   535  		return
   536  	}
   537  	fmt.Println("\n" + t.banner + v)
   538  }
   539  
   540  func (t *tester) extLink() bool {
   541  	pair := t.gohostos + "-" + t.goarch
   542  	switch pair {
   543  	case "android-arm",
   544  		"darwin-arm", "darwin-arm64",
   545  		"dragonfly-386", "dragonfly-amd64",
   546  		"freebsd-386", "freebsd-amd64", "freebsd-arm",
   547  		"linux-386", "linux-amd64", "linux-arm", "linux-arm64",
   548  		"netbsd-386", "netbsd-amd64",
   549  		"openbsd-386", "openbsd-amd64",
   550  		"windows-386", "windows-amd64":
   551  		return true
   552  	case "darwin-386", "darwin-amd64":
   553  		// linkmode=external fails on OS X 10.6 and earlier == Darwin
   554  		// 10.8 and earlier.
   555  		unameR, err := exec.Command("uname", "-r").Output()
   556  		if err != nil {
   557  			log.Fatalf("uname -r: %v", err)
   558  		}
   559  		major, _ := strconv.Atoi(string(unameR[:bytes.IndexByte(unameR, '.')]))
   560  		return major > 10
   561  	}
   562  	return false
   563  }
   564  
   565  func (t *tester) supportedBuildmode(mode string) bool {
   566  	pair := t.goos + "-" + t.goarch
   567  	switch mode {
   568  	case "c-archive":
   569  		if !t.extLink() {
   570  			return false
   571  		}
   572  		switch pair {
   573  		case "darwin-amd64", "darwin-arm", "darwin-arm64",
   574  			"linux-amd64", "linux-386":
   575  			return true
   576  		}
   577  		return false
   578  	case "c-shared":
   579  		// TODO(hyangah): add linux-386.
   580  		switch pair {
   581  		case "linux-amd64", "darwin-amd64", "android-arm", "linux-arm":
   582  			return true
   583  		}
   584  		return false
   585  	case "shared":
   586  		switch pair {
   587  		case "linux-amd64":
   588  			return true
   589  		}
   590  		return false
   591  	default:
   592  		log.Fatal("internal error: unknown buildmode %s", mode)
   593  		return false
   594  	}
   595  }
   596  
   597  func (t *tester) cgoTest() error {
   598  	env := mergeEnvLists([]string{"GOTRACEBACK=2"}, os.Environ())
   599  
   600  	if t.goos == "android" || t.iOS() {
   601  		cmd := t.dirCmd("misc/cgo/test", "go", "test", t.tags())
   602  		cmd.Env = env
   603  		return cmd.Run()
   604  	}
   605  
   606  	cmd := t.dirCmd("misc/cgo/test", "go", "test", t.tags(), "-ldflags", "-linkmode=auto")
   607  	cmd.Env = env
   608  	if err := cmd.Run(); err != nil {
   609  		return err
   610  	}
   611  
   612  	if t.gohostos != "dragonfly" {
   613  		// linkmode=internal fails on dragonfly since errno is a TLS relocation.
   614  		cmd := t.dirCmd("misc/cgo/test", "go", "test", "-ldflags", "-linkmode=internal")
   615  		cmd.Env = env
   616  		if err := cmd.Run(); err != nil {
   617  			return err
   618  		}
   619  	}
   620  
   621  	pair := t.gohostos + "-" + t.goarch
   622  	switch pair {
   623  	case "darwin-386", "darwin-amd64",
   624  		"openbsd-386", "openbsd-amd64",
   625  		"windows-386", "windows-amd64":
   626  		// test linkmode=external, but __thread not supported, so skip testtls.
   627  		if !t.extLink() {
   628  			break
   629  		}
   630  		cmd := t.dirCmd("misc/cgo/test", "go", "test", "-ldflags", "-linkmode=external")
   631  		cmd.Env = env
   632  		if err := cmd.Run(); err != nil {
   633  			return err
   634  		}
   635  		cmd = t.dirCmd("misc/cgo/test", "go", "test", "-ldflags", "-linkmode=external -s")
   636  		cmd.Env = env
   637  		if err := cmd.Run(); err != nil {
   638  			return err
   639  		}
   640  	case "android-arm",
   641  		"dragonfly-386", "dragonfly-amd64",
   642  		"freebsd-386", "freebsd-amd64", "freebsd-arm",
   643  		"linux-386", "linux-amd64", "linux-arm",
   644  		"netbsd-386", "netbsd-amd64":
   645  
   646  		cmd := t.dirCmd("misc/cgo/test", "go", "test", "-ldflags", "-linkmode=external")
   647  		cmd.Env = env
   648  		if err := cmd.Run(); err != nil {
   649  			return err
   650  		}
   651  		cmd = t.dirCmd("misc/cgo/testtls", "go", "test", "-ldflags", "-linkmode=auto")
   652  		cmd.Env = env
   653  		if err := cmd.Run(); err != nil {
   654  			return err
   655  		}
   656  		cmd = t.dirCmd("misc/cgo/testtls", "go", "test", "-ldflags", "-linkmode=external")
   657  		cmd.Env = env
   658  		if err := cmd.Run(); err != nil {
   659  			return err
   660  		}
   661  
   662  		switch pair {
   663  		case "netbsd-386", "netbsd-amd64":
   664  			// no static linking
   665  		case "freebsd-arm":
   666  			// -fPIC compiled tls code will use __tls_get_addr instead
   667  			// of __aeabi_read_tp, however, on FreeBSD/ARM, __tls_get_addr
   668  			// is implemented in rtld-elf, so -fPIC isn't compatible with
   669  			// static linking on FreeBSD/ARM with clang. (cgo depends on
   670  			// -fPIC fundamentally.)
   671  		default:
   672  			cc := mustEnv("CC")
   673  			cmd := t.dirCmd("misc/cgo/test",
   674  				cc, "-xc", "-o", "/dev/null", "-static", "-")
   675  			cmd.Env = env
   676  			cmd.Stdin = strings.NewReader("int main() {}")
   677  			if err := cmd.Run(); err != nil {
   678  				fmt.Println("No support for static linking found (lacks libc.a?), skip cgo static linking test.")
   679  			} else {
   680  				cmd = t.dirCmd("misc/cgo/testtls", "go", "test", "-ldflags", `-linkmode=external -extldflags "-static -pthread"`)
   681  				cmd.Env = env
   682  				if err := cmd.Run(); err != nil {
   683  					return err
   684  				}
   685  
   686  				cmd = t.dirCmd("misc/cgo/nocgo", "go", "test")
   687  				cmd.Env = env
   688  				if err := cmd.Run(); err != nil {
   689  					return err
   690  				}
   691  
   692  				cmd = t.dirCmd("misc/cgo/nocgo", "go", "test", "-ldflags", `-linkmode=external`)
   693  				cmd.Env = env
   694  				if err := cmd.Run(); err != nil {
   695  					return err
   696  				}
   697  
   698  				cmd = t.dirCmd("misc/cgo/nocgo", "go", "test", "-ldflags", `-linkmode=external -extldflags "-static -pthread"`)
   699  				cmd.Env = env
   700  				if err := cmd.Run(); err != nil {
   701  					return err
   702  				}
   703  			}
   704  
   705  			if pair != "freebsd-amd64" { // clang -pie fails to link misc/cgo/test
   706  				cmd := t.dirCmd("misc/cgo/test",
   707  					cc, "-xc", "-o", "/dev/null", "-pie", "-")
   708  				cmd.Env = env
   709  				cmd.Stdin = strings.NewReader("int main() {}")
   710  				if err := cmd.Run(); err != nil {
   711  					fmt.Println("No support for -pie found, skip cgo PIE test.")
   712  				} else {
   713  					cmd = t.dirCmd("misc/cgo/test", "go", "test", "-ldflags", `-linkmode=external -extldflags "-pie"`)
   714  					cmd.Env = env
   715  					if err := cmd.Run(); err != nil {
   716  						return fmt.Errorf("pie cgo/test: %v", err)
   717  					}
   718  					cmd = t.dirCmd("misc/cgo/testtls", "go", "test", "-ldflags", `-linkmode=external -extldflags "-pie"`)
   719  					cmd.Env = env
   720  					if err := cmd.Run(); err != nil {
   721  						return fmt.Errorf("pie cgo/testtls: %v", err)
   722  					}
   723  					cmd = t.dirCmd("misc/cgo/nocgo", "go", "test", "-ldflags", `-linkmode=external -extldflags "-pie"`)
   724  					cmd.Env = env
   725  					if err := cmd.Run(); err != nil {
   726  						return fmt.Errorf("pie cgo/nocgo: %v", err)
   727  					}
   728  				}
   729  			}
   730  		}
   731  	}
   732  
   733  	return nil
   734  }
   735  
   736  func (t *tester) cgoTestSOSupported() bool {
   737  	if t.goos == "android" || t.iOS() {
   738  		// No exec facility on Android or iOS.
   739  		return false
   740  	}
   741  	if t.goarch == "ppc64le" || t.goarch == "ppc64" {
   742  		// External linking not implemented on ppc64 (issue #8912).
   743  		return false
   744  	}
   745  	return true
   746  }
   747  
   748  func (t *tester) cgoTestSO(testpath string) error {
   749  	dir := filepath.Join(t.goroot, testpath)
   750  
   751  	// build shared object
   752  	output, err := exec.Command("go", "env", "CC").Output()
   753  	if err != nil {
   754  		return fmt.Errorf("Error running go env CC: %v", err)
   755  	}
   756  	cc := strings.TrimSuffix(string(output), "\n")
   757  	if cc == "" {
   758  		return errors.New("CC environment variable (go env CC) cannot be empty")
   759  	}
   760  	output, err = exec.Command("go", "env", "GOGCCFLAGS").Output()
   761  	if err != nil {
   762  		return fmt.Errorf("Error running go env GOGCCFLAGS: %v", err)
   763  	}
   764  	gogccflags := strings.Split(strings.TrimSuffix(string(output), "\n"), " ")
   765  
   766  	ext := "so"
   767  	args := append(gogccflags, "-shared")
   768  	switch t.goos {
   769  	case "darwin":
   770  		ext = "dylib"
   771  		args = append(args, "-undefined", "suppress", "-flat_namespace")
   772  	case "windows":
   773  		ext = "dll"
   774  		args = append(args, "-DEXPORT_DLL")
   775  	}
   776  	sofname := "libcgosotest." + ext
   777  	args = append(args, "-o", sofname, "cgoso_c.c")
   778  
   779  	if err := t.dirCmd(dir, cc, args...).Run(); err != nil {
   780  		return err
   781  	}
   782  	defer os.Remove(filepath.Join(dir, sofname))
   783  
   784  	if err := t.dirCmd(dir, "go", "build", "-o", "main.exe", "main.go").Run(); err != nil {
   785  		return err
   786  	}
   787  	defer os.Remove(filepath.Join(dir, "main.exe"))
   788  
   789  	cmd := t.dirCmd(dir, "./main.exe")
   790  	if t.goos != "windows" {
   791  		s := "LD_LIBRARY_PATH"
   792  		if t.goos == "darwin" {
   793  			s = "DYLD_LIBRARY_PATH"
   794  		}
   795  		cmd.Env = mergeEnvLists([]string{s + "=."}, os.Environ())
   796  	}
   797  	return cmd.Run()
   798  }
   799  
   800  func (t *tester) hasBash() bool {
   801  	switch t.gohostos {
   802  	case "windows", "plan9":
   803  		return false
   804  	}
   805  	return true
   806  }
   807  
   808  func (t *tester) raceDetectorSupported() bool {
   809  	switch t.gohostos {
   810  	case "linux", "darwin", "freebsd", "windows":
   811  		return t.cgoEnabled && t.goarch == "amd64" && t.gohostos == t.goos
   812  	}
   813  	return false
   814  }
   815  
   816  func (t *tester) raceTest() error {
   817  	if err := t.dirCmd("src", "go", "test", "-race", "-i", "runtime/race", "flag", "os/exec").Run(); err != nil {
   818  		return err
   819  	}
   820  	if err := t.dirCmd("src", "go", "test", "-race", "-run=Output", "runtime/race").Run(); err != nil {
   821  		return err
   822  	}
   823  	if err := t.dirCmd("src", "go", "test", "-race", "-short", "flag", "os/exec").Run(); err != nil {
   824  		return err
   825  	}
   826  	if t.cgoEnabled {
   827  		env := mergeEnvLists([]string{"GOTRACEBACK=2"}, os.Environ())
   828  		cmd := t.dirCmd("misc/cgo/test", "go", "test", "-race", "-short")
   829  		cmd.Env = env
   830  		if err := cmd.Run(); err != nil {
   831  			return err
   832  		}
   833  	}
   834  	if t.extLink() {
   835  		// Test with external linking; see issue 9133.
   836  		if err := t.dirCmd("src", "go", "test", "-race", "-short", "-ldflags=-linkmode=external", "flag", "os/exec").Run(); err != nil {
   837  			return err
   838  		}
   839  	}
   840  	return nil
   841  }
   842  
   843  func (t *tester) testDirTest(shard, shards int) error {
   844  	const runExe = "runtest.exe" // named exe for Windows, but harmless elsewhere
   845  	cmd := t.dirCmd("test", "go", "build", "-o", runExe, "run.go")
   846  	cmd.Env = mergeEnvLists([]string{"GOOS=" + t.gohostos, "GOARCH=" + t.gohostarch, "GOMAXPROCS="}, os.Environ())
   847  	if err := cmd.Run(); err != nil {
   848  		return err
   849  	}
   850  	absExe := filepath.Join(cmd.Dir, runExe)
   851  	defer os.Remove(absExe)
   852  	return t.dirCmd("test", absExe,
   853  		fmt.Sprintf("--shard=%d", shard),
   854  		fmt.Sprintf("--shards=%d", shards),
   855  	).Run()
   856  }
   857  
   858  func (t *tester) shootoutTests() []string {
   859  	sh, err := ioutil.ReadFile(filepath.Join(t.goroot, "test", "bench", "shootout", "timing.sh"))
   860  	if err != nil {
   861  		log.Fatal(err)
   862  	}
   863  	m := regexp.MustCompile(`(?m)^\s+run="([\w+ ]+)"\s*$`).FindSubmatch(sh)
   864  	if m == nil {
   865  		log.Fatal("failed to find run=\"...\" line in test/bench/shootout/timing.sh")
   866  	}
   867  	return strings.Fields(string(m[1]))
   868  }
   869  
   870  // mergeEnvLists merges the two environment lists such that
   871  // variables with the same name in "in" replace those in "out".
   872  // out may be mutated.
   873  func mergeEnvLists(in, out []string) []string {
   874  NextVar:
   875  	for _, inkv := range in {
   876  		k := strings.SplitAfterN(inkv, "=", 2)[0]
   877  		for i, outkv := range out {
   878  			if strings.HasPrefix(outkv, k) {
   879  				out[i] = inkv
   880  				continue NextVar
   881  			}
   882  		}
   883  		out = append(out, inkv)
   884  	}
   885  	return out
   886  }