Goddess2Dlg.cpp
上传用户:dzyhzl
上传日期:2019-04-29
资源大小:56270k
文件大小:11k
源码类别:

模拟服务器

开发平台:

C/C++

  1. // Goddess2Dlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "Goddess2.h"
  5. #include "Goddess2Dlg.h"
  6. #include "winsock2.h"
  7. #include <string>
  8. #include "tstring.h"
  9. #include "IniFile.h"
  10. #include "Utils.h"
  11. #include "src/IDBRoleServer.h"
  12. #include "src/KSG_RoleProcess.h"
  13. using namespace std;
  14. using namespace OnlineGameLib::Win32;
  15. #ifdef _DEBUG
  16. #define new DEBUG_NEW
  17. #undef THIS_FILE
  18. static char THIS_FILE[] = __FILE__;
  19. #endif
  20. bool CreateDatabaseEngine(HWND hwnd);
  21. void DestroyDatabaseEngine();
  22. void LoadSetting();
  23. void SaveSetting();
  24. static size_t g_nMaxRoleCount = 3;
  25. static unsigned short g_uServerPort = 5001;
  26. static size_t g_BackupSleepTime = 1;  //备份线程自动挂起时间
  27. static bool g_IsBackupSuspend = false;  //备份线程是否手工挂起
  28. int g_nKSG_RoleProcess_InitFlag = false;
  29. /////////////////////////////////////////////////////////////////////////////
  30. // CAboutDlg dialog used for App About
  31. class CAboutDlg : public CDialog
  32. {
  33. public:
  34. CAboutDlg();
  35. // Dialog Data
  36. //{{AFX_DATA(CAboutDlg)
  37. enum { IDD = IDD_ABOUTBOX };
  38. //}}AFX_DATA
  39. // ClassWizard generated virtual function overrides
  40. //{{AFX_VIRTUAL(CAboutDlg)
  41. protected:
  42. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  43. //}}AFX_VIRTUAL
  44. // Implementation
  45. protected:
  46. //{{AFX_MSG(CAboutDlg)
  47. //}}AFX_MSG
  48. DECLARE_MESSAGE_MAP()
  49. };
  50. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  51. {
  52. //{{AFX_DATA_INIT(CAboutDlg)
  53. //}}AFX_DATA_INIT
  54. }
  55. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  56. {
  57. CDialog::DoDataExchange(pDX);
  58. //{{AFX_DATA_MAP(CAboutDlg)
  59. //}}AFX_DATA_MAP
  60. }
  61. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  62. //{{AFX_MSG_MAP(CAboutDlg)
  63. // No message handlers
  64. //}}AFX_MSG_MAP
  65. END_MESSAGE_MAP()
  66. /////////////////////////////////////////////////////////////////////////////
  67. // CGoddess2Dlg dialog
  68. CGoddess2Dlg::CGoddess2Dlg(CWnd* pParent /*=NULL*/)
  69. : CDialog(CGoddess2Dlg::IDD, pParent)
  70. {
  71. //{{AFX_DATA_INIT(CGoddess2Dlg)
  72. m_OpenPort = 0;
  73. //}}AFX_DATA_INIT
  74. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  75. m_hIcon = AfxGetApp()->LoadIcon(IDI_GIDDESS2);
  76. }
  77. void CGoddess2Dlg::DoDataExchange(CDataExchange* pDX)
  78. {
  79. CDialog::DoDataExchange(pDX);
  80. //{{AFX_DATA_MAP(CGoddess2Dlg)
  81. DDX_Control(pDX, IDC_EDIT_OPEN_PORT, m_OpenPortCtl);
  82. DDX_Control(pDX, IDC_BUTTON_START_DB_SERVICE, m_StartDBService);
  83. DDX_Control(pDX, IDC_LIST_CLIENT, m_ClientList);
  84. DDX_Text(pDX, IDC_EDIT_OPEN_PORT, m_OpenPort);
  85. DDV_MinMaxUInt(pDX, m_OpenPort, 0, 65535);
  86. //}}AFX_DATA_MAP
  87. }
  88. BEGIN_MESSAGE_MAP(CGoddess2Dlg, CDialog)
  89. //{{AFX_MSG_MAP(CGoddess2Dlg)
  90. ON_WM_SYSCOMMAND()
  91. ON_WM_PAINT()
  92. ON_WM_QUERYDRAGICON()
  93. ON_BN_CLICKED(IDOK, OnOk)
  94. ON_WM_CLOSE()
  95.     ON_MESSAGE(WM_ADD_CLIENT, OnAddClient) 
  96.     ON_MESSAGE(WM_DEL_CLIENT, OnDelClient) 
  97. ON_BN_CLICKED(IDC_BUTTON_START_DB_SERVICE, OnButtonStartDbService)
  98. //}}AFX_MSG_MAP
  99. END_MESSAGE_MAP()
  100. /////////////////////////////////////////////////////////////////////////////
  101. // CGoddess2Dlg message handlers
  102. BOOL CGoddess2Dlg::OnInitDialog()
  103. {
  104. CDialog::OnInitDialog();
  105. // Add "About..." menu item to system menu.
  106. // IDM_ABOUTBOX must be in the system command range.
  107. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  108. ASSERT(IDM_ABOUTBOX < 0xF000);
  109. CMenu* pSysMenu = GetSystemMenu(FALSE);
  110. if (pSysMenu != NULL)
  111. {
  112. CString strAboutMenu;
  113. strAboutMenu.LoadString(IDS_ABOUTBOX);
  114. if (!strAboutMenu.IsEmpty())
  115. {
  116. pSysMenu->AppendMenu(MF_SEPARATOR);
  117. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  118. }
  119. }
  120. // Set the icon for this dialog.  The framework does this automatically
  121. //  when the application's main window is not a dialog
  122. SetIcon(m_hIcon, TRUE); // Set big icon
  123. SetIcon(m_hIcon, FALSE); // Set small icon
  124. // TODO: Add extra initialization here
  125.     LoadSetting();
  126. m_OpenPort = g_uServerPort;
  127.     UpdateData(FALSE);
  128. return TRUE;  // return TRUE  unless you set the focus to a control
  129. }
  130. void CGoddess2Dlg::OnSysCommand(UINT nID, LPARAM lParam)
  131. {
  132. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  133. {
  134. CAboutDlg dlgAbout;
  135. dlgAbout.DoModal();
  136. }
  137. else
  138. {
  139. CDialog::OnSysCommand(nID, lParam);
  140. }
  141. }
  142. // If you add a minimize button to your dialog, you will need the code below
  143. //  to draw the icon.  For MFC applications using the document/view model,
  144. //  this is automatically done for you by the framework.
  145. void CGoddess2Dlg::OnPaint() 
  146. {
  147. if (IsIconic())
  148. {
  149. CPaintDC dc(this); // device context for painting
  150. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  151. // Center icon in client rectangle
  152. int cxIcon = GetSystemMetrics(SM_CXICON);
  153. int cyIcon = GetSystemMetrics(SM_CYICON);
  154. CRect rect;
  155. GetClientRect(&rect);
  156. int x = (rect.Width() - cxIcon + 1) / 2;
  157. int y = (rect.Height() - cyIcon + 1) / 2;
  158. // Draw the icon
  159. dc.DrawIcon(x, y, m_hIcon);
  160. }
  161. else
  162. {
  163. CDialog::OnPaint();
  164. }
  165. }
  166. // The system calls this to obtain the cursor to display while the user drags
  167. //  the minimized window.
  168. HCURSOR CGoddess2Dlg::OnQueryDragIcon()
  169. {
  170. return (HCURSOR) m_hIcon;
  171. }
  172. void CGoddess2Dlg::OnOk() 
  173. {
  174.     SendMessage(WM_CLOSE, 0, 0);
  175. }
  176. void CGoddess2Dlg::OnClose() 
  177. {
  178.     UpdateData(TRUE);
  179. // TODO: Add your message handler code here and/or call default
  180.     int nRetCode = ::MessageBox(
  181.         GetSafeHwnd(), 
  182.         "Are you sure quit?", 
  183.         "Info", 
  184.         MB_YESNO | MB_ICONQUESTION
  185.     ); 
  186.     if (IDYES == nRetCode)
  187.     {
  188.         if (g_nKSG_RoleProcess_InitFlag)
  189.         {
  190.             KSG_RoleProcess_UnInit();
  191.             g_nKSG_RoleProcess_InitFlag = false;
  192.         }
  193.         DestroyDatabaseEngine();
  194.         SaveSetting();
  195.      CDialog::OnClose();
  196.     }
  197. }
  198. void CGoddess2Dlg::OnButtonStartDbService() 
  199. {
  200.     int nResult = false;
  201.     int nRetCode = false;
  202. // TODO: Add your control notification handler code here
  203.     UpdateData(TRUE);
  204.     nRetCode = CreateDatabaseEngine(m_hWnd);
  205.     if (!nRetCode)
  206.         goto Exit0;
  207.     nRetCode = KSG_RoleProcess_Init(m_OpenPort, m_hWnd);  // will call in OnStartDBService
  208.     if (!nRetCode)
  209.     {
  210. ::MessageBox(m_hWnd, "Role Process Init Failed!", "Error", MB_OK | MB_ICONERROR);
  211.         goto Exit0;
  212.     }
  213.     g_nKSG_RoleProcess_InitFlag = true;
  214.     nRetCode = KSG_RoleProcess_Start();
  215.     if (!nRetCode)
  216.     {   
  217. ::MessageBox(m_hWnd, "Role Process Start Fail!", "Error", MB_OK | MB_ICONERROR);
  218.         goto Exit0;
  219.     }
  220.     ::EnableWindow(m_StartDBService.m_hWnd, false);
  221.     ::EnableWindow(m_OpenPortCtl.m_hWnd, false);
  222.     SetWindowText("Goddess2 [Enable]");
  223.     nResult = true;
  224. Exit0:
  225.     if (!nResult)
  226.     {
  227.         if (g_nKSG_RoleProcess_InitFlag)
  228.         {
  229.             KSG_RoleProcess_UnInit();
  230.             g_nKSG_RoleProcess_InitFlag = false;
  231.         }
  232.     }
  233.     
  234. }
  235. LRESULT CGoddess2Dlg::OnAddClient(WPARAM wParam, LPARAM lParam)
  236. {
  237.     int addr_from_len = wParam;
  238.     struct sockaddr_in *paddr_from = (struct sockaddr_in *)lParam;
  239.     char szClientInfo[256];
  240.     char *pszRetAddress = NULL;
  241.     char szAddress[256];
  242.     if (!addr_from_len)
  243.         return 0;
  244.     if (!paddr_from)
  245.         return 0;
  246.     pszRetAddress = inet_ntoa(paddr_from->sin_addr);
  247.     if (!pszRetAddress)
  248.         return 0;
  249.     strncpy(szAddress, pszRetAddress, 256);
  250.     szAddress[256 - 1] = '';
  251.     sprintf(szClientInfo, "%s:%u", szAddress, (unsigned)ntohs(paddr_from->sin_port));
  252.     
  253.     m_ClientList.AddString(szClientInfo);
  254.     
  255.     return 0;
  256. }
  257. LRESULT CGoddess2Dlg::OnDelClient(WPARAM wParam, LPARAM lParam)
  258. {
  259.     int addr_from_len = wParam;
  260.     struct sockaddr_in *paddr_from = (struct sockaddr_in *)lParam;
  261.     char szClientInfo[256];
  262.     char *pszRetAddress = NULL;
  263.     char szAddress[256];
  264.     if (!addr_from_len)
  265.         return 0;
  266.     if (!paddr_from)
  267.         return 0;
  268.     pszRetAddress = inet_ntoa(paddr_from->sin_addr);
  269.     if (!pszRetAddress)
  270.         return 0;
  271.     strncpy(szAddress, pszRetAddress, 256);
  272.     szAddress[256 - 1] = '';
  273.     sprintf(szClientInfo, "%s:%u", szAddress, (unsigned)ntohs(paddr_from->sin_port));
  274.     int nRetIdx = m_ClientList.FindStringExact(-1, szClientInfo);
  275.     if (nRetIdx == LB_ERR)
  276.        return 0;
  277.     
  278.     m_ClientList.DeleteString(nRetIdx);
  279.     
  280.     return 0;
  281. }
  282. bool CreateDatabaseEngine( HWND hwnd )
  283. {
  284. if ( !InitDBInterface( g_nMaxRoleCount ) )
  285. {
  286. ::MessageBox( NULL, "Setup dbserver is failed!", "Warning", MB_OK | MB_ICONSTOP );
  287. return false;
  288. }
  289. // StartBackupTimer(g_BackupSleepTime); //开始运行备份线程
  290. // ::SetWindowText(GetDlgItem( hwnd, IDC_LAB_BACKUP_STATUS ),
  291. // "Backup thread status: Running");
  292. // /*
  293. //  * Open this server to client
  294. //  */
  295. // pfnCreateServerInterface pFactroyFun = ( pfnCreateServerInterface )( g_theHeavenLibrary.GetProcAddress( "CreateInterface" ) );
  296. //
  297. // IServerFactory *pServerFactory = NULL;
  298. //
  299. // if ( pFactroyFun && SUCCEEDED( pFactroyFun( IID_IServerFactory, reinterpret_cast< void ** >( &pServerFactory ) ) ) )
  300. // {
  301. // pServerFactory->SetEnvironment( 10, 10, 20, 1024 * 1024 );
  302. //
  303. // pServerFactory->CreateServerInterface( IID_IIOCPServer, reinterpret_cast< void ** >( &g_pServer ) );
  304. //
  305. // pServerFactory->Release();
  306. // }
  307. //
  308. // if ( !g_pServer )
  309. // {
  310. // ::MessageBox( NULL, "Initialization failed! Don't find a correct heaven.dll", "Warning", MB_OK | MB_ICONSTOP );
  311. //
  312. // return false;
  313. // }
  314. //
  315. // g_pServer->Startup();
  316. //
  317. // g_pServer->RegisterMsgFilter( ( void * )hwnd, ServerEventNotify );
  318. //
  319. // if ( FAILED( g_pServer->OpenService( INADDR_ANY, g_nServerPort ) ) )
  320. // {
  321. // return false;
  322. // }
  323. //
  324. // DWORD dwThreadID = 0L;
  325. //
  326. // IServer *pCloneServer = NULL;
  327. // g_pServer->QueryInterface( IID_IIOCPServer, ( void ** )&pCloneServer );
  328. //
  329. // g_hThread = ::CreateThread( NULL, 0, ThreadProcess, ( void * )pCloneServer, 0, &dwThreadID );
  330. //
  331. // if ( !g_hThread )
  332. // {
  333. // return false;
  334. // }
  335. //
  336. // CClientNode::Start( g_pServer );
  337. //
  338. // g_hStartWorkEvent.Set();
  339. return true;
  340. }
  341. void DestroyDatabaseEngine()
  342. {
  343. // CClientNode::End();
  344. //
  345. // g_hQuitEvent.Set();
  346. //
  347. // DWORD result = ::WaitForSingleObject( g_hThread, 50000 );
  348. //
  349. // if ( result == WAIT_TIMEOUT )
  350. // {
  351. // ::TerminateThread( g_hThread, ( DWORD )( -1 ) );
  352. // }
  353. //
  354. // SAFE_CLOSEHANDLE( g_hThread );
  355. //
  356. // if ( g_pServer )
  357. // {
  358. // g_pServer->CloseService();
  359. // g_pServer = NULL;
  360. // }
  361. //
  362. // SAFE_RELEASE( g_pServer );
  363. ReleaseDBInterface();
  364. }
  365. void LoadSetting()
  366. {
  367. CIniFile theConfigFile;
  368. _tstring sAppPath = GetAppFullPath( NULL );
  369. _tstring sConfigFile;
  370. sConfigFile = sAppPath + "Goddess.cfg";
  371. theConfigFile.SetFile( sConfigFile.c_str() );
  372. g_uServerPort = theConfigFile.ReadInteger( "Setting", "Port", 5001 );
  373. g_nMaxRoleCount = theConfigFile.ReadInteger( "Setting", "MaxRoleCount", 3 );
  374. g_BackupSleepTime = theConfigFile.ReadInteger( "Setting", "BackupSleepTime", 3 );
  375. }
  376. void SaveSetting()
  377. {
  378. CIniFile theConfigFile;
  379. _tstring sAppPath = GetAppFullPath( NULL );
  380. _tstring sConfigFile;
  381. sConfigFile = sAppPath + "Goddess.cfg";
  382. theConfigFile.SetFile( sConfigFile.c_str() );
  383. theConfigFile.WriteInteger( "Setting", "Port", g_uServerPort );
  384. theConfigFile.WriteInteger(  "Setting", "MaxRoleCount", g_nMaxRoleCount );
  385. theConfigFile.WriteInteger(  "Setting", "BackupSleepTime", g_BackupSleepTime );
  386. }