From 88ed6d1ad9113af2eac5811c3bedaddcaea82a0f Mon Sep 17 00:00:00 2001 From: gbrochar Date: Mon, 28 Oct 2024 13:56:33 +0100 Subject: [PATCH] fix bugs for correc --- src/main.c | 8 ++- src/nm.c | 175 +++++++++++++++++++++++++++++------------------------ src/nm32.c | 175 +++++++++++++++++++++++++++++------------------------ 3 files changed, 198 insertions(+), 160 deletions(-) diff --git a/src/main.c b/src/main.c index 326343c..a794e61 100644 --- a/src/main.c +++ b/src/main.c @@ -98,6 +98,7 @@ int nm(char *path, t_verbosity verbosity, t_ordering ordering) { if (!ident_ptr) { ft_printf("sss", "nm: ", path, ": file format not recognized\n"); munmap(mapped_file.ptr, mapped_file.size); + close(fd); return 0; } @@ -105,12 +106,14 @@ int nm(char *path, t_verbosity verbosity, t_ordering ordering) { if (check_ident(ident) == FT_NM_FAILURE) { ft_printf("sss", "nm: ", path, ": file format not recognized\n"); munmap(mapped_file.ptr, mapped_file.size); + close(fd); return 0; } if (ident[EI_CLASS] == ELFCLASS32) { if (nm32(mapped_file, path, verbosity, ordering) == FT_NM_FAILURE) { ft_printf("sss", "nm: ", path, ": file format not recognized\n"); munmap(mapped_file.ptr, mapped_file.size); + close(fd); return 0; } } @@ -118,6 +121,7 @@ int nm(char *path, t_verbosity verbosity, t_ordering ordering) { if (nm64(mapped_file, path, verbosity, ordering) == FT_NM_FAILURE) { ft_printf("sss", "nm: ", path, ": file format not recognized\n"); munmap(mapped_file.ptr, mapped_file.size); + close(fd); return 0; } } @@ -158,8 +162,8 @@ int main(int ac, char **av) { } t_list *tmp = env.list; while (tmp) { - ft_printf("\ns:\n", env.list->content); - nm(env.list->content, env.verbosity, env.ordering); + ft_printf("\ns:\n", tmp->content); + nm(tmp->content, env.verbosity, env.ordering); tmp = tmp->next; } free_env(env); diff --git a/src/nm.c b/src/nm.c index c4924d8..8fa6b23 100644 --- a/src/nm.c +++ b/src/nm.c @@ -1,122 +1,122 @@ #include "ft_nm.h" -char *get_sym_char(Elf64_Sym sym, Elf64_Shdr sec, int *tool) { +char *get_sym_char(Elf64_Sym *sym, Elf64_Shdr *sec, int *tool) { *tool = 0; - if (ELF64_ST_TYPE(sym.st_info) == STT_SECTION) { + if (ELF64_ST_TYPE(sym->st_info) == STT_SECTION) { *tool = 1; } - if (sym.st_shndx == SHN_ABS) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sym->st_shndx == SHN_ABS) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("a"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("A"); } } - if (ELF64_ST_BIND(sym.st_info) == STB_GNU_UNIQUE) { + if (ELF64_ST_BIND(sym->st_info) == STB_GNU_UNIQUE) { return ("u"); } - if (ELF64_ST_TYPE(sym.st_info) == STT_GNU_IFUNC) { + if (ELF64_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) { return ("i"); } - if (sec.sh_flags & SHF_EXECINSTR) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_flags & SHF_EXECINSTR) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("t"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("T"); } } - if (sec.sh_type == SHT_NOBITS) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_NOBITS) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("b"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("B"); } } - if (sec.sh_type == SHT_PROGBITS && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_PROGBITS && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_DYNAMIC && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_DYNAMIC && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_INIT_ARRAY && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_INIT_ARRAY && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_PREINIT_ARRAY && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_PREINIT_ARRAY && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_FINI_ARRAY && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_FINI_ARRAY && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_flags & SHF_ALLOC) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_flags & SHF_ALLOC) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("r"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("R"); } } - if (sec.sh_type == SHT_NOTE && sec.sh_flags == SHF_ALLOC) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_NOTE && sec->sh_flags == SHF_ALLOC) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("r"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("R"); } } - if (sym.st_shndx == SHN_COMMON) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sym->st_shndx == SHN_COMMON) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("c"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("C"); } } - if (sym.st_shndx == SHN_UNDEF) { - if (ELF64_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sym->st_shndx == SHN_UNDEF) { + if (ELF64_ST_BIND(sym->st_info) == STB_LOCAL) { return ("u"); } - else if (ELF64_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF64_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("U"); } } - if (ELF64_ST_BIND(sym.st_info) == STB_WEAK) { - if (ELF64_ST_TYPE(sym.st_info) == STT_OBJECT) { - if (sec.sh_type == SHT_NULL) { + if (ELF64_ST_BIND(sym->st_info) == STB_WEAK) { + if (ELF64_ST_TYPE(sym->st_info) == STT_OBJECT) { + if (sec->sh_type == SHT_NULL) { return ("v"); } else { return ("V"); } } else { - if (sec.sh_type == SHT_NULL) { + if (sec->sh_type == SHT_NULL) { return ("w"); } else { @@ -124,8 +124,8 @@ char *get_sym_char(Elf64_Sym sym, Elf64_Shdr sec, int *tool) { } } } - if (ELF64_ST_BIND(sym.st_info) == STB_WEAK) { - if (sec.sh_type == SHT_NULL) { + if (ELF64_ST_BIND(sym->st_info) == STB_WEAK) { + if (sec->sh_type == SHT_NULL) { return ("w"); } else { @@ -136,49 +136,67 @@ char *get_sym_char(Elf64_Sym sym, Elf64_Shdr sec, int *tool) { } int nm64(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_ordering ordering) { - t_root *tree = NULL; - t_entry *entry = (t_entry *)malloc(sizeof(t_entry)); - int tool; - if (!entry) - return FT_NM_FAILURE; Elf64_Ehdr header; if (get_header64(mapped_file, &header) == FT_NM_FAILURE) { return FT_NM_FAILURE; } uint64_t addr = header.e_shoff + header.e_shentsize * header.e_shstrndx; - Elf64_Shdr shstrtb; - shstrtb = *(Elf64_Shdr *)fetch(mapped_file, addr, header.e_shentsize); + Elf64_Shdr *shstrtb; + if (!(shstrtb = (Elf64_Shdr *)fetch(mapped_file, addr, header.e_shentsize))) { + return FT_NM_FAILURE; + } + t_root *tree = NULL; + t_entry *entry = (t_entry *)malloc(sizeof(t_entry)); + if (!entry) + return FT_NM_FAILURE; + int tool; for (int i = 0; i < header.e_shnum; i++) { uint64_t addr = header.e_shoff + header.e_shentsize * i; - Elf64_Shdr sh; - sh = *(Elf64_Shdr *)fetch(mapped_file, addr, header.e_shentsize); - if (sh.sh_type == SHT_SYMTAB) { - uint64_t addr2 = header.e_shoff + header.e_shentsize * sh.sh_link; - Elf64_Shdr strtab; - strtab = *(Elf64_Shdr *)fetch(mapped_file, addr2, header.e_shentsize); - char first = *(char *)fetch(mapped_file, strtab.sh_offset, 1); - char last = *(char *)fetch(mapped_file, strtab.sh_offset + strtab.sh_size - 1, 1); - if (first != '\0' || last != '\0') { + Elf64_Shdr *sh; + if (!(sh = (Elf64_Shdr *)fetch(mapped_file, addr, header.e_shentsize))) { + free(entry); + return FT_NM_FAILURE; + } + if (sh->sh_type == SHT_SYMTAB) { + uint64_t addr2 = header.e_shoff + header.e_shentsize * sh->sh_link; + Elf64_Shdr *strtab; + if (!(strtab = (Elf64_Shdr *)fetch(mapped_file, addr2, header.e_shentsize))) { + free(entry); return FT_NM_FAILURE; } - for (uint64_t j = sh.sh_entsize; j < sh.sh_size; j += sh.sh_entsize) { - Elf64_Sym sym; - Elf64_Shdr sec; - sym = *(Elf64_Sym *)fetch(mapped_file, sh.sh_offset + j, sh.sh_entsize); - if (sym.st_shndx < header.e_shnum) { - sec = *(Elf64_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym.st_shndx, header.e_shentsize); + char *first = (char *)fetch(mapped_file, strtab->sh_offset, 1); + char *last = (char *)fetch(mapped_file, strtab->sh_offset + strtab->sh_size - 1, 1); + if (!first || !last || *first != '\0' || *last != '\0') { + free(entry); + return FT_NM_FAILURE; + } + for (uint64_t j = sh->sh_entsize; j < sh->sh_size; j += sh->sh_entsize) { + Elf64_Sym *sym; + Elf64_Shdr *sec; + if (!(sym = (Elf64_Sym *)fetch(mapped_file, sh->sh_offset + j, sh->sh_entsize))) { + free(entry); + return FT_NM_FAILURE; + } + if (sym->st_shndx < header.e_shnum) { + if (!(sec = (Elf64_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym->st_shndx, header.e_shentsize))) { + free(entry); + return FT_NM_FAILURE; + } } char *str; char *sec_str; - Elf64_Shdr shdr; - if (sym.st_shndx != SHN_ABS) { - shdr = *(Elf64_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym.st_shndx, header.e_shentsize); - sec_str = ft_strdup(mapped_file.ptr + shstrtb.sh_offset + shdr.sh_name); + Elf64_Shdr *shdr; + if (sym->st_shndx != SHN_ABS) { + if (!(shdr = (Elf64_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym->st_shndx, header.e_shentsize))) { + free(entry); + return FT_NM_FAILURE; + } + sec_str = ft_strdup(mapped_file.ptr + shstrtb->sh_offset + shdr->sh_name); } else { sec_str = ft_strdup(""); } - if (sym.st_name) { - str = ft_strdup(mapped_file.ptr + strtab.sh_offset + sym.st_name); + if (sym->st_name) { + str = ft_strdup(mapped_file.ptr + strtab->sh_offset + sym->st_name); } else { str = sec_str; @@ -197,7 +215,7 @@ int nm64(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_orderin // ici le sym char a ete calculer, a partir de la on determine la verbosity // ici on fait if entry->verbosity >= verbosity, si c'est pas bon on skip toute la suite entry->verbosity = DEFAULT_VERBOSITY; - if ((sym_char[0] == 'a' && sym.st_info == 4) || sym_char[0] == 'N' || tool == 1) + if ((sym_char[0] == 'a' && sym->st_info == 4) || sym_char[0] == 'N' || tool == 1) entry->verbosity = ALL; if (sym_char[0] == 'u' || sym_char[0] == 'v' || sym_char[0] == 'w' || (sym_char[0] >= 'A' && sym_char[0] <= 'Z' && sym_char[0] != 'N')) entry->verbosity = GLOBAL; @@ -216,15 +234,15 @@ int nm64(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_orderin entry->string[17] = sym_char[0]; entry->string[18] = ' '; ft_memcpy(entry->string + 19, entry->symbol, ft_strlen(entry->symbol)); - if (sym.st_value && !ft_strstr(entry->symbol, "@@")) { + if (sym->st_value && !ft_strstr(entry->symbol, "@@")) { for (char i = 15; i >= 0; i--) { - entry->string[15 - i] = ft_get_hex_digit((sym.st_value >> i * 4) & 0xF); + entry->string[15 - i] = ft_get_hex_digit((sym->st_value >> i * 4) & 0xF); } } else { if (ft_strcmp(sym_char, "U") && ft_strcmp(sym_char, "w") && ft_strcmp(sym_char, "v")) { for (char i = 15; i >= 0; i--) { - entry->string[15 - i] = ft_get_hex_digit((sym.st_value >> i * 4) & 0xF); + entry->string[15 - i] = ft_get_hex_digit((sym->st_value >> i * 4) & 0xF); } } else { for (int i = 0; i < 16; i++) @@ -244,7 +262,6 @@ int nm64(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_orderin ft_rbt_insert(&tree, ft_rbt_new((void *)entry, sizeof(t_entry)), &strcmp_nm); break; } - fflush(stdout); } else { free(entry->symbol); free(sym_char); diff --git a/src/nm32.c b/src/nm32.c index a5a9f9d..5b9162b 100644 --- a/src/nm32.c +++ b/src/nm32.c @@ -1,122 +1,122 @@ #include "ft_nm.h" -char *get_sym_char32(Elf32_Sym sym, Elf32_Shdr sec, int *tool) { +char *get_sym_char32(Elf32_Sym *sym, Elf32_Shdr *sec, int *tool) { *tool = 0; - if (ELF32_ST_TYPE(sym.st_info) == STT_SECTION) { + if (ELF32_ST_TYPE(sym->st_info) == STT_SECTION) { *tool = 1; } - if (sym.st_shndx == SHN_ABS) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sym->st_shndx == SHN_ABS) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("a"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("A"); } } - if (ELF32_ST_BIND(sym.st_info) == STB_GNU_UNIQUE) { + if (ELF32_ST_BIND(sym->st_info) == STB_GNU_UNIQUE) { return ("u"); } - if (ELF32_ST_TYPE(sym.st_info) == STT_GNU_IFUNC) { + if (ELF32_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) { return ("i"); } - if (sec.sh_flags & SHF_EXECINSTR) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_flags & SHF_EXECINSTR) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("t"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("T"); } } - if (sec.sh_type == SHT_NOBITS) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_NOBITS) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("b"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("B"); } } - if (sec.sh_type == SHT_PROGBITS && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_PROGBITS && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_DYNAMIC && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_DYNAMIC && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_INIT_ARRAY && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_INIT_ARRAY && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_PREINIT_ARRAY && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_PREINIT_ARRAY && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_type == SHT_FINI_ARRAY && (sec.sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_FINI_ARRAY && (sec->sh_flags & (SHF_ALLOC | SHF_WRITE)) == (SHF_ALLOC | SHF_WRITE)) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("d"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("D"); } } - if (sec.sh_flags & SHF_ALLOC) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_flags & SHF_ALLOC) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("r"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("R"); } } - if (sec.sh_type == SHT_NOTE && sec.sh_flags == SHF_ALLOC) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sec->sh_type == SHT_NOTE && sec->sh_flags == SHF_ALLOC) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("r"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("R"); } } - if (sym.st_shndx == SHN_COMMON) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sym->st_shndx == SHN_COMMON) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("c"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("C"); } } - if (sym.st_shndx == SHN_UNDEF) { - if (ELF32_ST_BIND(sym.st_info) == STB_LOCAL) { + if (sym->st_shndx == SHN_UNDEF) { + if (ELF32_ST_BIND(sym->st_info) == STB_LOCAL) { return ("u"); } - else if (ELF32_ST_BIND(sym.st_info) == STB_GLOBAL) { + else if (ELF32_ST_BIND(sym->st_info) == STB_GLOBAL) { return ("U"); } } - if (ELF32_ST_BIND(sym.st_info) == STB_WEAK) { - if (ELF32_ST_TYPE(sym.st_info) == STT_OBJECT) { - if (sec.sh_type == SHT_NULL) { + if (ELF32_ST_BIND(sym->st_info) == STB_WEAK) { + if (ELF32_ST_TYPE(sym->st_info) == STT_OBJECT) { + if (sec->sh_type == SHT_NULL) { return ("v"); } else { return ("V"); } } else { - if (sec.sh_type == SHT_NULL) { + if (sec->sh_type == SHT_NULL) { return ("w"); } else { @@ -124,8 +124,8 @@ char *get_sym_char32(Elf32_Sym sym, Elf32_Shdr sec, int *tool) { } } } - if (ELF32_ST_BIND(sym.st_info) == STB_WEAK) { - if (sec.sh_type == SHT_NULL) { + if (ELF32_ST_BIND(sym->st_info) == STB_WEAK) { + if (sec->sh_type == SHT_NULL) { return ("w"); } else { @@ -136,49 +136,67 @@ char *get_sym_char32(Elf32_Sym sym, Elf32_Shdr sec, int *tool) { } int nm32(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_ordering ordering) { - t_root *tree = NULL; - t_entry *entry = (t_entry *)malloc(sizeof(t_entry)); - int tool; - if (!entry) - return FT_NM_FAILURE; Elf32_Ehdr header; if (get_header32(mapped_file, &header) == FT_NM_FAILURE) { return FT_NM_FAILURE; } uint32_t addr = header.e_shoff + header.e_shentsize * header.e_shstrndx; - Elf32_Shdr shstrtb; - shstrtb = *(Elf32_Shdr *)fetch(mapped_file, addr, header.e_shentsize); + Elf32_Shdr *shstrtb; + if (!(shstrtb = (Elf32_Shdr *)fetch(mapped_file, addr, header.e_shentsize))) { + return FT_NM_FAILURE; + } + t_root *tree = NULL; + t_entry *entry = (t_entry *)malloc(sizeof(t_entry)); + if (!entry) + return FT_NM_FAILURE; + int tool; for (int i = 0; i < header.e_shnum; i++) { uint32_t addr = header.e_shoff + header.e_shentsize * i; - Elf32_Shdr sh; - sh = *(Elf32_Shdr *)fetch(mapped_file, addr, header.e_shentsize); - if (sh.sh_type == SHT_SYMTAB) { - uint32_t addr2 = header.e_shoff + header.e_shentsize * sh.sh_link; - Elf32_Shdr strtab; - strtab = *(Elf32_Shdr *)fetch(mapped_file, addr2, header.e_shentsize); - char first = *(char *)fetch(mapped_file, strtab.sh_offset, 1); - char last = *(char *)fetch(mapped_file, strtab.sh_offset + strtab.sh_size - 1, 1); - if (first != '\0' || last != '\0') { + Elf32_Shdr *sh; + if (!(sh = (Elf32_Shdr *)fetch(mapped_file, addr, header.e_shentsize))) { + free(entry); + return FT_NM_FAILURE; + } + if (sh->sh_type == SHT_SYMTAB) { + uint32_t addr2 = header.e_shoff + header.e_shentsize * sh->sh_link; + Elf32_Shdr *strtab; + if (!(strtab = (Elf32_Shdr *)fetch(mapped_file, addr2, header.e_shentsize))) { + free(entry); return FT_NM_FAILURE; } - for (uint64_t j = sh.sh_entsize; j < sh.sh_size; j += sh.sh_entsize) { - Elf32_Sym sym; - Elf32_Shdr sec; - sym = *(Elf32_Sym *)fetch(mapped_file, sh.sh_offset + j, sh.sh_entsize); - if (sym.st_shndx < header.e_shnum) { - sec = *(Elf32_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym.st_shndx, header.e_shentsize); + char *first = (char *)fetch(mapped_file, strtab->sh_offset, 1); + char *last = (char *)fetch(mapped_file, strtab->sh_offset + strtab->sh_size - 1, 1); + if (!first || !last || *first != '\0' || *last != '\0') { + free(entry); + return FT_NM_FAILURE; + } + for (uint32_t j = sh->sh_entsize; j < sh->sh_size; j += sh->sh_entsize) { + Elf32_Sym *sym; + Elf32_Shdr *sec; + if (!(sym = (Elf32_Sym *)fetch(mapped_file, sh->sh_offset + j, sh->sh_entsize))) { + free(entry); + return FT_NM_FAILURE; + } + if (sym->st_shndx < header.e_shnum) { + if (!(sec = (Elf32_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym->st_shndx, header.e_shentsize))) { + free(entry); + return FT_NM_FAILURE; + } } char *str; char *sec_str; - Elf32_Shdr shdr; - if (sym.st_shndx != SHN_ABS) { - shdr = *(Elf32_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym.st_shndx, header.e_shentsize); - sec_str = ft_strdup(mapped_file.ptr + shstrtb.sh_offset + shdr.sh_name); + Elf32_Shdr *shdr; + if (sym->st_shndx != SHN_ABS) { + if (!(shdr = (Elf32_Shdr *)fetch(mapped_file, header.e_shoff + header.e_shentsize * sym->st_shndx, header.e_shentsize))) { + free(entry); + return FT_NM_FAILURE; + } + sec_str = ft_strdup(mapped_file.ptr + shstrtb->sh_offset + shdr->sh_name); } else { sec_str = ft_strdup(""); } - if (sym.st_name) { - str = ft_strdup(mapped_file.ptr + strtab.sh_offset + sym.st_name); + if (sym->st_name) { + str = ft_strdup(mapped_file.ptr + strtab->sh_offset + sym->st_name); } else { str = sec_str; @@ -197,7 +215,7 @@ int nm32(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_orderin // ici le sym char a ete calculer, a partir de la on determine la verbosity // ici on fait if entry->verbosity >= verbosity, si c'est pas bon on skip toute la suite entry->verbosity = DEFAULT_VERBOSITY; - if ((sym_char[0] == 'a' && sym.st_info == 4) || sym_char[0] == 'N' || tool == 1) + if ((sym_char[0] == 'a' && sym->st_info == 4) || sym_char[0] == 'N' || tool == 1) entry->verbosity = ALL; if (sym_char[0] == 'u' || sym_char[0] == 'v' || sym_char[0] == 'w' || (sym_char[0] >= 'A' && sym_char[0] <= 'Z' && sym_char[0] != 'N')) entry->verbosity = GLOBAL; @@ -216,15 +234,15 @@ int nm32(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_orderin entry->string[9] = sym_char[0]; entry->string[10] = ' '; ft_memcpy(entry->string + 11, entry->symbol, ft_strlen(entry->symbol)); - if (sym.st_value && !ft_strstr(entry->symbol, "@@")) { + if (sym->st_value && !ft_strstr(entry->symbol, "@@")) { for (char i = 7; i >= 0; i--) { - entry->string[7 - i] = ft_get_hex_digit((sym.st_value >> i * 4) & 0xF); + entry->string[7 - i] = ft_get_hex_digit((sym->st_value >> i * 4) & 0xF); } } else { if (ft_strcmp(sym_char, "U") && ft_strcmp(sym_char, "w") && ft_strcmp(sym_char, "v")) { for (char i = 7; i >= 0; i--) { - entry->string[7 - i] = ft_get_hex_digit((sym.st_value >> i * 4) & 0xF); + entry->string[7 - i] = ft_get_hex_digit((sym->st_value >> i * 4) & 0xF); } } else { for (int i = 0; i < 8; i++) @@ -244,7 +262,6 @@ int nm32(t_mapped_file mapped_file, char *path, t_verbosity verbosity, t_orderin ft_rbt_insert(&tree, ft_rbt_new((void *)entry, sizeof(t_entry)), &strcmp_nm); break; } - fflush(stdout); } else { free(entry->symbol); free(sym_char);