157 lines
3.5 KiB
C
157 lines
3.5 KiB
C
#include <stdint.h>
|
|
#include <stddef.h>
|
|
#include <ulib.h>
|
|
#include "interp.h"
|
|
#include "macros.h"
|
|
|
|
#define LINEBUF_MAX 1024
|
|
|
|
uint64_t PID;
|
|
|
|
struct {
|
|
char *modestr;
|
|
enum { MODE_INTERACTIVE = 1, MODE_RUNFILE = 2 } mode;
|
|
|
|
char *filepath;
|
|
|
|
bool logcmds;
|
|
|
|
char *preloadpath;
|
|
} CONFIG;
|
|
|
|
static Arg ARGS[] = {
|
|
ARG("-m", ARG_STRING, &CONFIG.modestr),
|
|
ARG("-f", ARG_STRING, &CONFIG.filepath),
|
|
ARG("-logcmds", ARG_BOOL, &CONFIG.logcmds),
|
|
ARG("-preload", ARG_STRING, &CONFIG.preloadpath),
|
|
ARG_END(),
|
|
};
|
|
|
|
void do_file(char *filepath);
|
|
|
|
void set_config(void) {
|
|
int32_t ret;
|
|
if ((ret = parse_args(args(), argslen(), ARGS)) < 0) {
|
|
uprintf("Could not parse args: %d\n", ret);
|
|
}
|
|
|
|
if (CONFIG.modestr != NULL) {
|
|
if (string_strcmp(CONFIG.modestr, "interactive") == 0) {
|
|
CONFIG.mode = MODE_INTERACTIVE;
|
|
} else if (string_strcmp(CONFIG.modestr, "runfile") == 0) {
|
|
CONFIG.mode = MODE_RUNFILE;
|
|
} else {
|
|
LOG(LOG_ERR, "Unknown mode %s\n", CONFIG.modestr);
|
|
}
|
|
} else {
|
|
CONFIG.mode = MODE_RUNFILE;
|
|
}
|
|
}
|
|
|
|
void do_file(char *filepath) {
|
|
int32_t ret;
|
|
|
|
int32_t ioh = ioctl(IOCTL_NOHANDLE, IOCTL_OPENF, (uint64_t)filepath, IOCTL_F_READ, 0);
|
|
if (ioh < 0) {
|
|
LOG(LOG_ERR, "Could not open %s: %s\n", filepath, ERRSTRING(ioh));
|
|
return;
|
|
}
|
|
|
|
IoctlStat statbuf = {0};
|
|
|
|
ioctl(ioh, IOCTL_STAT, (uint64_t)&statbuf, 0, 0);
|
|
if (statbuf.type != IOCTLSTAT_FILE) {
|
|
LOG(LOG_ERR, "%s is not a file (%d)\n", filepath, statbuf.type);
|
|
return;
|
|
}
|
|
uint8_t *buf = dlmalloc(statbuf.size+1);
|
|
string_memset(buf, 0, statbuf.size+1);
|
|
|
|
if ((ret = ioctl(ioh, IOCTL_READ, (uint64_t)buf, statbuf.size, 0)) < 0) {
|
|
LOG(LOG_ERR, "Could not read %s (%d): %s\n", filepath, ioh, ERRSTRING(ioh));
|
|
goto done;
|
|
}
|
|
|
|
InterpResult *res;
|
|
bool ok = interp_runstring((const char *)buf, &res, CONFIG.logcmds, CONFIG.mode == MODE_INTERACTIVE);
|
|
if (!ok) {
|
|
uprintf("Interpreter error:\n");
|
|
uprintf("%s\n", res->errmsg);
|
|
goto done;
|
|
}
|
|
|
|
done:
|
|
dlfree(buf);
|
|
ioctl(ioh, IOCTL_CLOSEF, 0, 0, 0);
|
|
}
|
|
|
|
void do_mode_interactive(void) {
|
|
char linebuf[LINEBUF_MAX];
|
|
size_t cursor;
|
|
for(;;) {
|
|
begin:
|
|
uprintf("tb# ");
|
|
|
|
cursor = 0;
|
|
string_memset(linebuf, 0, LINEBUF_MAX);
|
|
|
|
uint8_t b = 0;
|
|
for (;;) {
|
|
int32_t nrd = ipcpipe(PID, IPCPIPE_IN, IPCPIPE_READ, &b, 1);
|
|
if (nrd > 0) {
|
|
switch (b) {
|
|
case C('C'):
|
|
case 0xE9:
|
|
uprintf("\n");
|
|
goto begin;
|
|
break;
|
|
case C('L'):
|
|
uprintf(ANSIQ_CUR_SET(0, 0));
|
|
uprintf(ANSIQ_SCR_CLR_ALL);
|
|
goto begin;
|
|
break;
|
|
}
|
|
|
|
if (b == '\n') {
|
|
break;
|
|
}
|
|
|
|
if (string_chr_isascii(b) && b != 0 && cursor < LINEBUF_MAX) {
|
|
linebuf[cursor++] = b;
|
|
uprintf("%c", b);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (cursor < LINEBUF_MAX) {
|
|
linebuf[cursor] = '\0';
|
|
}
|
|
uprintf("\n");
|
|
InterpResult *res;
|
|
if (!interp_runstring(linebuf, &res, CONFIG.logcmds, CONFIG.mode == MODE_INTERACTIVE)) {
|
|
LOG(LOG_ERR, "%s\n", res->errmsg);
|
|
}
|
|
}
|
|
}
|
|
|
|
void main(void) {
|
|
PID = processctl(-1, PCTL_GETPID, 0, 0, 0);
|
|
|
|
set_config();
|
|
|
|
if (CONFIG.preloadpath != NULL) {
|
|
LOG(LOG_INF, "Preloading script: %s\n", CONFIG.preloadpath);
|
|
do_file(CONFIG.preloadpath);
|
|
}
|
|
|
|
if (CONFIG.mode == MODE_INTERACTIVE) {
|
|
do_mode_interactive();
|
|
} else if (CONFIG.mode == MODE_RUNFILE) {
|
|
if (CONFIG.filepath == NULL) {
|
|
uprintf("No file provided\n");
|
|
return;
|
|
}
|
|
do_file(CONFIG.filepath);
|
|
}
|
|
}
|