#include <elf.h>
#include <merror.h>
#include <stdlib.h>
#include <melf.h>

#include "link.h"

#define SYMTAB_INIT_LEN 8

int symtab_init(struct symbol_table *symtab)
{
	symtab->len = 1;
	symtab->size = SYMTAB_INIT_LEN;
	symtab->syms = malloc(sizeof(Elf32_Sym) * SYMTAB_INIT_LEN);

	if (symtab->syms == NULL) {
		PERROR("cannot alloc");
		return M_ERROR;
	}

	symtab->syms[0] = (Elf32_Sym){0};

	return M_SUCCESS;
}

void symtab_free(struct symbol_table *symtab)
{
	free(symtab->syms);
}

int symtab_push(struct symbol_table *symtab, const Elf32_Sym *sym)
{
	if (symtab->len >= symtab->size) {
		size_t size = symtab->size *= 2;
		void *new = realloc(symtab->syms, sizeof(Elf32_Sym) * size);
		if (new == NULL) {
			PERROR("cannot realloc");
			return M_ERROR;
		}
		symtab->size = size;
		symtab->syms = new;
	}

	symtab->syms[symtab->len++] = *sym;
	return M_SUCCESS;
}

int symtab_get(struct symbol_table *symtab, Elf32_Sym **res, const char *name)
{
	for (size_t i = 0; i < symtab->len; i++) {
		Elf32_Sym *sym = &symtab->syms[i];
		const char *symname = symtab->strtab->data + B32(sym->st_name);
		if (strcmp(name, symname) == 0) {
			if (res != NULL)
				*res = sym;
			return M_SUCCESS;
		}
	}
	return M_ERROR;
}