fix array-out-of-bounds issue
[picodrive.git] / pico / state.c
1 /*\r
2  * PicoDrive\r
3  * (C) notaz, 2010\r
4  *\r
5  * This work is licensed under the terms of MAME license.\r
6  * See COPYING file in the top-level directory.\r
7  */\r
8 \r
9 #include "pico_int.h"\r
10 #include <zlib/zlib.h>\r
11 \r
12 #include "../cpu/sh2/sh2.h"\r
13 #include "sound/ym2612.h"\r
14 #include "state.h"\r
15 \r
16 // sn76496\r
17 extern int *sn76496_regs;\r
18 \r
19 static arearw    *areaRead;\r
20 static arearw    *areaWrite;\r
21 static areaeof   *areaEof;\r
22 static areaseek  *areaSeek;\r
23 static areaclose *areaClose;\r
24 \r
25 carthw_state_chunk *carthw_chunks;\r
26 void (*PicoStateProgressCB)(const char *str);\r
27 void (*PicoLoadStateHook)(void);\r
28 \r
29 \r
30 /* I/O functions */\r
31 static size_t gzRead2(void *p, size_t _size, size_t _n, void *file)\r
32 {\r
33   return gzread(file, p, _size * _n);\r
34 }\r
35 \r
36 static size_t gzWrite2(void *p, size_t _size, size_t _n, void *file)\r
37 {\r
38   return gzwrite(file, p, _size * _n);\r
39 }\r
40 \r
41 static void set_cbs(int gz)\r
42 {\r
43   if (gz) {\r
44     areaRead  = gzRead2;\r
45     areaWrite = gzWrite2;\r
46     areaEof   = (areaeof *) gzeof;\r
47     areaSeek  = (areaseek *) gzseek;\r
48     areaClose = (areaclose *) gzclose;\r
49   } else {\r
50     areaRead  = (arearw *) fread;\r
51     areaWrite = (arearw *) fwrite;\r
52     areaEof   = (areaeof *) feof;\r
53     areaSeek  = (areaseek *) fseek;\r
54     areaClose = (areaclose *) fclose;\r
55   }\r
56 }\r
57 \r
58 static void *open_save_file(const char *fname, int is_save)\r
59 {\r
60   int len = strlen(fname);\r
61   void *afile = NULL;\r
62 \r
63   if (len > 3 && strcmp(fname + len - 3, ".gz") == 0)\r
64   {\r
65     if ( (afile = gzopen(fname, is_save ? "wb" : "rb")) ) {\r
66       set_cbs(1);\r
67       if (is_save)\r
68         gzsetparams(afile, 9, Z_DEFAULT_STRATEGY);\r
69     }\r
70   }\r
71   else\r
72   {\r
73     if ( (afile = fopen(fname, is_save ? "wb" : "rb")) ) {\r
74       set_cbs(0);\r
75     }\r
76   }\r
77 \r
78   return afile;\r
79 }\r
80 \r
81 // legacy savestate loading\r
82 #define SCANP(f, x) areaRead(&Pico.x, sizeof(Pico.x), 1, f)\r
83 \r
84 static int state_load_legacy(void *file)\r
85 {\r
86   unsigned char head[32];\r
87   unsigned char cpu[0x60];\r
88   unsigned char cpu_z80[Z80_STATE_SIZE];\r
89   void *ym2612_regs;\r
90   int ok;\r
91 \r
92   memset(&cpu,0,sizeof(cpu));\r
93   memset(&cpu_z80,0,sizeof(cpu_z80));\r
94 \r
95   memset(head, 0, sizeof(head));\r
96   areaRead(head, sizeof(head), 1, file);\r
97   if (strcmp((char *)head, "Pico") != 0)\r
98     return -1;\r
99 \r
100   elprintf(EL_STATUS, "legacy savestate");\r
101 \r
102   // Scan all the memory areas:\r
103   SCANP(file, ram);\r
104   SCANP(file, vram);\r
105   SCANP(file, zram);\r
106   SCANP(file, cram);\r
107   SCANP(file, vsram);\r
108 \r
109   // Pack, scan and unpack the cpu data:\r
110   areaRead(cpu, sizeof(cpu), 1, file);\r
111   SekUnpackCpu(cpu, 0);\r
112 \r
113   SCANP(file, m);\r
114   SCANP(file, video);\r
115 \r
116   ok = areaRead(cpu_z80, sizeof(cpu_z80), 1, file) == sizeof(cpu_z80);\r
117   // do not unpack if we fail to load z80 state\r
118   if (!ok) z80_reset();\r
119   else     z80_unpack(cpu_z80);\r
120 \r
121   ym2612_regs = YM2612GetRegs();\r
122   areaRead(sn76496_regs, 28*4, 1, file);\r
123   areaRead(ym2612_regs, 0x200+4, 1, file);\r
124   ym2612_unpack_state();\r
125 \r
126   return 0;\r
127 }\r
128 \r
129 // ---------------------------------------------------------------------------\r
130 \r
131 typedef enum {\r
132   CHUNK_M68K = 1,\r
133   CHUNK_RAM,\r
134   CHUNK_VRAM,\r
135   CHUNK_ZRAM,\r
136   CHUNK_CRAM,    // 5\r
137   CHUNK_VSRAM,\r
138   CHUNK_MISC,\r
139   CHUNK_VIDEO,\r
140   CHUNK_Z80,\r
141   CHUNK_PSG,     // 10\r
142   CHUNK_FM,\r
143   // CD stuff\r
144   CHUNK_S68K,\r
145   CHUNK_PRG_RAM,\r
146   CHUNK_WORD_RAM,\r
147   CHUNK_PCM_RAM, // 15\r
148   CHUNK_BRAM,\r
149   CHUNK_GA_REGS,\r
150   CHUNK_PCM,\r
151   CHUNK_CDC,\r
152   CHUNK_CDD,     // 20\r
153   CHUNK_SCD,\r
154   CHUNK_RC,\r
155   CHUNK_MISC_CD,\r
156   //\r
157   CHUNK_IOPORTS, // versions < 1.70 did not save that..\r
158   CHUNK_SMS,     // 25\r
159   // 32x\r
160   CHUNK_MSH2,\r
161   CHUNK_MSH2_DATA,\r
162   CHUNK_MSH2_PERI,\r
163   CHUNK_SSH2,\r
164   CHUNK_SSH2_DATA, // 30\r
165   CHUNK_SSH2_PERI,\r
166   CHUNK_32XSYS,\r
167   CHUNK_M68K_BIOS,\r
168   CHUNK_MSH2_BIOS,\r
169   CHUNK_SSH2_BIOS, // 35\r
170   CHUNK_SDRAM,\r
171   CHUNK_DRAM,\r
172   CHUNK_32XPAL,\r
173   CHUNK_32X_EVT,\r
174   CHUNK_32X_FIRST = CHUNK_MSH2,\r
175   CHUNK_32X_LAST = CHUNK_32X_EVT,\r
176   // add new stuff here\r
177   CHUNK_CD_EVT = 50,\r
178   //\r
179   CHUNK_DEFAULT_COUNT,\r
180   CHUNK_CARTHW_ = CHUNK_CARTHW,  // 64 (defined in PicoInt)\r
181 } chunk_name_e;\r
182 \r
183 static const char * const chunk_names[CHUNK_DEFAULT_COUNT] = {\r
184   "INVALID!",\r
185   "M68K state",\r
186   "RAM",\r
187   "VRAM",\r
188   "ZRAM",\r
189   "CRAM",      // 5\r
190   "VSRAM",\r
191   "emu state",\r
192   "VIDEO",\r
193   "Z80 state",\r
194   "PSG",       // 10\r
195   "FM",\r
196   // CD stuff\r
197   "S68K state",\r
198   "PRG_RAM",\r
199   "WORD_RAM",\r
200   "PCM_RAM",   // 15\r
201   "BRAM",\r
202   "GATE ARRAY regs",\r
203   "PCM state",\r
204   "CDC",\r
205   "CDD",       // 20\r
206   "SCD",\r
207   "GFX chip",\r
208   "MCD state",\r
209   //\r
210   "IO",\r
211   "SMS state", // 25\r
212   // 32x\r
213   "MSH2",\r
214   "MSH2 data",\r
215   "MSH2 peri",\r
216   "SSH2",\r
217   "SSH2 data", // 30\r
218   "SSH2 peri",\r
219   "32X system regs",\r
220   "M68K BIOS",\r
221   "MSH2 BIOS",\r
222   "SSH2 BIOS", // 35\r
223   "SDRAM",\r
224   "DRAM",\r
225   "PAL",\r
226   "events",\r
227 };\r
228 \r
229 static int write_chunk(chunk_name_e name, int len, void *data, void *file)\r
230 {\r
231   size_t bwritten = 0;\r
232   bwritten += areaWrite(&name, 1, 1, file);\r
233   bwritten += areaWrite(&len, 1, 4, file);\r
234   bwritten += areaWrite(data, 1, len, file);\r
235 \r
236   return (bwritten == len + 4 + 1);\r
237 }\r
238 \r
239 #define CHECKED_WRITE(name,len,data) { \\r
240   if (PicoStateProgressCB && name < CHUNK_DEFAULT_COUNT && chunk_names[name]) { \\r
241     strncpy(sbuff + 9, chunk_names[name], sizeof(sbuff) - 9); \\r
242     PicoStateProgressCB(sbuff); \\r
243   } \\r
244   if (!write_chunk(name, len, data, file)) return 1; \\r
245 }\r
246 \r
247 #define CHECKED_WRITE_BUFF(name,buff) { \\r
248   if (PicoStateProgressCB && name < CHUNK_DEFAULT_COUNT && chunk_names[name]) { \\r
249     strncpy(sbuff + 9, chunk_names[name], sizeof(sbuff) - 9); \\r
250     PicoStateProgressCB(sbuff); \\r
251   } \\r
252   if (!write_chunk(name, sizeof(buff), &buff, file)) return 1; \\r
253 }\r
254 \r
255 static int state_save(void *file)\r
256 {\r
257   char sbuff[32] = "Saving.. ";\r
258   unsigned char buff[0x60], buff_z80[Z80_STATE_SIZE];\r
259   void *ym2612_regs = YM2612GetRegs();\r
260   int ver = 0x0170; // not really used..\r
261 \r
262   areaWrite("PicoSEXT", 1, 8, file);\r
263   areaWrite(&ver, 1, 4, file);\r
264 \r
265   if (!(PicoAHW & PAHW_SMS)) {\r
266     memset(buff, 0, sizeof(buff));\r
267     SekPackCpu(buff, 0);\r
268     CHECKED_WRITE_BUFF(CHUNK_M68K,  buff);\r
269     CHECKED_WRITE_BUFF(CHUNK_RAM,   Pico.ram);\r
270     CHECKED_WRITE_BUFF(CHUNK_VSRAM, Pico.vsram);\r
271     CHECKED_WRITE_BUFF(CHUNK_IOPORTS, Pico.ioports);\r
272     ym2612_pack_state();\r
273     CHECKED_WRITE(CHUNK_FM, 0x200+4, ym2612_regs);\r
274   }\r
275   else {\r
276     CHECKED_WRITE_BUFF(CHUNK_SMS, Pico.ms);\r
277   }\r
278 \r
279   CHECKED_WRITE_BUFF(CHUNK_VRAM,  Pico.vram);\r
280   CHECKED_WRITE_BUFF(CHUNK_ZRAM,  Pico.zram);\r
281   CHECKED_WRITE_BUFF(CHUNK_CRAM,  Pico.cram);\r
282   CHECKED_WRITE_BUFF(CHUNK_MISC,  Pico.m);\r
283   CHECKED_WRITE_BUFF(CHUNK_VIDEO, Pico.video);\r
284 \r
285   z80_pack(buff_z80);\r
286   CHECKED_WRITE_BUFF(CHUNK_Z80, buff_z80);\r
287   CHECKED_WRITE(CHUNK_PSG, 28*4, sn76496_regs);\r
288 \r
289   if (PicoAHW & PAHW_MCD)\r
290   {\r
291     memset(buff, 0, sizeof(buff));\r
292     SekPackCpu(buff, 1);\r
293     if (Pico_mcd->s68k_regs[3] & 4) // 1M mode?\r
294       wram_1M_to_2M(Pico_mcd->word_ram2M);\r
295     memcpy(&Pico_mcd->m.hint_vector, Pico_mcd->bios + 0x72,\r
296       sizeof(Pico_mcd->m.hint_vector));\r
297 \r
298     CHECKED_WRITE_BUFF(CHUNK_S68K,     buff);\r
299     CHECKED_WRITE_BUFF(CHUNK_PRG_RAM,  Pico_mcd->prg_ram);\r
300     CHECKED_WRITE_BUFF(CHUNK_WORD_RAM, Pico_mcd->word_ram2M); // in 2M format\r
301     CHECKED_WRITE_BUFF(CHUNK_PCM_RAM,  Pico_mcd->pcm_ram);\r
302     CHECKED_WRITE_BUFF(CHUNK_BRAM,     Pico_mcd->bram);\r
303     CHECKED_WRITE_BUFF(CHUNK_GA_REGS,  Pico_mcd->s68k_regs); // GA regs, not CPU regs\r
304     CHECKED_WRITE_BUFF(CHUNK_PCM,      Pico_mcd->pcm);\r
305     CHECKED_WRITE_BUFF(CHUNK_CDD,      Pico_mcd->cdd);\r
306     CHECKED_WRITE_BUFF(CHUNK_CDC,      Pico_mcd->cdc);\r
307     CHECKED_WRITE_BUFF(CHUNK_SCD,      Pico_mcd->scd);\r
308     CHECKED_WRITE_BUFF(CHUNK_RC,       Pico_mcd->rot_comp);\r
309     CHECKED_WRITE_BUFF(CHUNK_MISC_CD,  Pico_mcd->m);\r
310     memset(buff, 0, 0x40);\r
311     memcpy(buff, pcd_event_times, sizeof(pcd_event_times));\r
312     CHECKED_WRITE(CHUNK_CD_EVT, 0x40, buff);\r
313 \r
314     if (Pico_mcd->s68k_regs[3] & 4) // convert back\r
315       wram_2M_to_1M(Pico_mcd->word_ram2M);\r
316   }\r
317 \r
318 #ifndef NO_32X\r
319   if (PicoAHW & PAHW_32X)\r
320   {\r
321     unsigned char cpubuff[SH2_STATE_SIZE];\r
322 \r
323     memset(cpubuff, 0, sizeof(cpubuff));\r
324 \r
325     sh2_pack(&sh2s[0], cpubuff);\r
326     CHECKED_WRITE_BUFF(CHUNK_MSH2,      cpubuff);\r
327     CHECKED_WRITE_BUFF(CHUNK_MSH2_DATA, sh2s[0].data_array);\r
328     CHECKED_WRITE_BUFF(CHUNK_MSH2_PERI, sh2s[0].peri_regs);\r
329 \r
330     sh2_pack(&sh2s[1], cpubuff);\r
331     CHECKED_WRITE_BUFF(CHUNK_SSH2,      cpubuff);\r
332     CHECKED_WRITE_BUFF(CHUNK_SSH2_DATA, sh2s[1].data_array);\r
333     CHECKED_WRITE_BUFF(CHUNK_SSH2_PERI, sh2s[1].peri_regs);\r
334 \r
335     CHECKED_WRITE_BUFF(CHUNK_32XSYS,    Pico32x);\r
336     CHECKED_WRITE_BUFF(CHUNK_M68K_BIOS, Pico32xMem->m68k_rom);\r
337     CHECKED_WRITE_BUFF(CHUNK_MSH2_BIOS, Pico32xMem->sh2_rom_m);\r
338     CHECKED_WRITE_BUFF(CHUNK_SSH2_BIOS, Pico32xMem->sh2_rom_s);\r
339     CHECKED_WRITE_BUFF(CHUNK_SDRAM,     Pico32xMem->sdram);\r
340     CHECKED_WRITE_BUFF(CHUNK_DRAM,      Pico32xMem->dram);\r
341     CHECKED_WRITE_BUFF(CHUNK_32XPAL,    Pico32xMem->pal);\r
342 \r
343     memset(buff, 0, 0x40);\r
344     memcpy(buff, p32x_event_times, sizeof(p32x_event_times));\r
345     CHECKED_WRITE(CHUNK_32X_EVT, 0x40, buff);\r
346   }\r
347 #endif\r
348 \r
349   if (carthw_chunks != NULL)\r
350   {\r
351     carthw_state_chunk *chwc;\r
352     if (PicoStateProgressCB)\r
353       PicoStateProgressCB("Saving.. cart hw state");\r
354     for (chwc = carthw_chunks; chwc->ptr != NULL; chwc++)\r
355       CHECKED_WRITE(chwc->chunk, chwc->size, chwc->ptr);\r
356   }\r
357 \r
358   return 0;\r
359 }\r
360 \r
361 static int g_read_offs = 0;\r
362 \r
363 #define R_ERROR_RETURN(error) \\r
364 { \\r
365   elprintf(EL_STATUS, "load_state @ %x: " error, g_read_offs); \\r
366   return 1; \\r
367 }\r
368 \r
369 // when is eof really set?\r
370 #define CHECKED_READ(len,data) { \\r
371   if (areaRead(data, 1, len, file) != len) { \\r
372     if (len == 1 && areaEof(file)) goto readend; \\r
373     R_ERROR_RETURN("areaRead: premature EOF\n"); \\r
374     return 1; \\r
375   } \\r
376   g_read_offs += len; \\r
377 }\r
378 \r
379 #define CHECKED_READ2(len2,data) { \\r
380   if (len2 != len) { \\r
381     elprintf(EL_STATUS, "unexpected len %i, wanted %i (%s)", len, len2, #len2); \\r
382     if (len > len2) R_ERROR_RETURN("failed."); \\r
383     /* else read anyway and hope for the best.. */ \\r
384   } \\r
385   CHECKED_READ(len, data); \\r
386 }\r
387 \r
388 #define CHECKED_READ_BUFF(buff) CHECKED_READ2(sizeof(buff), &buff);\r
389 \r
390 static int state_load(void *file)\r
391 {\r
392   unsigned char buff_m68k[0x60], buff_s68k[0x60];\r
393   unsigned char buff_z80[Z80_STATE_SIZE];\r
394   unsigned char buff_sh2[SH2_STATE_SIZE];\r
395   unsigned char buff[0x40];\r
396   unsigned char chunk;\r
397   void *ym2612_regs;\r
398   char header[8];\r
399   int ver, len;\r
400 \r
401   memset(buff_m68k, 0, sizeof(buff_m68k));\r
402   memset(buff_s68k, 0, sizeof(buff_s68k));\r
403   memset(buff_z80, 0, sizeof(buff_z80));\r
404 \r
405   g_read_offs = 0;\r
406   CHECKED_READ(8, header);\r
407   if (strncmp(header, "PicoSMCD", 8) && strncmp(header, "PicoSEXT", 8))\r
408     R_ERROR_RETURN("bad header");\r
409   CHECKED_READ(4, &ver);\r
410 \r
411   memset(pcd_event_times, 0, sizeof(pcd_event_times));\r
412   memset(p32x_event_times, 0, sizeof(p32x_event_times));\r
413 \r
414   while (!areaEof(file))\r
415   {\r
416     CHECKED_READ(1, &chunk);\r
417     CHECKED_READ(4, &len);\r
418     if (len < 0 || len > 1024*512) R_ERROR_RETURN("bad length");\r
419     if (CHUNK_S68K <= chunk && chunk <= CHUNK_MISC_CD && !(PicoAHW & PAHW_MCD))\r
420       R_ERROR_RETURN("cd chunk in non CD state?");\r
421     if (CHUNK_32X_FIRST <= chunk && chunk <= CHUNK_32X_LAST && !(PicoAHW & PAHW_32X))\r
422       Pico32xStartup();\r
423 \r
424     switch (chunk)\r
425     {\r
426       case CHUNK_M68K:\r
427         CHECKED_READ_BUFF(buff_m68k);\r
428         break;\r
429 \r
430       case CHUNK_Z80:\r
431         CHECKED_READ_BUFF(buff_z80);\r
432         break;\r
433 \r
434       case CHUNK_RAM:     CHECKED_READ_BUFF(Pico.ram); break;\r
435       case CHUNK_VRAM:    CHECKED_READ_BUFF(Pico.vram); break;\r
436       case CHUNK_ZRAM:    CHECKED_READ_BUFF(Pico.zram); break;\r
437       case CHUNK_CRAM:    CHECKED_READ_BUFF(Pico.cram); break;\r
438       case CHUNK_VSRAM:   CHECKED_READ_BUFF(Pico.vsram); break;\r
439       case CHUNK_MISC:    CHECKED_READ_BUFF(Pico.m); break;\r
440       case CHUNK_VIDEO:   CHECKED_READ_BUFF(Pico.video); break;\r
441       case CHUNK_IOPORTS: CHECKED_READ_BUFF(Pico.ioports); break;\r
442       case CHUNK_PSG:     CHECKED_READ2(28*4, sn76496_regs); break;\r
443       case CHUNK_FM:\r
444         ym2612_regs = YM2612GetRegs();\r
445         CHECKED_READ2(0x200+4, ym2612_regs);\r
446         ym2612_unpack_state();\r
447         break;\r
448 \r
449       case CHUNK_SMS:\r
450         CHECKED_READ_BUFF(Pico.ms);\r
451         break;\r
452 \r
453       // cd stuff\r
454       case CHUNK_S68K:\r
455         CHECKED_READ_BUFF(buff_s68k);\r
456         break;\r
457 \r
458       case CHUNK_PRG_RAM:  CHECKED_READ_BUFF(Pico_mcd->prg_ram); break;\r
459       case CHUNK_WORD_RAM: CHECKED_READ_BUFF(Pico_mcd->word_ram2M); break;\r
460       case CHUNK_PCM_RAM:  CHECKED_READ_BUFF(Pico_mcd->pcm_ram); break;\r
461       case CHUNK_BRAM:     CHECKED_READ_BUFF(Pico_mcd->bram); break;\r
462       case CHUNK_GA_REGS:  CHECKED_READ_BUFF(Pico_mcd->s68k_regs); break;\r
463       case CHUNK_PCM:      CHECKED_READ_BUFF(Pico_mcd->pcm); break;\r
464       case CHUNK_CDD:      CHECKED_READ_BUFF(Pico_mcd->cdd); break;\r
465       case CHUNK_CDC:      CHECKED_READ_BUFF(Pico_mcd->cdc); break;\r
466       case CHUNK_SCD:      CHECKED_READ_BUFF(Pico_mcd->scd); break;\r
467       case CHUNK_RC:       CHECKED_READ_BUFF(Pico_mcd->rot_comp); break;\r
468       case CHUNK_MISC_CD:  CHECKED_READ_BUFF(Pico_mcd->m); break;\r
469 \r
470       case CHUNK_CD_EVT:\r
471         CHECKED_READ_BUFF(buff);\r
472         memcpy(pcd_event_times, buff, sizeof(pcd_event_times));\r
473         break;\r
474 \r
475       // 32x stuff\r
476 #ifndef NO_32X\r
477       case CHUNK_MSH2:\r
478         CHECKED_READ_BUFF(buff_sh2);\r
479         sh2_unpack(&sh2s[0], buff_sh2);\r
480         break;\r
481 \r
482       case CHUNK_SSH2:\r
483         CHECKED_READ_BUFF(buff_sh2);\r
484         sh2_unpack(&sh2s[1], buff_sh2);\r
485         break;\r
486 \r
487       case CHUNK_MSH2_DATA:   CHECKED_READ_BUFF(sh2s[0].data_array); break;\r
488       case CHUNK_MSH2_PERI:   CHECKED_READ_BUFF(sh2s[0].peri_regs); break;\r
489       case CHUNK_SSH2_DATA:   CHECKED_READ_BUFF(sh2s[1].data_array); break;\r
490       case CHUNK_SSH2_PERI:   CHECKED_READ_BUFF(sh2s[1].peri_regs); break;\r
491       case CHUNK_32XSYS:      CHECKED_READ_BUFF(Pico32x); break;\r
492       case CHUNK_M68K_BIOS:   CHECKED_READ_BUFF(Pico32xMem->m68k_rom); break;\r
493       case CHUNK_MSH2_BIOS:   CHECKED_READ_BUFF(Pico32xMem->sh2_rom_m); break;\r
494       case CHUNK_SSH2_BIOS:   CHECKED_READ_BUFF(Pico32xMem->sh2_rom_s); break;\r
495       case CHUNK_SDRAM:       CHECKED_READ_BUFF(Pico32xMem->sdram); break;\r
496       case CHUNK_DRAM:        CHECKED_READ_BUFF(Pico32xMem->dram); break;\r
497       case CHUNK_32XPAL:      CHECKED_READ_BUFF(Pico32xMem->pal); break;\r
498 \r
499       case CHUNK_32X_EVT:\r
500         CHECKED_READ_BUFF(buff);\r
501         memcpy(p32x_event_times, buff, sizeof(p32x_event_times));\r
502         break;\r
503 #endif\r
504       default:\r
505         if (carthw_chunks != NULL)\r
506         {\r
507           carthw_state_chunk *chwc;\r
508           for (chwc = carthw_chunks; chwc->ptr != NULL; chwc++) {\r
509             if (chwc->chunk == chunk) {\r
510               CHECKED_READ2(chwc->size, chwc->ptr);\r
511               goto breakswitch;\r
512             }\r
513           }\r
514         }\r
515         elprintf(EL_STATUS, "load_state: skipping unknown chunk %i of size %i", chunk, len);\r
516         areaSeek(file, len, SEEK_CUR);\r
517         break;\r
518     }\r
519 breakswitch:;\r
520   }\r
521 \r
522 readend:\r
523   if (PicoAHW & PAHW_SMS)\r
524     PicoStateLoadedMS();\r
525 \r
526   if (PicoAHW & PAHW_32X)\r
527     Pico32xStateLoaded(1);\r
528 \r
529   // must unpack 68k and z80 after banks are set up\r
530   if (!(PicoAHW & PAHW_SMS))\r
531     SekUnpackCpu(buff_m68k, 0);\r
532   if (PicoAHW & PAHW_MCD)\r
533     SekUnpackCpu(buff_s68k, 1);\r
534 \r
535   z80_unpack(buff_z80);\r
536 \r
537   // due to dep from 68k cycles..\r
538   SekCycleAim = SekCycleCnt;\r
539   if (PicoAHW & PAHW_32X)\r
540     Pico32xStateLoaded(0);\r
541   if (PicoAHW & PAHW_MCD)\r
542   {\r
543     SekCycleAimS68k = SekCycleCntS68k;\r
544     pcd_state_loaded();\r
545 \r
546     if (!(Pico_mcd->s68k_regs[0x36] & 1) && (Pico_mcd->scd.Status_CDC & 1))\r
547       cdda_start_play();\r
548   }\r
549 \r
550   return 0;\r
551 }\r
552 \r
553 static int state_load_gfx(void *file)\r
554 {\r
555   int ver, len, found = 0, to_find = 4;\r
556   char buff[8];\r
557 \r
558   if (PicoAHW & PAHW_32X)\r
559     to_find += 2;\r
560 \r
561   g_read_offs = 0;\r
562   CHECKED_READ(8, buff);\r
563   if (strncmp((char *)buff, "PicoSMCD", 8) && strncmp((char *)buff, "PicoSEXT", 8))\r
564     R_ERROR_RETURN("bad header");\r
565   CHECKED_READ(4, &ver);\r
566 \r
567   while (!areaEof(file) && found < to_find)\r
568   {\r
569     CHECKED_READ(1, buff);\r
570     CHECKED_READ(4, &len);\r
571     if (len < 0 || len > 1024*512) R_ERROR_RETURN("bad length");\r
572     if (buff[0] > CHUNK_FM && buff[0] <= CHUNK_MISC_CD && !(PicoAHW & PAHW_MCD))\r
573       R_ERROR_RETURN("cd chunk in non CD state?");\r
574 \r
575     switch (buff[0])\r
576     {\r
577       case CHUNK_VRAM:  CHECKED_READ_BUFF(Pico.vram);  found++; break;\r
578       case CHUNK_CRAM:  CHECKED_READ_BUFF(Pico.cram);  found++; break;\r
579       case CHUNK_VSRAM: CHECKED_READ_BUFF(Pico.vsram); found++; break;\r
580       case CHUNK_VIDEO: CHECKED_READ_BUFF(Pico.video); found++; break;\r
581 \r
582 #ifndef NO_32X\r
583       case CHUNK_DRAM:\r
584         if (Pico32xMem != NULL)\r
585           CHECKED_READ_BUFF(Pico32xMem->dram);\r
586         break;\r
587 \r
588       case CHUNK_32XPAL:\r
589         if (Pico32xMem != NULL)\r
590           CHECKED_READ_BUFF(Pico32xMem->pal);\r
591         Pico32x.dirty_pal = 1;\r
592         break;\r
593 \r
594       case CHUNK_32XSYS:\r
595         CHECKED_READ_BUFF(Pico32x);\r
596         break;\r
597 #endif\r
598       default:\r
599         areaSeek(file, len, SEEK_CUR);\r
600         break;\r
601     }\r
602   }\r
603 \r
604 readend:\r
605   return 0;\r
606 }\r
607 \r
608 static int pico_state_internal(void *afile, int is_save)\r
609 {\r
610   int ret;\r
611 \r
612   if (is_save)\r
613     ret = state_save(afile);\r
614   else {\r
615     ret = state_load(afile);\r
616     if (ret != 0) {\r
617       areaSeek(afile, 0, SEEK_SET);\r
618       ret = state_load_legacy(afile);\r
619     }\r
620 \r
621     if (PicoLoadStateHook != NULL)\r
622       PicoLoadStateHook();\r
623     Pico.m.dirtyPal = 1;\r
624   }\r
625 \r
626   return ret;\r
627 }\r
628 \r
629 int PicoState(const char *fname, int is_save)\r
630 {\r
631   void *afile = NULL;\r
632   int ret;\r
633 \r
634   afile = open_save_file(fname, is_save);\r
635   if (afile == NULL)\r
636     return -1;\r
637 \r
638   ret = pico_state_internal(afile, is_save);\r
639   areaClose(afile);\r
640   return ret;\r
641 }\r
642 \r
643 int PicoStateFP(void *afile, int is_save,\r
644   arearw *read, arearw *write, areaeof *eof, areaseek *seek)\r
645 {\r
646   areaRead  = read;\r
647   areaWrite = write;\r
648   areaEof   = eof;\r
649   areaSeek  = seek;\r
650   areaClose = NULL;\r
651 \r
652   return pico_state_internal(afile, is_save);\r
653 }\r
654 \r
655 int PicoStateLoadGfx(const char *fname)\r
656 {\r
657   void *afile;\r
658   int ret;\r
659 \r
660   afile = open_save_file(fname, 0);\r
661   if (afile == NULL)\r
662     return -1;\r
663 \r
664   ret = state_load_gfx(afile);\r
665   if (ret != 0) {\r
666     // assume legacy\r
667     areaSeek(afile, 0x10020, SEEK_SET);  // skip header and RAM\r
668     areaRead(Pico.vram, 1, sizeof(Pico.vram), afile);\r
669     areaSeek(afile, 0x2000, SEEK_CUR);\r
670     areaRead(Pico.cram, 1, sizeof(Pico.cram), afile);\r
671     areaRead(Pico.vsram, 1, sizeof(Pico.vsram), afile);\r
672     areaSeek(afile, 0x221a0, SEEK_SET);\r
673     areaRead(&Pico.video, 1, sizeof(Pico.video), afile);\r
674   }\r
675   areaClose(afile);\r
676   return 0;\r
677 }\r
678 \r
679 // tmp state\r
680 struct PicoTmp\r
681 {\r
682   unsigned short vram[0x8000];\r
683   unsigned short cram[0x40];\r
684   unsigned short vsram[0x40];\r
685 \r
686   //struct PicoMisc m;\r
687   struct PicoVideo video;\r
688 \r
689   struct {\r
690     struct Pico32x p32x;\r
691     unsigned short dram[2][0x20000/2];\r
692     unsigned short pal[0x100];\r
693   } t32x;\r
694 };\r
695 \r
696 // returns data ptr to free() or PicoTmpStateRestore()\r
697 void *PicoTmpStateSave(void)\r
698 {\r
699   // gfx only for now\r
700   struct PicoTmp *t = malloc(sizeof(*t));\r
701   if (t == NULL)\r
702     return NULL;\r
703 \r
704   memcpy(t->vram, Pico.vram, sizeof(Pico.vram));\r
705   memcpy(t->cram, Pico.cram, sizeof(Pico.cram));\r
706   memcpy(t->vsram, Pico.vsram, sizeof(Pico.vsram));\r
707   memcpy(&t->video, &Pico.video, sizeof(Pico.video));\r
708 \r
709 #ifndef NO_32X\r
710   if (PicoAHW & PAHW_32X) {\r
711     memcpy(&t->t32x.p32x, &Pico32x, sizeof(Pico32x));\r
712     memcpy(t->t32x.dram, Pico32xMem->dram, sizeof(Pico32xMem->dram));\r
713     memcpy(t->t32x.pal, Pico32xMem->pal, sizeof(Pico32xMem->pal));\r
714   }\r
715 #endif\r
716 \r
717   return t;\r
718 }\r
719 \r
720 void PicoTmpStateRestore(void *data)\r
721 {\r
722   struct PicoTmp *t = data;\r
723   if (t == NULL)\r
724     return;\r
725 \r
726   memcpy(Pico.vram, t->vram, sizeof(Pico.vram));\r
727   memcpy(Pico.cram, t->cram, sizeof(Pico.cram));\r
728   memcpy(Pico.vsram, t->vsram, sizeof(Pico.vsram));\r
729   memcpy(&Pico.video, &t->video, sizeof(Pico.video));\r
730   Pico.m.dirtyPal = 1;\r
731 \r
732 #ifndef NO_32X\r
733   if (PicoAHW & PAHW_32X) {\r
734     memcpy(&Pico32x, &t->t32x.p32x, sizeof(Pico32x));\r
735     memcpy(Pico32xMem->dram, t->t32x.dram, sizeof(Pico32xMem->dram));\r
736     memcpy(Pico32xMem->pal, t->t32x.pal, sizeof(Pico32xMem->pal));\r
737     Pico32x.dirty_pal = 1;\r
738   }\r
739 #endif\r
740 }\r
741 \r
742 // vim:shiftwidth=2:ts=2:expandtab\r