《APUE》：为进程打印屏蔽字


《Unix环境高级编程》这本书附带了许多短小精美的小程序，我在阅读此书的时候，将书上的代码按照自己的理解重写了一遍（大部分是抄书上的），加深一下自己的理解（纯看书太困了，呵呵）。此例子在Ubuntu 10.04上测试通过。

相关链接

  • 《UNIX环境高级编程》（第二版）apue.h的错误
  • Unix环境高级编程 源代码地址

程序简介：这个例子显示了保护临界区，不被特定信号中断的正确方法

  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.bkjia.com @ubuntu:~/code$ gcc temp.c -o temp
www.bkjia.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：这一节看得我头晕晕的，所以上面说的可能有错，如果大家发现错误，请提出，大家共同学习，共同进步。

相关内容