QQ聊天工具实现，包含客户端服务器程序，使用MFC

// srvrdoc.cpp : implementation of the CServerDoc class // // This is a part of the Microsoft Foundation Classes C++ library. // Copyright (C) 1992-1997 Microsoft Corporation // All rights reserved. // // This source code is only intended as a supplement to the // Microsoft Foundation Classes Reference and related // electronic documentation provided with the library. // See these sources for detailed information regarding the // Microsoft Foundation Classes product. #include "stdafx.h" #include "chatsrvr.h" #include "srvrdoc.h" #include "srvrvw.h" #include "message.h" #include "CleanDlg.h" #ifdef _DEBUG #undef THIS_FILE static char BASED_CODE THIS_FILE[] = __FILE__; #endif ///////////////////////////////////////////////////////////////////////////// // CServerDoc IMPLEMENT_DYNCREATE(CServerDoc, CDocument) BEGIN_MESSAGE_MAP(CServerDoc, CDocument) //{{AFX_MSG_MAP(CServerDoc) ON_COMMAND(ID_CLEAN, OnClean) //}}AFX_MSG_MAP ON_UPDATE_COMMAND_UI(ID_MESSAGES, OnUpdateMessages) ON_UPDATE_COMMAND_UI(ID_CONNECTIONS, OnUpdateConnections) //ON_UPDATE_COMMAND_UI(ID_TEST,On_My) END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CServerDoc construction/destruction CServerDoc::CServerDoc() { m_pSocket = NULL; m_lCount = 0; } CServerDoc::~CServerDoc() { if(m_pSocket != NULL) delete m_pSocket; m_pSocket = NULL; } // //这里开始打开该端口的网络服务，等待客户的连接 // BOOL CServerDoc::OnNewDocument() { if (!CDocument::OnNewDocument()) return FALSE; m_pSocket = new CListeningSocket(this); if (m_pSocket->Create(PORT)){ if (m_pSocket->Listen()) return TRUE; } return FALSE; } //该函数在有客户连接时关闭服务器时调用 void CServerDoc::DeleteContents() { delete m_pSocket; m_pSocket = NULL; CString temp; temp="服务器已经关掉!"; CMessage* pMsg = new CMessage; while(!m_connectionList.IsEmpty()) { CClientSocket* pSocket = (CClientSocket*)m_connectionList.RemoveHead(); if(pSocket==NULL ) continue; if (pMsg == NULL) continue; pMsg->m_bClose = TRUE; pMsg->from = pSocket->name; pMsg->m_strText = _TEXT("服务器已关闭"); pMsg->to = _TEXT("所有人"); pMsg->type = -3; SendMsg(pSocket, pMsg); if(!pSocket->IsAborted()) { pSocket->ShutDown(); BYTE Buffer[50]; while (pSocket->Receive(Buffer,50) > 0); delete pSocket; } } delete pMsg; if (!m_viewList.IsEmpty()) ((CEditView*)m_viewList.GetHead())->SetWindowText(_T("")); CDocument::DeleteContents(); } ///////////////////////////////////////////////////////////////////////////// // CServerDoc serialization void CServerDoc::Serialize(CArchive& ar) { if (ar.IsStoring()) { // CEditView contains an edit control which handles all serialization ((CEditView*)m_viewList.GetHead())->SerializeRaw(ar);//如果是多文挡，该连表中可能含有多个视图类 } } ///////////////////////////////////////////////////////////////////////////// // CServerDoc diagnostics #ifdef _DEBUG void CServerDoc::AssertValid() const { CDocument::AssertValid(); } void CServerDoc::Dump(CDumpContext& dc) const { CDocument::Dump(dc); } #endif //_DEBUG ///////////////////////////////////////////////////////////////////////////// // CServerDoc Operations void CServerDoc::UpdateClients() { CMessage* pMsg = AssembleMsg(NULL); if(pMsg->type >= 0) m_lCount ++; for(POSITION pos = m_connectionList.GetHeadPosition(); pos != NULL;){ CClientSocket* pSocket = (CClientSocket*)m_connectionList.GetNext(pos); if (pMsg != NULL) SendMsg(pSocket, pMsg); } } void CServerDoc::ProcessPendingAccept() { CClientSocket* pSocket = new CClientSocket(this); if (m_pSocket->Accept(*pSocket)) { pSocket->Init(); m_connectionList.AddTail(pSocket); } else delete pSocket; } void CServerDoc::ProcessPendingRead(CClientSocket* pSocket) { if(pSocket != NULL) { do{ CMessage* pMsg = ReadMsg(pSocket); if(pMsg == NULL) return; if (pMsg->m_bClose) { CloseSocket(pSocket); pMsg->m_bClose = FALSE; break; } if(pMsg->type == -1){ UpdateList(pSocket); } } while (!pSocket->m_pArchiveIn->IsBufferEmpty()); } UpdateClients(); } CMessage* CServerDoc::AssembleMsg(CClientSocket* pSocket) { static CMessage msg; CMessage* tmp; tmp = m_lsMsg.RemoveHead(); if(tmp != NULL) memcpy(&msg, tmp, sizeof(CMessage)); return &msg; } CMessage* CServerDoc::ReadMsg(CClientSocket* pSocket) { static CMessage msg; CClientSocket* pSock; TRY { pSocket->ReceiveMsg(&msg); if(msg.type == -1) { bool found=false; for(POSITION posname=m_connectionList.GetHeadPosition();posname;){ pSock = (CClientSocket*)m_connectionList.GetNext(posname); if(pSock->name==msg.from ){ found=true; pSocket->needdel=true; DelSocket(); return NULL; } } if(found==false) { pSocket->name = msg.from; pSocket->image = msg.image; Message1(msg.from); Message1("风尘仆仆地推门而入\r\n"); } } if(msg.type == -2) { Message1(msg.from); Message1("静静地离开了，孤单的背影显得格外潇洒\r\n"); } int i = msg.type; if(i >= 0 && i <= 32){ talk(i,msg.from,msg.to,msg.m_strText); } if(i >= -2 && i <= 32){ m_lsMsg.AddTail(&msg); } } CATCH(CFileException, e) { CString strTemp; if (strTemp.LoadString(IDS_READERROR)) Message(strTemp); msg.m_bClose = TRUE; pSocket->Abort(); } END_CATCH return &msg; } void CServerDoc::SendMsg(CClientSocket* pSocket, CMessage* pMsg) { TRY { pSocket->SendMsg(pMsg); } CATCH(CFileException, e) { pSocket->Abort(); CString strTemp; if (strTemp.LoadString(IDS_SENDERROR)) Message(strTemp); } END_CATCH } void CServerDoc::CloseSocket(CClientSocket* pSocket) { pSocket->Close(); POSITION pos,temp; for(pos = m_connectionList.GetHeadPosition(); pos != NULL;) { temp = pos; CClientSocket* pSock = (CClientSocket*)m_connectionList.GetNext(pos); if (pSock == pSocket){ m_connectionList.RemoveAt(temp); break; } } delete pSocket; } void CServerDoc::Message(LPCTSTR lpszMessage) { ((CServerView*)m_viewList.GetHead())->Message(lpszMessage); } void CServerDoc::Message1(LPCTSTR lpszMessage) { ((CServerView*)m_viewList.GetHead())->Message1(lpszMessage); } ///////////////////////////////////////////////////////////////////////////// // CServerDoc Handlers void CServerDoc::OnUpdateMessages(CCmdUI* pCmdUI/*代表的就是那个状态栏中的资源*/) { pCmdUI->Enable(TRUE); CString strFmt; if (strFmt.LoadString(IDS_MESSAGESFMT)) { CString strTemp; wsprintf(strTemp.GetBuffer(50),strFmt,m_lCount); strTemp.ReleaseBuffer(); pCmdUI->SetText(strTemp); } } void CServerDoc::OnUpdateConnections(CCmdUI* pCmdUI) { pCmdUI->Enable(FALSE);//我终于明白了 CString strFmt; if (strFmt.LoadString(IDS_CONNECTIONSFMT)) { CString strTemp; wsprintf(strTemp.GetBuffer(50),strFmt,m_connectionList.GetCount()); strTemp.ReleaseBuffer(); pCmdUI->SetText(strTemp); } } void CServerDoc::talk(int type2, CString from1, CString to1, CString str1) { CString temp,to2,first,second; if(type2 > 32 || type2 < 0) return; temp.LoadString(IDS_TALK0 + type2); int i=temp.Find(","); if(i!=-1){ first=temp.Left(i); if(i!=temp.GetLength()-1){ second=temp.Mid(i+1); second+="："; } else{ second="："; } Message1(from1); Message1(first); Message1(to1); Message1(second); Message1(str1); Message1("\r\n"); } else{ first=temp; second="： "; Message1(from1); Message1(first); Message1(second); Message1(str1); Message1("\r\n"); } } void CServerDoc::DelSocket() { CClientSocket* pSock; POSITION pos, temp; for(pos = m_connection