Initial commit

2025-04-12 08:26:23 +08:00
commit aa66855054
57 changed files with 1702 additions and 0 deletions
--- a/main.c
+++ b/main.c
@@ -0,0 +1,116 @@
+#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;
+}