]> Cypherpunks.ru repositories - gostls13.git/blob - src/cmd/dist/buildtool.go
internal/saferio: new package to avoid OOM
[gostls13.git] / src / cmd / dist / buildtool.go
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 // Build toolchain using Go 1.4.
6 //
7 // The general strategy is to copy the source files we need into
8 // a new GOPATH workspace, adjust import paths appropriately,
9 // invoke the Go 1.4 go command to build those sources,
10 // and then copy the binaries back.
11
12 package main
13
14 import (
15         "fmt"
16         "os"
17         "path/filepath"
18         "regexp"
19         "strings"
20 )
21
22 // bootstrapDirs is a list of directories holding code that must be
23 // compiled with a Go 1.4 toolchain to produce the bootstrapTargets.
24 // All directories in this list are relative to and must be below $GOROOT/src.
25 //
26 // The list has two kinds of entries: names beginning with cmd/ with
27 // no other slashes, which are commands, and other paths, which are packages
28 // supporting the commands. Packages in the standard library can be listed
29 // if a newer copy needs to be substituted for the Go 1.4 copy when used
30 // by the command packages. Paths ending with /... automatically
31 // include all packages within subdirectories as well.
32 // These will be imported during bootstrap as bootstrap/name, like bootstrap/math/big.
33 var bootstrapDirs = []string{
34         "cmd/asm",
35         "cmd/asm/internal/...",
36         "cmd/cgo",
37         "cmd/compile",
38         "cmd/compile/internal/...",
39         "cmd/internal/archive",
40         "cmd/internal/bio",
41         "cmd/internal/codesign",
42         "cmd/internal/dwarf",
43         "cmd/internal/edit",
44         "cmd/internal/gcprog",
45         "cmd/internal/goobj",
46         "cmd/internal/notsha256",
47         "cmd/internal/obj/...",
48         "cmd/internal/objabi",
49         "cmd/internal/pkgpath",
50         "cmd/internal/quoted",
51         "cmd/internal/src",
52         "cmd/internal/sys",
53         "cmd/link",
54         "cmd/link/internal/...",
55         "compress/flate",
56         "compress/zlib",
57         "container/heap",
58         "debug/dwarf",
59         "debug/elf",
60         "debug/macho",
61         "debug/pe",
62         "go/constant",
63         "internal/buildcfg",
64         "internal/goexperiment",
65         "internal/goversion",
66         "internal/pkgbits",
67         "internal/race",
68         "internal/saferio",
69         "internal/unsafeheader",
70         "internal/xcoff",
71         "math/big",
72         "math/bits",
73         "sort",
74         "strconv",
75 }
76
77 // File prefixes that are ignored by go/build anyway, and cause
78 // problems with editor generated temporary files (#18931).
79 var ignorePrefixes = []string{
80         ".",
81         "_",
82         "#",
83 }
84
85 // File suffixes that use build tags introduced since Go 1.4.
86 // These must not be copied into the bootstrap build directory.
87 // Also ignore test files.
88 var ignoreSuffixes = []string{
89         "_arm64.s",
90         "_arm64.go",
91         "_loong64.s",
92         "_loong64.go",
93         "_riscv64.s",
94         "_riscv64.go",
95         "_wasm.s",
96         "_wasm.go",
97         "_test.s",
98         "_test.go",
99 }
100
101 var tryDirs = []string{
102         "sdk/go1.17",
103         "go1.17",
104 }
105
106 func bootstrapBuildTools() {
107         goroot_bootstrap := os.Getenv("GOROOT_BOOTSTRAP")
108         if goroot_bootstrap == "" {
109                 home := os.Getenv("HOME")
110                 goroot_bootstrap = pathf("%s/go1.4", home)
111                 for _, d := range tryDirs {
112                         if p := pathf("%s/%s", home, d); isdir(p) {
113                                 goroot_bootstrap = p
114                         }
115                 }
116         }
117         xprintf("Building Go toolchain1 using %s.\n", goroot_bootstrap)
118
119         mkbuildcfg(pathf("%s/src/internal/buildcfg/zbootstrap.go", goroot))
120         mkobjabi(pathf("%s/src/cmd/internal/objabi/zbootstrap.go", goroot))
121
122         // Use $GOROOT/pkg/bootstrap as the bootstrap workspace root.
123         // We use a subdirectory of $GOROOT/pkg because that's the
124         // space within $GOROOT where we store all generated objects.
125         // We could use a temporary directory outside $GOROOT instead,
126         // but it is easier to debug on failure if the files are in a known location.
127         workspace := pathf("%s/pkg/bootstrap", goroot)
128         xremoveall(workspace)
129         xatexit(func() { xremoveall(workspace) })
130         base := pathf("%s/src/bootstrap", workspace)
131         xmkdirall(base)
132
133         // Copy source code into $GOROOT/pkg/bootstrap and rewrite import paths.
134         writefile("module bootstrap\n", pathf("%s/%s", base, "go.mod"), 0)
135         for _, dir := range bootstrapDirs {
136                 recurse := strings.HasSuffix(dir, "/...")
137                 dir = strings.TrimSuffix(dir, "/...")
138                 filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
139                         if err != nil {
140                                 fatalf("walking bootstrap dirs failed: %v: %v", path, err)
141                         }
142
143                         name := filepath.Base(path)
144                         src := pathf("%s/src/%s", goroot, path)
145                         dst := pathf("%s/%s", base, path)
146
147                         if info.IsDir() {
148                                 if !recurse && path != dir || name == "testdata" {
149                                         return filepath.SkipDir
150                                 }
151
152                                 xmkdirall(dst)
153                                 if path == "cmd/cgo" {
154                                         // Write to src because we need the file both for bootstrap
155                                         // and for later in the main build.
156                                         mkzdefaultcc("", pathf("%s/zdefaultcc.go", src))
157                                         mkzdefaultcc("", pathf("%s/zdefaultcc.go", dst))
158                                 }
159                                 return nil
160                         }
161
162                         for _, pre := range ignorePrefixes {
163                                 if strings.HasPrefix(name, pre) {
164                                         return nil
165                                 }
166                         }
167                         for _, suf := range ignoreSuffixes {
168                                 if strings.HasSuffix(name, suf) {
169                                         return nil
170                                 }
171                         }
172
173                         text := bootstrapRewriteFile(src)
174                         writefile(text, dst, 0)
175                         return nil
176                 })
177         }
178
179         // Set up environment for invoking Go 1.4 go command.
180         // GOROOT points at Go 1.4 GOROOT,
181         // GOPATH points at our bootstrap workspace,
182         // GOBIN is empty, so that binaries are installed to GOPATH/bin,
183         // and GOOS, GOHOSTOS, GOARCH, and GOHOSTOS are empty,
184         // so that Go 1.4 builds whatever kind of binary it knows how to build.
185         // Restore GOROOT, GOPATH, and GOBIN when done.
186         // Don't bother with GOOS, GOHOSTOS, GOARCH, and GOHOSTARCH,
187         // because setup will take care of those when bootstrapBuildTools returns.
188
189         defer os.Setenv("GOROOT", os.Getenv("GOROOT"))
190         os.Setenv("GOROOT", goroot_bootstrap)
191
192         defer os.Setenv("GOPATH", os.Getenv("GOPATH"))
193         os.Setenv("GOPATH", workspace)
194
195         defer os.Setenv("GOBIN", os.Getenv("GOBIN"))
196         os.Setenv("GOBIN", "")
197
198         os.Setenv("GOOS", "")
199         os.Setenv("GOHOSTOS", "")
200         os.Setenv("GOARCH", "")
201         os.Setenv("GOHOSTARCH", "")
202
203         // Run Go 1.4 to build binaries. Use -gcflags=-l to disable inlining to
204         // workaround bugs in Go 1.4's compiler. See discussion thread:
205         // https://groups.google.com/d/msg/golang-dev/Ss7mCKsvk8w/Gsq7VYI0AwAJ
206         // Use the math_big_pure_go build tag to disable the assembly in math/big
207         // which may contain unsupported instructions.
208         // Use the purego build tag to disable other assembly code,
209         // such as in cmd/internal/notsha256.
210         // Note that if we are using Go 1.10 or later as bootstrap, the -gcflags=-l
211         // only applies to the final cmd/go binary, but that's OK: if this is Go 1.10
212         // or later we don't need to disable inlining to work around bugs in the Go 1.4 compiler.
213         cmd := []string{
214                 pathf("%s/bin/go", goroot_bootstrap),
215                 "install",
216                 "-gcflags=-l",
217                 "-tags=math_big_pure_go compiler_bootstrap purego",
218         }
219         if vflag > 0 {
220                 cmd = append(cmd, "-v")
221         }
222         if tool := os.Getenv("GOBOOTSTRAP_TOOLEXEC"); tool != "" {
223                 cmd = append(cmd, "-toolexec="+tool)
224         }
225         cmd = append(cmd, "bootstrap/cmd/...")
226         run(base, ShowOutput|CheckExit, cmd...)
227
228         // Copy binaries into tool binary directory.
229         for _, name := range bootstrapDirs {
230                 if !strings.HasPrefix(name, "cmd/") {
231                         continue
232                 }
233                 name = name[len("cmd/"):]
234                 if !strings.Contains(name, "/") {
235                         copyfile(pathf("%s/%s%s", tooldir, name, exe), pathf("%s/bin/%s%s", workspace, name, exe), writeExec)
236                 }
237         }
238
239         if vflag > 0 {
240                 xprintf("\n")
241         }
242 }
243
244 var ssaRewriteFileSubstring = filepath.FromSlash("src/cmd/compile/internal/ssa/rewrite")
245
246 // isUnneededSSARewriteFile reports whether srcFile is a
247 // src/cmd/compile/internal/ssa/rewriteARCHNAME.go file for an
248 // architecture that isn't for the given GOARCH.
249 //
250 // When unneeded is true archCaps is the rewrite base filename without
251 // the "rewrite" prefix or ".go" suffix: AMD64, 386, ARM, ARM64, etc.
252 func isUnneededSSARewriteFile(srcFile, goArch string) (archCaps string, unneeded bool) {
253         if !strings.Contains(srcFile, ssaRewriteFileSubstring) {
254                 return "", false
255         }
256         fileArch := strings.TrimSuffix(strings.TrimPrefix(filepath.Base(srcFile), "rewrite"), ".go")
257         if fileArch == "" {
258                 return "", false
259         }
260         b := fileArch[0]
261         if b == '_' || ('a' <= b && b <= 'z') {
262                 return "", false
263         }
264         archCaps = fileArch
265         fileArch = strings.ToLower(fileArch)
266         fileArch = strings.TrimSuffix(fileArch, "splitload")
267         if fileArch == goArch {
268                 return "", false
269         }
270         if fileArch == strings.TrimSuffix(goArch, "le") {
271                 return "", false
272         }
273         return archCaps, true
274 }
275
276 func bootstrapRewriteFile(srcFile string) string {
277         // During bootstrap, generate dummy rewrite files for
278         // irrelevant architectures. We only need to build a bootstrap
279         // binary that works for the current gohostarch.
280         // This saves 6+ seconds of bootstrap.
281         if archCaps, ok := isUnneededSSARewriteFile(srcFile, gohostarch); ok {
282                 return fmt.Sprintf(`// Code generated by go tool dist; DO NOT EDIT.
283
284 package ssa
285
286 func rewriteValue%s(v *Value) bool { panic("unused during bootstrap") }
287 func rewriteBlock%s(b *Block) bool { panic("unused during bootstrap") }
288 `, archCaps, archCaps)
289         }
290
291         return bootstrapFixImports(srcFile)
292 }
293
294 func bootstrapFixImports(srcFile string) string {
295         text := readfile(srcFile)
296         if !strings.Contains(srcFile, "/cmd/") && !strings.Contains(srcFile, `\cmd\`) {
297                 text = regexp.MustCompile(`\bany\b`).ReplaceAllString(text, "interface{}")
298         }
299         lines := strings.SplitAfter(text, "\n")
300         inBlock := false
301         for i, line := range lines {
302                 if strings.HasPrefix(line, "import (") {
303                         inBlock = true
304                         continue
305                 }
306                 if inBlock && strings.HasPrefix(line, ")") {
307                         inBlock = false
308                         continue
309                 }
310                 if strings.HasPrefix(line, `import "`) || strings.HasPrefix(line, `import . "`) ||
311                         inBlock && (strings.HasPrefix(line, "\t\"") || strings.HasPrefix(line, "\t. \"") || strings.HasPrefix(line, "\texec \"")) {
312                         line = strings.Replace(line, `"cmd/`, `"bootstrap/cmd/`, -1)
313                         for _, dir := range bootstrapDirs {
314                                 if strings.HasPrefix(dir, "cmd/") {
315                                         continue
316                                 }
317                                 line = strings.Replace(line, `"`+dir+`"`, `"bootstrap/`+dir+`"`, -1)
318                         }
319                         lines[i] = line
320                 }
321         }
322
323         lines[0] = "// Code generated by go tool dist; DO NOT EDIT.\n// This is a bootstrap copy of " + srcFile + "\n\n//line " + srcFile + ":1\n" + lines[0]
324
325         return strings.Join(lines, "")
326 }