《APUE》：同步信號處理

《Unix環境高級編程》這本書附帶了許多短小精美的小程序，我在閱讀此書的時候，將書上的代碼按照自己的理解重寫了一遍（大部分是抄書上的），加深一下自己的理解（純看書太困了，呵呵）。此例子在Ubuntu 10.04上測試通過。

相關鏈接

• 《UNIX環境高級編程》（第二版）apue.h的錯誤 http://www.linuxidc.com/Linux/2011-04/34662.htm
• Unix環境高級編程 源代碼地址 http://www.linuxidc.com/Linux/2011-04/34826.htm

程序簡介：在多線程程序中等侍信號設置標志，從而讓主程序退出。唯一可運行的控制線程應該是主線程和信號處理程序，所以阻塞信號足以避免錯失標志的修改。另外在線程中，需要使用互斥量來保護標志。以下這個程序演示了這方面的內容。

1. //《APUE》程序12－6：同步信號處理
2. #include <stdio.h>
3. #include <stdlib.h>
4. #include <unistd.h>
5. #include <signal.h>
6. #include <pthread.h>
8. int quitflag;
9. sigset_t mask;
11. pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
12. pthread_cond_t wait = PTHREAD_COND_INITIALIZER;
14. void *thr_fn(void *arg)
15. {
16. int signo;
17. while(1)
18. {
19. //設置阻塞信號集
20. sigwait(&mask, &signo);
21. switch(signo)
22. {
23. //收到中斷信號以後
24. case SIGINT:
25. printf("\ninterrupt\n");
26. break;
28. //收到退出信號以後
29. case SIGQUIT:
30. pthread_mutex_lock(&lock);
31. quitflag = 1;
32. pthread_mutex_unlock(&lock);
33. pthread_cond_signal(&wait);
34. return 0;
36. default:
37. printf("unexpected signal %d\n", signo);
38. exit(1);
39. }
40. }
41. }
43. int main(void)
44. {
45. sigset_t oldmask;
46. pthread_t tid;
48. sigemptyset(&mask);
49. sigaddset(&mask, SIGINT);
50. sigaddset(&mask, SIGQUIT);
52. pthread_sigmask(SIG_BLOCK, &mask, &oldmask);
53. pthread_create(&tid, NULL, thr_fn, 0);
55. /*
56. 經典的UNIX條件鎖三步曲：
57. 1。使用pthread_cond_wait前要先加鎖
58. 2。pthread_cond_wait內部會解鎖，然後等待條件變量被其它線程激活
59. 3。pthread_cond_wait被激活後會再自動加鎖
60. （所以還要我們視情況而決定是否手動解鎖）
61. */
62. pthread_mutex_lock(&lock);
63. while( 0 == quitflag )
64. pthread_cond_wait(&wait, &lock);
65. pthread_mutex_unlock(&lock);
67. //收到退出信號，但現在這個信號是阻塞的
68. quitflag = 0;
69. //重置阻塞信號集（讓程序退出）
70. sigprocmask(SIG_SETMASK, &oldmask, NULL);
71. return 0;
72. }

運行示例（紅色字體的為輸入）：

www.linuxidc.com @ubuntu:~/code$gcc temp.c -lpthread -o temp
www.linuxidc.com @ubuntu:~/code$ ./temp
^C #輸入中斷字符(ctrl+C)
interrupt
^C #輸入中斷字符(ctrl+C)
interrupt
^C #輸入中斷字符(ctrl+C)
interrupt
^\ #輸入退出字符(ctrl+\)