Linux父子进程同步

/*
 * 使用信号实现父子进程之间的同步
 *
 * TELL_WAIT(): set things up for TELL_xxx & WAIT_xxx 
 * TELL_PARENT():  tell parent we are done
 * WAIT_PARENT():  wait for parent
 * TELL_CHILD(): tell child we are done
 * WAIT_CHILD(): wait for child
 *
 * SIGUSR1: the signal parent sends to child
 * SIGUSR2: the signal child sends to parent
 */

 

 

#include <sys/types.h> #include <signal.h> #include <unistd.h> #include <stdio.h>   static volatile sig_atomic_t sigflag;  static sigset_t newmask, oldmask, zeromask; /* signal handler for SIGUSR1 and SIGUSR2 */ static void sig_usr(int signo) {  sigflag = 1;  return; } void TELL_WAIT() {  if(signal(SIGUSR1, sig_usr) == SIG_ERR)   printf("signal SIGUSR1 error\n");  if(signal(SIGUSR2, sig_usr) == SIG_ERR)   printf("signal SIGUSR2 error\n");     sigemptyset(&zeromask);    sigemptyset(&newmask);  sigaddset(&newmask, SIGUSR1);  sigaddset(&newmask, SIGUSR2);    /* block SIGUSR1 and SIGUSR2, and save current signal mask */  if(sigprocmask(SIG_BLOCK, &newmask, &oldmask) < 0)   printf("SIG_BLOCK error\n"); } void TELL_PARENT(pid_t pid) {  kill(pid, SIGUSR2); /* tell parent we are done */ } void WAIT_PARENT() {  while(sigflag == 0)   sigsuspend(&zeromask); /* wait for parent */    sigflag = 0;    /* reset signal mask */  if(sigprocmask(SIG_SETMASK, &oldmask, NULL) < 0)   printf("SIG_SETMASK error\n"); } void TELL_CHILD(pid_t pid) {  kill(pid, SIGUSR1); } void WAIT_CHILD() {  while(sigflag == 0)   sigsuspend(&zeromask); /* wait for parent */    sigflag = 0;    /* reset signal mask */  if(sigprocmask(SIG_SETMASK, &oldmask, NULL) < 0)   printf("SIG_SETMASK error\n"); } void do_task(char *task_str) {  printf("%s\n", task_str); } /* parent goes first program */ int main() {  pid_t pid;    TELL_WAIT();    pid = fork();  if(pid < 0) {   printf("fork error\n");  }  else if(pid == 0) {   WAIT_PARENT();   do_task("child task\n");  }  else {   do_task("parent task\n");   TELL_CHILD(pid);  }    return 0; } /* child goes first program*/ int main() {  pid_t pid;    TELL_WAIT();    pid = fork();  if(pid < 0) {   printf("fork error\n");  }  else if(pid == 0) {   do_task("child task\n");   TELL_PARENT(getppid());  }  else {   WAIT_CHILD();   do_task("parent task\n");  }    return 0; }

/*
 * 使用管道实现父子进程同步 
 *
 * 父进程在调用TELL_CHILD 时经由上一个管道写一个字符p，子进程在
 * 调用TELL_PARENT时，经由下一个管道写一个字符c。相应的WAIT_XXX
 * 函数调用read读一个字符，没有读到字符时阻塞（睡眠等待）。
 *
 */
 

 

 

static int pfd1[2], pfd[2]; void TELL_WAIT() {     if(pipe(pfd1) < 0 || pipe(pfd2) < 0)         printf("pipe error\n"); } void TELL_PARENT(pid_t pid) {     if(write(pfd2[1], "c", 1) != 1)         printf("write error\n"); } void WAIT_PARENT() {     char c;     if(read(pfd1[0], &c, 1) != 1)         printf("read error\n");     if(c != 'p')         printf("WAIT_PARENT: incorrect data\n"); } void TELL_CHILD(pid_t pid) {     if(write(pfd1[1], "p", 1) != 1)         printf("write error\n"); } void WAIT_CHILD() {     char c;     if(read(pfd1[0], &c, 1) != 1)         printf("read error\n");     if(c != 'c')         printf("WAIT_CHILD: incorrect data\n"); }

 

转自：http://blog.chinaunix.net/space.php?uid=20196318&do=blog&id=28793