blob: 0542f6488a470ce93a2ff864be6b2d3bd45b2285 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
package generate
import (
"bufio"
"fmt"
"io"
"regexp"
"strings"
)
var secRe = regexp.MustCompile(`^SEC.*sys_((?:enter|exit)_[a-z_0-9]+)`)
// ExtractTracepoints reads generated C code and extracts tracepoint names from
// SEC annotations, producing the generated_tracepoints.go content.
func ExtractTracepoints(r io.Reader) (string, error) {
scanner := bufio.NewScanner(r)
var tracepoints []string
for scanner.Scan() {
line := scanner.Text()
if m := secRe.FindStringSubmatch(line); m != nil {
tracepoints = append(tracepoints, "sys_"+m[1])
}
}
if err := scanner.Err(); err != nil {
return "", fmt.Errorf("scanning input: %w", err)
}
return formatTracepointsGo(tracepoints), nil
}
func formatTracepointsGo(tracepoints []string) string {
var b strings.Builder
b.WriteString("// Code generated - don't change manually!\n")
b.WriteString("package tracepoints\n\n")
b.WriteString("var List = []string{\n")
for _, tp := range tracepoints {
fmt.Fprintf(&b, "\t%q,\n", tp)
}
b.WriteString("}\n")
return b.String()
}
|
