自己动手实现socket的各种超时控制

在高并发的网络编程中，如果对socket的超时不加以控制，程序会死的很辛苦。

在这里，需要控制超时时间的地方基本上分为三个部分：连接超时、读超时以及写超时。

下面将按照上述超时问题依次解决之。

连接超时

处理连接超时有若干种方案，第一种比较trick， 使用socket本身的属性进行设置。

可以使用该属性进行设置的原因是connect的内核实现（ 源文件地址）采用了如下的实现方式：

 /*
  *      Connect to a remote host. There is regrettably still a little
  *      TCP 'magic' in here.
  */
 int __inet_stream_connect(struct socket *sock, struct sockaddr *uaddr,
                           int addr_len, int flags)
 {
        //省略之前的代码
        timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
 
         if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV)) {
                 int writebias = (sk->sk_protocol == IPPROTO_TCP) &&
                                 tcp_sk(sk)->fastopen_req &&
                                 tcp_sk(sk)->fastopen_req->data ? 1 : 0;
 
                 /* Error code is set above */
                 if (!timeo || !inet_wait_for_connect(sk, timeo, writebias))
                         goto out;
 
                 err = sock_intr_errno(timeo);
                 if (signal_pending(current))
                         goto out;
         }
         //省略之后的代码
 }
 EXPORT_SYMBOL(__inet_stream_connect);

该函数的实现使用了socket的发送超时时间的设定。

因此，对connect进行超时控制的时候，可以采用如下的代码对socket的发送超时时间进行设置，具体流程可以描述为：

1. 创建socket
2. 使用setsockopt函数设定socket的具体设置，在这里设定发送超时
3. 最后使用connect执行连接的操作。

代码如下：

    int sock_fd = socket(AF_INET, SOCK_STREAM, 0); 
    if (sock_fd < 0) {
        perror("create sock_fd");
        exit(-1);
    }   
    struct sockaddr_in addr;
    bzero(&addr,sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port=htons(g_conf.aod_port);
    addr.sin_addr.s_addr = inet_addr(g_conf.aod_host);

    struct timeval timeo = {0, 0}; 
    socklen_t len = sizeof(timeo);
    timeo.tv_usec = 10*1000;//10ms
    setsockopt(sock_fd, SOL_SOCKET, SO_SNDTIMEO, &timeo, len);

    if(connect(sock_fd, (sockaddr*)&addr,sizeof(addr))<0){
        fprintf(,stdout,"%s","CONNECT WITH AOD SERVER FAILED");
        return -1;
    }

另外一种对connect设定超时的方法是使用网上广为流传的select/poll方法。这种方法来自于connect函数的manuel，使用man connect，可以看见connect函数返回值的如下描述：

       EINPROGRESS
              The  socket  is non-blocking and the connection cannot be completed immediately.  It is possible to select(2) or
              poll(2) for completion by selecting the socket for writing. After select(2) indicates writability, use  getsock-
              opt(2)  to  read  the  SO_ERROR option at level SOL_SOCKET to determine whether connect() completed successfully
              (SO_ERROR is zero) or unsuccessfully (SO_ERROR is one of the usual error codes listed here, explaining the  rea-
              son for the failure).

根据上述的描述，我们可以使用connect的返回值做一些文章，从而完成对connect的设置。其实上述的手册已经给定了一个实现connect超时设定的基本步骤，这种方法比之前的方法复杂一些：

1. 建立socket的文件描述符
2. 将socket设置为非阻塞的连接模式
3. 调用connect函数，对connect的返回值进行判定
4. 使用select检查该socket的文件描述符是否可写
5. 将socket重新设置为阻塞的模式