服务器和客户端简单的使用select网络模型(二)
此版本在之前一的基础上进行了封装
Server部分:一共3个文件(server.cpp,EasyTcpServer.hpp和MessageHeader.hpp)
server.cpp:
#include"EasyTcpServer.hpp" #include<thread> void CmdThread(EsayTcpServer *server) { while (true) { char cmdBuf[256] = {}; scanf("%s", cmdBuf); if (0 == strcmp(cmdBuf, "exit")) { server->Close(); printf("退出cmdThread线程\n"); break; } else printf("不支持的命令。 \n"); } } int main() { EsayTcpServer server; server.initSocket(); server.Bind(nullptr,10000); server.Listen(5); std::thread t1(CmdThread, &server); t1.detach(); while (server.isRun()) { server.OnRun(); } return 0; }
EasyTcpServer.hpp:
#ifndef _EASY_TCP_SERVER_HPP_ #define _EASY_TCP_SERVER_HPP_ #define _AFXDLL #ifdef _WIN32 #include"afx.h" #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<iostream> #include<vector> #include<stdlib.h> #include"MessageHeader.hpp" class EsayTcpServer { public: SOCKET _sock; std::vector<SOCKET> g_clients; public: EsayTcpServer() { _sock = INVALID_SOCKET; } virtual ~EsayTcpServer() { Close(); } //初始化Socket SOCKET initSocket() { // 启动Windows socket 2.x环境 #ifdef _WIN32 WORD ver = MAKEWORD(2, 2); WSADATA dat; WSAStartup(ver, &dat); #endif //---------------- // 建立一个socket if (INVALID_SOCKET != _sock) { printf("<sock=%d>关闭之前的连接...\n", _sock); Close(); } _sock = socket(AF_INET, SOCK_STREAM, 0); if (INVALID_SOCKET == _sock) printf("错误,创建Socket失败...\n"); else printf("建立Socket成功...\n"); return _sock; } //绑定端口 int Bind(const char *ip,unsigned short port) { sockaddr_in sin = {}; sin.sin_family = AF_INET; sin.sin_port = htons(port); //host to net unsigned short #ifdef _WIN32 if(ip!=NULL) { sin.sin_addr.S_un.S_addr = inet_addr(ip); } else { sin.sin_addr.S_un.S_addr = INADDR_ANY; //本机上的外网和局域网ip任意一个拿来用 } #else if (ip!=NULL) { sin.sin_addr.s_addr = inet_addr(ip); //本机上的外网和局域网ip任意一个拿来用 } else { sin.sin_addr.s_addr = INADDR_ANY; //本机上的外网和局域网ip任意一个拿来用 } #endif //2绑定 if (SOCKET_ERROR == bind(_sock, (sockaddr *)&sin, sizeof(sockaddr_in))) { //TRACE(_T("ERROR:绑定socket失败\n")); return 0; } } //监听端口 int Listen(int n) { int ret = listen(_sock, n); if (SOCKET_ERROR == ret) { TRACE(_T("ERROR:socket监听失败")); } else { printf("监听开始\n"); } return ret; } //接受客户端连接 SOCKET Accept() { //4接收客户端连接 SOCKET _cSock = INVALID_SOCKET; sockaddr_in clientAddr = {}; int nAddrLen = sizeof(sockaddr_in); _cSock = accept(_sock, (sockaddr*)&clientAddr, &nAddrLen); if (INVALID_SOCKET == _cSock) { printf("接收客户端无效\n"); } else { //广播所有客户端,新加入一个新的客户端 NewUserJoin newUser; SendDataToAll(&newUser); g_clients.push_back(_cSock); printf("新客户端加入socket = %d,ip:%s\n", _cSock, inet_ntoa(clientAddr.sin_addr)); } return _cSock; } //关闭socket void Close() { if (_sock != INVALID_SOCKET) { #ifdef _WIN32 closesocket(_sock); WSACleanup(); #else close(_sock); #endif _sock = INVALID_SOCKET; } } //处理网络消息 bool OnRun() { fd_set fdRead; fd_set fdWrite; fd_set fdExp; FD_ZERO(&fdRead); FD_ZERO(&fdWrite); FD_ZERO(&fdExp); //将要监听的socket添加到数组 FD_SET(_sock, &fdRead); FD_SET(_sock, &fdWrite); FD_SET(_sock, &fdExp); SOCKET maxSock = _sock; //每次进来都把所有要监视的socket添加到相应的数组中 for (std::vector<SOCKET>::iterator it = g_clients.begin(); it != g_clients.end(); it++) { FD_SET(*it, &fdRead); if (maxSock < *it) { maxSock = *it; } } //开始查询,会清空3个数组所有的socket,然后如果哪个socket有事件发送,则会把此socket放入到数组返回 //当有客户端连接或者客户端发送消息都会进来 timeval time = { 2, 0 }; /*int select( IN int nfds, //windows下无意义,linux有意义IN OUT fd_set* readfds, //检查可读性 IN OUT fd_set* writefds, //检查可写性 IN OUT fd_set* exceptfds, //例外数据 IN const struct timeval* timeout); //函数的返回时间*/ int ret = select(maxSock + 1, &fdRead, &fdWrite, &fdExp, NULL); printf("空闲时间处理其他事情\n"); if (ret < 0) { printf("select任务结束\n"); Close(); return false; } if (FD_ISSET(_sock, &fdRead)) { FD_CLR(_sock, &fdRead); //4接收客户端连接 Accept(); } for (size_t n = 0; n < fdRead.fd_count; n++) { if (-1 == RecvData(fdRead.fd_array[n])) { auto iter = find(g_clients.begin(), g_clients.end(), fdRead.fd_array[n]); if (iter != g_clients.end()) { g_clients.erase(iter); } } } } //是否在工作中 bool isRun() { return _sock != INVALID_SOCKET; } // 接受数据,处理粘包,拆分包 int RecvData(SOCKET cSock) { char szRecv[1024] = {}; //接受客户端请求数据 int nLen = recv(cSock, szRecv, sizeof(DataHeader), 0); DataHeader *header = (DataHeader*)szRecv; if (nLen <= 0) { printf("客户端<socket=%d>已退出,任务结束\n", cSock); return -1; } printf("收到命令:%d 数据长度:%d\n", header->cmd, header->dataLength); recv(cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0); OnNetMsg(cSock, header); return 0; } // 响应网络消息 virtual void OnNetMsg(SOCKET _cSock, DataHeader* header) { switch (header->cmd) { case CMD_LOGIN: { Login* login = (Login*)header; printf("收到命令 用户名:%s密码%s\n", login->userName, login->PassWord); LoginResult ret = {}; ret.result = 0; send(_cSock, (const char *)&ret, sizeof(LoginResult), 0); break; } case CMD_LOGOUT: { Logout* loginOut = (Logout*)header; printf("收到命令: 数据长度:%d 用户名:%s密码%s\n", loginOut->dataLength, loginOut->userName); LogOutResult retOut = {}; retOut.result = 0; send(_cSock, (const char *)&retOut, sizeof(LoginResult), 0); break; } default: { DataHeader header = { CMD_ERROR, 0 }; send(_cSock, (const char *)&header, sizeof(DataHeader), 0); break; } } } // 发送数据给指定的socket int SendData(SOCKET _cSock, DataHeader* header) { if (isRun() && header) send(_cSock, (const char*)header, header->dataLength, 0); return SOCKET_ERROR; } //群发 void SendDataToAll( DataHeader* header) { if (isRun() && header) { //广播所有客户端,新加入一个新的客户端 for (std::vector<SOCKET>::iterator it = g_clients.begin(); it != g_clients.end(); it++) { SendData(*it, header); } } } private: }; #endif
MessageHeader.hpp:
#ifndef _MessageHeader_hpp_ #define _MessageHeader_hpp_ 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; }; #endif
Client部分:一共3个文件(Client.cpp,EasyTcpClient.hpp和MessageHeader.hpp)
Client.cpp:
#define _CRT_SECURE_NO_WARNINGS #include <thread> #include "EasyTcpClient.hpp" #pragma comment(lib, "ws2_32.lib") void cmdThread(EasyTcpClient *client) { while (true) { char cmdBuf[256] = {}; scanf("%s", cmdBuf); if (0 == strcmp(cmdBuf, "exit")) { client->Close(); printf("退出cmdThread线程\n"); break; } else if (0 == strcmp(cmdBuf, "login")) { Login login; strcpy(login.userName, "lyd"); strcpy(login.PassWord, "lydmm"); client->SendData(&login); } else if (0 == strcmp(cmdBuf, "logout")) { Login logout; strcpy(logout.userName, "lyd"); client->SendData(&logout); } else printf("不支持的命令。 \n"); } } int main() { EasyTcpClient client; client.initSocket(); client.Connect("127.0.0.1", 10000); std::thread t1(cmdThread, &client); t1.detach(); // 线程分离 while (client.isRun()) { client.OnRun(); // printf("空闲时间处理其他业务...\n"); // Sleep(1000); } client.Close(); printf("已退出。 \n"); getchar(); return 0; }
EasyTcpClient.hpp:
#pragma once #ifdef _WIN32 #include <WinSock2.h> #include <windows.h> #include "MessageHeader.hpp" #include <stdio.h> #else #include<unistd.h> #include<arpa/inet.h> #include<string.h> #define SOCKET int #define INVALID_SOCKET (SOCKET)(~0) #define SOCKET_ERROR (-1) #include <stdio.h> #endif class EasyTcpClient { SOCKET _sock; public: EasyTcpClient() { _sock = INVALID_SOCKET; } virtual ~EasyTcpClient() { Close(); } // 初始化socket void initSocket() { // 启动Windows socket 2.x环境 #ifdef _WIN32 WORD ver = MAKEWORD(2, 2); WSADATA dat; WSAStartup(ver, &dat); #endif //---------------- // 建立一个socket if (INVALID_SOCKET != _sock) { printf("<sock=%d>关闭之前的连接...\n", _sock); Close(); } _sock = socket(AF_INET, SOCK_STREAM, 0); if (INVALID_SOCKET == _sock) printf("错误,创建Socket失败...\n"); else printf("建立Socket成功...\n"); } // 连接服务器 int Connect(const char* ip, unsigned short port) { if (INVALID_SOCKET == _sock) { initSocket(); } // 连接服务器 connect sockaddr_in _sin = {}; _sin.sin_family = AF_INET; _sin.sin_port = htons(port); #ifdef _WIN32 _sin.sin_addr.S_un.S_addr = inet_addr(ip); #else _sin.sin_addr.s_addr = inet_addr(ip); #endif int ret = connect(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in)); if (SOCKET_ERROR == ret) printf("连接服务器错误...\n"); else printf("连接服务器成功...\n"); return ret; } void Close() { if (_sock != INVALID_SOCKET) { #ifdef _WIN32 closesocket(_sock); WSACleanup(); #else close(_sock); #endif _sock = INVALID_SOCKET; } } // 发送数据 // 接受数据 // 处理网络消息 bool OnRun() { if (isRun()) { 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("<socket=%d>select任务结束\n", _sock); return false; } if (FD_ISSET(_sock, &fdReads)) { FD_CLR(_sock, &fdReads); if (-1 == RecvData(_sock)) { printf("<socket=%d>select任务结束2\n", _sock); return false; } } return true; } return false; } bool isRun() { return _sock != INVALID_SOCKET; } // 接受数据,处理粘包,拆分包 int RecvData(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; } // 处理请求 recv(_cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0); OnNetMsg(header); return 0; } // 响应网络 void OnNetMsg(DataHeader* header) { switch (header->cmd) { case CMD_LOGIN_RESULT: { LoginResult* login = (LoginResult*)header; printf("收到服务端消息:CMD_LOGIN_RESULT,数据长度:%d\n", header->dataLength); break; } case CMD_LOGOUT_RESULT: { LogOutResult* logout = (LogOutResult*)header; printf("收到服务端消息:CMD_LOGOUT_RESULT,数据长度:%d\n", header->dataLength); break; } case CMD_NEW_USER_JOIN: { NewUserJoin* userJoin = (NewUserJoin*)header; printf("收到服务端消息:CMD_NEW_USER_JOIN,数据长度:%d\n", header->dataLength); break; } } } // 发送数据 int SendData(DataHeader* header) { if (isRun() && header) send(_sock, (const char*)header, header->dataLength, 0); return SOCKET_ERROR; } };
MessageHeader.hpp:
#ifndef _MessageHeader_hpp_ #define _MessageHeader_hpp_ 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; }; #endif
111