]> git.sesse.net Git - pistorm/blob - platforms/amiga/piscsi/piscsi.c
tidy up headers, remove extraneous duplicate decls
[pistorm] / platforms / amiga / piscsi / piscsi.c
1 #include <stdio.h>
2 #include <stdint.h>
3 #include <stdlib.h>
4 #include <string.h>
5 #include <fcntl.h>
6 #include <unistd.h>
7 #include <endian.h>
8
9 #include "config_file/config_file.h"
10 #include "gpio/ps_protocol.h"
11 #include "piscsi-enums.h"
12 #include "piscsi.h"
13 #include "platforms/amiga/hunk-reloc.h"
14
15 #define BE(val) be32toh(val)
16 #define BE16(val) be16toh(val)
17
18 // Uncomment the line below to enable debug output
19 //#define PISCSI_DEBUG
20
21 #ifdef PISCSI_DEBUG
22 #define DEBUG printf
23 //#define DEBUG_TRIVIAL printf
24 #define DEBUG_TRVIAL(...)
25
26 extern void stop_cpu_emulation(uint8_t disasm_cur);
27
28 static const char *op_type_names[4] = {
29     "BYTE",
30     "WORD",
31     "LONGWORD",
32     "MEM",
33 };
34 #else
35 #define DEBUG(...)
36 #define DEBUG_TRIVIAL(...)
37 #define stop_cpu_emulation(...)
38 #endif
39
40 #ifdef FAKESTORM
41 #define lseek64 lseek
42 #endif
43
44 extern struct emulator_config *cfg;
45
46 struct piscsi_dev devs[8];
47 struct piscsi_fs filesystems[NUM_FILESYSTEMS];
48
49 uint8_t piscsi_num_fs = 0;
50
51 uint8_t piscsi_cur_drive = 0;
52 uint32_t piscsi_u32[4];
53 uint32_t piscsi_dbg[8];
54 uint32_t piscsi_rom_size = 0;
55 uint8_t *piscsi_rom_ptr;
56
57 uint32_t rom_partitions[128];
58 uint32_t rom_partition_prio[128];
59 uint32_t rom_partition_dostype[128];
60 uint32_t rom_cur_partition = 0, rom_cur_fs = 0;
61
62 extern unsigned char ac_piscsi_rom[];
63
64 //static const char *partition_marker = "PART";
65
66 struct hunk_info piscsi_hinfo;
67 struct hunk_reloc piscsi_hreloc[256];
68
69 void piscsi_init() {
70     for (int i = 0; i < 8; i++) {
71         devs[i].fd = -1;
72         devs[i].lba = 0;
73         devs[i].c = devs[i].h = devs[i].s = 0;
74     }
75
76     FILE *in = fopen("./platforms/amiga/piscsi/piscsi.rom", "rb");
77     if (in == NULL) {
78         printf("[PISCSI] Could not open PISCSI Boot ROM file for reading!\n");
79         // Zero out the boot ROM offset from the autoconfig ROM.
80         ac_piscsi_rom[20] = 0;
81         ac_piscsi_rom[21] = 0;
82         ac_piscsi_rom[22] = 0;
83         ac_piscsi_rom[23] = 0;
84         return;
85     }
86     fseek(in, 0, SEEK_END);
87     piscsi_rom_size = ftell(in);
88     fseek(in, 0, SEEK_SET);
89     piscsi_rom_ptr = malloc(piscsi_rom_size);
90     fread(piscsi_rom_ptr, piscsi_rom_size, 1, in);
91
92     fseek(in, PISCSI_DRIVER_OFFSET, SEEK_SET);
93     process_hunks(in, &piscsi_hinfo, piscsi_hreloc, PISCSI_DRIVER_OFFSET);
94
95     fclose(in);
96     printf("[PISCSI] Loaded Boot ROM.\n");
97 }
98
99 void piscsi_find_partitions(struct piscsi_dev *d) {
100     int fd = d->fd;
101     int cur_partition = 0;
102     uint8_t tmp;
103
104     for (int i = 0; i < 16; i++) {
105         if (d->pb[i]) {
106             free(d->pb[i]);
107             d->pb[i] = NULL;
108         }
109     }
110
111     if (!d->rdb || d->rdb->rdb_PartitionList == 0) {
112         DEBUG("[PISCSI] No partitions on disk.\n");
113         return;
114     }
115
116     char *block = malloc(512);
117
118     lseek(fd, BE(d->rdb->rdb_PartitionList) * 512, SEEK_SET);
119 next_partition:;
120     read(fd, block, 512);
121
122     uint32_t first = be32toh(*((uint32_t *)&block[0]));
123     if (first != PART_IDENTIFIER) {
124         DEBUG("Entry at block %d is not a valid partition. Aborting.\n", BE(d->rdb->rdb_PartitionList));
125         return;
126     }
127
128     struct PartitionBlock *pb = (struct PartitionBlock *)block;
129     tmp = pb->pb_DriveName[0];
130     pb->pb_DriveName[tmp + 1] = 0x00;
131     DEBUG("[PISCSI] Partition %d: %s\n", cur_partition, pb->pb_DriveName + 1);
132     DEBUG("Checksum: %.8X HostID: %d\n", BE(pb->pb_ChkSum), BE(pb->pb_HostID));
133     DEBUG("Flags: %d (%.8X) Devflags: %d (%.8X)\n", BE(pb->pb_Flags), BE(pb->pb_Flags), BE(pb->pb_DevFlags), BE(pb->pb_DevFlags));
134     d->pb[cur_partition] = pb;
135
136     if (d->pb[cur_partition]->pb_Next != 0xFFFFFFFF) {
137         uint64_t next = be32toh(pb->pb_Next);
138         block = malloc(512);
139         lseek64(fd, next * 512, SEEK_SET);
140         cur_partition++;
141         DEBUG("[PISCSI] Next partition at block %d.\n", be32toh(pb->pb_Next));
142         goto next_partition;
143     }
144     DEBUG("[PISCSI] No more partitions on disk.\n");
145     d->num_partitions = cur_partition + 1;
146     d->fshd_offs = lseek64(fd, 0, SEEK_CUR);
147
148     return;
149 }
150
151 int piscsi_parse_rdb(struct piscsi_dev *d) {
152     int fd = d->fd;
153     int i = 0;
154     uint8_t *block = malloc(512);
155
156     lseek(fd, 0, SEEK_SET);
157     for (i = 0; i < RDB_BLOCK_LIMIT; i++) {
158         read(fd, block, 512);
159         uint32_t first = be32toh(*((uint32_t *)&block[0]));
160         if (first == RDB_IDENTIFIER)
161             goto rdb_found;
162     }
163     goto no_rdb_found;
164 rdb_found:;
165     struct RigidDiskBlock *rdb = (struct RigidDiskBlock *)block;
166     DEBUG("[PISCSI] RDB found at block %d.\n", i);
167     d->c = be32toh(rdb->rdb_Cylinders);
168     d->h = be32toh(rdb->rdb_Heads);
169     d->s = be32toh(rdb->rdb_Sectors);
170     d->num_partitions = 0;
171     DEBUG("[PISCSI] RDB - first partition at block %d.\n", be32toh(rdb->rdb_PartitionList));
172     if (d->rdb)
173         free(d->rdb);
174     d->rdb = rdb;
175     sprintf(d->rdb->rdb_DriveInitName, "pi-scsi.device");
176     return 0;
177
178 no_rdb_found:;
179     if (block)
180         free(block);
181
182     return -1;
183 }
184
185 void piscsi_refresh_drives() {
186     piscsi_num_fs = 0;
187
188     for (int i = 0; i < NUM_FILESYSTEMS; i++) {
189         if (filesystems[i].binary_data) {
190             free(filesystems[i].binary_data);
191             filesystems[i].binary_data = NULL;
192         }
193         if (filesystems[i].fhb) {
194             free(filesystems[i].fhb);
195             filesystems[i].fhb = NULL;
196         }
197         filesystems[i].h_info.current_hunk = 0;
198         filesystems[i].h_info.reloc_hunks = 0;
199         filesystems[i].FS_ID = 0;
200         filesystems[i].handler = 0;
201     }
202
203     rom_cur_fs = 0;
204
205     for (int i = 0; i < NUM_UNITS; i++) {
206         if (devs[i].fd != -1) {
207             piscsi_parse_rdb(&devs[i]);
208             piscsi_find_partitions(&devs[i]);
209             piscsi_find_filesystems(&devs[i]);
210         }
211     }
212 }
213
214 void piscsi_find_filesystems(struct piscsi_dev *d) {
215     if (!d->num_partitions)
216         return;
217
218     uint8_t fs_found = 0;
219
220     uint8_t *fhb_block = malloc(512);
221
222     lseek64(d->fd, d->fshd_offs, SEEK_SET);
223
224     struct FileSysHeaderBlock *fhb = (struct FileSysHeaderBlock *)fhb_block;
225     read(d->fd, fhb_block, 512);
226
227     while (BE(fhb->fhb_ID) == FS_IDENTIFIER) {
228         char *dosID = (char *)&fhb->fhb_DosType;
229 #ifdef PISCSI_DEBUG
230         uint16_t *fsVer = (uint16_t *)&fhb->fhb_Version;
231
232         DEBUG("[FSHD] FSHD Block found.\n");
233         DEBUG("[FSHD] HostID: %d Next: %d Size: %d\n", BE(fhb->fhb_HostID), BE(fhb->fhb_Next), BE(fhb->fhb_SummedLongs));
234         DEBUG("[FSHD] Flags: %.8X DOSType: %c%c%c/%d\n", BE(fhb->fhb_Flags), dosID[0], dosID[1], dosID[2], dosID[3]);
235         DEBUG("[FSHD] Version: %d.%d\n", BE16(fsVer[0]), BE16(fsVer[1]));
236         DEBUG("[FSHD] Patchflags: %d Type: %d\n", BE(fhb->fhb_PatchFlags), BE(fhb->fhb_Type));
237         DEBUG("[FSHD] Task: %d Lock: %d\n", BE(fhb->fhb_Task), BE(fhb->fhb_Lock));
238         DEBUG("[FSHD] Handler: %d StackSize: %d\n", BE(fhb->fhb_Handler), BE(fhb->fhb_StackSize));
239         DEBUG("[FSHD] Prio: %d Startup: %d\n", BE(fhb->fhb_Priority), BE(fhb->fhb_Startup));
240         DEBUG("[FSHD] SegListBlocks: %d GlobalVec: %d\n", BE(fhb->fhb_Priority), BE(fhb->fhb_Startup));
241         DEBUG("[FSHD] FileSysName: %s\n", fhb->fhb_FileSysName + 1);
242 #endif
243
244         for (int i = 0; i < NUM_FILESYSTEMS; i++) {
245             if (filesystems[i].FS_ID == fhb->fhb_DosType) {
246                 DEBUG("[FSHD] File system %c%c%c/%d already loaded. Skipping.\n", dosID[0], dosID[1], dosID[2], dosID[3]);
247                 if (BE(fhb->fhb_Next) == 0xFFFFFFFF)
248                     goto fs_done;
249
250                 goto skip_fs_load_lseg;
251             }
252         }
253
254         if (load_lseg(d->fd, &filesystems[piscsi_num_fs].binary_data, &filesystems[piscsi_num_fs].h_info, filesystems[piscsi_num_fs].relocs) != -1) {
255             filesystems[piscsi_num_fs].FS_ID = fhb->fhb_DosType;
256             filesystems[piscsi_num_fs].fhb = fhb;
257             printf("[FSHD] Loaded and set up file system %d: %c%c%c/%d\n", piscsi_num_fs + 1, dosID[0], dosID[1], dosID[2], dosID[3]);
258             piscsi_num_fs++;
259         }
260
261 skip_fs_load_lseg:;
262         fs_found++;
263         lseek64(d->fd, BE(fhb->fhb_Next) * 512, SEEK_SET);
264         fhb_block = malloc(512);
265         fhb = (struct FileSysHeaderBlock *)fhb_block;
266         read(d->fd, fhb_block, 512);
267     }
268
269     if (!fs_found) {
270         DEBUG("[!!!FSHD] No file systems found on hard drive!\n");
271     }
272
273 fs_done:;
274     if (fhb_block)
275         free(fhb_block);
276 }
277
278 void piscsi_map_drive(char *filename, uint8_t index) {
279     if (index > 7) {
280         printf("[PISCSI] Drive index %d out of range.\nUnable to map file %s to drive.\n", index, filename);
281         return;
282     }
283
284     int32_t tmp_fd = open(filename, O_RDWR);
285     if (tmp_fd == -1) {
286         printf("[PISCSI] Failed to open file %s, could not map drive %d.\n", filename, index);
287         return;
288     }
289
290     struct piscsi_dev *d = &devs[index];
291
292     uint64_t file_size = lseek(tmp_fd, 0, SEEK_END);
293     d->fs = file_size;
294     d->fd = tmp_fd;
295     lseek(tmp_fd, 0, SEEK_SET);
296     printf("[PISCSI] Map %d: [%s] - %llu bytes.\n", index, filename, file_size);
297
298     if (piscsi_parse_rdb(d) == -1) {
299         DEBUG("[PISCSI] No RDB found on disk, making up some CHS values.\n");
300         d->h = 16;
301         d->s = 63;
302         d->c = (file_size / 512) / (d->s * d->h);
303     }
304     printf("[PISCSI] CHS: %d %d %d\n", d->c, d->h, d->s);
305
306     piscsi_find_partitions(d);
307     piscsi_find_filesystems(d);
308 }
309
310 void piscsi_unmap_drive(uint8_t index) {
311     if (devs[index].fd != -1) {
312         DEBUG("[PISCSI] Unmapped drive %d.\n", index);
313         close (devs[index].fd);
314         devs[index].fd = -1;
315     }
316 }
317
318 char *io_cmd_name(int index) {
319     switch (index) {
320         case CMD_INVALID: return "INVALID";
321         case CMD_RESET: return "RESET";
322         case CMD_READ: return "READ";
323         case CMD_WRITE: return "WRITE";
324         case CMD_UPDATE: return "UPDATE";
325         case CMD_CLEAR: return "CLEAR";
326         case CMD_STOP: return "STOP";
327         case CMD_START: return "START";
328         case CMD_FLUSH: return "FLUSH";
329         case TD_MOTOR: return "TD_MOTOR";
330         case TD_SEEK: return "SEEK";
331         case TD_FORMAT: return "FORMAT";
332         case TD_REMOVE: return "REMOVE";
333         case TD_CHANGENUM: return "CHANGENUM";
334         case TD_CHANGESTATE: return "CHANGESTATE";
335         case TD_PROTSTATUS: return "PROTSTATUS";
336         case TD_RAWREAD: return "RAWREAD";
337         case TD_RAWWRITE: return "RAWWRITE";
338         case TD_GETDRIVETYPE: return "GETDRIVETYPE";
339         case TD_GETNUMTRACKS: return "GETNUMTRACKS";
340         case TD_ADDCHANGEINT: return "ADDCHANGEINT";
341         case TD_REMCHANGEINT: return "REMCHANGEINT";
342         case TD_GETGEOMETRY: return "GETGEOMETRY";
343         case TD_EJECT: return "EJECT";
344         case TD_LASTCOMM: return "LASTCOMM/READ64";
345         case TD_WRITE64: return "WRITE64";
346         case HD_SCSICMD: return "HD_SCSICMD";
347         case NSCMD_DEVICEQUERY: return "NSCMD_DEVICEQUERY";
348         case NSCMD_TD_READ64: return "NSCMD_TD_READ64";
349         case NSCMD_TD_WRITE64: return "NSCMD_TD_WRITE64";
350         case NSCMD_TD_FORMAT64: return "NSCMD_TD_FORMAT64";
351
352         default:
353             return "[!!!PISCSI] Unhandled IO command";
354     }
355 }
356
357 #define GETSCSINAME(a) case a: return ""#a"";
358 #define SCSIUNHANDLED(a) return "[!!!PISCSI] Unhandled SCSI command "#a"";
359
360 char *scsi_cmd_name(int index) {
361     switch(index) {
362         GETSCSINAME(SCSICMD_TEST_UNIT_READY);
363         GETSCSINAME(SCSICMD_INQUIRY);
364         GETSCSINAME(SCSICMD_READ_6);
365         GETSCSINAME(SCSICMD_WRITE_6);
366         GETSCSINAME(SCSICMD_READ_10);
367         GETSCSINAME(SCSICMD_WRITE_10);
368         GETSCSINAME(SCSICMD_READ_CAPACITY_10);
369         GETSCSINAME(SCSICMD_MODE_SENSE_6);
370         GETSCSINAME(SCSICMD_READ_DEFECT_DATA_10);
371         default:
372             return "[!!!PISCSI] Unhandled SCSI command";
373     }
374 }
375
376 void print_piscsi_debug_message(int index) {
377     int32_t r = 0;
378
379     switch (index) {
380         case DBG_INIT:
381             DEBUG("[PISCSI] Initializing devices.\n");
382             break;
383         case DBG_OPENDEV:
384             if (piscsi_dbg[0] != 255) {
385                 DEBUG("[PISCSI] Opening device %d (%d). Flags: %d (%.2X)\n", piscsi_dbg[0], piscsi_dbg[2], piscsi_dbg[1], piscsi_dbg[1]);
386             }
387             break;
388         case DBG_CLEANUP:
389             DEBUG("[PISCSI] Cleaning up.\n");
390             break;
391         case DBG_CHS:
392             DEBUG("[PISCSI] C/H/S: %d / %d / %d\n", piscsi_dbg[0], piscsi_dbg[1], piscsi_dbg[2]);
393             break;
394         case DBG_BEGINIO:
395             DEBUG("[PISCSI] BeginIO: io_Command: %d (%s) - io_Flags = %d - quick: %d\n", piscsi_dbg[0], io_cmd_name(piscsi_dbg[0]), piscsi_dbg[1], piscsi_dbg[2]);
396             break;
397         case DBG_ABORTIO:
398             DEBUG("[PISCSI] AbortIO!\n");
399             break;
400         case DBG_SCSICMD:
401             DEBUG("[PISCSI] SCSI Command %d (%s)\n", piscsi_dbg[1], scsi_cmd_name(piscsi_dbg[1]));
402             DEBUG("Len: %d - %.2X %.2X %.2X - Command Length: %d\n", piscsi_dbg[0], piscsi_dbg[1], piscsi_dbg[2], piscsi_dbg[3], piscsi_dbg[4]);
403             break;
404         case DBG_SCSI_UNKNOWN_MODESENSE:
405             DEBUG("[!!!PISCSI] SCSI: Unknown modesense %.4X\n", piscsi_dbg[0]);
406             break;
407         case DBG_SCSI_UNKNOWN_COMMAND:
408             DEBUG("[!!!PISCSI] SCSI: Unknown command %.4X\n", piscsi_dbg[0]);
409             break;
410         case DBG_SCSIERR:
411             DEBUG("[!!!PISCSI] SCSI: An error occured: %.4X\n", piscsi_dbg[0]);
412             break;
413         case DBG_IOCMD:
414             DEBUG_TRIVIAL("[PISCSI] IO Command %d (%s)\n", piscsi_dbg[0], io_cmd_name(piscsi_dbg[0]));
415             break;
416         case DBG_IOCMD_UNHANDLED:
417             DEBUG("[!!!PISCSI] WARN: IO command %.4X (%s) is unhandled by driver.\n", piscsi_dbg[0], io_cmd_name(piscsi_dbg[0]));
418             break;
419         case DBG_SCSI_FORMATDEVICE:
420             DEBUG("[PISCSI] Get SCSI FormatDevice MODE SENSE.\n");
421             break;
422         case DBG_SCSI_RDG:
423             DEBUG("[PISCSI] Get SCSI RDG MODE SENSE.\n");
424             break;
425         case DBG_SCSICMD_RW10:
426 #ifdef PISCSI_DEBUG
427             r = get_mapped_item_by_address(cfg, piscsi_dbg[0]);
428             struct SCSICmd_RW10 *rwdat = NULL;
429             char data[10];
430             if (r != -1) {
431                 uint32_t addr = piscsi_dbg[0] - cfg->map_offset[r];
432                 rwdat = (struct SCSICmd_RW10 *)(&cfg->map_data[r][addr]);
433             }
434             else {
435                 DEBUG_TRIVIAL("[RW10] scsiData: %.8X\n", piscsi_dbg[0]);
436                 for (int i = 0; i < 10; i++) {
437                     data[i] = read8(piscsi_dbg[0] + i);
438                 }
439                 rwdat = data;
440             }
441             if (rwdat) {
442                 DEBUG_TRIVIAL("[RW10] CMD: %.2X\n", rwdat->opcode);
443                 DEBUG_TRIVIAL("[RW10] RDP: %.2X\n", rwdat->rdprotect_flags);
444                 DEBUG_TRIVIAL("[RW10] Block: %d (%d)\n", rwdat->block, BE(rwdat->block));
445                 DEBUG_TRIVIAL("[RW10] Res_Group: %.2X\n", rwdat->res_groupnum);
446                 DEBUG_TRIVIAL("[RW10] Len: %d (%d)\n", rwdat->len, BE16(rwdat->len));
447             }
448 #endif
449             break;
450         case DBG_SCSI_DEBUG_MODESENSE_6:
451             DEBUG_TRIVIAL("[PISCSI] SCSI ModeSense debug. Data: %.8X\n", piscsi_dbg[0]);
452             r = get_mapped_item_by_address(cfg, piscsi_dbg[0]);
453             if (r != -1) {
454 #ifdef PISCSI_DEBUG
455                 uint32_t addr = piscsi_dbg[0] - cfg->map_offset[r];
456                 struct SCSICmd_ModeSense6 *sense = (struct SCSICmd_ModeSense6 *)(&cfg->map_data[r][addr]);
457                 DEBUG_TRIVIAL("[SenseData] CMD: %.2X\n", sense->opcode);
458                 DEBUG_TRIVIAL("[SenseData] DBD: %d\n", sense->reserved_dbd & 0x04);
459                 DEBUG_TRIVIAL("[SenseData] PC: %d\n", (sense->pc_pagecode & 0xC0 >> 6));
460                 DEBUG_TRIVIAL("[SenseData] PageCodes: %.2X %.2X\n", (sense->pc_pagecode & 0x3F), sense->subpage_code);
461                 DEBUG_TRIVIAL("[SenseData] AllocLen: %d\n", sense->alloc_len);
462                 DEBUG_TRIVIAL("[SenseData] Control: %.2X (%d)\n", sense->control, sense->control);
463 #endif
464             }
465             else {
466                 DEBUG("[!!!PISCSI] ModeSense data not immediately available.\n");
467             }
468             break;
469         default:
470             DEBUG("[!!!PISCSI] No debug message available for index %d.\n", index);
471             break;
472     }
473 }
474
475 #define DEBUGME_SIMPLE(i, s) case i: DEBUG(s); break;
476
477 void piscsi_debugme(uint32_t index) {
478     switch (index) {
479         DEBUGME_SIMPLE(1, "[PISCSI-DEBUGME] Arrived at DiagEntry.\n");
480         DEBUGME_SIMPLE(3, "[PISCSI-DEBUGME] Init: Interrupt disable.\n");
481         DEBUGME_SIMPLE(4, "[PISCSI-DEBUGME] Init: Copy/reloc driver.\n");
482         DEBUGME_SIMPLE(5, "[PISCSI-DEBUGME] Init: InitResident.\n");
483         DEBUGME_SIMPLE(7, "[PISCSI-DEBUGME] Init: Begin partition loop.\n");
484         DEBUGME_SIMPLE(8, "[PISCSI-DEBUGME] Init: Partition loop done. Cleaning up and returning to Exec.\n");
485         DEBUGME_SIMPLE(9, "[PISCSI-DEBUGME] Init: Load file systems.\n");
486         DEBUGME_SIMPLE(10, "[PISCSI-DEBUGME] Init: AllocMem for resident.\n");
487         DEBUGME_SIMPLE(11, "[PISCSI-DEBUGME] Init: Checking if resident is loaded.\n");
488         DEBUGME_SIMPLE(22, "[PISCSI-DEBUGME] Arrived at BootEntry.\n");
489         DEBUGME_SIMPLE(30, "[PISCSI-DEBUGME] LoadFileSystems: Opening FileSystem.resource.\n");
490         DEBUGME_SIMPLE(33, "[PISCSI-DEBUGME] FileSystem.resource not available, creating.\n");
491         case 31:
492             DEBUG("[PISCSI-DEBUGME] OpenResource result: %d\n", piscsi_u32[0]);
493             break;
494         case 32:
495             DEBUG("AAAAHH!\n");
496             break;
497         default:
498             DEBUG("[!!!PISCSI-DEBUGME] No debugme message for index %d!\n", index);
499             break;
500     }
501
502     if (index == 8) {
503         stop_cpu_emulation(1);
504     }
505 }
506
507 void handle_piscsi_write(uint32_t addr, uint32_t val, uint8_t type) {
508     int32_t r;
509 #ifndef PISCSI_DEBUG
510     if (type) {}
511 #endif
512
513     struct piscsi_dev *d = &devs[piscsi_cur_drive];
514
515     uint16_t cmd = (addr & 0xFFFF);
516
517     switch (cmd) {
518         case PISCSI_CMD_READ64:
519         case PISCSI_CMD_READ:
520             d = &devs[val];
521             if (d->fd == -1) {
522                 DEBUG("[!!!PISCSI] BUG: Attempted read from unmapped drive %d.\n", val);
523                 break;
524             }
525
526             if (cmd == PISCSI_CMD_READ) {
527                 DEBUG("[PISCSI-%d] %d byte READ from block %d to address %.8X\n", val, piscsi_u32[1], piscsi_u32[0], piscsi_u32[2]);
528                 d->lba = piscsi_u32[0];
529                 lseek(d->fd, (piscsi_u32[0] * 512), SEEK_SET);
530             }
531             else {
532                 uint64_t src = piscsi_u32[3];
533                 src = (src << 32) | piscsi_u32[0];
534                 DEBUG("[PISCSI-%d] %d byte READ64 from block %lld to address %.8X\n", val, piscsi_u32[1], (src / 512), piscsi_u32[2]);
535                 d->lba = (src / 512);
536                 lseek64(d->fd, src, SEEK_SET);
537             }
538
539             r = get_mapped_item_by_address(cfg, piscsi_u32[2]);
540             if (r != -1 && cfg->map_type[r] == MAPTYPE_RAM) {
541                 DEBUG_TRIVIAL("[PISCSI-%d] \"DMA\" Read goes to mapped range %d.\n", val, r);
542                 read(d->fd, cfg->map_data[r] + piscsi_u32[2] - cfg->map_offset[r], piscsi_u32[1]);
543             }
544             else {
545                 DEBUG_TRIVIAL("[PISCSI-%d] No mapped range found for read.\n", val);
546                 uint8_t c = 0;
547                 for (uint32_t i = 0; i < piscsi_u32[1]; i++) {
548                     read(d->fd, &c, 1);
549                     write8(piscsi_u32[2] + i, (uint32_t)c);
550                 }
551             }
552             break;
553         case PISCSI_CMD_WRITE64:
554         case PISCSI_CMD_WRITE:
555             d = &devs[val];
556             if (d->fd == -1) {
557                 DEBUG ("[PISCSI] BUG: Attempted write to unmapped drive %d.\n", val);
558                 break;
559             }
560
561             if (cmd == PISCSI_CMD_WRITE) {
562                 DEBUG("[PISCSI-%d] %d byte WRITE to block %d from address %.8X\n", val, piscsi_u32[1], piscsi_u32[0], piscsi_u32[2]);
563                 d->lba = piscsi_u32[0];
564                 lseek(d->fd, (piscsi_u32[0] * 512), SEEK_SET);
565             }
566             else {
567                 uint64_t src = piscsi_u32[3];
568                 src = (src << 32) | piscsi_u32[0];
569                 DEBUG("[PISCSI-%d] %d byte WRITE64 to block %lld from address %.8X\n", val, piscsi_u32[1], (src / 512), piscsi_u32[2]);
570                 d->lba = (src / 512);
571                 lseek64(d->fd, src, SEEK_SET);
572             }
573
574             r = get_mapped_item_by_address(cfg, piscsi_u32[2]);
575             if (r != -1) {
576                 DEBUG_TRIVIAL("[PISCSI-%d] \"DMA\" Write comes from mapped range %d.\n", val, r);
577                 write(d->fd, cfg->map_data[r] + piscsi_u32[2] - cfg->map_offset[r], piscsi_u32[1]);
578             }
579             else {
580                 DEBUG_TRIVIAL("[PISCSI-%d] No mapped range found for write.\n", val);
581                 uint8_t c = 0;
582                 for (uint32_t i = 0; i < piscsi_u32[1]; i++) {
583                     c = read8(piscsi_u32[2] + i);
584                     write(d->fd, &c, 1);
585                 }
586             }
587             break;
588         case PISCSI_CMD_ADDR1: case PISCSI_CMD_ADDR2: case PISCSI_CMD_ADDR3: case PISCSI_CMD_ADDR4: {
589             int i = ((addr & 0xFFFF) - PISCSI_CMD_ADDR1) / 4;
590             piscsi_u32[i] = val;
591             break;
592         }
593         case PISCSI_CMD_DRVNUM:
594             //printf("%d ", val);
595             if (val % 10 != 0)
596                 piscsi_cur_drive = 255;
597             else
598                 piscsi_cur_drive = val / 10;
599             if (piscsi_cur_drive > NUM_UNITS)
600                 piscsi_cur_drive = 255;
601
602             if (piscsi_cur_drive != 255) {
603                 DEBUG("[PISCSI] (%s) Drive number set to %d (%d)\n", op_type_names[type], piscsi_cur_drive, val);
604             }
605             break;
606         case PISCSI_CMD_DEBUGME:
607             piscsi_debugme(val);
608             break;
609         case PISCSI_CMD_DRIVER: {
610             DEBUG("[PISCSI] Driver copy/patch called, destination address %.8X.\n", val);
611             int r = get_mapped_item_by_address(cfg, val);
612             if (r != -1) {
613                 uint32_t addr = val - cfg->map_offset[r];
614                 uint8_t *dst_data = cfg->map_data[r];
615                 uint8_t cur_partition = 0;
616                 memcpy(dst_data + addr, piscsi_rom_ptr + PISCSI_DRIVER_OFFSET, 0x4000 - PISCSI_DRIVER_OFFSET);
617
618                 piscsi_hinfo.base_offset = val;
619
620                 reloc_hunks(piscsi_hreloc, dst_data + addr, &piscsi_hinfo);
621
622                 #define PUTNODELONG(val) *(uint32_t *)&dst_data[p_offs] = htobe32(val); p_offs += 4;
623                 #define PUTNODELONGBE(val) *(uint32_t *)&dst_data[p_offs] = val; p_offs += 4;
624
625                 for (int i = 0; i < 128; i++) {
626                     rom_partitions[i] = 0;
627                     rom_partition_prio[i] = 0;
628                     rom_partition_dostype[i] = 0;
629                 }
630                 rom_cur_partition = 0;
631
632                 uint32_t data_addr = addr + 0x3F00;
633                 sprintf((char *)dst_data + data_addr, "pi-scsi.device");
634                 uint32_t addr2 = addr + 0x4000;
635                 for (int i = 0; i < NUM_UNITS; i++) {
636                     if (devs[i].fd != -1)
637                         piscsi_find_partitions(&devs[i]);
638                     else
639                         goto skip_disk;
640
641                     if (devs[i].num_partitions) {
642                         uint32_t p_offs = addr2;
643                         DEBUG("[PISCSI] Adding %d partitions for unit %d\n", devs[i].num_partitions, i);
644                         for (uint32_t j = 0; j < devs[i].num_partitions; j++) {
645                             DEBUG("Partition %d: %s\n", j, devs[i].pb[j]->pb_DriveName + 1);
646                             sprintf((char *)dst_data + p_offs, "%s", devs[i].pb[j]->pb_DriveName + 1);
647                             p_offs += 0x20;
648                             PUTNODELONG(addr2 + cfg->map_offset[r]);
649                             PUTNODELONG(data_addr + cfg->map_offset[r]);
650                             PUTNODELONG((i * 10));
651                             PUTNODELONG(0);
652                             uint32_t nodesize = (be32toh(devs[i].pb[j]->pb_Environment[0]) + 1) * 4;
653                             memcpy(dst_data + p_offs, devs[i].pb[j]->pb_Environment, nodesize);
654
655                             struct pihd_dosnode_data *dat = (struct pihd_dosnode_data *)(&dst_data[addr2+0x20]);
656
657                             if (BE(devs[i].pb[j]->pb_Flags) & 0x01) {
658                                 DEBUG("Partition is bootable.\n");
659                                 rom_partition_prio[cur_partition] = 0;
660                                 dat->priority = 0;
661                             }
662                             else {
663                                 DEBUG("Partition is not bootable.\n");
664                                 rom_partition_prio[cur_partition] = -128;
665                                 dat->priority = htobe32(-128);
666                             }
667
668                             DEBUG("DOSNode Data:\n");
669                             DEBUG("Name: %s Device: %s\n", dst_data + addr2, dst_data + data_addr);
670                             DEBUG("Unit: %d Flags: %d Pad1: %d\n", BE(dat->unit), BE(dat->flags), BE(dat->pad1));
671                             DEBUG("Node len: %d Block len: %d\n", BE(dat->node_len) * 4, BE(dat->block_len) * 4);
672                             DEBUG("H: %d SPB: %d BPS: %d\n", BE(dat->surf), BE(dat->secs_per_block), BE(dat->blocks_per_track));
673                             DEBUG("Reserved: %d Prealloc: %d\n", BE(dat->reserved_blocks), BE(dat->pad2));
674                             DEBUG("Interleaved: %d Buffers: %d Memtype: %d\n", BE(dat->interleave), BE(dat->buffers), BE(dat->mem_type));
675                             DEBUG("Lowcyl: %d Highcyl: %d Prio: %d\n", BE(dat->lowcyl), BE(dat->highcyl), BE(dat->priority));
676                             DEBUG("Maxtransfer: %.8X Mask: %.8X\n", BE(dat->maxtransfer), BE(dat->transfer_mask));
677                             DEBUG("DOSType: %.8X\n", BE(dat->dostype));
678
679                             rom_partitions[cur_partition] = addr2 + 0x20 + cfg->map_offset[r];
680                             rom_partition_dostype[cur_partition] = dat->dostype;
681                             cur_partition++;
682                             addr2 += 0x100;
683                             p_offs = addr2;
684                         }
685                     }
686 skip_disk:;
687                 }
688             }
689
690             break;
691         }
692         case PISCSI_CMD_NEXTPART:
693             DEBUG("[PISCSI] Switch partition %d -> %d\n", rom_cur_partition, rom_cur_partition + 1);
694             rom_cur_partition++;
695             break;
696         case PISCSI_CMD_NEXTFS:
697             DEBUG("[PISCSI] Switch file file system %d -> %d\n", rom_cur_fs, rom_cur_fs + 1);
698             rom_cur_fs++;
699             break;
700         case PISCSI_CMD_COPYFS:
701             DEBUG("[PISCSI] Copy file system %d to %.8X and reloc.\n", rom_cur_fs, piscsi_u32[2]);
702             r = get_mapped_item_by_address(cfg, piscsi_u32[2]);
703             if (r != -1) {
704                 uint32_t addr = piscsi_u32[2] - cfg->map_offset[r];
705                 memset(cfg->map_data[r] + addr, 0x00, filesystems[rom_cur_fs].h_info.alloc_size);
706                 memcpy(cfg->map_data[r] + addr, filesystems[rom_cur_fs].binary_data, filesystems[rom_cur_fs].h_info.byte_size);
707                 filesystems[rom_cur_fs].h_info.base_offset = piscsi_u32[2];
708                 reloc_hunks(filesystems[rom_cur_fs].relocs, cfg->map_data[r] + addr, &filesystems[rom_cur_fs].h_info);
709                 filesystems[rom_cur_fs].handler = piscsi_u32[2];
710             }
711             break;
712         case PISCSI_CMD_SETFSH: {
713             int i = 0;
714             DEBUG("[PISCSI] Set handler for partition %d (DeviceNode: %.8X)\n", rom_cur_partition, val);
715             r = get_mapped_item_by_address(cfg, val);
716             if (r != -1) {
717                 uint32_t addr = val - cfg->map_offset[r];
718                 struct DeviceNode *node = (struct DeviceNode *)(cfg->map_data[r] + addr);
719 #ifdef PISCSI_DEBUG
720                 char *dosID = (char *)&rom_partition_dostype[rom_cur_partition];
721 #endif
722                 DEBUG("[PISCSI] Partition DOSType is %c%c%c/%d\n", dosID[0], dosID[1], dosID[2], dosID[3]);
723                 for (i = 0; i < piscsi_num_fs; i++) {
724                     if (rom_partition_dostype[rom_cur_partition] == filesystems[i].FS_ID) {
725                         node->dn_SegList = htobe32((filesystems[i].handler >> 2));
726                         node->dn_GlobalVec = 0xFFFFFFFF;
727                         goto fs_found;
728                     }
729                 }
730                 DEBUG("[!!!PISCSI] Found no handler for file system!\n");
731 fs_found:;
732                 DEBUG("[FS-HANDLER] Next: %d Type: %.8X\n", BE(node->dn_Next), BE(node->dn_Type));
733                 DEBUG("[FS-HANDLER] Task: %d Lock: %d\n", BE(node->dn_Task), BE(node->dn_Lock));
734                 DEBUG("[FS-HANDLER] Handler: %d Stacksize: %d\n", BE((uint32_t)node->dn_Handler), BE(node->dn_StackSize));
735                 DEBUG("[FS-HANDLER] Priority: %d Startup: %d\n", BE((uint32_t)node->dn_Priority), BE(node->dn_Startup));
736                 DEBUG("[FS-HANDLER] SegList: %.8X GlobalVec: %d\n", BE((uint32_t)node->dn_SegList), BE(node->dn_GlobalVec));
737                 DEBUG("[PISCSI] Handler for partition %.8X set to %.8X (%.8X).\n", BE((uint32_t)node->dn_Name), filesystems[i].FS_ID, filesystems[i].handler);
738             }
739             break;
740         }
741         case PISCSI_DBG_VAL1: case PISCSI_DBG_VAL2: case PISCSI_DBG_VAL3: case PISCSI_DBG_VAL4:
742         case PISCSI_DBG_VAL5: case PISCSI_DBG_VAL6: case PISCSI_DBG_VAL7: case PISCSI_DBG_VAL8: {
743             int i = ((addr & 0xFFFF) - PISCSI_DBG_VAL1) / 4;
744             piscsi_dbg[i] = val;
745             break;
746         }
747         case PISCSI_DBG_MSG:
748             print_piscsi_debug_message(val);
749             break;
750         default:
751             DEBUG("[!!!PISCSI] WARN: Unhandled %s register write to %.8X: %d\n", op_type_names[type], addr, val);
752             break;
753     }
754 }
755
756 #define PIB 0x00
757
758 uint32_t handle_piscsi_read(uint32_t addr, uint8_t type) {
759     if (type) {}
760
761     if ((addr & 0xFFFF) >= PISCSI_CMD_ROM) {
762         uint32_t romoffs = (addr & 0xFFFF) - PISCSI_CMD_ROM;
763         if (romoffs < (piscsi_rom_size + PIB)) {
764             //DEBUG("[PISCSI] %s read from Boot ROM @$%.4X (%.8X): ", op_type_names[type], romoffs, addr);
765             uint32_t v = 0;
766             switch (type) {
767                 case OP_TYPE_BYTE:
768                     v = piscsi_rom_ptr[romoffs - PIB];
769                     //DEBUG("%.2X\n", v);
770                     break;
771                 case OP_TYPE_WORD:
772                     v = be16toh(*((uint16_t *)&piscsi_rom_ptr[romoffs - PIB]));
773                     //DEBUG("%.4X\n", v);
774                     break;
775                 case OP_TYPE_LONGWORD:
776                     v = be32toh(*((uint32_t *)&piscsi_rom_ptr[romoffs - PIB]));
777                     //DEBUG("%.8X\n", v);
778                     break;
779             }
780             return v;
781         }
782         return 0;
783     }
784
785     switch (addr & 0xFFFF) {
786         case PISCSI_CMD_ADDR1: case PISCSI_CMD_ADDR2: case PISCSI_CMD_ADDR3: case PISCSI_CMD_ADDR4: {
787             int i = ((addr & 0xFFFF) - PISCSI_CMD_ADDR1) / 4;
788             return piscsi_u32[i];
789             break;
790         }
791         case PISCSI_CMD_DRVTYPE:
792             if (devs[piscsi_cur_drive].fd == -1) {
793                 DEBUG("[PISCSI] %s Read from DRVTYPE %d, drive not attached.\n", op_type_names[type], piscsi_cur_drive);
794                 return 0;
795             }
796             DEBUG("[PISCSI] %s Read from DRVTYPE %d, drive attached.\n", op_type_names[type], piscsi_cur_drive);
797             return 1;
798             break;
799         case PISCSI_CMD_DRVNUM:
800             return piscsi_cur_drive;
801             break;
802         case PISCSI_CMD_CYLS:
803             DEBUG("[PISCSI] %s Read from CYLS %d: %d\n", op_type_names[type], piscsi_cur_drive, devs[piscsi_cur_drive].c);
804             return devs[piscsi_cur_drive].c;
805             break;
806         case PISCSI_CMD_HEADS:
807             DEBUG("[PISCSI] %s Read from HEADS %d: %d\n", op_type_names[type], piscsi_cur_drive, devs[piscsi_cur_drive].h);
808             return devs[piscsi_cur_drive].h;
809             break;
810         case PISCSI_CMD_SECS:
811             DEBUG("[PISCSI] %s Read from SECS %d: %d\n", op_type_names[type], piscsi_cur_drive, devs[piscsi_cur_drive].s);
812             return devs[piscsi_cur_drive].s;
813             break;
814         case PISCSI_CMD_BLOCKS: {
815             uint32_t blox = devs[piscsi_cur_drive].fs / 512;
816             DEBUG("[PISCSI] %s Read from BLOCKS %d: %d\n", op_type_names[type], piscsi_cur_drive, (uint32_t)(devs[piscsi_cur_drive].fs / 512));
817             DEBUG("fs: %lld (%d)\n", devs[piscsi_cur_drive].fs, blox);
818             return blox;
819             break;
820         }
821         case PISCSI_CMD_GETPART: {
822             DEBUG("[PISCSI] Get ROM partition %d offset: %.8X\n", rom_cur_partition, rom_partitions[rom_cur_partition]);
823             return rom_partitions[rom_cur_partition];
824             break;
825         }
826         case PISCSI_CMD_GETPRIO:
827             DEBUG("[PISCSI] Get partition %d boot priority: %d\n", rom_cur_partition, rom_partition_prio[rom_cur_partition]);
828             return rom_partition_prio[rom_cur_partition];
829             break;
830         case PISCSI_CMD_CHECKFS:
831             DEBUG("[PISCSI] Get current loaded file system: %.8X\n", filesystems[rom_cur_fs].FS_ID);
832             return filesystems[rom_cur_fs].FS_ID;
833         case PISCSI_CMD_FSSIZE:
834             DEBUG("[PISCSI] Get alloc size of loaded file system: %d\n", filesystems[rom_cur_fs].h_info.alloc_size);
835             return filesystems[rom_cur_fs].h_info.alloc_size;
836         default:
837             DEBUG("[!!!PISCSI] WARN: Unhandled %s register read from %.8X\n", op_type_names[type], addr);
838             break;
839     }
840
841     return 0;
842 }