connect_timeo.c

来自「非常高效的扫描器」· C语言 代码 · 共 42 行

#define TIMEOUT 5
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <errno.h>
#include <stdlib.h>
#include <signal.h>
#ifdef LINUX
#include <sys/time.h>
#endif
typedef void Sigfunc (int);

void connect_alarm(int signo);


int
connect_timeo(int sockfd, struct sockaddr *saptr, int salen, int
nsec)

{
int n;
alarm(0);
signal(SIGALRM,connect_alarm);
alarm(TIMEOUT);

if( (n = connect(sockfd, (struct sockaddr *) saptr, salen)) < 0) {
	close(sockfd);
	if(errno == EINTR)
	errno = ETIMEDOUT;
}
alarm(0);
signal(SIGALRM, SIG_DFL);
return(n);
}

void connect_alarm(int signo)
{
// printf("DEBUG: connect timed out\n");
return;
}