From b663f827057fc9fb199293bc1920cf27315d1846 Mon Sep 17 00:00:00 2001
From: Freya Murphy <freya@freyacat.org>
Date: Wed, 9 Oct 2024 12:07:59 -0400
Subject: refactor elf32 assembler, add support for multiple isa's in cmdline

---
 masm/asm/strtab.c | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 masm/asm/strtab.c

(limited to 'masm/asm/strtab.c')

diff --git a/masm/asm/strtab.c b/masm/asm/strtab.c
new file mode 100644
index 0000000..799f0dc
--- /dev/null
+++ b/masm/asm/strtab.c
@@ -0,0 +1,54 @@
+#include <merror.h>
+#include <string.h>
+#include <stdlib.h>
+
+#include "elf32.h"
+
+int strtab_get_str(struct elf_str_table *strtab, const char *str, size_t *res)
+{
+	for (size_t i = 0; i < strtab->size; i ++) {
+		if (strcmp(strtab->ptr + i, str) == 0) {
+			if (res != NULL)
+				*res = i;
+			return M_SUCCESS;
+		}
+	}
+
+	return M_ERROR;
+}
+
+int strtab_write_str(struct elf_str_table *strtab, const char *str, size_t *res)
+{
+	if (strtab_get_str(strtab, str, res) == M_SUCCESS)
+		return M_SUCCESS;
+
+	size_t len = strlen(str);
+	char *new = realloc(strtab->ptr, strtab->size + len + 1);
+	if (new == NULL)
+		return M_ERROR;
+	strtab->ptr = new;
+	memcpy(strtab->ptr + strtab->size, str, len + 1);
+
+	if (res != NULL)
+		*res = strtab->size;
+
+	strtab->size += len + 1;
+	return M_SUCCESS;
+}
+
+int strtab_init(struct elf_str_table *strtab)
+{
+	strtab->size = 1;
+	strtab->ptr = malloc(1);
+	if (strtab->ptr == NULL) {
+		PERROR("cannot alloc");
+		return M_ERROR;
+	}
+	*strtab->ptr = '\0';
+	return M_SUCCESS;
+}
+
+void strtab_free(struct elf_str_table *strtab)
+{
+	free(strtab->ptr);
+}
-- 
cgit v1.2.3-freya