项目需要写一个主动连接且定时发送数据的客户端程序,并保证传输数据的可靠性和稳定性。

如果用阻塞方式,程序将阻塞在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所有连接到这个端口的连接;客户端发起连接



Logo

更多推荐