refactor

1 files changed, 43 insertions, 0 deletions

diff --git a/masm/reltab.c b/masm/reltab.c
new file mode 100644
index 0000000..482ed44
--- /dev/null
+++ b/masm/reltab.c
@@ -0,0 +1,43 @@
+#include <elf.h>
+#include <stdlib.h>
+#include <merror.h>
+
+#include "asm.h"
+
+#define RELTAB_INIT_LEN 8
+
+int reltab_init(struct relocation_table *reltab)
+{
+	reltab->size = RELTAB_INIT_LEN;
+	reltab->len = 0;
+	reltab->data = malloc(sizeof(Elf32_Rela) * RELTAB_INIT_LEN);
+
+	if (reltab->data == NULL) {
+		ERROR("cannot alloc");
+		return M_ERROR;
+	}
+
+	return M_SUCCESS;
+}
+
+void reltab_free(struct relocation_table *reltab)
+{
+	free(reltab->data);
+}
+
+int reltab_push(struct relocation_table *reltab, const Elf32_Rela rel)
+{
+	if (reltab->len >= reltab->size) {
+		reltab->size *= 2;
+		reltab->data = realloc(reltab->data, sizeof(Elf32_Rela)
+			 * reltab->size);
+
+		if (reltab->data == NULL) {
+			ERROR("cannot realloc");
+			return M_ERROR;
+		}
+	}
+
+	reltab->data[reltab->len++] = rel;
+	return M_SUCCESS;
+}