301 lines
6.6 KiB
C
301 lines
6.6 KiB
C
/* Threading for AtheOS.
|
|
Based on thread_beos.h. */
|
|
|
|
#include <atheos/threads.h>
|
|
#include <atheos/semaphore.h>
|
|
#include <atheos/atomic.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
/* Missing decl from threads.h */
|
|
extern int exit_thread(int);
|
|
|
|
|
|
/* Undefine FASTLOCK to play with simple semaphores. */
|
|
#define FASTLOCK
|
|
|
|
|
|
#ifdef FASTLOCK
|
|
|
|
/* Use an atomic counter and a semaphore for maximum speed. */
|
|
typedef struct fastmutex {
|
|
sem_id sem;
|
|
atomic_t count;
|
|
} fastmutex_t;
|
|
|
|
|
|
static int fastmutex_create(const char *name, fastmutex_t * mutex);
|
|
static int fastmutex_destroy(fastmutex_t * mutex);
|
|
static int fastmutex_lock(fastmutex_t * mutex);
|
|
static int fastmutex_timedlock(fastmutex_t * mutex, bigtime_t timeout);
|
|
static int fastmutex_unlock(fastmutex_t * mutex);
|
|
|
|
|
|
static int fastmutex_create(const char *name, fastmutex_t * mutex)
|
|
{
|
|
mutex->count = 0;
|
|
mutex->sem = create_semaphore(name, 0, 0);
|
|
return (mutex->sem < 0) ? -1 : 0;
|
|
}
|
|
|
|
|
|
static int fastmutex_destroy(fastmutex_t * mutex)
|
|
{
|
|
if (fastmutex_timedlock(mutex, 0) == 0 || errno == EWOULDBLOCK) {
|
|
return delete_semaphore(mutex->sem);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int fastmutex_lock(fastmutex_t * mutex)
|
|
{
|
|
atomic_t prev = atomic_add(&mutex->count, 1);
|
|
if (prev > 0)
|
|
return lock_semaphore(mutex->sem);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int fastmutex_timedlock(fastmutex_t * mutex, bigtime_t timeout)
|
|
{
|
|
atomic_t prev = atomic_add(&mutex->count, 1);
|
|
if (prev > 0)
|
|
return lock_semaphore_x(mutex->sem, 1, 0, timeout);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int fastmutex_unlock(fastmutex_t * mutex)
|
|
{
|
|
atomic_t prev = atomic_add(&mutex->count, -1);
|
|
if (prev > 1)
|
|
return unlock_semaphore(mutex->sem);
|
|
return 0;
|
|
}
|
|
|
|
|
|
#endif /* FASTLOCK */
|
|
|
|
|
|
/*
|
|
* Initialization.
|
|
*
|
|
*/
|
|
static void PyThread__init_thread(void)
|
|
{
|
|
/* Do nothing. */
|
|
return;
|
|
}
|
|
|
|
|
|
/*
|
|
* Thread support.
|
|
*
|
|
*/
|
|
|
|
static atomic_t thread_count = 0;
|
|
|
|
long PyThread_start_new_thread(void (*func) (void *), void *arg)
|
|
{
|
|
status_t success = -1;
|
|
thread_id tid;
|
|
char name[OS_NAME_LENGTH];
|
|
atomic_t this_thread;
|
|
|
|
dprintf(("PyThread_start_new_thread called\n"));
|
|
|
|
this_thread = atomic_add(&thread_count, 1);
|
|
PyOS_snprintf(name, sizeof(name), "python thread (%d)", this_thread);
|
|
|
|
tid = spawn_thread(name, func, NORMAL_PRIORITY, 0, arg);
|
|
if (tid < 0) {
|
|
dprintf(("PyThread_start_new_thread spawn_thread failed: %s\n", strerror(errno)));
|
|
} else {
|
|
success = resume_thread(tid);
|
|
if (success < 0) {
|
|
dprintf(("PyThread_start_new_thread resume_thread failed: %s\n", strerror(errno)));
|
|
}
|
|
}
|
|
|
|
return (success < 0 ? -1 : tid);
|
|
}
|
|
|
|
|
|
long PyThread_get_thread_ident(void)
|
|
{
|
|
return get_thread_id(NULL);
|
|
}
|
|
|
|
|
|
static void do_PyThread_exit_thread(int no_cleanup)
|
|
{
|
|
dprintf(("PyThread_exit_thread called\n"));
|
|
|
|
/* Thread-safe way to read a variable without a mutex: */
|
|
if (atomic_add(&thread_count, 0) == 0) {
|
|
/* No threads around, so exit main(). */
|
|
if (no_cleanup)
|
|
_exit(0);
|
|
else
|
|
exit(0);
|
|
} else {
|
|
/* We're a thread */
|
|
exit_thread(0);
|
|
}
|
|
}
|
|
|
|
|
|
void PyThread_exit_thread(void)
|
|
{
|
|
do_PyThread_exit_thread(0);
|
|
}
|
|
|
|
|
|
void PyThread__exit_thread(void)
|
|
{
|
|
do_PyThread_exit_thread(1);
|
|
}
|
|
|
|
|
|
#ifndef NO_EXIT_PROG
|
|
static void do_PyThread_exit_prog(int status, int no_cleanup)
|
|
{
|
|
dprintf(("PyThread_exit_prog(%d) called\n", status));
|
|
|
|
/* No need to do anything, the threads get torn down if main()exits. */
|
|
if (no_cleanup)
|
|
_exit(status);
|
|
else
|
|
exit(status);
|
|
}
|
|
|
|
|
|
void PyThread_exit_prog(int status)
|
|
{
|
|
do_PyThread_exit_prog(status, 0);
|
|
}
|
|
|
|
|
|
void PyThread__exit_prog(int status)
|
|
{
|
|
do_PyThread_exit_prog(status, 1);
|
|
}
|
|
#endif /* NO_EXIT_PROG */
|
|
|
|
|
|
/*
|
|
* Lock support.
|
|
*
|
|
*/
|
|
|
|
static atomic_t lock_count = 0;
|
|
|
|
PyThread_type_lock PyThread_allocate_lock(void)
|
|
{
|
|
#ifdef FASTLOCK
|
|
fastmutex_t *lock;
|
|
#else
|
|
sem_id sema;
|
|
#endif
|
|
char name[OS_NAME_LENGTH];
|
|
atomic_t this_lock;
|
|
|
|
dprintf(("PyThread_allocate_lock called\n"));
|
|
|
|
#ifdef FASTLOCK
|
|
lock = (fastmutex_t *) malloc(sizeof(fastmutex_t));
|
|
if (lock == NULL) {
|
|
dprintf(("PyThread_allocate_lock failed: out of memory\n"));
|
|
return (PyThread_type_lock) NULL;
|
|
}
|
|
#endif
|
|
this_lock = atomic_add(&lock_count, 1);
|
|
PyOS_snprintf(name, sizeof(name), "python lock (%d)", this_lock);
|
|
|
|
#ifdef FASTLOCK
|
|
if (fastmutex_create(name, lock) < 0) {
|
|
dprintf(("PyThread_allocate_lock failed: %s\n",
|
|
strerror(errno)));
|
|
free(lock);
|
|
lock = NULL;
|
|
}
|
|
dprintf(("PyThread_allocate_lock()-> %p\n", lock));
|
|
return (PyThread_type_lock) lock;
|
|
#else
|
|
sema = create_semaphore(name, 1, 0);
|
|
if (sema < 0) {
|
|
dprintf(("PyThread_allocate_lock failed: %s\n",
|
|
strerror(errno)));
|
|
sema = 0;
|
|
}
|
|
dprintf(("PyThread_allocate_lock()-> %p\n", sema));
|
|
return (PyThread_type_lock) sema;
|
|
#endif
|
|
}
|
|
|
|
|
|
void PyThread_free_lock(PyThread_type_lock lock)
|
|
{
|
|
dprintf(("PyThread_free_lock(%p) called\n", lock));
|
|
|
|
#ifdef FASTLOCK
|
|
if (fastmutex_destroy((fastmutex_t *) lock) < 0) {
|
|
dprintf(("PyThread_free_lock(%p) failed: %s\n", lock,
|
|
strerror(errno)));
|
|
}
|
|
free(lock);
|
|
#else
|
|
if (delete_semaphore((sem_id) lock) < 0) {
|
|
dprintf(("PyThread_free_lock(%p) failed: %s\n", lock,
|
|
strerror(errno)));
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
int PyThread_acquire_lock(PyThread_type_lock lock, int waitflag)
|
|
{
|
|
int retval;
|
|
|
|
dprintf(("PyThread_acquire_lock(%p, %d) called\n", lock,
|
|
waitflag));
|
|
|
|
#ifdef FASTLOCK
|
|
if (waitflag)
|
|
retval = fastmutex_lock((fastmutex_t *) lock);
|
|
else
|
|
retval = fastmutex_timedlock((fastmutex_t *) lock, 0);
|
|
#else
|
|
if (waitflag)
|
|
retval = lock_semaphore((sem_id) lock);
|
|
else
|
|
retval = lock_semaphore_x((sem_id) lock, 1, 0, 0);
|
|
#endif
|
|
if (retval < 0) {
|
|
dprintf(("PyThread_acquire_lock(%p, %d) failed: %s\n",
|
|
lock, waitflag, strerror(errno)));
|
|
}
|
|
dprintf(("PyThread_acquire_lock(%p, %d)-> %d\n", lock, waitflag,
|
|
retval));
|
|
return retval < 0 ? 0 : 1;
|
|
}
|
|
|
|
|
|
void PyThread_release_lock(PyThread_type_lock lock)
|
|
{
|
|
dprintf(("PyThread_release_lock(%p) called\n", lock));
|
|
|
|
#ifdef FASTLOCK
|
|
if (fastmutex_unlock((fastmutex_t *) lock) < 0) {
|
|
dprintf(("PyThread_release_lock(%p) failed: %s\n", lock,
|
|
strerror(errno)));
|
|
}
|
|
#else
|
|
if (unlock_semaphore((sem_id) lock) < 0) {
|
|
dprintf(("PyThread_release_lock(%p) failed: %s\n", lock,
|
|
strerror(errno)));
|
|
}
|
|
#endif
|
|
}
|