wut/libraries/wutstdc++/wut_gthread_keys.cc

#include "wut_gthread.h"
#include <string.h>
#include <sys/errno.h>

struct __wut_key_table_entry
{
   bool in_use;
   void (*dtor) (void *);
};

static __wut_key_table_entry key_table[__WUT_MAX_KEYS];

static OSMutex key_mutex;
static __wut_once_t init_once_control = __WUT_ONCE_VALUE_INIT;

static void init()
{
   __wut_mutex_init_function(&key_mutex);
   memset(key_table, 0, sizeof(key_table));
}

int
__wut_key_create(__wut_key_t *key,
                 void (*dtor) (void *))
{
   int res = EAGAIN;
   __wut_once(&init_once_control, init);
   __wut_mutex_lock(&key_mutex);

   for (int i = 0; i < __WUT_MAX_KEYS; ++i) {
      if (key_table[i].in_use) {
         continue;
      }

      key_table[i].in_use = 1;
      key_table[i].dtor = dtor;

      res = 0;
      *key = (__wut_key_t)i;
      break;
   }

   __wut_mutex_unlock(&key_mutex);
   return res;
}

int
__wut_key_delete(__wut_key_t key)
{
   __wut_mutex_lock(&key_mutex);
   key_table[key].in_use = 0;
   key_table[key].dtor = NULL;
   __wut_mutex_unlock(&key_mutex);
   return -1;
}

static const void **
__wut_get_thread_keys()
{
   const void **keys = (const void **)OSGetThreadSpecific(__WUT_KEY_THREAD_SPECIFIC_ID);
   if (!keys) {
      MEMExpandedHeap *heap = (MEMExpandedHeap *)MEMGetBaseHeapHandle(MEM_BASE_HEAP_MEM2);
      keys = (const void **)MEMAllocFromExpHeapEx(heap, sizeof(void *) * sizeof(__WUT_MAX_KEYS), 4);
      memset(keys, 0, sizeof(void *) * sizeof(__WUT_MAX_KEYS));
      OSSetThreadSpecific(__WUT_KEY_THREAD_SPECIFIC_ID, keys);
   }

   return keys;
}

void *
__wut_getspecific(__wut_key_t key)
{
   return (void *)__wut_get_thread_keys()[key];
}

int
__wut_setspecific(__wut_key_t key,
                  const void *ptr)
{
   __wut_get_thread_keys()[key] = ptr;
   return 0;
}

void
__wut_key_cleanup(OSThread *thread)
{
   void **keys = (void **)OSGetThreadSpecific(__WUT_KEY_THREAD_SPECIFIC_ID);
   if (!keys) {
      return;
   }

   __wut_mutex_lock(&key_mutex);

   for (int i = 0; i < __WUT_MAX_KEYS; ++i) {
      if (key_table[i].in_use && key_table[i].dtor && keys[i]) {
         key_table[i].dtor(keys[i]);
      }
   }

   __wut_mutex_unlock(&key_mutex);
}