关于Linux信号的备忘录

最近在做一个动态日志模块，需要用到linux信号，虽然用到的有关信号的东西都比较简单，但是我感觉还是要系统的学一下信号，为加强理解和今后便于记忆，写了下学习和使用中的一些备忘录。

安全的结束进程

循环的终止

有时候对于某些程序，我们不知道何时停止他们运行，只是在他们运行过程中我们达到了我们需要的结果我们才让他们停止运行，这样的程序我们一般是写一个while(true)的循环，当然我们终止他们直接是强制杀死进程，现在我们用信号来让他们更优雅的退出，在他们运行过程中，向他们发送一个信号，程序接收到信号后停止while循环，然后优雅的退出，我们这边使用SIGUSR1信号，代码如下：

#include <stdio.h>
#include <signal.h>
#include <unistd.h>

static void sig_usr(int);
static volatile int g_stop = 0;

int main(void)
{
    struct sigaction act;
    act.sa_handler = sig_usr;          
    act.sa_flags = 0;
    if ((sigemptyset(&act.sa_mask) == -1) ||
        (sigaction(SIGUSR1, &act, NULL) == -1)) {
        printf("error\n");
        return -1;
    }

    while(!g_stop){
        sleep(2);
    }
    printf("Process exit\n");
    return 0;
}

static void sig_usr(int signo)
{
    g_stop = 1;
}

采用ps命令获得程序的进程号，然后发送USR1信号，命令如：kill -s USR1 10120

信号等待的终止

我们可以通过pause函数去等待一个信号的到来，这个函数会一直阻塞，直到有信号到来，采用这个函数相对于定期检测更省cpu，上述代码检测进程退出我们可以这样写：

#include <stdio.h>
#include <signal.h>
#include <unistd.h>

static void sig_usr(int);
static volatile int g_stop = 0;

int main(void)
{
    struct sigaction act;
    act.sa_handler = sig_usr;          
    act.sa_flags = 0;
    if ((sigemptyset(&act.sa_mask) == -1) ||
        (sigaction(SIGUSR1, &act, NULL) == -1)) {
        printf("error\n");
        return -1;
    }

    while(!g_stop){
        pause();
    }
    printf("Process exit\n");
    return 0;
}

static void sig_usr(int signo)
{
    g_stop = 1;
}

但是这样会有一些风险，假如信号正好在while判断和pause函数之间到来，那么等到pause函数执行的时候就无法捕捉到信号了，解决这个问题可以用sigsuspend函数，在执行sigsuspend函数之前先阻塞掉我们要捕获的信号，sigsuspend函数执行两个动作，解除信号阻塞，等待信号到来，这两个动作结合成一个原子操作，也就是两个动作的间隙是不会有其他信号来干扰的，实现代码如下，为简单明了，以下省去了一些返回值处理。

#include <stdio.h>
#include <signal.h>
#include <unistd.h>

static void sig_usr(int);
static volatile int g_stop = 0;

int main(void)
{
    sigset_t maskblocked, maskunblocked, maskold;
    sigprocmask(SIG_SETMASK, NULL, &maskblocked);
    sigaddset(&maskblocked, SIGUSR1);
    sigdelset(&maskunblocked, SIGUSR1);

    struct sigaction act;
    act.sa_handler = sig_usr;          
    act.sa_flags = 0;

    if ((sigemptyset(&act.sa_mask) == -1) ||
        (sigaction(SIGUSR1, &act, NULL) == -1)) {
        printf("error\n");
        return -1;
    }
    sigprocmask(SIG_SETMASK, &maskblocked, &maskold);

    while(!g_stop){
        sigsuspend(&maskunblocked);
    }

    sigprocmask(SIG_SETMASK, &maskold, NULL);

    printf("Process exit\n");
    return 0;
}

static void sig_usr(int signo)
{
    g_stop = 1;
}

线程中等待信号

我们一般都在进程中注册信号回调函数，然后进程在执行过程中，如果有信号发生，就会转去执行信号回调函数，信号回调函数同注册信号的线程是属于同一线程，所以当信号来的时候进程就会暂时终止原来的操作，转向信号回调函数。有时候我们想专门开一个线程去等待信号，有什么信号执行什么操作，且都是在线程中执行，这样就不会影响主线程的操作了，我们可以在线程中执行sigwait函数来等待信号的到来。

#include <stdio.h>
#include <signal.h>
#include <unistd.h>
#include <pthread.h>

void *func_wait_signal(void *arg)
{
    int signo;
    sigset_t *sigset = (sigset_t *)arg;
    sigwait(sigset, &signo);/* 等待信号, signo会返回接收到的信号值 */
    return NULL;
}

int main(void)
{
    sigset_t sigset;
    pthread_t thread;
    if ((sigemptyset(&sigset) == -1) ||
        (sigaddset(&sigset, SIGUSR1) == -1) ||
        (sigprocmask(SIG_BLOCK, &sigset, NULL) == -1) ){
       fprintf(stderr, "Failed to block signals before sigwait\n");
       return -1;
    }
    pthread_create(&thread, NULL, func_wait_signal, &sigset);
    pthread_join(thread, NULL);/* 等待线程退出 */
    printf("Process exit\n");
    return 0;
}

注意：sigsuspend和sigwait都是接收一个信号集作为参数，但是两者的处理是有区别的，sisuspend是将信号集中没有包括的项给他unblock，当信号集中不包含的信号到来时sigsuspend才会返回，传给sigwait的信号集是当信号集中的信号到来时sigwait才返回，该信号集在传递给sigwait之前应当使用sigmaskproc阻塞掉，且sigwait不会改变信号集掩码。