Guard madvise ENOMEM and stabilize pool/tiny front v3

This commit is contained in:
Moe Charm (CI)
2025-12-09 21:50:15 +09:00
parent e274d5f6a9
commit a905e0ffdd
45 changed files with 3154 additions and 242 deletions

View File

@ -5,9 +5,12 @@
#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
@ -56,12 +59,16 @@ void hkm_sys_madvise_dontneed(void* ptr, size_t size) {
HKM_TIME_START(t0);
int ret = madvise(ptr, size, MADV_DONTNEED);
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\n", ptr, size);
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();
}
}
}
@ -70,11 +77,15 @@ void hkm_sys_madvise_willneed(void* ptr, size_t size) {
HKM_TIME_START(t0);
int ret = madvise(ptr, size, MADV_WILLNEED);
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\n", ptr, size);
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();
}
}
}