kasan.cpp

Go to the documentation of this file.

 
#include <runtime/kasan.h>
#include <common/logger.h>
 
using namespace MaxOS;
using namespace MaxOS::runtime;
 
KASanHandler::KASanHandler() = default;
 
KASanHandler::~KASanHandler() = default;
 
void KASanHandler::handle(bool write, uintptr_t address, size_t size, void* rip) {
    ASSERT(false, "KASAN ERROR: %s of size %zu at address 0x%lx from RIP 0x%p", write ? "WRITE" : "READ", size, address, rip);
}
 
void KASanHandler::initialize() {
 
}
 
void KASanHandler::poison_stack(uintptr_t address, size_t size) {
 
}
 
void KASanHandler::unpoison_stack(uintptr_t address, size_t size) {
 
}
 
extern "C" void __asan_load1_noabort(uintptr_t address) {
    KASanHandler::handle(false, address, 1, __builtin_return_address(0));
}
 
extern "C" void __asan_load2_noabort(uintptr_t address) {
    KASanHandler::handle(false, address, 2, __builtin_return_address(0));
}
 
extern "C" void __asan_load4_noabort(uintptr_t address) {
    KASanHandler::handle(false, address, 4, __builtin_return_address(0));
}
 
extern "C" void __asan_load8_noabort(uintptr_t address) {
    KASanHandler::handle(false, address, 8, __builtin_return_address(0));
}
 
extern "C" void __asan_load16_noabort(uintptr_t address) {
    KASanHandler::handle(false, address, 16, __builtin_return_address(0));
}
 
extern "C" void __asan_loadN_noabort(uintptr_t address, size_t size) {
    KASanHandler::handle(false, address, size, __builtin_return_address(0));
}
 
extern "C" void __asan_store1_noabort(uintptr_t address) {
    KASanHandler::handle(true, address, 1, __builtin_return_address(0));
}
 
extern "C" void __asan_store2_noabort(uintptr_t address) {
    KASanHandler::handle(true, address, 2, __builtin_return_address(0));
}
 
extern "C" void __asan_store4_noabort(uintptr_t address) {
    KASanHandler::handle(true, address, 4, __builtin_return_address(0));
}
 
extern "C" void __asan_store8_noabort(uintptr_t address) {
    KASanHandler::handle(true, address, 8, __builtin_return_address(0));
}
 
extern "C" void __asan_store16_noabort(uintptr_t address) {
    KASanHandler::handle(true, address, 16, __builtin_return_address(0));
}
 
extern "C" void __asan_storeN_noabort(uintptr_t address, size_t size) {
    KASanHandler::handle(false, address, size, __builtin_return_address(0));
}
 
extern "C" void __asan_handle_no_return() {
    ASSERT(false, "KASAN ERROR: Function marked 'noreturn' has returned");
}
 
extern "C" void __asan_alloca_poison(uintptr_t address, size_t size) {
    //TODO: Implement stack poisoning
    ASSERT(false, "KASAN ERROR: Stack poisoning not implemented for address 0x%lx of size %zu", address, size);
}
 
extern "C" void __asan_alloca_unpoison(void* stack_top, void* stack_bottom) {
    //TODO: Implement stack unpoisoning
    ASSERT(false, "KASAN ERROR: Stack unpoisoning not implemented for range 0x%p - 0x%p", stack_bottom, stack_top);
}
 
extern "C" void __asan_set_shadow_00(void* pointer, size_t size) {
 
    // Clear the memory to mark it as unpoisoned
    for(int i = 0; i < size; ++i)
        ((int8_t*)pointer)[i] = 0;
 
}
 
extern "C" void __asan_set_shadow_f8(void *pointer, size_t size){
 
    // Fill the memory to mark it as poisoned
    for(int i = 0; i < size; ++i)
        ((int8_t*)pointer)[i] = 0xF8;
 
}
 
extern "C" void __asan_before_dynamic_init(const void *module_name) {
    // Nothing to do
}
 
 extern "C" void __asan_after_dynamic_init(const void *module_name) {
    // Nothing to do
 }