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 | d0ba098 | 2013-04-26 16:16:04 -0400 | [diff] [blame^] | 15 | sigset_t set; |
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 | 6e531f9 | 2013-04-26 16:04:30 -0400 | [diff] [blame] | 35 | /* Block all signals before decrementing the live thread count. |
| 36 | * This is important to ensure that dynamically allocated TLS |
| 37 | * is not under-allocated/over-committed, and possibly for other |
| 38 | * reasons as well. */ |
Rich Felker | d0ba098 | 2013-04-26 16:16:04 -0400 | [diff] [blame^] | 39 | __syscall(SYS_rt_sigprocmask, SIG_BLOCK, SIGALL_SET, &set, _NSIG/8); |
Rich Felker | 23f21c3 | 2013-04-26 15:47:44 -0400 | [diff] [blame] | 40 | |
Rich Felker | d0ba098 | 2013-04-26 16:16:04 -0400 | [diff] [blame^] | 41 | /* It's impossible to determine whether this is "the last thread" |
| 42 | * until performing the atomic decrement, since multiple threads |
| 43 | * could exit at the same time. For the last thread, revert the |
| 44 | * decrement and unblock signals to give the atexit handlers and |
| 45 | * stdio cleanup code a consistent state. */ |
| 46 | if (a_fetch_add(&libc.threads_minus_1, -1)==0) { |
| 47 | libc.threads_minus_1 = 0; |
| 48 | __syscall(SYS_rt_sigprocmask, SIG_SETMASK, &set, 0, _NSIG/8); |
| 49 | exit(0); |
| 50 | } |
Rich Felker | fb11b6b | 2011-02-19 10:38:57 -0500 | [diff] [blame] | 51 | |
Rich Felker | 5fcebcd | 2011-03-10 18:31:37 -0500 | [diff] [blame] | 52 | if (self->detached && self->map_base) { |
Rich Felker | 6e531f9 | 2013-04-26 16:04:30 -0400 | [diff] [blame] | 53 | /* Detached threads must avoid the kernel clear_child_tid |
| 54 | * feature, since the virtual address will have been |
| 55 | * unmapped and possibly already reused by a new mapping |
| 56 | * at the time the kernel would perform the write. In |
| 57 | * the case of threads that started out detached, the |
| 58 | * initial clone flags are correct, but if the thread was |
| 59 | * detached later (== 2), we need to clear it here. */ |
| 60 | if (self->detached == 2) __syscall(SYS_set_tid_address, 0); |
| 61 | |
| 62 | /* The following call unmaps the thread's stack mapping |
| 63 | * and then exits without touching the stack. */ |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 64 | __unmapself(self->map_base, self->map_size); |
Rich Felker | 5fcebcd | 2011-03-10 18:31:37 -0500 | [diff] [blame] | 65 | } |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 66 | |
Rich Felker | 0c05bd3 | 2012-09-06 23:34:10 -0400 | [diff] [blame] | 67 | for (;;) __syscall(SYS_exit, 0); |
Rich Felker | 1a9a2ff | 2011-02-13 19:58:30 -0500 | [diff] [blame] | 68 | } |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 69 | |
Rich Felker | cfd892f | 2012-05-23 14:13:54 -0400 | [diff] [blame] | 70 | void __do_cleanup_push(struct __ptcb *cb) |
Rich Felker | 5f37fc1 | 2011-08-03 19:57:46 -0400 | [diff] [blame] | 71 | { |
| 72 | struct pthread *self = pthread_self(); |
| 73 | cb->__next = self->cancelbuf; |
| 74 | self->cancelbuf = cb; |
| 75 | } |
| 76 | |
Rich Felker | cfd892f | 2012-05-23 14:13:54 -0400 | [diff] [blame] | 77 | void __do_cleanup_pop(struct __ptcb *cb) |
Rich Felker | 5f37fc1 | 2011-08-03 19:57:46 -0400 | [diff] [blame] | 78 | { |
Rich Felker | afc35d5 | 2012-02-09 02:33:08 -0500 | [diff] [blame] | 79 | __pthread_self()->cancelbuf = cb->__next; |
Rich Felker | 5f37fc1 | 2011-08-03 19:57:46 -0400 | [diff] [blame] | 80 | } |
| 81 | |
Rich Felker | 3f72cda | 2011-09-18 10:14:37 -0400 | [diff] [blame] | 82 | static int start(void *p) |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 83 | { |
Rich Felker | 3f72cda | 2011-09-18 10:14:37 -0400 | [diff] [blame] | 84 | pthread_t self = p; |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 85 | if (self->startlock[0]) { |
| 86 | __wait(self->startlock, 0, 1, 1); |
| 87 | if (self->startlock[0]) { |
| 88 | self->detached = 2; |
| 89 | pthread_exit(0); |
| 90 | } |
| 91 | __syscall(SYS_rt_sigprocmask, SIG_SETMASK, |
Rich Felker | ccc7b4c | 2013-03-26 23:07:31 -0400 | [diff] [blame] | 92 | self->sigmask, 0, _NSIG/8); |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 93 | } |
Rich Felker | 99b8a25 | 2011-05-07 23:23:58 -0400 | [diff] [blame] | 94 | if (self->unblock_cancel) |
Rich Felker | 2f43704 | 2012-08-09 22:52:13 -0400 | [diff] [blame] | 95 | __syscall(SYS_rt_sigprocmask, SIG_UNBLOCK, |
Rich Felker | ccc7b4c | 2013-03-26 23:07:31 -0400 | [diff] [blame] | 96 | SIGPT_SET, 0, _NSIG/8); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 97 | pthread_exit(self->start(self->start_arg)); |
Rich Felker | 3f72cda | 2011-09-18 10:14:37 -0400 | [diff] [blame] | 98 | return 0; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 99 | } |
| 100 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 101 | #define ROUND(x) (((x)+PAGE_SIZE-1)&-PAGE_SIZE) |
| 102 | |
| 103 | /* pthread_key_create.c overrides this */ |
| 104 | static const size_t dummy = 0; |
| 105 | weak_alias(dummy, __pthread_tsd_size); |
| 106 | |
Rich Felker | dba68bf | 2011-07-30 08:02:14 -0400 | [diff] [blame] | 107 | static FILE *const dummy_file = 0; |
| 108 | weak_alias(dummy_file, __stdin_used); |
| 109 | weak_alias(dummy_file, __stdout_used); |
| 110 | weak_alias(dummy_file, __stderr_used); |
| 111 | |
| 112 | static void init_file_lock(FILE *f) |
| 113 | { |
| 114 | if (f && f->lock<0) f->lock = 0; |
| 115 | } |
| 116 | |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 117 | void *__copy_tls(unsigned char *); |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 118 | |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 119 | 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] | 120 | { |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 121 | int ret; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 122 | size_t size, guard; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 123 | struct pthread *self = pthread_self(), *new; |
Rich Felker | 14a835b | 2013-03-31 23:25:55 -0400 | [diff] [blame] | 124 | unsigned char *map = 0, *stack = 0, *tsd = 0, *stack_limit; |
Rich Felker | 92f8396 | 2012-07-11 23:36:46 -0400 | [diff] [blame] | 125 | unsigned flags = 0x7d8f00; |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 126 | int do_sched = 0; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 127 | pthread_attr_t attr = {0}; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 128 | |
Rich Felker | 7fd3995 | 2011-04-03 16:15:15 -0400 | [diff] [blame] | 129 | if (!self) return ENOSYS; |
Rich Felker | 9080cc1 | 2011-04-17 16:53:54 -0400 | [diff] [blame] | 130 | if (!libc.threaded) { |
Rich Felker | dba68bf | 2011-07-30 08:02:14 -0400 | [diff] [blame] | 131 | for (FILE *f=libc.ofl_head; f; f=f->next) |
| 132 | init_file_lock(f); |
| 133 | init_file_lock(__stdin_used); |
| 134 | init_file_lock(__stdout_used); |
| 135 | init_file_lock(__stderr_used); |
Rich Felker | 9080cc1 | 2011-04-17 16:53:54 -0400 | [diff] [blame] | 136 | libc.threaded = 1; |
| 137 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 138 | if (attrp) attr = *attrp; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 139 | |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 140 | __acquire_ptc(); |
| 141 | |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 142 | if (attr._a_stackaddr) { |
| 143 | size_t need = libc.tls_size + __pthread_tsd_size; |
| 144 | size = attr._a_stacksize + DEFAULT_STACK_SIZE; |
| 145 | stack = (void *)(attr._a_stackaddr & -16); |
Rich Felker | ced6499 | 2013-04-06 01:15:08 -0400 | [diff] [blame] | 146 | stack_limit = (void *)(attr._a_stackaddr - size); |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 147 | /* Use application-provided stack for TLS only when |
| 148 | * it does not take more than ~12% or 2k of the |
| 149 | * application's stack space. */ |
| 150 | if (need < size/8 && need < 2048) { |
| 151 | tsd = stack - __pthread_tsd_size; |
| 152 | stack = tsd - libc.tls_size; |
| 153 | } else { |
| 154 | size = ROUND(need); |
| 155 | guard = 0; |
Rich Felker | 819006a | 2012-06-09 19:53:29 -0400 | [diff] [blame] | 156 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 157 | } else { |
| 158 | guard = ROUND(DEFAULT_GUARD_SIZE + attr._a_guardsize); |
| 159 | size = guard + ROUND(DEFAULT_STACK_SIZE + attr._a_stacksize |
| 160 | + libc.tls_size + __pthread_tsd_size); |
| 161 | } |
| 162 | |
| 163 | if (!tsd) { |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 164 | if (guard) { |
| 165 | map = mmap(0, size, PROT_NONE, MAP_PRIVATE|MAP_ANON, -1, 0); |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame] | 166 | if (map == MAP_FAILED) goto fail; |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 167 | if (mprotect(map+guard, size-guard, PROT_READ|PROT_WRITE)) { |
| 168 | munmap(map, size); |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame] | 169 | goto fail; |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 170 | } |
| 171 | } else { |
| 172 | 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] | 173 | if (map == MAP_FAILED) goto fail; |
Rich Felker | 8431d79 | 2012-10-04 16:35:46 -0400 | [diff] [blame] | 174 | } |
Rich Felker | 819006a | 2012-06-09 19:53:29 -0400 | [diff] [blame] | 175 | tsd = map + size - __pthread_tsd_size; |
Rich Felker | 14a835b | 2013-03-31 23:25:55 -0400 | [diff] [blame] | 176 | if (!stack) { |
| 177 | stack = tsd - libc.tls_size; |
| 178 | stack_limit = map + guard; |
| 179 | } |
Rich Felker | 11e4b92 | 2011-05-07 23:39:48 -0400 | [diff] [blame] | 180 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 181 | |
| 182 | new = __copy_tls(tsd - libc.tls_size); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 183 | new->map_base = map; |
| 184 | new->map_size = size; |
Rich Felker | 14a835b | 2013-03-31 23:25:55 -0400 | [diff] [blame] | 185 | new->stack = stack; |
| 186 | new->stack_size = stack - stack_limit; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 187 | new->pid = self->pid; |
| 188 | new->errno_ptr = &new->errno_val; |
| 189 | new->start = entry; |
| 190 | new->start_arg = arg; |
| 191 | new->self = new; |
| 192 | new->tsd = (void *)tsd; |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 193 | if (attr._a_detach) { |
Rich Felker | 92f8396 | 2012-07-11 23:36:46 -0400 | [diff] [blame] | 194 | new->detached = 1; |
| 195 | flags -= 0x200000; |
| 196 | } |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 197 | if (attr._a_sched) { |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 198 | do_sched = new->startlock[0] = 1; |
| 199 | __syscall(SYS_rt_sigprocmask, SIG_BLOCK, |
Rich Felker | ccc7b4c | 2013-03-26 23:07:31 -0400 | [diff] [blame] | 200 | SIGALL_SET, self->sigmask, _NSIG/8); |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 201 | } |
Rich Felker | bf619d8 | 2011-03-29 12:58:22 -0400 | [diff] [blame] | 202 | new->unblock_cancel = self->cancel; |
Rich Felker | 0a96a37 | 2012-10-07 21:43:46 -0400 | [diff] [blame] | 203 | new->canary = self->canary; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 204 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 205 | a_inc(&libc.threads_minus_1); |
Rich Felker | 9ec4283 | 2012-10-15 18:51:53 -0400 | [diff] [blame] | 206 | 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] | 207 | |
Rich Felker | dcd6037 | 2012-10-05 11:51:50 -0400 | [diff] [blame] | 208 | __release_ptc(); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 209 | |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 210 | if (do_sched) { |
| 211 | __syscall(SYS_rt_sigprocmask, SIG_SETMASK, |
Rich Felker | ccc7b4c | 2013-03-26 23:07:31 -0400 | [diff] [blame] | 212 | new->sigmask, 0, _NSIG/8); |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 213 | } |
| 214 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 215 | if (ret < 0) { |
| 216 | a_dec(&libc.threads_minus_1); |
Rich Felker | 077549e | 2013-02-01 22:23:24 -0500 | [diff] [blame] | 217 | if (map) munmap(map, size); |
Rich Felker | 5966680 | 2011-02-15 02:20:21 -0500 | [diff] [blame] | 218 | return EAGAIN; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 219 | } |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 220 | |
| 221 | if (do_sched) { |
| 222 | ret = __syscall(SYS_sched_setscheduler, new->tid, |
Rich Felker | d514264 | 2013-02-01 22:10:40 -0500 | [diff] [blame] | 223 | attr._a_policy, &attr._a_prio); |
Rich Felker | 1e21e78 | 2012-11-11 15:38:04 -0500 | [diff] [blame] | 224 | a_store(new->startlock, ret<0 ? 2 : 0); |
| 225 | __wake(new->startlock, 1, 1); |
| 226 | if (ret < 0) return -ret; |
| 227 | } |
| 228 | |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 229 | *res = new; |
| 230 | return 0; |
Rich Felker | 72768ea | 2013-02-01 22:25:19 -0500 | [diff] [blame] | 231 | fail: |
| 232 | __release_ptc(); |
| 233 | return EAGAIN; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 234 | } |