Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 1 | #include "pthread_impl.h" |
Rich Felker | dba68bf | 2011-07-30 08:02:14 -0400 | [diff] [blame] | 2 | #include "stdio_impl.h" |
Rich Felker | efd4d87 | 2012-11-08 17:04:20 -0500 | [diff] [blame] | 3 | #include <sys/mman.h> |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 4 | |
Rich Felker | b2486a8 | 2011-04-06 20:27:07 -0400 | [diff] [blame] | 5 | static void dummy_0() |
| 6 | { |
| 7 | } |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 8 | weak_alias(dummy_0, __acquire_ptc); |
| 9 | weak_alias(dummy_0, __release_ptc); |
Rich Felker | a6054e3 | 2011-04-19 23:09:14 -0400 | [diff] [blame] | 10 | weak_alias(dummy_0, __pthread_tsd_run_dtors); |
Rich Felker | fd80cfa | 2011-04-03 02:33:50 -0400 | [diff] [blame] | 11 | |
Rich Felker | 0c05bd3 | 2012-09-06 23:34:10 -0400 | [diff] [blame] | 12 | _Noreturn void pthread_exit(void *result) |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 13 | { |
Rich Felker | 1ebde9c | 2011-04-17 17:06:05 -0400 | [diff] [blame] | 14 | pthread_t self = pthread_self(); |
Rich Felker | feee989 | 2011-04-17 11:43:03 -0400 | [diff] [blame] | 15 | int n; |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 16 | |
Rich Felker | afc35d5 | 2012-02-09 02:33:08 -0500 | [diff] [blame] | 17 | self->result = result; |
| 18 | |
| 19 | while (self->cancelbuf) { |
| 20 | void (*f)(void *) = self->cancelbuf->__f; |
| 21 | void *x = self->cancelbuf->__x; |
| 22 | self->cancelbuf = self->cancelbuf->__next; |
| 23 | f(x); |
Rich Felker | 1ebde9c | 2011-04-17 17:06:05 -0400 | [diff] [blame] | 24 | } |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 25 | |
Rich Felker | a6054e3 | 2011-04-19 23:09:14 -0400 | [diff] [blame] | 26 | __pthread_tsd_run_dtors(); |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 27 | |
Rich Felker | bbbe87e | 2012-07-12 11:23:43 -0400 | [diff] [blame] | 28 | __lock(self->exitlock); |
Rich Felker | f58c8a0 | 2011-06-14 01:25:17 -0400 | [diff] [blame] | 29 | |
Rich Felker | 5fcebcd | 2011-03-10 18:31:37 -0500 | [diff] [blame] | 30 | /* Mark this thread dead before decrementing count */ |
Rich Felker | bbbe87e | 2012-07-12 11:23:43 -0400 | [diff] [blame] | 31 | __lock(self->killlock); |
Rich Felker | 5fcebcd | 2011-03-10 18:31:37 -0500 | [diff] [blame] | 32 | self->dead = 1; |
Rich Felker | bbbe87e | 2012-07-12 11:23:43 -0400 | [diff] [blame] | 33 | __unlock(self->killlock); |
Rich Felker | 19eb13b | 2011-02-19 11:04:36 -0500 | [diff] [blame] | 34 | |
Rich Felker | feee989 | 2011-04-17 11:43:03 -0400 | [diff] [blame] | 35 | do n = libc.threads_minus_1; |
| 36 | while (n && a_cas(&libc.threads_minus_1, n, n-1)!=n); |
| 37 | if (!n) exit(0); |
Rich Felker | fb11b6b | 2011-02-19 10:38:57 -0500 | [diff] [blame] | 38 | |
Rich Felker | 5fcebcd | 2011-03-10 18:31:37 -0500 | [diff] [blame] | 39 | if (self->detached && self->map_base) { |
Rich Felker | 92f8396 | 2012-07-11 23:36:46 -0400 | [diff] [blame] | 40 | if (self->detached == 2) |
| 41 | __syscall(SYS_set_tid_address, 0); |
Rich Felker | 2f43704 | 2012-08-09 22:52:13 -0400 | [diff] [blame] | 42 | __syscall(SYS_rt_sigprocmask, SIG_BLOCK, |
| 43 | SIGALL_SET, 0, __SYSCALL_SSLEN); |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 44 | __unmapself(self->map_base, self->map_size); |
Rich Felker | 5fcebcd | 2011-03-10 18:31:37 -0500 | [diff] [blame] | 45 | } |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 46 | |
Rich Felker | 0c05bd3 | 2012-09-06 23:34:10 -0400 | [diff] [blame] | 47 | for (;;) __syscall(SYS_exit, 0); |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 48 | } |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 49 | |
Rich Felker | cfd892f | 2012-05-23 14:13:54 -0400 | [diff] [blame] | 50 | void __do_cleanup_push(struct __ptcb *cb) |
Rich Felker | 5f37fc1 | 2011-08-03 19:57:46 -0400 | [diff] [blame] | 51 | { |
| 52 | struct pthread *self = pthread_self(); |
| 53 | cb->__next = self->cancelbuf; |
| 54 | self->cancelbuf = cb; |
| 55 | } |
| 56 | |
Rich Felker | cfd892f | 2012-05-23 14:13:54 -0400 | [diff] [blame] | 57 | void __do_cleanup_pop(struct __ptcb *cb) |
Rich Felker | 5f37fc1 | 2011-08-03 19:57:46 -0400 | [diff] [blame] | 58 | { |
Rich Felker | afc35d5 | 2012-02-09 02:33:08 -0500 | [diff] [blame] | 59 | __pthread_self()->cancelbuf = cb->__next; |
Rich Felker | 5f37fc1 | 2011-08-03 19:57:46 -0400 | [diff] [blame] | 60 | } |
| 61 | |
Rich Felker | 3f72cda | 2011-09-18 10:14:37 -0400 | [diff] [blame] | 62 | static int start(void *p) |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 63 | { |
Rich Felker | 3f72cda | 2011-09-18 10:14:37 -0400 | [diff] [blame] | 64 | pthread_t self = p; |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 65 | if (self->startlock[0]) { |
| 66 | __wait(self->startlock, 0, 1, 1); |
| 67 | if (self->startlock[0]) { |
| 68 | self->detached = 2; |
| 69 | pthread_exit(0); |
| 70 | } |
| 71 | __syscall(SYS_rt_sigprocmask, SIG_SETMASK, |
| 72 | self->sigmask, 0, __SYSCALL_SSLEN); |
| 73 | } |
Rich Felker | 99b8a25 | 2011-05-07 23:23:58 -0400 | [diff] [blame] | 74 | if (self->unblock_cancel) |
Rich Felker | 2f43704 | 2012-08-09 22:52:13 -0400 | [diff] [blame] | 75 | __syscall(SYS_rt_sigprocmask, SIG_UNBLOCK, |
| 76 | SIGPT_SET, 0, __SYSCALL_SSLEN); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 77 | pthread_exit(self->start(self->start_arg)); |
Rich Felker | 3f72cda | 2011-09-18 10:14:37 -0400 | [diff] [blame] | 78 | return 0; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 79 | } |
| 80 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 81 | #define ROUND(x) (((x)+PAGE_SIZE-1)&-PAGE_SIZE) |
| 82 | |
| 83 | /* pthread_key_create.c overrides this */ |
| 84 | static const size_t dummy = 0; |
| 85 | weak_alias(dummy, __pthread_tsd_size); |
| 86 | |
Rich Felker | dba68bf | 2011-07-30 08:02:14 -0400 | [diff] [blame] | 87 | static FILE *const dummy_file = 0; |
| 88 | weak_alias(dummy_file, __stdin_used); |
| 89 | weak_alias(dummy_file, __stdout_used); |
| 90 | weak_alias(dummy_file, __stderr_used); |
| 91 | |
| 92 | static void init_file_lock(FILE *f) |
| 93 | { |
| 94 | if (f && f->lock<0) f->lock = 0; |
| 95 | } |
| 96 | |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 97 | void *__copy_tls(unsigned char *); |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 98 | |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 99 | int pthread_create(pthread_t *restrict res, const pthread_attr_t *restrict attrp, void *(*entry)(void *), void *restrict arg) |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 100 | { |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 101 | int ret; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 102 | size_t size, guard; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 103 | struct pthread *self = pthread_self(), *new; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 104 | unsigned char *map = 0, *stack = 0, *tsd = 0; |
Rich Felker | 92f8396 | 2012-07-11 23:36:46 -0400 | [diff] [blame] | 105 | unsigned flags = 0x7d8f00; |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 106 | int do_sched = 0; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 107 | pthread_attr_t attr = {0}; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 108 | |
Rich Felker | 7fd3995 | 2011-04-03 16:15:15 -0400 | [diff] [blame] | 109 | if (!self) return ENOSYS; |
Rich Felker | 9080cc1 | 2011-04-17 16:53:54 -0400 | [diff] [blame] | 110 | if (!libc.threaded) { |
Rich Felker | dba68bf | 2011-07-30 08:02:14 -0400 | [diff] [blame] | 111 | for (FILE *f=libc.ofl_head; f; f=f->next) |
| 112 | init_file_lock(f); |
| 113 | init_file_lock(__stdin_used); |
| 114 | init_file_lock(__stdout_used); |
| 115 | init_file_lock(__stderr_used); |
Rich Felker | 9080cc1 | 2011-04-17 16:53:54 -0400 | [diff] [blame] | 116 | libc.threaded = 1; |
| 117 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 118 | if (attrp) attr = *attrp; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 119 | |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 120 | __acquire_ptc(); |
| 121 | |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 122 | if (attr._a_stackaddr) { |
| 123 | size_t need = libc.tls_size + __pthread_tsd_size; |
| 124 | size = attr._a_stacksize + DEFAULT_STACK_SIZE; |
| 125 | stack = (void *)(attr._a_stackaddr & -16); |
| 126 | /* Use application-provided stack for TLS only when |
| 127 | * it does not take more than ~12% or 2k of the |
| 128 | * application's stack space. */ |
| 129 | if (need < size/8 && need < 2048) { |
| 130 | tsd = stack - __pthread_tsd_size; |
| 131 | stack = tsd - libc.tls_size; |
| 132 | } else { |
| 133 | size = ROUND(need); |
| 134 | guard = 0; |
Rich Felker | 819006a | 2012-06-09 19:53:29 -0400 | [diff] [blame] | 135 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 136 | } else { |
| 137 | guard = ROUND(DEFAULT_GUARD_SIZE + attr._a_guardsize); |
| 138 | size = guard + ROUND(DEFAULT_STACK_SIZE + attr._a_stacksize |
| 139 | + libc.tls_size + __pthread_tsd_size); |
| 140 | } |
| 141 | |
| 142 | if (!tsd) { |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 143 | if (guard) { |
| 144 | map = mmap(0, size, PROT_NONE, MAP_PRIVATE|MAP_ANON, -1, 0); |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame^] | 145 | if (map == MAP_FAILED) goto fail; |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 146 | if (mprotect(map+guard, size-guard, PROT_READ|PROT_WRITE)) { |
| 147 | munmap(map, size); |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame^] | 148 | goto fail; |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 149 | } |
| 150 | } else { |
| 151 | map = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame^] | 152 | if (map == MAP_FAILED) goto fail; |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 153 | } |
Rich Felker | 819006a | 2012-06-09 19:53:29 -0400 | [diff] [blame] | 154 | tsd = map + size - __pthread_tsd_size; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 155 | if (!stack) stack = tsd - libc.tls_size; |
Rich Felker | 11e4b92 | 2011-05-07 23:39:48 -0400 | [diff] [blame] | 156 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 157 | |
| 158 | new = __copy_tls(tsd - libc.tls_size); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 159 | new->map_base = map; |
| 160 | new->map_size = size; |
| 161 | new->pid = self->pid; |
| 162 | new->errno_ptr = &new->errno_val; |
| 163 | new->start = entry; |
| 164 | new->start_arg = arg; |
| 165 | new->self = new; |
| 166 | new->tsd = (void *)tsd; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 167 | if (attr._a_detach) { |
Rich Felker | 92f8396 | 2012-07-11 23:36:46 -0400 | [diff] [blame] | 168 | new->detached = 1; |
| 169 | flags -= 0x200000; |
| 170 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 171 | if (attr._a_sched) { |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 172 | do_sched = new->startlock[0] = 1; |
| 173 | __syscall(SYS_rt_sigprocmask, SIG_BLOCK, |
| 174 | SIGALL_SET, self->sigmask, __SYSCALL_SSLEN); |
| 175 | } |
Rich Felker | bf619d8 | 2011-03-29 12:58:22 -0400 | [diff] [blame] | 176 | new->unblock_cancel = self->cancel; |
Rich Felker | 0a96a37 | 2012-10-07 21:43:46 -0400 | [diff] [blame] | 177 | new->canary = self->canary; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 178 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 179 | a_inc(&libc.threads_minus_1); |
Rich Felker | 9ec4283 | 2012-10-15 18:51:53 -0400 | [diff] [blame] | 180 | ret = __clone(start, stack, flags, new, &new->tid, TP_ADJ(new), &new->tid); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 181 | |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 182 | __release_ptc(); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 183 | |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 184 | if (do_sched) { |
| 185 | __syscall(SYS_rt_sigprocmask, SIG_SETMASK, |
| 186 | new->sigmask, 0, __SYSCALL_SSLEN); |
| 187 | } |
| 188 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 189 | if (ret < 0) { |
| 190 | a_dec(&libc.threads_minus_1); |
Rich Felker | 077549e | 2013-02-01 22:23:24 -0500 | [diff] [blame] | 191 | if (map) munmap(map, size); |
Rich Felker | 5966680 | 2011-02-15 02:20:21 -0500 | [diff] [blame] | 192 | return EAGAIN; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 193 | } |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 194 | |
| 195 | if (do_sched) { |
| 196 | ret = __syscall(SYS_sched_setscheduler, new->tid, |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 197 | attr._a_policy, &attr._a_prio); |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 198 | a_store(new->startlock, ret<0 ? 2 : 0); |
| 199 | __wake(new->startlock, 1, 1); |
| 200 | if (ret < 0) return -ret; |
| 201 | } |
| 202 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 203 | *res = new; |
| 204 | return 0; |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame^] | 205 | fail: |
| 206 | __release_ptc(); |
| 207 | return EAGAIN; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 208 | } |