117 lines
2.5 KiB
C
117 lines
2.5 KiB
C
#include "sdb.h"
|
|
#include "vector.h"
|
|
#include "logger.h"
|
|
|
|
#include <stddef.h>
|
|
#include <stdbool.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <linux/limits.h>
|
|
|
|
static inline enum instruction prompt(void)
|
|
{
|
|
static char *input;
|
|
static size_t input_len;
|
|
|
|
OUTPUT("(sdb) ");
|
|
getline(&input, &input_len, stdin);
|
|
|
|
char *op = strtok(input, " \t\n");
|
|
if (op == NULL) return INST_NOP;
|
|
|
|
if (strcmp(op, "exit") == 0) return INST_EXIT;
|
|
if (strcmp(op, "load") == 0) return INST_LOAD;
|
|
if (strcmp(op, "show") == 0) return INST_SHOW;
|
|
if (strcmp(op, "info") == 0) return INST_INFO;
|
|
if (strcmp(op, "break") == 0) return INST_BREAK;
|
|
if (strcmp(op, "delete") == 0) return INST_DELETE;
|
|
if (strcmp(op, "patch") == 0) return INST_PATCH;
|
|
if (strcmp(op, "si") == 0) return INST_SI;
|
|
if (strcmp(op, "cont") == 0) return INST_CONT;
|
|
if (strcmp(op, "syscall") == 0) return INST_SYSCALL;
|
|
|
|
return INST_INVALID;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
setvbuf(stdin, NULL, _IONBF, 0);
|
|
setvbuf(stdout, NULL, _IONBF, 0);
|
|
setvbuf(stderr, NULL, _IONBF, 0);
|
|
|
|
char *filename = "";
|
|
if (argc > 1) {
|
|
filename = malloc(strlen(argv[1]) + 1);
|
|
strcpy(filename, argv[1]);
|
|
if (access(filename, F_OK) != 0) {
|
|
perror("main file open");
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
if (access(filename, F_OK) != 0) {
|
|
load_retry:
|
|
enum instruction op = prompt();
|
|
|
|
switch (op) {
|
|
case INST_NOP:
|
|
goto load_retry;
|
|
case INST_EXIT:
|
|
goto exit;
|
|
|
|
case INST_LOAD:
|
|
inst_load(&filename);
|
|
if (access(filename, F_OK) == 0)
|
|
break;
|
|
else {
|
|
INFO("file invalid\n");
|
|
goto load_retry;
|
|
}
|
|
|
|
default:
|
|
INFO("please load a program first.\n");
|
|
goto load_retry;
|
|
}
|
|
}
|
|
|
|
DEBUG("filename = %s\n", filename);
|
|
run(filename);
|
|
|
|
sync_regs();
|
|
disassemble_addr = regs.rip;
|
|
disassemble();
|
|
|
|
for (;;) {
|
|
enum instruction op = prompt();
|
|
DEBUG("op = 0x%x\n", (int)op);
|
|
|
|
switch (op) {
|
|
case INST_NOP: break;
|
|
case INST_EXIT: goto exit;
|
|
case INST_SHOW: disassemble(); break;
|
|
case INST_INFO: inst_info(); break;
|
|
|
|
case INST_BREAK: inst_break(); break;
|
|
case INST_DELETE: inst_delete(); break;
|
|
case INST_PATCH: inst_patch(); break;
|
|
|
|
case INST_SI: inst_si(); disassemble(); break;
|
|
case INST_CONT: inst_cont(); disassemble(); break;
|
|
case INST_SYSCALL: inst_syscall(); disassemble(); break;
|
|
|
|
default:
|
|
INFO("invalid command\n");
|
|
}
|
|
|
|
DEBUG("disassemble_addr = %p\n", (void *)disassemble_addr);
|
|
if (disassemble_addr == 0x00)
|
|
goto exit;
|
|
}
|
|
|
|
exit:
|
|
free(filename);
|
|
free(bps);
|
|
return 0;
|
|
}
|