catch_child.cpp

/*
 * function: 父程式使用sigaction()函式捕捉SIGCHLD訊號子程式
 *
 * 2020-12-27
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/wait.h>

void func(int signo)
{
    pid_t wpid;
    int status;

    while ((wpid = waitpid(-1, &status, 0)) != -1)         // 阻塞回收, -1表示沒有子程式
    {
        if (WIFEXITED(status))
        {
            printf("catch a child %d, normal exit status:%d\n", wpid,  WEXITSTATUS(status));
        }
        else if (WIFSIGNALED(status))
        {
            printf("catch a child %d, kill by signal:%d\n", wpid, WTERMSIG(status));
        }
        else if (WCOREDUMP(status))
        {
            printf("catch a child %d, stop status:%d\n", wpid, WIFSTOPPED(status));
        }
    }
}

int main(int argc, char *argv[])
{
    pid_t pid = 0;

    // 阻塞SIGCHLD訊號，防止在父程式註冊捕捉函式之前子程式退出
    sigset_t set;
    sigemptyset(&set);
    sigaddset(&set, SIGCHLD);
    sigprocmask(SIG_BLOCK, &set, NULL);

    int ii = 0;
    for (ii = 0; ii < 5; ++ii)
    {
        pid = fork();
        if (0 == pid)
        {
            break;      // 子程式跳出迴圈
        }
        else if (-1 == pid)
        {
            perror("fork failed");
            exit(1);
        }
    }

    if (pid > 0)    // 父程式
    {
        printf("I'm parent:%d\n", getpid());

        struct sigaction act;
        act.sa_handler = func;       // 設定回撥函式
        sigemptyset(&act.sa_mask);  // 清空sa_mask
        act.sa_flags = 0;           // 設定預設屬性

        sigaction(SIGCHLD, &act, NULL);     // 註冊訊號捕捉

        // 解除阻塞
        sigprocmask(SIG_UNBLOCK, &set, NULL);

        while(1);       // 模擬父程式後續操作
    }
    else if (0 == pid)   // 子程式
    {
        printf("I'm a child:%d\n", getpid());
        return ii;
    }

    return 0;
}