套接字模式—非阻塞模式开发（3）—— 客户端类实现

经过四个晚上的认真阅读，这套远程算术运算的代码总算全部通过，作者对于每个模块的设计深感佩服！

 

客户端CClient类对应每个服务器接受的客户端，为其实现接收数据、计算数据和发送计算结果的功能。

 

在没有完全读懂前，我确实不敢贴出来，现在既然读懂了，就贴出来大家交流一下嘛！

 

具体代码如下：

 

// // CClient.h #ifndef _CLIENT_H #define _CLIENT_H #include <winsock2.h> #define TIMEFOR_THREAD_CLIENT 500 //线程睡眠时间 #define MAX_NUM_CLIENT 10 //接受的客户端连接最多数量 #define MAX_NUM_BUF 48 //缓冲区的最大长度 #define INVALID_OPERATOR 1 //无效的操作符 #define INVALID_NUM 2 //分母为零 #define ZERO 0 //零 //数据包类型 #define EXPRESSION 'E' //算数表达式 #define BYEBYE 'B' //消息byebye #define HEADERLEN (sizeof(hdr)) //头长度 //数据包头结构，该结构在win32下为4byte typedef struct _head { char type; //类型 unsigned short len; //数据包的长度(包括头的长度) }hdr, *phdr; //数据包中的数据结构 typedef struct _data { char buf[MAX_NUM_BUF];//数据 }DATABUF, *pDataBuf; class CClient { public: CClient(const SOCKET sClient,const sockaddr_in &addrClient); virtual ~CClient(); public: BOOL StartRuning(void); //创建发送和接收数据线程 void HandleData(const char* pExpr); //计算表达式 BOOL IsConning(void){ //是否连接存在 return m_bConning; } void DisConning(void){ //断开与客户端的连接 m_bConning = FALSE; } BOOL IsExit(void){ //接收和发送线程是否已经退出 return m_bExit; } public: static DWORD __stdcall RecvDataThread(void* pParam); //接收客户端数据 static DWORD __stdcall SendDataThread(void* pParam); //向客户端发送数据 private: CClient(); private: SOCKET m_socket; //套接字 sockaddr_in m_addr; //地址 DATABUF m_data; //数据 HANDLE m_hEvent; //事件对象 HANDLE m_hThreadSend; //发送数据线程句柄 HANDLE m_hThreadRecv; //接收数据线程句柄 CRITICAL_SECTION m_cs; //临界区对象 BOOL m_bConning; //客户端连接状态 BOOL m_bExit; //线程退出 }; #endif

 

 

// // CClient.cpp #include "StdAfx.h" #include <process.h> #include "Client.h" /* * 构造函数 */ CClient::CClient(const SOCKET sClient, const sockaddr_in &addrClient) { //初始化变量 m_hThreadRecv = NULL; m_hThreadSend = NULL; m_socket = sClient; m_addr = addrClient; m_bConning = FALSE; m_bExit = FALSE; memset(m_data.buf, 0, MAX_NUM_BUF); //创建事件 m_hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);//手动设置信号状态，初始化为无信号状态 //初始化临界区 InitializeCriticalSection(&m_cs); } /* * 析构函数 */ CClient::~CClient() { closesocket(m_socket); //关闭套接字 m_socket = INVALID_SOCKET; //套接字无效 DeleteCriticalSection(&m_cs); //释放临界区对象 CloseHandle(m_hEvent); //释放事件对象 } /* * 创建发送和接收数据线程 */ BOOL CClient::StartRuning(void) { m_bConning = TRUE;//设置连接状态 //创建接收数据线程 unsigned long ulThreadId; m_hThreadRecv = CreateThread(NULL, 0, RecvDataThread, this, 0, &ulThreadId); if(NULL == m_hThreadRecv) { return FALSE; }else{ CloseHandle(m_hThreadRecv); } //创建接收客户端数据的线程 m_hThreadSend = CreateThread(NULL, 0, SendDataThread, this, 0, &ulThreadId); if(NULL == m_hThreadSend) { return FALSE; }else{ CloseHandle(m_hThreadSend); } return TRUE; } /* * 接收客户端数据 */ DWORD CClient::RecvDataThread(void* pParam) { CClient *pClient = (CClient*)pParam; //客户端对象指针 int reVal; //返回值 char temp[MAX_NUM_BUF]; //临时变量 memset(temp, 0, MAX_NUM_BUF); for (;pClient->m_bConning;) //连接状态 { reVal = recv(pClient->m_socket, temp, MAX_NUM_BUF, 0); //接收数据 //处理错误返回值 if (SOCKET_ERROR == reVal) { int nErrCode = WSAGetLastError(); if ( WSAEWOULDBLOCK == nErrCode ) //接受数据缓冲区不可用 { continue; //继续循环 }else if (WSAENETDOWN == nErrCode ||//客户端关闭了连接 WSAETIMEDOUT == nErrCode || WSAECONNRESET == nErrCode ) { break; //线程退出 } } //客户端关闭了连接 if ( reVal == 0) { break; } //收到数据 if (reVal > HEADERLEN) { pClient->HandleData(temp); //处理数据 SetEvent(pClient->m_hEvent); //通知发送数据线程 memset(temp, 0, MAX_NUM_BUF); //清空临时变量 } Sleep(TIMEFOR_THREAD_CLIENT); //线程睡眠 } pClient->m_bConning = FALSE; //与客户端的连接断开 SetEvent(pClient->m_hEvent); //通知发送数据线程退出 return 0; //线程退出 } /* * //向客户端发送数据 */ DWORD CClient::SendDataThread(void* pParam) { CClient *pClient = (CClient*)pParam;//转换数据类型为CClient指针 for (;pClient->m_bConning;)//连接状态 { //收到事件通知 if (WAIT_OBJECT_0 == WaitForSingleObject(pClient->m_hEvent, INFINITE)) { //当客户端的连接断开时，接收数据线程先退出，然后该线程后退出，并设置退出标志 if (!pClient->m_bConning) { pClient->m_bExit = TRUE; break ; } //进入临界区 EnterCriticalSection(&pClient->m_cs); //发送数据 phdr pHeader = (phdr)pClient->m_data.buf; int nSendlen = pHeader->len; int val = send(pClient->m_socket, pClient->m_data.buf, nSendlen,0); //处理返回错误 if (SOCKET_ERROR == val) { int nErrCode = WSAGetLastError(); if (nErrCode == WSAEWOULDBLOCK)//发送数据缓冲区不可用 { continue; }else if ( WSAENETDOWN == nErrCode || WSAETIMEDOUT == nErrCode || WSAECONNRESET == nErrCode)//客户端关闭了连接 { //离开临界区 LeaveCriticalSection(&pClient->m_cs); pClient->m_bConning = FALSE; //连接断开 pClient->m_bExit = TRUE; //线程退出 break; }else { //离开临界区 LeaveCriticalSection(&pClient->m_cs); pClient->m_bConning = FALSE; //连接断开 pClient->m_bExit = TRUE; //线程退出 break; } } //成功发送数据 //离开临界区 LeaveCriticalSection(&pClient->m_cs); //设置事件为无信号状态 ResetEvent(&pClient->m_hEvent); } } return 0; } /* * 计算表达式,打包数据 */ void CClient::HandleData(const char* pExpr) { memset(m_data.buf, 0, MAX_NUM_BUF);//清空m_data //如果是“byebye”或者“Byebye” if (BYEBYE == ((phdr)pExpr)->type) { EnterCriticalSection(&m_cs); phdr pHeaderSend = (phdr)m_data.buf; //发送的数据 pHeaderSend->type = BYEBYE; //单词类型 pHeaderSend->len = HEADERLEN + strlen("OK"); //数据包长度 memcpy(m_data.buf + HEADERLEN, "OK", strlen("OK")); //复制数据到m_data" LeaveCriticalSection(&m_cs); }else{//算数表达式 int nFirNum; //第一个数字 int nSecNum; //第二个数字 char cOper; //算数运算符 int nResult; //计算结果 //格式化读入数据 sscanf(pExpr + HEADERLEN, "%d%c%d", &nFirNum, &cOper, &nSecNum); //计算 switch(cOper) { case '+'://加 { nResult = nFirNum + nSecNum; break; } case '-'://减 { nResult = nFirNum - nSecNum; break; } case '*'://乘 { nResult = nFirNum * nSecNum; break; } case '/'://除 { if (ZERO == nSecNum)//无效的数字 { nResult = INVALID_NUM; }else { nResult = nFirNum / nSecNum; } break; } default: nResult = INVALID_OPERATOR;//无效操作符 break; } //将算数表达式和计算的结果写入字符数组中 char temp[MAX_NUM_BUF]; char cEqu = '='; sprintf(temp, "%d%c%d%c%d",nFirNum, cOper, nSecNum,cEqu, nResult); //打包数据 EnterCriticalSection(&m_cs); phdr pHeaderSend = (phdr)m_data.buf; //发送的数据 pHeaderSend->type = EXPRESSION; //数据类型为算数表达式 pHeaderSend->len = HEADERLEN + strlen(temp); //数据包的长度 memcpy(m_data.buf + HEADERLEN, temp, strlen(temp)); //复制数据到m_data LeaveCriticalSection(&m_cs); } }