Category

Development Platform
Home > SourceCode/Document > Game Program > Game Server Simulator > View Code
MainWnd.cpp
Package: SwordOnline.rar　[view]
Upload User: dzyhzl
Upload Date: 2019-04-29
Package Size: 56270k
Code Size: 23k
Category:

Game Server Simulator

Development Platform:

C/C++

MainWnd.cpp：Code Content
  1. // MainWnd.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "S3DBClient.h"
  6. #include "PerformanceDlg.h"
  7. #include "LoginDlg.h"
  8. #include "MainWnd.h"
  9. #include "AddUserDlg.h"
  11. #ifdef _DEBUG
  12. #define new DEBUG_NEW
  13. #undef THIS_FILE
  14. static char THIS_FILE[] = __FILE__;
  15. #endif
  17. extern CS3DBClientApp theApp;
  19. /////////////////////////////////////////////////////////////////////////////
  20. // CAboutDlg dialog used for App About
  22. class CAboutDlg : public CDialog
  23. {
  24. public:
  25. CAboutDlg();
  27. // Dialog Data
  28. //{{AFX_DATA(CAboutDlg)
  29. enum { IDD = IDD_ABOUTBOX };
  30. //}}AFX_DATA
  32. // ClassWizard generated virtual function overrides
  33. //{{AFX_VIRTUAL(CAboutDlg)
  34. protected:
  35. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  36. //}}AFX_VIRTUAL
  38. // Implementation
  39. protected:
  40. //{{AFX_MSG(CAboutDlg)
  41. //}}AFX_MSG
  42. DECLARE_MESSAGE_MAP()
  43. };
  45. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  46. {
  47. //{{AFX_DATA_INIT(CAboutDlg)
  48. //}}AFX_DATA_INIT
  49. }
  51. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  52. {
  53. CDialog::DoDataExchange(pDX);
  54. //{{AFX_DATA_MAP(CAboutDlg)
  55. //}}AFX_DATA_MAP
  56. }
  58. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  59. //{{AFX_MSG_MAP(CAboutDlg)
  60. // No message handlers
  61. //}}AFX_MSG_MAP
  62. END_MESSAGE_MAP()
  64. /////////////////////////////////////////////////////////////////////////////
  65. // CMainWnd
  67. IMPLEMENT_DYNAMIC(CMainWnd, CPropertySheet)
  69. CMainWnd::CMainWnd(UINT nIDCaption, CWnd* pParentWnd, UINT iSelectPage)
  70. :CPropertySheet(nIDCaption, pParentWnd, iSelectPage)
  71. {
  72. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  74. AddPage( &m_page1 );
  75. AddPage( &m_page2 );
  76. }
  78. CMainWnd::CMainWnd(LPCTSTR pszCaption, CWnd* pParentWnd, UINT iSelectPage)
  79. :CPropertySheet(pszCaption, pParentWnd, iSelectPage)
  80. {
  81. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  83. AddPage( &m_page1 );
  84. AddPage( &m_page2 );
  85. }
  87. CMainWnd::~CMainWnd()
  88. {
  89. }
  92. BEGIN_MESSAGE_MAP(CMainWnd, CPropertySheet)
  93. //{{AFX_MSG_MAP(CMainWnd)
  94. ON_WM_PAINT()
  95. ON_WM_SYSCOMMAND()
  96. ON_COMMAND(ID_ABOUT, OnAbout)
  97. ON_COMMAND(ID_LOGIN, OnLogin)
  98. ON_COMMAND(ID_LOGOUT, OnLogout)
  99. ON_COMMAND(ID_ADDUSER, OnAdduser)
  100. ON_COMMAND(ID_DELETEUSER, OnDeleteuser)
  101. ON_COMMAND(ID_PREFERENCES, OnPreferences)
  102. ON_COMMAND(ID_EXIT, OnExit)
  103. ON_WM_DESTROY()
  104. ON_COMMAND(ID_MAINFACE, OnMainface)
  105. ON_COMMAND(ID_DATABASEUSER, OnDatabaseuser)
  106. ON_COMMAND(ID_REFRESH, OnRefresh)
  107. //}}AFX_MSG_MAP
  108. ON_MESSAGE(WM_INPUTLANGCHANGEREQUEST,OnIMEInputLangChangeRequest)
  109. ON_MESSAGE(WM_TURNTOPAGE2,OnTurntoPage2)
  110. END_MESSAGE_MAP()
  112. /////////////////////////////////////////////////////////////////////////////
  113. // CMainWnd message handlers
  115. BOOL CMainWnd::OnInitDialog() 
  116. {
  117. CPropertySheet::OnInitDialog();
  119. InitClientSocket();
  121. CTabCtrl* pTabCtrl = GetTabControl();
  122. CRect oldTabRc;
  123. CRect newTabRc;
  124. CRect rc;
  125. GetWindowRect( &rc );
  126. if ( NULL != pTabCtrl )
  127. {
  128. pTabCtrl->GetWindowRect( &oldTabRc );
  129. }
  130. m_menu.LoadMenu( IDR_MAINMENU );
  131. SetMenu( &m_menu );
  132. if ( NULL != pTabCtrl )
  133. {
  134. pTabCtrl->GetWindowRect( &newTabRc );
  135. rc.InflateRect( 0, abs( newTabRc.top - oldTabRc.top ) / 2 );
  136. MoveWindow( &rc );
  137. }
  138. // TODO: Add extra initialization here
  139. // Add "About..." menu item to system menu.
  141. // IDM_ABOUTBOX must be in the system command range.
  142. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  143. ASSERT(IDM_ABOUTBOX < 0xF000);
  145. CMenu* pSysMenu = GetSystemMenu(FALSE);
  146. if (pSysMenu != NULL)
  147. {
  148. CString strAboutMenu;
  149. strAboutMenu.LoadString(IDS_ABOUTBOX);
  150. if (!strAboutMenu.IsEmpty())
  151. {
  152. pSysMenu->AppendMenu(MF_SEPARATOR);
  153. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  154. }
  155. }
  157. // Set the icon for this dialog.  The framework does this automatically
  158. //  when the application's main window is not a dialog
  159. SetIcon(m_hIcon, TRUE); // Set big icon
  160. SetIcon(m_hIcon, FALSE); // Set small icon
  162. SetActivePage( &m_page1 );
  164. UpdateMenuState();
  165. return TRUE;  // return TRUE unless you set the focus to a control
  166.               // EXCEPTION: OCX Property Pages should return FALSE
  167. }
  169. void CMainWnd::OnPaint() 
  170. {
  171. if (IsIconic())
  172. {
  173. CPaintDC dc(this); // device context for painting
  175. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  177. // Center icon in client rectangle
  178. int cxIcon = GetSystemMetrics(SM_CXICON);
  179. int cyIcon = GetSystemMetrics(SM_CYICON);
  180. CRect rect;
  181. GetClientRect(&rect);
  182. int x = (rect.Width() - cxIcon + 1) / 2;
  183. int y = (rect.Height() - cyIcon + 1) / 2;
  185. // Draw the icon
  186. dc.DrawIcon(x, y, m_hIcon);
  187. }
  188. else
  189. {
  190. CPropertySheet::OnPaint();
  191. }
  192. }
  194. void CMainWnd::OnAbout()
  195. {
  196. CAboutDlg dlg;
  197. dlg.DoModal();
  198. }
  200. void CMainWnd::OnSysCommand(UINT nID, LPARAM lParam) 
  201. {
  202. // TODO: Add your message handler code here and/or call default
  203. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  204. {
  205. CAboutDlg dlgAbout;
  206. dlgAbout.DoModal();
  207. }
  208. else
  209. {
  210. CPropertySheet::OnSysCommand(nID, lParam);
  211. }
  212. }
  214. HCURSOR CMainWnd::OnQueryDragIcon()
  215. {
  216. return ( HCURSOR )m_hIcon;
  217. }
  219. static DWORD CALLBACK LoginCallback( DWORD dwPara1, DWORD dwPara2 )
  220. {
  221. DWORD dwRet = 0;
  222. IBYTE* pRecvMsg = ( IBYTE* )dwPara1;
  223. int* piRet = ( int* )dwPara2;
  224. if ( ( NULL != pRecvMsg )
  225. || ( NULL != piRet ) )
  226. {
  227. DWORD dwMsgParam;
  228. memcpy( &dwMsgParam, &( pRecvMsg[1+sizeof( DWORD )] ), sizeof( DWORD ) );
  229. if ( ( DBLOGIN_R_LOGIN | DBLOGIN_R_SUCCESS ) == dwMsgParam )
  230. {
  231. *piRet = 1;
  232. }
  233. else if ( ( DBLOGIN_R_LOGIN | DBLOGIN_R_HASLOGGEDIN ) == dwMsgParam )
  234. {
  235. *piRet = -1;
  236. }
  237. else
  238. {
  239. *piRet = 0;
  240. }
  241. }
  242. return dwRet;
  243. }
  244. void CMainWnd::OnLogin() 
  245. {
  246. // TODO: Add your command handler code here
  247. if ( theApp.m_bConnected )
  248. {
  249. return;
  250. }
  251. if ( NULL != theApp.m_pClientSocket )
  252. {
  253. CLoginDlg dlg;
  254. if ( IDOK == dlg.DoModal() )
  255. {
  256. _DBLOGINSTRUCT dbLoginStruct;
  257. memset( &dbLoginStruct, 0, sizeof( _DBLOGINSTRUCT ) );
  258. memcpy( dbLoginStruct.szUserName,
  259. dlg.m_cstrUserName,
  260. dlg.m_cstrUserName.GetLength() );
  261. theApp.m_cstrCurUsername = dlg.m_cstrUserName;
  262. memcpy( dbLoginStruct.szPassword,
  263. dlg.m_cstrPassword,
  264. dlg.m_cstrPassword.GetLength() );
  265. theApp.m_cstrCurPassword = dlg.m_cstrPassword;
  266. dbLoginStruct.Size = sizeof( _DBLOGINSTRUCT );
  267. dbLoginStruct.Param = DBLOGIN_A_LOGIN;
  269. int iResult = 0;
  270. BOOL bResult =
  271. theApp.m_pClientSocket->RemoteQuery( &dbLoginStruct,
  272. LoginCallback, ( DWORD )( &iResult ), theApp.m_serverInfo.dwWaitTime );
  273. if ( bResult )
  274. {
  275. CString cstrMsg;
  276. if ( 1 == iResult ) // Log in successfully
  277. {
  278. cstrMsg = _T("Successful to log in.");
  279. UpdateMenuState( TRUE );
  280. OnRefresh();
  281. }
  282. else if ( -1 == iResult ) // Has logged in
  283. {
  284. //cstrMsg = _T("The same user have connected the database server from other machine.");
  285. //UpdateMenuState( FALSE );
  286. cstrMsg = _T("Successful to log in.");
  287. UpdateMenuState( TRUE );
  288. OnRefresh();
  289. }
  290. else
  291. {
  292. cstrMsg = _T("Failed to log in.rnMaybe the username or password you input is wrong.");
  293. }
  294. MessageBox( cstrMsg );
  295. }
  296. else
  297. {
  298. MessageBox( _T("Failed to log in"), NULL, MB_OK | MB_ICONASTERISK );
  299. }
  300. }
  301. }
  302. else
  303. {
  304. MessageBox( _T("No availd socket"), NULL, MB_OK | MB_ICONASTERISK );
  305. }
  306. }
  308. static DWORD CALLBACK LogoutCallback( DWORD dwPara1, DWORD dwPara2 )
  309. {
  310. DWORD dwRet = 0;
  311. IBYTE* pRecvMsg = ( IBYTE* )dwPara1;
  312. int* piRet = ( int* )dwPara2;
  313. if ( ( NULL != pRecvMsg )
  314. || ( NULL != piRet ) )
  315. {
  316. DWORD dwMsgParam;
  317. memcpy( &dwMsgParam, &( pRecvMsg[1+sizeof( DWORD )] ), sizeof( DWORD ) );
  318. if ( ( DBLOGIN_R_LOGOUT | DBLOGIN_R_SUCCESS ) == dwMsgParam )
  319. {
  320. *piRet = 1;
  321. }
  322. else
  323. {
  324. *piRet = -1;
  325. }
  326. }
  327. return dwRet;
  328. }
  329. void CMainWnd::OnLogout() 
  330. {
  331. // TODO: Add your command handler code here
  332. if ( !theApp.m_bConnected )
  333. {
  334. return;
  335. }
  336. if ( NULL != theApp.m_pClientSocket )
  337. {
  338. _DBLOGINSTRUCT dbLoginStruct;
  339. memset( &dbLoginStruct, 0, sizeof( _DBLOGINSTRUCT ) );
  340. memcpy( dbLoginStruct.szUserName,
  341. theApp.m_cstrCurUsername,
  342. theApp.m_cstrCurUsername.GetLength() );
  343. memcpy( dbLoginStruct.szPassword,
  344. theApp.m_cstrCurPassword,
  345. theApp.m_cstrCurPassword.GetLength() );
  346. dbLoginStruct.Size = sizeof( _DBLOGINSTRUCT );
  347. dbLoginStruct.Param = DBLOGIN_A_LOGOUT;
  349. int iResult = 0;
  350. BOOL bResult =
  351. theApp.m_pClientSocket->RemoteQuery( &dbLoginStruct,
  352. LogoutCallback, ( DWORD )( &iResult ), theApp.m_serverInfo.dwWaitTime );
  353. if ( bResult )
  354. {
  355. CString cstrMsg;
  356. if ( 1 == iResult ) // Log in successfully
  357. {
  358. cstrMsg = _T("Successful to log out.");
  359. UpdateMenuState( FALSE );
  360. OnRefresh();
  361. }
  362. else
  363. {
  364. cstrMsg = _T("Failed to log out.rnMaybe the net is busy, please try again later.");
  365. MessageBox( cstrMsg );
  366. }
  367. }
  368. else
  369. {
  370. MessageBox( _T("Failed to log out"), NULL, MB_OK | MB_ICONASTERISK );
  371. }
  372. }
  373. else
  374. {
  375. MessageBox( _T("No availd socket"), NULL, MB_OK | MB_ICONASTERISK );
  376. }
  377. }
  379. static DWORD CALLBACK AddUserCallback( DWORD dwPara1, DWORD dwPara2 )
  380. {
  381. DWORD dwRet = 0;
  382. IBYTE* pRecvMsg = ( IBYTE* )dwPara1;
  383. int* piRet = ( int* )dwPara2;
  384. if ( ( NULL != pRecvMsg )
  385. || ( NULL != piRet ) )
  386. {
  387. DWORD dwMsgParam;
  388. memcpy( &dwMsgParam, &( pRecvMsg[1+sizeof( DWORD )] ), sizeof( DWORD ) );
  389. if ( ( DBLOGIN_R_ADDDBUSER | DBLOGIN_R_SUCCESS ) == dwMsgParam )
  390. {
  391. *piRet = 1;
  392. }
  393. else if ( ( DBLOGIN_R_ADDDBUSER | DBLOGIN_R_HASNOENOUGHPRIORITY ) == dwMsgParam )
  394. {
  395. *piRet = -1;
  396. }
  397. else if ( ( DBLOGIN_R_ADDDBUSER | DBLOGIN_R_USERHASEXISTED ) == dwMsgParam )
  398. {
  399. *piRet = -2;
  400. }
  401. else
  402. {
  403. *piRet = 0;
  404. }
  405. }
  406. return dwRet;
  407. }
  408. void CMainWnd::OnAdduser() 
  409. {
  410. // TODO: Add your command handler code here
  411. if ( !theApp.m_bConnected )
  412. {
  413. return;
  414. }
  415. if ( NULL != theApp.m_pClientSocket )
  416. {
  417. CAddUserDlg dlg;
  418. if ( IDOK == dlg.DoModal() )
  419. {
  420. _DBLOGINSTRUCT dbLoginStruct;
  421. memset( &dbLoginStruct, 0, sizeof( _DBLOGINSTRUCT ) );
  422. memcpy( dbLoginStruct.szUserName,
  423. theApp.m_cstrCurUsername,
  424. theApp.m_cstrCurUsername.GetLength() );
  425. memcpy( dbLoginStruct.szPassword,
  426. theApp.m_cstrCurPassword,
  427. theApp.m_cstrCurPassword.GetLength() );
  428. dbLoginStruct.Size =
  429. sizeof( _DBLOGINSTRUCT )
  430. + def_DBUSERNAME_MAX_LEN
  431. + def_DBPASSWORD_MAX_LEN
  432. + sizeof( short int );
  433. dbLoginStruct.Param = DBLOGIN_A_ADDDBUSER;
  434. IBYTE* pSendData = new IBYTE[dbLoginStruct.Size];
  435. if ( NULL != pSendData )
  436. {
  437. memset( pSendData, 0, dbLoginStruct.Size );
  438. DWORD dwOffset = 0;
  439. memcpy( pSendData, &dbLoginStruct, sizeof( _DBLOGINSTRUCT ) );
  440. dwOffset += sizeof( _DBLOGINSTRUCT );
  441. memcpy( pSendData + dwOffset, dlg.m_cstrName, dlg.m_cstrName.GetLength() );
  442. dwOffset += def_DBUSERNAME_MAX_LEN;
  443. memcpy( pSendData + dwOffset, dlg.m_cstrPassword, dlg.m_cstrPassword.GetLength() );
  444. dwOffset += def_DBPASSWORD_MAX_LEN;
  445. memcpy( pSendData + dwOffset, &( dlg.m_siPriority ), sizeof( short int ) );
  446. dwOffset += sizeof( short int );
  447. int iResult = 0;
  448. BOOL bResult =
  449. theApp.m_pClientSocket->RemoteQuery( ( _PDBLOGINSTRUCT )pSendData,//&dbLoginStruct,
  450. AddUserCallback, ( DWORD )( &iResult ), theApp.m_serverInfo.dwWaitTime );
  451. if ( bResult )
  452. {
  453. CString cstrMsg;
  454. if ( 1 == iResult ) // Log in successfully
  455. {
  456. OnRefresh();
  457. }
  458. else if ( -1 == iResult )
  459. {
  460. cstrMsg =
  461. _T("Has no enough priority to carry out this operation.");
  462. MessageBox( cstrMsg );
  463. }
  464. else if ( -2 == iResult )
  465. {
  466. cstrMsg.Format( _T("User "%s" has existed."), dlg.m_cstrName );
  467. }
  468. else
  469. {
  470. cstrMsg =
  471. _T("Failed to access database.rnMaybe the net is busy, please try again later.");
  472. MessageBox( cstrMsg );
  473. }
  474. }
  475. else
  476. {
  477. MessageBox( _T("Failed to access database.rnMaybe the net is busy, please try again later."), NULL, MB_OK | MB_ICONASTERISK );
  478. }
  479. delete []pSendData;
  480. pSendData = NULL;
  481. }
  482. else
  483. {
  484. MessageBox( _T("Failed to allocate memory.") );
  485. }
  486. }
  487. }
  488. else
  489. {
  490. MessageBox( _T("No availd socket"), NULL, MB_OK | MB_ICONASTERISK );
  491. }
  492. }
  494. static DWORD CALLBACK DeleteUserCallBack( DWORD dwPara1, DWORD dwPara2 )
  495. {
  496. DWORD dwRet = 0;//DBLOGIN_R_HASNOENOUGHPRIORITY
  497. IBYTE* pRecvMsg = ( IBYTE* )dwPara1;
  498. int* piRet = ( int* )dwPara2;
  499. if ( ( NULL != pRecvMsg )
  500. || ( NULL != piRet ) )
  501. {
  502. DWORD dwMsgParam;
  503. memcpy( &dwMsgParam, &( pRecvMsg[1+sizeof( DWORD )] ), sizeof( DWORD ) );
  504. if ( ( DBLOGIN_R_DELDBUSER | DBLOGIN_R_SUCCESS ) == dwMsgParam )
  505. {
  506. *piRet = 1;
  507. }
  508. else if ( ( DBLOGIN_R_DELDBUSER | DBLOGIN_R_HASNOENOUGHPRIORITY ) == dwMsgParam )
  509. {
  510. *piRet = -1;
  511. }
  512. else
  513. {
  514. *piRet = 0;
  515. }
  516. }
  517. return dwRet;
  518. }
  519. void CMainWnd::OnDeleteuser() 
  520. {
  521. // TODO: Add your command handler code here
  522. if ( ( &m_page2 == GetActivePage() )
  523. && theApp.m_bConnected )
  524. {
  525. UINT uSelectedCount = m_page2.m_dbUserListCtrl.GetSelectedCount();
  526. int  nItem = -1;
  528. if ( 1 == uSelectedCount )
  529. {
  530. nItem = m_page2.m_dbUserListCtrl.GetNextItem( nItem, LVNI_SELECTED );
  531. ASSERT(nItem != -1);
  532. CString cstrSelName = m_page2.m_dbUserListCtrl.GetItemText( nItem, 0 );
  534. if ( NULL != theApp.m_pClientSocket )
  535. {
  536. _DBLOGINSTRUCT dbLoginStruct;
  537. memset( &dbLoginStruct, 0, sizeof( _DBLOGINSTRUCT ) );
  538. memcpy( dbLoginStruct.szUserName,
  539. theApp.m_cstrCurUsername,
  540. theApp.m_cstrCurUsername.GetLength() );
  541. memcpy( dbLoginStruct.szPassword,
  542. theApp.m_cstrCurPassword,
  543. theApp.m_cstrCurPassword.GetLength() );
  544. dbLoginStruct.Size = sizeof( _DBLOGINSTRUCT ) + def_DBUSERNAME_MAX_LEN;
  545. dbLoginStruct.Param = DBLOGIN_A_DELDBUSER;
  546. IBYTE* pSendBuf = new IBYTE[dbLoginStruct.Size];
  547. memset( pSendBuf, 0, dbLoginStruct.Size );
  548. if ( NULL != pSendBuf )
  549. {
  550. memcpy( pSendBuf, &dbLoginStruct, sizeof( _DBLOGINSTRUCT ) );
  551. memcpy( pSendBuf + sizeof( _DBLOGINSTRUCT ), cstrSelName, cstrSelName.GetLength() );
  552. CString cstrMsg;
  553. int iResult = 0;
  554. BOOL bResult =
  555. theApp.m_pClientSocket->RemoteQuery( ( _PDBLOGINSTRUCT )pSendBuf,
  556. DeleteUserCallBack, ( DWORD )( &iResult ), theApp.m_serverInfo.dwWaitTime );
  557. if ( bResult )
  558. {
  559. if ( 1 == iResult )
  560. {
  561. if ( theApp.m_cstrCurUsername == cstrSelName )
  562. {
  563. UpdateMenuState( FALSE );
  564. }
  565. OnRefresh();
  566. }
  567. else if ( -1 == iResult )
  568. {
  569. cstrMsg = ( _T("You have no enough priority to carry out this operation.") );
  570. MessageBox( cstrMsg );
  571. }
  572. else
  573. {
  574. cstrMsg.Format( _T("Failed to delete "%s".rnMaybe the net is busy, please try again later."), cstrSelName );
  575. MessageBox( cstrMsg );
  576. }
  577. }
  578. else
  579. {
  580. cstrMsg.Format( _T("Failed to delete "%s".rnMaybe the net is busy, please try again later."), cstrSelName );
  581. MessageBox( cstrMsg );
  582. }
  583. delete []pSendBuf;
  584. pSendBuf = NULL;
  585. }
  586. else
  587. {
  588. MessageBox( _T("Failed to allocate memory.") );
  589. }
  590. }
  591. else
  592. {
  593. MessageBox( _T("No availd socket"), NULL, MB_OK | MB_ICONASTERISK );
  594. }
  595. }
  596. }
  597. }
  599. void CMainWnd::OnPreferences() 
  600. {
  601. // TODO: Add your command handler code here
  602. CPerformanceDlg dlg;
  603. dlg.SetAddress( theApp.m_serverInfo.szDBServerIP );
  604. dlg.m_siDBServerPort = theApp.m_serverInfo.siDBServerPort;
  605. dlg.m_siLocalPort = theApp.m_serverInfo.siLocalPort;
  606. dlg.m_dwWaitTime = theApp.m_serverInfo.dwWaitTime;
  607. if ( IDOK == dlg.DoModal() )
  608. {
  609. theApp.m_serverInfo.siDBServerPort = dlg.m_siDBServerPort;
  610. BYTE byField[4];
  611. dlg.GetAddress( byField[0], byField[1],
  612. byField[2], byField[3] );
  613. sprintf( theApp.m_serverInfo.szDBServerIP, "%d.%d.%d.%d",
  614. byField[0], byField[1], byField[2], byField[3] );
  615. theApp.m_serverInfo.siLocalPort = dlg.m_siLocalPort;
  616. theApp.m_serverInfo.dwWaitTime = dlg.m_dwWaitTime;
  617. }
  618. }
  620. void CMainWnd::OnExit() 
  621. {
  622. // TODO: Add your command handler code here
  623. m_page1.OnCancel();
  624. m_page2.OnCancel();
  625. EndDialog( IDCANCEL );
  626. }
  628. void CMainWnd::OnDestroy() 
  629. {
  630. CPropertySheet::OnDestroy();
  632. // TODO: Add your message handler code here
  633. char szSetupFilePath[MAX_PATH+1];
  634. KPIGetExePath( szSetupFilePath, MAX_PATH );
  635. strcat( szSetupFilePath, "setup.ini" );
  636. char szPort[32];
  637. sprintf( szPort, "%d", theApp.m_serverInfo.siLocalPort );
  638. KPIWritePrivateProfileString( _T("Local"), _T("port"), szPort, szSetupFilePath );
  639. sprintf( szPort, "%d", theApp.m_serverInfo.dwWaitTime );
  640. KPIWritePrivateProfileString( _T("Local"), _T("waittime"), szPort, szSetupFilePath );
  641. KPIWritePrivateProfileString( _T("DBServer"), _T("addr"), theApp.m_serverInfo.szDBServerIP, szSetupFilePath );
  642. sprintf( szPort, "%d", theApp.m_serverInfo.siDBServerPort );
  643. KPIWritePrivateProfileString( _T("DBServer"), _T("port"), szPort, szSetupFilePath );
  645. OnLogout();
  647. CClientSocket::ReleaseInstance();
  648. }
  650. void CMainWnd::UpdateMenuState( BOOL bEnable /* = FALSE  */ )
  651. {
  652. theApp.m_bConnected = bEnable;
  653. if ( m_menu.GetSafeHmenu() )
  654. {
  655. m_menu.EnableMenuItem( ID_LOGIN, theApp.m_bConnected );
  656. m_menu.EnableMenuItem( ID_LOGOUT, !theApp.m_bConnected );
  657. m_menu.EnableMenuItem( ID_NEWACCOUNT, !theApp.m_bConnected );
  658. m_menu.EnableMenuItem( ID_DELETEACCOUNT, !theApp.m_bConnected );
  659. m_menu.EnableMenuItem( ID_ADDUSER, !theApp.m_bConnected );
  660. m_menu.EnableMenuItem( ID_DELETEUSER, !theApp.m_bConnected );
  661. m_menu.EnableMenuItem( ID_REFRESH, !theApp.m_bConnected );
  662. m_menu.EnableMenuItem( ID_PREFERENCES, theApp.m_bConnected );
  663. }
  664. }
  666. void CMainWnd::InitClientSocket()
  667. {
  668. char szSetupFilePath[MAX_PATH+1];
  669. KPIGetExePath( szSetupFilePath, MAX_PATH );
  670. strcat( szSetupFilePath, "setup.ini" );
  671. char szPort[32];
  672. KPIGetPrivateProfileString( _T("Local"), _T("port"), _T("8000"),
  673. szPort, 32, szSetupFilePath );
  674. theApp.m_serverInfo.siLocalPort = atoi( szPort );
  675. KPIGetPrivateProfileString( _T("Local"), _T("waittime"), _T("6000"),
  676. szPort, 32, szSetupFilePath );
  677. theApp.m_serverInfo.dwWaitTime = atoi( szPort );
  678. KPIGetPrivateProfileString( _T("DBServer"), _T("addr"), _T(""),
  679. theApp.m_serverInfo.szDBServerIP, 16 * sizeof( TCHAR ), szSetupFilePath );
  680. KPIGetPrivateProfileString( _T("DBServer"), _T("port"), _T(""),
  681. szPort, 32, szSetupFilePath );
  682. theApp.m_serverInfo.siDBServerPort = atoi( szPort );
  684. theApp.m_pClientSocket = CClientSocket::Instance( &( theApp.m_serverInfo.siLocalPort ) );
  685. if ( NULL != theApp.m_pClientSocket )
  686. {
  687. theApp.m_pClientSocket->SetServerAddr( theApp.m_serverInfo.szDBServerIP,
  688. theApp.m_serverInfo.siDBServerPort );
  689. theApp.m_pClientSocket->Connect( theApp.m_serverInfo.szDBServerIP,
  690. theApp.m_serverInfo.siDBServerPort );
  691. theApp.m_pClientSocket->Start();
  692. }
  693. }
  695. void CMainWnd::OnIMEInputLangChangeRequest( WPARAM wParam, LPARAM lParam )
  696. {
  697. ActivateKeyboardLayout( HKL( lParam ), wParam );
  698. }
  700. void CMainWnd::OnMainface() 
  701. {
  702. // TODO: Add your command handler code here
  703. SetActivePage( &m_page1 );
  704. }
  706. void CMainWnd::OnDatabaseuser() 
  707. {
  708. // TODO: Add your command handler code here
  709. SetActivePage( &m_page2 );
  710. if ( theApp.m_bConnected )
  711. {
  712. OnRefresh();
  713. }
  714. }
  716. void CMainWnd::OnRefresh() 
  717. {
  718. // TODO: Add your command handler code here
  719. if ( m_page2.m_dbUserListCtrl.GetSafeHwnd() )
  720. //if ( GetActivePage() == &m_page2 )
  721. {
  722. if ( theApp.m_bConnected )
  723. QueryUserlist( theApp.m_cstrCurUsername, theApp.m_cstrCurPassword );
  724. else
  725. m_page2.m_dbUserListCtrl.DeleteAllItems();
  726. }
  727. if ( GetActivePage() == &m_page1 )
  728. {
  729. m_page1.UpdateUI();
  730. }
  731. }
  733. static DWORD CALLBACK QueryUserlistCallback( DWORD dwPara1, DWORD dwPara2 )
  734. {
  735. DWORD dwRet = 0;
  736. IBYTE* pRecvBuf = ( IBYTE* )dwPara1;
  737. CListCtrl* pListCtrl = ( CListCtrl* )dwPara2;
  738. if ( ( NULL != pRecvBuf )
  739. && ( NULL != pListCtrl ) )
  740. {
  741. if ( pListCtrl->GetSafeHwnd() )
  742. {
  743. CClientSocket::_USERINFO item;
  744. DWORD dwSize = 0;
  745. DWORD dwUserListSize = 0;
  746. memcpy( &dwSize, &pRecvBuf[1], sizeof( DWORD ) );
  747. dwUserListSize = dwSize - sizeof( _DBLOGINSTRUCT ) - sizeof( GUID );
  748. if ( 0 == dwUserListSize % sizeof( CClientSocket::_USERINFO ) )
  749. {
  750. int iItemNum =
  751. dwUserListSize / sizeof( CClientSocket::_USERINFO );
  752. pListCtrl->DeleteAllItems();
  753. for ( int i = 0; i < iItemNum; i++ )
  754. {
  755. memset( &item, 0, sizeof( CClientSocket::_USERINFO ) );
  756. memcpy( &item,
  757. &pRecvBuf[1+sizeof( _DBLOGINSTRUCT )+sizeof( GUID )+i*sizeof( CClientSocket::_USERINFO )],
  758. sizeof( CClientSocket::_USERINFO ) );
  759. pListCtrl->InsertItem( i, item.szUserName );
  760. pListCtrl->SetItemText( i, 1, item.szHostAddr );
  761. CString cstrPriority;
  762. if ( 0 == item.siPriority )
  763. {
  764. cstrPriority = _T("Read");
  765. }
  766. else if ( 1 == item.siPriority )
  767. {
  768. cstrPriority = _T("Write");
  769. }
  770. else if ( 2 == item.siPriority )
  771. {
  772. cstrPriority = _T("Read-Write");
  773. }
  774. else if ( 3 == item.siPriority )
  775. {
  776. cstrPriority = _T("Administrator");
  777. }
  778. pListCtrl->SetItemText( i, 2, cstrPriority );
  779. CString cstrState;
  780. if ( TRUE == item.bLoggedin )
  781. {
  782. cstrState = _T("Connected");
  783. }
  784. else
  785. {
  786. cstrState = _T("Disconnected");
  787. }
  788. pListCtrl->SetItemText( i, 3, cstrState );
  789. pListCtrl->SetItemText( i, 4, item.szLastLoginTime );
  790. pListCtrl->SetItemText( i, 5, item.szLastLogoutTime );
  791. }
  792. }
  793. }
  794. }
  795. return dwRet;
  796. }
  797. BOOL CMainWnd::QueryUserlist(const char *lpszUserName, const char *lpszPassword)
  798. {
  799. BOOL bRet = FALSE;
  800. if ( ( NULL == lpszUserName )
  801. || ( NULL == lpszPassword ) )
  802. {
  803. bRet = FALSE;
  804. return bRet;
  805. }
  806. if ( ( 0 == lpszUserName[0] )
  807. || ( 0 == lpszPassword[0] ) )
  808. {
  809. bRet = FALSE;
  810. return bRet;
  811. }
  812. if ( NULL != theApp.m_pClientSocket )
  813. {
  814. _DBLOGINSTRUCT dbLoginStruct;
  815. memset( &dbLoginStruct, 0, sizeof( _DBLOGINSTRUCT ) );
  816. memcpy( dbLoginStruct.szUserName,
  817. theApp.m_cstrCurUsername,
  818. theApp.m_cstrCurUsername.GetLength() );
  819. memcpy( dbLoginStruct.szPassword,
  820. theApp.m_cstrCurPassword,
  821. theApp.m_cstrCurPassword.GetLength() );
  822. dbLoginStruct.Size = sizeof( _DBLOGINSTRUCT );
  823. dbLoginStruct.Param = DBLOGIN_A_QUERYUSERLIST;
  825. CListCtrl* pListCtrl = &( m_page2.m_dbUserListCtrl );
  826. BOOL bResult =
  827. theApp.m_pClientSocket->RemoteQuery( &dbLoginStruct,
  828. QueryUserlistCallback, ( DWORD )( pListCtrl ), theApp.m_serverInfo.dwWaitTime );
  829. if ( bResult )
  830. {
  831. bRet = TRUE;
  832. }
  833. else
  834. {
  835. MessageBox( _T("Failed to query user list.rnMaybe the net is busy, please try again later."),
  836. NULL, MB_OK | MB_ICONASTERISK );
  837. }
  838. }
  839. else
  840. {
  841. MessageBox( _T("No availd socket"), NULL, MB_OK | MB_ICONASTERISK );
  842. }
  843. return bRet;
  844. }
  846. void CMainWnd::OnTurntoPage2( WPARAM wParam, LPARAM lParam )
  847. {
  848. OnRefresh();
  849. }