6fd3fa90bf
ysyx_22040000 李心杨 Linux calcite 6.6.19 #1-NixOS SMP PREEMPT_DYNAMIC Fri Mar 1 12:35:11 UTC 2024 x86_64 GNU/Linux 15:46:59 up 2:52, 2 users, load average: 1.25, 0.75, 0.47
100 lines
3.4 KiB
C
100 lines
3.4 KiB
C
#include "macro.h"
|
|
#include <assert.h>
|
|
#include <common.h>
|
|
#include <elf.h>
|
|
#include <func.h>
|
|
|
|
// Put this into another file
|
|
#ifdef CONFIG_FTRACE
|
|
// static vaddr_t g_function_stack[CONFIG_FTRACE_STACK_SIZE] = {0};
|
|
// static vaddr_t *g_function_start = NULL;
|
|
func_t *func_table = NULL;
|
|
int func_table_len = 0, func_table_size = 8;
|
|
#endif
|
|
|
|
static int cmp_func_t(const void *a, const void *b) {
|
|
return ((func_t *)a)->start > ((func_t *)b)->start;
|
|
}
|
|
|
|
func_t *get_func(vaddr_t addr) {
|
|
int l = 0, r = func_table_len - 1;
|
|
while(l < r) {
|
|
int mid = l + (r - l) / 2;
|
|
if(func_table[mid].start < addr) l = mid + 1;
|
|
else r = mid;
|
|
}
|
|
return &func_table[l];
|
|
}
|
|
|
|
void init_elf(const char *path) {
|
|
bool success = false;
|
|
FILE *elf_file = fopen(path, "rb");
|
|
Elf32_Ehdr header;
|
|
Elf32_Shdr section_header[200], *psh;
|
|
|
|
func_table = (func_t *)calloc(func_table_size, sizeof(func_t));
|
|
FAILED_GOTO(failed_nosym, func_table == NULL);
|
|
|
|
FAILED_GOTO(failed_nosym, fread(&header, sizeof(Elf32_Ehdr), 1, elf_file) <= 0);
|
|
FAILED_GOTO(failed_nosym, fseek(elf_file, header.e_shoff, SEEK_SET) != 0);
|
|
FAILED_GOTO(failed_nosym, fread(section_header, header.e_shentsize, header.e_shnum, elf_file) <= 0);
|
|
|
|
char *shstrtab = calloc(1, section_header[header.e_shstrndx].sh_size);
|
|
FAILED_GOTO(failed_shstrtab, fseek(elf_file, section_header[header.e_shstrndx].sh_offset, SEEK_SET) != 0);
|
|
FAILED_GOTO(failed_shstrtab, fread(shstrtab, section_header[header.e_shstrndx].sh_size, 1, elf_file) <= 0);
|
|
|
|
Elf32_Shdr *symtab = NULL, *strtab = NULL;
|
|
for(int i = 0; i < header.e_shnum; i++) {
|
|
psh = section_header + i;
|
|
if (psh->sh_type == SHT_SYMTAB) {
|
|
symtab = psh;
|
|
} else if (psh->sh_type == SHT_STRTAB && strncmp(shstrtab + psh->sh_name, ".strtab", 8) == 0) {
|
|
strtab = psh;
|
|
}
|
|
}
|
|
|
|
int sym_length = symtab->sh_size / sizeof(Elf32_Sym);
|
|
Elf32_Sym *sym = calloc(sym_length, sizeof(Elf32_Sym));
|
|
FAILED_GOTO(failed_nosym, sym == NULL);
|
|
FAILED_GOTO(failed, fseek(elf_file, symtab->sh_offset, SEEK_SET) != 0);
|
|
FAILED_GOTO(failed, fread(sym, sizeof(Elf32_Sym), sym_length, elf_file) <= 0);
|
|
|
|
for(int j = 0; j < sym_length; j++) {
|
|
if(ELF32_ST_TYPE(sym[j].st_info) != STT_FUNC) continue;
|
|
// Only read function type symbol
|
|
func_t *f = &func_table[func_table_len];
|
|
char *func = (char *)malloc(30);
|
|
FAILED_GOTO(failed, fseek(elf_file, strtab->sh_offset + sym[j].st_name, SEEK_SET) != 0);
|
|
FAILED_GOTO(failed, fgets(func, 30, elf_file) <= 0);
|
|
f->start = sym[j].st_value;
|
|
f->len = sym[j].st_size;
|
|
f->name = func;
|
|
++func_table_len;
|
|
if(func_table_len >= func_table_size) {
|
|
Assert(func_table_size * 2 > func_table_size, "Function table exceed memory limit");
|
|
func_table_size *= 2;
|
|
func_table = realloc(func_table, func_table_size * sizeof(func_t));
|
|
Assert(func_table, "Function table exceed memory limit");
|
|
}
|
|
}
|
|
qsort(func_table, func_table_len, sizeof(func_t), cmp_func_t);
|
|
for(int i = 0; i < func_table_len; i++) {
|
|
func_t *f = &func_table[i];
|
|
printf("%s: 0x%x - 0x%x\n", f->name, f->start, f->start + f->len);
|
|
}
|
|
printf("%s\n", get_func(0x80000010)->name);
|
|
printf("%s\n", get_func(0x80000026)->name);
|
|
success = true;
|
|
failed:
|
|
for(int i = 0; i < func_table_len; i++) {
|
|
func_t *f = &func_table[i];
|
|
if(f->name) { free(f->name); }
|
|
}
|
|
free(func_table);
|
|
free(sym);
|
|
failed_shstrtab:
|
|
free(shstrtab);
|
|
failed_nosym:
|
|
if(success) return;
|
|
else Error("Failed reading elf file");
|
|
}
|