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

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

 

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

 

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

 

具体代码如下：

 

//////////////////////////////////////////////////////////////////////////// CClient.h#ifndef _CLIENT_H#define _CLIENT_H#include <winsock2.h>#define TIMEFOR_THREAD_CLIENT500//线程睡眠时间#defineMAX_NUM_CLIENT10//接受的客户端连接最多数量#defineMAX_NUM_BUF48//缓冲区的最大长度#define INVALID_OPERATOR1//无效的操作符#define INVALID_NUM2//分母为零#define ZERO0//零//数据包类型#define EXPRESSION'E'//算数表达式#define BYEBYE'B'//消息byebye#define HEADERLEN(sizeof(hdr))//头长度//数据包头结构，该结构在win32下为4bytetypedef struct _head{chartype;//类型unsigned shortlen;//数据包的长度(包括头的长度)}hdr, *phdr;//数据包中的数据结构typedef struct _data {charbuf[MAX_NUM_BUF];//数据}DATABUF, *pDataBuf;class CClient{public:CClient(const SOCKET sClient,const sockaddr_in &addrClient);virtual ~CClient();public:BOOLStartRuning(void);//创建发送和接收数据线程voidHandleData(const char* pExpr);//计算表达式BOOLIsConning(void){//是否连接存在return m_bConning;} voidDisConning(void){//断开与客户端的连接m_bConning = FALSE;}BOOLIsExit(void){//接收和发送线程是否已经退出return m_bExit;}public:static DWORD __stdcall RecvDataThread(void* pParam);//接收客户端数据static DWORD __stdcall SendDataThread(void* pParam);//向客户端发送数据private:CClient();private:SOCKETm_socket;//套接字sockaddr_inm_addr;//地址DATABUFm_data;//数据HANDLEm_hEvent;//事件对象HANDLEm_hThreadSend;//发送数据线程句柄HANDLEm_hThreadRecv;//接收数据线程句柄CRITICAL_SECTION m_cs;//临界区对象BOOLm_bConning;//客户端连接状态BOOLm_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;//客户端对象指针intreVal;//返回值chartemp[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_dataLeaveCriticalSection(&m_cs);}}