Linux下非阻塞connect
#include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <stdlib.h> #include <assert.h> #include <stdio.h> #include <time.h> #include <errno.h> #include <fcntl.h> #include <sys/ioctl.h> #include <unistd.h> #include <string.h> #define BUFFER_SIZE 1023 int setnonblocking(int fd) { int old_option=fcntl(fd,F_GETFL); int new_option=old_option | O_NONBLOCK; fcntl(fd,F_SETFL,new_option); return old_option; } int unblock_connect(const char *ip,int port,int time) { int ret=0; struct sockaddr_in address; bzero(&address,sizeof(address)); address.sin_family=AF_INET; inet_pton(AF_INET,ip,&address.sin_addr); address.sin_port=htons(port); int sockfd=socket(PF_INET,SOCK_STREAM,0); int fdopt=setnonblocking(sockfd); ret=connect(sockfd,(struct sockaddr *)&address,sizeof(address)); if(ret==0) { printf("connect with server immediately\n"); fcntl(sockfd,F_SETFL,fdopt); return sockfd; } else if(errno!=EINPROGRESS) { printf("unblock connect not support\n"); return -1; } fd_set readfds; fd_set writefds; struct timeval timeout; timeout.tv_sec=time; timeout.tv_usec=0; ret=select(sockfd+1,NULL,&writefds,NULL,&timeout); if(ret<=0) { printf("connection time out\n"); close(sockfd); return -1; } if(!FD_ISSET(sockfd,&writefds)) { printf("no events on sockfd found\n"); close(sockfd); return -1; } int error=0; socklen_t length=sizeof(error); if(getsockopt(sockfd,SOL_SOCKET,SO_ERROR,&error,&length)<0) { printf("get socket option failed"); close(sockfd); return -1; } if(error!=0) { printf("connection failed after select with the error:% d \n",error); close(sockfd); return -1; } printf("connection ready after select with the socket: %d \n",sockfd); fcntl(sockfd,F_SETFL,fdopt); return sockfd; } int main() { const char *ip="127.0.0.1"; int port=12345; int sockfd=unblock_connect(ip,port,10); if(sockfd<0) { return 1; } close(sockfd); return 0; }
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。