socket异步通信：如何设置成非阻塞模式，非阻塞模式下判断connect成功(失败)，判断recv/recvfrom成功(失败)，判断send/sendto

原文：

将一个socket 设置成 阻塞模式 和 非阻塞模式， 使用fcntl方法，即：

设置成非阻塞模式：

先用fcntl的F_GETFL获取flags,用F_SETFL设置flags|O_NONBLOCK;        

即：

 

      flags = fcntl(sockfd, F_GETFL, 0);                        //获取文件的flags值。

      fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);   //设置成非阻塞模式；

同时在接收和发送数据时，需要使用MSG_DONTWAIT标志

即：

      在recv，recvfrom和send，sendto数据时，将flag设置为MSG_DONTWAIT。

设置成阻塞模式：

 

先用fcntl的F_GETFL获取flags,用F_SETFL设置flags&~O_NONBLOCK;      

即：

     flags  = fcntl(sockfd,F_GETFL,0);                          //获取文件的flags值。

     fcntl(sockfd,F_SETFL,flags&~O_NONBLOCK);    //设置成阻塞模式；            

同时在接收和发送数据时，需要使用阻塞标志

即：

        在recv，recvfrom和send，sendto数据时，将flag设置为0，默认是阻塞。       

 

在将socket设置成非阻塞模式后，每次的对于sockfd 的操作都是非阻塞的；

非阻塞模式下：

connect   

       =0   当返回0时，表示立即创建了socket链接，

       <0   当返回-1时，需要判断errno是否是EINPROGRESS(表示当前进程正在处理)，否则失败。

       例如：下面会有select或epoll监听fd是否建立链接，

        select监听connect是否成功的例子，注意getsockopt验证，因为三次握手的第三个ACK有可能会丢失，但是客户端认为链接已经建立:

int ret = ::connect(_socket_fd, add.addr(), add.length());
if(ret == 0)
{
            //建立链接成功
}
else if(ret < 0 && errno == EINPROGRESS)          //errno == EINPROGRESS表示正在建立链接
{
     // 等待连接完成，errno == EINPROGRESS表示正在建立链接
     fd_set set;
     FD_ZERO(&set);
     FD_SET(_socket_fd,&set);  //相反的是FD_CLR(_sock_fd,&set)

     time_t = 10;          //(超时时间设置为10毫秒)
     struct timeval timeo;
     timeo.tv_sec = timeout / 1000; 
     timeo.tv_usec = (timeout % 1000) * 1000;

 

     int retval = select(_socket_fd + 1, NULL, &set, NULL, &timeo);           //事件监听
     if(retval < 0)   
     {
            //建立链接错误close(_socket_fd)
     }
     else if(retval == 0) // 超时
     {
            //超时链接没有建立close(_socket_fd)
     }

     //将检测到_socket_fd读事件或写时间，并不能说明connect成功
     if(FD_ISSET(_socket_fd,&set))
     {
           int error = 0;
           socklen_t len = sizeof(error);
           if(getsockopt(_socket_fd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)
           {
                  //建立简介失败close(_socket_fd)
           }
           if(error != 0) // 失败
           {
                  //建立链接失败close(_socket_fd)
           }
           else
            {
                  //建立链接成功
            }
     }
}
else
{
      //出现错误 close(_sock_fd)
}

注意：这里主要是想强调当epoll或select监听到sockfd上有EPOLL_IN或EPOLL_OUT时，即读写事件时，并不能说明链接已经建立，如上面的代码。

/*

int error = 0;
socklen_t ilen = sizeof(error);
ret = getsockopt(fd,SOL_SOCKET,SO_ERROR,&error,&ilen);
if(ret < 0)
{
      //说明链接建立失败，close(fd);
}
else if(error != 0 )
{

     //说明链接建立失败，close(fd);
}

else

{

       //说明链接建立成功。即可以向fd上写数据。

}

 */                      

 recv 和 recvfrom

       =0  当返回值为0时，表示对端已经关闭了这个链接，我们应该自己关闭这个链接，即close(sockfd)。另外因为异步操作会用select或epoll做事件触发，所以:

       1、如果使用select，应该使用FD_CLR(sockfd,fd_set)将sockfd清除掉，不再监听。

       2、如果使用epoll，系统会自己将sockfd清除掉，不再进行监听。

       >0 当返回值大于0 且 小于sizeof(buffer)时，表示数据肯定读完。（如果等于sizeof(buffer)，可能有数据还没读，应该继续读，不可能有大于）

       <0 当返回值小于0，即等于-1时，分情况判断：

        1、如果   errno   为  EAGAINE  或 EWOULDBLOCK                                      

                表示暂时无数据可读，可以继续读，或者等待epoll或select的后续通知。（EAGAINE，EWOULDBLOCK产生的

         原因：可能是多进程读同一个sockfd，可能一个进程读到数据，其他进程就读取不到数据（类似惊群效应），当然

         单个进程也可能出现这种情况。对于这种错误，不需用close(sockfd)。可以等待select或epoll的下一次触发，

         继续读。）

         2、如果   errno   为  EINTR

                表示被中断了，可以继续读，或者等待epoll或select后续的通知。

                否则，真的是读取数据失败。(此时应该close(sockfd))

 

send和sendto      

        返回值是实际发送的字符数，因为我们知道要发送的总长度，所以，如果没有发送完，我们可以继续发送。

          <0 当返回值为 -1   时， 我们需要判断  errno:

                1、如果errno为  EAGAINE   或 EWOULDBLOCK ，表示当前缓冲区写满，可以继续写，

                      或者等待epoll或select的后续通知，一旦有缓冲区，就会触发写操作，这个也是经常利用的一个特性。  

                 2、如果errno为EINTR  ，表示被中断了，可以继续写，或者等待epoll或select的后续通知。

                       否则真的出错了，即errno不为EAGAINE或EWOULDBLOCK或EINTR，此时应该close(sockfd)

          >=0 >=0且不等于要求发送的长度，应该继续send，如果等于要求发送的长度，发送完毕。

转自：http://aigo.iteye.com/blog/1939118