-
Notifications
You must be signed in to change notification settings - Fork 6
/
LinkCableMultiboot.hpp
277 lines (230 loc) 路 9.45 KB
/
LinkCableMultiboot.hpp
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
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
#ifndef LINK_CABLE_MULTIBOOT_H
#define LINK_CABLE_MULTIBOOT_H
// --------------------------------------------------------------------------
// A Multiboot tool to send small programs from one GBA to up to 3 slaves.
// --------------------------------------------------------------------------
// Usage:
// - 1) Include this header in your main.cpp file and add:
// LinkCableMultiboot* linkCableMultiboot = new LinkCableMultiboot();
// - 2) Send the ROM:
// LinkCableMultiboot::Result result = linkCableMultiboot->sendRom(
// romBytes, // for current ROM, use: ((const u8*)MEM_EWRAM)
// romLength, // in bytes, should be multiple of 0x10
// []() {
// u16 keys = ~REG_KEYS & KEY_ANY;
// return keys & KEY_START;
// // (when this returns true, the transfer will be canceled)
// }
// );
// // `result` should be LinkCableMultiboot::Result::SUCCESS
// --------------------------------------------------------------------------
// considerations:
// - for better results, turn on the GBAs after calling the `sendRom` method!
// --------------------------------------------------------------------------
#include <tonc_bios.h>
#include <tonc_core.h>
#include "LinkRawCable.hpp"
#define LINK_CABLE_MULTIBOOT_MIN_ROM_SIZE (0x100 + 0xc0)
#define LINK_CABLE_MULTIBOOT_MAX_ROM_SIZE (256 * 1024)
#define LINK_CABLE_MULTIBOOT_WAIT_BEFORE_RETRY ((160 + 68) * 60)
#define LINK_CABLE_MULTIBOOT_WAIT_BEFORE_TRANSFER 50
#define LINK_CABLE_MULTIBOOT_DETECTION_TRIES 16
#define LINK_CABLE_MULTIBOOT_PALETTE_DATA 0x93
#define LINK_CABLE_MULTIBOOT_CLIENTS 3
#define LINK_CABLE_MULTIBOOT_CLIENT_NO_DATA 0xff
#define LINK_CABLE_MULTIBOOT_HANDSHAKE 0x6200
#define LINK_CABLE_MULTIBOOT_HANDSHAKE_RESPONSE 0x7200
#define LINK_CABLE_MULTIBOOT_CONFIRM_CLIENTS 0x6100
#define LINK_CABLE_MULTIBOOT_SEND_PALETTE 0x6300
#define LINK_CABLE_MULTIBOOT_HANDSHAKE_DATA 0x11
#define LINK_CABLE_MULTIBOOT_CONFIRM_HANDSHAKE_DATA 0x6400
#define LINK_CABLE_MULTIBOOT_ACK_RESPONSE 0x73
#define LINK_CABLE_MULTIBOOT_HEADER_SIZE 0xC0
#define LINK_CABLE_MULTIBOOT_SWI_MULTIPLAYER_MODE 1
#define LINK_CABLE_MULTIBOOT_MAX_BAUD_RATE LinkRawCable::BaudRate::BAUD_RATE_3
#define LINK_CABLE_MULTIBOOT_TRY(CALL) \
do { \
partialResult = CALL; \
} while (partialResult == NEEDS_RETRY); \
if (partialResult == ABORTED) \
return error(CANCELED); \
else if (partialResult == ERROR) \
return error(FAILURE_DURING_HANDSHAKE);
static volatile char LINK_CABLE_MULTIBOOT_VERSION[] =
"LinkCableMultiboot/v6.3.0";
const u8 LINK_CABLE_MULTIBOOT_CLIENT_IDS[] = {0b0010, 0b0100, 0b1000};
class LinkCableMultiboot {
public:
enum Result {
SUCCESS,
INVALID_SIZE,
CANCELED,
FAILURE_DURING_HANDSHAKE,
FAILURE_DURING_TRANSFER
};
template <typename F>
Result sendRom(const u8* rom, u32 romSize, F cancel) {
if (romSize < LINK_CABLE_MULTIBOOT_MIN_ROM_SIZE)
return INVALID_SIZE;
if (romSize > LINK_CABLE_MULTIBOOT_MAX_ROM_SIZE)
return INVALID_SIZE;
if ((romSize % 0x10) != 0)
return INVALID_SIZE;
PartialResult partialResult;
MultiBootParam multiBootParameters;
multiBootParameters.client_data[0] = LINK_CABLE_MULTIBOOT_CLIENT_NO_DATA;
multiBootParameters.client_data[1] = LINK_CABLE_MULTIBOOT_CLIENT_NO_DATA;
multiBootParameters.client_data[2] = LINK_CABLE_MULTIBOOT_CLIENT_NO_DATA;
multiBootParameters.palette_data = LINK_CABLE_MULTIBOOT_PALETTE_DATA;
multiBootParameters.client_bit = 0;
multiBootParameters.boot_srcp = (u8*)rom + LINK_CABLE_MULTIBOOT_HEADER_SIZE;
multiBootParameters.boot_endp = (u8*)rom + romSize;
LINK_CABLE_MULTIBOOT_TRY(detectClients(multiBootParameters, cancel))
LINK_CABLE_MULTIBOOT_TRY(confirmClients(multiBootParameters, cancel))
LINK_CABLE_MULTIBOOT_TRY(sendHeader(rom, cancel))
LINK_CABLE_MULTIBOOT_TRY(confirmHeader(multiBootParameters, cancel))
LINK_CABLE_MULTIBOOT_TRY(reconfirm(multiBootParameters, cancel))
LINK_CABLE_MULTIBOOT_TRY(sendPalette(multiBootParameters, cancel))
multiBootParameters.handshake_data = (LINK_CABLE_MULTIBOOT_HANDSHAKE_DATA +
multiBootParameters.client_data[0] +
multiBootParameters.client_data[1] +
multiBootParameters.client_data[2]) %
256;
LINK_CABLE_MULTIBOOT_TRY(confirmHandshakeData(multiBootParameters, cancel))
int result = MultiBoot(&multiBootParameters,
LINK_CABLE_MULTIBOOT_SWI_MULTIPLAYER_MODE);
linkRawCable->deactivate();
return result == 1 ? FAILURE_DURING_TRANSFER : SUCCESS;
}
~LinkCableMultiboot() { delete linkRawCable; }
private:
LinkRawCable* linkRawCable = new LinkRawCable();
enum PartialResult { NEEDS_RETRY, FINISHED, ABORTED, ERROR };
struct Responses {
u16 d[LINK_CABLE_MULTIBOOT_CLIENTS];
};
template <typename F>
PartialResult detectClients(MultiBootParam& multiBootParameters, F cancel) {
linkRawCable->activate(LINK_CABLE_MULTIBOOT_MAX_BAUD_RATE);
for (u32 t = 0; t < LINK_CABLE_MULTIBOOT_DETECTION_TRIES; t++) {
auto response =
linkRawCable->transfer(LINK_CABLE_MULTIBOOT_HANDSHAKE, cancel);
if (cancel())
return ABORTED;
for (u32 i = 0; i < LINK_CABLE_MULTIBOOT_CLIENTS; i++) {
if ((response.data[1 + i] & 0xfff0) ==
LINK_CABLE_MULTIBOOT_HANDSHAKE_RESPONSE) {
auto clientId = response.data[1 + i] & 0xf;
switch (clientId) {
case 0b0010:
case 0b0100:
case 0b1000: {
multiBootParameters.client_bit |= clientId;
break;
}
default:
return NEEDS_RETRY;
}
}
}
}
if (multiBootParameters.client_bit == 0) {
linkRawCable->deactivate();
wait(LINK_CABLE_MULTIBOOT_WAIT_BEFORE_RETRY);
return NEEDS_RETRY;
}
return FINISHED;
}
template <typename F>
PartialResult confirmClients(MultiBootParam& multiBootParameters, F cancel) {
return compare(
multiBootParameters,
LINK_CABLE_MULTIBOOT_CONFIRM_CLIENTS | multiBootParameters.client_bit,
LINK_CABLE_MULTIBOOT_HANDSHAKE_RESPONSE, cancel);
}
template <typename F>
PartialResult confirmHeader(MultiBootParam& multiBootParameters, F cancel) {
return compare(multiBootParameters, LINK_CABLE_MULTIBOOT_HANDSHAKE, 0,
cancel);
}
template <typename F>
PartialResult reconfirm(MultiBootParam& multiBootParameters, F cancel) {
return compare(multiBootParameters, LINK_CABLE_MULTIBOOT_HANDSHAKE,
LINK_CABLE_MULTIBOOT_HANDSHAKE_RESPONSE, cancel);
}
template <typename F>
PartialResult sendHeader(const u8* rom, F cancel) {
u16* headerOut = (u16*)rom;
for (int i = 0; i < LINK_CABLE_MULTIBOOT_HEADER_SIZE; i += 2) {
linkRawCable->transfer(*(headerOut++), cancel);
if (cancel())
return ABORTED;
}
return FINISHED;
}
template <typename F>
PartialResult sendPalette(MultiBootParam& multiBootParameters, F cancel) {
auto data =
LINK_CABLE_MULTIBOOT_SEND_PALETTE | LINK_CABLE_MULTIBOOT_PALETTE_DATA;
auto response = linkRawCable->transfer(data, cancel);
if (cancel())
return ABORTED;
for (u32 i = 0; i < LINK_CABLE_MULTIBOOT_CLIENTS; i++) {
if (response.data[1 + i] >> 8 == LINK_CABLE_MULTIBOOT_ACK_RESPONSE)
multiBootParameters.client_data[i] = response.data[1 + i] & 0xff;
}
for (u32 i = 0; i < LINK_CABLE_MULTIBOOT_CLIENTS; i++) {
u8 clientId = LINK_CABLE_MULTIBOOT_CLIENT_IDS[i];
bool isClientConnected = multiBootParameters.client_bit & clientId;
if (isClientConnected && multiBootParameters.client_data[i] ==
LINK_CABLE_MULTIBOOT_CLIENT_NO_DATA)
return NEEDS_RETRY;
}
return FINISHED;
}
template <typename F>
PartialResult confirmHandshakeData(MultiBootParam& multiBootParameters,
F cancel) {
u16 data = LINK_CABLE_MULTIBOOT_CONFIRM_HANDSHAKE_DATA |
multiBootParameters.handshake_data;
auto response = linkRawCable->transfer(data, cancel);
if (cancel())
return ABORTED;
return (response.data[1] >> 8) == LINK_CABLE_MULTIBOOT_ACK_RESPONSE
? FINISHED
: ERROR;
}
template <typename F>
PartialResult compare(MultiBootParam& multiBootParameters,
u16 data,
u16 expectedResponse,
F cancel) {
auto response = linkRawCable->transfer(data, cancel);
if (cancel())
return ABORTED;
for (u32 i = 0; i < LINK_CABLE_MULTIBOOT_CLIENTS; i++) {
u8 clientId = LINK_CABLE_MULTIBOOT_CLIENT_IDS[i];
u16 expectedResponseWithId = expectedResponse | clientId;
bool isClientConnected = multiBootParameters.client_bit & clientId;
if (isClientConnected && response.data[1 + i] != expectedResponseWithId)
return ERROR;
}
return FINISHED;
}
Result error(Result error) {
linkRawCable->deactivate();
return error;
}
void wait(u32 verticalLines) {
u32 count = 0;
u32 vCount = REG_VCOUNT;
while (count < verticalLines) {
if (REG_VCOUNT != vCount) {
count++;
vCount = REG_VCOUNT;
}
};
}
};
extern LinkCableMultiboot* linkCableMultiboot;
#endif // LINK_CABLE_MULTIBOOT_H