【Linux】模拟实现sleep函数

本次用到的相关函数介绍

alarm函数

头文件

#include<unistd.h>

unsigned int alarm(unsigned int seconds);

作用

调用alarm函数可以设定一个闹钟,也就是告诉内核在seconds秒之后给当前进程发SIGALRM信号, 该信号的默认处理动作是终止当前进程。 

返回值

这个函数的返回值是0或者是以前设定的闹钟时间还余下的秒数。

pause函数

头文件

#include<unistd.h>

int pause(void);

作用

pause函数使调用进程挂起直到有信号递达。

返回值

1、如果信号的处理动作是终止进程,则进程终止,pause函数没有机会返回

2、如果信号的处理动作是忽略,则进程继续处于挂起状态,pause不返回

3、如果信号的处理动作是捕捉,则调用了信号处理函数之后pause返回-1,errno设置为EINTR, 所以pause只有出错的返回值。错误码EINTR表示“被信号中断”。

模拟实现sleep函数

步骤

（1）mysleep函数中注册了SIGALRM信号的处理函数，该处理函数为sigAlrm()

（2）调用alarm函数进行闹钟的设定

（3）利用pause函数进行挂起等待，直到有信号递达

（4）闹钟时间到后，内核发送SIGALRM信号到该进程

（5）当这个进程从内核态返回用户态时，进行SIGALRM信号的自定义处理，调用处理函数sigAlrm

（6）处理完毕后，pause函数返回-1

（7）调用alarm(0)来进行闹钟的取消，并将其作为返回值

（8）恢复SIGALRM信号以前的动作

具体实现

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

void sigAlrm(int signal)
{}

unsigned int mysleep(unsigned int second)
{
	struct sigaction sig,osig;
	sig.sa_handler = sigAlrm;//设置处理函数为sigAlrm();
	sigemptyset(&sig.sa_mask);//将表示阻塞的标记初始化归零
	sig.sa_flags = 0;

	sigaction(SIGALRM,&sig,&osig);//将SIGALRM信号的处理方式设置成sig,并用osig保存sig的设置方式

	alarm(second);//等待second秒后发送SIGALRM信号
	pause();//进程挂起直到有信号递达

	unsigned int ret = alarm(0);//清除闹钟，并保存函数返回值，与sleep函数返回值一致

	sigaction(SIGALRM,&osig,NULL);//将SIGALRM信号的处理方式设置成osig

	return ret;
}
int main()
{
	while(1)
	{
		mysleep(1);
		printf("sleep 1 second\n");
	}
	return 0;
} 

运行结果

竞态条件

基本概念

当程序由于不恰当时序引起的错误就称为竞态条件

分析上述代码

假设在我们设定闹钟，进程进行挂起等待后，由于内核的调度让更高优先级的进程去执行，而我们这份代码的优先级很低，很久可能都轮不到我们跑这份代码。

在alarm时间到后，内核发送SIGALRM到进程，处于未决状态。

后来其他进程执行完了，轮到我们这个进程时，SLGALRM信号递达，进入处理函数sigAlrm后再次进入内核。然后返回进程的主控制流程，alarm函数返回，调用pause函数进行挂起等待。

可是，此刻SIGALRM信号已经处理完了，这样等待只会造成永远等待下去。

优化原理

（1）屏蔽SIGALRM信号

（2）alarm(seconds)

（3）解除对SIGALRM信号的屏蔽

（4）pause()

但是呢，从解除信号屏蔽到调用pause之间仍存间隙，SIGALRM信号任然有可能在这个间隙之间递达。

这里呢，我们就需要利用sigsuspend函数来进行处理

这个函数可以将解除信号屏蔽和挂起等待信号两个步骤和为一步，从而解决了上述代码导致的错误

sigsuspend函数

函数头文件：#include<signal.h>

函数原型：int sigsuspend(const sigset_t *sigmask);

更改代码

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

void sigAlrm(int signal)
{}


unsigned int mySleep(unsigned int second)
{
	struct sigaction sig,osig;
	sigset_t mask,omask,smask;//定义三个mask变量 
	sig.sa_handler = sigAlrm;//设置处理函数为sigAlrm()
	sigemptyset(&sig.sa_mask);//将阻塞字段初始化为0
	sig.sa_flags = 0;
	
	sigaction(SIGALRM,&sig,&osig);//先用osig保留之前的处理方式，然后将当前的SIGALRM处理方式设置成sig

	sigemptyset(&mask);//初始化信号屏蔽字段mask
	sigaddset(&mask,SIGALRM);//将SIGALRM信号添加到mask字段中
	sigprocmask(SIG_BLOCK,&mask,&omask);//先将当前信号屏蔽字段保存到omask中，然后将mask字段设置成当前的信号屏蔽字段

	alarm(second);//设置在second秒后收到SIGALRM信号

	smask = omask;
	sigdelset(&smask,SIGALRM);//删除smask保存之前的SIGALRM字段
	sigsuspend(&smask);//解除smask中的信号屏蔽，并且挂起等待直到收到一个信号

	unsigned int ret = alarm(0);//清除信号，并保存返回值
	sigaction(SIGALRM,&osig,NULL);//还原之前的屏蔽信号

	sigprocmask(SIG_SETMASK,&omask,NULL);//还原之前的信号屏蔽字段

	return ret;
}

int main()
{
	while(1)
	{
		mySleep(1);
		printf("sleep 1 second\n");
	}
	return 0;
}

运行结果

如果在调用mysleep函数时SIGALRM信号没有屏蔽，我们根据下列步骤来防范竞态条件

（1） 调用sigprocmask(SIG_BLOCK, &mask, &omask);来屏蔽SIGALRM信号，不让该信号递达

（2） 调用sigsuspend(&suspmask);来解除对SIGALRM的屏蔽，然后将该进程进行挂起等待操作

（3） SIGALRM信号递达后，利用suspend返回，自动恢复原来的屏蔽字，也就是再次屏蔽SIGALRM。

（4） 调用sigprocmask(SIG_SETMASK, &omask, NULL);用来再次解除对SIGALRM的屏蔽。

这样我们便也进行竞态条件的防范了