《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》程序10－10：為進程打印屏蔽字
2. //《APUE》程序10－15：保護臨界區不被信號中斷
3. #include <unistd.h>
4. #include <stdio.h>
5. #include <stdlib.h>
6. #include <signal.h>
7. #include <errno.h>
9. //輸出錯誤信息並退出
10. void error_quit(const char *str)
11. {
12. fprintf(stderr, "%s\n", str);
13. exit(1);
14. }
16. //打印調用進程的依賴屏蔽字中信號的名稱（只測試了我們用到的幾種）
17. void pr_mark(const char *str)
18. {
19. sigset_t sigset;
20. int error_save;
22. error_save = errno;
23. int temp = sigprocmask(0, NULL, &sigset);
24. if( temp < 0 )
25. error_quit("sigpromask error");
27. printf("%s", str);
28. if( sigismember(&sigset, SIGINT) )
29. printf("SIGINT ");
30. if( sigismember(&sigset, SIGQUIT) )
31. printf("SIGQUIT ");
32. if( sigismember(&sigset, SIGUSR1) )
33. printf("SIGUSR1 ");
34. if( sigismember(&sigset, SIGALRM) )
35. printf("SIGALRM ");
37. printf("\n");
38. errno = error_save;
39. }
41. void sig_int(int signo)
42. {
43. pr_mark("\nin sig_int: ");
44. }
46. int main(void)
47. {
48. sigset_t newmask, oldmask, waitmask;
49. pr_mark("program start: ");
51. //設置對SIGINT信號的處理函數
52. if( SIG_ERR == signal(SIGINT, sig_int) )
53. error_quit("signal(SIGINT) error");
54. //將waitmask信號集初始化為空
55. sigemptyset(&waitmask);
56. //在waitmask信號集中增加SIGUSR1信號
57. sigaddset(&waitmask, SIGUSR1);
59. //將newmask信號集初始化為空
60. sigemptyset(&newmask);
61. //在newmask信號集中增加SIGINT信號
62. sigaddset(&newmask, SIGINT);
64. //將oldmask設置為當前的屏蔽信號集（方便以後恢復）
65. //在當前的屏蔽信號添加增加newmask信號集
66. int temp = sigprocmask(SIG_BLOCK, &newmask, &oldmask);
67. if( temp < 0 )
68. error_quit("SIG_BLOCK error");
69. pr_mark("in critical region: ");
71. //這段操作比較難理解，請見注解2
72. //進入臨界區，等待SIGINT信號
73. temp = sigsuspend(&waitmask);
74. if( temp != -1 )
75. error_quit("sigsuspend error");
76. //收到SIGINT信號後，離開臨界區
78. pr_mark("after return from sigsuspend: ");
80. //恢復屏蔽信號集
81. temp = sigprocmask(SIG_SETMASK, &oldmask, NULL);
82. if( temp < 0 )
83. error_quit("SIG_SETMASK error");
85. pr_mark("program exit: ");
86. return 0;
87. }

運行示例（紅色字體的為輸入）：
www.linuxidc.com @ubuntu:~/code$ gcc temp.c -o temp
www.linuxidc.com @ubuntu:~/code$ ./temp
program start:
in critical region: SIGINT
^C
in sig_int: SIGINT SIGUSR1
after return from sigsuspend: SIGINT
program exit:

注解：
1：臨界區指的就是不希望被某類信號打斷的代碼段（這裡是指SIGUSR1）

2：sigsuspend的整個原子操作過程為：
(1) 設置新的mask阻塞當前進程；
(2) 收到信號，恢復原先mask；
(3) 調用該進程設置的信號處理函數；
(4) 待信號處理函數返回後，sigsuspend返回。

3：sigsuspend和sigprocmask(); pause();區別是：
sigsuspend把信號捕捉及信號處理函數的調用整合在一起了，也就是說它設置新的mask阻塞當前進程後如果有信號我就自己收信號，調用信號處理函數了，而sigprocmask() ; pause();的問題在於它希望在pause之後收信號，而不是sigprocmask() 和pause();之間。

PS：這一節看得我頭暈暈的，所以上面說的可能有錯，如果大家發現錯誤，請提出，大家共同學習，共同進步。