2 // ELFManager.cpp: ELF format manager
6 // JPM = Jean-Paul Mari <djipi.mari@gmail.com>
9 // --- ---------- ------------------------------------------------------------
10 // JPM Jan./2016 Created this file and added ELF format support
11 // JPM 07/13/2017 ELF DWARF format support improvement
12 // JPM 10/20/2018 Added function name support from ELF structure
17 #include "libelf/libelf.h"
18 #include "libelf/gelf.h"
21 #include "ELFManager.h"
22 #include "DwarfManager.h"
25 //#define LOG_SUPPORT // Support log
29 const char *SectionName
;
44 ELFSectionType ELFTabSectionType
[] = {
45 { "", ELF_NULL_TYPE
},
46 { ".text", ELF_text_TYPE
},
47 { ".rodata", ELF_rodata_TYPE
},
48 { ".data", ELF_data_TYPE
},
49 { ".bss", ELF_bss_TYPE
},
50 { ".heap", ELF_heap_TYPE
},
51 { ".debug_aranges", ELF_debug_aranges_TYPE
},
52 { ".debug_info", ELF_debug_info_TYPE
},
53 { ".debug_abbrev", ELF_debug_abbrev_TYPE
},
54 { ".debug_line", ELF_debug_line_TYPE
},
55 { ".debug_frame", ELF_debug_frame_TYPE
},
56 { ".debug_str", ELF_debug_str_TYPE
},
57 { ".debug_loc", ELF_debug_loc_TYPE
},
58 { ".debug_ranges", ELF_debug_ranges_TYPE
},
59 { ".comment", ELF_comment_TYPE
},
60 { ".shstrtab", ELF_shstrtab_TYPE
},
61 { ".symtab", ELF_symtab_TYPE
},
62 { ".strtab", ELF_strtab_TYPE
}
68 void *PtrExec
; // ELF executable
70 // ELF Dwarf management
74 size_t NbELFtabStruct
;
78 char *ELFManager_GetSymbolnameFromSymbolindex(size_t Index
);
81 // ELF section type detection
82 size_t ELFManager_GetSectionType(char *SectionName
)
86 for (i
= 0; i
< ELF_END_TYPE
; i
++)
88 if (!strcmp(ELFTabSectionType
[i
].SectionName
, SectionName
))
90 return (ELFTabSectionType
[i
].SectionType
);
94 return (size_t)ELF_NO_TYPE
;
98 // ELF manager executable copy
99 void *ELFManager_ExeCopy(void *src
, size_t size
)
101 if ((PtrExec
= malloc(size
)) != NULL
)
103 memcpy(PtrExec
, src
, size
);
110 Elf
*ELFManager_MemOpen(char *PtrELFExe
, size_t Size
)
112 return(ElfMem
= elf_memory(PtrELFExe
, Size
));
116 int ELFManager_MemEnd(void)
122 err
= elf_end(ElfMem
);
130 // ELF manager Initialisation
131 void ELFManager_Init(void)
141 // ELF manager Dwarf Initialisation
142 bool ELFManager_DwarfInit(Elf
*PtrElfMem
)
144 return (ElfDwarf
= (DWARFManager_ElfInit(PtrElfMem
) == DW_DLV_OK
) ? true : false);
149 void ELFManager_Reset(void)
153 DWARFManager_Reset();
165 while (NbELFtabStruct
)
167 free(ELFtab
[--NbELFtabStruct
]);
178 // ELF manager Closing
179 void ELFManager_Close(void)
185 // Save the ELF tab in memory
186 bool ELFManager_AddTab(void *Ptr
, size_t type
)
190 if ((ELFtab
= (ELFTab
**)calloc(1, sizeof(ELFTab
*))) == NULL
)
197 if ((ELFtab
= (ELFTab
**)realloc(ELFtab
, sizeof(ELFTab
*)*(NbELFtabStruct
+1))) == NULL
)
203 if ((ELFtab
[NbELFtabStruct
] = (ELFTab
*)calloc(1, sizeof(ELFTab
))) == NULL
)
209 ELFtab
[NbELFtabStruct
]->Type
= type
;
210 ELFtab
[NbELFtabStruct
]->PtrTab
= Ptr
;
217 // Get Address from his Symbol Name
218 // Return 0 if Symbol name is not found
219 size_t ELFManager_GetAdrFromSymbolName(char *SymbolName
)
224 if (ELFtab
&& SymbolName
)
226 for (size_t i
= 0; i
< NbELFtabStruct
; i
++)
228 if ((ELFtab
[i
]->Type
== ELF_symtab_TYPE
) && ((ELFtab
[i
]->PtrDataTab
) != NULL
))
232 while ((PtrST
= gelf_getsym(ELFtab
[i
]->PtrDataTab
, j
++, &ST
)) != NULL
)
234 if (!strcmp(ELFManager_GetSymbolnameFromSymbolindex(PtrST
->st_name
), SymbolName
))
236 Adr
= PtrST
->st_value
;
247 // Get function name from his address
248 // Return NULL if function name is not found
249 char *ELFManager_GetFunctionName(size_t Adr
)
251 char *SymbolName
= NULL
;
256 for (size_t i
= 0; i
< NbELFtabStruct
; i
++)
258 if ((ELFtab
[i
]->Type
== ELF_symtab_TYPE
) && ((ELFtab
[i
]->PtrDataTab
) != NULL
))
262 while ((PtrST
= gelf_getsym(ELFtab
[i
]->PtrDataTab
, j
++, &ST
)) != NULL
)
264 if (PtrST
->st_value
== Adr
)
266 if (ELF32_ST_TYPE(PtrST
->st_info
) == STT_FUNC
)
268 SymbolName
= ELFManager_GetSymbolnameFromSymbolindex(PtrST
->st_name
);
280 // Get Symbol name from his address
281 // Return NULL if Symbol name is not found
282 char *ELFManager_GetSymbolnameFromAdr(size_t Adr
)
284 char *SymbolName
= NULL
;
289 for (size_t i
= 0; i
< NbELFtabStruct
; i
++)
291 if ((ELFtab
[i
]->Type
== ELF_symtab_TYPE
) && ((ELFtab
[i
]->PtrDataTab
) != NULL
))
295 while ((PtrST
= gelf_getsym(ELFtab
[i
]->PtrDataTab
, j
++, &ST
)) != NULL
)
297 if (PtrST
->st_value
== Adr
)
300 WriteLog("ELF: .symtab: DATA: st_info=%0x, st_name=%0x, st_other=%0x, st_shndx=%0x, st_size=%0x, st_value=%0x\n", PtrST
->st_info
, PtrST
->st_name
, PtrST
->st_other
, PtrST
->st_shndx
, PtrST
->st_size
, PtrST
->st_value
);
302 SymbolName
= ELFManager_GetSymbolnameFromSymbolindex(PtrST
->st_name
);
313 // Get Symbol name from his Symbol index
314 char *ELFManager_GetSymbolnameFromSymbolindex(size_t Index
)
318 for (size_t i
= 0; i
< NbELFtabStruct
; i
++)
320 if ((ELFtab
[i
]->Type
== ELF_strtab_TYPE
) && ((ELFtab
[i
]->PtrDataTab
) != NULL
))
322 return ((char *)(ELFtab
[i
]->PtrDataTab
->d_buf
) + Index
);