《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>   
8.   
9. //輸出錯誤信息並退出       
10. void error_quit(const char *str)      
11. {      
12.     fprintf(stderr, "%s\n", str);      
13.     exit(1);      
14. }    
15.   
16. //打印調用進程的依賴屏蔽字中信號的名稱（只測試了我們用到的幾種）   
17. void pr_mark(const char *str)  
18. {  
19.     sigset_t sigset;  
20.     int error_save;  
21.   
22.     error_save = errno;  
23.     int temp = sigprocmask(0, NULL, &sigset);  
24.     if( temp < 0 )  
25.         error_quit("sigpromask error");  
26.   
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 ");  
36.   
37.     printf("\n");  
38.     errno = error_save;  
39. }  
40.   
41. void sig_int(int signo)  
42. {  
43.     pr_mark("\nin sig_int: ");  
44. }  
45.   
46. int main(void)  
47. {  
48.     sigset_t newmask, oldmask, waitmask;  
49.     pr_mark("program start: ");  
50.   
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);  
58.   
59.     //將newmask信號集初始化為空   
60.     sigemptyset(&newmask);  
61.     //在newmask信號集中增加SIGINT信號    
62.     sigaddset(&newmask, SIGINT);  
63.   
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: ");  
70.   
71.     //這段操作比較難理解，請見注解2   
72.     //進入臨界區，等待SIGINT信號   
73.     temp = sigsuspend(&waitmask);  
74.     if( temp != -1 )  
75.         error_quit("sigsuspend error");  
76.     //收到SIGINT信號後，離開臨界區   
77.   
78.     pr_mark("after return from sigsuspend: ");  
79.   
80.     //恢復屏蔽信號集     
81.     temp = sigprocmask(SIG_SETMASK, &oldmask, NULL);  
82.     if( temp < 0 )  
83.         error_quit("SIG_SETMASK error");  
84.   
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：這一節看得我頭暈暈的，所以上面說的可能有錯，如果大家發現錯誤，請提出，大家共同學習，共同進步。