r0.c

/* MIT 2009-2021 -- pancake /at/ nopcode /dot/ org */

#define ut64 unsigned long long 
#define ut8 unsigned char

static int earlyquit = 0;
static int verbose = 1;
static int scriptn = 0;
static const char **scripts = 0;
static int cmdn = 0;
static char **cmds = 0;
static ut64 oldseek, curseek = 0LL;
static int obsize, bsize = 256;
int r0_cmd(const char *cmd); // XXX : recursive depenency
#if defined(__WATCOMC__)
#define HAVE_FTRUNCATE 0
#define BUFSZ 1024
#elif __wasi__
#define BUFSZ 128
#else
#define BUFSZ 128*1024
#endif

ut64 calc(const char *str);
#include "r0.h"
#include "util.c"
#include "calc.c"
#include "cmd.c"

static void r0_slurpin() {
	ut8 buf[BUFSZ];
	for(;;) {
		int len = read(0, buf, sizeof(buf));
		if(len<1) break;
		hexdump(buf, len, 16);
		curseek += len;
	}
}

static int r0_interpret(const char *file) {
	char buf[BUFSZ];
	FILE *fd = fopen(file, "r");
	if(fd != NULL) {
		for(;;) {
			if(fgets(buf, sizeof(buf), fd) == NULL)
				break;
			r0_cmd(buf);
		}
		fclose(fd);
	} else if (file)
		fprintf(stderr, "Cannot open script file '%s'\n", file);
	return 1;
}

int r0_cmd(const char *cmd) {
	const char *arg = cmd+1;
	SKIPSPACES(arg);
	switch(*cmd) {
	case 'q': return 0;
	case ';': case '#': break; // comment
	case '>': return cmd_dump(arg); break;
	case '<': return cmd_load(arg); break;
	case '.': return r0_interpret(arg); break;
	case 's': return cmd_seek(arg); break;
	case 'b': return cmd_bsize(arg); break;
	case '/': return cmd_search(arg); break;
	case 'd': return cmd_disasm(arg); break;
	case 'p': return cmd_print(arg); break;
	case 'r': return cmd_resize(arg); break;
	case 'x': return cmd_hexdump(arg); break;
	case 'X': return cmd_bytedump(arg); break;
	case 'w': return cmd_write(arg); break;
	case '!': return cmd_system(arg); break;
	case 'V': return cmd_system("vr0 $FILE"); break;
	case '?': return cmd_help(arg); break;
	default: fprintf(stderr, "? %s\n", cmd);
	}
	return 1;
}

static int r0_prompt() {
	char *at, *at2, line[BUFSZ];
	if(verbose) {
		printf("[0x%08"LLF"x]> ", curseek);
		fflush(stdout);
	}
	if(fgets(line, sizeof(line), stdin) == NULL)
		return 0;
	line[strlen(line)-1] = '\0';
	if(*line != '!') {
		at = strchr(line, '@');
		oldseek = curseek;
		obsize = bsize;
		if(at) {
			*at = 0;
			at2 = strchr(++at, ':');
			if(at2) {
				*at2 = 0; at2++;
				if(*at2) bsize = (int)calc(at2);
			}
			if(*at) curseek = calc(at);
		}
	}
	at = line;
	SKIPSPACES(at);
	return r0_cmd(at);
}

int r0_open(const char *file) {
	int ret = io_open(file);
	if(ret != -1) {
		oldseek = 0;
		setenv("FILE", file, 1);
		if(scripts)
			for (ret=0;ret<scriptn;ret++)
				r0_interpret(scripts[ret]);
		if(cmds)
			for (ret=0;ret<cmdn;ret++)
				r0_cmd(cmds[ret]);
		if(earlyquit)
			return 0;
		while((ret=r0_prompt())>0) {
			curseek = oldseek;
			bsize = obsize;
		}
		io_close();
	} else if (file)
		fprintf(stderr, "Cannot open '%s'\n", file);
	return ret==-1 ?1:0;
}

static int r0_help() {
	puts("r0 [-qhnv] [-c cmd] [-i script] [-|file ..]");
	return 0;
}

int main(int argc, char **argv) {
	int i, ret = 1;
	argc++;
	cmdn = 0;
	scriptn = 0;
	scripts = calloc(sizeof(const char*), argc);
	cmds = calloc(sizeof(const char*), argc);
	if(argc > 1 && argv[1]) {
		for(i = 1; i < argc; i++) {
			if(argv[i] && argv[i][0]=='-')
				switch(argv[i][1]) {
				case 'q': earlyquit = 1; break;
				case 'i': scripts[scriptn++] = argv[++i]; break;
				case 'c': cmds[cmdn++] = argv[++i]; break;
				case 'n': verbose = 0; break;
				case 'v': puts(VERSION); ret = 0; break;
				case 'h': ret = r0_help(); break;
				case 0x0: r0_slurpin(); ret = 0; break;
			} else {
				if (!argv[i]) break;
				ret = r0_open(argv[i]);
			}
		}
	} else {
		ret = r0_help();
	}
	free(scripts);
        return ret;
}