C++ SOCKET中实现超时的一种方法

C++ SOCKET中实现超时的一种方法

         原始SOCKET貌似没有直接可以使用的超时参数，所以只能使用别的方法来实现超时。这是学习到的一种方法，写下来，做个记录。

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

using namespace std;

void timeout_alarm(int sig)
{
	cout << "time out " << endl;
//	exit(0);
}

int main(int argc, char* argv[])
{
	sighandler_t sig;

	sig = signal(SIGALRM, timeout_alarm);
	signal(SIGALRM, timeout_alarm);

	if(alarm(3) != 0)
		cout << "alarm was already set" << endl;
	sleep(2);
	//取消alarm
	alarm(0);
	cout << "ok, goes here1" << endl;

	//alarm是一次性的，所以需要重新设置
	if(alarm(3) != 0)
		cout << "alarm was already set" << endl;
	sleep(5);
	cout << "ok, goes here2" << endl;
	alarm(0);	
	
	if(alarm(3) != 0)
		cout << "alarm was already set" << endl;
	sleep(5);
		cout << "ok, goes here3" << endl;
	alarm(0);	

	return 0;
}

      signal( )函数用来设置信号的响应函数，alarm（）设定一个时间，当超时就会发送一个SIGALRM信号，这个时候signal( )就可以响应咯。alarm(0) 用于关闭。

      这个方法可用于阻塞型的业务,这里用了一个sleep( )表示一个阻塞业务。若小于三秒则不会调用timeout_alarm( )，若sleep()超过三秒则会超时，调用timeout_alarm()。