X-Git-Url: http://git.harvie.cz/?a=blobdiff_plain;f=c%2Fpthread_extra%2Fpthread_pause.c;h=a9ce5dce3c487e87613fff24d90c39fe045f5741;hb=3109d0d18e330e166a45042a2d363ec1b285c493;hp=4107b38004fac9c2cb1686b91ff546d157ca7518;hpb=70bd1c94dee46dac0885831ccce3d1840388656e;p=mirrors%2FPrograms.git diff --git a/c/pthread_extra/pthread_pause.c b/c/pthread_extra/pthread_pause.c index 4107b38..a9ce5dc 100644 --- a/c/pthread_extra/pthread_pause.c +++ b/c/pthread_extra/pthread_pause.c @@ -3,6 +3,7 @@ #include #include #include +#include #include #include #include @@ -12,29 +13,35 @@ #include //#include +//Mutex that ensures proper serialization of (un)pause calls +//pthread_mutex_t pthread_pause_mutex = PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP; + +//Semaphore that ensures proper serialization of (un)pause signals +sem_t pthread_pause_sem; +//Once control to init the semaphore (and possibly other stuff) +pthread_once_t pthread_pause_once_ctrl = PTHREAD_ONCE_INIT; +void pthread_pause_once(void) { + sem_init(&pthread_pause_sem, 0, 1); +} +void pthread_pause_init() { pthread_once(&pthread_pause_once_ctrl, &pthread_pause_once); } + ///When this variable is nonzero, only referenced thread is allowed to run -///Access has to be protected by pthread_user_data_lock() -pthread_t pthread_pause_holder = 0; +///Access has to be protected by pthread_user_data_lock() and pthread_pause_sem; +pthread_t pthread_pause_holder = PTHREAD_XNULL; void pthread_pause_handler(const int signal, siginfo_t *info, void *ptr) { (void)signal; (void)info; (void)ptr; int run = info->si_value.sival_int; //(void)td; - //Do nothing when there are more signals pending (to cleanup the queue) - sigset_t pending; - sigpending(&pending); - if(sigismember(&pending, PTHREAD_XSIG_STOP)) return; + //Post semaphore to confirm that signal is handled + sem_post(&pthread_pause_sem); //Keep waiting for signals until we are supposed to be running - sigset_t sigset; - sigfillset(&sigset); - sigdelset(&sigset, PTHREAD_XSIG_STOP); - - //printf("RCV: %p = %p\n", (void *)pthread_user_data_internal(pthread_self()), (void *)td); - - //if(!pthread_user_data_internal(pthread_self())->running) { if(!run) { + sigset_t sigset; + sigfillset(&sigset); + sigdelset(&sigset, PTHREAD_XSIG_STOP); sigsuspend(&sigset); } } @@ -46,8 +53,10 @@ void pthread_pause_enable() { //Nesting signals too deep is not good for stack //You can get runtime stats using following command: //grep -i sig /proc/$(pgrep binary)/status - struct rlimit sigq = {.rlim_cur = 32, .rlim_max=32}; - setrlimit(RLIMIT_SIGPENDING, &sigq); + //struct rlimit sigq = {.rlim_cur = 32, .rlim_max=32}; + //setrlimit(RLIMIT_SIGPENDING, &sigq); + + pthread_pause_init(); //Make sure semaphore is init'd //Prepare signal mask sigset_t sigset; @@ -59,7 +68,7 @@ void pthread_pause_enable() { const struct sigaction pause_sa = { .sa_sigaction = pthread_pause_handler, .sa_mask = sigset, - .sa_flags = SA_SIGINFO, + .sa_flags = SA_SIGINFO | SA_RESTART, .sa_restorer = NULL }; sigaction(PTHREAD_XSIG_STOP, &pause_sa, NULL); @@ -72,11 +81,17 @@ void pthread_pause_disable() { //Add thread to internal registry //pthread_user_data_internal(pthread_self()); + pthread_pause_init(); //Make sure semaphore is init'd + //Block signal sigset_t sigset; sigemptyset(&sigset); sigaddset(&sigset, PTHREAD_XSIG_STOP); + + //Make sure all signals are dispatched before we block them + sem_wait(&pthread_pause_sem); pthread_sigmask(SIG_BLOCK, &sigset, NULL); + sem_post(&pthread_pause_sem); } /* @@ -92,13 +107,20 @@ int pthread_pause_reschedule(pthread_t thread) { */ int pthread_pause_reschedule(pthread_t thread) { - //Decide if the thread should run + //Decide if the thread should run and signal it + + //Wait for semaphore which means signal queue is empty + pthread_pause_init(); //Make sure semaphore is init'd + sem_wait(&pthread_pause_sem); + + //Only call this if you already acquired pthread_pause_sem semaphore!!!! + //Otherwise call pthread_pause_reschedule() pthread_user_data_lock(); //Check if thread has running flag int run = (pthread_user_data_internal(thread)->running); //Check if privileged (single thread) mode is active - if((pthread_pause_holder != 0) && !pthread_equal(pthread_pause_holder, thread)) { + if((pthread_pause_holder != PTHREAD_XNULL) && !pthread_equal(pthread_pause_holder, thread)) { run = 0; } pthread_user_data_unlock(); @@ -108,6 +130,11 @@ int pthread_pause_reschedule(pthread_t thread) { while(pthread_sigqueue(thread, PTHREAD_XSIG_STOP, (const union sigval){.sival_int=run} ) == EAGAIN) usleep(1000); + + //Wait for signal to be delivered + sem_wait(&pthread_pause_sem); + sem_post(&pthread_pause_sem); + return 0; } @@ -121,8 +148,8 @@ int pthread_pause(pthread_t thread) { //Set thread as paused and notify it via signal (wait when queue full) pthread_user_data_lock(); pthread_user_data_internal(thread)->running = 0; - pthread_user_data_unlock(); pthread_pause_reschedule(thread); + pthread_user_data_unlock(); return 0; } @@ -130,26 +157,26 @@ int pthread_unpause(pthread_t thread) { //Set thread as running and notify it via signal (wait when queue full) pthread_user_data_lock(); pthread_user_data_internal(thread)->running = 1; - pthread_user_data_unlock(); pthread_pause_reschedule(thread); + pthread_user_data_unlock(); return 0; } int pthread_pause_all() { pthread_user_data_lock(); - if(pthread_pause_holder!=0) assert(pthread_equal(pthread_pause_holder, pthread_self())); + if(pthread_pause_holder!=PTHREAD_XNULL) assert(pthread_equal(pthread_pause_holder, pthread_self())); pthread_pause_holder = pthread_self(); + pthread_user_data_internal_iterate(&pthread_pause_reschedule, NULL); pthread_user_data_unlock(); - //todo reschedule all return 0; } int pthread_unpause_all() { pthread_user_data_lock(); - if(pthread_pause_holder!=0) assert(pthread_equal(pthread_pause_holder, pthread_self())); - pthread_pause_holder = 0; + if(pthread_pause_holder!=PTHREAD_XNULL) assert(pthread_equal(pthread_pause_holder, pthread_self())); + pthread_pause_holder = PTHREAD_XNULL; + pthread_user_data_internal_iterate(&pthread_pause_reschedule, NULL); pthread_user_data_unlock(); - //todo reschedule return 0; }