mysleep

一、普通版本的mysleep函数

1.main函数调用mysleep函数,后者调用sigaction注册了SIGALRM信号的处理函数handler。 
2.调用alarm(timeout）设定闹钟。 
3.调用pause等待,内核切换到别的进程运行。 
4.timeout秒之后,闹钟超时,内核发SIGALRM给这个进程。 
5.从内核态返回这个进程的用户态之前处理未决信号,发现有SIGALRM信号,其处理函数是handler。 
6.切换到用户态执行handler函数,进入handler函数时SIGALRM信号被自动屏蔽, 从handler函数返回时SIGALRM信号自动解除屏蔽。然后自动执行系统调用sigreturn再次进入 内核,再返回用户态继续执行进程的主控制流程(main函数调用的mysleep函数)。 
7.pause函数返回-1,然后调用alarm(0)取消闹钟,调用sigaction恢复SIGALRM信号以前的处理动作。

代码实现：

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

void handler(int signum)
{}

int mysleep(int time)
{
    struct sigaction act,oldact;
    act.sa_handler = handler;
    int sa_flags = 0;
    sigset_t sa_mask;
    sigemptyset(&sa_mask);
    sigaction(SIGALRM, &act, &oldact);
    alarm(time);
    pause();
    int ret = alarm(0);
    sigaction(SIGALRM, &act, NULL);
    return ret;
}

int main()
{
    while(1)
    {
        mysleep(1);
        printf("I am sleeping!!\n");
    }
    return 0;
}

运行结果：

二、规避竞态条件的mysleep函数

1.屏蔽SIGALRM信号； 
2.alarm(time); 
3.解除对SIGALRM的屏蔽； 
4.pause(); 

代码实现：

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

void handler(int signum)
{}

int mysleep(int time)
{
    struct sigaction act,oldact;
    act.sa_handler = handler;
    int sa_flags = 0;
    sigset_t sa_mask, oldmask, suspmask;
    sigaction(SIGALRM, &act, &oldact);
    sigemptyset(&sa_mask);
    sigaddset(&sa_mask, SIGALRM);
    sigprocmask(SIG_BLOCK, &sa_mask, &oldmask);
    alarm(time);
    suspmask = oldmask;
    sigdelset(&suspmask, SIGALRM);
    sigsuspend(&suspmask);
    int ret = alarm(0);
    sigaction(SIGALRM, &act, NULL);
    sigprocmask(SIG_SETMASK, &oldmask, NULL);
    return ret;
}

int main()
{
    while(1)
    {
        mysleep(1);
        printf("I am sleeping!!\n");
    }
    return 0;
}

运行结果：