Linux下实现connect超时
项目需要写一个主动连接且定时发送数据的客户端程序,并保证传输数据的可靠性和稳定性。如果用阻塞方式,程序将阻塞在connect、send函数上;如果采用非阻塞方式,程序又会出现connect返回值判定、select超时设置问题、send/recv 返回值、close(socket)后send/recv数据等问题。【1】非阻塞socket编程问题小结里面有详细的介绍。其中关于非阻塞conn
·
项目需要写一个主动连接且定时发送数据的客户端程序,并保证传输数据的可靠性和稳定性。
如果用阻塞方式,程序将阻塞在connect、send函数上;如果采用非阻塞方式,程序又会出现connect返回值判定、select超时设置问题、send/recv 返回值、close(socket)后send/recv数据等问题。【1】
非阻塞socket编程问题小结里面有详细的介绍。其中关于非阻塞connect的问题,【2】
Linux下Socket通信中非阻塞connect的注意事项 介绍的很详细。
为了减少代码量,保证稳定下,借鉴【3】
linux下socket connect 阻塞方式 阻塞时间控制 的思路进行,采用alarm+sigaction的方式,为什么不采用signal?因为signal与sigaction函数区别,
signal是重启动函数,超时以后会自动启动已阻塞的函数,而不是中断它的执行,比如recvfrom,给人的感觉就是使用了alarm,程序依然阻塞在了recvfrom上,不往下执行,如果在信号处理函数中使用printf可以看到超时后输出了一条超时信息,然后signal又启动了recvfrom,继续阻塞。。。sigaction可以自己设置是否重启动函数,
alrmact.sa_flags = SA_NOMASK选项,
SA_NOMASK为不重启动,中断已阻塞的函数recvfrom,使程序继续往下执行,SA_RESTART为重启动函数,与signal相同,继续阻塞在recvfrom上。。。
参考【4】
linux C使用alarm设置套接字接收函数recvfrom超时
【5】
那年,一步一步学linux c ---signal 和sigaction
为什么不用sigrelse?【6】
SIGSET中强调sigset, sighold, sigrelse, sigignore - System V 信号 API,这些函数在 glibc 里为使用历史上的 System V 信号 API 的程序提供一个兼容接口。这个 API 是历史遗留物:新程序应该使用 POSIX 信号 API(sigaction、sigprocmask) 等等)。
问题最终解决,总结linux 下实现connect 超时的两种方法,(1) alarm+sigaction,(2) 非阻塞的select.示例代码参考如下【7】
linux 下实现connect 超时
sigaction
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netdb.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <unistd.h>
#include <asm/ioctls.h>
#include <signal.h>
#define SERVPORT 13333
#define TIMEOUT_TIME 6
void connect_sigalarm()
{
printf("connect alarm.\n");
return;
}
int main(int argc,char *argv[])
{
int sockfd;
int retval;
int error=-1;
char *dest_ip;
int ret=0;
struct sockaddr_in serv_info;
struct sigaction act,oldact;
struct timeval tm;
fd_set rset,wset;
char buf[1024];
if(argc == 2)
{
dest_ip=argv[1];
}
act.sa_handler = connect_sigalarm;
sigemptyset(&act.sa_mask);
sigaddset(&act.sa_mask, SIGALRM);
act.sa_flags = SA_INTERRUPT; //由此信号中断的系统调用不会自动重启
sigaction(SIGALRM, &act, &oldact);
if(alarm(5) != 0)
{
printf("alarm has already set.\n");
}
memset(&serv_info,0,sizeof(serv_info));
serv_info.sin_family = AF_INET;
serv_info.sin_port = htons(SERVPORT);
serv_info.sin_addr.s_addr = inet_addr(dest_ip);
sockfd = socket(PF_INET,SOCK_STREAM,0);
if(sockfd<0)
{
perror("socket");
return 1;
}
if(connect(sockfd,(struct sockaddr *)&serv_info,sizeof(struct sockaddr))<0)
{
close(sockfd);
if (errno == EINTR)
{
//errno=TIMEOUT;
printf("connect timeout\n");
}
}
alarm(0);
return 0;
}
非阻塞的select
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netdb.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <unistd.h>
#include <asm/ioctls.h>
#include <signal.h>
#include <fcntl.h>
#define SERVPORT 13333
#define TIMEOUT_TIME 6
int main(int argc,char *argv[])
{
int sockfd;
int retval;
int error=-1;
char *dest_ip;
int len;
int ret=0;
struct sockaddr_in serv_info;
struct sigaction act,oldact;
struct timeval tm;
fd_set rset,wset;
char buf[1024];
if(argc == 2)
{
dest_ip=argv[1];
}
memset(&serv_info,0,sizeof(serv_info));
serv_info.sin_family = AF_INET;
serv_info.sin_port = htons(SERVPORT);
serv_info.sin_addr.s_addr = inet_addr(dest_ip);
sockfd = socket(PF_INET,SOCK_STREAM,0);
if(sockfd<0)
{
perror("socket");
return 1;
}
if(fcntl(sockfd,F_SETFL,fcntl(sockfd,F_GETFL) | O_NONBLOCK)<0)
{
perror("fcntl");
close(sockfd);
return 1;
}
if(connect(sockfd,(struct sockaddr *)&serv_info,sizeof(struct sockaddr))<0)
{
if(errno != EINPROGRESS)
{
perror("connect");
return 1;
}
tm.tv_sec = TIMEOUT_TIME;
tm.tv_usec = 0;
FD_ZERO(&wset);
FD_SET(sockfd,&wset);
retval = select(sockfd+1,NULL,&wset,NULL,&tm);
switch(retval)
{
case -1:
perror("select");
return 1;
case 0:
printf("timeout\n");
break;
default:
if(FD_ISSET(sockfd,&wset))
{
if(getsockopt(sockfd,SOL_SOCKET,SO_ERROR,&error,(socklen_t *)&len)<0)
{
return 1;
}
printf("error=%d\n",error);
if(error==0)
{
ret=1;
}
else
{
ret=0;
errno=error;
}
}
break;
}
}
fcntl(sockfd,F_SETFL,fcntl(sockfd,F_GETFL) & O_NONBLOCK); //set back to block mode
if(!ret)
{
close(sockfd);
perror("connect failed");
return 1;
}
sprintf(buf,"%s","hello world!");
if(send(sockfd,buf,strlen(buf),0)<0)
{
perror("send");
}
printf("connected\n");
return 0;
}
均已测试通过,测试方式,服务端监听端口13333,设置iptables drop所有连接到这个端口的连接;客户端发起连接
更多推荐
已为社区贡献1条内容
所有评论(0)