123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925 |
- #include <assert.h>
- #include <errno.h>
- #include <signal.h>
- #include <stdint.h>
- #include <string.h>
- #include <unistd.h>
- #include <sys/mman.h>
- #include <sys/param.h>
- #include <tls.h>
- #include <lowlevellock.h>
- #include <link.h>
- #define __getpagesize getpagesize
- #ifndef NEED_SEPARATE_REGISTER_STACK
- # define STACK_VARIABLES void *stackaddr = 0
- # define STACK_VARIABLES_ARGS stackaddr
- # define STACK_VARIABLES_PARMS void *stackaddr
- # define ALLOCATE_STACK_PARMS void **stack
- # define ALLOCATE_STACK(attr, pd) allocate_stack (attr, pd, &stackaddr)
- #else
- # define STACK_VARIABLES void *stackaddr = 0; size_t stacksize
- # define STACK_VARIABLES_ARGS stackaddr, stacksize
- # define STACK_VARIABLES_PARMS void *stackaddr, size_t stacksize
- # define ALLOCATE_STACK_PARMS void **stack, size_t *stacksize
- # define ALLOCATE_STACK(attr, pd) \
- allocate_stack (attr, pd, &stackaddr, &stacksize)
- #endif
- #ifndef STACK_ALIGN
- # define STACK_ALIGN __alignof__ (long double)
- #endif
- #ifndef MINIMAL_REST_STACK
- # define MINIMAL_REST_STACK 4096
- #endif
- #ifndef ARCH_MAP_FLAGS
- # define ARCH_MAP_FLAGS 0
- #endif
- #if defined(TLS_TCB_AT_TP)
- # define TLS_TPADJ(pd) (pd)
- #elif defined(TLS_DTV_AT_TP)
- # define TLS_TPADJ(pd) ((struct pthread *)((char *) (pd) + TLS_PRE_TCB_SIZE))
- #endif
- static size_t stack_cache_maxsize = 40 * 1024 * 1024;
- static size_t stack_cache_actsize;
- static lll_lock_t stack_cache_lock = LLL_LOCK_INITIALIZER;
- static LIST_HEAD (stack_cache);
- static LIST_HEAD (stack_used);
- list_t __stack_user __attribute__ ((nocommon));
- hidden_def (__stack_user)
- #if COLORING_INCREMENT != 0
- static unsigned int nptl_ncreated;
- #endif
- #define FREE_P(descr) ((descr)->tid <= 0)
- static struct pthread *
- get_cached_stack (size_t *sizep, void **memp)
- {
- size_t size = *sizep;
- struct pthread *result = NULL;
- list_t *entry;
- lll_lock (stack_cache_lock);
-
- list_for_each (entry, &stack_cache)
- {
- struct pthread *curr;
- curr = list_entry (entry, struct pthread, list);
- if (FREE_P (curr) && curr->stackblock_size >= size)
- {
- if (curr->stackblock_size == size)
- {
- result = curr;
- break;
- }
- if (result == NULL
- || result->stackblock_size > curr->stackblock_size)
- result = curr;
- }
- }
- if (__builtin_expect (result == NULL, 0)
-
- || __builtin_expect (result->stackblock_size > 4 * size, 0))
- {
-
- lll_unlock (stack_cache_lock);
- return NULL;
- }
-
- list_del (&result->list);
-
- list_add (&result->list, &stack_used);
-
- stack_cache_actsize -= result->stackblock_size;
-
- lll_unlock (stack_cache_lock);
-
- *sizep = result->stackblock_size;
- *memp = result->stackblock;
-
- result->cancelhandling = 0;
- result->cleanup = NULL;
-
- result->nextevent = NULL;
-
- dtv_t *dtv = GET_DTV (TLS_TPADJ (result));
- memset (dtv, '\0', (dtv[-1].counter + 1) * sizeof (dtv_t));
-
- _dl_allocate_tls_init (TLS_TPADJ (result));
- return result;
- }
- static inline void
- __attribute ((always_inline))
- queue_stack (struct pthread *stack)
- {
-
- list_add (&stack->list, &stack_cache);
- stack_cache_actsize += stack->stackblock_size;
- if (__builtin_expect (stack_cache_actsize > stack_cache_maxsize, 0))
- {
-
- list_t *entry;
- list_t *prev;
-
- list_for_each_prev_safe (entry, prev, &stack_cache)
- {
- struct pthread *curr;
- curr = list_entry (entry, struct pthread, list);
- if (FREE_P (curr))
- {
-
- list_del (entry);
-
- stack_cache_actsize -= curr->stackblock_size;
-
- _dl_deallocate_tls (TLS_TPADJ (curr), false);
-
- if (munmap (curr->stackblock, curr->stackblock_size) != 0)
- abort ();
-
- if (stack_cache_actsize <= stack_cache_maxsize)
- break;
- }
- }
- }
- }
- static int
- internal_function
- change_stack_perm (struct pthread *pd
- #ifdef NEED_SEPARATE_REGISTER_STACK
- , size_t pagemask
- #endif
- )
- {
- #ifdef NEED_SEPARATE_REGISTER_STACK
- void *stack = (pd->stackblock
- + (((((pd->stackblock_size - pd->guardsize) / 2)
- & pagemask) + pd->guardsize) & pagemask));
- size_t len = pd->stackblock + pd->stackblock_size - stack;
- #else
- void *stack = pd->stackblock + pd->guardsize;
- size_t len = pd->stackblock_size - pd->guardsize;
- #endif
- if (mprotect (stack, len, PROT_READ | PROT_WRITE | PROT_EXEC) != 0)
- return errno;
- return 0;
- }
- static int
- allocate_stack (const struct pthread_attr *attr, struct pthread **pdp,
- ALLOCATE_STACK_PARMS)
- {
- struct pthread *pd;
- size_t size;
- size_t pagesize_m1 = __getpagesize () - 1;
- void *stacktop;
- assert (attr != NULL);
- assert (powerof2 (pagesize_m1 + 1));
- assert (TCB_ALIGNMENT >= STACK_ALIGN);
-
- size = attr->stacksize ?: __default_stacksize;
-
- if (__builtin_expect (attr->flags & ATTR_FLAG_STACKADDR, 0))
- {
- uintptr_t adj;
-
- if (attr->stacksize != 0
- && attr->stacksize < (__static_tls_size + MINIMAL_REST_STACK))
- return EINVAL;
-
- #if defined(TLS_TCB_AT_TP)
- adj = ((uintptr_t) attr->stackaddr - TLS_TCB_SIZE)
- & __static_tls_align_m1;
- assert (size > adj + TLS_TCB_SIZE);
- #elif defined(TLS_DTV_AT_TP)
- adj = ((uintptr_t) attr->stackaddr - __static_tls_size)
- & __static_tls_align_m1;
- assert (size > adj);
- #endif
-
- #if defined(TLS_TCB_AT_TP)
- pd = (struct pthread *) ((uintptr_t) attr->stackaddr
- - TLS_TCB_SIZE - adj);
- #elif defined(TLS_DTV_AT_TP)
- pd = (struct pthread *) (((uintptr_t) attr->stackaddr
- - __static_tls_size - adj)
- - TLS_PRE_TCB_SIZE);
- #endif
-
- memset (pd, '\0', sizeof (struct pthread));
-
- pd->specific[0] = pd->specific_1stblock;
-
- pd->stackblock = (char *) attr->stackaddr - size;
- pd->stackblock_size = size;
-
- pd->user_stack = true;
-
- pd->header.multiple_threads = 1;
- #ifndef TLS_MULTIPLE_THREADS_IN_TCB
- __pthread_multiple_threads = *__libc_multiple_threads_ptr = 1;
- #endif
- #ifdef NEED_DL_SYSINFO
-
- THREAD_SYSINFO(pd) = THREAD_SELF_SYSINFO;
- #endif
-
- pd->pid = THREAD_GETMEM (THREAD_SELF, pid);
-
- if (_dl_allocate_tls (TLS_TPADJ (pd)) == NULL)
- {
-
- assert (errno == ENOMEM);
- return EAGAIN;
- }
-
- lll_lock (stack_cache_lock);
-
- list_add (&pd->list, &__stack_user);
- lll_unlock (stack_cache_lock);
- }
- else
- {
-
- size_t guardsize;
- size_t reqsize;
- void *mem = 0;
- const int prot = (PROT_READ | PROT_WRITE);
- #if COLORING_INCREMENT != 0
-
- if (size <= 16 * pagesize_m1)
- size += pagesize_m1 + 1;
- #endif
-
- size &= ~__static_tls_align_m1;
- assert (size != 0);
-
- guardsize = (attr->guardsize + pagesize_m1) & ~pagesize_m1;
- if (__builtin_expect (size < (guardsize + __static_tls_size
- + MINIMAL_REST_STACK + pagesize_m1 + 1),
- 0))
-
- return EINVAL;
-
- reqsize = size;
- pd = get_cached_stack (&size, &mem);
- if (pd == NULL)
- {
-
- #if MULTI_PAGE_ALIASING != 0
- if ((size % MULTI_PAGE_ALIASING) == 0)
- size += pagesize_m1 + 1;
- #endif
- mem = mmap (NULL, size, prot,
- MAP_PRIVATE | MAP_ANONYMOUS | ARCH_MAP_FLAGS, -1, 0);
- if (__builtin_expect (mem == MAP_FAILED, 0))
- {
- #ifdef ARCH_RETRY_MMAP
- mem = ARCH_RETRY_MMAP (size);
- if (__builtin_expect (mem == MAP_FAILED, 0))
- #endif
- return errno;
- }
-
- assert (mem != NULL);
- #if COLORING_INCREMENT != 0
-
- unsigned int ncreated = atomic_increment_val (&nptl_ncreated);
-
- size_t coloring = (ncreated * COLORING_INCREMENT) & pagesize_m1;
-
- if (__builtin_expect ((coloring & __static_tls_align_m1) != 0, 0))
- coloring = (((coloring + __static_tls_align_m1)
- & ~(__static_tls_align_m1))
- & ~pagesize_m1);
- #else
-
- # define coloring 0
- #endif
-
- #if defined(TLS_TCB_AT_TP)
- pd = (struct pthread *) ((char *) mem + size - coloring) - 1;
- #elif defined(TLS_DTV_AT_TP)
- pd = (struct pthread *) ((((uintptr_t) mem + size - coloring
- - __static_tls_size)
- & ~__static_tls_align_m1)
- - TLS_PRE_TCB_SIZE);
- #endif
-
- pd->stackblock = mem;
- pd->stackblock_size = size;
-
- pd->specific[0] = pd->specific_1stblock;
-
- pd->header.multiple_threads = 1;
- #ifndef TLS_MULTIPLE_THREADS_IN_TCB
- __pthread_multiple_threads = *__libc_multiple_threads_ptr = 1;
- #endif
- #ifdef NEED_DL_SYSINFO
-
- THREAD_SYSINFO(pd) = THREAD_SELF_SYSINFO;
- #endif
-
- pd->pid = THREAD_GETMEM (THREAD_SELF, pid);
-
- if (_dl_allocate_tls (TLS_TPADJ (pd)) == NULL)
- {
-
- assert (errno == ENOMEM);
-
- (void) munmap (mem, size);
- return EAGAIN;
- }
-
- lll_lock (stack_cache_lock);
-
- list_add (&pd->list, &stack_used);
- lll_unlock (stack_cache_lock);
-
- }
-
- if (__builtin_expect (guardsize > pd->guardsize, 0))
- {
- #ifdef NEED_SEPARATE_REGISTER_STACK
- char *guard = mem + (((size - guardsize) / 2) & ~pagesize_m1);
- #else
- char *guard = mem;
- #endif
- if (mprotect (guard, guardsize, PROT_NONE) != 0)
- {
- int err;
- mprot_error:
- err = errno;
- lll_lock (stack_cache_lock);
-
- list_del (&pd->list);
- lll_unlock (stack_cache_lock);
-
- _dl_deallocate_tls (TLS_TPADJ (pd), false);
-
- (void) munmap (mem, size);
- return err;
- }
- pd->guardsize = guardsize;
- }
- else if (__builtin_expect (pd->guardsize - guardsize > size - reqsize,
- 0))
- {
-
- #ifdef NEED_SEPARATE_REGISTER_STACK
- char *guard = mem + (((size - guardsize) / 2) & ~pagesize_m1);
- char *oldguard = mem + (((size - pd->guardsize) / 2) & ~pagesize_m1);
- if (oldguard < guard
- && mprotect (oldguard, guard - oldguard, prot) != 0)
- goto mprot_error;
- if (mprotect (guard + guardsize,
- oldguard + pd->guardsize - guard - guardsize,
- prot) != 0)
- goto mprot_error;
- #else
- if (mprotect ((char *) mem + guardsize, pd->guardsize - guardsize,
- prot) != 0)
- goto mprot_error;
- #endif
- pd->guardsize = guardsize;
- }
-
- pd->reported_guardsize = guardsize;
- }
-
- pd->lock = LLL_LOCK_INITIALIZER;
-
- *pdp = pd;
- #if defined(TLS_TCB_AT_TP)
-
- stacktop = ((char *) (pd + 1) - __static_tls_size);
- #elif defined(TLS_DTV_AT_TP)
- stacktop = (char *) (pd - 1);
- #endif
- #ifdef NEED_SEPARATE_REGISTER_STACK
- *stack = pd->stackblock;
- *stacksize = stacktop - *stack;
- #else
- *stack = stacktop;
- #endif
- return 0;
- }
- void
- internal_function
- __deallocate_stack (struct pthread *pd)
- {
- lll_lock (stack_cache_lock);
-
- list_del (&pd->list);
-
- if (__builtin_expect (! pd->user_stack, 1))
- (void) queue_stack (pd);
- else
-
- _dl_deallocate_tls (TLS_TPADJ (pd), false);
- lll_unlock (stack_cache_lock);
- }
- int
- internal_function
- __make_stacks_executable (void **stack_endp)
- {
-
- int err = EPERM;
- if (err != 0)
- return err;
- #ifdef NEED_SEPARATE_REGISTER_STACK
- const size_t pagemask = ~(__getpagesize () - 1);
- #endif
- lll_lock (stack_cache_lock);
- list_t *runp;
- list_for_each (runp, &stack_used)
- {
- err = change_stack_perm (list_entry (runp, struct pthread, list)
- #ifdef NEED_SEPARATE_REGISTER_STACK
- , pagemask
- #endif
- );
- if (err != 0)
- break;
- }
-
- if (err == 0)
- list_for_each (runp, &stack_cache)
- {
- err = change_stack_perm (list_entry (runp, struct pthread, list)
- #ifdef NEED_SEPARATE_REGISTER_STACK
- , pagemask
- #endif
- );
- if (err != 0)
- break;
- }
- lll_unlock (stack_cache_lock);
- return err;
- }
- void
- __reclaim_stacks (void)
- {
- struct pthread *self = (struct pthread *) THREAD_SELF;
-
-
- list_t *runp;
- list_for_each (runp, &stack_used)
- {
- struct pthread *curp;
- curp = list_entry (runp, struct pthread, list);
- if (curp != self)
- {
-
- curp->tid = 0;
-
- curp->pid = self->pid;
-
- stack_cache_actsize += curp->stackblock_size;
- }
- }
-
- list_splice (&stack_used, &stack_cache);
-
- list_del (&self->list);
-
- INIT_LIST_HEAD (&stack_used);
- INIT_LIST_HEAD (&__stack_user);
- if (__builtin_expect (THREAD_GETMEM (self, user_stack), 0))
- list_add (&self->list, &__stack_user);
- else
- list_add (&self->list, &stack_used);
-
- __nptl_nthreads = 1;
-
- stack_cache_lock = LLL_LOCK_INITIALIZER;
- }
- #if HP_TIMING_AVAIL
- # undef __find_thread_by_id
- attribute_hidden
- struct pthread *
- __find_thread_by_id (pid_t tid)
- {
- struct pthread *result = NULL;
- lll_lock (stack_cache_lock);
-
- list_t *runp;
- list_for_each (runp, &stack_used)
- {
- struct pthread *curp;
- curp = list_entry (runp, struct pthread, list);
- if (curp->tid == tid)
- {
- result = curp;
- goto out;
- }
- }
-
- list_for_each (runp, &__stack_user)
- {
- struct pthread *curp;
- curp = list_entry (runp, struct pthread, list);
- if (curp->tid == tid)
- {
- result = curp;
- goto out;
- }
- }
- out:
- lll_unlock (stack_cache_lock);
- return result;
- }
- #endif
- int
- attribute_hidden
- __nptl_setxid (struct xid_command *cmdp)
- {
- int result;
- lll_lock (stack_cache_lock);
- __xidcmd = cmdp;
- cmdp->cntr = 0;
- INTERNAL_SYSCALL_DECL (err);
- struct pthread *self = THREAD_SELF;
-
- list_t *runp;
- list_for_each (runp, &stack_used)
- {
- struct pthread *t = list_entry (runp, struct pthread, list);
- if (t != self)
- {
- int val;
- #if __ASSUME_TGKILL
- val = INTERNAL_SYSCALL (tgkill, err, 3,
- THREAD_GETMEM (THREAD_SELF, pid),
- t->tid, SIGSETXID);
- #else
- # ifdef __NR_tgkill
- val = INTERNAL_SYSCALL (tgkill, err, 3,
- THREAD_GETMEM (THREAD_SELF, pid),
- t->tid, SIGSETXID);
- if (INTERNAL_SYSCALL_ERROR_P (val, err)
- && INTERNAL_SYSCALL_ERRNO (val, err) == ENOSYS)
- # endif
- val = INTERNAL_SYSCALL (tkill, err, 2, t->tid, SIGSETXID);
- #endif
- if (!INTERNAL_SYSCALL_ERROR_P (val, err))
- atomic_increment (&cmdp->cntr);
- }
- }
-
- list_for_each (runp, &__stack_user)
- {
- struct pthread *t = list_entry (runp, struct pthread, list);
- if (t != self)
- {
- int val;
- #if __ASSUME_TGKILL
- val = INTERNAL_SYSCALL (tgkill, err, 3,
- THREAD_GETMEM (THREAD_SELF, pid),
- t->tid, SIGSETXID);
- #else
- # ifdef __NR_tgkill
- val = INTERNAL_SYSCALL (tgkill, err, 3,
- THREAD_GETMEM (THREAD_SELF, pid),
- t->tid, SIGSETXID);
- if (INTERNAL_SYSCALL_ERROR_P (val, err)
- && INTERNAL_SYSCALL_ERRNO (val, err) == ENOSYS)
- # endif
- val = INTERNAL_SYSCALL (tkill, err, 2, t->tid, SIGSETXID);
- #endif
- if (!INTERNAL_SYSCALL_ERROR_P (val, err))
- atomic_increment (&cmdp->cntr);
- }
- }
- int cur = cmdp->cntr;
- while (cur != 0)
- {
- lll_futex_wait (&cmdp->cntr, cur);
- cur = cmdp->cntr;
- }
-
- result = INTERNAL_SYSCALL_NCS (cmdp->syscall_no, err, 3,
- cmdp->id[0], cmdp->id[1], cmdp->id[2]);
- if (INTERNAL_SYSCALL_ERROR_P (result, err))
- {
- __set_errno (INTERNAL_SYSCALL_ERRNO (result, err));
- result = -1;
- }
- lll_unlock (stack_cache_lock);
- return result;
- }
- static inline void __attribute__((always_inline))
- init_one_static_tls (struct pthread *curp, struct link_map *map)
- {
- dtv_t *dtv = GET_DTV (TLS_TPADJ (curp));
- # if defined(TLS_TCB_AT_TP)
- void *dest = (char *) curp - map->l_tls_offset;
- # elif defined(TLS_DTV_AT_TP)
- void *dest = (char *) curp + map->l_tls_offset + TLS_PRE_TCB_SIZE;
- # else
- # error "Either TLS_TCB_AT_TP or TLS_DTV_AT_TP must be defined"
- # endif
-
- dtv[map->l_tls_modid].pointer.val = dest;
- dtv[map->l_tls_modid].pointer.is_static = true;
-
- memset (mempcpy (dest, map->l_tls_initimage, map->l_tls_initimage_size),
- '\0', map->l_tls_blocksize - map->l_tls_initimage_size);
- }
- void
- attribute_hidden
- __pthread_init_static_tls (struct link_map *map)
- {
- lll_lock (stack_cache_lock);
-
- list_t *runp;
- list_for_each (runp, &stack_used)
- init_one_static_tls (list_entry (runp, struct pthread, list), map);
-
- list_for_each (runp, &__stack_user)
- init_one_static_tls (list_entry (runp, struct pthread, list), map);
- lll_unlock (stack_cache_lock);
- }
|