非阻塞connect的测试

工作在ET模式的epoll，描述符基本都是设置为非阻塞（nonblocking）的。阻塞的connect，最多可以等待75s（第二次握手syn+ack一直没到来），效率很低。对于非阻塞connect，调用函数后，通常返回-1，但并不代表出错，还要检查errno值，如果是115（EINPROGESS），则表示连接正在进行中。但如果对端的端口一直没打开，则connect后，errno还是115，不会是111（ECONNREFUSED），那如何判断当前连接是否成功了呢？这就必须在注册当前socket到epoll中时，注册进EPOLLIN事件，因为如果服务器端口没打开，客户端连接的时候，服务端会发送一个RST数据包来，这个RST数据包会触发EPOLLIN事件，然后在EPOLLIN中read，根据read的返回值来判断，返回值通常是-1，并且errno会变成111，由此来判断当前socket是被远端服务器拒绝了。

测试代码如下，用的是我另一篇文章的代码，Makefile一样：

测试客户端代码如下，服务端启用 nc -l 12439：

#include <iostream>
#include <sys/epoll.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>
#include <fcntl.h>
#include <cstring>

# define SERADDR "192.168.101.13"
# define SERPORT 12439
int main(void)
{
    int clifd = socket(AF_INET, SOCK_STREAM, 0);
    int epfd = epoll_create(1024);
    int ret;
    if(clifd < 0)
        std::cout << "Create clifd fail!\n";
    fcntl(clifd, F_SETFL, O_NONBLOCK);

    struct sockaddr_in seraddr;
    bzero(&seraddr, sizeof(seraddr));
    seraddr.sin_family = AF_INET;
    seraddr.sin_port = htons(SERPORT);
    seraddr.sin_addr.s_addr = inet_addr(SERADDR);

    ret = connect(clifd, (struct sockaddr*)&seraddr, sizeof(seraddr));
    std::cout << "ret:" << ret << std::endl;
    if(errno == EINPROGRESS)
        std::cout << strerror(errno) << std::endl;
    if(ret < 0 && errno != EINPROGRESS)
    {
        std::cout << "Connect to server error!\n";
        std::cout << strerror(errno) << std::endl;
        close(clifd);
        return -1;
    }
    struct epoll_event ev;
    struct epoll_event events[1024];
    ev.events =
#ifdef IN
        EPOLLIN |
#endif
#ifdef OUT
        EPOLLOUT |
#endif
        EPOLLET ;
    ev.data.fd = clifd;
    epoll_ctl(epfd, EPOLL_CTL_ADD, clifd, &ev);

    while(1)
    {
        int num = epoll_wait(epfd, events, 1024, -1);
        char buf[1024];
        for(int i = 0; i < num; i++)
        {
            if(events[i].events & EPOLLIN){
                std::cout << "Ready to read!\n";
                std::cout << "ret=" << read(events[i].data.fd, buf,1024)
                          << std::endl;
                std::cout << "errno=" << errno << '\t'
                          << strerror(errno) << std::endl;

            }
            if(events[i].events & EPOLLOUT){
                std::cout << "Ready to write!\n";
            }
        }

    }
}

makefile如下：

CLIENT  =   epollclient
CLISRC  =   epollcli.cpp
CC      =   g++

ifneq ($(ALL),)
    MACROALL = -DIN -DOUT
else
ifneq ($(IN),)
    MACROIN = -DIN
else
ifneq ($(OUT),)
    MACROOUT = -DOUT
endif
endif
endif

.PHONY:all clean
all: clean  $(CLIENT)
$(CLIENT): $(CLISRC)
             $(CC) $^ -o $@ $(MACROALL) $(MACROIN) $(MACROOUT)
clean:
    @rm -rf  epollclient

测试结果如下：

(1) 远端服务器监听关闭，只注册EPOLLOUT到events中，make OUT=1

      

        返回-1，且提示连接正在进行中，这个描述符永远不知道是否连接成功。

(2)远端服务器监听打开 （nc -l 12439），只注册EPOLLOUT到events中，make OUT=1

     

      返回-1，且提示连接正在进行中，后面提示可写，证明连接成功了。

(3) 远端服务器监听关闭，只注册EPOLLIN到events中，make IN=1

     

     触发了可读事件（由于服务器发了一个RST回来），读取后返回值为-1，且此时的errno为111，连接被拒绝

(4) 远端服务器监听开启（nc -l 12439），只注册EPOLLIN到events中，make IN=1

     

     连接成功了（服务器没有发RST），因此等待读事件发生。

(5) 远端服务器监听关闭，同时注册EPOLLIN || EPOLLOUT到events中，make ALL=1

  

   触发了可读事件(服务器发回了RST)，读取后，知道是连接被拒绝错误；由于没有成功EPOLLOUT事件没有被触发。

(5) 远端服务器监听开 启（nc -l 12439），同时注册EPOLLIN || EPOLLOUT到events中，make ALL=1

  

  触发了可写事件，因为此时连接已经成功建立，此描述符可写。

注：以上完全个人测试结果，刚接触，有不对的地方，愿意接受大家的批评、指正，共同学习，共同进步。