/* Copyright (C) 2002 Free Software Foundation, Inc. Contributed by Zack Weinberg <zack@codesourcery.com> This file is part of GCC. GCC is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2, or (at your option) any later version. GCC is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with GCC; see the file COPYING. If not, write to the Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ /* Threads compatibility routines for libgcc2 for VxWorks. These are out-of-line routines called from gthr-vxworks.h. */ /* As a special exception, if you link this library with other files, some of which are compiled with GCC, to produce an executable, this library does not by itself cause the resulting executable to be covered by the GNU General Public License. This exception does not however invalidate any other reasons why the executable file might be covered by the GNU General Public License. */ #include "tconfig.h" #include "tsystem.h" #include "gthr.h" #include <vxWorks.h> #include <vxLib.h> #include <taskLib.h> #include <taskHookLib.h> /* Init-once operation. This would be a clone of the implementation from gthr-solaris.h, except that we have a bootstrap problem - the whole point of this exercise is to prevent double initialization, but if two threads are racing with each other, once->mutex is liable to be initialized by both. Then each thread will lock its own mutex, and proceed to call the initialization routine. So instead we use a bare atomic primitive (vxTas()) to handle mutual exclusion. Threads losing the race then busy-wait, calling taskDelay() to yield the processor, until the initialization is completed. Inefficient, but reliable. */ int __gthread_once (__gthread_once_t *guard, void (*func)(void)) { if (guard->done) return 0; while (!vxTas ((void *)&guard->busy)) taskDelay (1); /* Only one thread at a time gets here. Check ->done again, then go ahead and call func() if no one has done it yet. */ if (!guard->done) { func (); guard->done = 1; } guard->busy = 0; return 0; } /* Thread-specific data. We reserve a field in the TCB to point to a dynamically allocated array which is used to store TSD values. A TSD key is simply an offset in this array. The exact location of the TCB field is not known to this code nor to vxlib.c -- all access to it indirects through the routines __gthread_get_tsd_data and __gthread_set_tsd_data, which are provided by the VxWorks kernel. There is also a global array which records which keys are valid and which have destructors. A task delete hook is installed to execute key destructors. The routines __gthread_enter_tsd_dtor_context and __gthread_leave_tsd_dtor_context, which are also provided by the kernel, ensure that it is safe to call free() on memory allocated by the task being deleted. (This is a no-op on VxWorks 5, but a major undertaking on AE.) Since this interface is used to allocate only a small number of keys, the table size is small and static, which simplifies the code quite a bit. Revisit this if and when it becomes necessary. */ #define MAX_KEYS 4 /* This is the structure pointed to by the pointer returned by __gthread_get_tsd_data. */ struct tsd_data { void *values[MAX_KEYS]; unsigned int generation[MAX_KEYS]; }; /* kernel provided routines */ extern void *__gthread_get_tsd_data (WIND_TCB *tcb); extern void __gthread_set_tsd_data (WIND_TCB *tcb, void *data); extern void __gthread_enter_tsd_dtor_context (WIND_TCB *tcb); extern void __gthread_leave_tsd_dtor_context (WIND_TCB *tcb); typedef void (*fet_callback_t) (WIND_TCB *, unsigned int); extern void __gthread_for_all_tasks (fet_callback_t fun, unsigned int number); /* This is a global structure which records all of the active keys. A key is potentially valid (i.e. has been handed out by __gthread_key_create) iff its generation count in this structure is even. In that case, the matching entry in the dtors array is a routine to be called when a thread terminates with a valid, non-NULL specific value for that key. A key is actually valid in a thread T iff the generation count stored in this structure is equal to the generation count stored in T's specific-value structure. */ typedef void (*tsd_dtor) (void *); struct tsd_keys { tsd_dtor dtor[MAX_KEYS]; unsigned int generation[MAX_KEYS]; }; #define KEY_VALID_P(key) !(tsd_keys.generation[key] & 1) /* Note: if MAX_KEYS is increased, this initializer must be updated to match. All the generation counts begin at 1, which means no key is valid. */ static struct tsd_keys tsd_keys = { { 0, 0, 0, 0 }, { 1, 1, 1, 1 } }; /* This lock protects the tsd_keys structure. */ static __gthread_mutex_t tsd_lock; static __gthread_once_t tsd_init_guard = __GTHREAD_ONCE_INIT; /* Internal routines. */ /* The task TCB has just been deleted. Call the destructor function for each TSD key that has both a destructor and a non-NULL specific value in this thread. This routine does not need to take tsd_lock; the generation count protects us from calling a stale destructor. It does need to read tsd_keys.dtor[key] atomically. */ static void tsd_delete_hook (WIND_TCB *tcb) { struct tsd_data *data = __gthread_get_tsd_data (tcb); __gthread_key_t key; if (data) { __gthread_enter_tsd_dtor_context (tcb); for (key = 0; key < MAX_KEYS; key++) { if (data->generation[key] == tsd_keys.generation[key]) { tsd_dtor dtor = tsd_keys.dtor[key]; if (dtor) dtor (data->values[key]); } } free (data); __gthread_set_tsd_data (tcb, 0); __gthread_leave_tsd_dtor_context (tcb); } } /* Initialize global data used by the TSD system. */ static void tsd_init (void) { taskDeleteHookAdd ((FUNCPTR)tsd_delete_hook); __GTHREAD_MUTEX_INIT_FUNCTION (&tsd_lock); } /* External interface */ /* Store in KEYP a value which can be passed to __gthread_setspecific/ __gthread_getspecific to store and retrieve a value which is specific to each calling thread. If DTOR is not NULL, it will be called when a thread terminates with a non-NULL specific value for this key, with the value as its sole argument. */ int __gthread_key_create (__gthread_key_t *keyp, tsd_dtor dtor) { __gthread_key_t key; __gthread_once (&tsd_init_guard, tsd_init); if (__gthread_mutex_lock (&tsd_lock) == ERROR) return errno; for (key = 0; key < MAX_KEYS; key++) if (!KEY_VALID_P (key)) goto found_slot; /* no room */ __gthread_mutex_unlock (&tsd_lock); return EAGAIN; found_slot: tsd_keys.generation[key]++; /* making it even */ tsd_keys.dtor[key] = dtor; *keyp = key; __gthread_mutex_unlock (&tsd_lock); return 0; } /* Invalidate KEY; it can no longer be used as an argument to setspecific/getspecific. Note that this does NOT call destructor functions for any live values for this key. */ int __gthread_key_delete (__gthread_key_t key) { if (key >= MAX_KEYS) return EINVAL; __gthread_once (&tsd_init_guard, tsd_init); if (__gthread_mutex_lock (&tsd_lock) == ERROR) return errno; if (!KEY_VALID_P (key)) { __gthread_mutex_unlock (&tsd_lock); return EINVAL; } tsd_keys.generation[key]++; /* making it odd */ tsd_keys.dtor[key] = 0; __gthread_mutex_unlock (&tsd_lock); return 0; } /* Retrieve the thread-specific value for KEY. If it has never been set in this thread, or KEY is invalid, returns NULL. It does not matter if this function races with key_create or key_delete; the worst that can happen is you get a value other than the one that a serialized implementation would have provided. */ void * __gthread_getspecific (__gthread_key_t key) { struct tsd_data *data; if (key >= MAX_KEYS) return 0; data = __gthread_get_tsd_data (taskTcb (taskIdSelf ())); if (!data) return 0; if (data->generation[key] != tsd_keys.generation[key]) return 0; return data->values[key]; } /* Set the thread-specific value for KEY. If KEY is invalid, or memory allocation fails, returns -1, otherwise 0. The generation count protects this function against races with key_create/key_delete; the worst thing that can happen is that a value is successfully stored into a dead generation (and then immediately becomes invalid). However, we do have to make sure to read tsd_keys.generation[key] atomically. */ int __gthread_setspecific (__gthread_key_t key, void *value) { struct tsd_data *data; WIND_TCB *tcb; unsigned int generation; if (key >= MAX_KEYS) return EINVAL; tcb = taskTcb (taskIdSelf ()); data = __gthread_get_tsd_data (tcb); if (!data) { data = malloc (sizeof (struct tsd_data)); if (!data) return ENOMEM; memset (data, 0, sizeof (struct tsd_data)); __gthread_set_tsd_data (tcb, data); } generation = tsd_keys.generation[key]; if (generation & 1) return EINVAL; data->generation[key] = generation; data->values[key] = value; return 0; }