C++ 跨平台网络客户端编程实现

使用多线程加自定义结构体数据发送以及select函数，实现跨平台网络客户端

#define WIN32_LEAN_AND_MEAN
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS

#ifdef _WIN32
  #include <windows.h>
  #include <WinSock2.h>
#else
  #include<unistd.h>
  #include<arpa/inet.h>
  #include<string.h>

  #define SOCKET int
  #define INVALID_SOCKET (SOCKET)(~0)
  #define SOCKET_ERROR           (-1)
#endif

#include <stdio.h>
#include <thread>

#pragma comment(lib, "ws2_32.lib")

enum CMD {
	CMD_LOGIN,
	CMD_LOGIN_RESULT,
	CMD_LOGOUT,
	CMD_LOGOUT_RESULT,
	CMD_NEW_USER_JOIN,
	CMD_ERROR,

};

// 消息头
struct DataHeader {
	short dataLength;
	short cmd;
};


struct Login :public DataHeader {

	Login() {

		dataLength = sizeof(Login);
		cmd = CMD_LOGIN;
	}
	char userName[32];
	char PassWord[32];
};

struct LoginResult : public DataHeader {

	LoginResult() {
		dataLength = sizeof(LoginResult);
		cmd = CMD_LOGIN_RESULT;
		result = 0;
	}
	int result;

};

struct Logout :public DataHeader {

	Logout() {
		dataLength = sizeof(Logout);
		cmd = CMD_LOGOUT;
	}
	char userName[32];
};

struct LogOutResult :public DataHeader {

	LogOutResult() {
		dataLength = sizeof(LogOutResult);
		cmd = CMD_LOGOUT_RESULT;
		result = 0;
	}
	int result;
};

struct NewUserJoin :public DataHeader
{
	NewUserJoin() {
		dataLength = sizeof(NewUserJoin);
		cmd = CMD_NEW_USER_JOIN;
		sock = 0;
	}
	int sock;
};

int processor(SOCKET _cSock)
{
	// 缓冲区
	char szRecv[4096] = {};
	int nLen = recv(_cSock, (char*)&szRecv, sizeof(DataHeader), 0);
	DataHeader* header = (DataHeader*)szRecv;
	if (nLen <= 0) {
		printf("与服务器断开连接，任务结束。\n");
		return -1;
	}
	// 处理请求
	switch (header->cmd) {
	case CMD_LOGIN_RESULT: {
		recv(_cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);
		LoginResult* login = (LoginResult*)szRecv;
		printf("收到服务端%d消息:CMD_LOGIN_RESULT,数据长度:%d\n",
			_cSock, header->dataLength);
	}
    break;
	case CMD_LOGOUT_RESULT: {
		recv(_cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);
		LogOutResult* logout = (LogOutResult*)szRecv;
		printf("收到服务端%d消息:CMD_LOGOUT_RESULT,数据长度:%d\n",
			_cSock, header->dataLength);
	}
	break;
	case CMD_NEW_USER_JOIN: {
		recv(_cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);
		NewUserJoin* userJoin = (NewUserJoin*)szRecv;
		printf("收到服务端%d消息:CMD_NEW_USER_JOIN,数据长度:%d\n",
			_cSock, header->dataLength);
	}
	break;
	}
	return 0;
}
bool g_bRun = true;
void cmdThread(SOCKET _sock)
{
	while (true) {
		char cmdBuf[256] = {};
		scanf("%s", cmdBuf);
		if (0 == strcmp(cmdBuf, "exit")) {
			g_bRun = false;
			printf("退出\n");
			return;
		}
		else if (0 == strcmp(cmdBuf, "login")) {
			Login login;
			strcpy(login.userName, "lyd");
			strcpy(login.PassWord, "lyd");
			send(_sock, (const char*)&login, sizeof(Login), 0);
		}
		else if (0 == strcmp(cmdBuf, "logout")) {
			Login logout;
			strcpy(logout.userName, "lyd");
			strcpy(logout.PassWord, "lyd");
			send(_sock, (const char*)&logout, sizeof(Logout), 0);
		}

	}
	

}
int main()
{
	// 启动Windows socket 2.x环境
#ifdef _WIN32
    WORD ver = MAKEWORD(2, 2);
	WSADATA dat;
	WSAStartup(ver, &dat);
#endif
	//----------------
	// 建立一个socket
	SOCKET _sock = socket(AF_INET, SOCK_STREAM, 0);
	if (INVALID_SOCKET == _sock)
		printf("创建套接字错误...\n");
	// 连接服务器 connect
	sockaddr_in _sin = {};
	_sin.sin_family = AF_INET;
	_sin.sin_port = htons(9527);
#ifdef _WIN32
	_sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
#else
	_sin.sin_addr.s_addr = inet_addr("192.168.95.1");
#endif
	int ret = connect(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in));
	if (SOCKET_ERROR == ret)
		printf("连接服务器错误...\n");
	else
		printf("连接Socket成功...\n");

	// 启动线程函数
	std::thread t1(cmdThread,_sock);
	t1.detach();                           // 与主线程分离
	while (g_bRun) {
		fd_set fdReads;
		FD_ZERO(&fdReads);
		FD_SET(_sock, &fdReads);
		timeval t = { 1,0 };
		int ret = select(_sock+1, &fdReads, 0, 0, &t);
		if (ret < 0) {
			printf("select任务结束\n");
			break;
		}
		if (FD_ISSET(_sock, &fdReads)) {
			FD_CLR(_sock, &fdReads);
			if (-1 == processor(_sock)) {
				printf("select任务结束2\n");
				break;
			}
		}
		
	}
	// 关闭套接字
#ifdef _WIN32
	closesocket(_sock);
	WSACleanup();
#else
	close(_sock);
#endif
	return 0;
}