如何让recv函数有点脾气？（让你知道select函数的威力）

最新推荐文章于 2024-08-01 10:47:47 发布

涛歌依旧

最新推荐文章于 2024-08-01 10:47:47 发布

阅读量1.3w

点赞数 5

本文链接：https://blog.csdn.net/stpeace/article/details/21416645

版权

s2: 软件进阶同时被 3 个专栏收录

1053 篇文章 76 订阅

订阅专栏

S1: C/C++

972 篇文章 329 订阅

订阅专栏

s2: 网络编程

147 篇文章 46 订阅

订阅专栏

我们知道， recv是一个阻塞的函数，没有到数据的时候，会傻傻地等待数据。那么怎么让recv函数有点脾气，不傻傻地等待呢？

服务器为(先跑起来)：

#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));

	// 将socket设置为监听模式，5表示等待连接队列的最大长度
	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;
}

我们看到，当建立一个TCP连接后，服务器需要等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;
}

运行客户端程序，可以发现，尽管服务器故意延时了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);
	
	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;
}

这个客户的耐心只有4秒，他发飙说：服务器，你妈的，在我跟你建立TCP关系后，你如果4秒之内没发数据，我就不鸟你了。运行程序，可以看到，这个客户依然可以接受到数据。

下面我们再看看，耐心只有2秒的客户：

#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 = {2, 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;
}

这个客户的脾气够火爆啊，等了2秒，就不等了，所以就没有收到数据。

在实际的软件开发中，常常要用到超时机制，可以避免客户端死等。

多的不闲扯了。最后，非常非常有趣的是：如果改为