简易C/S模式1.3
将server端升级为select模型
server
#include <stdio.h>
#include <WinSock2.h>
#include <vector>
using namespace std;
#pragma comment(lib, "ws2_32.lib")
//DataPackage
enum CMD{
CMD_LOGIN,
CMD_LOGIN_RESULT,
CMD_LOGOUT,
CMD_LOGOUT_RESULT,
CMD_ERROR
};
struct DataHeader{
short cmd; //命令
short dataLength; //数据长度
};
struct Login : public DataHeader{
Login(){
this->cmd = CMD_LOGIN;
this->dataLength = sizeof(Login);
}
char userName[32];
char password[32];
};
struct LoginResult : public DataHeader{
LoginResult(){
this->cmd = CMD_LOGIN_RESULT;
this->dataLength = sizeof(LoginResult);
result = 0;
}
int result;
};
struct Logout : public DataHeader{
Logout(){
this->cmd = CMD_LOGIN;
this->dataLength = sizeof(Logout);
}
char userName[32];
};
struct LogoutResult : public DataHeader{
LogoutResult(){
this->cmd = CMD_LOGOUT_RESULT;
this->dataLength = sizeof(LogoutResult);
result = 0;
}
int result;
};
vector<SOCKET> g_clients;
int process(SOCKET clientsock){
char szRecv[1024] = {};
//接受客户端数据
int nLen = recv(clientsock, szRecv, sizeof(DataHeader), 0);
DataHeader* header = (DataHeader*)szRecv;
if (nLen <= 0){
printf("client exit.\n");
return -1;
}
switch (header->cmd){
case CMD_LOGIN:
{
recv(clientsock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);
Login* login = (Login*)szRecv;
printf("recv cmd : CMD_LOGIN, dataLength : %d, userName=%s, password=%s\n", login->dataLength, login->userName, login->password);
//用户名密码判断过程
LoginResult ret;
send(clientsock, (const char*)&ret, sizeof(LoginResult), 0);
}
break;
case CMD_LOGOUT:
{
recv(clientsock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);
Logout* logout = (Logout*)szRecv;
printf("recv cmd : CMD_LOGOUT, dataLength : %d, userName=%s\n", logout->dataLength, logout->userName);
//用户名密码判断过程
LogoutResult ret;
send(clientsock, (const char*)&ret, sizeof(LogoutResult), 0);
}
break;
default:
{
DataHeader header = { CMD_ERROR, 0 };
send(clientsock, (const char*)&header, sizeof(DataHeader), 0);
}
break;
}
return 0;
}
int main(int argc, char* argvp[])
{
//初始化WSA
WORD sockVersion = MAKEWORD(2, 2);
WSADATA wsaData;
if (WSAStartup(sockVersion, &wsaData) != 0){
return -1;
}
//创建套接字
SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sock == INVALID_SOCKET){
printf("socket errot\n");
WSACleanup();
return -2;
}
//套接字结构体
SOCKADDR_IN ser_addr;
ser_addr.sin_family = AF_INET;
ser_addr.sin_port = htons(8888);
ser_addr.sin_addr.S_un.S_addr = INADDR_ANY; //inet_addr("127.0.0.1");
//绑定IP和端口
if (bind(sock, (struct sockaddr*)&ser_addr, sizeof(ser_addr)) == SOCKET_ERROR){
printf("bind error\n");
closesocket(sock);
WSACleanup();
return -3;
}
//监听端口
if (listen(sock, 5) == SOCKET_ERROR){
printf("listen error\n");
closesocket(sock);
WSACleanup();
return -4;
}
#if 0
//定义接收客户端的套接字
SOCKET clientsock = INVALID_SOCKET;
SOCKADDR_IN client_addr = {};
int client_addr_len = sizeof(client_addr);
//接受一个客户端连接
clientsock = accept(sock, (struct sockaddr*)&client_addr, &client_addr_len);
if (clientsock == INVALID_SOCKET){
printf("accept error\n");
closesocket(sock);
WSACleanup();
return -5;
}
printf("new client add, socket = %d, IP = %s\n", clientsock, inet_ntoa(client_addr.sin_addr));
#endif
while (true){
fd_set fdRead;
fd_set fdWrite;
fd_set fdExcept;
FD_ZERO(&fdRead);
FD_ZERO(&fdWrite);
FD_ZERO(&fdExcept);
FD_SET(sock, &fdRead);
FD_SET(sock, &fdWrite);
FD_SET(sock, &fdExcept);
for (size_t i = 0; i < g_clients.size(); ++i){
FD_SET(g_clients[i], &fdRead);
}
timeval t = { 0, 10 };
int ret = select(sock + 1, &fdRead, &fdWrite, &fdExcept, &t);
if (ret < 0){
printf("select error.\n");
break;
}
if (FD_ISSET(sock, &fdRead)){
FD_CLR(sock, &fdRead);
//定义接收客户端的套接字
SOCKET clientsock = INVALID_SOCKET;
SOCKADDR_IN client_addr = {};
int client_addr_len = sizeof(client_addr);
//接受一个客户端连接
clientsock = accept(sock, (struct sockaddr*)&client_addr, &client_addr_len);
if (clientsock == INVALID_SOCKET){
printf("accept error\n");
closesocket(sock);
WSACleanup();
return -5;
}
printf("new client add, socket = %d, IP = %s\n", clientsock, inet_ntoa(client_addr.sin_addr));
g_clients.push_back(clientsock);
}
for (size_t i = 0; i < fdRead.fd_count; ++i){
if (-1 == process(fdRead.fd_array[i])){
auto it = find(g_clients.begin(), g_clients.end(), fdRead.fd_array[i]);
if (it != g_clients.end()){
g_clients.erase(it);
}
}
}
}
//关闭套接字
for (size_t i = 0; i < g_clients.size(); ++i){
closesocket(g_clients[i]);
}
//清除window socket环境
WSACleanup();
getchar();
return 0;
}