rt-thread/components/pthreads/pthread_tls.c

#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*))
{
	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;
}