SplitScript/src/codegen/runtime_helpers/gba.rs at master · LiveSplit/SplitScript · GitHub

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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
//! Game Boy Advance hardware-address translation.

use wasm_encoder::{BlockType, Function, Instruction, ValType};

use crate::{abi::AbiImportId, stdlib::StdlibTypeId};

use super::super::GcLayout;
use super::super::imports::Abi;
use super::super::memory_plan::AbiReadScratch;

const BACKEND_POINTER_32: i32 = 2;
const BACKEND_POINTER_64: i32 = 3;
const BACKEND_NOCASH: i32 = 4;

pub(super) fn compile_translate_address(
    abi: &Abi,
    gc: &GcLayout,
    abi_read: AbiReadScratch,
) -> Function {
    let mut function = Function::new([(2, ValType::I64), (1, ValType::I32)]);
    let emulator = 1;
    let address = 2;
    let size = 3;
    let base = 4;
    let pointer = 5;
    let backend = 6;
    let emulator_type = gc.standard_index(StdlibTypeId::GBAEmulator);

    function
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::RefIsNull)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::I64Const(0))
        .instruction(&Instruction::Return)
        .instruction(&Instruction::End)
        // Ensure the selected backend is initialized.
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::StructGet {
            struct_type_index: emulator_type,
            field_index: gc.standard_field_index(crate::stdlib::StdlibFieldId::GBAEmulatorBackend),
        })
        .instruction(&Instruction::LocalTee(backend))
        .instruction(&Instruction::I32Eqz)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::I64Const(0))
        .instruction(&Instruction::Return)
        .instruction(&Instruction::End)
        .instruction(&Instruction::Block(BlockType::Empty))
        // EWRAM: 0x02000000..0x02040000.
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0200_0000))
        .instruction(&Instruction::I32GeU)
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0204_0000))
        .instruction(&Instruction::I32LtU)
        .instruction(&Instruction::I32And)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::LocalGet(size))
        .instruction(&Instruction::I32Const(0x0204_0000))
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Sub)
        .instruction(&Instruction::I32LeU)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::StructGet {
            struct_type_index: emulator_type,
            field_index: gc.standard_field_index(crate::stdlib::StdlibFieldId::GBAEmulatorEwram),
        })
        .instruction(&Instruction::LocalSet(base))
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::StructGet {
            struct_type_index: emulator_type,
            field_index: gc.standard_field_index(crate::stdlib::StdlibFieldId::GBAEmulatorAux1),
        })
        .instruction(&Instruction::LocalSet(pointer))
        .instruction(&Instruction::Br(2))
        .instruction(&Instruction::End)
        .instruction(&Instruction::End)
        // IWRAM: 0x03000000..0x03008000.
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0300_0000))
        .instruction(&Instruction::I32GeU)
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0300_8000))
        .instruction(&Instruction::I32LtU)
        .instruction(&Instruction::I32And)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::LocalGet(size))
        .instruction(&Instruction::I32Const(0x0300_8000))
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Sub)
        .instruction(&Instruction::I32LeU)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::StructGet {
            struct_type_index: emulator_type,
            field_index: gc.standard_field_index(crate::stdlib::StdlibFieldId::GBAEmulatorIwram),
        })
        .instruction(&Instruction::LocalSet(base))
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::StructGet {
            struct_type_index: emulator_type,
            field_index: gc.standard_field_index(crate::stdlib::StdlibFieldId::GBAEmulatorAux2),
        })
        .instruction(&Instruction::LocalSet(pointer))
        .instruction(&Instruction::Br(2))
        .instruction(&Instruction::End)
        .instruction(&Instruction::End)
        .instruction(&Instruction::I64Const(0))
        .instruction(&Instruction::Return)
        .instruction(&Instruction::End)
        // Pointer-backed emulators can move RAM when emulation starts or a
        // ROM is reloaded. Resolve their current base for every script read.
        .instruction(&Instruction::LocalGet(backend))
        .instruction(&Instruction::I32Const(BACKEND_POINTER_32))
        .instruction(&Instruction::I32Eq)
        .instruction(&Instruction::If(BlockType::Empty));
    emit_translate_process_read(&mut function, abi, abi_read, pointer, 4);
    function
        .instruction(&Instruction::I32Const(abi_read.start()))
        .instruction(&Instruction::I32Load(super::super::memarg()))
        .instruction(&Instruction::I64ExtendI32U)
        .instruction(&Instruction::LocalSet(base))
        .instruction(&Instruction::Else)
        .instruction(&Instruction::LocalGet(backend))
        .instruction(&Instruction::I32Const(BACKEND_POINTER_64))
        .instruction(&Instruction::I32Eq)
        .instruction(&Instruction::If(BlockType::Empty));
    emit_translate_process_read(&mut function, abi, abi_read, pointer, 8);
    function
        .instruction(&Instruction::I32Const(abi_read.start()))
        .instruction(&Instruction::I64Load(super::super::memarg()))
        .instruction(&Instruction::LocalSet(base))
        .instruction(&Instruction::Else)
        .instruction(&Instruction::LocalGet(backend))
        .instruction(&Instruction::I32Const(BACKEND_NOCASH))
        .instruction(&Instruction::I32Eq)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::LocalGet(emulator))
        .instruction(&Instruction::StructGet {
            struct_type_index: emulator_type,
            field_index: gc.standard_field_index(crate::stdlib::StdlibFieldId::GBAEmulatorAux1),
        })
        .instruction(&Instruction::LocalSet(pointer));
    emit_translate_process_read(&mut function, abi, abi_read, pointer, 4);
    function
        .instruction(&Instruction::I32Const(abi_read.start()))
        .instruction(&Instruction::I32Load(super::super::memarg()))
        .instruction(&Instruction::I64ExtendI32U)
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0300_0000))
        .instruction(&Instruction::I32LtU)
        .instruction(&Instruction::If(BlockType::Result(ValType::I64)))
        .instruction(&Instruction::I64Const(0x9394))
        .instruction(&Instruction::Else)
        .instruction(&Instruction::I64Const(0x95d4))
        .instruction(&Instruction::End)
        .instruction(&Instruction::I64Add)
        .instruction(&Instruction::LocalSet(pointer));
    emit_translate_process_read(&mut function, abi, abi_read, pointer, 4);
    function
        .instruction(&Instruction::I32Const(abi_read.start()))
        .instruction(&Instruction::I32Load(super::super::memarg()))
        .instruction(&Instruction::I64ExtendI32U)
        .instruction(&Instruction::LocalSet(base))
        .instruction(&Instruction::End)
        .instruction(&Instruction::End)
        .instruction(&Instruction::End)
        .instruction(&Instruction::LocalGet(base))
        .instruction(&Instruction::I64Eqz)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::I64Const(0))
        .instruction(&Instruction::Return)
        .instruction(&Instruction::End)
        .instruction(&Instruction::LocalGet(base))
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0300_0000))
        .instruction(&Instruction::I32LtU)
        .instruction(&Instruction::If(BlockType::Result(ValType::I32)))
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0200_0000))
        .instruction(&Instruction::I32Sub)
        .instruction(&Instruction::Else)
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(0x0300_0000))
        .instruction(&Instruction::I32Sub)
        .instruction(&Instruction::End)
        .instruction(&Instruction::I64ExtendI32U)
        .instruction(&Instruction::I64Add)
        .instruction(&Instruction::End);
    function
}

fn emit_translate_process_read(
    function: &mut Function,
    abi: &Abi,
    abi_read: AbiReadScratch,
    address: u32,
    size: u32,
) {
    function
        .instruction(&Instruction::LocalGet(0))
        .instruction(&Instruction::LocalGet(address))
        .instruction(&Instruction::I32Const(abi_read.destination(size)))
        .instruction(&Instruction::I32Const(size as i32))
        .instruction(&Instruction::Call(abi.function(AbiImportId::ProcessRead)))
        .instruction(&Instruction::I32Eqz)
        .instruction(&Instruction::If(BlockType::Empty))
        .instruction(&Instruction::I64Const(0))
        .instruction(&Instruction::Return)
        .instruction(&Instruction::End);
}