Bananymous
/
banan-os
1
1
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

LibC: Implement POSIX TLD area (pthread_key*)

This commit is contained in:
Bananymous 2025-04-19 22:16:36 +03:00
parent f1585d09e1
commit fe6c4cd0b5
3 changed files with 77 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
userspace/libraries/LibC/include/bits/pthread_types.h
View File

@ -36,7 +36,7 @@ __BEGIN_DECLS
#if !defined(__pthread_key_t_defined) && (defined(__need_all_types) || defined(__need_pthread_key_t))
#define __pthread_key_t_defined 1
typedef int pthread_key_t;
typedef unsigned pthread_key_t;
#endif
#undef __need_pthread_key_t

3
userspace/libraries/LibC/include/limits.h
View File

@ -66,6 +66,9 @@ __BEGIN_DECLS
#define HOST_NAME_MAX 255
#define TTY_NAME_MAX PATH_MAX
#define PTHREAD_KEYS_MAX _POSIX_THREAD_KEYS_MAX
#define PTHREAD_DESTRUCTOR_ITERATIONS _POSIX_THREAD_DESTRUCTOR_ITERATIONS
#define CHAR_MAX SCHAR_MAX
#define CHAR_MIN SCHAR_MIN

73
userspace/libraries/LibC/pthread.cpp
View File

@ -108,6 +108,62 @@ void pthread_cleanup_push(void (*routine)(void*), void* arg)
}
#endif
#if not __disable_thread_local_storage
static thread_local struct {
void* value;
void (*destructor)(void*);
} s_pthread_keys[PTHREAD_KEYS_MAX] {};
static thread_local uint8_t s_pthread_keys_allocated[(PTHREAD_KEYS_MAX + 7) / 8];
static inline bool is_pthread_key_allocated(pthread_key_t key)
{
if (key >= PTHREAD_KEYS_MAX)
return false;
return s_pthread_keys_allocated[key / 8] & (1 << (key % 8));
}
int pthread_key_create(pthread_key_t* key, void (*destructor)(void*))
{
for (pthread_key_t i = 0; i < PTHREAD_KEYS_MAX; i++)
{
if (is_pthread_key_allocated(i))
continue;
s_pthread_keys[i].value = nullptr;
s_pthread_keys[i].destructor = destructor;
s_pthread_keys_allocated[i / 8] |= 1 << (i % 8);
*key = i;
return 0;
}
return EAGAIN;
}
int pthread_key_delete(pthread_key_t key)
{
if (!is_pthread_key_allocated(key))
return EINVAL;
s_pthread_keys[key].value = nullptr;
s_pthread_keys[key].destructor = nullptr;
s_pthread_keys_allocated[key / 8] &= ~(1 << (key % 8));
return 0;
}
void* pthread_getspecific(pthread_key_t key)
{
if (!is_pthread_key_allocated(key))
return nullptr;
return s_pthread_keys[key].value;
}
int pthread_setspecific(pthread_key_t key, const void* value)
{
if (!is_pthread_key_allocated(key))
return EINVAL;
s_pthread_keys[key].value = const_cast<void*>(value);
return 0;
}
#endif
int pthread_attr_init(pthread_attr_t* attr)
{
*attr = 0;
@ -189,6 +245,23 @@ void pthread_exit(void* value_ptr)
#if not __disable_thread_local_storage
while (s_cleanup_stack)
pthread_cleanup_pop(1);
for (size_t iteration = 0; iteration < PTHREAD_DESTRUCTOR_ITERATIONS; iteration++)
{
bool called = false;
for (pthread_key_t i = 0; i < PTHREAD_KEYS_MAX; i++)
{
if (!is_pthread_key_allocated(i))
continue;
if (!s_pthread_keys[i].value || !s_pthread_keys[i].destructor)
continue;
void* old_value = s_pthread_keys[i].value;
s_pthread_keys[i].value = nullptr;
s_pthread_keys[i].destructor(old_value);
called = true;
}
if (!called)
break;
}
#endif
free_uthread(get_uthread());
syscall(SYS_PTHREAD_EXIT, value_ptr);