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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
// Copyright 2024 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.
// gen generates instruction tables (ifuzz_types/insns.go) from ARM64 JSON.
package main
import (
"bytes"
"encoding/json"
"fmt"
"os"
"strconv"
"strings"
"github.com/google/syzkaller/pkg/ifuzz/arm64"
"github.com/google/syzkaller/pkg/osutil"
"github.com/google/syzkaller/pkg/serializer"
"github.com/google/syzkaller/pkg/tool"
)
func main() {
if len(os.Args) != 3 {
tool.Failf("usage: gen arm64.json output.file")
}
jsonStr, err := os.ReadFile(os.Args[1])
if err != nil {
tool.Failf("failed to open input file: %v", err)
}
insns := JSONToInsns(jsonStr)
out := new(bytes.Buffer)
fmt.Fprintf(out, `// Code generated by pkg/ifuzz/gen. DO NOT EDIT.
// go:build !codeanalysis
package generated
import (
. "github.com/google/syzkaller/pkg/ifuzz/arm64"
)
func init() {
Register(insns_arm64)
}
var insns_arm64 =
`)
serializer.Write(out, insns)
if err := osutil.WriteFileAtomically(os.Args[2], out.Bytes()); err != nil {
tool.Fail(err)
}
fmt.Fprintf(os.Stderr, "handled %v\n", len(insns))
}
type insnDesc struct {
Name string
Bits string
Arch string
Syntax string
Code string
Alias string
}
func isPrivateInsn(insn arm64.Insn) bool {
switch insn.Name {
case "AT", "DC", "IC", "SYS", "SYSL", "TLBI":
return true
}
return false
}
func JSONToInsns(jsonStr []byte) []*arm64.Insn {
var insnDescriptions []insnDesc
err := json.Unmarshal(jsonStr, &insnDescriptions)
if err != nil {
return nil
}
ret := []*arm64.Insn{}
for _, desc := range insnDescriptions {
mask := uint32(0)
opcode := uint32(0)
curBit := uint(31)
fields := []arm64.InsnField{}
pieces := strings.Split(desc.Bits, "|")
for _, piece := range pieces {
size := uint(1)
pair := strings.Split(piece, ":")
var pattern = piece
if len(pair) == 2 {
size64, err := strconv.ParseUint(pair[1], 10, 0)
if err != nil {
return nil
}
size = uint(size64)
pattern = pair[0]
}
updateOpcode := true
opPart := uint32(0)
maskPart := uint32(0)
if pattern[0:1] != "(" {
number, err := strconv.ParseUint(pattern, 2, 32)
if err != nil {
// This is a named region.
field := arm64.InsnField{
Name: pattern,
Start: curBit,
Length: size,
}
fields = append(fields, field)
updateOpcode = false
} else {
// This is a binary mask.
opPart = uint32(number)
maskPart = ((1 << size) - 1)
}
}
opcode <<= size
mask <<= size
curBit -= size
if updateOpcode {
opcode |= opPart
mask |= maskPart
}
}
templ := arm64.Insn{
Name: desc.Name,
OpcodeMask: mask,
Opcode: opcode,
Fields: fields,
AsUInt32: opcode,
}
templ.Priv = isPrivateInsn(templ)
insn := new(arm64.Insn)
*insn = templ
ret = append(ret, insn)
}
return ret
}
|
