113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package opcodescsv
|
|
|
|
import (
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"golang.org/x/arch/x86/x86csv"
|
|
)
|
|
|
|
type Alias struct {
|
|
Opcode string
|
|
DataSize int
|
|
NumOperands int
|
|
}
|
|
|
|
// BuildAliasMap constructs a map from AT&T/GNU/Intel to Go syntax.
|
|
func BuildAliasMap(is []*x86csv.Inst) (map[Alias]string, error) {
|
|
m := map[Alias]string{}
|
|
for _, i := range is {
|
|
if skip(i) {
|
|
continue
|
|
}
|
|
|
|
s, err := strconv.Atoi("0" + i.DataSize)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, alt := range []string{i.IntelOpcode(), i.GNUOpcode()} {
|
|
if strings.ToUpper(alt) != i.GoOpcode() {
|
|
a := Alias{
|
|
Opcode: strings.ToLower(alt),
|
|
DataSize: s,
|
|
NumOperands: len(i.GoArgs()),
|
|
}
|
|
m[a] = i.GoOpcode()
|
|
}
|
|
}
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
type OperandOrder uint8
|
|
|
|
const (
|
|
UnknownOrder = iota
|
|
IntelOrder
|
|
ReverseIntelOrder
|
|
CMP3Order
|
|
)
|
|
|
|
// BuildOrderMap collects operand order information from the instruction list.
|
|
func BuildOrderMap(is []*x86csv.Inst) map[string]OperandOrder {
|
|
s := map[string]OperandOrder{}
|
|
for _, i := range is {
|
|
if skip(i) {
|
|
continue
|
|
}
|
|
s[i.GoOpcode()] = order(i)
|
|
}
|
|
return s
|
|
}
|
|
|
|
// order categorizes the operand order of an instruction.
|
|
func order(i *x86csv.Inst) OperandOrder {
|
|
// Is it Intel order already?
|
|
intel := i.IntelArgs()
|
|
if reflect.DeepEqual(i.GoArgs(), intel) {
|
|
return IntelOrder
|
|
}
|
|
|
|
// Check if it's reverse Intel.
|
|
for l, r := 0, len(intel)-1; l < r; l, r = l+1, r-1 {
|
|
intel[l], intel[r] = intel[r], intel[l]
|
|
}
|
|
if reflect.DeepEqual(i.GoArgs(), intel) {
|
|
return ReverseIntelOrder
|
|
}
|
|
|
|
// Otherwise we could be in the bizarre special-case of 3-argument CMP instructions.
|
|
//
|
|
// Reference: https://github.com/golang/arch/blob/b19384d3c130858bb31a343ea8fce26be71b5998/x86/x86spec/format.go#L138-L144
|
|
//
|
|
// case "CMPPD", "CMPPS", "CMPSD", "CMPSS":
|
|
// // rotate destination to end but don't swap comparison operands
|
|
// if len(args) == 3 {
|
|
// args[0], args[1], args[2] = args[2], args[0], args[1]
|
|
// break
|
|
// }
|
|
// fallthrough
|
|
//
|
|
switch i.GoOpcode() {
|
|
case "CMPPD", "CMPPS", "CMPSD", "CMPSS":
|
|
if len(i.GoArgs()) == 3 {
|
|
return CMP3Order
|
|
}
|
|
}
|
|
|
|
return UnknownOrder
|
|
}
|
|
|
|
// skip decides whether to ignore the instruction for analysis purposes.
|
|
func skip(i *x86csv.Inst) bool {
|
|
switch {
|
|
case strings.Contains(i.GoOpcode(), "/"):
|
|
return true
|
|
case i.Mode64 == "I": // Invalid in 64-bit mode.
|
|
return true
|
|
}
|
|
return false
|
|
}
|