《Unix环境高级编程》：为进程打印屏蔽字

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

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

//《APUE》程序10－10：为进程打印屏蔽字
//《APUE》程序10－15：保护临界区不被信号中断
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <errno.h>

//输出错误信息并退出    
void error_quit(const char *str)    
{    
	fprintf(stderr, "%s\n", str);    
	exit(1);    
}  

//打印调用进程的依赖屏蔽字中信号的名称（只测试了我们用到的几种）
void pr_mark(const char *str)
{
	sigset_t sigset;
	int error_save;

	error_save = errno;
	int temp = sigprocmask(0, NULL, &sigset);
	if( temp < 0 )
		error_quit("sigpromask error");

	printf("%s", str);
	if( sigismember(&sigset, SIGINT) )
		printf("SIGINT ");
	if( sigismember(&sigset, SIGQUIT) )
		printf("SIGQUIT ");
	if( sigismember(&sigset, SIGUSR1) )
		printf("SIGUSR1 ");
	if( sigismember(&sigset, SIGALRM) )
		printf("SIGALRM ");

	printf("\n");
	errno = error_save;
}

void sig_int(int signo)
{
	pr_mark("\nin sig_int: ");
}

int main(void)
{
	sigset_t newmask, oldmask, waitmask;
	pr_mark("program start: ");

	//设置对SIGINT信号的处理函数  
	if( SIG_ERR == signal(SIGINT, sig_int) )
		error_quit("signal(SIGINT) error");
	//将waitmask信号集初始化为空  
	sigemptyset(&waitmask);
	//在waitmask信号集中增加SIGUSR1信号  
	sigaddset(&waitmask, SIGUSR1);

	//将newmask信号集初始化为空
	sigemptyset(&newmask);
	//在newmask信号集中增加SIGINT信号 
	sigaddset(&newmask, SIGINT);

	//将oldmask设置为当前的屏蔽信号集（方便以后恢复）  
	//在当前的屏蔽信号添加增加newmask信号集  
	int temp = sigprocmask(SIG_BLOCK, &newmask, &oldmask);
	if( temp < 0 )
		error_quit("SIG_BLOCK error");
	pr_mark("in critical region: ");

	//这段操作比较难理解，请见注解2
	//进入临界区，等待SIGINT信号
	temp = sigsuspend(&waitmask);
	if( temp != -1 )
		error_quit("sigsuspend error");
	//收到SIGINT信号后，离开临界区

	pr_mark("after return from sigsuspend: ");

	//恢复屏蔽信号集  
	temp = sigprocmask(SIG_SETMASK, &oldmask, NULL);
	if( temp < 0 )
		error_quit("SIG_SETMASK error");

	pr_mark("program exit: ");
	return 0;
}

运行示例（红色字体的为输入）：
qch@ubuntu:~/code$ gcc temp.c -o temp
qch@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：这一节看得我头晕晕的，所以上面说的可能有错，如果大家发现错误，请提出，大家共同学习，共同进步。