path: root/test/elf_parser.cc

#include "elf_parser.h"

#include <cstddef>
#include <cstdio>

#include <elf.h>

#include "elf_parser/libtest-32.h"
#include "elf_parser/libtest-64.h"

void dump_dynsyms(const unsigned char* bytes, size_t len) {
  auto elf = elf::parse(bytes, len);

  elf->dynsyms([](const char* name, char type, char bind) {  // NOLINT
    const char* bind_str = "<unnown>";
    switch (bind) {
      case STB_GLOBAL:
        bind_str = "GLOBAL";
        break;
      case STB_LOCAL:
        bind_str = "LOCAL";
        break;
      case STB_WEAK:
        bind_str = "WEAK";
        break;
      default:
        break;
    }

    const char* type_str = "<unknown>";
    switch (type) {
      case STT_NOTYPE:
        type_str = "NOTYPE";
        break;
      case STT_FUNC:
        type_str = "FUNC";
        break;
      case STT_OBJECT:
        type_str = "OBJECT";
        break;
      case STT_SECTION:
        type_str = "SECTION";
        break;
      case STT_FILE:
        type_str = "FILE";
        break;
      default:
        break;
    }

    printf("syms type: %10s bind: %10s name: %s\n", type_str, bind_str, name);
    return true;
  });
}

int main() {
  dump_dynsyms(libtest_32_so, libtest_32_so_len);
  dump_dynsyms(libtest_64_so, libtest_64_so_len);
}