Guard madvise ENOMEM and stabilize pool/tiny front v3
This commit is contained in:
@ -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();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user