X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Flttng-elf.c;h=b2aa88487e1dff286c21d0442f6cdf175a68681f;hp=47dc89900237f4e88aa537ec8a8d6be548c95998;hb=22fae25a70cc5e87ab76e5e2071c88c0c181e2c3;hpb=8bd52288b87c954916e836066241790f23e9d5e6 diff --git a/src/common/lttng-elf.c b/src/common/lttng-elf.c index 47dc89900..b2aa88487 100644 --- a/src/common/lttng-elf.c +++ b/src/common/lttng-elf.c @@ -38,6 +38,8 @@ #define TEXT_SECTION_NAME ".text" #define SYMBOL_TAB_SECTION_NAME ".symtab" #define STRING_TAB_SECTION_NAME ".strtab" +#define DYNAMIC_SYMBOL_TAB_SECTION_NAME ".dynsym" +#define DYNAMIC_STRING_TAB_SECTION_NAME ".dynstr" #define NOTE_STAPSDT_SECTION_NAME ".note.stapsdt" #define NOTE_STAPSDT_NAME "stapsdt" #define NOTE_STAPSDT_TYPE 3 @@ -54,13 +56,13 @@ do { \ switch (sizeof(x)) { \ case 8: \ - x = be64toh(x); \ + x = be64toh((uint64_t)x); \ break; \ case 4: \ - x = be32toh(x); \ + x = be32toh((uint32_t)x); \ break; \ case 2: \ - x = be16toh(x); \ + x = be16toh((uint16_t)x); \ break; \ case 1: \ break; \ @@ -307,39 +309,32 @@ error: * Retrieve the nth (where n is the `index` argument) shdr (section * header) from the given elf instance. * - * A pointer to the shdr is returned on success, NULL on failure. + * 0 is returned on succes, -1 on failure. */ static -struct lttng_elf_shdr *lttng_elf_get_section_hdr(struct lttng_elf *elf, - uint16_t index) +int lttng_elf_get_section_hdr(struct lttng_elf *elf, + uint16_t index, struct lttng_elf_shdr *out_header) { - struct lttng_elf_shdr *section_header = NULL; int ret = 0; if (!elf) { + ret = -1; goto error; } if (index >= elf->ehdr->e_shnum) { + ret = -1; goto error; } - section_header = zmalloc(sizeof(struct lttng_elf_shdr)); - if (!section_header) { - goto error; - } - - ret = populate_section_header(elf, section_header, index); + ret = populate_section_header(elf, out_header, index); if (ret) { - ret = LTTNG_ERR_ELF_PARSING; DBG("Error populating section header."); goto error; } - return section_header; error: - free(section_header); - return NULL; + return ret; } /* @@ -533,7 +528,7 @@ end: static struct lttng_elf *lttng_elf_create(int fd) { - struct lttng_elf_shdr *section_names_shdr; + struct lttng_elf_shdr section_names_shdr; struct lttng_elf *elf = NULL; int ret; @@ -558,15 +553,14 @@ struct lttng_elf *lttng_elf_create(int fd) goto error; } - section_names_shdr = lttng_elf_get_section_hdr(elf, elf->ehdr->e_shstrndx); - if (!section_names_shdr) { + ret = lttng_elf_get_section_hdr( + elf, elf->ehdr->e_shstrndx, §ion_names_shdr); + if (ret) { goto error; } - elf->section_names_offset = section_names_shdr->sh_offset; - elf->section_names_size = section_names_shdr->sh_size; - - free(section_names_shdr); + elf->section_names_offset = section_names_shdr.sh_offset; + elf->section_names_size = section_names_shdr.sh_size; return elf; error: @@ -605,15 +599,19 @@ void lttng_elf_destroy(struct lttng_elf *elf) static int lttng_elf_get_section_hdr_by_name(struct lttng_elf *elf, - const char *section_name, struct lttng_elf_shdr **section_hdr) + const char *section_name, struct lttng_elf_shdr *section_hdr) { int i; char *curr_section_name; + for (i = 0; i < elf->ehdr->e_shnum; ++i) { - *section_hdr = lttng_elf_get_section_hdr(elf, i); - curr_section_name = lttng_elf_get_section_name(elf, - (*section_hdr)->sh_name); + int ret = lttng_elf_get_section_hdr(elf, i, section_hdr); + if (ret) { + break; + } + curr_section_name = lttng_elf_get_section_name(elf, + section_hdr->sh_name); if (!curr_section_name) { continue; } @@ -677,7 +675,7 @@ int lttng_elf_convert_addr_in_text_to_offset(struct lttng_elf *elf_handle, off_t text_section_addr_beg; off_t text_section_addr_end; off_t offset_in_section; - struct lttng_elf_shdr *text_section_hdr = NULL; + struct lttng_elf_shdr text_section_hdr; if (!elf_handle) { DBG("Invalid ELF handle."); @@ -694,9 +692,10 @@ int lttng_elf_convert_addr_in_text_to_offset(struct lttng_elf *elf_handle, goto error; } - text_section_offset = text_section_hdr->sh_offset; - text_section_addr_beg = text_section_hdr->sh_addr; - text_section_addr_end = text_section_addr_beg + text_section_hdr->sh_size; + text_section_offset = text_section_hdr.sh_offset; + text_section_addr_beg = text_section_hdr.sh_addr; + text_section_addr_end = + text_section_addr_beg + text_section_hdr.sh_size; /* * Verify that the address is within the .text section boundaries. @@ -737,8 +736,9 @@ int lttng_elf_get_symbol_offset(int fd, char *symbol, uint64_t *offset) char *curr_sym_str = NULL; char *symbol_table_data = NULL; char *string_table_data = NULL; - struct lttng_elf_shdr *symtab_hdr = NULL; - struct lttng_elf_shdr *strtab_hdr = NULL; + char *string_table_name = NULL; + struct lttng_elf_shdr symtab_hdr; + struct lttng_elf_shdr strtab_hdr; struct lttng_elf *elf = NULL; if (!symbol || !offset ) { @@ -752,16 +752,31 @@ int lttng_elf_get_symbol_offset(int fd, char *symbol, uint64_t *offset) goto end; } - /* Get the symbol table section header. */ + /* + * The .symtab section might not exist on stripped binaries. + * Try to get the symbol table section header first. If it's absent, + * try to get the dynamic symbol table. All symbols in the dynamic + * symbol tab are in the (normal) symbol table if it exists. + */ ret = lttng_elf_get_section_hdr_by_name(elf, SYMBOL_TAB_SECTION_NAME, &symtab_hdr); if (ret) { - DBG("Cannot get ELF Symbol Table section."); - ret = LTTNG_ERR_ELF_PARSING; - goto destroy_elf; + DBG("Cannot get ELF Symbol Table section. Trying to get ELF Dynamic Symbol Table section."); + /* Get the dynamic symbol table section header. */ + ret = lttng_elf_get_section_hdr_by_name(elf, DYNAMIC_SYMBOL_TAB_SECTION_NAME, + &symtab_hdr); + if (ret) { + DBG("Cannot get ELF Symbol Table nor Dynamic Symbol Table sections."); + ret = LTTNG_ERR_ELF_PARSING; + goto destroy_elf; + } + string_table_name = DYNAMIC_STRING_TAB_SECTION_NAME; + } else { + string_table_name = STRING_TAB_SECTION_NAME; } + /* Get the data associated with the symbol table section. */ - symbol_table_data = lttng_elf_get_section_data(elf, symtab_hdr); + symbol_table_data = lttng_elf_get_section_data(elf, &symtab_hdr); if (symbol_table_data == NULL) { DBG("Cannot get ELF Symbol Table data."); ret = LTTNG_ERR_ELF_PARSING; @@ -769,7 +784,7 @@ int lttng_elf_get_symbol_offset(int fd, char *symbol, uint64_t *offset) } /* Get the string table section header. */ - ret = lttng_elf_get_section_hdr_by_name(elf, STRING_TAB_SECTION_NAME, + ret = lttng_elf_get_section_hdr_by_name(elf, string_table_name, &strtab_hdr); if (ret) { DBG("Cannot get ELF string table section."); @@ -777,7 +792,7 @@ int lttng_elf_get_symbol_offset(int fd, char *symbol, uint64_t *offset) } /* Get the data associated with the string table section. */ - string_table_data = lttng_elf_get_section_data(elf, strtab_hdr); + string_table_data = lttng_elf_get_section_data(elf, &strtab_hdr); if (string_table_data == NULL) { DBG("Cannot get ELF string table section data."); ret = LTTNG_ERR_ELF_PARSING; @@ -785,7 +800,7 @@ int lttng_elf_get_symbol_offset(int fd, char *symbol, uint64_t *offset) } /* Get the number of symbol in the table for the iteration. */ - sym_count = symtab_hdr->sh_size / symtab_hdr->sh_entsize; + sym_count = symtab_hdr.sh_size / symtab_hdr.sh_entsize; /* Loop over all symbol. */ for (sym_idx = 0; sym_idx < sym_count; sym_idx++) { @@ -844,7 +859,7 @@ int lttng_elf_get_symbol_offset(int fd, char *symbol, uint64_t *offset) */ ret = lttng_elf_convert_addr_in_text_to_offset(elf, addr, offset); if (ret) { - DBG("Cannot convet addr to offset."); + DBG("Cannot convert addr to offset."); goto free_string_table_data; } @@ -871,11 +886,11 @@ int lttng_elf_get_sdt_probe_offsets(int fd, const char *provider_name, const char *probe_name, uint64_t **offsets, uint32_t *nb_probes) { int ret = 0, nb_match = 0; - struct lttng_elf_shdr *stap_note_section_hdr = NULL; + struct lttng_elf_shdr stap_note_section_hdr; struct lttng_elf *elf = NULL; char *stap_note_section_data = NULL; char *curr_note_section_begin, *curr_data_ptr, *curr_probe, *curr_provider; - char *next_note_ptr, *curr_desc_beg; + char *next_note_ptr; uint32_t name_size, desc_size, note_type; uint64_t curr_probe_location, curr_probe_offset, curr_semaphore_location; uint64_t *probe_locs = NULL, *new_probe_locs = NULL; @@ -902,14 +917,14 @@ int lttng_elf_get_sdt_probe_offsets(int fd, const char *provider_name, } /* Get the data associated with the stap note section. */ - stap_note_section_data = lttng_elf_get_section_data(elf, stap_note_section_hdr); + stap_note_section_data = + lttng_elf_get_section_data(elf, &stap_note_section_hdr); if (stap_note_section_data == NULL) { DBG("Cannot get ELF stap note section data."); ret = LTTNG_ERR_ELF_PARSING; goto destroy_elf_error; } - curr_data_ptr = stap_note_section_data; next_note_ptr = stap_note_section_data; curr_note_section_begin = stap_note_section_data; @@ -918,7 +933,8 @@ int lttng_elf_get_sdt_probe_offsets(int fd, const char *provider_name, curr_data_ptr = next_note_ptr; /* Check if we have reached the end of the note section. */ if (curr_data_ptr >= - curr_note_section_begin + stap_note_section_hdr->sh_size) { + curr_note_section_begin + + stap_note_section_hdr.sh_size) { *nb_probes = nb_match; *offsets = probe_locs; ret = 0; @@ -965,9 +981,6 @@ int lttng_elf_get_sdt_probe_offsets(int fd, const char *provider_name, curr_data_ptr += name_size; - /* Get description field. */ - curr_desc_beg = curr_data_ptr; - /* Get probe location. */ curr_probe_location = *(uint64_t *) curr_data_ptr; curr_data_ptr += sizeof(uint64_t); @@ -984,9 +997,6 @@ int lttng_elf_get_sdt_probe_offsets(int fd, const char *provider_name, /* Get probe name. */ curr_probe = curr_data_ptr; - curr_data_ptr += strlen(curr_probe) + 1; - - curr_data_ptr = curr_desc_beg + desc_size; /* Check if the provider and probe name match */ if (strcmp(provider_name, curr_provider) == 0 && @@ -1000,7 +1010,7 @@ int lttng_elf_get_sdt_probe_offsets(int fd, const char *provider_name, */ if (curr_semaphore_location != 0) { ret = LTTNG_ERR_SDT_PROBE_SEMAPHORE; - goto end; + goto realloc_error; } new_size = (++nb_match) * sizeof(uint64_t);