1/* Copyright (c) 2013-2015 Jeffrey Pfau
  2 *
  3 * This Source Code Form is subject to the terms of the Mozilla Public
  4 * License, v. 2.0. If a copy of the MPL was not distributed with this
  5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
  6#include <mgba/internal/gba/sharkport.h>
  7
  8#include <mgba/internal/arm/macros.h>
  9#include <mgba/internal/gba/gba.h>
 10#include <mgba-util/vfs.h>
 11
 12static const char* const SHARKPORT_HEADER = "SharkPortSave";
 13
 14bool GBASavedataImportSharkPort(struct GBA* gba, struct VFile* vf, bool testChecksum) {
 15	union {
 16		char c[0x1C];
 17		int32_t i;
 18	} buffer;
 19	if (vf->read(vf, &buffer.i, 4) < 4) {
 20		return false;
 21	}
 22	int32_t size;
 23	LOAD_32(size, 0, &buffer.i);
 24	if (size != (int32_t) strlen(SHARKPORT_HEADER)) {
 25		return false;
 26	}
 27	if (vf->read(vf, buffer.c, size) < size) {
 28		return false;
 29	}
 30	if (memcmp(SHARKPORT_HEADER, buffer.c, size) != 0) {
 31		return false;
 32	}
 33	if (vf->read(vf, &buffer.i, 4) < 4) {
 34		return false;
 35	}
 36	LOAD_32(size, 0, &buffer.i);
 37	if (size != 0x000F0000) {
 38		// What is this value?
 39		return false;
 40	}
 41
 42	// Skip first three fields
 43	if (vf->read(vf, &buffer.i, 4) < 4) {
 44		return false;
 45	}
 46	LOAD_32(size, 0, &buffer.i);
 47	if (vf->seek(vf, size, SEEK_CUR) < 0) {
 48		return false;
 49	}
 50
 51	if (vf->read(vf, &buffer.i, 4) < 4) {
 52		return false;
 53	}
 54	LOAD_32(size, 0, &buffer.i);
 55	if (vf->seek(vf, size, SEEK_CUR) < 0) {
 56		return false;
 57	}
 58
 59	if (vf->read(vf, &buffer.i, 4) < 4) {
 60		return false;
 61	}
 62	LOAD_32(size, 0, &buffer.i);
 63	if (vf->seek(vf, size, SEEK_CUR) < 0) {
 64		return false;
 65	}
 66
 67	// Read payload
 68	if (vf->read(vf, &buffer.i, 4) < 4) {
 69		return false;
 70	}
 71	LOAD_32(size, 0, &buffer.i);
 72	if (size < 0x1C || size > SIZE_CART_FLASH1M + 0x1C) {
 73		return false;
 74	}
 75	char* payload = malloc(size);
 76	if (vf->read(vf, payload, size) < size) {
 77		goto cleanup;
 78	}
 79
 80	struct GBACartridge* cart = (struct GBACartridge*) gba->memory.rom;
 81	memcpy(buffer.c, &cart->title, 16);
 82	buffer.c[0x10] = 0;
 83	buffer.c[0x11] = 0;
 84	buffer.c[0x12] = cart->checksum;
 85	buffer.c[0x13] = cart->maker;
 86	buffer.c[0x14] = 1;
 87	buffer.c[0x15] = 0;
 88	buffer.c[0x16] = 0;
 89	buffer.c[0x17] = 0;
 90	buffer.c[0x18] = 0;
 91	buffer.c[0x19] = 0;
 92	buffer.c[0x1A] = 0;
 93	buffer.c[0x1B] = 0;
 94	if (memcmp(buffer.c, payload, 0x1C) != 0) {
 95		goto cleanup;
 96	}
 97
 98	uint32_t checksum;
 99	if (vf->read(vf, &buffer.i, 4) < 4) {
100		goto cleanup;
101	}
102	LOAD_32(checksum, 0, &buffer.i);
103
104	if (testChecksum) {
105		uint32_t calcChecksum = 0;
106		int i;
107		for (i = 0; i < size; ++i) {
108			calcChecksum += ((int32_t) payload[i]) << (calcChecksum % 24);
109		}
110
111		if (calcChecksum != checksum) {
112			goto cleanup;
113		}
114	}
115
116	uint32_t copySize = size - 0x1C;
117	switch (gba->memory.savedata.type) {
118	case SAVEDATA_FLASH512:
119		if (copySize > SIZE_CART_FLASH512) {
120			GBASavedataForceType(&gba->memory.savedata, SAVEDATA_FLASH1M);
121		}
122	// Fall through
123	default:
124		if (copySize > GBASavedataSize(&gba->memory.savedata)) {
125			copySize = GBASavedataSize(&gba->memory.savedata);
126		}
127		break;
128	case SAVEDATA_FORCE_NONE:
129	case SAVEDATA_AUTODETECT:
130		goto cleanup;
131	}
132
133	if (gba->memory.savedata.type == SAVEDATA_EEPROM) {
134		size_t i;
135		for (i = 0; i < copySize; i += 8) {
136			uint32_t lo, hi;
137			LOAD_32BE(lo, i + 0x1C, payload);
138			LOAD_32BE(hi, i + 0x20, payload);
139			STORE_32LE(hi, i, gba->memory.savedata.data);
140			STORE_32LE(lo, i + 4, gba->memory.savedata.data);
141		}
142	} else {
143		memcpy(gba->memory.savedata.data, &payload[0x1C], copySize);
144	}
145	if (gba->memory.savedata.vf) {
146		gba->memory.savedata.vf->sync(gba->memory.savedata.vf, gba->memory.savedata.data, size);
147	}
148
149	free(payload);
150	return true;
151
152cleanup:
153	free(payload);
154	return false;
155}
156
157bool GBASavedataExportSharkPort(const struct GBA* gba, struct VFile* vf) {
158	union {
159		char c[0x1C];
160		int32_t i;
161	} buffer;
162	uint32_t size = strlen(SHARKPORT_HEADER);
163	STORE_32(size, 0, &buffer.i);
164	if (vf->write(vf, &buffer.i, 4) < 4) {
165		return false;
166	}
167	if (vf->write(vf, SHARKPORT_HEADER, size) < size) {
168		return false;
169	}
170
171	size = 0x000F0000;
172	STORE_32(size, 0, &buffer.i);
173	if (vf->write(vf, &buffer.i, 4) < 4) {
174		return false;
175	}
176
177	const struct GBACartridge* cart = (const struct GBACartridge*) gba->memory.rom;
178	size = sizeof(cart->title);
179	STORE_32(size, 0, &buffer.i);
180	if (vf->write(vf, &buffer.i, 4) < 4) {
181		return false;
182	}
183	if (vf->write(vf, cart->title, size) < 4) {
184		return false;
185	}
186
187	time_t t = time(0);
188	struct tm* tm = localtime(&t);
189	size = strftime(&buffer.c[4], sizeof(buffer.c) - 4, "%m/%d/%Y %I:%M:%S %p", tm);
190	STORE_32(size, 0, &buffer.i);
191	if (vf->write(vf, buffer.c, size + 4) < size + 4) {
192		return false;
193	}
194
195	// Last field is blank
196	size = 0;
197	STORE_32(size, 0, &buffer.i);
198	if (vf->write(vf, &buffer.i, 4) < 4) {
199		return false;
200	}
201
202	// Write payload
203	size = 0x1C + GBASavedataSize(&gba->memory.savedata);
204	if (size == 0x1C) {
205		return false;
206	}
207	STORE_32(size, 0, &buffer.i);
208	if (vf->write(vf, &buffer.i, 4) < 4) {
209		return false;
210	}
211	size -= 0x1C;
212
213	memcpy(buffer.c, &cart->title, 16);
214	buffer.c[0x10] = 0;
215	buffer.c[0x11] = 0;
216	buffer.c[0x12] = cart->checksum;
217	buffer.c[0x13] = cart->maker;
218	buffer.c[0x14] = 1;
219	buffer.c[0x15] = 0;
220	buffer.c[0x16] = 0;
221	buffer.c[0x17] = 0;
222	buffer.c[0x18] = 0;
223	buffer.c[0x19] = 0;
224	buffer.c[0x1A] = 0;
225	buffer.c[0x1B] = 0;
226	if (vf->write(vf, buffer.c, 0x1C) < 0x1C) {
227		return false;
228	}
229
230	uint32_t checksum = 0;
231	size_t i;
232	for (i = 0; i < 0x1C; ++i) {
233		checksum += buffer.c[i] << (checksum % 24);
234	}
235
236
237	if (gba->memory.savedata.type == SAVEDATA_EEPROM) {
238		for (i = 0; i < size; ++i) {
239			char byte = gba->memory.savedata.data[i ^ 7];
240			checksum += byte << (checksum % 24);
241			vf->write(vf, &byte, 1);
242		}
243	} else if (vf->write(vf, gba->memory.savedata.data, size) < size) {
244		return false;
245	} else {
246		for (i = 0; i < size; ++i) {
247			checksum += ((char) gba->memory.savedata.data[i]) << (checksum % 24);
248		}
249	}
250
251	STORE_32(checksum, 0, &buffer.i);
252	if (vf->write(vf, &buffer.i, 4) < 4) {
253		return false;
254	}
255
256	return true;
257}