:: limine / common / lib / trace.s2.c 2.0 KB raw

#include <stddef.h>

#include <stdint.h>

#include <lib/trace.h>

#include <lib/misc.h>

#include <lib/config.h>

#include <lib/print.h>

#include <lib/uri.h>

#include <fs/file.h>

#include <mm/pmm.h>

​

#if defined (BIOS)

extern symbol stage2_map;

#elif defined (UEFI)

extern symbol __slide;

#endif

​

extern symbol full_map;

​

static char *trace_address(size_t *off, size_t addr) {

    char *limine_map;

​

#if defined (BIOS)

    if (!stage3_loaded)

        limine_map = stage2_map;

    else

        limine_map = full_map;

#elif defined (UEFI)

    limine_map = full_map;

​

    addr -= (size_t)__slide;

#endif

​

    uintptr_t prev_addr = 0;

    char     *prev_sym  = NULL;

​

    for (size_t i = 0;;) {

        if (*((uintptr_t *)&limine_map[i]) >= addr) {

            *off = addr - prev_addr;

            return prev_sym;

        }

        prev_addr = *((uintptr_t *)&limine_map[i]);

        i += sizeof(uintptr_t);

        prev_sym  = &limine_map[i];

        while (limine_map[i++] != 0);

    }

}

​

void print_stacktrace(size_t *base_ptr) {

    if (base_ptr == NULL) {

        asm volatile (

#if defined (__i386__)

            "movl %%ebp, %0"

#elif defined (__x86_64__)

            "movq %%rbp, %0"

#elif defined (__aarch64__)

            "mov %0, x29"

#elif defined (__riscv)

            "mv %0, fp;  addi %0, %0, -16"

#elif defined (__loongarch64)

            "move %0, $fp;  addi.d %0, %0, -16"

#endif

            : "=r"(base_ptr)

            :: "memory"

        );

    }

    print("Stacktrace:\n");

    for (;;) {

        size_t old_bp = base_ptr[0];

        size_t ret_addr = base_ptr[1];

        if (!ret_addr)

            break;

        size_t off;

        char *name = trace_address(&off, ret_addr);

        if (name)

            print("  [%p] <%s+%p>\n", ret_addr, name, off);

        else

            print("  [%p]\n", ret_addr);

        if (!old_bp)

            break;

#if defined (__riscv) || defined (__loongarch64)

        base_ptr = (void *)(old_bp - 16);

#else

        base_ptr = (void*)old_bp;

#endif

    }

    print("End of trace. ");

}