Windows CE 下的 TCP 客户端类

以前一直在使用 UDP 与服务器进行通讯，这次一个新的项目需要采用 TCP 来实现与服务器的通讯。

先写了一个 TCP 客户端的类，同时也做了一个服务器用于测试。先把客户端的 TCP 类代码分分享出来吧。

头文件：

 1 // CeTcpClient.h: interface for the CCeTcpClient class.  
 2 //  
 3 //////////////////////////////////////////////////////////////////////  
 4   
 5 #if !defined(AFX_CETCPCLIENT_H__B7856B99_69E7_4868_9BA3_96152245C65E__INCLUDED_)  
 6 #define AFX_CETCPCLIENT_H__B7856B99_69E7_4868_9BA3_96152245C65E__INCLUDED_  
 7   
 8 #if _MSC_VER > 1000  
 9 #pragma once  
10 #endif // _MSC_VER > 1000  
11   
12 #include <winsock.h>  
13   
14 // 连接断开  
15 typedef void (CALLBACK *ONDISCONNECT)(CWnd *);  
16 // 数据接收  
17 typedef void (CALLBACK *ONREAD)(CWnd *,const char *,int);  
18 // Socket 错误  
19 typedef void (CALLBACK *ONERROR)(CWnd *,int);  
20   
21 class CCeTcpClient  
22 {  
23 public:  
24     CCeTcpClient();  
25     virtual ~CCeTcpClient();  
26   
27 public:  
28     // 服务器 IP 地址  
29     CString  m_csRemoteHost;  
30     // 服务器端口  
31     int m_iPort;  
32       
33     // 连接断开  
34     ONDISCONNECT OnDisConnect;  
35     // 接收数据  
36     ONREAD OnRead;  
37     // 发生错误  
38     ONERROR OnError;  
39       
40 public:  
41     // 打开 Socket  
42     bool Open(CWnd *pWnd);  
43     // 关闭 Socket  
44     bool Close();  
45     // 与服务器端建立连接  
46     bool Connect();  
47     // 向服务器端发送数据  
48     bool SendData(const char *pcBuffer,int iLen);  
49   
50 private:  
51     // Socket 句柄  
52     SOCKET m_socket;  
53     // 通讯线程退出事件  
54     HANDLE m_hExitThreadEvent;  
55     // 通讯线程  
56     HANDLE m_hTcpThreadHandle;  
57     // 父窗口句柄  
58     CWnd *m_pOwnerWnd;  
59   
60 private:  
61     // 通讯线程  
62     static DWORD SocketControlThread(LPVOID lparam);  
63 };  
64   
65 #endif // !defined(AFX_TCPCLIENT_CE_H__B7856B99_69E7_4868_9BA3_96152245C65E__INCLUDED_)

源文件：

  1 // CeTcpClient.cpp: implementation of the CCeTcpClient class.  
  2 // Leo.Zheng 2012.04.09  
  3 //////////////////////////////////////////////////////////////////////  
  4   
  5 #include "stdafx.h"  
  6 #include "TCPClient.h"  
  7 #include "CeTcpClient.h"  
  8   
  9 #ifdef _DEBUG  
 10 #undef THIS_FILE  
 11 static char THIS_FILE[]=__FILE__;  
 12 #define new DEBUG_NEW  
 13 #endif  
 14   
 15 //////////////////////////////////////////////////////////////////////  
 16 // Construction/Destruction  
 17 //////////////////////////////////////////////////////////////////////  
 18 CCeTcpClient::CCeTcpClient()  
 19 {  
 20     int iInitWSA = 0;  
 21     // 初始化 Socket, 版本: 1.1  
 22     WSADATA wsd;  
 23   
 24     iInitWSA = WSAStartup(MAKEWORD(1,1),&wsd);  
 25     if(0 != iInitWSA)  
 26     {  
 27         RETAILMSG(1,(L"[CCeTcpClient::CCeTcpClient]WSA start up failed: %d
",iInitWSA));  
 28     }  
 29     // 创建线程退出事件  
 30     m_hExitThreadEvent = CreateEvent(NULL,FALSE,FALSE,NULL);  
 31 }  
 32   
 33 CCeTcpClient::~CCeTcpClient()  
 34 {  
 35     // 释放 Socket  
 36     WSACleanup();  
 37     // 关闭线程退出事件  
 38     CloseHandle(m_hExitThreadEvent);  
 39 }  
 40   
 41 /* 
 42  * 功能: 打开 Socket 
 43  * 参数: pWnd 用于指定父窗口句柄 
 44  * 返回值: TRUE 成功; FALSE 失败 
 45 */  
 46 bool CCeTcpClient::Open(CWnd *pWnd)  
 47 {  
 48     ResetEvent(m_hExitThreadEvent);  
 49     m_pOwnerWnd = pWnd;  
 50       
 51     // 创建 TCP 套接字   
 52     m_socket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);  
 53     if(SOCKET_ERROR == m_socket)  
 54     {  
 55         return FALSE;  
 56     }  
 57   
 58     // 创建通讯线程  
 59     m_hTcpThreadHandle = CreateThread(NULL,0,SocketControlThread,this,0,NULL);  
 60     if(NULL == m_hTcpThreadHandle)  
 61     {  
 62         closesocket(m_socket);  
 63         return FALSE;  
 64     }  
 65       
 66     return TRUE;  
 67 }  
 68   
 69 /* 
 70  * 功能: 关闭 Socket 
 71  * 参数: 无 
 72  * 返回值: TRUE 成功; FALSE 失败 
 73 */  
 74 bool CCeTcpClient::Close()  
 75 {  
 76     // 设置通讯线程结束事件  
 77     SetEvent(m_hExitThreadEvent);  
 78     Sleep(1000);  
 79   
 80     // 关闭 Socket  
 81     int iError = closesocket(m_socket);  
 82     if(SOCKET_ERROR == iError)  
 83     {  
 84         return FALSE;  
 85     }  
 86   
 87     return TRUE;  
 88 }  
 89   
 90 /* 
 91  * 功能: 与 TCP 服务器建立连接 
 92  * 参数: 无 
 93  * 返回值: TRUE 成功; FALSE 失败 
 94 */  
 95 bool CCeTcpClient::Connect()  
 96 {  
 97     struct sockaddr_in addr;  
 98     int iError = 0;  
 99     char cAnsiRemoteHost[255 + 1];  
100   
101     addr.sin_family = AF_INET;  
102     addr.sin_port = htons(m_iPort);  
103     ZeroMemory(cAnsiRemoteHost,sizeof(char) * (255 + 1));  
104     WideCharToMultiByte(CP_ACP,WC_COMPOSITECHECK,m_csRemoteHost,wcslen(m_csRemoteHost),cAnsiRemoteHost,wcslen(m_csRemoteHost),NULL,NULL);  
105   
106    addr.sin_addr.s_addr = inet_addr(cAnsiRemoteHost);  
107    // 采用同步连接方式, connect 直接返回成功或是失败  
108    iError = connect(m_socket,(struct sockaddr *)&addr,sizeof(addr));  
109    if(SOCKET_ERROR == iError)   
110    {  
111        return FALSE;  
112    }  
113    // 设置通讯模式为异步模式  
114    DWORD dwMode = 1;  
115    ioctlsocket(m_socket,FIONBIO,&dwMode);  
116   
117    return TRUE;  
118 }  
119   
120 /* 
121  * 功能: 向服务器端发送数据 
122  * 参数: buf 待发送的数据 
123     len 待发送的数据长度 
124  * 返回值: TRUE 成功; FALSE 失败 
125 */  
126 bool CCeTcpClient::SendData(const char *pcBuffer,int iLen)  
127 {  
128     int iBytes = 0;  
129     int iSendBytes = 0;  
130               
131     while(iSendBytes < iLen)  
132     {  
133         iBytes = send(m_socket,pcBuffer + iSendBytes,iLen - iSendBytes,0);  
134         if(SOCKET_ERROR == iBytes)  
135         {  
136             int iErrorCode = WSAGetLastError();  
137             OnError(m_pOwnerWnd,iErrorCode);  
138             OnDisConnect(m_pOwnerWnd);  
139             //关闭 Socket  
140             Close();  
141             return FALSE;  
142         }  
143   
144         iSendBytes = iSendBytes + iBytes;  
145           
146         if(iSendBytes < iLen)  
147         {  
148             Sleep(1000);  
149         }  
150     }   
151     return TRUE;   
152 }  
153   
154 /* 
155  * 功能: 此线程用于监听 TCP 客户端通讯的事件 
156     例如: 当接收到数据、连接断开或通讯过程发生错误等事件 
157  * 参数: lparam: 可以通过此参数，向线程中传入需要用到的资源 
158     在这里我们将 CCeTcpClient 类实例指针传进来 
159  * 返回值: 无意义，将返回值设为 0。 
160 */  
161 DWORD CCeTcpClient::SocketControlThread(LPVOID lparam)  
162 {  
163     CCeTcpClient *pSocket = NULL;  
164     fd_set fdRead;  
165     TIMEVAL aTime;  
166     int iRet = 0;  
167   
168     // 得到 CCeTcpClient 实例指针  
169     pSocket = (CCeTcpClient *)lparam;  
170     // 事件等待时间设置  
171     aTime.tv_sec = 1;  
172     aTime.tv_usec = 0;  
173   
174     while(TRUE)  
175     {  
176         // 退出事件，结束线程  
177         if(WAIT_OBJECT_0 == WaitForSingleObject(pSocket->m_hExitThreadEvent,0))  
178         {  
179             break;  
180         }  
181         // 置 fdRead 为空  
182         FD_ZERO(&fdRead);  
183         // Socket 设置读事件  
184         FD_SET(pSocket->m_socket,&fdRead);  
185   
186         // 判断是否有读事件  
187         iRet = select(0,&fdRead,NULL,NULL,&aTime);  
188         if(SOCKET_ERROR == iRet)  
189         {  
190             pSocket->OnError(pSocket->m_pOwnerWnd,1);  
191             pSocket->OnDisConnect(pSocket->m_pOwnerWnd);  
192             //关闭客户端 Socket  
193             closesocket(pSocket->m_socket);  
194             break;  
195         }  
196           
197         if(iRet > 0)  
198         {  
199             if(FD_ISSET(pSocket->m_socket,&fdRead))  
200             {  
201                 char cRecvBuffer[1024 + 1];  
202                 int iRecvLength = 0;  
203   
204                 ZeroMemory(cRecvBuffer,sizeof(char) * (1024 + 1));  
205                 // 接收数据  
206                 iRecvLength = recv(pSocket->m_socket,cRecvBuffer,1024,0);   
207                 if(SOCKET_ERROR == iRecvLength)  
208                 {  
209                     int iError = WSAGetLastError();  
210                     pSocket->OnError(pSocket->m_pOwnerWnd,iError);  
211                     pSocket->OnDisConnect(pSocket->m_pOwnerWnd);  
212                     // 关闭客户端 Socket  
213                     closesocket(pSocket->m_socket);  
214                     break;  
215                 }  
216                 else if(0 == iRecvLength)  
217                 {  
218                     pSocket->OnDisConnect(pSocket->m_pOwnerWnd);  
219                     // 关闭客户端 Socket  
220                     closesocket(pSocket->m_socket);  
221                     break;  
222                 }  
223                 else  
224                 {  
225                    // 触发数据接收事件  
226                    pSocket->OnRead(pSocket->m_pOwnerWnd,cRecvBuffer,iRecvLength);  
227                 }  
228             }  
229         }  
230     }  
231   
232     return 0;  
233 }

测试代码如下：

  1 BOOL CTCPClientDlg::OnInitDialog()  
  2 {  
  3     //m_bFullScreen = FALSE;  
  4     CDialog::OnInitDialog();  
  5       
  6     // Set the icon for this dialog.  The framework does this automatically  
  7     //  when the application's main window is not a dialog  
  8     SetIcon(m_hIcon, TRUE);         // Set big icon  
  9     SetIcon(m_hIcon, FALSE);        // Set small icon  
 10       
 11     CenterWindow(GetDesktopWindow());   // center to the hpc screen  
 12       
 13     // 初始化输入值  
 14     m_csRemoteHost = GetLocalIP();      // 在模拟器上测试时用: 客户端与服务器都运行在模拟器上  
 15     m_iRemotePort = 5000;  
 16     UpdateData(FALSE);  
 17     return TRUE;  // return TRUE  unless you set the focus to a control  
 18 }  
 19   
 20   
 21 // 连接断开  
 22 void CALLBACK CTCPClientDlg::OnDisConnect(CWnd *pWnd)  
 23 {  
 24     CTCPClientDlg *pDlg = (CTCPClientDlg *)pWnd;  
 25       
 26     CStatic *pStatus = (CStatic *)pDlg->GetDlgItem(IDC_LBLCONNSTATUS);  
 27     ASSERT(NULL != pStatus);  
 28     pStatus->SetWindowText(L"连接断开");  
 29       
 30     CButton *pBtnConn = (CButton *)pDlg->GetDlgItem(IDC_BTNCONN);  
 31     CButton *pBtnDisConn = (CButton *)pDlg->GetDlgItem(IDC_BTNDISCONN);  
 32     CButton *pBtnSendData =  (CButton *)pDlg->GetDlgItem(IDC_BTNSENDDATA);  
 33     ASSERT(NULL != pBtnConn);  
 34     ASSERT(NULL != pBtnDisConn);  
 35     ASSERT(NULL != pBtnSendData);  
 36     pBtnConn->EnableWindow(TRUE);  
 37     pBtnDisConn->EnableWindow(FALSE);  
 38     pBtnSendData->EnableWindow(FALSE);  
 39 }  
 40   
 41 // 接收的数据显示  
 42 void CALLBACK CTCPClientDlg::OnRead(CWnd *pWnd,const char *pcBuffer,int iLen)  
 43 {  
 44     CTCPClientDlg *pDlg = (CTCPClientDlg *)pWnd;  
 45     CString csRecvStr = pcBuffer;  
 46     CEdit *pEdtRecv = (CEdit *)pDlg->GetDlgItem(IDC_EDTRECV);  
 47   
 48     ASSERT(NULL != pEdtRecv);  
 49     //将接收的数据显示到接收文本框上  
 50     pEdtRecv->SetWindowText(csRecvStr);  
 51 }  
 52   
 53 // Socket 错误显示  
 54 void CALLBACK CTCPClientDlg::OnError(CWnd *pWnd,int iErrorCode)  
 55 {  
 56     CString csErrorInfo;  
 57   
 58     csErrorInfo.Format(L"%s:%d",L"客户端socket发生错误",iErrorCode);  
 59     AfxMessageBox(csErrorInfo);  
 60 }  
 61   
 62 // 建立连接  
 63 void CTCPClientDlg::OnBtnconn()   
 64 {  
 65     CStatic *pStatus = (CStatic *)GetDlgItem(IDC_LBLCONNSTATUS);  
 66     CButton *pBtnConn = (CButton *)GetDlgItem(IDC_BTNCONN);  
 67     CButton *pBtnDisConn = (CButton *)GetDlgItem(IDC_BTNDISCONN);  
 68     CButton *pBtnSendData =  (CButton *)GetDlgItem(IDC_BTNSENDDATA);  
 69       
 70     ASSERT(NULL != pStatus);  
 71     ASSERT(NULL != pBtnConn);  
 72     ASSERT(NULL != pBtnDisConn);  
 73     ASSERT(NULL != pBtnSendData);  
 74   
 75     UpdateData(TRUE);  
 76     // 设置 m_tcpClient 属性  
 77     m_tcpClient.m_csRemoteHost = m_csRemoteHost;  
 78     m_tcpClient.m_iPort = m_iRemotePort;  
 79     m_tcpClient.OnDisConnect = OnDisConnect;  
 80     m_tcpClient.OnRead = OnRead;  
 81     m_tcpClient.OnError = OnError;  
 82     // 打开客户端 socket  
 83     m_tcpClient.Open(this);  
 84       
 85     // 与服务器端连接  
 86     if(m_tcpClient.Connect())  
 87     {  
 88         pStatus->SetWindowText(L"建立连接");  
 89         UpdateData(FALSE);  
 90     }  
 91     else  
 92     {  
 93         AfxMessageBox(L"建立连接失败");  
 94         pStatus->SetWindowText(L"连接断开");  
 95         return;  
 96     }  
 97   
 98     pBtnConn->EnableWindow(FALSE);  
 99     pBtnDisConn->EnableWindow(TRUE);  
100     pBtnSendData->EnableWindow(TRUE);  
101 }  
102   
103 // 断开连接  
104 void CTCPClientDlg::OnBtndisconn()   
105 {  
106     if(m_tcpClient.Close())  
107     {  
108         CStatic *pStatus = (CStatic *)GetDlgItem(IDC_LBLCONNSTATUS);  
109         CButton *pBtnConn =(CButton*)GetDlgItem(IDC_BTNCONN);  
110         CButton *pBtnDisConn = (CButton*)GetDlgItem(IDC_BTNDISCONN);  
111         CButton *pBtnSendData = (CButton*)GetDlgItem(IDC_BTNSENDDATA);  
112   
113         ASSERT(NULL != pStatus);  
114         ASSERT(NULL != pBtnConn);  
115         ASSERT(NULL != pBtnDisConn);  
116         ASSERT(NULL != pBtnSendData);  
117   
118         pStatus->SetWindowText(L"连接断开");  
119         pBtnConn->EnableWindow(TRUE);  
120         pBtnDisConn->EnableWindow(FALSE);  
121         pBtnSendData->EnableWindow(FALSE);  
122     }  
123     else  
124     {  
125         AfxMessageBox(L"连接断开失败");  
126     }     
127 }  
128   
129 // 发送数据  
130 void CTCPClientDlg::OnBtnsenddata()   
131 {  
132     char *pcSendBuf;  
133     int iSendLength = 0;  
134     UpdateData(TRUE);  
135       
136     iSendLength = m_csSendData.GetLength();  
137     pcSendBuf = new char[iSendLength * 2];  
138     ASSERT(NULL != pcSendBuf);  
139     wcstombs(pcSendBuf,m_csSendData,iSendLength);  
140     if(!m_tcpClient.SendData(pcSendBuf,iSendLength))  
141     {  
142         AfxMessageBox(L"发送失败");  
143     }  
144     delete[] pcSendBuf;  
145     pcSendBuf = NULL;         
146 }