// The OS/K Team licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#include "arch.h"

#define FWPROGSIZE (1024 * 1024 * 1024)
static ssize_t fwsize;
static ushort *fwprog;

ushort bget(ctx_t *ctx)
{
    if (ctx->r[RIP].val % 2) {
        _except(ctx, E_ALI, "Misaligned RIP register: 0x%016lX",
            ctx->r[RIP].val);
    }

    if (addr2real(ctx->r[RIP].val) >= ctx->mz) {
        _except(ctx, E_ACC, "Executing out of memory: 0x%016lX",
            ctx->r[RIP].val);
    }

    ushort c = ctx->mp[addr2real(ctx->r[RIP].val)];

    ctx->r[RIP].val += 2;

    return c;
}

ushort dget(ctx_t *ctx)
{
    static int i = 0;

    if (i >= fwsize) {
        log("Finished disassembling\n");
        fclose(ctx->disf);
        exit(0);
    }

    ctx->r[RIP].val += 2;
    return fwprog[i++];
}

int main(int argc, char **argv)
{
    ctx_t main_ctx;
    FILE *fwfile;

    main_ctx.r = arch_r;
    main_ctx.i = arch_i;

    if (argc < 2) {
        log("Not enough arguments\n");
        exit(-3);
    }

    fwprog = malloc(FWPROGSIZE);
    fwfile = fopen(argv[1], "rb");

    if (!fwprog) {
        log("Couldn't allocate firmware buffer\n");
        exit(-1);
    }

    if (!fwfile) {
        log("Couldn't open program file\n");
        exit(-2);
    }

    fwsize = fread(fwprog, 1, FWPROGSIZE, fwfile);

    //log("Loaded %lu bytes

    if (fwsize < 2) {
        log("Program file too small or empty\n");
        exit(-3);
    }

    if (!(argc > 2 && !strcmp(argv[2], "-d"))) {
        main_ctx.mp = malloc(MEMSIZE + 16);
        main_ctx.mz = MEMSIZE;

        main_ctx.get = bget;
        main_ctx.disf = NULL;

        main_ctx.r[RIP].val = MEMOFF;

        if (main_ctx.mp == 0) {
            log("Couldn't allocate RAM\n");
            exit(-1);
        }

        memcpy(&main_ctx.mp[addr2real(main_ctx.r[RIP].val)], fwprog, fwsize);

        while (1) {
            decode(&main_ctx);
        }
    }

    // Disassembly
    else {
        main_ctx.disf = fopen("fwprog.dis", "w");

        if (main_ctx.disf == NULL) {
            log("Couldn't open fwprog.dis\n");
            exit(-2);
        }

        main_ctx.mp = NULL;
        main_ctx.mz = 0;

        main_ctx.get = dget;

        while (1) {
            disasm(&main_ctx);
        }
    }


    return 0;
}