1
2
3
4
5 package debug
6
7 import (
8 "fmt"
9 "runtime"
10 "strconv"
11 "strings"
12 )
13
14
15 func modinfo() string
16
17
18
19
20 func ReadBuildInfo() (info *BuildInfo, ok bool) {
21 data := modinfo()
22 if len(data) < 32 {
23 return nil, false
24 }
25 data = data[16 : len(data)-16]
26 bi, err := ParseBuildInfo(data)
27 if err != nil {
28 return nil, false
29 }
30
31
32
33
34
35 bi.GoVersion = runtime.Version()
36
37 return bi, true
38 }
39
40
41 type BuildInfo struct {
42
43
44 GoVersion string
45
46
47
48 Path string
49
50
51 Main Module
52
53
54
55 Deps []*Module
56
57
58 Settings []BuildSetting
59 }
60
61
62 type Module struct {
63 Path string
64 Version string
65 Sum string
66 Replace *Module
67 }
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89 type BuildSetting struct {
90
91
92
93 Key, Value string
94 }
95
96
97 func quoteKey(key string) bool {
98 return len(key) == 0 || strings.ContainsAny(key, "= \t\r\n\"`")
99 }
100
101
102 func quoteValue(value string) bool {
103 return strings.ContainsAny(value, " \t\r\n\"`")
104 }
105
106
107 func (bi *BuildInfo) String() string {
108 buf := new(strings.Builder)
109 if bi.GoVersion != "" {
110 fmt.Fprintf(buf, "go\t%s\n", bi.GoVersion)
111 }
112 if bi.Path != "" {
113 fmt.Fprintf(buf, "path\t%s\n", bi.Path)
114 }
115 var formatMod func(string, Module)
116 formatMod = func(word string, m Module) {
117 buf.WriteString(word)
118 buf.WriteByte('\t')
119 buf.WriteString(m.Path)
120 buf.WriteByte('\t')
121 buf.WriteString(m.Version)
122 if m.Replace == nil {
123 buf.WriteByte('\t')
124 buf.WriteString(m.Sum)
125 } else {
126 buf.WriteByte('\n')
127 formatMod("=>", *m.Replace)
128 }
129 buf.WriteByte('\n')
130 }
131 if bi.Main != (Module{}) {
132 formatMod("mod", bi.Main)
133 }
134 for _, dep := range bi.Deps {
135 formatMod("dep", *dep)
136 }
137 for _, s := range bi.Settings {
138 key := s.Key
139 if quoteKey(key) {
140 key = strconv.Quote(key)
141 }
142 value := s.Value
143 if quoteValue(value) {
144 value = strconv.Quote(value)
145 }
146 fmt.Fprintf(buf, "build\t%s=%s\n", key, value)
147 }
148
149 return buf.String()
150 }
151
152
153
154
155
156
157
158 func ParseBuildInfo(data string) (bi *BuildInfo, err error) {
159 lineNum := 1
160 defer func() {
161 if err != nil {
162 err = fmt.Errorf("could not parse Go build info: line %d: %w", lineNum, err)
163 }
164 }()
165
166 const (
167 pathLine = "path\t"
168 modLine = "mod\t"
169 depLine = "dep\t"
170 repLine = "=>\t"
171 buildLine = "build\t"
172 newline = "\n"
173 tab = "\t"
174 )
175
176 readModuleLine := func(elem []string) (Module, error) {
177 if len(elem) != 2 && len(elem) != 3 {
178 return Module{}, fmt.Errorf("expected 2 or 3 columns; got %d", len(elem))
179 }
180 version := elem[1]
181 sum := ""
182 if len(elem) == 3 {
183 sum = elem[2]
184 }
185 return Module{
186 Path: elem[0],
187 Version: version,
188 Sum: sum,
189 }, nil
190 }
191
192 bi = new(BuildInfo)
193 var (
194 last *Module
195 line string
196 ok bool
197 )
198
199 for len(data) > 0 {
200 line, data, ok = strings.Cut(data, newline)
201 if !ok {
202 break
203 }
204 switch {
205 case strings.HasPrefix(line, pathLine):
206 elem := line[len(pathLine):]
207 bi.Path = elem
208 case strings.HasPrefix(line, modLine):
209 elem := strings.Split(line[len(modLine):], tab)
210 last = &bi.Main
211 *last, err = readModuleLine(elem)
212 if err != nil {
213 return nil, err
214 }
215 case strings.HasPrefix(line, depLine):
216 elem := strings.Split(line[len(depLine):], tab)
217 last = new(Module)
218 bi.Deps = append(bi.Deps, last)
219 *last, err = readModuleLine(elem)
220 if err != nil {
221 return nil, err
222 }
223 case strings.HasPrefix(line, repLine):
224 elem := strings.Split(line[len(repLine):], tab)
225 if len(elem) != 3 {
226 return nil, fmt.Errorf("expected 3 columns for replacement; got %d", len(elem))
227 }
228 if last == nil {
229 return nil, fmt.Errorf("replacement with no module on previous line")
230 }
231 last.Replace = &Module{
232 Path: elem[0],
233 Version: elem[1],
234 Sum: elem[2],
235 }
236 last = nil
237 case strings.HasPrefix(line, buildLine):
238 kv := line[len(buildLine):]
239 if len(kv) < 1 {
240 return nil, fmt.Errorf("build line missing '='")
241 }
242
243 var key, rawValue string
244 switch kv[0] {
245 case '=':
246 return nil, fmt.Errorf("build line with missing key")
247
248 case '`', '"':
249 rawKey, err := strconv.QuotedPrefix(kv)
250 if err != nil {
251 return nil, fmt.Errorf("invalid quoted key in build line")
252 }
253 if len(kv) == len(rawKey) {
254 return nil, fmt.Errorf("build line missing '=' after quoted key")
255 }
256 if c := kv[len(rawKey)]; c != '=' {
257 return nil, fmt.Errorf("unexpected character after quoted key: %q", c)
258 }
259 key, _ = strconv.Unquote(rawKey)
260 rawValue = kv[len(rawKey)+1:]
261
262 default:
263 var ok bool
264 key, rawValue, ok = strings.Cut(kv, "=")
265 if !ok {
266 return nil, fmt.Errorf("build line missing '=' after key")
267 }
268 if quoteKey(key) {
269 return nil, fmt.Errorf("unquoted key %q must be quoted", key)
270 }
271 }
272
273 var value string
274 if len(rawValue) > 0 {
275 switch rawValue[0] {
276 case '`', '"':
277 var err error
278 value, err = strconv.Unquote(rawValue)
279 if err != nil {
280 return nil, fmt.Errorf("invalid quoted value in build line")
281 }
282
283 default:
284 value = rawValue
285 if quoteValue(value) {
286 return nil, fmt.Errorf("unquoted value %q must be quoted", value)
287 }
288 }
289 }
290
291 bi.Settings = append(bi.Settings, BuildSetting{Key: key, Value: value})
292 }
293 lineNum++
294 }
295 return bi, nil
296 }
297
View as plain text