仿QQ即时通信源码(包括客户端和服务器的IM源码)

// CChatServer.cpp : Defines the class behaviors for the application. // #include "stdafx.h" #include "CChatServer.h" #include "SetServerPortDlg.h" #include "MyTreeView.h" #include "ClientSocket.h" #include "CIniFile.h" #include "MainFrm.h" #include "CChatServerDoc.h" #include "CChatServerView.h" #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif ///////////////////////////////////////////////////////////////////////////// // CCChatServerApp BEGIN_MESSAGE_MAP(CCChatServerApp, CWinApp) //{{AFX_MSG_MAP(CCChatServerApp) ON_COMMAND(ID_APP_ABOUT, OnAppAbout) // NOTE - the ClassWizard will add and remove mapping macros here. // DO NOT EDIT what you see in these blocks of generated code! //}}AFX_MSG_MAP // Standard file based document commands ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew) ON_COMMAND(ID_FILE_OPEN, CWinApp::OnFileOpen) // Standard print setup command ON_COMMAND(ID_FILE_PRINT_SETUP, CWinApp::OnFilePrintSetup) END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CCChatServerApp construction CCChatServerApp::CCChatServerApp():bInit(FALSE) { // TODO: add construction code here, // Place all significant initialization in InitInstance m_pClientSocketList = new CPtrList(); CIniFile m_IniFile; bInit = m_IniFile.Create("RoomList.ini"); if(bInit) { int iRoomNum; m_IniFile.GetVarInt("ROOMNUM","RoomNumber",iRoomNum); for(int i = 1; i<=iRoomNum; i++) { CString *strRoom = new CString ; CString strTemp ;; strTemp.Format("Room%d",i); m_IniFile.GetVarStr("ROOMNAME",strTemp,*strRoom); m_ChatRoomList.AddTail(strRoom); } } } ///////////////////////////////////////////////////////////////////////////// // The one and only CCChatServerApp object CCChatServerApp theApp; CCriticalSection g_cs; ///////////////////////////////////////////////////////////////////////////// // CCChatServerApp initialization BOOL CCChatServerApp::InitInstance() { if(!bInit) { AfxMessageBox("初始化文件不存在"); return FALSE; } if (!AfxSocketInit()) { return FALSE; } AfxEnableControlContainer(); // Standard initialization // If you are not using these features and wish to reduce the size // of your final executable, you should remove from the following // the specific initialization routines you do not need. #ifdef _AFXDLL Enable3dControls(); // Call this when using MFC in a shared DLL #else Enable3dControlsStatic(); // Call this when linking to MFC statically #endif // Change the registry key under which our settings are stored. // TODO: You should modify this string to be something appropriate // such as the name of your company or organization. SetRegistryKey(_T("Local AppWizard-Generated Applications")); LoadStdProfileSettings(); // Load standard INI file options (including MRU) // Register the application's document templates. Document templates // serve as the connection between documents, frame windows and views. CSingleDocTemplate* pDocTemplate; pDocTemplate = new CSingleDocTemplate( IDR_MAINFRAME, RUNTIME_CLASS(CCChatServerDoc), RUNTIME_CLASS(CMainFrame), // main SDI frame window //RUNTIME_CLASS(CMyTreeView)); NULL); AddDocTemplate(pDocTemplate); // Parse command line for standard shell commands, DDE, file open CCommandLineInfo cmdInfo; ParseCommandLine(cmdInfo); // Dispatch commands specified on the command line if (!ProcessShellCommand(cmdInfo)) { return FALSE; } // The one and only window has been initialized, so show and update it. // // Place all significant initialization in InitInstance m_pMainWnd->ShowWindow(SW_SHOW); m_pMainWnd->UpdateWindow(); m_pMainWnd->SetWindowText("会议中心"); CSetServerPortDlg Dlg; if(Dlg.DoModal() == IDCANCEL) { return FALSE; } else { if(!m_skServerSocket.Create(Dlg.m_iServerPort)) { if(m_ChatRoomList.GetCount()) { m_ChatRoomList.RemoveAll(); } return FALSE; } } if(!m_skServerSocket.Listen()) { return FALSE; } TaskIcon.cbSize = sizeof(TaskIcon); TaskIcon.hWnd = AfxGetMainWnd()->GetSafeHwnd(); lstrcpy(TaskIcon.szTip,_T("服务器正在运行...")); TaskIcon.uFlags = 0x07; TaskIcon.hIcon = LoadIcon(IDR_MAINFRAME); TaskIcon.uCallbackMessage = WM_TASKICON; Shell_NotifyIcon(NIM_ADD,&TaskIcon); return TRUE; } ///////////////////////////////////////////////////////////////////////////// // CAboutDlg dialog used for App About class CAboutDlg : public CDialog { public: CAboutDlg(); // Dialog Data //{{AFX_DATA(CAboutDlg) enum { IDD = IDD_ABOUTBOX }; //}}AFX_DATA // ClassWizard generated virtual function overrides //{{AFX_VIRTUAL(CAboutDlg) protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support //}}AFX_VIRTUAL // Implementation protected: //{{AFX_MSG(CAboutDlg) // No message handlers //}}AFX_MSG DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD) { //{{AFX_DATA_INIT(CAboutDlg) //}}AFX_DATA_INIT } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); //{{AFX_DATA_MAP(CAboutDlg) //}}AFX_DATA_MAP } BEGIN_MESSAGE_MAP(CAboutDlg, CDialog) //{{AFX_MSG_MAP(CAboutDlg) // No message handlers //}}AFX_MSG_MAP END_MESSAGE_MAP() // App command to run the dialog void CCChatServerApp::OnAppAbout() { CAboutDlg aboutDlg; aboutDlg.DoModal(); } ///////////////////////////////////////////////////////////////////////////// // CCChatServerApp message handlers int CCChatServerApp::ExitInstance() { // TODO: Add your specialized code here and/or call the base class Shell_NotifyIcon(NIM_DELETE,&TaskIcon); //向所有用户发送退出消息 Message msObj; msObj.iType = SYSERROR; msObj.iSubType = SERVERQUIT; CPtrList * m_pClientList = theApp.m_pClientSocketList; POSITION pos = m_pClientList ->GetHeadPosition(); if(pos) { CClientSocket * m_pClientSocket; for(int i = 0; i < m_pClientList ->GetCount(); i++) { // if(m_pClientSocket ->GetState() == HAVELOGIN) { m_pClientSocket = static_cast < CClientSocket *>(m_pClientList->GetNext(pos)); m_pClientSocket ->Send(&msObj,sizeof(Message)); Sleep(100); } } } //删除 DeleteAllList(); return CWinApp::ExitInstance(); } void CCChatServerApp::DeleteAllList() { //删除 POSITION pos = m_pClientSocketList ->GetHeadPosition(); if(pos) { CClientSocket * m_pClientSocket; for(int i = 0; i < m_pClientSocketList ->GetCount(); i++) { m_pClientSocket = static_cast < CClientSocket *>(m_pClientSocketList->GetNext(pos)); ASSERT(m_pClientSocket != NULL); ASSERT_VALID(m_pClientSocket); delete m_pClientSocket; } } // if(m_pClientSocketList ->GetCount()) // delete m_pClientSocketList; pos = m_ChatRoomList.GetHeadPosition(); if(pos) { CString * m_pRoomName; for(int i = 0; i < m_ChatRoomList .GetCount(); i++) { m_pRoomName = static_cast < CString *>(m_ChatRoomList.GetNext(pos)); ASSERT(m_pRoomName!=NULL); delete m_pRoomName; } } // m_ChatRoomList.RemoveAll(); }