Windows网络编程 I/O复用的多人聊天室

代码：

https://github.com/ChristmasError/TCP-IP-Network-programming/tree/master/WINDOWS%20IO%E5%A4%8D%E7%94%A8%E5%A4%9A%E4%BA%BA%E8%81%8A%E5%A4%A9%E5%AE%A4%E5%AE%9E%E7%8E%B0

请多开几个客户端进行测试。

//server
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
#include  
#include 
#include 
#include//定义socklen_t

using namespace std;


#pragma comment(lib, "WS2_32")  // 链接到WS2_32.lib


#define SERVER_IP "127.0.0.1"// 默认服务器端IP地址
#define SERVER_PORT 8888// 服务器端口号


class server
{
public:
	server();
	void init();
	void process();

private:
	int listener;//监听套接字
	sockaddr_in  serverAddr;//IPV4的地址方式
	vector  socksArr;//存放创建的套接字
};
int main()
{
	server ser;
	ser.process();

	return 0;
}


server::server()
{
	listener = 0;
	serverAddr.sin_family = PF_INET;
	serverAddr.sin_port = SERVER_PORT;
	serverAddr.sin_addr.s_addr = inet_addr(SERVER_IP);//将字符串类型转换uint32_t
}
//初始化函数，创建监听套接字，绑定端口，并进行监听
void server::init()
{
	WSADATA   wsaData;                   
	WSAStartup(MAKEWORD(2, 2), &wsaData);

	listener = socket(AF_INET, SOCK_STREAM, 0);//采用ipv4,TCP传输
	if (listener == -1) { printf("Error at socket(): %ld\n", WSAGetLastError()); perror("listener failed"); exit(1); }
	else printf("Server is running......\n");

	if (bind(listener, (sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
	{
		perror("bind error");
		exit(1);
	}
	listen(listener, 5);//listener这个套接字监听申请的链接，最大等待连接队列为5,等待accept()
	socksArr.push_back(listener);//将监听套接字加入套接字数组，数组内首个套接字就是服务器的套接字
}

void server::process()
{

	int mount = 0;
	fd_set fds,fds_copy;
	FD_ZERO(&fds);//将fds清零

	init();
	//下面就是不断的检查申请的连接队列
	printf("Server is waiting......\n");
	while (1)
	{
		mount = socksArr.size();
		//每次循环更新一次fds数组
		for (int i = 0; i

//Client
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include
#include 
#include 
#include  
#include 
#include//定义socklen_t

#pragma comment(lib, "WS2_32")  // 链接到WS2_32.lib

using namespace std;



#define SERVER_IP "127.0.0.1"// 默认服务器端IP地址
#define SERVER_PORT 8888// 服务器端口号



class client
{
public:
	client();
	void init();
	void process();
private:
	int user; //客户端套接字（句柄）
	int writing;
	sockaddr_in  serverAddr;
	void sendata();
};
int main()
{
	client user;
	user.process();

	return 0;
}

client::client()
{
	user = 0;
	writing = 0;
	serverAddr.sin_family = PF_INET;
	serverAddr.sin_port = SERVER_PORT;
	serverAddr.sin_addr.s_addr = inet_addr(SERVER_IP);//将字符串类型转换uint32_t
}

void client::init()
{
	WSADATA   wsaData;                											
	WSAStartup(MAKEWORD(2, 2), &wsaData);

	user = socket(AF_INET, SOCK_STREAM, 0);//采用ipv4,TCP传输，成功时返回非负数socket描述符

	if (user <= 0)
	{
		perror("establish client faild");
		printf("Error at socket(): %ld\n", WSAGetLastError());
		exit(1);
	};
	printf("establish succesfully\n");//创建成功，阻塞式的等待服务器连接
	if (connect(user, (const sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
	{
		perror("connect to server faild");
		printf("Error at socket(): %ld\n", WSAGetLastError());
		exit(1);
	}
	printf("connect IP:%s  Port:%d succesfully\n", SERVER_IP, SERVER_PORT);//创建成功
}

void client::process()
{
	char recvbuf[1024];
	fd_set fdread, fedwrite;
	FD_ZERO(&fdread);//将fdread清零
	FD_ZERO(&fedwrite);//将fedwrite清零

	init();

	while (1)
	{
		FD_SET(user, &fdread);
		if (writing == 0)
			FD_SET(user, &fedwrite);
		struct timeval timeout = { 1,0 };//每个Select等待三秒
		switch (select(0, &fdread, &fedwrite, NULL, &timeout))
		{
		case -1:
		{
			printf("Error at socket(): %ld\n", WSAGetLastError());
			break;
		}
		case 0:		
			break;
		default:
		{
			if (FD_ISSET(user, &fdread))//有待读事件
			{
				int size = recv(user, recvbuf, sizeof(recvbuf) - 1, 0);
				if (size > 0)
				{
					printf("server : %s\n", recvbuf);
					memset(recvbuf, '\0', sizeof(recvbuf));
				}
				else if (size == 0)
				{
					printf("server is closed\n");
					exit(1);
				}
			}
			if (FD_ISSET(user, &fedwrite))
			{
				FD_ZERO(&fedwrite);//将fedwrite清零
				writing = 1;//表示正在写入
				thread sendtask(bind(&client::sendata, this));
				sendtask.detach();//将子线程和主进程分离且互相不影响
			}
			break;
		}
	}
	}
}

void client::sendata()
{
	char sendbuf[1024];
	char middle[1024];

	cin.getline(sendbuf, 1024);//读取一行
	send(user, sendbuf, sizeof(sendbuf) - 1, 0);
	writing = 0;
}