-
Notifications
You must be signed in to change notification settings - Fork 183
/
Copy pathjmp_riscv64.go
105 lines (95 loc) · 3.53 KB
/
jmp_riscv64.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
102
103
104
105
package gomonkey
import (
"encoding/binary"
)
// buildJmpDirective 为 riscv64 架构生成一段跳转指令，
// 将传入的 64 位地址加载到寄存器 x6（t1）中，
// 然后执行 JALR x0, 0(x6) 实现无条件跳转。
func buildJmpDirective(double uintptr) []byte {
var res []byte
// 将地址转换为 64 位无符号整数
d := uint64(double)
// 将 64 位地址拆分成若干部分：
// imm0: 位 [7:0]
// imm1: 位 [19:8] (12 位)
// imm2: 位 [31:20] (12 位)
// imm3: 位 [43:32] (12 位)
// imm4: 位 [63:44] (20 位)
imm0 := d & 0xff
imm1 := (d >> 8) & 0xfff
imm2 := (d >> 20) & 0xfff
imm3 := (d >> 32) & 0xfff
imm4 := (d >> 44) & 0xfffff
// 依次生成指令：
// 1. LUI x6, imm4 // 将最高 20 位加载到 x6
res = append(res, encodeLUI(6, uint32(imm4))...)
// 2. ADDI x6, x6, imm3 // 加载接下来的 12 位
res = append(res, encodeADDI(6, 6, int32(imm3))...)
// 3. SLLI x6, x6, 12 // 左移 12 位
res = append(res, encodeSLLI(6, 6, 12)...)
// 4. ADDI x6, x6, imm2 // 加载接下来的 12 位
res = append(res, encodeADDI(6, 6, int32(imm2))...)
// 5. SLLI x6, x6, 12 // 再次左移 12 位
res = append(res, encodeSLLI(6, 6, 12)...)
// 6. ADDI x6, x6, imm1 // 加载接下来的 12 位
res = append(res, encodeADDI(6, 6, int32(imm1))...)
// 7. SLLI x6, x6, 8 // 左移 8 位
res = append(res, encodeSLLI(6, 6, 8)...)
// 8. ORI x6, x6, imm0 // 最后加载最低 8 位
res = append(res, encodeORI(6, 6, int32(imm0))...)
// 9. JALR x0, 0(x6) // 跳转到 x6 指定的地址
res = append(res, encodeJALR(0, 6, 0)...)
return res
}
// 以下辅助函数生成各条指令的机器码，均返回 4 字节小端表示。
// LUI 指令格式：
// 31 12 11 7 6 0
// [ imm[31:12] ] [ rd ] [ opcode ]
// opcode LUI 为 0x37。
func encodeLUI(rd int, imm20 uint32) []byte {
inst := (imm20 << 12) | (uint32(rd) << 7) | 0x37
res := make([]byte, 4)
binary.LittleEndian.PutUint32(res, inst)
return res
}
// ADDI 指令格式（用于加载 12 位立即数）：
// 31 20 19 15 14 12 11 7 6 0
// [ imm[11:0] ] [ rs1 ] [funct3] [ rd ] [ opcode ]
// opcode ADDI 为 0x13，funct3 为 0。
func encodeADDI(rd, rs1 int, imm int32) []byte {
inst := ((uint32(imm) & 0xfff) << 20) | (uint32(rs1) << 15) | (uint32(rd) << 7) | 0x13
res := make([]byte, 4)
binary.LittleEndian.PutUint32(res, inst)
return res
}
// SLLI 指令格式：
// 31 26 25 20 19 15 14 12 11 7 6 0
// [ 0 ] [ shamt ] [ rs1 ] [funct3] [ rd ] [ opcode ]
// opcode 为 0x13，funct3 为 1。
func encodeSLLI(rd, rs1, shamt int) []byte {
inst := (uint32(shamt) << 20) | (uint32(rs1) << 15) | (1 << 12) | (uint32(rd) << 7) | 0x13
res := make([]byte, 4)
binary.LittleEndian.PutUint32(res, inst)
return res
}
// ORI 指令格式：
// 31 20 19 15 14 12 11 7 6 0
// [ imm[11:0] ] [ rs1 ] [funct3] [ rd ] [ opcode ]
// opcode 为 0x13，funct3 为 6。
func encodeORI(rd, rs1 int, imm int32) []byte {
inst := ((uint32(imm) & 0xfff) << 20) | (uint32(rs1) << 15) | (6 << 12) | (uint32(rd) << 7) | 0x13
res := make([]byte, 4)
binary.LittleEndian.PutUint32(res, inst)
return res
}
// JALR 指令格式：
// 31 20 19 15 14 12 11 7 6 0
// [ imm[11:0] ] [ rs1 ] [funct3] [ rd ] [ opcode ]
// opcode 为 0x67，funct3 为 0。
// JALR x0, 0(x6) 用于无条件跳转。
func encodeJALR(rd, rs1 int, imm int32) []byte {
inst := ((uint32(imm) & 0xfff) << 20) | (uint32(rs1) << 15) | (uint32(rd) << 7) | 0x67
res := make([]byte, 4)
binary.LittleEndian.PutUint32(res, inst)
return res
}