这是一个精练的客户端套接字类,类名、函数名和变量名均采用匈牙利命名法。小写的x代表我的姓氏首字母(谢欣伦),个人习惯而已,如有雷同,纯属巧合。
CxClientSocket的使用如下(以某个叫做CSomeClass的类的相关代码为例):
一、声明客户端套接字对象实例。
CxClientSocket m_csRemoteHost;
二、设置客户端套接字事件通知方式。消息响应或回调函数,任选其一。客户端套接字事件有:缓冲区有数据可读取、与远程主机的连接被断开。
void CSomeClass::PreSubclassWindow()
{
// TODO: Add your specialized code here and/or call the base class
HWND hwnd = GetSafeHwnd();
MSG msg = {hwnd, WM_SOCKET_EVENT};
m_usRemoteHost.SetWindowMessage(&msg);
m_csRemoteHost.SetWindowMessage(&msg);
m_ssLocalHost.SetWindowMessage(&msg); if (m_btCSMode == CSM_SERVER)
ListenRemoteHost(); CStatic::PreSubclassWindow();
}
三、连接远程主机。
BOOL CSomeClass::ConnectToRemoteHost()
{
if (m_csRemoteHost.IsConnected())
m_csRemoteHost.DisConnect(); BOOL bRet = m_csRemoteHost.Connect(m_ipapRemoteHost.szAddr,
m_ipapRemoteHost.nPort); return bRet;
}
四、处理消息响应或函数回调。
1.消息响应
BEGIN_MESSAGE_MAP(CSomeClass, CStatic)
//{{AFX_MSG_MAP(CSomeClass)
ON_WM_DESTROY()
//}}AFX_MSG_MAP
ON_MESSAGE(WM_SOCKET_EVENT, OnSocketEvent)
END_MESSAGE_MAP() /////////////////////////////////////////////////////////////////////////////
// CSomeClass message handlers LRESULT CSomeClass::OnSocketEvent(WPARAM wParam, LPARAM lParam)
{
if (WSAGETSELECTERROR(lParam) != )
return ; int nEvent = WSAGETSELECTEVENT(lParam);
switch (nEvent)
{
case FD_ACCEPT: OnSocketAccept(wParam); break;
case FD_CLOSE: OnSocketClose(wParam); break;
case FD_READ: OnSocketRead(wParam); break;
default: break;
} return ;
}
void CSomeClass::OnSocketRead(WPARAM wParam)
{
if (m_csRemoteHost.GetSocket() == (SOCKET)wParam)
{
m_csRemoteHost.SelectEvent(FD_CLOSE);
SOME_DATA_STRUCT data = {};
if (m_csRemoteHost.Recv((LPBYTE)&data, sizeof(data)))
{
//do something with data;
}
m_csRemoteHost.SelectEvent(FD_READ | FD_CLOSE);
}
} void CSomeClass::OnSocketClose(WPARAM wParam)
{
if (m_csRemoteHost.GetSocket() == (SOCKET)wParam)
{
m_csRemoteHost.DisConnect();
char szDebug[MAX_PATH];
sprintf(szDebug, "%s disconnect.\n", m_szRemoteHostName);
OutputDebugString(szDebug);
GetParent()->PostMessage(WM_SOCKET_EVENT, (WPARAM)this, FD_CLOSE);
}
}
2.回调函数
LRESULT CALLBACK CSomeClass::SocketProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
LRESULT lRet = ;
CSomeClass* pThis = (CSomeClass*)hwnd;
if (pThis != )
lRet = pThis->OnSocketEvent(wParam, lParam); return lRet;
}
注意,在使用回调函数时应设置客户端套接字事件通知方式为回调函数,并且设法把this指针传入设置函数,方便在回调函数中调用成员函数。
MSG msg = {(HWND)this, WM_SOCKET_EVENT};
m_csRemoteHost.SetWindowMessage(&msg, SocketProc);
五、断开与远程主机的连接。
void CSomeClass::OnDestroy()
{
CStatic::OnDestroy(); // TODO: Add your message handler code here
TerminateWaitComm(); m_csRemoteHost.DisConnect();
m_ssLocalHost.DisListen();
m_usRemoteHost.Disbind();
}
至于客户端套接字数据发送,很简单:在需要处调用以下函数:
m_csRemoteHost.Send((LPBYTE)pData, sizeof(SOME_DATA_STRUCT));
精练的代码不需要过多解释,你们懂的。To be continued...
下载