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, gba->memory.savedata.realisticTiming);
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	memcpy(gba->memory.savedata.data, &payload[0x1C], copySize);
134	gba->memory.savedata.vf && gba->memory.savedata.vf->sync(gba->memory.savedata.vf, gba->memory.savedata.data, size);
135
136	free(payload);
137	return true;
138
139cleanup:
140	free(payload);
141	return false;
142}
143
144bool GBASavedataExportSharkPort(const struct GBA* gba, struct VFile* vf) {
145	union {
146		char c[0x1C];
147		int32_t i;
148	} buffer;
149	int32_t size = strlen(SHARKPORT_HEADER);
150	STORE_32(size, 0, &buffer.i);
151	if (vf->write(vf, &buffer.i, 4) < 4) {
152		return false;
153	}
154	if (vf->write(vf, SHARKPORT_HEADER, size) < size) {
155		return false;
156	}
157
158	size = 0x000F0000;
159	STORE_32(size, 0, &buffer.i);
160	if (vf->write(vf, &buffer.i, 4) < 4) {
161		return false;
162	}
163
164	const struct GBACartridge* cart = (const struct GBACartridge*) gba->memory.rom;
165	size = sizeof(cart->title);
166	STORE_32(size, 0, &buffer.i);
167	if (vf->write(vf, &buffer.i, 4) < 4) {
168		return false;
169	}
170	if (vf->write(vf, cart->title, size) < 4) {
171		return false;
172	}
173
174	time_t t = time(0);
175	struct tm* tm = localtime(&t);
176	size = strftime(&buffer.c[4], sizeof(buffer.c) - 4, "%m/%d/%Y %I:%M:%S %p", tm);
177	STORE_32(size, 0, &buffer.i);
178	if (vf->write(vf, buffer.c, size + 4) < size + 4) {
179		return false;
180	}
181
182	// Last field is blank
183	size = 0;
184	STORE_32(size, 0, &buffer.i);
185	if (vf->write(vf, &buffer.i, 4) < 4) {
186		return false;
187	}
188
189	// Write payload
190	size = 0x1C;
191	switch (gba->memory.savedata.type) {
192	case SAVEDATA_SRAM:
193		size += SIZE_CART_SRAM;
194		break;
195	case SAVEDATA_FLASH512:
196		size += SIZE_CART_FLASH512;
197		break;
198	case SAVEDATA_FLASH1M:
199		size += SIZE_CART_FLASH1M;
200		break;
201	case SAVEDATA_EEPROM:
202		size += SIZE_CART_EEPROM;
203		break;
204	case SAVEDATA_FORCE_NONE:
205	case SAVEDATA_AUTODETECT:
206		return false;
207	}
208	STORE_32(size, 0, &buffer.i);
209	if (vf->write(vf, &buffer.i, 4) < 4) {
210		return false;
211	}
212	size -= 0x1C;
213
214	memcpy(buffer.c, &cart->title, 16);
215	buffer.c[0x10] = 0;
216	buffer.c[0x11] = 0;
217	buffer.c[0x12] = cart->checksum;
218	buffer.c[0x13] = cart->maker;
219	buffer.c[0x14] = 1;
220	buffer.c[0x15] = 0;
221	buffer.c[0x16] = 0;
222	buffer.c[0x17] = 0;
223	buffer.c[0x18] = 0;
224	buffer.c[0x19] = 0;
225	buffer.c[0x1A] = 0;
226	buffer.c[0x1B] = 0;
227	if (vf->write(vf, buffer.c, 0x1C) < 0x1C) {
228		return false;
229	}
230
231	uint32_t checksum = 0;
232	int i;
233	for (i = 0; i < 0x1C; ++i) {
234		checksum += buffer.c[i] << (checksum % 24);
235	}
236
237	if (vf->write(vf, gba->memory.savedata.data, size) < size) {
238		return false;
239	}
240
241	for (i = 0; i < size; ++i) {
242		checksum += ((char) gba->memory.savedata.data[i]) << (checksum % 24);
243	}
244
245	STORE_32(checksum, 0, &buffer.i);
246	if (vf->write(vf, &buffer.i, 4) < 4) {
247		return false;
248	}
249
250	return true;
251}