Showing the changes to sys/src/cmd/python/Python/thread_pth.h (1 of 3706 files). Show all
Unified · Split
| @@ -0,0 +1,213 @@ | ||
| 1 | + | |
| 2 | +/* GNU pth threads interface | |
| 3 | + http://www.gnu.org/software/pth | |
| 4 | + 2000-05-03 Andy Dustman <andy@dustman.net> | |
| 5 | + | |
| 6 | + Adapted from Posix threads interface | |
| 7 | + 12 May 1997 -- david arnold <davida@pobox.com> | |
| 8 | + */ | |
| 9 | + | |
| 10 | +#include <stdlib.h> | |
| 11 | +#include <string.h> | |
| 12 | +#include <pth.h> | |
| 13 | + | |
| 14 | +/* A pth mutex isn't sufficient to model the Python lock type | |
| 15 | + * because pth mutexes can be acquired multiple times by the | |
| 16 | + * same thread. | |
| 17 | + * | |
| 18 | + * The pth_lock struct implements a Python lock as a "locked?" bit | |
| 19 | + * and a <condition, mutex> pair. In general, if the bit can be acquired | |
| 20 | + * instantly, it is, else the pair is used to block the thread until the | |
| 21 | + * bit is cleared. | |
| 22 | + */ | |
| 23 | + | |
| 24 | +typedef struct { | |
| 25 | + char locked; /* 0=unlocked, 1=locked */ | |
| 26 | + /* a <cond, mutex> pair to handle an acquire of a locked lock */ | |
| 27 | + pth_cond_t lock_released; | |
| 28 | + pth_mutex_t mut; | |
| 29 | +} pth_lock; | |
| 30 | + | |
| 31 | +#define CHECK_STATUS(name) if (status == -1) { printf("%d ", status); perror(name); error = 1; } | |
| 32 | + | |
| 33 | +pth_attr_t PyThread_attr; | |
| 34 | + | |
| 35 | +/* | |
| 36 | + * Initialization. | |
| 37 | + */ | |
| 38 | + | |
| 39 | +static void PyThread__init_thread(void) | |
| 40 | +{ | |
| 41 | + pth_init(); | |
| 42 | + PyThread_attr = pth_attr_new(); | |
| 43 | + pth_attr_set(PyThread_attr, PTH_ATTR_STACK_SIZE, 1<<18); | |
| 44 | + pth_attr_set(PyThread_attr, PTH_ATTR_JOINABLE, FALSE); | |
| 45 | +} | |
| 46 | + | |
| 47 | +/* | |
| 48 | + * Thread support. | |
| 49 | + */ | |
| 50 | + | |
| 51 | + | |
| 52 | +long PyThread_start_new_thread(void (*func)(void *), void *arg) | |
| 53 | +{ | |
| 54 | + pth_t th; | |
| 55 | + dprintf(("PyThread_start_new_thread called\n")); | |
| 56 | + if (!initialized) | |
| 57 | + PyThread_init_thread(); | |
| 58 | + | |
| 59 | + th = pth_spawn(PyThread_attr, | |
| 60 | + (void* (*)(void *))func, | |
| 61 | + (void *)arg | |
| 62 | + ); | |
| 63 | + | |
| 64 | + return th; | |
| 65 | +} | |
| 66 | + | |
| 67 | +long PyThread_get_thread_ident(void) | |
| 68 | +{ | |
| 69 | + volatile pth_t threadid; | |
| 70 | + if (!initialized) | |
| 71 | + PyThread_init_thread(); | |
| 72 | + /* Jump through some hoops for Alpha OSF/1 */ | |
| 73 | + threadid = pth_self(); | |
| 74 | + return (long) *(long *) &threadid; | |
| 75 | +} | |
| 76 | + | |
| 77 | +static void do_PyThread_exit_thread(int no_cleanup) | |
| 78 | +{ | |
| 79 | + dprintf(("PyThread_exit_thread called\n")); | |
| 80 | + if (!initialized) { | |
| 81 | + if (no_cleanup) | |
| 82 | + _exit(0); | |
| 83 | + else | |
| 84 | + exit(0); | |
| 85 | + } | |
| 86 | +} | |
| 87 | + | |
| 88 | +void PyThread_exit_thread(void) | |
| 89 | +{ | |
| 90 | + do_PyThread_exit_thread(0); | |
| 91 | +} | |
| 92 | + | |
| 93 | +void PyThread__exit_thread(void) | |
| 94 | +{ | |
| 95 | + do_PyThread_exit_thread(1); | |
| 96 | +} | |
| 97 | + | |
| 98 | +#ifndef NO_EXIT_PROG | |
| 99 | +static void do_PyThread_exit_prog(int status, int no_cleanup) | |
| 100 | +{ | |
| 101 | + dprintf(("PyThread_exit_prog(%d) called\n", status)); | |
| 102 | + if (!initialized) | |
| 103 | + if (no_cleanup) | |
| 104 | + _exit(status); | |
| 105 | + else | |
| 106 | + exit(status); | |
| 107 | +} | |
| 108 | + | |
| 109 | +void PyThread_exit_prog(int status) | |
| 110 | +{ | |
| 111 | + do_PyThread_exit_prog(status, 0); | |
| 112 | +} | |
| 113 | + | |
| 114 | +void PyThread__exit_prog(int status) | |
| 115 | +{ | |
| 116 | + do_PyThread_exit_prog(status, 1); | |
| 117 | +} | |
| 118 | +#endif /* NO_EXIT_PROG */ | |
| 119 | + | |
| 120 | +/* | |
| 121 | + * Lock support. | |
| 122 | + */ | |
| 123 | +PyThread_type_lock PyThread_allocate_lock(void) | |
| 124 | +{ | |
| 125 | + pth_lock *lock; | |
| 126 | + int status, error = 0; | |
| 127 | + | |
| 128 | + dprintf(("PyThread_allocate_lock called\n")); | |
| 129 | + if (!initialized) | |
| 130 | + PyThread_init_thread(); | |
| 131 | + | |
| 132 | + lock = (pth_lock *) malloc(sizeof(pth_lock)); | |
| 133 | + memset((void *)lock, '\0', sizeof(pth_lock)); | |
| 134 | + if (lock) { | |
| 135 | + lock->locked = 0; | |
| 136 | + status = pth_mutex_init(&lock->mut); | |
| 137 | + CHECK_STATUS("pth_mutex_init"); | |
| 138 | + status = pth_cond_init(&lock->lock_released); | |
| 139 | + CHECK_STATUS("pth_cond_init"); | |
| 140 | + if (error) { | |
| 141 | + free((void *)lock); | |
| 142 | + lock = NULL; | |
| 143 | + } | |
| 144 | + } | |
| 145 | + dprintf(("PyThread_allocate_lock() -> %p\n", lock)); | |
| 146 | + return (PyThread_type_lock) lock; | |
| 147 | +} | |
| 148 | + | |
| 149 | +void PyThread_free_lock(PyThread_type_lock lock) | |
| 150 | +{ | |
| 151 | + pth_lock *thelock = (pth_lock *)lock; | |
| 152 | + | |
| 153 | + dprintf(("PyThread_free_lock(%p) called\n", lock)); | |
| 154 | + | |
| 155 | + free((void *)thelock); | |
| 156 | +} | |
| 157 | + | |
| 158 | +int PyThread_acquire_lock(PyThread_type_lock lock, int waitflag) | |
| 159 | +{ | |
| 160 | + int success; | |
| 161 | + pth_lock *thelock = (pth_lock *)lock; | |
| 162 | + int status, error = 0; | |
| 163 | + | |
| 164 | + dprintf(("PyThread_acquire_lock(%p, %d) called\n", lock, waitflag)); | |
| 165 | + | |
| 166 | + status = pth_mutex_acquire(&thelock->mut, !waitflag, NULL); | |
| 167 | + CHECK_STATUS("pth_mutex_acquire[1]"); | |
| 168 | + success = thelock->locked == 0; | |
| 169 | + if (success) thelock->locked = 1; | |
| 170 | + status = pth_mutex_release( &thelock->mut ); | |
| 171 | + CHECK_STATUS("pth_mutex_release[1]"); | |
| 172 | + | |
| 173 | + if ( !success && waitflag ) { | |
| 174 | + /* continue trying until we get the lock */ | |
| 175 | + | |
| 176 | + /* mut must be locked by me -- part of the condition | |
| 177 | + * protocol */ | |
| 178 | + status = pth_mutex_acquire( &thelock->mut, !waitflag, NULL ); | |
| 179 | + CHECK_STATUS("pth_mutex_acquire[2]"); | |
| 180 | + while ( thelock->locked ) { | |
| 181 | + status = pth_cond_await(&thelock->lock_released, | |
| 182 | + &thelock->mut, NULL); | |
| 183 | + CHECK_STATUS("pth_cond_await"); | |
| 184 | + } | |
| 185 | + thelock->locked = 1; | |
| 186 | + status = pth_mutex_release( &thelock->mut ); | |
| 187 | + CHECK_STATUS("pth_mutex_release[2]"); | |
| 188 | + success = 1; | |
| 189 | + } | |
| 190 | + if (error) success = 0; | |
| 191 | + dprintf(("PyThread_acquire_lock(%p, %d) -> %d\n", lock, waitflag, success)); | |
| 192 | + return success; | |
| 193 | +} | |
| 194 | + | |
| 195 | +void PyThread_release_lock(PyThread_type_lock lock) | |
| 196 | +{ | |
| 197 | + pth_lock *thelock = (pth_lock *)lock; | |
| 198 | + int status, error = 0; | |
| 199 | + | |
| 200 | + dprintf(("PyThread_release_lock(%p) called\n", lock)); | |
| 201 | + | |
| 202 | + status = pth_mutex_acquire( &thelock->mut, 0, NULL ); | |
| 203 | + CHECK_STATUS("pth_mutex_acquire[3]"); | |
| 204 | + | |
| 205 | + thelock->locked = 0; | |
| 206 | + | |
| 207 | + status = pth_mutex_release( &thelock->mut ); | |
| 208 | + CHECK_STATUS("pth_mutex_release[3]"); | |
| 209 | + | |
| 210 | + /* wake up someone (anyone, if any) waiting on the lock */ | |
| 211 | + status = pth_cond_notify( &thelock->lock_released, 0 ); | |
| 212 | + CHECK_STATUS("pth_cond_notify"); | |
| 213 | +} | |