LibC: merge atexit and __cxa_atexit into common function

This commit is contained in:
2025-04-15 22:20:40 +03:00
parent 4f49d60e4a
commit 066ed7e4a1
2 changed files with 18 additions and 30 deletions

View File

@@ -1,41 +1,39 @@
#include <BAN/Assert.h>
#include <stdint.h>
#include <stddef.h>
#define ATEXIT_MAX_FUNCS 128
struct atexit_func_entry_t
{
void(*func)(void*);
void (*func)(void*);
void* arg;
void* dso_handle;
};
static atexit_func_entry_t __atexit_funcs[ATEXIT_MAX_FUNCS];
static size_t __atexit_func_count = 0;
static atexit_func_entry_t s_atexit_funcs[ATEXIT_MAX_FUNCS];
static size_t s_atexit_func_count = 0;
extern "C" int __cxa_atexit(void(*func)(void*), void* arg, void* dso_handle)
{
if (__atexit_func_count >= ATEXIT_MAX_FUNCS)
if (s_atexit_func_count >= ATEXIT_MAX_FUNCS)
return -1;
auto& atexit_func = __atexit_funcs[__atexit_func_count++];
atexit_func.func = func;
atexit_func.arg = arg;
atexit_func.dso_handle = dso_handle;
s_atexit_funcs[s_atexit_func_count++] = {
.func = func,
.arg = arg,
.dso_handle = dso_handle,
};
return 0;
};
extern "C" void __cxa_finalize(void* f)
extern "C" void __cxa_finalize(void* dso_handle)
{
for (size_t i = __atexit_func_count; i > 0; i--)
for (size_t i = s_atexit_func_count; i > 0; i--)
{
auto& atexit_func = __atexit_funcs[i - 1];
auto& atexit_func = s_atexit_funcs[i - 1];
if (atexit_func.func == nullptr)
continue;
if (f == nullptr || f == atexit_func.func)
{
atexit_func.func(atexit_func.arg);
atexit_func.func = nullptr;
}
if (dso_handle && dso_handle != atexit_func.dso_handle)
continue;
atexit_func.func(atexit_func.arg);
atexit_func.func = nullptr;
}
};