1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
|
#define _GNU_SOURCE #include <pthread.h> #include <stdio.h> #include <stdlib.h> #include <signal.h> #include <string.h> #include <stdlib.h> #include <errno.h>
#define CHECK(x) do { if(!(x)) { fprintf(stderr, "ERROR: %s\nfile=%s, line=%d\n", strerror(errno), __FILE__, __LINE__); exit(1); } } while(0) #define THREAD_CNT 5 pthread_mutex_t sig_mutex; pthread_cond_t sig_cond; int thread_ready_count;
void handler(int sig, siginfo_t *info, void *context) { printf("recv sig = %d: %s, from = %d\n", sig, strsignal(sig), info->si_value.sival_int); }
pthread_t tid2pthread[THREAD_CNT];
int tid2sig(int tid) { return SIGRTMAX - 1 - tid; }
void *fun_thread(void *arg) { int tid = *(int *)arg; free(arg); sigset_t set; CHECK(sigemptyset(&set) == 0); CHECK(sigaddset(&set, tid2sig(tid)) == 0); CHECK(pthread_sigmask(SIG_SETMASK, &set, NULL) == 0); CHECK(sigaction(tid2sig(tid), &(struct sigaction) { .sa_sigaction =handler, .sa_flags = SA_SIGINFO, .sa_mask = 0 }, NULL) == 0); printf("Thread-%d, sigaction %d\n", tid, tid2sig(tid)); CHECK(pthread_mutex_lock(&sig_mutex) == 0); thread_ready_count++; tid2pthread[tid] = pthread_self(); CHECK(pthread_mutex_unlock(&sig_mutex) == 0); CHECK(pthread_cond_broadcast(&sig_cond) == 0);
CHECK(pthread_mutex_lock(&sig_mutex) == 0); while(thread_ready_count < THREAD_CNT){ CHECK(pthread_cond_wait(&sig_cond, &sig_mutex) == 0); } CHECK(pthread_mutex_unlock(&sig_mutex) == 0); int sig = tid2sig((tid + 1) % THREAD_CNT); printf("Thread-%d send %d(%s) to %d\n", tid, sig, strsignal(sig), (tid + 1) % THREAD_CNT); CHECK(pthread_sigqueue(tid2pthread[(tid + 1) % THREAD_CNT], sig, (union sigval) { .sival_int=tid }) == 0); do { printf("Thread-%d sig not pending %d\n", tid, tid2sig(tid)); sleep(1); CHECK(sigemptyset(&set) == 0); CHECK(sigpending(&set) == 0); } while(sigismember(&set, tid2sig(tid)) == 0); printf("Thread-%d sig pending %d\n", tid, tid2sig(tid)); CHECK(sigemptyset(&set) == 0); CHECK(pthread_sigmask(SIG_SETMASK, &set, NULL) == 0); CHECK(sigpending(&set) == 0); sig = -1; while(sig != tid2sig(tid)) CHECK(sigwait(&set, &sig) == 0); printf("Thread-%d exit\n", tid); return NULL; }
int main(int argc, char *argv[]) { CHECK(pthread_mutex_init(&sig_mutex, NULL) == 0); CHECK(pthread_cond_init(&sig_cond, NULL) == 0); thread_ready_count = 0; for(int i = 0; i < THREAD_CNT; i++) { pthread_t thread; int *data = malloc(sizeof(int)); *data = i; CHECK(pthread_create(&thread, NULL, fun_thread, data) == 0); } pthread_exit(NULL); }
|