网络编程二、回声服务器

涉及到的Windows系统的API参考：https://blog.csdn.net/SNAKEpc12138/article/details/130370222

服务器端

#include <WinSock2.h>
#pragma comment(lib, "ws2_32.lib")

int main()
{
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
	{
		return -1;
	}

	SOCKET srvSock = socket(AF_INET, SOCK_STREAM, 0);
	if (INVALID_SOCKET == srvSock)
	{
		WSACleanup();
		return -1;
	}

	SOCKADDR_IN srvAddr;
	memset(&srvAddr, 0, sizeof(srvAddr));
	srvAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
	srvAddr.sin_family = AF_INET;
	srvAddr.sin_port = htons(6000);
	if (SOCKET_ERROR == bind(srvSock, (SOCKADDR*)&srvAddr, sizeof(SOCKADDR)))
	{
		closesocket(srvSock);
		WSACleanup();
		return -1;
	}

	if (SOCKET_ERROR == listen(srvSock, 5))
	{
		closesocket(srvSock);
		WSACleanup();
		return -1;
	}

	SOCKET cltSock;
	SOCKADDR_IN cltAddr;
	memset(&cltAddr, 0, sizeof(cltAddr));
	int cltAddrLen = sizeof(cltAddr);
	char buf[100] = { 0 };

	fputs("服务器初始化成功,等待客户端连接.\n", stdout);
	cltSock = accept(srvSock, (SOCKADDR*)&cltAddr, &cltAddrLen);
	if (cltSock == INVALID_SOCKET)
	{
		closesocket(srvSock);
		WSACleanup();
		return -1;
	}
	
	while (1)
	{
		int recvLen = recv(cltSock, buf, 99, 0);
		//判断接收状态
		if (recvLen == -1)
		{
			//客户端强制掉线
			fputs("客户端已掉线...\n", stdout);
			break;
		}
		else if (recvLen == 0)
		{
			//客户端请求断开连接
			fputs("客户端请求断开连接...\n", stdout);
			break;
		}

		fputs(buf, stdout);
		//回发
		int sendLen = send(cltSock, buf, recvLen, 0);
	}
	closesocket(cltSock);
	closesocket(srvSock);
	WSACleanup();
	return 0;
}

客户端

#include <WinSock2.h>
#include <WS2tcpip.h>
#pragma comment(lib, "ws2_32.lib")


int main()
{
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
	{
		return -1;
	}

	SOCKET cltSock = socket(AF_INET, SOCK_STREAM, 0);
	if (INVALID_SOCKET == cltSock)
	{
		WSACleanup();
		return -1;
	}
	
	//服务器端地址
	SOCKADDR_IN srvAddr;
	memset(&srvAddr, 0, sizeof(srvAddr));
	inet_pton(AF_INET, "127.0.0.1", &srvAddr.sin_addr.S_un.S_addr); //inet_addr("127.0.0.1");
	srvAddr.sin_family = AF_INET;
	srvAddr.sin_port = htons(6000);

	//连接到服务器端
	if (SOCKET_ERROR == connect(cltSock, (SOCKADDR*)&srvAddr, sizeof(SOCKADDR)))
	{
		closesocket(cltSock);
		WSACleanup();
		return -1;
	}

	char buf[100] = { 0 };
	while (TRUE)
	{
		fputs("Input send msg(q to quit): ", stdout);
		fgets(buf, 99, stdin);
		if (!strcmp(buf, "q\n") || !strcmp(buf, "Q\n"))
		{
			fputs("请求关闭连接.", stdout);
			break;
		}

		int sendLen = send(cltSock, buf, strlen(buf), 0);

		int recvLen = recv(cltSock, buf, 99, 0);
		if (recvLen == -1)
		{
			//服务器掉线
			fputs("服务器已掉线.", stdout);
			break;
		}
		fputs(buf, stdout);
	}
	closesocket(cltSock);
	WSACleanup();

	return 0;
}

回声客户端的问题

上面客户端代码存在接受数据不完整的问题，即客户端发送数据后马上就调用recv接收输入缓冲区数据。倘若服务器还未回发完成，或因数据过长，服务器分两次回发均会导致客户端马上调用recv接收到的数据是不完整的。

这里我们多发几次模拟问题：

#include <WinSock2.h>
#include <WS2tcpip.h>
#pragma comment(lib, "ws2_32.lib")


int main()
{
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
	{
		return -1;
	}

	SOCKET cltSock = socket(AF_INET, SOCK_STREAM, 0);
	if (INVALID_SOCKET == cltSock)
	{
		WSACleanup();
		return -1;
	}
	
	//服务器端地址
	SOCKADDR_IN srvAddr;
	memset(&srvAddr, 0, sizeof(srvAddr));
	inet_pton(AF_INET, "127.0.0.1", &srvAddr.sin_addr.S_un.S_addr); //inet_addr("127.0.0.1");
	srvAddr.sin_family = AF_INET;
	srvAddr.sin_port = htons(6000);

	//连接到服务器端
	if (SOCKET_ERROR == connect(cltSock, (SOCKADDR*)&srvAddr, sizeof(SOCKADDR)))
	{
		closesocket(cltSock);
		WSACleanup();
		return -1;
	}

	char buf[100] = { 0 };
	while (TRUE)
	{
		fputs("Input send msg(q to quit): ", stdout);
		fgets(buf, 99, stdin);
		if (!strcmp(buf, "q\n") || !strcmp(buf, "Q\n"))
		{
			fputs("请求关闭连接.", stdout);
			break;
		}

		//这里多次发送模拟回声客户端问题
		int sendLen = 0;
		sendLen  += send(cltSock, buf, strlen(buf), 0);
		sendLen  += send(cltSock, buf, strlen(buf), 0);
		sendLen  += send(cltSock, buf, strlen(buf), 0);
		sendLen  += send(cltSock, buf, strlen(buf), 0);
		sendLen  += send(cltSock, buf, strlen(buf), 0);

		int recvLen = recv(cltSock, buf, 99, 0);
		if (recvLen == -1)
		{
			//服务器掉线
			fputs("服务器已掉线.", stdout);
			break;
		}
		fputs(buf, stdout);
	}
	closesocket(cltSock);
	WSACleanup();

	return 0;
}

运行结果如下：

可以看到我们第一次发了5次123，按理来说服务器应该会回发5次123才对。然而这里客户端只收到了一次123。

是服务器只回了一次吗？

其实服务器是有回5次的，但客户端并没有等服务器5次回复完才recv，即服务器可能只回复了一次，客户端就recv了，当前客户端的输入缓冲中就只有一次回复123.

下次输入456，再recv时，会把客户端输入缓冲中的所有数据读出来，所以才会看到四次123（上次未读完的）和一次456（此次未回复完的）数据。

解决回声客户端问题

这里有两种方式解决上述问题。第一是延时recv，第二种是明确接收的数据长度。

延时recv

调整客户端代码如下：

//上下文同上，不再赘述
//这里只调整收发部分的代码
		sendLen = 0;
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		

		//回声客户端的问题：这里send后马上recv，当send的字符串过大，服务器防分多次会发时，这里马上recv有可能没有收到全部的数据
		//延时recv解决：问题是不知延时多久。
		Sleep(3000);
		strLen = recv(srvSock, Msg, BUF_SIZE - 1, 0);
		Msg[strLen] = 0;

再次运行可以看到能够完整的接收到回声数据：

当然这样也有问题，具体延时多长时间需要根据需要接收数据的长度来确定，即我们不知道服务器会发多少数据到客户端，所以这并不是完美解决回声客户端问题的方法。

明确接收的数据长度

虽然我们不知道应该服务器会发多少到客户端，但客户端知道自己发了多少数据到服务器。因此接收时只接收发送的数据长度即可。这才是回声客户端的完美实现。

		sendLen = 0;
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		sendLen += send(srvSock, Msg, strlen(Msg), 0);
		

		//回声客户端的问题：这里send后马上recv，当send的字符串过大，服务器防分多次会发时，这里马上recv有可能没有收到全部的数据
		//延时recv解决：问题是不知延时多久。
		//Sleep(3000);
		//strLen = recv(srvSock, Msg, BUF_SIZE - 1, 0);
		//Msg[strLen] = 0;

		//正确解决：
		strLen = 0;
		while (strLen < sendLen)
		{
			int recvLen = recv(srvSock, &Msg[strLen], BUF_SIZE - 1, 0);
			if (recvLen == -1)
			{
				closesocket(srvSock);
				WSACleanup();
				return -1;
			}
			strLen += recvLen;
		}

运行结果如下：

总结

通过回声服务器/客户端实例，我们更加深刻的理解了TCP传输数据不存在边界的特性，即recv函数和send函数的调用次数并无太大意义，因为每次recv都会获取输入缓冲中的所有内容。也更加深刻理解了recv和send函数的应用，这里通过send和recv函数的返回值来实现完美的回声客户端。