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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
// Copyright 2017 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
//go:build ignore
// +build ignore
package main
import (
"bytes"
"fmt"
"io/ioutil"
"os"
"regexp"
"github.com/google/syzkaller/pkg/tool"
)
func main() {
out, err := os.Create("generated.go")
if err != nil {
tool.Fail(err)
}
defer out.Close()
data, err := ioutil.ReadFile("../../executor/common.h")
if err != nil {
tool.Fail(err)
}
executorFilenames := []string{
"common_linux.h",
"common_akaros.h",
"common_bsd.h",
"common_openbsd.h",
"common_fuchsia.h",
"common_windows.h",
"common_test.h",
"common_kvm_amd64.h",
"common_kvm_arm64.h",
"common_kvm_ppc64.h",
"common_usb_linux.h",
"common_usb_netbsd.h",
"common_usb.h",
"common_ext.h",
"android/android_seccomp.h",
"kvm.h",
"kvm_amd64.S.h",
"kvm_ppc64le.S.h",
"common_zlib.h",
}
data = replaceIncludes(executorFilenames, "../../executor/", data)
androidFilenames := []string{
"arm64_app_policy.h",
"arm_app_policy.h",
"x86_64_app_policy.h",
"x86_app_policy.h",
"arm64_system_policy.h",
"arm_system_policy.h",
"x86_64_system_policy.h",
"x86_system_policy.h",
}
data = replaceIncludes(androidFilenames, "../../executor/android/", data)
// Remove `//` comments, but keep lines which start with `//%`.
for _, remove := range []string{
"(\n|^)\\s*//$",
"(\n|^)\\s*//[^%].*",
"\\s*//$",
"\\s*//[^%].*",
} {
data = regexp.MustCompile(remove).ReplaceAll(data, nil)
}
fmt.Fprintf(out, "// Code generated by gen.go from executor/*.h. DO NOT EDIT.\n\n")
fmt.Fprintf(out, "package csource\n\nvar commonHeader = `\n")
out.Write(data)
fmt.Fprintf(out, "`\n")
}
func replaceIncludes(filenames []string, location string, data []byte) []byte {
for _, include := range filenames {
contents, err := ioutil.ReadFile(location + include)
if err != nil {
tool.Fail(err)
}
replace := []byte("#include \"" + include + "\"")
if bytes.Index(data, replace) == -1 {
tool.Failf("can't find %v include", include)
}
data = bytes.Replace(data, replace, contents, -1)
}
return data
}
|
