LibC: Implement POSIX TLD area (pthread_key*)

userspace/libraries/LibC/include/bits/pthread_types.h

@@ -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
 

userspace/libraries/LibC/include/limits.h

@@ -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
 

userspace/libraries/LibC/pthread.cpp

@@ -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);