非阻塞Connect对于select时应注意问题

 http://www.cnitblog.com/zouzheng/archive/2010/11/25/71711.html 

 对于面向连接的socket类型(SOCK_STREAM,SOCK_SEQPACKET)在读数据之前必须建立连接,首先服务器端socket必须在一个客户端知道的地址进行监听,也就是创建socket之后必须调用bind绑定到一个指定的地址,然后调用int listen(int sockfd, int backlog);进行监听。此时服务器socket允许客户端进行连接,backlog提示没被accept的客户连接请求队列的大小,系统决定实际的值,最大值定义为SOMAXCONN在头文件<sys/socket.h>里面。如果某种原因导致服务器端进程未及时accpet客户连接而导致此队列满了的话则新的客户端连接请求被拒绝(在工作中遇到过此情况,IONA ORBIX(CORBA中间件)由于没有配置超时时间结果在WIFI网络中传输数据出现异常情况一直阻塞而无机会调用accept接受新的客户请求,于是最终队列满导致新的客户连接被拒绝)。

  调用listen之后当有客户端连接到达的时候调用int accept(int sockfd, struct sockaddr *restrict addr, socklen_t *restrict len);接受客户端连接建立起连接返回用于连接数据传送的socket描述符,进行监听的socket可以用于继续监听客户端的连接请求,返回的socket描述符跟监听的socket类型一致。如果addr不为NULL,则客户端发起连接请求的socket地址信息会通过addr进行返回。如果监听的socket描述符为阻塞模式则accept一直会阻塞直到有客户发起连接请求,如果监听的socket描述符为非阻塞模式则如果当前没有可用的客户连接请求,则返回-1(errno设置为EAGAIN)。可以使用select函数对监听的socket描述符进行多路分离,如果有客户连接请求则select将监听的socket描述符设置为(注意,如果监听的socket为阻塞模式而使用select进行多路分离则可能造成select返回但是调用accept会被阻塞住的情况,原因是在调用accept之前客户端可能主动关闭连接或者发送RST异常关闭连接,因此select最好跟非阻塞socket搭配使用)

  客户端调用int connect(int sockfd, const struct sockaddr *addr, socklen_t len);发起对服务器的socket的连接请求,如果客户端socket描述符为阻塞模式则会一直阻塞到连接建立或者连接失败(注意阻塞模式的超时时间可能为75秒到几分钟之间),而如果为非阻塞模式,则调用connect之后如果连接不能马上建立则返回-1(errno设置为EINPROGRESS,注意连接也可能马上建立成功比如连接本机的服务器进程),如果没有马上建立返回,此时TCP的三路握手动作在背后继续,而程序可以做其他的东西,然后调用select检测非阻塞connect是否完成(此时可以指定select的超时时间,这个超时时间可以设置为比connect的超时时间短),如果select超时则关闭socket,然后可以尝试创建新的socket重新连接,如果select返回非阻塞socket描述符可写则表明连接建立成功,如果select返回非阻塞socket描述符既可读又可写则表明连接出错(注意:这儿必须跟另外一种连接正常的情况区分开来,就是连接建立好了之后,服务器端发送了数据给客户端,此时select同样会返回非阻塞socket描述符既可读又可写,这时可以通过以下方法区分:
  1.调用getpeername获取对端的socket地址.如果getpeername返回ENOTCONN,表示连接建立失败,然后用SO_ERROR调用getsockopt得到套接口描述符上的待处理错误;
  2.调用read,读取长度为0字节的数据.如果read调用失败,则表示连接建立失败,而且read返回的errno指明了连接失败的原因.如果连接建立成功,read应该返回0;
  3.再调用一次connect.它应该失败,如果错误errno是EISCONN,就表示套接口已经建立,而且第一次连接是成功的;否则,连接就是失败的;
  对于无连接的socket类型(SOCK_DGRAM),客户端也可以调用connect进行连接,此连接实际上并不建立类似SOCK_STREAM的连接,而仅仅是在本地保存了对端的地址,这样后续的读写操作可以默认以连接的对端为操作对象。

  当对端机器crash或者网络连接被断开(比如路由器不工作,网线断开等),此时发送数据给对端然后取本端socket会返回ETIMEDOUT或者EHOSTUNREACH 或者ENETUNREACH(后两个是中间路由器判断服务器主机不可达的情况)。

  当对端机器crash之后重新启动,然后客户端再向原来的连接发送数据,因为服务器端已经没有原来的连接信息,此时服务器端回送RST给客户端,此时客户端本地端口返回ECONNRESET错误。

  当服务器所在的进程正常或者异常关闭时,会对所有打开的文件描述符进行close,因此对于连接的socket描述符则会向对端发送FIN分节进行正常关闭流程。对端在收到FIN之后端口变得,此时取端口会返回0表示到了文件结尾(对端不会再发送数据) 

  当一端收到RST导致socket返回ECONNRESET,此时如果再次调用write发送数据给对端则触发SIGPIPE信号,信号默认终止进程,如果忽略此信号或者从SIGPIPE的信号处理程序返回则write出错返回EPIPE。

  可以看出只有当本地端口主动发送消息给对端才能检测出连接异常中断的情况,搭配select进行多路分离的时候,socket收到RST或者FIN时候,select返回(心跳消息就是用于检测连接的状态)。也可以使用socket的KEEPLIVE选项,依赖socket本身侦测socket连接异常中断的情况。


  发送socket数据有以下方法:

  调用ssize_t send(int sockfd, const void *buf, size_t nbytes, int flags);,只能用于建立好了连接的socket(面向连接的SOCK_STREAM或者调用了connect的SOCK_DGRAM)。flags取值如下:

  MSG_DONTROUTE 对数据不进行路由

  MSG_DONTWAIT 不等待数据发送完成

  MSG_EOR 数据包结尾

  MSG_OOB 带外数据

  注意send函数成功返回并不代表对端一定收到了发送的消息,另外对于数据报协议如果发送的数据大于一个数据报长度则发送失败(errno设置为EMSGSIZE)。

linux 客户端 Socket 非阻塞connect编程(正文)

linux 客户端 Socket 非阻塞connect编程(正文)/*开发过程与源码解析

  开发测试环境:虚拟机CentOS,windows网络调试助手
  非阻塞模式有3种用途

  1.三次握手同时做其他的处理。connect要花一个往返时间完成,从几毫秒的局域网到几百毫秒或几秒的广域网。这段时间可能有一些其他的处理要执行,比如数据准备,预处理等。
  2.用这种技术建立多个连接。这在web浏览器中很普遍.
  3.由于程序用select等待连接完成,可以设置一个select等待时间限制,从而缩短connect超时时间。多数实现中,connect的超时时间在75秒到几分钟之间。有时程序希望在等待一定时间内结束,使用非阻塞connect可以防止阻塞75秒,在多线程网络编程中,尤其必要。 例如有一个通过建立线程与其他主机进行socket通信的应用程序,如果建立的线程使用阻塞connect与远程通信,当有几百个线程并发的时候,由于网络延迟而全部阻塞,阻塞的线程不会释放系统的资源,同一时刻阻塞线程超过一定数量时候,系统就不再允许建立新的线程(每个进程由于进程空间的原因能产生的线程有限),如果使用非阻塞的connect,连接失败使用select等待很短时间,如果还没有连接后,线程立刻结束释放资源,防止大量线程阻塞而使程序崩溃。

  目前connect非阻塞编程的普遍思路是:
  在一个TCP套接口设置为非阻塞后,调用connect,connect会在系统提供的errno变量中返回一个EINRPOCESS错误,此时TCP的三路握手继续进行。之后可以用select函数检查这个连接是否建立成功。以下实验基于unix网络编程和网络上给出的普遍示例,在经过大量测试之后,发现其中有很多方法,在linux中,并不适用。

  我先给出了重要源码的逐步分析,在最后给出完整的connect非阻塞源码。
  1.首先填写套接字结构,包括远程的ip,通信端口如下: */
  struct sockaddr_in serv_addr;
  serv_addr.sin_family=AF_INET;
  serv_addr.sin_port=htons(9999);
  serv_addr.sin_addr.s_addr = inet_addr("58.31.231.255"); //inet_addr转换为网络字节序
  bzero(&(serv_addr.sin_zero),8);

  // 2.建立socket套接字:
  if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
  {
  perror("socket creat error");
  return 1;
  }

  // 3.将socket建立为非阻塞,此时socket被设置为非阻塞模式
  flags = fcntl(sockfd,F_GETFL,0);//获取建立的sockfd的当前状态(非阻塞)
  fcntl(sockfd,F_SETFL,flags|O_NONBLOCK);//将当前sockfd设置为非阻塞
  /*4. 建立connect连接,此时socket设置为非阻塞,connect调用后,无论连接是否建立立即返回-1,同时将errno(包含errno.h就可以直接使用)设置为EINPROGRESS, 表示此时tcp三次握手仍旧进行,如果errno不是EINPROGRESS,则说明连接错误,程序结束。
  当客户端和服务器端在同一台主机上的时候,connect回马上结束,并返回0;无需等待,所以使用goto函数跳过select等待函数,直接进入连接后的处理部分。*/

  if ( ( n = connect( sockfd, ( struct sockaddr *)&serv_addr , sizeof(struct sockaddr)) ) < 0 )
  {
  if(errno != EINPROGRESS) return 1;
  }

  if(n==0)
  {
  printf("connect completed immediately");
  goto done;
  }

  /* 5.设置等待时间,使用select函数等待正在后台连接的connect函数,这里需要说明的是使用select监听socket描述符是否可读或者可写,如果只可写,说明连接成功,可以进行下面的操作。如果描述符既可读又可写,分为两种情况,第一种情况是socket连接出现错误(不要问为什么,这是系统规定的,可读可写时候有可能是connect连接成功后远程主机断开了连接close(socket)),第二种情况是connect连接成功,socket读缓冲区得到了远程主机发送的数据。需要通过connect连接后返回给errno的值来进行判定,或者通过调用 getsockopt(sockfd,SOL_SOCKET,SO_ERROR,&error,&len); 函数返回值来判断是否发生错误,这里存在一个可移植性问题,在solaris中发生错误返回-1,但在其他系统中可能返回0.我首先按unix网络编程的源码进行实现。如下:*/

  FD_ZERO(&rset);
  FD_SET(sockfd,&rset);
  wset = rset;
  tval.tv_sec = 0;
  tval.tv_usec = 300000;
  int error;
  socklen_t len;

  if(( n = select(sockfd+1, &rset, &wset, NULL,&tval)) <= 0)
  {
  printf("time out connect error");
  close(sockfd);
  return -1;
  }

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

  /* 这里我测试了一下,按照unix网络编程的描述,当网络发生错误的时候,getsockopt返回-1,return -1,程序结束。网络正常时候返回0,程序继续执行。
  可是我在linux下,无论网络是否发生错误,getsockopt始终返回0,不返回-1,说明linux与unix网络编程还是有些细微的差别。就是说当socket描述符可读可写的时候,这段代码不起作用。不能检测出网络是否出现故障。
  我测试的方法是,当调用connect后,sleep(2)休眠2秒,借助这两秒时间将网络助手断开连接,这时候select返回2,说明套接口可读又可写,应该是网络连接的出错情况。
  此时,getsockopt返回0,不起作用。获取errno的值,指示为EINPROGRESS,没有返回unix网络编程中说的ENOTCONN,EINPROGRESS表示正在试图连接,不能表示网络已经连接失败。
针对这种情况,unix网络编程中提出了另外3种方法,这3种方法,也是网络上给出的常用的非阻塞connect示例:
  a.再调用connect一次。失败返回errno是EISCONN说明连接成功,表示刚才的connect成功,否则返回失败。 代码如下:*/

  int connect_ok;

  connect(sockfd, (struct sockaddr *)&serv_addr, sizeof(struct sockaddr) );
  switch (errno)
  {
  case EISCONN: //connect ok
  printf("connect OK \n");
  connect_ok = 1;
  break;
  case EALREADY:
  connect_0k = -1
  break;
  case EINPROGRESS: // is connecting, need to check again
  connect_ok = -1
  break;
  default: 
  printf("connect fail err=%d \n",errno);
  connect_ok = -1;
  break;
  }

  /*如程序所示,根据再次调用的errno返回值将connect_ok的值,来进行下面的处理,connect_ok为1继续执行其他操作,否则程序结束。
  但这种方法我在linux下测试了,当发生错误的时候,socket描述符(我的程序里是sockfd)变成可读且可写,但第二次调用connect 后,errno并没有返回EISCONN,,也没有返回连接失败的错误,仍旧是EINPROGRESS,而当网络不发生故障的时候,第二次使用 connect连接也返回EINPROGRESS,因此也无法通过再次connect来判断连接是否成功。
  b.unix网络编程中说使用read函数,如果失败,表示connect失败,返回的errno指明了失败原因,但这种方法在linux上行不通,linux在socket描述符为可读可写的时候,read返回0,并不会置errno为错误。
   c.unix网络编程中说使用getpeername函数,如果连接失败,调用该函数后,通过errno来判断第一次连接是否成功,但我试过了,无论网络连接是否成功,errno都没变化,都为EINPROGRESS,无法判断。
  悲哀啊,即使调用getpeername函数,getsockopt函数仍旧不行。
  综上方法,既然都不能确切知道非阻塞connect是否成功,所以我直接当描述符可读可写的情况下进行发送,通过能否获取服务器的返回值来判断是否成功。(如果服务器端的设计不发送数据,那就悲哀了。)
  程序的书写形式出于可移植性考虑,按照unix网络编程推荐写法,使用getsocketopt进行判断,但不通过返回值来判断,而通过函数的返回参数来判断。
  6. 用select查看接收描述符,如果可读,就读出数据,程序结束。在接收数据的时候注意要先对先前的rset重新赋值为描述符,因为select会对 rset清零,当调用select后,如果socket没有变为可读,则rset在select会被置零。所以如果在程序中使用了rset,最好在使用时候重新对rset赋值。

  程序如下:*/

  FD_ZERO(&rset);
  FD_SET(sockfd,&rset);//如果前面select使用了rset,最好重新赋值

  if( ( n = select(sockfd+1,&rset,NULL, NULL,&tval)) <= 0 )
  {
  close(sockfd);
  return -1;
  } 

  if ((recvbytes=recv(sockfd, buf, 1024, 0)) ==-1)
  {
  perror("recv error!");
  close(sockfd);
  return 1;

  }
  printf("receive num %d\n",recvbytes);

  printf("%s\n",buf);

  */

非阻塞connect

在一个TCP套接口被设置为非阻塞之后调用connect,connect会立即返回EINPROGRESS错误,表示连接操作正在进行中,但是仍未完成;同时TCP的三路握手操作继续进行;在这之后,我们可以调用select来检查这个链接是否建立成功;非阻塞connect有三种用途:
1.我们可以在三路握手的同时做一些其它的处理.connect操作要花一个往返时间完成,而且可以是在任何地方,从几个毫秒的局域网到几百毫秒或几秒的广域网.在这段时间内我们可能有一些其他的处理想要执行;
2.可以用这种技术同时建立多个连接.在Web浏览器中很普遍;
3.由于我们使用select来等待连接的完成,因此我们可以给select设置一个时间限制,从而缩短connect的超时时间.在大多数实现中,connect的超时时间在75秒到几分钟之间.有时候应用程序想要一个更短的超时时间,使用非阻塞connect就是一种方法;
非阻塞connect听起来虽然简单,但是仍然有一些细节问题要处理:
1.即使套接口是非阻塞的,如果连接的服务器在同一台主机上,那么在调用connect建立连接时,连接通常会立即建立成功.我们必须处理这种情况;
2.源自Berkeley的实现(和Posix.1g)有两条与select和非阻塞IO相关的规则:
  A:当连接建立成功时,套接口描述符变成可写;
  B:当连接出错时,套接口描述符变成既可读又可写;
  注意:当一个套接口出错时,它会被select调用标记为既可读又可写;

非阻塞connect有这么多好处,但是处理非阻塞connect时会遇到很多可移植性问题;

处理非阻塞connect的步骤:
第一步:创建socket,返回套接口描述符;
第二步:调用fcntl把套接口描述符设置成非阻塞;
第三步:调用connect开始建立连接;
第四步:判断连接是否成功建立;
       A:如果connect返回0,表示连接简称成功(服务器可客户端在同一台机器上时就有可能发生这种情况);
       B:调用select来等待连接建立成功完成;
         如果select返回0,则表示建立连接超时;我们返回超时错误给用户,同时关闭连接,以防止三路握手操作继续进行下去;
         如果select返回大于0的值,则需要检查套接口描述符是否可读或可写;如果套接口描述符可读或可写,则我们可以通过调用getsockopt来得到套接口上待处理的错误(SO_ERROR),如果连接建立成功,这个错误值将是0,如果建立连接时遇到错误,则这个值是连接错误所对应的errno值(比如:ECONNREFUSED,ETIMEDOUT等).
"读取套接口上的错误"是遇到的第一个可移植性问题;如果出现问题,getsockopt源自Berkeley的实现是返回0,等待处理的错误在变量errno中返回;但是Solaris会让getsockopt返回-1,errno置为待处理的错误;我们对这两种情况都要处理;

这样,在处理非阻塞connect时,在不同的套接口实现的平台中存在的移植性问题,首先,有可能在调用select之前,连接就已经建立成功,而且对方的数据已经到来.在这种情况下,连接成功时套接口将既可读又可写.这和连接失败时是一样的.这个时候我们还得通过getsockopt来读取错误值;这是第二个可移植性问题;
移植性问题总结:
1.对于出错的套接口描述符,getsockopt的返回值源自Berkeley的实现是返回0,待处理的错误值存储在errno中;而源自Solaris的实现是返回0,待处理的错误存储在errno中;(套接口描述符出错时调用getsockopt的返回值不可移植)
2.有可能在调用select之前,连接就已经建立成功,而且对方的数据已经到来,在这种情况下,套接口描述符是既可读又可写;这与套接口描述符出错时是一样的;(怎样判断连接是否建立成功的条件不可移植)

这样的话,在我们判断连接是否建立成功的条件不唯一时,我们可以有以下的方法来解决这个问题:
1.调用getpeername代替getsockopt.如果调用getpeername失败,getpeername返回ENOTCONN,表示连接建立失败,我们必须以SO_ERROR调用getsockopt得到套接口描述符上的待处理错误;
2.调用read,读取长度为0字节的数据.如果read调用失败,则表示连接建立失败,而且read返回的errno指明了连接失败的原因.如果连接建立成功,read应该返回0;
3.再调用一次connect.它应该失败,如果错误errno是EISCONN,就表示套接口已经建立,而且第一次连接是成功的;否则,连接就是失败的;

被中断的connect:
如果在一个阻塞式套接口上调用connect,在TCP的三路握手操作完成之前被中断了,比如说,被捕获的信号中断,将会发生什么呢?假定connect不会自动重启,它将返回EINTR.那么,这个时候,我们就不能再调用connect等待连接建立完成了,如果再次调用connect来等待连接建立完成的话,connect将会返回错误值EADDRINUSE.在这种情况下,应该做的是调用select,就像在非阻塞式connect中所做的一样.然后,select在连接建立成功(使套接口描述符可写)或连接建立失败(使套接口描述符既可读又可写)时返回;

 

 

在一个 CLIENT/SERVER模型的网络应用中,客户端的调用序列大致如下:

        socket -> connect -> recv/send -> close

        其中socket没有什么可疑问的,主要是创建一个套接字用于与服务端交换数据,并且通常它会迅速返回,此时并没有数据通过网卡发送出去,而紧随其后的 connect函数则会产生网络数据的发送,TCP的三次握手也正是在此时开始,connect会先发送一个SYN包给服务端,并从最初始的CLOSED 状态进入到SYN_SENT状态,在此状态等待服务端的确认包,通常情况下这个确认包会很快到达,以致于我们根本无法使用netstat命令看到 SYN_SENT状态的存在,不过我们可以做一个极端情况的模拟,让客户端去连接一个随意指定服务器(如IP地址为88.88.88.88),因为该服务 器很明显不会反馈给我们SYN包的确认包(SYN ACK),客户端就会在一定时间内处于SYN_SENT状态,并在预定的超时时间(比如3分钟)之后从connect函数返回,connect调用一旦失 败(没能到达ESTABLISHED状态)这个套接字便不可用,若要再次调用connect函数则必须要重新使用socket函数创建新的套接字。

 

下面结合实例分析,客户端代码如下:

  1. /** 
  2.  * client.c 
  3.  * 
  4.  * TCP client program, it is a simple example only. 
  5.  * Writen By: Zhou Jianchun 
  6.  * Date: 2011.08.11 
  7.  * 
  8.  * Compiled With: gcc -o client client.c 
  9.  * Tested On: Ubuntu 11.04 LTS 
  10.  * gcc version: 4.5.2 
  11.  * 
  12.  */  
  13.   
  14. #include <stdio.h>  
  15. #include <sys/socket.h>  
  16. #include <unistd.h>  
  17. #include <sys/types.h>  
  18. #include <netinet/in.h>  
  19. #include <stdlib.h>  
  20. #include <string.h>  
  21. #include <errno.h>  
  22.   
  23. #define SERVER_PORT 20000  
  24.   
  25. void usage(char *name)  
  26. {  
  27.     printf("usage: %s IP\n", name);  
  28. }  
  29. int main(int argc, char **argv)  
  30. {  
  31.     int server_fd, client_fd, length = 0;  
  32.     struct sockaddr_in server_addr, client_addr;  
  33.     socklen_t socklen = sizeof(server_addr);  
  34.   
  35.     if(argc < 2)  
  36.     {  
  37.         usage(argv[0]);  
  38.         exit(1);  
  39.     }  
  40.     if((client_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)  
  41.     {  
  42.         printf("create socket error, exit!\n");  
  43.         exit(1);  
  44.     }  
  45.     srand(time(NULL));  
  46.     bzero(&client_addr, sizeof(client_addr));  
  47.     client_addr.sin_family = AF_INET;  
  48.     client_addr.sin_addr.s_addr = htons(INADDR_ANY);  
  49.   
  50.     bzero(&server_addr, sizeof(server_addr));  
  51.     server_addr.sin_family = AF_INET;  
  52.     inet_aton(argv[1], &server_addr.sin_addr);  
  53.     server_addr.sin_port = htons(SERVER_PORT);  
  54.   
  55.     if(connect(client_fd, (struct sockaddr*)&server_addr, socklen) < 0)  
  56.     {  
  57.         printf("can not connect to %s, exit!\n", argv[1]);  
  58.         printf("%s\n", strerror(errno));  
  59.         exit(1);  
  60.     }  
  61.     return 0;  
  62. }  

编译完成之后执行:
  1. zhou@neptune:~/data/source$ ./client 88.88.88.88  

此时程序会在connect函数中阻塞等待,约180秒之后输出:

  1. can not connect to 88.88.88.88, exit!  
  2. Connection timed out  

此刻connect的返回值为ETIMEOUT。

在此过程中我们可以用netstat命令查询连接状态:

  1. zhou@neptune:~/data/source$ sudo netstat -natp |grep 20000  
  2. tcp        0      1 192.168.0.4:44203       88.88.88.88:20000       SYN_SENT    5954/client      

可以看到此时的TCP连接状态为SYN_SENT,也就意味着发送了SYN包之后一直未得到服务端回馈SYN ACK包。

接下来我们使用这个客户端程序来连接自己的机器,测试时我的IP地址是192.168.0.4,是一个无线局域网,结果如下:

  1. zhou@neptune:~/data/source$ ./client 192.168.0.4  
  2. can not connect to 192.168.0.4, exit!  
  3. Connection refused  

因为我的机器上并没有跑在指定端口(20000)上监听的服务端程序,所以这个连接直接被协议栈拒绝(通过发送RST类型的TCP包),connect立刻返回,返回值为ECONNREFUSED。

再来看看去连接同一局域网中一台不存在的主机时的情形,比如这台想象的主机的IP地址为192.168.0.188:

 

  1. zhou@neptune:~/data/source$ ./client 192.168.0.188  
  2. can not connect to 192.168.0.188, exit!  
  3. No route to host  

因为本地局域网中的该主机并不存在,ARP请求得不到回应,网关会回应主机不可达的ICMP报文,connect返回EHOSTUNREACH。

至此connect函数的分析就结束了,由于本人水平有限,博客中的不妥或错误之处在所难免,殷切希望读者批评指正。同时也欢迎读者共同探讨相关的内容,如果乐意交流的话请留下您宝贵的意见,谢谢。

 

 

 

原来我们实现connect()超时基本上都使用unix网络编程一书的非阻塞方式(connect_nonb),今天在网上看到一篇文章,觉得很有意思,转载如下:

读Linux内核源码的时候偶然发现其connect的超时参数竟然和用SO_SNDTIMO操作的参数一致:

  File: net/ipv4/af_inet.c

    559       timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
    560
    561       if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV)) {
    562           /* Error code is set above */
    563           if (!timeo || !inet_wait_for_connect(sk, timeo))
    564               goto out;
    565
    566           err = sock_intr_errno(timeo);
    567           if (signal_pending(current))
    568               goto out;
    569       }

  这意味着:在Linux平台下,可以通过在connect之前设置SO_SNDTIMO来达到控制连接超时的目的。简单的写了份测试代码:


#include <stdlib.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <errno.h>

int main(int argc, char *argv[])
{
        int fd;
        struct sockaddr_in addr;
        struct timeval timeo = {3, 0};
        socklen_t len = sizeof(timeo);

         fd = socket(AF_INET, SOCK_STREAM, 0);
        if (argc == 4)
                 timeo.tv_sec = atoi(argv[3]);
        setsockopt(fd, SOL_SOCKET, SO_SNDTIMEO, &timeo, len);
         addr.sin_family = AF_INET;
         addr.sin_addr.s_addr = inet_addr(argv[1]);
         addr.sin_port = htons(atoi(argv[2]));
        if (connect(fd, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
                if (errno == EINPROGRESS) {
                        fprintf(stderr, "timeout/n");
                        return -1;
                }
                perror("connect");
                return 0;
        }
        printf("connected/n");

        return 0;
}

在一个 CLIENT/SERVER模型的网络应用中,客户端的调用序列大致如下:

        socket -> connect -> recv/send -> close

        其中socket没有什么可疑问的,主要是创建一个套接字用于与服务端交换数据,并且通常它会迅速返回,此时并没有数据通过网卡发送出去,而紧随其后的 connect函数则会产生网络数据的发送,TCP的三次握手也正是在此时开始,connect会先发送一个SYN包给服务端,并从最初始的CLOSED 状态进入到SYN_SENT状态,在此状态等待服务端的确认包,通常情况下这个确认包会很快到达,以致于我们根本无法使用netstat命令看到 SYN_SENT状态的存在,不过我们可以做一个极端情况的模拟,让客户端去连接一个随意指定服务器(如IP地址为88.88.88.88),因为该服务 器很明显不会反馈给我们SYN包的确认包(SYN ACK),客户端就会在一定时间内处于SYN_SENT状态,并在预定的超时时间(比如3分钟)之后从connect函数返回,connect调用一旦失 败(没能到达ESTABLISHED状态)这个套接字便不可用,若要再次调用connect函数则必须要重新使用socket函数创建新的套接字。

 

下面结合实例分析,客户端代码如下:

  1. /** 
  2.  * client.c 
  3.  * 
  4.  * TCP client program, it is a simple example only. 
  5.  * Writen By: Zhou Jianchun 
  6.  * Date: 2011.08.11 
  7.  * 
  8.  * Compiled With: gcc -o client client.c 
  9.  * Tested On: Ubuntu 11.04 LTS 
  10.  * gcc version: 4.5.2 
  11.  * 
  12.  */  
  13.   
  14. #include <stdio.h>  
  15. #include <sys/socket.h>  
  16. #include <unistd.h>  
  17. #include <sys/types.h>  
  18. #include <netinet/in.h>  
  19. #include <stdlib.h>  
  20. #include <string.h>  
  21. #include <errno.h>  
  22.   
  23. #define SERVER_PORT 20000  
  24.   
  25. void usage(char *name)  
  26. {  
  27.     printf("usage: %s IP\n", name);  
  28. }  
  29. int main(int argc, char **argv)  
  30. {  
  31.     int server_fd, client_fd, length = 0;  
  32.     struct sockaddr_in server_addr, client_addr;  
  33.     socklen_t socklen = sizeof(server_addr);  
  34.   
  35.     if(argc < 2)  
  36.     {  
  37.         usage(argv[0]);  
  38.         exit(1);  
  39.     }  
  40.     if((client_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)  
  41.     {  
  42.         printf("create socket error, exit!\n");  
  43.         exit(1);  
  44.     }  
  45.     srand(time(NULL));  
  46.     bzero(&client_addr, sizeof(client_addr));  
  47.     client_addr.sin_family = AF_INET;  
  48.     client_addr.sin_addr.s_addr = htons(INADDR_ANY);  
  49.   
  50.     bzero(&server_addr, sizeof(server_addr));  
  51.     server_addr.sin_family = AF_INET;  
  52.     inet_aton(argv[1], &server_addr.sin_addr);  
  53.     server_addr.sin_port = htons(SERVER_PORT);  
  54.   
  55.     if(connect(client_fd, (struct sockaddr*)&server_addr, socklen) < 0)  
  56.     {  
  57.         printf("can not connect to %s, exit!\n", argv[1]);  
  58.         printf("%s\n", strerror(errno));  
  59.         exit(1);  
  60.     }  
  61.     return 0;  
  62. }  

编译完成之后执行:
  1. zhou@neptune:~/data/source$ ./client 88.88.88.88  

此时程序会在connect函数中阻塞等待,约180秒之后输出:

  1. can not connect to 88.88.88.88, exit!  
  2. Connection timed out  

此刻connect的返回值为ETIMEOUT。

在此过程中我们可以用netstat命令查询连接状态:

  1. zhou@neptune:~/data/source$ sudo netstat -natp |grep 20000  
  2. tcp        0      1 192.168.0.4:44203       88.88.88.88:20000       SYN_SENT    5954/client      

可以看到此时的TCP连接状态为SYN_SENT,也就意味着发送了SYN包之后一直未得到服务端回馈SYN ACK包。

接下来我们使用这个客户端程序来连接自己的机器,测试时我的IP地址是192.168.0.4,是一个无线局域网,结果如下:

  1. zhou@neptune:~/data/source$ ./client 192.168.0.4  
  2. can not connect to 192.168.0.4, exit!  
  3. Connection refused  

因为我的机器上并没有跑在指定端口(20000)上监听的服务端程序,所以这个连接直接被协议栈拒绝(通过发送RST类型的TCP包),connect立刻返回,返回值为ECONNREFUSED。

再来看看去连接同一局域网中一台不存在的主机时的情形,比如这台想象的主机的IP地址为192.168.0.188:

 

  1. zhou@neptune:~/data/source$ ./client 192.168.0.188  
  2. can not connect to 192.168.0.188, exit!  
  3. No route to host  

因为本地局域网中的该主机并不存在,ARP请求得不到回应,网关会回应主机不可达的ICMP报文,connect返回EHOSTUNREACH。

至此connect函数的分析就结束了,由于本人水平有限,博客中的不妥或错误之处在所难免,殷切希望读者批评指正。同时也欢迎读者共同探讨相关的内容,如果乐意交流的话请留下您宝贵的意见,谢谢。

使用TCP协议进行网络通讯时,通信的两端首先需要建立起一条连接链路,当然这并不表示使用UDP通信不需要“连接链路”,这里说的连接链路指的是通信协 议范畴的东东,并不是物理介质或者电磁波信号,只所以说TCP是面向连接的网络通信协议,主要是指双方在通信时都会保持一些连接相关的信息,比如已收到的 分组的序列号,下一次需要收到的分组的序号,对方的滑动窗口信息等等。

        OK,闲话少扯,我们进入主题,下面结合一个简单的TCP服务端与客户端代码,借助tcpdump命令来分析一下TCP建立连接时的三次握手过程(Three-way handshake process)。

服务端代码如下:

  1. /** 
  2.  * server.c 
  3.  * 
  4.  * TCP server program, it is a simple example only. 
  5.  * 
  6.  * Writen By: Zhou Jianchun 
  7.  * Date: 2011.08.12 
  8.  * 
  9.  * Compiled With: gcc -o client client.c 
  10.  * Tested On: Ubuntu 11.04 LTS 
  11.  * 
  12.  * gcc version: 4.5.2 
  13.  * 
  14.  */  
  15.   
  16. #include <stdio.h>  
  17. #include <sys/socket.h>  
  18. #include <unistd.h>  
  19. #include <sys/types.h>  
  20. #include <netinet/in.h>  
  21. #include <stdlib.h>  
  22. #include <time.h>  
  23. #include <strings.h>  
  24. #include <string.h>  
  25.   
  26. #define SERVER_PORT 20000  
  27. #define LENGTH_OF_LISTEN_QUEUE 10  
  28. #define BUFFER_SIZE 255  
  29. #define WELCOME_MESSAGE "welcome to our server."  
  30.   
  31. int main(int argc, char **argv)  
  32. {  
  33.     int server_fd, client_fd;  
  34.     struct sockaddr_in server_addr, client_addr;  
  35.   
  36.     if((server_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)  
  37.     {  
  38.         printf("create socket error, exit!\n");  
  39.         exit(1);  
  40.     }  
  41.   
  42.     bzero(&server_addr, sizeof(server_addr));  
  43.     server_addr.sin_family = AF_INET;  
  44.     server_addr.sin_port = htons(SERVER_PORT);  
  45.     server_addr.sin_addr.s_addr = htons(INADDR_ANY);  
  46.   
  47.     if(bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0)  
  48.     {  
  49.         printf("bind to port %d failed, exit!\n", SERVER_PORT);  
  50.         exit(1);  
  51.     }  
  52.   
  53.     if(listen(server_fd, LENGTH_OF_LISTEN_QUEUE) < 0)  
  54.     {  
  55.         printf("failed to listen, exit!\n");  
  56.         exit(1);  
  57.     }  
  58.   
  59.     while(1)  
  60.     {  
  61.         char buf[BUFFER_SIZE];  
  62.         long timestamp;  
  63.         socklen_t length = sizeof(client_addr);  
  64.         client_fd = accept(server_fd, (struct sockaddr*)&client_addr, &length);  
  65.         if(client_fd <0)  
  66.         {  
  67.             printf("call accept error, break from while loop!\n");  
  68.             break;  
  69.         }  
  70.         strcpy(buf, WELCOME_MESSAGE);  
  71.         printf("connect from client: IP: %s, Port: %d\n", (char *)inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));  
  72.         timestamp = time(NULL);  
  73.         strcat(buf, "timestamp on server:");  
  74.         strcat(buf, ctime(×tamp));  
  75.         send(client_fd, buf, BUFFER_SIZE, 0);  
  76.         close(client_fd);  
  77.   
  78.         close(server_fd);  
  79.         return 0;  
  80.     }  
  81. }  

客户端代码:
  1. /** 
  2.  * client.c 
  3.  * 
  4.  * TCP client program, it is a simple example only. 
  5.  * 
  6.  * Writen By: Zhou Jianchun 
  7.  * Date: 2011.08.12 
  8.  * 
  9.  * Compiled With: gcc -o client client.c 
  10.  * Tested On: Ubuntu 11.04 LTS 
  11.  * 
  12.  * gcc version: 4.5.2 
  13.  * 
  14.  */  
  15.   
  16. #include <stdio.h>  
  17. #include <sys/socket.h>  
  18. #include <unistd.h>  
  19. #include <sys/types.h>  
  20. #include <netinet/in.h>  
  21. #include <stdlib.h>  
  22. #include <string.h>  
  23.   
  24. #define SERVER_PORT 20000  
  25. #define CLIENT_PORT ((20001 + rand()) % 65536)  
  26. #define BUFFER_SIZE 255  
  27. #define REQUEST_MESSAGE "welcome to connect the server.\n"  
  28.   
  29. void usage(char *name)  
  30. {  
  31.     printf("usage: %s IP\n", name);  
  32. }  
  33. int main(int argc, char **argv)  
  34. {  
  35.     int server_fd, client_fd, length = 0;  
  36.     struct sockaddr_in server_addr, client_addr;  
  37.     socklen_t socklen = sizeof(server_addr);  
  38.     char buf[BUFFER_SIZE];  
  39.   
  40.     if(argc < 2)  
  41.     {  
  42.         usage(argv[0]);  
  43.         exit(1);  
  44.     }  
  45.     if((client_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)  
  46.     {  
  47.         printf("create socket error, exit!\n");  
  48.         exit(1);  
  49.     }  
  50.     srand(time(NULL));  
  51.     bzero(&client_addr, sizeof(client_addr));  
  52.     client_addr.sin_family = AF_INET;  
  53.     //client_addr.sin_port = htons(CLIENT_PORT);  
  54.     client_addr.sin_port = htons(40000);  
  55.     client_addr.sin_addr.s_addr = htons(INADDR_ANY);  
  56.   
  57.     bzero(&server_addr, sizeof(server_addr));  
  58.     server_addr.sin_family = AF_INET;  
  59.     inet_aton(argv[1], &server_addr.sin_addr);  
  60.     server_addr.sin_port = htons(SERVER_PORT);  
  61.   
  62.     /*if(bind(client_fd, (struct sockaddr*)&client_addr, sizeof(client_addr)) < 0) 
  63.     { 
  64.         printf("bind to port %d failed, exit!\n", CLIENT_PORT); 
  65.         exit(1); 
  66.     }*/  
  67.   
  68.     if(connect(client_fd, (struct sockaddr*)&server_addr, socklen) < 0)  
  69.     {  
  70.         printf("can not connect to %s, exit!\n", argv[1]);  
  71.         exit(1);  
  72.     }  
  73.   
  74.     /*length = recv(client_fd, buf, BUFFER_SIZE, 0); 
  75.     if(length < 0) 
  76.     { 
  77.         printf("recieve data from %s error, exit!\n", argv[1]); 
  78.         exit(1); 
  79.     } 
  80.     */  
  81.     char *tmp = buf;  
  82.     while((length = read(client_fd, tmp, BUFFER_SIZE)) > 0)  
  83.     {  
  84.         tmp += length;  
  85.     }  
  86.     printf("frome server %s:\n\t%s", argv[1], buf);  
  87.     close(client_fd);  
  88.     return 0;  
  89. }  

代码逻辑十分简单,服务端程序启动后监听在20000端口,等待外部连接,客户端启动后连接过来,服务端发送一串字符串信息给客户端,然后退出,客户端在读取完信息后也退出。

运行程序之前先在另一个终端下输入如下命令:

tcpdump 'port 20000' -i lo -S

待两端程序退出后可以看到该命令输出如下信息:

  1. 17:05:35.358403 IP neptune.local.49493 > neptune.local.20000: Flags [S], seq 1317094743, win 32792, options [mss 16396,sackOK,TS val 7083694 ecr 0,nop,wscale 6], length 0  
  2. 17:05:35.358439 IP neptune.local.20000 > neptune.local.49493: Flags [S.], seq 1311370954, ack 1317094744, win 32768, options [mss 16396,sackOK,TS val 7083694 ecr 7083694,nop,wscale 6], length 0  
  3. 17:05:35.358468 IP neptune.local.49493 > neptune.local.20000: Flags [.], ack 1311370955, win 513, options [nop,nop,TS val 7083694 ecr 7083694], length 0  
  4. 17:05:35.358871 IP neptune.local.20000 > neptune.local.49493: Flags [P.], seq 1311370955:1311371210, ack 1317094744, win 512, options [nop,nop,TS val 7083694 ecr 7083694], length 255  
  5. 17:05:35.358890 IP neptune.local.49493 > neptune.local.20000: Flags [.], ack 1311371210, win 530, options [nop,nop,TS val 7083694 ecr 7083694], length 0  
  6. 17:05:35.358913 IP neptune.local.20000 > neptune.local.49493: Flags [F.], seq 1311371210, ack 1317094744, win 512, options [nop,nop,TS val 7083694 ecr 7083694], length 0  
  7. 17:05:35.359419 IP neptune.local.49493 > neptune.local.20000: Flags [F.], seq 1317094744, ack 1311371211, win 530, options [nop,nop,TS val 7083694 ecr 7083694], length 0  
  8. 17:05:35.359441 IP neptune.local.20000 > neptune.local.49493: Flags [.], ack 1317094745, win 512, options [nop,nop,TS val 7083694 ecr 7083694], length 0  

下面我们逐条进行分析:

1.客户端通过49493端口向服务端的20000端口发送一个SYN同步请求包,展开第一次握手,其中Flags [S]表求数据包的类型为SYN, 即同步请求包,seq字段标识数据包序列号。

2.服务端发送ACK确认包,同时附代一个SYN请求包,在确认客户端同步请求的同时 向客户端发送同步请求,其中Flags [S.]中的点号表示这是个确认包(ACK),S表示它同时又是一个SYN请求包。因为TCP是双工通信协议,连接建立之后双方可以同时收发数据,所以双 方都发送了SYN包请求同步。

3.客户端发送ACK包确认服务端的SYN同步请求,可以看到此时Flags中只有一个小数点,表示这个包只是用来做确认的。

到此为止,三次握手过程就结束了,双方如果都收到了ACK包,则都进入到ESTABLISHED状态,表明此时可以进行数据发送了。

4.服务端向客户端发送一个数据包,包中的内容就是一个字符串,可以看到此时的Flags标识中有个字母P,意为PUSH DATA,就是发送数据的意思。

至此TCP三次握手过程的分析就结束了,由于本人水平有限,博客中的不妥或错误之处在所难免,殷切希望读者批评指正。同时也欢迎读者共同探讨相关的内容,如果乐意交流的话请留下您宝贵的意见,谢谢。

1、建立连接协议(三次握手)

1)客户端发送一个带SYN标志的TCP报文到服务器。这是三次握手过程中的报文1

2) 服务器端回应客户端的,这是三次握手中的第2个报文,这个报文同时带ACK标志和SYN标志。因此它表示对刚才客户端SYN报文的回应;同时又标志SYN给客户端,询问客户端是否准备好进行数据通讯。

3) 客户必须再次回应服务段一个ACK报文,这是报文段3

2、连接终止协议(四次握手)
   由于TCP连接是全双工的,因此每个方向都必须单独进行关闭。这原则是当一方完成它的数据发送任务后就能发送一个FIN来终止这个方向的连接。收到一个 FIN只意味着这一方向上没有数据流动,一个TCP连接在收到一个FIN后仍能发送数据。首先进行关闭的一方将执行主动关闭,而另一方执行被动关闭。

 (1 TCP客户端发送一个FIN,用来关闭客户到服务器的数据传送(报文段4)。
 (2) 服务器收到这个FIN,它发回一个ACK,确认序号为收到的序号加1(报文段5)。和SYN一样,一个FIN将占用一个序号。
 (3) 服务器关闭客户端的连接,发送一个FIN给客户端(报文段6)。
 (4) 客户段发回ACK报文确认,并将确认序号设置为收到序号加1(报文段7)。

CLOSED: 这个没什么好说的了,表示初始状态。

LISTEN: 这个也是非常容易理解的一个状态,表示服务器端的某个SOCKET处于监听状态,可以接受连接了。

SYN_RCVD: 这个状态表示接受到了SYN报文,在正常情况下,这个状态是服务器端的SOCKET在建立TCP连接时的三次握手会话过程中的一个中间状态,很短暂,基本上用netstat你是很难看到这种状态的,除非你特意写了一个客户端测试程序,故意将三次TCP握手过程中最后一个ACK报文不予发送。因此这种状态时,当收到客户端的ACK报文后,它会进入到ESTABLISHED状态。

SYN_SENT: 这个状态与SYN_RCVD遥想呼应,当客户端SOCKET执行CONNECT连接时,它首先发送SYN报文,因此也随即它会进入到了SYN_SENT状态,并等待服务端的发送三次握手中的第2个报文。SYN_SENT状态表示客户端已发送SYN报文。

ESTABLISHED:这个容易理解了,表示连接已经建立了。

FIN_WAIT_1: 这个状态要好好解释一下,其实FIN_WAIT_1FIN_WAIT_2状态的真正含义都是表示等待对方的FIN报文。而这两种状态的区别是:FIN_WAIT_1状态实际上是当SOCKETESTABLISHED状态时,它想主动关闭连接,向对方发送了FIN报文,此时该SOCKET即进入到FIN_WAIT_1状态。而当对方回应ACK报文后,则进入到FIN_WAIT_2状态,当然在实际的正常情况下,无论对方何种情况下,都应该马上回应ACK报文,所以FIN_WAIT_1状态一般是比较难见到的,而FIN_WAIT_2状态还有时常常可以用netstat看到。

FIN_WAIT_2:上面已经详细解释了这种状态,实际上FIN_WAIT_2状态下的SOCKET,表示半连接,也即有一方要求close连接,但另外还告诉对方,我暂时还有点数据需要传送给你,稍后再关闭连接。

TIME_WAIT: 表示收到了对方的FIN报文,并发送出了ACK报文,就等2MSL后即可回到CLOSED可用状态了。如果FIN_WAIT_1状态下,收到了对方同时带FIN标志和ACK标志的报文时,可以直接进入到TIME_WAIT状态,而无须经过FIN_WAIT_2状态。

CLOSING: 这种状态比较特殊,实际情况中应该是很少见,属于一种比较罕见的例外状态。正常情况下,当你发送FIN报文后,按理来说是应该先收到(或同时收到)对方的ACK报文,再收到对方的FIN报文。但是CLOSING状态表示你发送FIN报文后,并没有收到对方的ACK报文,反而却也收到了对方的FIN报文。什么情况下会出现此种情况呢?其实细想一下,也不难得出结论:那就是如果双方几乎在同时close一个SOCKET的话,那么就出现了双方同时发送FIN报文的情况,也即会出现CLOSING状态,表示双方都正在关闭SOCKET连接。

CLOSE_WAIT: 这种状态的含义其实是表示在等待关闭。怎么理解呢?当对方close一个SOCKET后发送FIN报文给自己,你系统毫无疑问地会回应一个ACK报文给对方,此时则进入到CLOSE_WAIT状态。接下来呢,实际上你真正需要考虑的事情是察看你是否还有数据发送给对方,如果没有的话,那么你也就可以close这个SOCKET,发送FIN报文给对方,也即关闭连接。所以你在CLOSE_WAIT状态下,需要完成的事情是等待你去关闭连接。

LAST_ACK: 这个状态还是比较容易好理解的,它是被动关闭一方在发送FIN报文后,最后等待对方的ACK报文。当收到ACK报文后,也即可以进入到CLOSED可用状态了。

最后有2个问题的回答,我自己分析后的结论(不一定保证100%正确)

1、 为什么建立连接协议是三次握手,而关闭连接却是四次握手呢?

这是因为服务端的LISTEN状态下的SOCKET当收到SYN报文的建连请求后,它可以把ACKSYNACK起应答作用,而SYN起同步作用)放在一个报文里来发送。但关闭连接时,当收到对方的FIN报文通知时,它仅仅表示对方没有数据发送给你了;但未必你所有的数据都全部发送给对方了,所以你可以未必会马上会关闭SOCKET,也即你可能还需要发送一些数据给对方之后,再发送FIN报文给对方来表示你同意现在可以关闭连接了,所以它这里的ACK报文和FIN报文多数情况下都是分开发送的。

2、 为什么TIME_WAIT状态还需要等2MSL后才能返回到CLOSED状态?

这是因为:虽然双方都同意关闭连接了,而且握手的4个报文也都协调和发送完毕,按理可以直接回到CLOSED状态(就好比从SYN_SEND状态到ESTABLISH状态那样);但是因为我们必须要假想网络是不可靠的,你无法保证你最后发送的ACK报文会一定被对方收到,因此对方处于LAST_ACK状态下的SOCKET可能会因为超时未收到ACK报文,而重发FIN报文,所以这个TIME_WAIT状态的作用就是用来重发可能丢失的ACK报文,并保证于此。

 

posted @ 2013-04-02 16:47  晨光静默  阅读(8496)  评论(0编辑  收藏  举报