github.com/bir3/gocompiler@v0.9.2202/src/cmd/link/internal/ld/util.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 ld 6 7 import ( 8 "github.com/bir3/gocompiler/src/cmd/link/internal/loader" 9 "encoding/binary" 10 "fmt" 11 "os" 12 ) 13 14 var atExitFuncs []func() 15 16 func AtExit(f func()) { 17 atExitFuncs = append(atExitFuncs, f) 18 } 19 20 // runAtExitFuncs runs the queued set of AtExit functions. 21 func runAtExitFuncs() { 22 for i := len(atExitFuncs) - 1; i >= 0; i-- { 23 atExitFuncs[i]() 24 } 25 atExitFuncs = nil 26 } 27 28 // Exit exits with code after executing all atExitFuncs. 29 func Exit(code int) { 30 runAtExitFuncs() 31 os.Exit(code) 32 } 33 34 // Exitf logs an error message then calls Exit(2). 35 func Exitf(format string, a ...interface{}) { 36 fmt.Fprintf(os.Stderr, os.Args[0]+": "+format+"\n", a...) 37 nerrors++ 38 if *flagH { 39 panic("error") 40 } 41 Exit(2) 42 } 43 44 // afterErrorAction updates 'nerrors' on error and invokes exit or 45 // panics in the proper circumstances. 46 func afterErrorAction() { 47 nerrors++ 48 if *flagH { 49 panic("error") 50 } 51 if nerrors > 20 { 52 Exitf("too many errors") 53 } 54 } 55 56 // Errorf logs an error message. 57 // 58 // If more than 20 errors have been printed, exit with an error. 59 // 60 // Logging an error means that on exit cmd/link will delete any 61 // output file and return a non-zero error code. 62 // 63 // TODO: remove. Use ctxt.Errorf instead. 64 // All remaining calls use nil as first arg. 65 func Errorf(dummy *int, format string, args ...interface{}) { 66 format += "\n" 67 fmt.Fprintf(os.Stderr, format, args...) 68 afterErrorAction() 69 } 70 71 // Errorf method logs an error message. 72 // 73 // If more than 20 errors have been printed, exit with an error. 74 // 75 // Logging an error means that on exit cmd/link will delete any 76 // output file and return a non-zero error code. 77 func (ctxt *Link) Errorf(s loader.Sym, format string, args ...interface{}) { 78 if ctxt.loader != nil { 79 ctxt.loader.Errorf(s, format, args...) 80 return 81 } 82 // Note: this is not expected to happen very often. 83 format = fmt.Sprintf("sym %d: %s", s, format) 84 format += "\n" 85 fmt.Fprintf(os.Stderr, format, args...) 86 afterErrorAction() 87 } 88 89 func artrim(x []byte) string { 90 i := 0 91 j := len(x) 92 for i < len(x) && x[i] == ' ' { 93 i++ 94 } 95 for j > i && x[j-1] == ' ' { 96 j-- 97 } 98 return string(x[i:j]) 99 } 100 101 func stringtouint32(x []uint32, s string) { 102 for i := 0; len(s) > 0; i++ { 103 var buf [4]byte 104 s = s[copy(buf[:], s):] 105 x[i] = binary.LittleEndian.Uint32(buf[:]) 106 } 107 } 108 109 // contains reports whether v is in s. 110 func contains(s []string, v string) bool { 111 for _, x := range s { 112 if x == v { 113 return true 114 } 115 } 116 return false 117 }