1 files changed, 56 insertions, 0 deletions

diff --git a/rom.c b/rom.c
new file mode 100644
index 0000000..d432cc2
--- /dev/null
+++ b/rom.c
@@ -0,0 +1,56 @@
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "rom.h"
+
+void
+parse_rom(const uint8_t *data, size_t data_len, struct Rom *rom)
+{
+        size_t prg_rom_offset = 16;
+        size_t chr_rom_offset = 0;
+
+        if (data_len < 16) {
+                fprintf(stderr, "Invalid ROM: size less than 16 bytes\n");
+                exit(1);
+        }
+
+        if (data[0] != 'N' || data[1] != 'E' || data[2] != 'S' || data[3] != 0x1A) {
+                fprintf(stderr, "Invalid ROM: not in iNES format\n");
+                exit(1);
+        }
+
+        /* trainer data present */
+        if (data[6] & (1 << 2))
+                prg_rom_offset += 512;
+
+        rom->prg_rom_size = data[4] * 0x4000;
+        rom->prg_rom = calloc(rom->prg_rom_size, 1);
+        memcpy(rom->prg_rom, data + prg_rom_offset, rom->prg_rom_size);
+
+        rom->chr_rom_size = data[5] * 0x2000;
+        if (rom->chr_rom_size > 0) {
+                rom->chr_rom = calloc(rom->chr_rom_size, 1);
+                chr_rom_offset = prg_rom_offset + rom->prg_rom_size;
+                memcpy(rom->chr_rom, data + chr_rom_offset, rom->chr_rom_size);
+        }
+
+        if ((data[6] & (1 << 3)) != 0)
+                rom->mirror = M_FOUR;
+        else if ((data[6] & 1) == 1)
+                rom->mirror = M_VERTICAL;
+        else if ((data[6] & 1) == 0)
+                rom->mirror = M_HORIZONTAL;
+
+        rom->mapper = (data[6] & (0xF << 4)) >> 4;
+        rom->mapper |= data[7] & (0xF << 4);
+}
+
+void
+free_rom(struct Rom *rom)
+{
+        free(rom->prg_rom);
+        if (rom->chr_rom_size > 0)
+                free(rom->chr_rom);
+}