c/c++ llinux epoll系列4 利用epoll_wait实现非阻塞的connect
程序员文章站
2022-03-10 15:10:25
llinux epoll系列4 利用epoll_wait实现非阻塞的connect connect函数是阻塞的,而且不能设置connect函数的timeout时间,所以一旦阻塞太长时间,影响用户的体验,所以就出来一个需求,硬要设置connect的timeout时间。 实现方法:先把connect函数 ......
llinux epoll系列4 利用epoll_wait实现非阻塞的connect
connect函数是阻塞的,而且不能设置connect函数的timeout时间,所以一旦阻塞太长时间,影响用户的体验,所以就出来一个需求,硬要设置connect的timeout时间。
实现方法:先把connect函数变成非阻塞的,然后用设置epoll_wait的timeout时间,用epoll_wait等待connect的完成。
#include <stdio.h> #include <unistd.h> #include <string.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <sys/epoll.h> #include <sys/ioctl.h> #include <errno.h> #include <arpa/inet.h> #include <fcntl.h> #define events 10 int main(){ sockaddr_in server; int sock, epfd; char buf[32]; int nfds, n; int val; epoll_event ev, ev_ret[events]; sock = socket(af_inet, sock_stream, 0); server.sin_family = af_inet; server.sin_port = htons(12345); inet_pton(af_inet, "127.0.0.1", &server.sin_addr.s_addr); epfd = epoll_create(1); if(epfd < 0){ perror("epfd"); return 1; } memset(&ev, 0, sizeof(ev)); ev.events = epollin; ev.data.fd = sock; if(epoll_ctl(epfd, epoll_ctl_add, sock, &ev) != 0){ perror("epoll_ctl"); return 1; } val = 1; //本来下面的connect函数是阻塞的,但是用fionbio的ioctl函数后,就变成非阻塞的了, //所以不管connect函数成功与否,都立即结束了。 ioctl(sock, fionbio, &val); n = connect(sock, (sockaddr*)&server, sizeof(server)); if(n != 0){ if(errno == einprogress){ printf("before epoll_wait\n"); nfds = epoll_wait(epfd, ev_ret, events, 1000*10);//timeout is 1 sec if(nfds < 0){ perror("epoll_wait\n"); return 1; } printf("after epoll_wait : nfds=%d\n", nfds); } else{ perror("connect"); return 1; } } n = read(sock, buf, sizeof(buf)); write(fileno(stdout), buf, n); close(sock); return 0; }