122 lines
2.3 KiB
C
122 lines
2.3 KiB
C
// 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;
|
|
}
|
|
|