This repository has been archived by the owner on Feb 27, 2025. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathLoad_test.go
101 lines (96 loc) · 4.54 KB
/
Load_test.go
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
package asm_test
import (
"testing"
"github.com/akyoto/asm"
"github.com/akyoto/assert"
)
func TestLoadRegister(t *testing.T) {
usagePatterns := []struct {
Destination string
Source string
Offset byte
ByteCount byte
Code []byte
}{
{"rax", "rax", 1, 8, []byte{0x48, 0x8b, 0x40, 0x01}},
{"eax", "rax", 1, 4, []byte{0x8b, 0x40, 0x01}},
{"ax", "rax", 1, 2, []byte{0x66, 0x8b, 0x40, 0x01}},
{"al", "rax", 1, 1, []byte{0x8a, 0x40, 0x01}},
{"rcx", "rcx", 1, 8, []byte{0x48, 0x8b, 0x49, 0x01}},
{"ecx", "rcx", 1, 4, []byte{0x8b, 0x49, 0x01}},
{"cx", "rcx", 1, 2, []byte{0x66, 0x8b, 0x49, 0x01}},
{"cl", "rcx", 1, 1, []byte{0x8a, 0x49, 0x01}},
{"rdx", "rdx", 1, 8, []byte{0x48, 0x8b, 0x52, 0x01}},
{"edx", "rdx", 1, 4, []byte{0x8b, 0x52, 0x01}},
{"dx", "rdx", 1, 2, []byte{0x66, 0x8b, 0x52, 0x01}},
{"dl", "rdx", 1, 1, []byte{0x8a, 0x52, 0x01}},
{"rbx", "rbx", 1, 8, []byte{0x48, 0x8b, 0x5b, 0x01}},
{"ebx", "rbx", 1, 4, []byte{0x8b, 0x5b, 0x01}},
{"bx", "rbx", 1, 2, []byte{0x66, 0x8b, 0x5b, 0x01}},
{"bl", "rbx", 1, 1, []byte{0x8a, 0x5b, 0x01}},
{"rdi", "rdi", 1, 8, []byte{0x48, 0x8b, 0x7f, 0x01}},
{"edi", "rdi", 1, 4, []byte{0x8b, 0x7f, 0x01}},
{"di", "rdi", 1, 2, []byte{0x66, 0x8b, 0x7f, 0x01}},
{"dl", "rdi", 1, 1, []byte{0x8a, 0x57, 0x01}},
{"rsi", "rsi", 1, 8, []byte{0x48, 0x8b, 0x76, 0x01}},
{"esi", "rsi", 1, 4, []byte{0x8b, 0x76, 0x01}},
{"si", "rsi", 1, 2, []byte{0x66, 0x8b, 0x76, 0x01}},
{"sil", "rsi", 1, 1, []byte{0x40, 0x8a, 0x76, 0x01}},
{"rbp", "rbp", 1, 8, []byte{0x48, 0x8b, 0x6d, 0x01}},
{"ebp", "rbp", 1, 4, []byte{0x8b, 0x6d, 0x01}},
{"bp", "rbp", 1, 2, []byte{0x66, 0x8b, 0x6d, 0x01}},
{"bl", "rbp", 1, 1, []byte{0x8a, 0x5d, 0x01}},
{"rsp", "rsp", 1, 8, []byte{0x48, 0x8b, 0x64, 0x24, 0x01}},
{"esp", "rsp", 1, 4, []byte{0x8b, 0x64, 0x24, 0x01}},
{"sp", "rsp", 1, 2, []byte{0x66, 0x8b, 0x64, 0x24, 0x01}},
{"spl", "rsp", 1, 1, []byte{0x40, 0x8a, 0x64, 0x24, 0x01}},
{"r8", "r8", 1, 8, []byte{0x4d, 0x8b, 0x40, 0x01}},
{"r8d", "r8", 1, 4, []byte{0x45, 0x8b, 0x40, 0x01}},
{"r8w", "r8", 1, 2, []byte{0x66, 0x45, 0x8b, 0x40, 0x01}},
{"r8b", "r8", 1, 1, []byte{0x45, 0x8a, 0x40, 0x01}},
{"r9", "r9", 1, 8, []byte{0x4d, 0x8b, 0x49, 0x01}},
{"r9d", "r9", 1, 4, []byte{0x45, 0x8b, 0x49, 0x01}},
{"r9w", "r9", 1, 2, []byte{0x66, 0x45, 0x8b, 0x49, 0x01}},
{"r9b", "r9", 1, 1, []byte{0x45, 0x8a, 0x49, 0x01}},
{"r10", "r10", 1, 8, []byte{0x4d, 0x8b, 0x52, 0x01}},
{"r10d", "r10", 1, 4, []byte{0x45, 0x8b, 0x52, 0x01}},
{"r10w", "r10", 1, 2, []byte{0x66, 0x45, 0x8b, 0x52, 0x01}},
{"r10b", "r10", 1, 1, []byte{0x45, 0x8a, 0x52, 0x01}},
{"r11", "r11", 1, 8, []byte{0x4d, 0x8b, 0x5b, 0x01}},
{"r11d", "r11", 1, 4, []byte{0x45, 0x8b, 0x5b, 0x01}},
{"r11w", "r11", 1, 2, []byte{0x66, 0x45, 0x8b, 0x5b, 0x01}},
{"r11b", "r11", 1, 1, []byte{0x45, 0x8a, 0x5b, 0x01}},
{"r12", "r12", 1, 8, []byte{0x4d, 0x8b, 0x64, 0x24, 0x01}},
{"r12d", "r12", 1, 4, []byte{0x45, 0x8b, 0x64, 0x24, 0x01}},
{"r12w", "r12", 1, 2, []byte{0x66, 0x45, 0x8b, 0x64, 0x24, 0x01}},
{"r12b", "r12", 1, 1, []byte{0x45, 0x8a, 0x64, 0x24, 0x01}},
{"r13", "r13", 1, 8, []byte{0x4d, 0x8b, 0x6d, 0x01}},
{"r13d", "r13", 1, 4, []byte{0x45, 0x8b, 0x6d, 0x01}},
{"r13w", "r13", 1, 2, []byte{0x66, 0x45, 0x8b, 0x6d, 0x01}},
{"r13b", "r13", 1, 1, []byte{0x45, 0x8a, 0x6d, 0x01}},
{"r14", "r14", 1, 8, []byte{0x4d, 0x8b, 0x76, 0x01}},
{"r14d", "r14", 1, 4, []byte{0x45, 0x8b, 0x76, 0x01}},
{"r14w", "r14", 1, 2, []byte{0x66, 0x45, 0x8b, 0x76, 0x01}},
{"r14b", "r14", 1, 1, []byte{0x45, 0x8a, 0x76, 0x01}},
{"r15", "r15", 1, 8, []byte{0x4d, 0x8b, 0x7f, 0x01}},
{"r15d", "r15", 1, 4, []byte{0x45, 0x8b, 0x7f, 0x01}},
{"r15w", "r15", 1, 2, []byte{0x66, 0x45, 0x8b, 0x7f, 0x01}},
{"r15b", "r15", 1, 1, []byte{0x45, 0x8a, 0x7f, 0x01}},
// Extra tests
{"rcx", "rbx", 0, 8, []byte{0x48, 0x8b, 0x0b}},
{"r12", "rbx", 0, 8, []byte{0x4c, 0x8b, 0x23}},
{"rcx", "rbx", 1, 8, []byte{0x48, 0x8b, 0x4b, 0x01}},
{"r12", "rbx", 1, 8, []byte{0x4c, 0x8b, 0x63, 0x01}},
{"rsp", "rbx", 0, 8, []byte{0x48, 0x8b, 0x23}},
{"rbp", "rbx", 0, 8, []byte{0x48, 0x8b, 0x2b}},
{"r13", "rbx", 0, 8, []byte{0x4c, 0x8b, 0x2b}},
{"r14", "rbx", 0, 8, []byte{0x4c, 0x8b, 0x33}},
{"r15", "rbx", 0, 8, []byte{0x4c, 0x8b, 0x3b}},
}
for _, pattern := range usagePatterns {
a := asm.New()
t.Logf("load %dB %s, [%s+%d]", pattern.ByteCount, pattern.Destination, pattern.Source, pattern.Offset)
a.LoadRegister(pattern.Destination, pattern.Source, pattern.Offset, pattern.ByteCount)
assert.DeepEqual(t, a.Code(), pattern.Code)
}
}