socket 非阻塞模式下的connect封装实现

最新推荐文章于 2023-09-19 15:44:05 发布

转载最新推荐文章于 2023-09-19 15:44:05 发布 · 925 阅读

Linux 学习与总结专栏收录该内容

22 篇文章

订阅专栏

本文介绍在非阻塞模式下如何使用TCP的connect函数建立连接，并通过select或者epoll来判断连接是否成功。文章详细解释了如何处理EINPROGRESS错误，以及如何通过getsockopt来检查socket错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

connect 在非阻塞模式下，返回值 为0的时候表示 已建立连接；返回值为-1，且此时错误值是 EINPROGRESS时，，表示连接进行的3次握手正在进行。需要后续通过select或者是epoll来判断，如果select返回值为1，此时任然需要判断 socket是否有错。通过getsockopt来获取socket的错误值。

flags = fcntl(sockfd, F_GETFL, 0);
 fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);

 if ((retVal = connect(sockfd, saptr, salen)) < 0)
     if (errno != EINPROGRESS)
  return (-1);

 if (retVal == 0)
 {
     // restore file status flags
     fcntl(sockfd, F_SETFL, flags);  
     return 0;
 }

 FD_ZERO(&rset);
 FD_SET(sockfd, &rset);
 wset = rset;
 tval.tv_sec = nsec;
 tval.tv_usec = 0;

 if ((retVal = select(sockfd + 1, &rset, &wset, NULL, &tval)) == 0) 
 {
     // timeout
     close(sockfd);          
     errno = ETIMEDOUT;
     return (-1);
 }

 if (retVal < 0) 
 {
     // select() failed
     return (-1);
 }

 if (FD_ISSET(sockfd, &rset) || FD_ISSET(sockfd, &wset)) 
 {
     len = sizeof(error);
     error = 0;
     if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)
  return (-1);     

     if (error > 0) //<<<<< error == EINPROGRESS >>>
     { 
  close(sockfd);
  errno = error;
  return (-1);
     }
 } 
 else
 {
     return (-1);
 }

 // restore file status flags
 fcntl(sockfd, F_SETFL, flags);