Showing the changes to sys/src/cmd/python/Python/thread_beos.h (1 of 3706 files). Show all
Unified · Split
| @@ -0,0 +1,287 @@ | ||
| 1 | +#include <kernel/OS.h> | |
| 2 | +#include <support/SupportDefs.h> | |
| 3 | +#include <errno.h> | |
| 4 | + | |
| 5 | +/* ---------------------------------------------------------------------- | |
| 6 | + * Fast locking mechanism described by Benoit Schillings (benoit@be.com) | |
| 7 | + * in the Be Developer's Newsletter, Issue #26 (http://www.be.com/). | |
| 8 | + */ | |
| 9 | +typedef struct benaphore { | |
| 10 | + sem_id _sem; | |
| 11 | + int32 _atom; | |
| 12 | +} benaphore_t; | |
| 13 | + | |
| 14 | +static status_t benaphore_create( const char *name, benaphore_t *ben ); | |
| 15 | +static status_t benaphore_destroy( benaphore_t *ben ); | |
| 16 | +static status_t benaphore_lock( benaphore_t *ben ); | |
| 17 | +static status_t benaphore_timedlock( benaphore_t *ben, bigtime_t micros ); | |
| 18 | +static status_t benaphore_unlock( benaphore_t *ben ); | |
| 19 | + | |
| 20 | +static status_t benaphore_create( const char *name, benaphore_t *ben ) | |
| 21 | +{ | |
| 22 | + if( ben != NULL ) { | |
| 23 | + ben->_atom = 0; | |
| 24 | + ben->_sem = create_sem( 0, name ); | |
| 25 | + | |
| 26 | + if( ben->_sem < B_NO_ERROR ) { | |
| 27 | + return B_BAD_SEM_ID; | |
| 28 | + } | |
| 29 | + } else { | |
| 30 | + return EFAULT; | |
| 31 | + } | |
| 32 | + | |
| 33 | + return EOK; | |
| 34 | +} | |
| 35 | + | |
| 36 | +static status_t benaphore_destroy( benaphore_t *ben ) | |
| 37 | +{ | |
| 38 | + if( ben->_sem >= B_NO_ERROR ) { | |
| 39 | + status_t retval = benaphore_timedlock( ben, 0 ); | |
| 40 | + | |
| 41 | + if( retval == EOK || retval == EWOULDBLOCK ) { | |
| 42 | + status_t del_retval = delete_sem( ben->_sem ); | |
| 43 | + | |
| 44 | + return del_retval; | |
| 45 | + } | |
| 46 | + } | |
| 47 | + | |
| 48 | + return B_BAD_SEM_ID; | |
| 49 | +} | |
| 50 | + | |
| 51 | +static status_t benaphore_lock( benaphore_t *ben ) | |
| 52 | +{ | |
| 53 | + int32 prev = atomic_add( &(ben->_atom), 1 ); | |
| 54 | + | |
| 55 | + if( prev > 0 ) { | |
| 56 | + return acquire_sem( ben->_sem ); | |
| 57 | + } | |
| 58 | + | |
| 59 | + return EOK; | |
| 60 | +} | |
| 61 | + | |
| 62 | +static status_t benaphore_timedlock( benaphore_t *ben, bigtime_t micros ) | |
| 63 | +{ | |
| 64 | + int32 prev = atomic_add( &(ben->_atom), 1 ); | |
| 65 | + | |
| 66 | + if( prev > 0 ) { | |
| 67 | + status_t retval = acquire_sem_etc( ben->_sem, 1, B_TIMEOUT, micros ); | |
| 68 | + | |
| 69 | + switch( retval ) { | |
| 70 | + case B_WOULD_BLOCK: /* Fall through... */ | |
| 71 | + case B_TIMED_OUT: | |
| 72 | + return EWOULDBLOCK; | |
| 73 | + break; | |
| 74 | + case B_OK: | |
| 75 | + return EOK; | |
| 76 | + break; | |
| 77 | + default: | |
| 78 | + return retval; | |
| 79 | + break; | |
| 80 | + } | |
| 81 | + } | |
| 82 | + | |
| 83 | + return EOK; | |
| 84 | +} | |
| 85 | + | |
| 86 | +static status_t benaphore_unlock( benaphore_t *ben ) | |
| 87 | +{ | |
| 88 | + int32 prev = atomic_add( &(ben->_atom), -1 ); | |
| 89 | + | |
| 90 | + if( prev > 1 ) { | |
| 91 | + return release_sem( ben->_sem ); | |
| 92 | + } | |
| 93 | + | |
| 94 | + return EOK; | |
| 95 | +} | |
| 96 | + | |
| 97 | +/* ---------------------------------------------------------------------- | |
| 98 | + * Initialization. | |
| 99 | + */ | |
| 100 | +static void PyThread__init_thread( void ) | |
| 101 | +{ | |
| 102 | + /* Do nothing. */ | |
| 103 | + return; | |
| 104 | +} | |
| 105 | + | |
| 106 | +/* ---------------------------------------------------------------------- | |
| 107 | + * Thread support. | |
| 108 | + * | |
| 109 | + * Only ANSI C, renamed functions here; you can't use K&R on BeOS, | |
| 110 | + * and there's no legacy thread module to support. | |
| 111 | + */ | |
| 112 | + | |
| 113 | +static int32 thread_count = 0; | |
| 114 | + | |
| 115 | +long PyThread_start_new_thread( void (*func)(void *), void *arg ) | |
| 116 | +{ | |
| 117 | + status_t success = 0; | |
| 118 | + thread_id tid; | |
| 119 | + char name[B_OS_NAME_LENGTH]; | |
| 120 | + int32 this_thread; | |
| 121 | + | |
| 122 | + dprintf(("PyThread_start_new_thread called\n")); | |
| 123 | + | |
| 124 | + /* We are so very thread-safe... */ | |
| 125 | + this_thread = atomic_add( &thread_count, 1 ); | |
| 126 | + PyOS_snprintf(name, sizeof(name), | |
| 127 | + "python thread (%d)", this_thread ); | |
| 128 | + | |
| 129 | + tid = spawn_thread( (thread_func)func, name, | |
| 130 | + B_NORMAL_PRIORITY, arg ); | |
| 131 | + if( tid > B_NO_ERROR ) { | |
| 132 | + success = resume_thread( tid ); | |
| 133 | + } | |
| 134 | + | |
| 135 | + return ( success == B_NO_ERROR ? tid : -1 ); | |
| 136 | +} | |
| 137 | + | |
| 138 | +long PyThread_get_thread_ident( void ) | |
| 139 | +{ | |
| 140 | + /* Presumed to return the current thread's ID... */ | |
| 141 | + thread_id tid; | |
| 142 | + tid = find_thread( NULL ); | |
| 143 | + | |
| 144 | + return ( tid != B_NAME_NOT_FOUND ? tid : -1 ); | |
| 145 | +} | |
| 146 | + | |
| 147 | +static void do_PyThread_exit_thread( int no_cleanup ) | |
| 148 | +{ | |
| 149 | + int32 threads; | |
| 150 | + | |
| 151 | + dprintf(("PyThread_exit_thread called\n")); | |
| 152 | + | |
| 153 | + /* Thread-safe way to read a variable without a mutex: */ | |
| 154 | + threads = atomic_add( &thread_count, 0 ); | |
| 155 | + | |
| 156 | + if( threads == 0 ) { | |
| 157 | + /* No threads around, so exit main(). */ | |
| 158 | + if( no_cleanup ) { | |
| 159 | + _exit(0); | |
| 160 | + } else { | |
| 161 | + exit(0); | |
| 162 | + } | |
| 163 | + } else { | |
| 164 | + /* Oh, we're a thread, let's try to exit gracefully... */ | |
| 165 | + exit_thread( B_NO_ERROR ); | |
| 166 | + } | |
| 167 | +} | |
| 168 | + | |
| 169 | +void PyThread_exit_thread( void ) | |
| 170 | +{ | |
| 171 | + do_PyThread_exit_thread(0); | |
| 172 | +} | |
| 173 | + | |
| 174 | +void PyThread__exit_thread( void ) | |
| 175 | +{ | |
| 176 | + do_PyThread_exit_thread(1); | |
| 177 | +} | |
| 178 | + | |
| 179 | +#ifndef NO_EXIT_PROG | |
| 180 | +static void do_PyThread_exit_prog( int status, int no_cleanup ) | |
| 181 | +{ | |
| 182 | + dprintf(("PyThread_exit_prog(%d) called\n", status)); | |
| 183 | + | |
| 184 | + /* No need to do anything, the threads get torn down if main() exits. */ | |
| 185 | + | |
| 186 | + if (no_cleanup) { | |
| 187 | + _exit(status); | |
| 188 | + } else { | |
| 189 | + exit(status); | |
| 190 | + } | |
| 191 | +} | |
| 192 | + | |
| 193 | +void PyThread_exit_prog( int status ) | |
| 194 | +{ | |
| 195 | + do_PyThread_exit_prog(status, 0); | |
| 196 | +} | |
| 197 | + | |
| 198 | +void PyThread__exit_prog( int status ) | |
| 199 | +{ | |
| 200 | + do_PyThread_exit_prog(status, 1); | |
| 201 | +} | |
| 202 | +#endif /* NO_EXIT_PROG */ | |
| 203 | + | |
| 204 | +/* ---------------------------------------------------------------------- | |
| 205 | + * Lock support. | |
| 206 | + */ | |
| 207 | + | |
| 208 | +static int32 lock_count = 0; | |
| 209 | + | |
| 210 | +PyThread_type_lock PyThread_allocate_lock( void ) | |
| 211 | +{ | |
| 212 | + benaphore_t *lock; | |
| 213 | + status_t retval; | |
| 214 | + char name[B_OS_NAME_LENGTH]; | |
| 215 | + int32 this_lock; | |
| 216 | + | |
| 217 | + dprintf(("PyThread_allocate_lock called\n")); | |
| 218 | + | |
| 219 | + lock = (benaphore_t *)malloc( sizeof( benaphore_t ) ); | |
| 220 | + if( lock == NULL ) { | |
| 221 | + /* TODO: that's bad, raise MemoryError */ | |
| 222 | + return (PyThread_type_lock)NULL; | |
| 223 | + } | |
| 224 | + | |
| 225 | + this_lock = atomic_add( &lock_count, 1 ); | |
| 226 | + PyOS_snprintf(name, sizeof(name), "python lock (%d)", this_lock); | |
| 227 | + | |
| 228 | + retval = benaphore_create( name, lock ); | |
| 229 | + if( retval != EOK ) { | |
| 230 | + /* TODO: that's bad, raise an exception */ | |
| 231 | + return (PyThread_type_lock)NULL; | |
| 232 | + } | |
| 233 | + | |
| 234 | + dprintf(("PyThread_allocate_lock() -> %p\n", lock)); | |
| 235 | + return (PyThread_type_lock) lock; | |
| 236 | +} | |
| 237 | + | |
| 238 | +void PyThread_free_lock( PyThread_type_lock lock ) | |
| 239 | +{ | |
| 240 | + status_t retval; | |
| 241 | + | |
| 242 | + dprintf(("PyThread_free_lock(%p) called\n", lock)); | |
| 243 | + | |
| 244 | + retval = benaphore_destroy( (benaphore_t *)lock ); | |
| 245 | + if( retval != EOK ) { | |
| 246 | + /* TODO: that's bad, raise an exception */ | |
| 247 | + return; | |
| 248 | + } | |
| 249 | +} | |
| 250 | + | |
| 251 | +int PyThread_acquire_lock( PyThread_type_lock lock, int waitflag ) | |
| 252 | +{ | |
| 253 | + int success; | |
| 254 | + status_t retval; | |
| 255 | + | |
| 256 | + dprintf(("PyThread_acquire_lock(%p, %d) called\n", lock, waitflag)); | |
| 257 | + | |
| 258 | + if( waitflag ) { | |
| 259 | + retval = benaphore_lock( (benaphore_t *)lock ); | |
| 260 | + } else { | |
| 261 | + retval = benaphore_timedlock( (benaphore_t *)lock, 0 ); | |
| 262 | + } | |
| 263 | + | |
| 264 | + if( retval == EOK ) { | |
| 265 | + success = 1; | |
| 266 | + } else { | |
| 267 | + success = 0; | |
| 268 | + | |
| 269 | + /* TODO: that's bad, raise an exception */ | |
| 270 | + } | |
| 271 | + | |
| 272 | + dprintf(("PyThread_acquire_lock(%p, %d) -> %d\n", lock, waitflag, success)); | |
| 273 | + return success; | |
| 274 | +} | |
| 275 | + | |
| 276 | +void PyThread_release_lock( PyThread_type_lock lock ) | |
| 277 | +{ | |
| 278 | + status_t retval; | |
| 279 | + | |
| 280 | + dprintf(("PyThread_release_lock(%p) called\n", lock)); | |
| 281 | + | |
| 282 | + retval = benaphore_unlock( (benaphore_t *)lock ); | |
| 283 | + if( retval != EOK ) { | |
| 284 | + /* TODO: that's bad, raise an exception */ | |
| 285 | + return; | |
| 286 | + } | |
| 287 | +} | |