C++Socket网络通信多路复用

C++Socket网络通信多路复用

1.select

可以在windows上面使用！

服务端：

/*
 * @Description:SOCKET
 * @Author: szq
 * @Github: https://github.com/MrQqqq
 * @Date: 2020-06-26 00:04:39
 * @LastEditors: szq
 * @LastEditTime: 2020-06-26 00:04:39
 * @FilePath: \cpp\src\BinaryTree\client_TCp.cpp
 */
#include<iostream>
#include<winsock.h>
#include<vector>
#pragma comment(lib,"ws2_32.lib")
using namespace std;
void initialization() {
   
	//初始化套接字库
	WORD w_req = MAKEWORD(2, 2);//版本号
	WSADATA wsadata;
	int err;
	err = WSAStartup(w_req, &wsadata);
	if (err != 0) {
   
		cout << "初始化套接字库失败！" << endl;
	}
	else {
   
		cout << "初始化套接字库成功！" << endl;
	}
	//检测版本号
	if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wHighVersion) != 2) {
   
		cout << "套接字库版本号不符！" << endl;
		WSACleanup();
	}
	else {
   
		cout << "套接字库版本正确！" << endl;
	}
}

enum MyCMD {
   
	LOGIN,
	LOGOUT,
	LOGERROR,
	RESULT
};
class DataHeader {
   
public:
	int dataLength;
	MyCMD cmd;
};

class Login : public DataHeader{
   
public:
	Login() {
   
		dataLength = sizeof(Login);
		cmd = LOGIN;
	}
	char username[30];
	char password[30];
};

class Result : public DataHeader {
   
public:
	int result;
	Result(int a) {
   
		dataLength = sizeof(Result);
		cmd = RESULT;
		result = a;
	}
};
vector<SOCKET> g_clients;
int main() {
   
	//定义长度变量
	int send_len = 0;
	int recv_len = 0;
	int len = 0;

	//定义发送缓冲区和接收缓冲区
	char send_buf[1024];
	char recv_buf[1024];
	//定义服务器套接字,接受请求套接字
	SOCKET s_server;
	

	//服务器地址和客户端地址
	SOCKADDR_IN server_addr;
	
	initialization();

	//填充服务端信息
	server_addr.sin_family = AF_INET;
	server_addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
	server_addr.sin_port = htons(5010);

	//创建套接字
	s_server = socket(AF_INET, SOCK_STREAM, 0);
	if (bind(s_server, (SOCKADDR*)&server_addr, sizeof(SOCKADDR)) == SOCKET_ERROR) {
   
		cout << "套接字绑定失败！" << endl;
	}
	else {
   
		cout << "套接字绑定成功！" << endl;
	}
	if (listen(s_server, SOMAXCONN) < 0) {
   
		cout << "设置监听状态失败！" << endl;
	}
	else {
   
		cout << "设置监听状态成功！" << endl;
	}
	cout << "服务器正在监听连接，请稍候..." << endl;

	while (1) {
   
		fd_set fdRead;
		fd_set fdWrite;
		fd_set fdExecpt;
		//清空
		FD_ZERO(&fdRead);
		FD_ZERO(&fdWrite);
		FD_ZERO(&fdExecpt);

		FD_SET(s_server, &fdRead);
		FD_SET(s_server, &fdWrite);
		FD_SET(s_server, &fdExecpt);

		for (int i = 0; i < g_clients.size(); i++) {
   
			FD_SET(g_clients[i], &fdRead);
		}
		//nfds是一个整数值，是指fd_set集合中所有描述符的范围，而不是数量，即最大值
		timeval t = {
    0,0 };//设置成非阻塞
		int ret = select(s_server + 1, &fdRead, &fdWrite, &fdExecpt,&t);
		if (ret < 0) {
   
			cout << "select任务结束！" << endl;
			break;
		}
		if (FD_ISSET(s_server, &fdRead)) {
   
			FD_CLR(s_server, &fdRead);
			SOCKET s_accept = INVALID_SOCKET;
			SOCKADDR_IN accept_addr;
			len = sizeof(SOCKADDR);
			s_accept = accept(s_server, (SOCKADDR*)&accept_addr, &len);
			if (s_accept == SOCKET_ERROR) {
   
				cout << "连接失败！" << endl;
				return 0;
			}
			cout << "连接建立，准备接受数据" << endl;

			g_clients.push_back(s_accept);
		}

		for (int i =