www.pudn.com > MultiLineChatRoomVCSrc.zip > ChatClient.cpp

// ChatClient.cpp : Defines the class behaviors for the application. 
// 
 
#include "stdafx.h" 
#include "ChatClient.h" 
 
#include "MainFrm.h" 
#include "ChatClientDoc.h" 
#include "ChatClientView.h" 
#include "ConnectServerDlg.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
///////////////////////////////////////////////////////////////////////////// 
// CChatClientApp 
 
BEGIN_MESSAGE_MAP(CChatClientApp, CWinApp) 
//{{AFX_MSG_MAP(CChatClientApp) 
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() 
 
///////////////////////////////////////////////////////////////////////////// 
// CChatClientApp construction 
 
CChatClientApp::CChatClientApp() 
{ 
	// TODO: add construction code here, 
	// Place all significant initialization in InitInstance 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// The one and only CChatClientApp object 
 
CChatClientApp theApp; 
 
///////////////////////////////////////////////////////////////////////////// 
// CChatClientApp initialization 
 
BOOL CChatClientApp::InitInstance() 
{ 
 
	if (!AfxSocketInit()) 
	{ 
		AfxMessageBox(IDP_SOCKETS_INIT_FAILED); 
		return FALSE; 
	} 
	if(!AfxInitRichEdit() ) 
	{ 
		AfxMessageBox("Error"); 
		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(CChatClientDoc), 
	 
		RUNTIME_CLASS(CMainFrame),       // main SDI frame window 
		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. 
	m_pMainWnd->ShowWindow(SW_SHOW); 
	m_pMainWnd->UpdateWindow(); 
	m_pMainWnd->SetWindowText("客户端"); 
	if(!m_skMainSocket.Create()) 
	{ 
		AfxMessageBox("端口创建错误"); 
		return FALSE; 
	} 
	 
	int iTryTimes = 0; 
		CString strName; 
	if(!IniFile.Create("username.ini")) 
	{ 
		AfxMessageBox("初始化文件不存在"); 
		return FALSE; 
	} 
	 
	IniFile.GetVarStr("USERNAME","DEFUALTNAME",strName); 
 
	while(iTryTimes < 3) 
	{ 
		CConnectServerDlg Dlg(strName); 
		 
		if(Dlg.DoModal() == IDCANCEL) 
			return FALSE; 
 
	 
		if(!m_skMainSocket.Connect(Dlg.m_strServerIp,Dlg.m_iServerPort)) 
		{ 
			 
			if(AfxMessageBox("服务器连接错误",IDRETRY) == IDNO) 
			{ 
			//	ExitInstance(); 
				return FALSE; 
			} 
			iTryTimes ++; 
		} 
		else 
		{ 
			Message msObj; 
			msObj.iType = FIRSTLOG; 
			msObj.iSubType = FIRSTTIME; 
			CString strTemp = Dlg.m_strUserName; 
			int iLen = strTemp.GetLength(); 
			lstrcpy(msObj.strName,strTemp.GetBuffer(iLen)); 
			strTemp.ReleaseBuffer(); 
			m_skMainSocket.Send(&msObj,sizeof(Message)); 
			m_skMainSocket.SetUserName(strTemp); 
			return TRUE; 
		} 
		if(iTryTimes >=3) 
		{ 
			AfxMessageBox("已经重试3次，退出"); 
			return FALSE; 
		} 
	} 
 
	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 CChatClientApp::OnAppAbout() 
{ 
	CAboutDlg aboutDlg; 
	aboutDlg.DoModal(); 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CChatClientApp message handlers 
 
int CChatClientApp::ExitInstance()  
{ 
	// TODO: Add your specialized code here and/or call the base class 
 
	Message msObj; 
	memset(&msObj,0,sizeof(Message)); 
	msObj.iType = USERSESSION; 
	msObj.iSubType = USERQUIT; 
	CString strTemp = m_skMainSocket.GetUserName(); 
	int iLen = strTemp.GetLength(); 
	lstrcpy(msObj.strName,strTemp.GetBuffer(iLen)); 
	strTemp.ReleaseBuffer(); 
	strTemp = m_skMainSocket.GetRoomName(); 
	iLen = strTemp.GetLength(); 
	lstrcpy(msObj.strRoom,strTemp.GetBuffer(iLen)); 
	strTemp.ReleaseBuffer(); 
	m_skMainSocket.Send(&msObj,sizeof(Message)); 
	Sleep(100); 
	m_skMainSocket.ShutDown(); 
	m_skMainSocket.Close(); 
	return CWinApp::ExitInstance(); 
}