add thread local storage feature.

git-svn-id: https://rt-thread.googlecode.com/svn/trunk@1049 bbd45198-f89e-11dd-88c7-29a3b14d5316

components/pthreads/pthread_tls.c

@@ -1,19 +1,84 @@
 #include <pthread.h>
+#include "pthread_internal.h"
+
+struct _pthread_key_data
+{
+	int is_used;
+	void* (*destructor)(void* parameter);
+};
+typedef struct _pthread_key_data _pthread_key_data_t;
+static _pthread_key_data_t _thread_keys[PTHREAD_KEY_MAX];
+
+void pthread_key_system_init()
+{
+	rt_memset(&_thread_keys[0], 0, sizeof(_thread_keys));
+}
 
 void *pthread_getspecific(pthread_key_t key)
 {
+	struct _pthread_data* ptd;
+
+	ptd = _pthread_get_data(rt_thread_self());
+	RT_ASSERT(ptd != NULL);
+
+	if (ptd->tls == NULL) return NULL;
+
+	if ((key < PTHREAD_KEY_MAX) && (_thread_keys[key].is_used))
+		return ptd->tls[key];
+
+	return NULL;
 }
 
 int pthread_setspecific(pthread_key_t key, const void *value)
 {
+	struct _pthread_data* ptd;
+
+	ptd = _pthread_get_data(rt_thread_self());
+	RT_ASSERT(ptd != NULL);
+
+	/* check tls area */
+	if (ptd->tls == NULL) ptd->tls = rt_malloc(sizeof(void*) * PTHREAD_KEY_MAX);
+
+	if ((key < PTHREAD_KEY_MAX) && _thread_keys[key].is_used)
+	{
+		ptd->tls[key] = (void *)value;
+		return 0;
+	}
+
+	return EINVAL;
 }
 
 int pthread_key_create(pthread_key_t *key, void (*destructor)(void*))
 {
-	return 0;
+	rt_uint32_t index;
+
+	rt_enter_critical();
+	for (index = 0; index < PTHREAD_KEY_MAX; index ++)
+	{
+		if (_thread_keys[index].is_used == 0)
+		{
+			_thread_keys[index].is_used = 1;
+			_thread_keys[index].destructor = destructor;
+
+			*key = index;
+
+			rt_exit_critical();
+			return 0;
+		}
+	}
+
+	rt_exit_critical();
+	return EAGAIN;
 }
 
 int pthread_key_delete(pthread_key_t key)
 {
+	if (key >= PTHREAD_KEY_MAX) return EINVAL;
+
+	rt_enter_critical();
+	_thread_keys[key].is_used = 0;
+	_thread_keys[key].destructor = 0;
+	rt_exit_critical();
+
 	return 0;
 }