如何让recv有点脾气？- C

大家好，我是涛哥。

今天，我们来聊聊TCP网络编程中的recv函数。我们知道，recv是一个阻塞的函数，没有收到数据的时候，会傻傻地等待数据。

那么，怎么让recv函数有点脾气，而不傻傻地等待呢？select函数可以顶上。接下来，我们实战一下，加深对网络编程的理解。

涛哥手绘

一. 没脾气的recv函数

当recv函数没脾气时，仅会傻傻地等待，直到地老天荒。我们看下服务端程序，并先让服务端跑起来。

我们看到，当建立一个TCP连接后，服务器故意等3秒才发送数据，那么客户端等得起吗？有耐心吗？


 
#include <stdio.h>
#include <winsock2.h> // winsock接口
#pragma comment(lib, "ws2_32.lib") // winsock实现
 
int main()
{
  WORD wVersionRequested;  // 双字节，winsock库的版本
  WSADATA wsaData;         // winsock库版本的相关信息
  
  wVersionRequested = MAKEWORD(1, 1); // 0x0101 即:257
  
  // 加载winsock库并确定winsock版本，系统会把数据填入wsaData中
  WSAStartup( wVersionRequested, &wsaData );
 
  // AF_INET 表示采用TCP/IP协议族
  // SOCK_STREAM 表示采用TCP协议
  // 0是通常的默认情况
  unsigned int sockSrv = socket(AF_INET, SOCK_STREAM, 0);
 
  SOCKADDR_IN addrSrv;
 
  addrSrv.sin_family = AF_INET; // TCP/IP协议族
  addrSrv.sin_addr.S_un.S_addr = inet_addr("0.0.0.0"); 
  addrSrv.sin_port = htons(8888); // socket对应的端口
 
  // 将socket绑定到某个IP和端口（IP标识主机，端口标识通信进程）
  bind(sockSrv,(SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
 
  listen(sockSrv, 5);
 
  SOCKADDR_IN addrClient;
  int len = sizeof(SOCKADDR);
  
  while(1)
  {
    unsigned int sockConn = accept(sockSrv,(SOCKADDR*)&addrClient, &len);
    printf("accept is returned\n");
 
    Sleep(3000);
    char sendBuf[100] = "hello";
    send(sockConn, sendBuf, strlen(sendBuf) + 1, 0); // 发送数据到客户端，最后一个参数一般设置为0
    //closesocket(sockConn);  
  }
 
  closesocket(sockSrv);
  WSACleanup();
  
  return 0;
}

接着，我们看客户端程序，并让客户端也跑起来。运行客户端程序，可以发现，尽管服务器故意延时了3秒，但客户有无限的耐心，仍然能收到数据。


 
#include <winsock2.h>
#include <stdio.h>
#pragma comment(lib, "ws2_32.lib")
 
int main()
{
  WORD wVersionRequested;
  WSADATA wsaData;
  wVersionRequested = MAKEWORD(1, 1);
  
  WSAStartup( wVersionRequested, &wsaData );
 
  SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0);
 
  SOCKADDR_IN addrSrv;
  addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
  addrSrv.sin_family = AF_INET;
  addrSrv.sin_port = htons(8888);
 
  int ret = connect(sockClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
  printf("connect ret is %d\n", ret);
 
      
  char recvBuf[200] = "x";
  ret = recv(sockClient, recvBuf, 100, 0);
  if(0 > ret)
  {
    printf("recv error");
    return -3;
  }
 
  if(0 == ret)
  {
    printf("recv ret is %d\n", ret);
    return -4;
  }
  
  printf("%s\n", recvBuf);
 
  while(1);
 
  closesocket(sockClient);
  WSACleanup();
 
  return 0;
}

二. 有脾气的recv函数

recv函数该有点脾气了，等待4秒后，你还不来，我就不等了。怎么实现？可用select函数的超时特性，客户端程序如下：


 
#include <winsock2.h>
#include <stdio.h>
#pragma comment(lib, "ws2_32.lib")
 
int main()
{
  WORD wVersionRequested;
  WSADATA wsaData;
  wVersionRequested = MAKEWORD(1, 1);
  
  WSAStartup( wVersionRequested, &wsaData );
 
  SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0);
 
  SOCKADDR_IN addrSrv;
  addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
  addrSrv.sin_family = AF_INET;
  addrSrv.sin_port = htons(8888);
 
  int ret = connect(sockClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
  printf("connect ret is %d\n", ret);
  
  fd_set rfds;
  struct timeval timeout = {4, 0};
 
  FD_ZERO(&rfds); 
  FD_SET(sockClient, &rfds);
  ret = select(-1, &rfds, NULL, NULL, &timeout);
  printf("select ret is %d\n",ret);
  if(0 > ret)
  {
    printf("select error\n");
    return -1;
  }
   
  if(0 == ret)
  {
    printf("time out\n");
    return -2;
  }
    
  if(FD_ISSET(sockClient, &rfds))
  {
    char recvBuf[200] = "x";
    ret = recv(sockClient, recvBuf, 100, 0);
    if(0 > ret)
    {
      printf("recv error");
      return -3;
    }
 
    if(0 == ret)
    {
      printf("recv ret is %d, buf is %s\n", ret, recvBuf);
      return -4;
    }
    
    printf("%s\n", recvBuf);
    }
 
  while(1);
  closesocket(sockClient);
  WSACleanup();
 
  return 0;
}