按键盘上方向键 ← 或 → 可快速上下翻页,按键盘上的 Enter 键可回到本书目录页,按键盘上方向键 ↑ 可回到本页顶部!
————未阅读完?加入书签已便下次继续阅读!
//创建 socket 失败
if (!m_pSocket…》Create())
{
delete m_pSocket;
m_pSocket = NULL;
AfxMessageBox(〃创建 socket 失败!〃);
return FALSE;
}
//连接失败
if(!m_pSocket…》Connect(lpszAddress; nPort))
{
delete m_pSocket;
m_pSocket = NULL;
return FALSE;
}
m_pFile = new CSocketFile(m_pSocket);
//用于数据接收和发送
m_pArchiveIn = new CArchive(m_pFile;CArchive::load);
m_pArchiveOut = new CArchive(m_pFile;CArchive::store);
//将用户名发送给服务器
SendMsg(m_strName; SENDING_NICKNAME; false);
·316 ·
…………………………………………………………Page 328……………………………………………………………
第 11 章 网络编程
CString strTemp;
strTemp。Format(〃进入聊天室〃);
//作为普通消息发送给服务器
SendMsg(strTemp; NORMAL_MESSAGE; true);
return TRUE;
}
为 SendMsg()函数编写如下代码:
void CChatClientDoc::SendMsg(CString& strText; int mCode; BOOL bSendHandle)
{
if (m_pArchiveOut != NULL)
{
CMsg msg;
msg。code = mCode;
msg。m_strText = (bSendHandle ? m_strName + _T(〃: 〃) + strText : strText);
TRY
{
msg。Serialize(*m_pArchiveOut);
m_pArchiveOut…》Flush();
}
CATCH(CFileException; e)
{
m_pArchiveOut…》Abort();
delete m_pArchiveOut;
m_pArchiveOut = NULL;
CString strTemp;
strTemp。Format(〃发送失败〃);
DisplayMsg(strTemp);
}
END_CATCH
}
}
为 ProcessReceive() 函数编写如下代码:
void CChatClientDoc::ProcessReceive()
{
do
{
ReceiveMsg();
if (m_pSocket == NULL)
return;
·317 ·
…………………………………………………………Page 329……………………………………………………………
Visual C++ 6。0 程序设计从入门到精通
}
while(!m_pArchiveIn…》IsBufferEmpty());
}
为 ReceiveMsg() 函数编写如下代码:
void CChatClientDoc::ReceiveMsg()
{
CMsg msg;
TRY
{
msg。Serialize(*m_pArchiveIn);
if(msg。code == SENDING_CHATTERS_LIST)
{
//更新用户列表
UpdateChattersList(&msg);
return;
}
if(msg。code == USED_NAME)
{
//如果是用户名已存在信息则关闭连接,退出
AfxMessageBox(msg。m_strText);
msg。m_bClose = TRUE;
OnDisconnect();
return;
}
//显示消息
DisplayMsg(msg。m_strText);
}
CATCH(CFileException; e)
{
msg。m_bClose = TRUE;
m_pArchiveOut…》Abort();
CString strTemp;
strTemp。Format(〃接收数据失败〃);
DisplayMsg(strTemp);
}
END_CATCH
//如果连接已关闭则删除各对象
if (msg。m_bClose)
{
delete m_pArchiveIn;
·318 ·
…………………………………………………………Page 330……………………………………………………………
第 11 章 网络编程
m_pArchiveIn = NULL;
delete m_pArchiveOut;
m_pArchiveOut = NULL;
delete m_pFile;
m_pFile = NULL;
delete m_pSocket;
m_pSocket = NULL;
}
}
为 DisplayMsg() 函数编写如下代码:
void CChatClientDoc::DisplayMsg(LPCTSTR lpszText)
{
for(POSITION pos=GetFirstViewPosition();pos!=NULL;)
{
CView* pView = GetNextView(pos);
CMessageView* pChatView = DYNAMIC_DOWNCAST(CMessageView; pView);
if (pChatView != NULL)
pChatView…》ShowMessage(lpszText);
}
}
为 UpdateChattersList() 函数编写如下代码:
void CChatClientDoc::UpdateChattersList(CMsg* pMsg)
{
CChattersView* pChattersView;
for(POSITION pos=GetFirstViewPosition();pos!=NULL;)
{
CView* pView = GetNextView(pos);
pChattersView = DYNAMIC_DOWNCAST(CChattersView; pView);
//首先清空用户列表视图
if (pChattersView != NULL)
pChattersView…》ClearChattersList();
}
CString strTemp = pMsg…》m_strText;
//得到所有的用户名并加入到用户列表中
do
{
CString sName = strTemp。Left(strTemp。Find(〃:〃;0));
pChattersView…》AddToChattersList(sName);
strTemp = strTemp。Mid(strTemp。Find(〃:〃 ; 0)+1);
}while(strTemp。Find(〃:〃;0) != …1);
·319 ·
…………………………………………………………Page 331……………………………………………………………
Visual C++ 6。0 程序设计从入门到精通
}
为 DeleteContents() 函数编写如下代码:
void CChatClientDoc::DeleteContents()
{
if ((m_pSocket != NULL) && (m_pFile != NULL) && (m_pArchiveOut != NULL))
{
//首先发送用户名
SendMsg(m_strName; LEAVING_CHAT; false);
CMsg msg;
CString strTemp;
//发送普通消息
strTemp。Format(〃:离开了聊天室〃);
msg。code = NORMAL_MESSAGE;
msg。m_bClose = TRUE;
msg。m_strText = m_strName + strTemp;
msg。Serialize(*m_pArchiveOut);
m_pArchiveOut…》Flush();
}
//删除各对象
delete m_pArchiveOut;
m_pArchiveOut = NULL;
delete m_pArchiveIn;
m_pArchiveIn = NULL;
delete m_pFile;
m_pFile = NULL;
//关闭连接
if (m_pSocket != NULL)
{
m_pSocket…》ShutDown(2);
delete m_pSocket;
m_pSocket = NULL;