1 #include "traceback.h" 2 #include <common/printk.h> 3 #include <process/process.h> 4 5 static int lookup_kallsyms(uint64_t addr, int level) 6 { 7 const char *str = (const char *)&kallsyms_names; 8 9 // 暴力查找符合要求的symbol 10 // todo: 改用二分搜索。 11 // 由于符号表使用nm -n生成，因此是按照地址升序排列的，因此可以二分 12 uint64_t index = 0; 13 for (index = 0; index < kallsyms_num - 1; ++index) 14 { 15 if (addr > kallsyms_address[index] && addr <= kallsyms_address[index + 1]) 16 break; 17 } 18 19 if (index < kallsyms_num) // 找到对应的函数 20 { 21 // 依次输出函数名称、rip离函数起始处的偏移量、函数执行的rip 22 printk("function:%s() \t(+) %04d address:%#018lx\n", &str[kallsyms_names_index[index]], addr - kallsyms_address[index], addr); 23 return 0; 24 } 25 else 26 return -1; 27 } 28 29 /** 30 * @brief 追溯内核栈调用情况 31 * 32 * @param regs 内核栈结构体 33 */ 34 void traceback(struct pt_regs *regs) 35 { 36 // 先检验是否为用户态出错，若为用户态出错，则直接返回 37 if (verify_area(regs->rbp, 0)) 38 { 39 printk_color(YELLOW, BLACK, "Kernel traceback: Fault in userland. pid=%ld, rbp=%#018lx\n", rs_current_pcb_pid(), regs->rbp); 40 return; 41 } 42 43 uint64_t *rbp = (uint64_t *)regs->rbp; 44 printk_color(YELLOW, BLACK, "======== Kernel traceback =======\n"); 45 // printk("&kallsyms_address:%#018lx,kallsyms_address:%#018lx\n", &kallsyms_address, kallsyms_address); 46 // printk("&kallsyms_syms_num:%#018lx,kallsyms_syms_num:%d\n", &kallsyms_num, kallsyms_num); 47 // printk("&kallsyms_index:%#018lx\n", &kallsyms_names_index); 48 // printk("&kallsyms_names:%#018lx,kallsyms_names:%s\n", &kallsyms_names, &kallsyms_names); 49 50 uint64_t ret_addr = regs->rip; 51 // 最大追踪10层调用栈 52 for (int i = 0; i < 10; ++i) 53 { 54 if (lookup_kallsyms(ret_addr, i) != 0) 55 break; 56 57 // 当前栈帧的rbp的地址大于等于内核栈的rbp的时候，表明调用栈已经到头了，追踪结束。 58 // 当前rbp的地址为用户空间时，直接退出 59 if ((uint64_t)(rbp) >= rs_current_pcb_thread_rbp() || ((uint64_t)rbp < regs->rsp)) 60 break; 61 62 printk_color(ORANGE, BLACK, "rbp:%#018lx,*rbp:%#018lx\n", rbp, *rbp); 63 64 // 由于x86处理器在执行call指令时，先将调用返回地址压入栈中，然后再把函数的rbp入栈，最后将rsp设为新的rbp。 65 // 因此，此处的rbp就是上一层的rsp，那么，*(rbp+1)得到的就是上一层函数的返回地址 66 ret_addr = *(rbp + 1); 67 rbp = (uint64_t *)(*rbp); 68 printk("\n"); 69 } 70 printk_color(YELLOW, BLACK, "======== Kernel traceback end =======\n"); 71 }