ChatClient.cpp
上传用户:power_led
上传日期:2013-04-11
资源大小:373k
文件大小:6k
源码类别:

ICQ/即时通讯

开发平台:

Visual C++

  1. // ChatClient.cpp : Defines the class behaviors for the application.
  2. //
  3. #include "stdafx.h"
  4. #include "ChatClient.h"
  5. #include "MainFrm.h"
  6. #include "ChatClientDoc.h"
  7. #include "ChatClientView.h"
  8. #include "ConnectServerDlg.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CChatClientApp
  16. BEGIN_MESSAGE_MAP(CChatClientApp, CWinApp)
  17. //{{AFX_MSG_MAP(CChatClientApp)
  18. ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
  19. // NOTE - the ClassWizard will add and remove mapping macros here.
  20. //    DO NOT EDIT what you see in these blocks of generated code!
  21. //}}AFX_MSG_MAP
  22. // Standard file based document commands
  23. ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew)
  24. ON_COMMAND(ID_FILE_OPEN, CWinApp::OnFileOpen)
  25. // Standard print setup command
  26. ON_COMMAND(ID_FILE_PRINT_SETUP, CWinApp::OnFilePrintSetup)
  27. END_MESSAGE_MAP()
  28. /////////////////////////////////////////////////////////////////////////////
  29. // CChatClientApp construction
  30. CChatClientApp::CChatClientApp()
  31. {
  32. // TODO: add construction code here,
  33. // Place all significant initialization in InitInstance
  34. }
  35. /////////////////////////////////////////////////////////////////////////////
  36. // The one and only CChatClientApp object
  37. CChatClientApp theApp;
  38. /////////////////////////////////////////////////////////////////////////////
  39. // CChatClientApp initialization
  40. BOOL CChatClientApp::InitInstance()
  41. {
  42. if (!AfxSocketInit())
  43. {
  44. AfxMessageBox(IDP_SOCKETS_INIT_FAILED);
  45. return FALSE;
  46. }
  47. if(!AfxInitRichEdit() )
  48. {
  49. AfxMessageBox(_T("Error"));
  50. return FALSE;
  51. }
  52. AfxEnableControlContainer();
  53. // Standard initialization
  54. // If you are not using these features and wish to reduce the size
  55. //  of your final executable, you should remove from the following
  56. //  the specific initialization routines you do not need.
  57. #ifdef _AFXDLL
  58. Enable3dControls(); // Call this when using MFC in a shared DLL
  59. #else
  60. Enable3dControlsStatic(); // Call this when linking to MFC statically
  61. #endif
  62. // Change the registry key under which our settings are stored.
  63. // TODO: You should modify this string to be something appropriate
  64. // such as the name of your company or organization.
  65. SetRegistryKey(_T("Local AppWizard-Generated Applications"));
  66. LoadStdProfileSettings();  // Load standard INI file options (including MRU)
  67. // Register the application's document templates.  Document templates
  68. //  serve as the connection between documents, frame windows and views.
  69. CSingleDocTemplate* pDocTemplate;
  70. pDocTemplate = new CSingleDocTemplate(
  71. IDR_MAINFRAME,
  72. RUNTIME_CLASS(CChatClientDoc),
  73. RUNTIME_CLASS(CMainFrame),       // main SDI frame window
  74. NULL);
  75. AddDocTemplate(pDocTemplate);
  76. // Parse command line for standard shell commands, DDE, file open
  77. CCommandLineInfo cmdInfo;
  78. ParseCommandLine(cmdInfo);
  79. // Dispatch commands specified on the command line
  80. if (!ProcessShellCommand(cmdInfo))
  81. return FALSE;
  82. // The one and only window has been initialized, so show and update it.
  83. m_pMainWnd->ShowWindow(SW_HIDE);
  84. m_pMainWnd->UpdateWindow();
  85. m_pMainWnd->SetWindowText(_T("QQ-Client"));
  86. if(!m_skMainSocket.Create())
  87. {
  88. AfxMessageBox("端口创建错误");
  89. return FALSE;
  90. }
  91. int iTryTimes = 0;
  92. CString strName;
  93. if(!IniFile.Create("username.ini",1))
  94. {
  95. AfxMessageBox("初始化文件不存在");
  96. return FALSE;
  97. }
  98. CString strServerIp;
  99. int iServerPort;
  100. IniFile.GetVarStr("USERNAME","DEFUALTNAME",strName);
  101. if(!IniFile.GetVarStr("SERVERIP","DEFUALTSERVERIP",strServerIp))
  102. {
  103. strServerIp = _T("127.0.0.1");
  104. }
  105. if(!IniFile.GetVarInt("SERVERPORT","DEFAULTSERVERPORT",iServerPort))
  106. {
  107. iServerPort = 1080;
  108. }
  109. while(iTryTimes < 3)
  110. {
  111. CConnectServerDlg Dlg(strName,strServerIp,iServerPort);
  112. if(Dlg.DoModal() == IDCANCEL)
  113. return FALSE;
  114. if(!m_skMainSocket.Connect(Dlg.m_strServerIp,Dlg.m_iServerPort))
  115. {
  116. if(MessageBox(NULL,"服务器连接错误","重试?",IDRETRY) == IDNO)
  117. {
  118. // ExitInstance();
  119. return FALSE;
  120. }
  121. iTryTimes ++;
  122. }
  123. else
  124. {
  125. Message msObj;
  126. msObj.iType = FIRSTLOG;
  127. msObj.iSubType = FIRSTTIME;
  128. CString strTemp = Dlg.m_strUserName;
  129. int iLen = strTemp.GetLength();
  130. lstrcpy(msObj.strName,_T(strTemp.GetBuffer(iLen)));
  131. strTemp.ReleaseBuffer();
  132. m_skMainSocket.Send(&msObj,sizeof(Message));
  133. m_skMainSocket.SetUserName(strTemp);
  134. return TRUE;
  135. }
  136. if(iTryTimes >=3)
  137. {
  138. AfxMessageBox("已经重试3次,退出");
  139. return FALSE;
  140. }
  141. }
  142. return TRUE;
  143. }
  144. /////////////////////////////////////////////////////////////////////////////
  145. // CAboutDlg dialog used for App About
  146. class CAboutDlg : public CDialog
  147. {
  148. public:
  149. CAboutDlg();
  150. // Dialog Data
  151. //{{AFX_DATA(CAboutDlg)
  152. enum { IDD = IDD_ABOUTBOX };
  153. //}}AFX_DATA
  154. // ClassWizard generated virtual function overrides
  155. //{{AFX_VIRTUAL(CAboutDlg)
  156. protected:
  157. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  158. //}}AFX_VIRTUAL
  159. // Implementation
  160. protected:
  161. //{{AFX_MSG(CAboutDlg)
  162. // No message handlers
  163. //}}AFX_MSG
  164. DECLARE_MESSAGE_MAP()
  165. };
  166. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  167. {
  168. //{{AFX_DATA_INIT(CAboutDlg)
  169. //}}AFX_DATA_INIT
  170. }
  171. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  172. {
  173. CDialog::DoDataExchange(pDX);
  174. //{{AFX_DATA_MAP(CAboutDlg)
  175. //}}AFX_DATA_MAP
  176. }
  177. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  178. //{{AFX_MSG_MAP(CAboutDlg)
  179. // No message handlers
  180. //}}AFX_MSG_MAP
  181. END_MESSAGE_MAP()
  182. // App command to run the dialog
  183. void CChatClientApp::OnAppAbout()
  184. {
  185. CAboutDlg aboutDlg;
  186. aboutDlg.DoModal();
  187. }
  188. /////////////////////////////////////////////////////////////////////////////
  189. // CChatClientApp message handlers
  190. int CChatClientApp::ExitInstance() 
  191. {
  192. // TODO: Add your specialized code here and/or call the base class
  193. Shell_NotifyIcon(NIM_DELETE,&TaskIcon);
  194. Message msObj;
  195. memset(&msObj,0,sizeof(Message));
  196. msObj.iType = USERSESSION;
  197. msObj.iSubType = USERQUIT;
  198. CString strTemp = m_skMainSocket.GetUserName();
  199. int iLen = strTemp.GetLength();
  200. lstrcpy(msObj.strName,_T(strTemp.GetBuffer(iLen)));
  201. strTemp.ReleaseBuffer();
  202. strTemp = m_skMainSocket.GetRoomName();
  203. iLen = strTemp.GetLength();
  204. lstrcpy(msObj.strRoom,_T(strTemp.GetBuffer(iLen)));
  205. strTemp.ReleaseBuffer();
  206. m_skMainSocket.Send(&msObj,sizeof(Message));
  207. Sleep(100);
  208. m_skMainSocket.ShutDown();
  209. m_skMainSocket.Close();
  210. return CWinApp::ExitInstance();
  211. }