#include <windows.h>
#include <stdio.h>

BOOL WINAPI VirtualCopy(LPVOID lpvDest, LPVOID lpvSrc, DWORD cbSize, DWORD fdwProtect);

enum {
  PAGESIZE = 0x10000,
};

static BOOL writefile(const void *p, unsigned size, const wchar_t *fname)
{
  FILE *f = 0;
  const char *pb = p;
  BOOL ret = FALSE;

  f = _wfopen(fname, L"wb");
  if (!f) {
    wprintf(L"\nError: fopen failed! Aborting.\n");
    goto cleanup;
  }
  while (size) {
    const unsigned wsize = (size < PAGESIZE) ? size : PAGESIZE;
    if (fwrite(pb, 1, wsize, f) != wsize) {
      wprintf(L"\nError: fwrite failed! Aborting.\n");
      goto cleanup;
    }
    wprintf(L".");
    fflush(stdout);
    pb += wsize;
    size -= wsize;
  }
  ret = TRUE;
cleanup:
  if (f) fclose(f);
  return ret;
}


struct data {
  unsigned addr;
  unsigned size;
  const wchar_t *name;
};

const struct data roms[] = {
  {0x00000000, 0x02000000, L"rom.dat"},  // 0x0000_0000 - 0x01ff_ffff
  {0x0c000000, 0x00400000, L"flash.dat"}, // 0x0c00_0000 - 0x0c3f_ffff
  {0, 0, 0},
};

int wmain(int argc, wchar_t **argv)
{
  wchar_t fname[MAX_PATH];
  wchar_t *pathend;
  const struct data *i;

  if (!GetModuleFileName(0, fname, MAX_PATH)) {
    wprintf(L"Error: Cannot get executable file path! Aborting.\n");
    goto end;
  }
  pathend = wcsrchr(fname, L'\\');
  if (!pathend) {
    wprintf(L"Error: Cannot get directory to write rom files! Aborting\n");
    goto end;
  }
  for (i = roms; i->name; i++) {
    void *p;

    /* too lazy to check string length */
    pathend[1] = 0;
    wcscat(fname, i->name);
    wprintf(L"Dumping %ls", fname);
    fflush(stdout);
    p = VirtualAlloc(0, i->size, MEM_RESERVE, PAGE_NOACCESS);
    if (!p) {
      wprintf(L"\nError: VirtualAlloc failed! Aborting.\n");
      goto end;
    }
    if (!VirtualCopy(p, (void *)(i->addr >> 8), i->size, PAGE_READONLY | PAGE_PHYSICAL)) {
      wprintf(L"\nError: VirtualCopy failed! Aborting.\n");
      goto end;
    }
    if (!writefile(p, i->size, fname)) return 1;
    wprintf(L"OK!\n");
  }
end:
  wprintf(L"Press enter to continue...\n");
  getwchar();
  return 0;
}