// Copyright 2015 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 main import ( "bytes" "encoding/json" "flag" "fmt" "io" "io/fs" "log" "os" "os/exec" "path/filepath" "reflect" "regexp" "runtime" "strconv" "strings" "time" ) func cmdtest() { gogcflags = os.Getenv("GO_GCFLAGS") setNoOpt() var t tester var noRebuild bool flag.BoolVar(&t.listMode, "list", false, "list available tests") flag.BoolVar(&t.rebuild, "rebuild", false, "rebuild everything first") flag.BoolVar(&noRebuild, "no-rebuild", false, "overrides -rebuild (historical dreg)") flag.BoolVar(&t.keepGoing, "k", false, "keep going even when error occurred") flag.BoolVar(&t.race, "race", false, "run in race builder mode (different set of tests)") flag.BoolVar(&t.compileOnly, "compile-only", false, "compile tests, but don't run them") flag.StringVar(&t.banner, "banner", "##### ", "banner prefix; blank means no section banners") flag.StringVar(&t.runRxStr, "run", "", "run only those tests matching the regular expression; empty means to run all. "+ "Special exception: if the string begins with '!', the match is inverted.") flag.BoolVar(&t.msan, "msan", false, "run in memory sanitizer builder mode") flag.BoolVar(&t.asan, "asan", false, "run in address sanitizer builder mode") flag.BoolVar(&t.json, "json", false, "report test results in JSON") xflagparse(-1) // any number of args if noRebuild { t.rebuild = false } t.run() } // tester executes cmdtest. type tester struct { race bool msan bool asan bool listMode bool rebuild bool failed bool keepGoing bool compileOnly bool // just try to compile all tests, but no need to run runRxStr string runRx *regexp.Regexp runRxWant bool // want runRx to match (true) or not match (false) runNames []string // tests to run, exclusive with runRx; empty means all banner string // prefix, or "" for none lastHeading string // last dir heading printed short bool cgoEnabled bool partial bool json bool tests []distTest // use addTest to extend testNames map[string]bool timeoutScale int worklist []*work } type work struct { dt *distTest cmd *exec.Cmd // Must write stdout/stderr to work.out flush func() // If non-nil, called after cmd.Run start chan bool out bytes.Buffer err error end chan bool } // printSkip prints a skip message for all of work. func (w *work) printSkip(t *tester, msg string) { if t.json { type event struct { Time time.Time Action string Package string Output string `json:",omitempty"` } enc := json.NewEncoder(&w.out) ev := event{Time: time.Now(), Package: w.dt.name, Action: "start"} enc.Encode(ev) ev.Action = "output" ev.Output = msg enc.Encode(ev) ev.Action = "skip" ev.Output = "" enc.Encode(ev) return } fmt.Fprintln(&w.out, msg) } // A distTest is a test run by dist test. // Each test has a unique name and belongs to a group (heading) type distTest struct { name string // unique test name; may be filtered with -run flag heading string // group section; this header is printed before the test is run. fn func(*distTest) error } func (t *tester) run() { timelog("start", "dist test") os.Setenv("PATH", fmt.Sprintf("%s%c%s", gorootBin, os.PathListSeparator, os.Getenv("PATH"))) t.short = true if v := os.Getenv("GO_TEST_SHORT"); v != "" { short, err := strconv.ParseBool(v) if err != nil { fatalf("invalid GO_TEST_SHORT %q: %v", v, err) } t.short = short } cmd := exec.Command(gorootBinGo, "env", "CGO_ENABLED") cmd.Stderr = new(bytes.Buffer) slurp, err := cmd.Output() if err != nil { fatalf("Error running %s: %v\n%s", cmd, err, cmd.Stderr) } parts := strings.Split(string(slurp), "\n") if nlines := len(parts) - 1; nlines < 1 { fatalf("Error running %s: output contains <1 lines\n%s", cmd, cmd.Stderr) } t.cgoEnabled, _ = strconv.ParseBool(parts[0]) if flag.NArg() > 0 && t.runRxStr != "" { fatalf("the -run regular expression flag is mutually exclusive with test name arguments") } t.runNames = flag.Args() // Set GOTRACEBACK to system if the user didn't set a level explicitly. // Since we're running tests for Go, we want as much detail as possible // if something goes wrong. // // Set it before running any commands just in case something goes wrong. if ok := isEnvSet("GOTRACEBACK"); !ok { if err := os.Setenv("GOTRACEBACK", "system"); err != nil { if t.keepGoing { log.Printf("Failed to set GOTRACEBACK: %v", err) } else { fatalf("Failed to set GOTRACEBACK: %v", err) } } } if t.rebuild { t.out("Building packages and commands.") // Force rebuild the whole toolchain. goInstall(toolenv(), gorootBinGo, append([]string{"-a"}, toolchain...)...) } if !t.listMode { if builder := os.Getenv("GO_BUILDER_NAME"); builder == "" { // Ensure that installed commands are up to date, even with -no-rebuild, // so that tests that run commands end up testing what's actually on disk. // If everything is up-to-date, this is a no-op. // We first build the toolchain twice to allow it to converge, // as when we first bootstrap. // See cmdbootstrap for a description of the overall process. // // On the builders, we skip this step: we assume that 'dist test' is // already using the result of a clean build, and because of test sharding // and virtualization we usually start with a clean GOCACHE, so we would // end up rebuilding large parts of the standard library that aren't // otherwise relevant to the actual set of packages under test. goInstall(toolenv(), gorootBinGo, toolchain...) goInstall(toolenv(), gorootBinGo, toolchain...) goInstall(toolenv(), gorootBinGo, "cmd") } } t.timeoutScale = 1 if s := os.Getenv("GO_TEST_TIMEOUT_SCALE"); s != "" { t.timeoutScale, err = strconv.Atoi(s) if err != nil { fatalf("failed to parse $GO_TEST_TIMEOUT_SCALE = %q as integer: %v", s, err) } } if t.runRxStr != "" { if t.runRxStr[0] == '!' { t.runRxWant = false t.runRxStr = t.runRxStr[1:] } else { t.runRxWant = true } t.runRx = regexp.MustCompile(t.runRxStr) } t.registerTests() if t.listMode { for _, tt := range t.tests { fmt.Println(tt.name) } return } for _, name := range t.runNames { if !t.testNames[name] { fatalf("unknown test %q", name) } } // On a few builders, make GOROOT unwritable to catch tests writing to it. if strings.HasPrefix(os.Getenv("GO_BUILDER_NAME"), "linux-") { if os.Getuid() == 0 { // Don't bother making GOROOT unwritable: // we're running as root, so permissions would have no effect. } else { xatexit(t.makeGOROOTUnwritable()) } } if !t.json { if err := t.maybeLogMetadata(); err != nil { t.failed = true if t.keepGoing { log.Printf("Failed logging metadata: %v", err) } else { fatalf("Failed logging metadata: %v", err) } } } for _, dt := range t.tests { if !t.shouldRunTest(dt.name) { t.partial = true continue } dt := dt // dt used in background after this iteration if err := dt.fn(&dt); err != nil { t.runPending(&dt) // in case that hasn't been done yet t.failed = true if t.keepGoing { log.Printf("Failed: %v", err) } else { fatalf("Failed: %v", err) } } } t.runPending(nil) timelog("end", "dist test") if !t.json { if t.failed { fmt.Println("\nFAILED") } else if t.partial { fmt.Println("\nALL TESTS PASSED (some were excluded)") } else { fmt.Println("\nALL TESTS PASSED") } } if t.failed { xexit(1) } } func (t *tester) shouldRunTest(name string) bool { if t.runRx != nil { return t.runRx.MatchString(name) == t.runRxWant } if len(t.runNames) == 0 { return true } for _, runName := range t.runNames { if runName == name { return true } } return false } func (t *tester) maybeLogMetadata() error { if t.compileOnly { // We need to run a subprocess to log metadata. Don't do that // on compile-only runs. return nil } t.out("Test execution environment.") // Helper binary to print system metadata (CPU model, etc). This is a // separate binary from dist so it need not build with the bootstrap // toolchain. // // TODO(prattmic): If we split dist bootstrap and dist test then this // could be simplified to directly use internal/sysinfo here. return t.dirCmd(filepath.Join(goroot, "src/cmd/internal/metadata"), gorootBinGo, []string{"run", "main.go"}).Run() } // testName returns the dist test name for a given package and variant. func testName(pkg, variant string) string { name := pkg if variant != "" { name += ":" + variant } return name } // goTest represents all options to a "go test" command. The final command will // combine configuration from goTest and tester flags. type goTest struct { timeout time.Duration // If non-zero, override timeout short bool // If true, force -short tags []string // Build tags race bool // Force -race bench bool // Run benchmarks (briefly), not tests. runTests string // Regexp of tests to run cpu string // If non-empty, -cpu flag gcflags string // If non-empty, build with -gcflags=all=X ldflags string // If non-empty, build with -ldflags=X buildmode string // If non-empty, -buildmode flag env []string // Environment variables to add, as KEY=VAL. KEY= unsets a variable runOnHost bool // When cross-compiling, run this test on the host instead of guest // variant, if non-empty, is a name used to distinguish different // configurations of the same test package(s). If set and omitVariant is false, // the Package field in test2json output is rewritten to pkg:variant. variant string // omitVariant indicates that variant is used solely for the dist test name and // that the set of test names run by each variant (including empty) of a package // is non-overlapping. omitVariant bool // We have both pkg and pkgs as a convenience. Both may be set, in which // case they will be combined. At least one must be set. pkgs []string // Multiple packages to test pkg string // A single package to test testFlags []string // Additional flags accepted by this test } // bgCommand returns a go test Cmd and a post-Run flush function. The result // will write its output to stdout and stderr. If stdout==stderr, bgCommand // ensures Writes are serialized. The caller should call flush() after Cmd exits. func (opts *goTest) bgCommand(t *tester, stdout, stderr io.Writer) (cmd *exec.Cmd, flush func()) { build, run, pkgs, testFlags, setupCmd := opts.buildArgs(t) // Combine the flags. args := append([]string{"test"}, build...) if t.compileOnly { args = append(args, "-c", "-o", os.DevNull) } else { args = append(args, run...) } args = append(args, pkgs...) if !t.compileOnly { args = append(args, testFlags...) } cmd = exec.Command(gorootBinGo, args...) setupCmd(cmd) if t.json && opts.variant != "" && !opts.omitVariant { // Rewrite Package in the JSON output to be pkg:variant. When omitVariant // is true, pkg.TestName is already unambiguous, so we don't need to // rewrite the Package field. // // We only want to process JSON on the child's stdout. Ideally if // stdout==stderr, we would also use the same testJSONFilter for // cmd.Stdout and cmd.Stderr in order to keep the underlying // interleaving of writes, but then it would see even partial writes // interleaved, which would corrupt the JSON. So, we only process // cmd.Stdout. This has another consequence though: if stdout==stderr, // we have to serialize Writes in case the Writer is not concurrent // safe. If we were just passing stdout/stderr through to exec, it would // do this for us, but since we're wrapping stdout, we have to do it // ourselves. if stdout == stderr { stdout = &lockedWriter{w: stdout} stderr = stdout } f := &testJSONFilter{w: stdout, variant: opts.variant} cmd.Stdout = f flush = f.Flush } else { cmd.Stdout = stdout flush = func() {} } cmd.Stderr = stderr return cmd, flush } // run runs a go test and returns an error if it does not succeed. func (opts *goTest) run(t *tester) error { cmd, flush := opts.bgCommand(t, os.Stdout, os.Stderr) err := cmd.Run() flush() return err } // buildArgs is in internal helper for goTest that constructs the elements of // the "go test" command line. build is the flags for building the test. run is // the flags for running the test. pkgs is the list of packages to build and // run. testFlags is the list of flags to pass to the test package. // // The caller must call setupCmd on the resulting exec.Cmd to set its directory // and environment. func (opts *goTest) buildArgs(t *tester) (build, run, pkgs, testFlags []string, setupCmd func(*exec.Cmd)) { run = append(run, "-count=1") // Disallow caching if opts.timeout != 0 { d := opts.timeout * time.Duration(t.timeoutScale) run = append(run, "-timeout="+d.String()) } else if t.timeoutScale != 1 { const goTestDefaultTimeout = 10 * time.Minute // Default value of go test -timeout flag. run = append(run, "-timeout="+(goTestDefaultTimeout*time.Duration(t.timeoutScale)).String()) } if opts.short || t.short { run = append(run, "-short") } var tags []string if t.iOS() { tags = append(tags, "lldb") } if noOpt { tags = append(tags, "noopt") } tags = append(tags, opts.tags...) if len(tags) > 0 { build = append(build, "-tags="+strings.Join(tags, ",")) } if t.race || opts.race { build = append(build, "-race") } if t.msan { build = append(build, "-msan") } if t.asan { build = append(build, "-asan") } if opts.bench { // Run no tests. run = append(run, "-run=^$") // Run benchmarks briefly as a smoke test. run = append(run, "-bench=.*", "-benchtime=.1s") } else if opts.runTests != "" { run = append(run, "-run="+opts.runTests) } if opts.cpu != "" { run = append(run, "-cpu="+opts.cpu) } if t.json { run = append(run, "-json") } if opts.gcflags != "" { build = append(build, "-gcflags=all="+opts.gcflags) } if opts.ldflags != "" { build = append(build, "-ldflags="+opts.ldflags) } if opts.buildmode != "" { build = append(build, "-buildmode="+opts.buildmode) } pkgs = opts.packages() runOnHost := opts.runOnHost && (goarch != gohostarch || goos != gohostos) needTestFlags := len(opts.testFlags) > 0 || runOnHost if needTestFlags { testFlags = append([]string{"-args"}, opts.testFlags...) } if runOnHost { // -target is a special flag understood by tests that can run on the host testFlags = append(testFlags, "-target="+goos+"/"+goarch) } setupCmd = func(cmd *exec.Cmd) { setDir(cmd, filepath.Join(goroot, "src")) if len(opts.env) != 0 { for _, kv := range opts.env { if i := strings.Index(kv, "="); i < 0 { unsetEnv(cmd, kv[:len(kv)-1]) } else { setEnv(cmd, kv[:i], kv[i+1:]) } } } if runOnHost { setEnv(cmd, "GOARCH", gohostarch) setEnv(cmd, "GOOS", gohostos) } } return } // packages returns the full list of packages to be run by this goTest. This // will always include at least one package. func (opts *goTest) packages() []string { pkgs := opts.pkgs if opts.pkg != "" { pkgs = append(pkgs[:len(pkgs):len(pkgs)], opts.pkg) } if len(pkgs) == 0 { panic("no packages") } return pkgs } // ranGoTest and stdMatches are state closed over by the stdlib // testing func in registerStdTest below. The tests are run // sequentially, so there's no need for locks. // // ranGoBench and benchMatches are the same, but are only used // in -race mode. var ( ranGoTest bool stdMatches []string ranGoBench bool benchMatches []string ) func (t *tester) registerStdTest(pkg string) { const stdTestHeading = "Testing packages." // known to addTest for a safety check gcflags := gogcflags name := testName(pkg, "") if t.runRx == nil || t.runRx.MatchString(name) == t.runRxWant { stdMatches = append(stdMatches, pkg) } t.addTest(name, stdTestHeading, func(dt *distTest) error { if ranGoTest { return nil } t.runPending(dt) timelog("start", dt.name) defer timelog("end", dt.name) ranGoTest = true timeoutSec := 180 * time.Second for _, pkg := range stdMatches { if pkg == "cmd/go" { timeoutSec *= 3 break } } return (&goTest{ timeout: timeoutSec, gcflags: gcflags, pkgs: stdMatches, }).run(t) }) } func (t *tester) registerRaceBenchTest(pkg string) { const raceBenchHeading = "Running benchmarks briefly." // known to addTest for a safety check name := testName(pkg, "racebench") if t.runRx == nil || t.runRx.MatchString(name) == t.runRxWant { benchMatches = append(benchMatches, pkg) } t.addTest(name, raceBenchHeading, func(dt *distTest) error { if ranGoBench { return nil } t.runPending(dt) timelog("start", dt.name) defer timelog("end", dt.name) ranGoBench = true return (&goTest{ variant: "racebench", omitVariant: true, // The only execution of benchmarks in dist; benchmark names are guaranteed not to overlap with test names. timeout: 1200 * time.Second, // longer timeout for race with benchmarks race: true, bench: true, cpu: "4", pkgs: benchMatches, }).run(t) }) } func (t *tester) registerTests() { // registerStdTestSpecially tracks import paths in the standard library // whose test registration happens in a special way. // // These tests *must* be able to run normally as part of "go test std cmd", // even if they are also registered separately by dist, because users often // run go test directly. Use skips or build tags in preference to expanding // this list. registerStdTestSpecially := map[string]bool{ // testdir can run normally as part of "go test std cmd", but because // it's a very large test, we register is specially as several shards to // enable better load balancing on sharded builders. Ideally the build // system would know how to shard any large test package. "cmd/internal/testdir": true, } // Fast path to avoid the ~1 second of `go list std cmd` when // the caller lists specific tests to run. (as the continuous // build coordinator does). if len(t.runNames) > 0 { for _, name := range t.runNames { if !strings.Contains(name, ":") { t.registerStdTest(name) } else if strings.HasSuffix(name, ":racebench") { t.registerRaceBenchTest(strings.TrimSuffix(name, ":racebench")) } } } else { // Use a format string to only list packages and commands that have tests. const format = "{{if (or .TestGoFiles .XTestGoFiles)}}{{.ImportPath}}{{end}}" cmd := exec.Command(gorootBinGo, "list", "-f", format) if t.race { cmd.Args = append(cmd.Args, "-tags=race") } cmd.Args = append(cmd.Args, "std", "cmd") cmd.Stderr = new(bytes.Buffer) all, err := cmd.Output() if err != nil { fatalf("Error running go list std cmd: %v:\n%s", err, cmd.Stderr) } pkgs := strings.Fields(string(all)) for _, pkg := range pkgs { if registerStdTestSpecially[pkg] { continue } t.registerStdTest(pkg) } if t.race { for _, pkg := range pkgs { if t.packageHasBenchmarks(pkg) { t.registerRaceBenchTest(pkg) } } } } if t.race { return } // Test the os/user package in the pure-Go mode too. if !t.compileOnly { t.registerTest("os/user with tag osusergo", &goTest{ variant: "osusergo", timeout: 300 * time.Second, tags: []string{"osusergo"}, pkg: "os/user", }) t.registerTest("hash/maphash purego implementation", &goTest{ variant: "purego", timeout: 300 * time.Second, tags: []string{"purego"}, pkg: "hash/maphash", }) } // Test ios/amd64 for the iOS simulator. if goos == "darwin" && goarch == "amd64" && t.cgoEnabled { t.registerTest("GOOS=ios on darwin/amd64", &goTest{ variant: "amd64ios", timeout: 300 * time.Second, runTests: "SystemRoots", env: []string{"GOOS=ios", "CGO_ENABLED=1"}, pkg: "crypto/x509", }) } // Runtime CPU tests. if !t.compileOnly && t.hasParallelism() { t.registerTest("GOMAXPROCS=2 runtime -cpu=1,2,4 -quick", &goTest{ variant: "cpu124", timeout: 300 * time.Second, cpu: "1,2,4", short: true, testFlags: []string{"-quick"}, // We set GOMAXPROCS=2 in addition to -cpu=1,2,4 in order to test runtime bootstrap code, // creation of first goroutines and first garbage collections in the parallel setting. env: []string{"GOMAXPROCS=2"}, pkg: "runtime", }) } // morestack tests. We only run these on in long-test mode // (with GO_TEST_SHORT=false) because the runtime test is // already quite long and mayMoreStackMove makes it about // twice as slow. if !t.compileOnly && !t.short { // hooks is the set of maymorestack hooks to test with. hooks := []string{"mayMoreStackPreempt", "mayMoreStackMove"} // hookPkgs is the set of package patterns to apply // the maymorestack hook to. hookPkgs := []string{"runtime/...", "reflect", "sync"} // unhookPkgs is the set of package patterns to // exclude from hookPkgs. unhookPkgs := []string{"runtime/testdata/..."} for _, hook := range hooks { // Construct the build flags to use the // maymorestack hook in the compiler and // assembler. We pass this via the GOFLAGS // environment variable so that it applies to // both the test itself and to binaries built // by the test. goFlagsList := []string{} for _, flag := range []string{"-gcflags", "-asmflags"} { for _, hookPkg := range hookPkgs { goFlagsList = append(goFlagsList, flag+"="+hookPkg+"=-d=maymorestack=runtime."+hook) } for _, unhookPkg := range unhookPkgs { goFlagsList = append(goFlagsList, flag+"="+unhookPkg+"=") } } goFlags := strings.Join(goFlagsList, " ") t.registerTest("maymorestack="+hook, &goTest{ variant: hook, timeout: 600 * time.Second, short: true, env: []string{"GOFLAGS=" + goFlags}, pkgs: []string{"runtime", "reflect", "sync"}, }) } } // Test that internal linking of standard packages does not // require libgcc. This ensures that we can install a Go // release on a system that does not have a C compiler // installed and still build Go programs (that don't use cgo). for _, pkg := range cgoPackages { if !t.internalLink() { break } // ARM libgcc may be Thumb, which internal linking does not support. if goarch == "arm" { break } // What matters is that the tests build and start up. // Skip expensive tests, especially x509 TestSystemRoots. run := "^Test[^CS]" if pkg == "net" { run = "TestTCPStress" } t.registerTest("Testing without libgcc.", &goTest{ variant: "nolibgcc", ldflags: "-linkmode=internal -libgcc=none", runTests: run, pkg: pkg, }) } // Stub out following test on alpine until 54354 resolved. builderName := os.Getenv("GO_BUILDER_NAME") disablePIE := strings.HasSuffix(builderName, "-alpine") // Test internal linking of PIE binaries where it is supported. if t.internalLinkPIE() && !disablePIE { t.registerTest("internal linking of -buildmode=pie", &goTest{ variant: "pie_internal", timeout: 60 * time.Second, buildmode: "pie", ldflags: "-linkmode=internal", env: []string{"CGO_ENABLED=0"}, pkg: "reflect", }) // Also test a cgo package. if t.cgoEnabled && t.internalLink() && !disablePIE { t.registerTest("internal linking of -buildmode=pie", &goTest{ variant: "pie_internal", timeout: 60 * time.Second, buildmode: "pie", ldflags: "-linkmode=internal", pkg: "os/user", }) } } // sync tests if t.hasParallelism() { t.registerTest("sync -cpu=10", &goTest{ variant: "cpu10", timeout: 120 * time.Second, cpu: "10", pkg: "sync", }) } if t.raceDetectorSupported() { t.registerRaceTests() } const cgoHeading = "Testing cgo" if t.cgoEnabled { t.registerCgoTests(cgoHeading) } if goos == "wasip1" { t.registerTest("wasip1 host tests", &goTest{ variant: "host", pkg: "runtime/internal/wasitest", timeout: 1 * time.Minute, runOnHost: true, }) } if goos != "android" && !t.iOS() { // Only start multiple test dir shards on builders, // where they get distributed to multiple machines. // See issues 20141 and 31834. nShards := 1 if os.Getenv("GO_BUILDER_NAME") != "" { nShards = 10 } if n, err := strconv.Atoi(os.Getenv("GO_TEST_SHARDS")); err == nil { nShards = n } for shard := 0; shard < nShards; shard++ { id := fmt.Sprintf("%d_%d", shard, nShards) t.registerTest("../test", &goTest{ variant: id, omitVariant: true, // Shards of the same Go package; tests are guaranteed not to overlap. pkg: "cmd/internal/testdir", testFlags: []string{fmt.Sprintf("-shard=%d", shard), fmt.Sprintf("-shards=%d", nShards)}, runOnHost: true, }, ) } } // Only run the API check on fast development platforms. // Every platform checks the API on every GOOS/GOARCH/CGO_ENABLED combination anyway, // so we really only need to run this check once anywhere to get adequate coverage. // To help developers avoid trybot-only failures, we try to run on typical developer machines // which is darwin,linux,windows/amd64 and darwin/arm64. if goos == "darwin" || ((goos == "linux" || goos == "windows") && goarch == "amd64") { t.registerTest("API check", &goTest{variant: "check", pkg: "cmd/api", timeout: 5 * time.Minute, testFlags: []string{"-check"}}) } } // addTest adds an arbitrary test callback to the test list. // // name must uniquely identify the test and heading must be non-empty. func (t *tester) addTest(name, heading string, fn func(*distTest) error) { if t.testNames[name] { panic("duplicate registered test name " + name) } if heading == "" { panic("empty heading") } // Two simple checks for cases that would conflict with the fast path in registerTests. if !strings.Contains(name, ":") && heading != "Testing packages." { panic("empty variant is reserved exclusively for registerStdTest") } else if strings.HasSuffix(name, ":racebench") && heading != "Running benchmarks briefly." { panic("racebench variant is reserved exclusively for registerRaceBenchTest") } if t.testNames == nil { t.testNames = make(map[string]bool) } t.testNames[name] = true t.tests = append(t.tests, distTest{ name: name, heading: heading, fn: fn, }) } type registerTestOpt interface { isRegisterTestOpt() } // rtSkipFunc is a registerTest option that runs a skip check function before // running the test. type rtSkipFunc struct { skip func(*distTest) (string, bool) // Return message, true to skip the test } func (rtSkipFunc) isRegisterTestOpt() {} // registerTest registers a test that runs the given goTest. // // Each Go package in goTest will have a corresponding test // ":", which must uniquely identify the test. // // heading and test.variant must be non-empty. func (t *tester) registerTest(heading string, test *goTest, opts ...registerTestOpt) { var skipFunc func(*distTest) (string, bool) for _, opt := range opts { switch opt := opt.(type) { case rtSkipFunc: skipFunc = opt.skip } } // Register each test package as a separate test. register1 := func(test *goTest) { if test.variant == "" { panic("empty variant") } name := testName(test.pkg, test.variant) t.addTest(name, heading, func(dt *distTest) error { if skipFunc != nil { msg, skip := skipFunc(dt) if skip { t.printSkip(test, msg) return nil } } w := &work{dt: dt} w.cmd, w.flush = test.bgCommand(t, &w.out, &w.out) t.worklist = append(t.worklist, w) return nil }) } if test.pkg != "" && len(test.pkgs) == 0 { // Common case. Avoid copying. register1(test) return } // TODO(dmitshur,austin): It might be better to unify the execution of 'go test pkg' // invocations for the same variant to be done with a single 'go test pkg1 pkg2 pkg3' // command, just like it's already done in registerStdTest and registerRaceBenchTest. // Those methods accumulate matched packages in stdMatches and benchMatches slices, // and we can extend that mechanism to work for all other equal variant registrations. // Do the simple thing to start with. for _, pkg := range test.packages() { test1 := *test test1.pkg, test1.pkgs = pkg, nil register1(&test1) } } func (t *tester) printSkip(test *goTest, msg string) { if !t.json { fmt.Println(msg) return } type event struct { Time time.Time Action string Package string Output string `json:",omitempty"` } out := json.NewEncoder(os.Stdout) for _, pkg := range test.packages() { ev := event{Time: time.Now(), Package: testName(pkg, test.variant), Action: "start"} out.Encode(ev) ev.Action = "output" ev.Output = msg out.Encode(ev) ev.Action = "skip" ev.Output = "" out.Encode(ev) } } // dirCmd constructs a Cmd intended to be run in the foreground. // The command will be run in dir, and Stdout and Stderr will go to os.Stdout // and os.Stderr. func (t *tester) dirCmd(dir string, cmdline ...interface{}) *exec.Cmd { bin, args := flattenCmdline(cmdline) cmd := exec.Command(bin, args...) if filepath.IsAbs(dir) { setDir(cmd, dir) } else { setDir(cmd, filepath.Join(goroot, dir)) } cmd.Stdout = os.Stdout cmd.Stderr = os.Stderr if vflag > 1 { errprintf("%s\n", strings.Join(cmd.Args, " ")) } return cmd } // flattenCmdline flattens a mixture of string and []string as single list // and then interprets it as a command line: first element is binary, then args. func flattenCmdline(cmdline []interface{}) (bin string, args []string) { var list []string for _, x := range cmdline { switch x := x.(type) { case string: list = append(list, x) case []string: list = append(list, x...) default: panic("invalid dirCmd argument type: " + reflect.TypeOf(x).String()) } } bin = list[0] if !filepath.IsAbs(bin) { panic("command is not absolute: " + bin) } return bin, list[1:] } func (t *tester) iOS() bool { return goos == "ios" } func (t *tester) out(v string) { if t.json { return } if t.banner == "" { return } fmt.Println("\n" + t.banner + v) } // extLink reports whether the current goos/goarch supports // external linking. This should match the test in determineLinkMode // in cmd/link/internal/ld/config.go. func (t *tester) extLink() bool { if goarch == "ppc64" && goos != "aix" { return false } return true } func (t *tester) internalLink() bool { if gohostos == "dragonfly" { // linkmode=internal fails on dragonfly since errno is a TLS relocation. return false } if goos == "android" { return false } if goos == "ios" { return false } if goos == "windows" && goarch == "arm64" { return false } // Internally linking cgo is incomplete on some architectures. // https://golang.org/issue/10373 // https://golang.org/issue/14449 if goarch == "loong64" || goarch == "mips64" || goarch == "mips64le" || goarch == "mips" || goarch == "mipsle" || goarch == "riscv64" { return false } if goos == "aix" { // linkmode=internal isn't supported. return false } return true } func (t *tester) internalLinkPIE() bool { switch goos + "-" + goarch { case "darwin-amd64", "darwin-arm64", "linux-amd64", "linux-arm64", "linux-ppc64le", "android-arm64", "windows-amd64", "windows-386", "windows-arm": return true } return false } // supportedBuildMode reports whether the given build mode is supported. func (t *tester) supportedBuildmode(mode string) bool { switch mode { case "c-archive", "c-shared", "shared", "plugin", "pie": default: fatalf("internal error: unknown buildmode %s", mode) return false } return buildModeSupported("gc", mode, goos, goarch) } func (t *tester) registerCgoTests(heading string) { cgoTest := func(variant string, subdir, linkmode, buildmode string, opts ...registerTestOpt) *goTest { gt := &goTest{ variant: variant, pkg: "cmd/cgo/internal/" + subdir, buildmode: buildmode, } var ldflags []string if linkmode != "auto" { // "auto" is the default, so avoid cluttering the command line for "auto" ldflags = append(ldflags, "-linkmode="+linkmode) } if linkmode == "internal" { gt.tags = append(gt.tags, "internal") if buildmode == "pie" { gt.tags = append(gt.tags, "internal_pie") } } if buildmode == "static" { // This isn't actually a Go buildmode, just a convenient way to tell // cgoTest we want static linking. gt.buildmode = "" if linkmode == "external" { ldflags = append(ldflags, `-extldflags "-static -pthread"`) } else if linkmode == "auto" { gt.env = append(gt.env, "CGO_LDFLAGS=-static -pthread") } else { panic("unknown linkmode with static build: " + linkmode) } gt.tags = append(gt.tags, "static") } gt.ldflags = strings.Join(ldflags, " ") t.registerTest(heading, gt, opts...) return gt } // test, testtls, and testnocgo are run with linkmode="auto", buildmode="" // as part of go test cmd. Here we only have to register the non-default // build modes of these tests. // Stub out various buildmode=pie tests on alpine until 54354 resolved. builderName := os.Getenv("GO_BUILDER_NAME") disablePIE := strings.HasSuffix(builderName, "-alpine") if t.internalLink() { cgoTest("internal", "test", "internal", "") } os := gohostos p := gohostos + "/" + goarch switch { case os == "darwin", os == "windows": if !t.extLink() { break } // test linkmode=external, but __thread not supported, so skip testtls. cgoTest("external", "test", "external", "") gt := cgoTest("external-s", "test", "external", "") gt.ldflags += " -s" if t.supportedBuildmode("pie") && !disablePIE { cgoTest("auto-pie", "test", "auto", "pie") if t.internalLink() && t.internalLinkPIE() { cgoTest("internal-pie", "test", "internal", "pie") } } case os == "aix", os == "android", os == "dragonfly", os == "freebsd", os == "linux", os == "netbsd", os == "openbsd": gt := cgoTest("external-g0", "test", "external", "") gt.env = append(gt.env, "CGO_CFLAGS=-g0 -fdiagnostics-color") cgoTest("external", "testtls", "external", "") switch { case os == "aix": // no static linking case p == "freebsd/arm": // -fPIC compiled tls code will use __tls_get_addr instead // of __aeabi_read_tp, however, on FreeBSD/ARM, __tls_get_addr // is implemented in rtld-elf, so -fPIC isn't compatible with // static linking on FreeBSD/ARM with clang. (cgo depends on // -fPIC fundamentally.) default: // Check for static linking support var staticCheck rtSkipFunc ccName := compilerEnvLookup("CC", defaultcc, goos, goarch) cc, err := exec.LookPath(ccName) if err != nil { staticCheck.skip = func(*distTest) (string, bool) { return fmt.Sprintf("$CC (%q) not found, skip cgo static linking test.", ccName), true } } else { cmd := t.dirCmd("src/cmd/cgo/internal/test", cc, "-xc", "-o", "/dev/null", "-static", "-") cmd.Stdin = strings.NewReader("int main() {}") cmd.Stdout, cmd.Stderr = nil, nil // Discard output if err := cmd.Run(); err != nil { // Skip these tests staticCheck.skip = func(*distTest) (string, bool) { return "No support for static linking found (lacks libc.a?), skip cgo static linking test.", true } } } // Doing a static link with boringcrypto gets // a C linker warning on Linux. // in function `bio_ip_and_port_to_socket_and_addr': // warning: Using 'getaddrinfo' in statically linked applications requires at runtime the shared libraries from the glibc version used for linking if staticCheck.skip == nil && goos == "linux" && strings.Contains(goexperiment, "boringcrypto") { staticCheck.skip = func(*distTest) (string, bool) { return "skipping static linking check on Linux when using boringcrypto to avoid C linker warning about getaddrinfo", true } } // Static linking tests if goos != "android" && p != "netbsd/arm" { // TODO(#56629): Why does this fail on netbsd-arm? cgoTest("static", "testtls", "external", "static", staticCheck) } cgoTest("external", "testnocgo", "external", "", staticCheck) if goos != "android" { cgoTest("static", "testnocgo", "external", "static", staticCheck) cgoTest("static", "test", "external", "static", staticCheck) // -static in CGO_LDFLAGS triggers a different code path // than -static in -extldflags, so test both. // See issue #16651. if goarch != "loong64" { // TODO(#56623): Why does this fail on loong64? cgoTest("auto-static", "test", "auto", "static", staticCheck) } } // PIE linking tests if t.supportedBuildmode("pie") && !disablePIE { cgoTest("auto-pie", "test", "auto", "pie") if t.internalLink() && t.internalLinkPIE() { cgoTest("internal-pie", "test", "internal", "pie") } cgoTest("auto-pie", "testtls", "auto", "pie") cgoTest("auto-pie", "testnocgo", "auto", "pie") } } } } // run pending test commands, in parallel, emitting headers as appropriate. // When finished, emit header for nextTest, which is going to run after the // pending commands are done (and runPending returns). // A test should call runPending if it wants to make sure that it is not // running in parallel with earlier tests, or if it has some other reason // for needing the earlier tests to be done. func (t *tester) runPending(nextTest *distTest) { worklist := t.worklist t.worklist = nil for _, w := range worklist { w.start = make(chan bool) w.end = make(chan bool) // w.cmd must be set up to write to w.out. We can't check that, but we // can check for easy mistakes. if w.cmd.Stdout == nil || w.cmd.Stdout == os.Stdout || w.cmd.Stderr == nil || w.cmd.Stderr == os.Stderr { panic("work.cmd.Stdout/Stderr must be redirected") } go func(w *work) { if !<-w.start { timelog("skip", w.dt.name) w.printSkip(t, "skipped due to earlier error") } else { timelog("start", w.dt.name) w.err = w.cmd.Run() if w.flush != nil { w.flush() } if w.err != nil { if isUnsupportedVMASize(w) { timelog("skip", w.dt.name) w.out.Reset() w.printSkip(t, "skipped due to unsupported VMA") w.err = nil } } } timelog("end", w.dt.name) w.end <- true }(w) } started := 0 ended := 0 var last *distTest for ended < len(worklist) { for started < len(worklist) && started-ended < maxbg { w := worklist[started] started++ w.start <- !t.failed || t.keepGoing } w := worklist[ended] dt := w.dt if t.lastHeading != dt.heading { t.lastHeading = dt.heading t.out(dt.heading) } if dt != last { // Assumes all the entries for a single dt are in one worklist. last = w.dt if vflag > 0 { fmt.Printf("# go tool dist test -run=^%s$\n", dt.name) } } if vflag > 1 { errprintf("%s\n", strings.Join(w.cmd.Args, " ")) } ended++ <-w.end os.Stdout.Write(w.out.Bytes()) // We no longer need the output, so drop the buffer. w.out = bytes.Buffer{} if w.err != nil { log.Printf("Failed: %v", w.err) t.failed = true } } if t.failed && !t.keepGoing { fatalf("FAILED") } if dt := nextTest; dt != nil { if t.lastHeading != dt.heading { t.lastHeading = dt.heading t.out(dt.heading) } if vflag > 0 { fmt.Printf("# go tool dist test -run=^%s$\n", dt.name) } } } func (t *tester) hasBash() bool { switch gohostos { case "windows", "plan9": return false } return true } // hasParallelism is a copy of the function // internal/testenv.HasParallelism, which can't be used here // because cmd/dist can not import internal packages during bootstrap. func (t *tester) hasParallelism() bool { switch goos { case "js", "wasip1": return false } return true } func (t *tester) raceDetectorSupported() bool { if gohostos != goos { return false } if !t.cgoEnabled { return false } if !raceDetectorSupported(goos, goarch) { return false } // The race detector doesn't work on Alpine Linux: // golang.org/issue/14481 if isAlpineLinux() { return false } // NetBSD support is unfinished. // golang.org/issue/26403 if goos == "netbsd" { return false } return true } func isAlpineLinux() bool { if runtime.GOOS != "linux" { return false } fi, err := os.Lstat("/etc/alpine-release") return err == nil && fi.Mode().IsRegular() } func (t *tester) registerRaceTests() { hdr := "Testing race detector" t.registerTest(hdr, &goTest{ variant: "race", race: true, runTests: "Output", pkg: "runtime/race", }) t.registerTest(hdr, &goTest{ variant: "race", race: true, runTests: "TestParse|TestEcho|TestStdinCloseRace|TestClosedPipeRace|TestTypeRace|TestFdRace|TestFdReadRace|TestFileCloseRace", pkgs: []string{"flag", "net", "os", "os/exec", "encoding/gob"}, }) // We don't want the following line, because it // slows down all.bash (by 10 seconds on my laptop). // The race builder should catch any error here, but doesn't. // TODO(iant): Figure out how to catch this. // t.registerTest(hdr, &goTest{variant: "race", race: true, runTests: "TestParallelTest", pkg: "cmd/go"}) if t.cgoEnabled { // Building cmd/cgo/internal/test takes a long time. // There are already cgo-enabled packages being tested with the race detector. // We shouldn't need to redo all of cmd/cgo/internal/test too. // The race buildler will take care of this. // t.registerTest(hdr, &goTest{variant: "race", race: true, env: []string{"GOTRACEBACK=2"}, pkg: "cmd/cgo/internal/test"}) } if t.extLink() { // Test with external linking; see issue 9133. t.registerTest(hdr, &goTest{ variant: "race-external", race: true, ldflags: "-linkmode=external", runTests: "TestParse|TestEcho|TestStdinCloseRace", pkgs: []string{"flag", "os/exec"}, }) } } // cgoPackages is the standard packages that use cgo. var cgoPackages = []string{ "net", "os/user", } var funcBenchmark = []byte("\nfunc Benchmark") // packageHasBenchmarks reports whether pkg has benchmarks. // On any error, it conservatively returns true. // // This exists just to eliminate work on the builders, since compiling // a test in race mode just to discover it has no benchmarks costs a // second or two per package, and this function returns false for // about 100 packages. func (t *tester) packageHasBenchmarks(pkg string) bool { pkgDir := filepath.Join(goroot, "src", pkg) d, err := os.Open(pkgDir) if err != nil { return true // conservatively } defer d.Close() names, err := d.Readdirnames(-1) if err != nil { return true // conservatively } for _, name := range names { if !strings.HasSuffix(name, "_test.go") { continue } slurp, err := os.ReadFile(filepath.Join(pkgDir, name)) if err != nil { return true // conservatively } if bytes.Contains(slurp, funcBenchmark) { return true } } return false } // makeGOROOTUnwritable makes all $GOROOT files & directories non-writable to // check that no tests accidentally write to $GOROOT. func (t *tester) makeGOROOTUnwritable() (undo func()) { dir := os.Getenv("GOROOT") if dir == "" { panic("GOROOT not set") } type pathMode struct { path string mode os.FileMode } var dirs []pathMode // in lexical order undo = func() { for i := range dirs { os.Chmod(dirs[i].path, dirs[i].mode) // best effort } } filepath.WalkDir(dir, func(path string, d fs.DirEntry, err error) error { if suffix := strings.TrimPrefix(path, dir+string(filepath.Separator)); suffix != "" { if suffix == ".git" { // Leave Git metadata in whatever state it was in. It may contain a lot // of files, and it is highly unlikely that a test will try to modify // anything within that directory. return filepath.SkipDir } } if err != nil { return nil } info, err := d.Info() if err != nil { return nil } mode := info.Mode() if mode&0222 != 0 && (mode.IsDir() || mode.IsRegular()) { dirs = append(dirs, pathMode{path, mode}) } return nil }) // Run over list backward to chmod children before parents. for i := len(dirs) - 1; i >= 0; i-- { err := os.Chmod(dirs[i].path, dirs[i].mode&^0222) if err != nil { dirs = dirs[i:] // Only undo what we did so far. undo() fatalf("failed to make GOROOT read-only: %v", err) } } return undo } // raceDetectorSupported is a copy of the function // internal/platform.RaceDetectorSupported, which can't be used here // because cmd/dist can not import internal packages during bootstrap. // The race detector only supports 48-bit VMA on arm64. But we don't have // a good solution to check VMA size(See https://golang.org/issue/29948) // raceDetectorSupported will always return true for arm64. But race // detector tests may abort on non 48-bit VMA configuration, the tests // will be marked as "skipped" in this case. func raceDetectorSupported(goos, goarch string) bool { switch goos { case "linux": return goarch == "amd64" || goarch == "ppc64le" || goarch == "arm64" || goarch == "s390x" case "darwin": return goarch == "amd64" || goarch == "arm64" case "freebsd", "netbsd", "openbsd", "windows": return goarch == "amd64" default: return false } } // buildModeSupports is a copy of the function // internal/platform.BuildModeSupported, which can't be used here // because cmd/dist can not import internal packages during bootstrap. func buildModeSupported(compiler, buildmode, goos, goarch string) bool { if compiler == "gccgo" { return true } platform := goos + "/" + goarch switch buildmode { case "archive": return true case "c-archive": switch goos { case "aix", "darwin", "ios", "windows": return true case "linux": switch goarch { case "386", "amd64", "arm", "armbe", "arm64", "arm64be", "loong64", "ppc64le", "riscv64", "s390x": // linux/ppc64 not supported because it does // not support external linking mode yet. return true default: // Other targets do not support -shared, // per ParseFlags in // cmd/compile/internal/base/flag.go. // For c-archive the Go tool passes -shared, // so that the result is suitable for inclusion // in a PIE or shared library. return false } case "freebsd": return goarch == "amd64" } return false case "c-shared": switch platform { case "linux/amd64", "linux/arm", "linux/arm64", "linux/loong64", "linux/386", "linux/ppc64le", "linux/riscv64", "linux/s390x", "android/amd64", "android/arm", "android/arm64", "android/386", "freebsd/amd64", "darwin/amd64", "darwin/arm64", "windows/amd64", "windows/386", "windows/arm64": return true } return false case "default": return true case "exe": return true case "pie": switch platform { case "linux/386", "linux/amd64", "linux/arm", "linux/arm64", "linux/loong64", "linux/ppc64le", "linux/riscv64", "linux/s390x", "android/amd64", "android/arm", "android/arm64", "android/386", "freebsd/amd64", "darwin/amd64", "darwin/arm64", "ios/amd64", "ios/arm64", "aix/ppc64", "windows/386", "windows/amd64", "windows/arm", "windows/arm64": return true } return false case "shared": switch platform { case "linux/386", "linux/amd64", "linux/arm", "linux/arm64", "linux/ppc64le", "linux/s390x": return true } return false case "plugin": switch platform { case "linux/amd64", "linux/arm", "linux/arm64", "linux/386", "linux/s390x", "linux/ppc64le", "android/amd64", "android/386", "darwin/amd64", "darwin/arm64", "freebsd/amd64": return true } return false default: return false } } // isUnsupportedVMASize reports whether the failure is caused by an unsupported // VMA for the race detector (for example, running the race detector on an // arm64 machine configured with 39-bit VMA) func isUnsupportedVMASize(w *work) bool { unsupportedVMA := []byte("unsupported VMA range") return w.dt.name == "race" && bytes.Contains(w.out.Bytes(), unsupportedVMA) } // isEnvSet reports whether the environment variable evar is // set in the environment. func isEnvSet(evar string) bool { evarEq := evar + "=" for _, e := range os.Environ() { if strings.HasPrefix(e, evarEq) { return true } } return false }