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