hakmem/core/hakmem_sys.c

// hakmem_sys.c - Syscall Wrappers Implementation
//
// License: MIT
// Date: 2025-10-21

#include "hakmem_sys.h"
#include "hakmem_debug.h"
#include "hakmem_env_cache.h" // For HAK_ENV_SS_MADVISE_STRICT
#include "box/ss_os_acquire_box.h"
#include <sys/mman.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h> // For errno values

// madvise constants (Linux)
#ifndef MADV_DONTNEED
#define MADV_DONTNEED 4
#endif
#ifndef MADV_WILLNEED
#define MADV_WILLNEED 3
#endif

// ============================================================================
// Syscall Wrappers (計測付き)
// ============================================================================

void* hkm_sys_mmap(size_t size) {
    HKM_TIME_START(t0);

    void* ptr = mmap(NULL, size, PROT_READ | PROT_WRITE,
                     MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

    HKM_TIME_END(HKM_CAT_SYSCALL_MMAP, t0);

    if (ptr == MAP_FAILED) {
        fprintf(stderr, "[HAKMEM SYS] mmap(%zu) failed\n", size);
        return NULL;
    }

    return ptr;
}

void hkm_sys_munmap(void* ptr, size_t size) {
    if (!ptr) return;

    HKM_TIME_START(t0);

    int ret = munmap(ptr, size);

    HKM_TIME_END(HKM_CAT_SYSCALL_MUNMAP, t0);

    if (ret != 0) {
        fprintf(stderr, "[HAKMEM SYS] munmap(%p, %zu) failed\n", ptr, size);
    }
}

void hkm_sys_madvise_dontneed(void* ptr, size_t size) {
    if (!ptr) return;

    HKM_TIME_START(t0);

    int ret = ss_os_madvise_guarded(ptr, size, MADV_DONTNEED, "hakmem_sys_dontneed");

    HKM_TIME_END(HKM_CAT_SYSCALL_MADVISE, t0);

    if (ret != 0) {
        fprintf(stderr, "[HAKMEM SYS] madvise(DONTNEED, %p, %zu) failed errno=%d\n", ptr, size, errno);
        if (HAK_ENV_SS_MADVISE_STRICT() && errno == EINVAL) {
            fprintf(stderr, "[HAKMEM SYS] Critical: madvise(DONTNEED) failed with EINVAL in strict mode. Aborting.\n");
            abort();
        }
    }
}

void hkm_sys_madvise_willneed(void* ptr, size_t size) {
    if (!ptr) return;

    HKM_TIME_START(t0);

    int ret = ss_os_madvise_guarded(ptr, size, MADV_WILLNEED, "hakmem_sys_willneed");

    HKM_TIME_END(HKM_CAT_SYSCALL_MADVISE, t0);

    if (ret != 0) {
        fprintf(stderr, "[HAKMEM SYS] madvise(WILLNEED, %p, %zu) failed errno=%d\n", ptr, size, errno);
        if (HAK_ENV_SS_MADVISE_STRICT() && errno == EINVAL) {
            fprintf(stderr, "[HAKMEM SYS] Critical: madvise(WILLNEED) failed with EINVAL in strict mode. Aborting.\n");
            abort();
        }
    }
}