开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
Goddess.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：15k
源码类别：

模拟服务器

开发平台：

C/C++

Goddess.cpp：源码内容
  1. // Goddess.cpp : Defines the entry point for the application.
  2. //
  4. #include "stdafx.h"
  5. #include "resource.h"
  7. #include <objbase.h>
  8. #include <initguid.h>
  9. #include <winsock2.h>
  10. #include <process.h>
  12. #include "kprotocoldef.h"
  14. #include "IDBRoleServer.h"
  16. #include "console.h"
  17. #include "Event.h"
  18. #include "macro.h"
  19. #include "Inifile.h"
  20. #include "Utils.h"
  21. #include "tstring.h"
  22. #include "Library.h"
  23. #include "Buffer.h"
  24. #include "CriticalSection.h"
  26. #include "IServer.h"
  27. #include "HeavenInterface.h"
  29. #include "ClientNode.h"
  31. #include <iostream>
  32. #include <strstream>
  33. #include <string>
  34. #include <list>
  36. #include "FilterTextLib.h"
  37. #include "RoleNameFilter.h"
  40. using namespace std;
  42. using OnlineGameLib::Win32::Console::clrscr;
  43. using OnlineGameLib::Win32::Console::setcolor;
  44. using OnlineGameLib::Win32::CEvent;
  45. using OnlineGameLib::Win32::CIniFile;
  46. using OnlineGameLib::Win32::GetAppFullPath;
  47. using OnlineGameLib::Win32::_tstring;
  48. using OnlineGameLib::Win32::CLibrary;
  49. using OnlineGameLib::Win32::ToString;
  50. using OnlineGameLib::Win32::CPackager;
  51. using OnlineGameLib::Win32::CBuffer;
  52. using OnlineGameLib::Win32::CCriticalSection;
  54. /*
  55.  * User data
  56.  */
  58. static CLibrary g_theHeavenLibrary( "heaven.dll" );
  60. static CEvent g_hQuitEvent( NULL, true, false );
  61. static CEvent g_hStartWorkEvent( NULL, true, false );
  63. static size_t g_nMaxRoleCount = 3;
  65. static unsigned short g_nServerPort = 5001;
  66. static IServer *g_pServer = NULL;
  68. static HANDLE g_hThread = NULL;
  70. static CCriticalSection g_csPlayer;
  72. typedef map< size_t, CClientNode * > stdPlayerMap;
  73. static  stdPlayerMap g_thePlayer;
  75. static CPackager g_theRecv;
  76. static CPackager g_theSend;
  78. UINT g_nDBEngineLoop = 0;
  79. UINT g_nServiceLoop = 0;
  81. extern HWND hListOutput;
  83. #define WM_CREATE_ENGINE WM_USER + 0x100
  84. #define WM_ADD_CLIENT WM_USER + 0x200
  85. #define WM_DEL_CLIENT WM_USER + 0x300
  88. CFilterTextLib g_libFilterText;
  89. CRoleNameFilter g_fltRoleName;
  91. static size_t g_BackupSleepTime = 1;  //备份线程自动挂起时间
  92. static bool g_IsBackupSuspend = false;  //备份线程是否手工挂起
  94. /*
  95.  * Helper function
  96.  */
  97. BOOL CALLBACK MainWndProc( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam );
  99. bool CreateDatabaseEngine( HWND hwnd );
  100. void DestroyDatabaseEngine();
  102. void LoadSetting();
  103. void SaveSetting();
  105. //==========监视的线程
  106. DWORD WINAPI MonitorProc(LPVOID lpParameter) {
  107. while(true) {
  108. Sleep(60 * 1000); //1次
  109. }
  110. return 0;
  111. }
  114. /*
  115.  * Callback function
  116.  */
  117. typedef HRESULT ( __stdcall * pfnCreateServerInterface )(
  118. REFIID riid,
  119. void **ppv
  120. );
  122. void __stdcall ServerEventNotify(
  123. LPVOID lpParam,
  124. const unsigned long &ulnID,
  125. const unsigned long &ulnEventType )
  126. {
  127. HWND hwnd = ( HWND )lpParam;
  129. switch( ulnEventType )
  130. {
  131. case enumClientConnectCreate:
  132. {
  133. CClientNode *pNode = CClientNode::AddNode( g_pServer, ulnID );
  135. if ( pNode )
  136. {
  137. CCriticalSection::Owner lock( g_csPlayer );
  139. g_thePlayer.insert( stdPlayerMap::value_type( ulnID, pNode ) );
  141. ::PostMessage( hwnd, WM_ADD_CLIENT, ulnID, 0 );
  142. }
  143. }
  144. break;
  146. case enumClientConnectClose:
  147. {
  148. {
  149. CCriticalSection::Owner lock( g_csPlayer );
  151. stdPlayerMap::iterator it;
  152. if ( g_thePlayer.end() != ( it = g_thePlayer.find( ulnID ) ) )
  153. {
  154. g_thePlayer.erase( ulnID );
  156. ::PostMessage( hwnd, WM_DEL_CLIENT, ulnID, 0 );
  157. }
  158. }
  160. CClientNode::DelNode( ulnID );
  161. }
  162. break;
  163. }
  164. }
  166. DWORD WINAPI ThreadProcess( void *pParam )
  167. {
  168. IServer *pServer = ( IServer * )pParam;
  170. ASSERT( pServer );
  172. g_hStartWorkEvent.Wait();
  174. //try
  175. //{
  176. while ( !g_hQuitEvent.Wait( 0 ) )
  177. {
  178. {
  179. CCriticalSection::Owner lock( g_csPlayer );
  181. stdPlayerMap::iterator it;
  182. for ( it = g_thePlayer.begin(); it != g_thePlayer.end(); it ++ )
  183. {
  184. UINT index = ( *it ).first;
  185. CClientNode *pClientNode = ( *it ).second;
  187. size_t datalength = 0;
  188. const char *pData = ( const char * )pServer->GetPackFromClient( index, datalength );
  190. while ( pClientNode && pData && datalength )
  191. {
  192. pClientNode->AppendData( pData, datalength );
  194. pData = ( const char * )pServer->GetPackFromClient( index, datalength );
  195. }
  196. }
  197. }
  199. if ( ++ g_nServiceLoop & 0x80000000 )
  200. {
  201. g_nServiceLoop = 0;
  202. }
  204. if ( g_nServiceLoop & 0x1 )
  205. {
  206. ::Sleep( 1 );
  207. }
  208. }
  209. //}
  210. //catch(...)
  211. //{
  212. // ::MessageBox( NULL, "ThreadProcess was error!", "Warning", MB_OK );
  213. //}
  215. return 0L;
  216. }
  218. int APIENTRY WinMain(HINSTANCE hInstance,
  219.                      HINSTANCE hPrevInstance,
  220.                      LPSTR     lpCmdLine,
  221.                      int       nCmdShow)
  222. {
  223. //role name filter
  224. if (!g_libFilterText.Initialize()
  225. || !g_fltRoleName.Initialize())
  226. {
  227. MessageBox(NULL, "text filter's initing has failed", "error", MB_OK|MB_ICONERROR);
  228. return -1;
  229. }
  233. HWND hwndMain = ::CreateDialog( hInstance, 
  234. MAKEINTRESOURCE( IDD_DLG_GODDESS ),
  235. NULL, 
  236. (DLGPROC)MainWndProc );
  238. ASSERT( hwndMain );
  240. ::ShowWindow( hwndMain, SW_SHOWNORMAL );
  241. ::UpdateWindow( hwndMain );
  243. hListOutput = ::GetDlgItem( hwndMain, IDC_OUTPUT );//输出list
  245. BOOL bRet;
  246. MSG msg;
  248. while ( ( bRet = ::GetMessage( &msg, NULL, 0, 0 ) ) != 0 )
  249. {
  250. if ( !IsWindow( hwndMain ) || !IsDialogMessage( hwndMain, &msg ) )
  251. {
  252. ::TranslateMessage( &msg );
  253. ::DispatchMessage( &msg );
  254. }
  255. }
  258. //rolename filter
  259. g_fltRoleName.Uninitialize();
  260. g_libFilterText.Uninitialize();
  263. return msg.wParam;
  264. }
  266. /*
  267.  *
  268.  * MainWndProc() - Main window callback procedure.
  269.  *  
  270.  */
  272. BOOL CALLBACK MainWndProc( HWND hwnd, 
  273.                        UINT msg,
  274.                        WPARAM wParam,
  275.                        LPARAM lParam )
  276. {
  277. static const UINT g_nIDEvent = 0x100;
  278. static const UINT g_nElapse = 500;
  280. static const char g_szBaseInfo[] = "....................";
  281. static const int g_nDots = sizeof( g_szBaseInfo ) - 1;
  283. static UINT g_nLastServiceLoop = g_nServiceLoop;
  284. static int g_nServiceStep = g_nDots;
  285. static UINT g_nLastDBEngineLoop= g_nDBEngineLoop;
  286. static int g_nDBEngineStep = g_nDots;
  288. BOOL bTranslated = TRUE;
  290. switch ( msg )
  291. {
  292. case WM_TIMER:
  293. {
  294. if ( g_nLastServiceLoop != g_nServiceLoop )
  295. {
  296. g_nServiceStep = ( -- g_nServiceStep > 0 ) ? g_nServiceStep : g_nDots;
  298. ::SetDlgItemText( hwnd, IDC_STATIC_SERVICE, ( const char * )g_szBaseInfo + g_nServiceStep );
  300. g_nLastServiceLoop = g_nServiceLoop;
  301. }
  303. if ( g_nLastDBEngineLoop != g_nDBEngineLoop )
  304. {
  305. g_nDBEngineStep = ( -- g_nDBEngineStep > 0 ) ? g_nDBEngineStep : g_nDots;
  307. ::SetDlgItemText( hwnd, IDC_STATIC_DATABASE, ( const char * )g_szBaseInfo + g_nDBEngineStep );
  309. g_nLastDBEngineLoop = g_nDBEngineLoop;
  310. }
  312. if(IsBackupThreadWorking())
  313. {
  314. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_SUS_RES ), TRUE );
  315. ::SetWindowText(GetDlgItem( hwnd, IDC_LAB_BACKUP_STATUS ),
  316. "Backup thread status: Sleeping");
  317. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_MANUAL ), TRUE );
  318. if(IsBackupWorking())
  319. {
  320. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_SUS_RES ), FALSE );
  321. ::SetWindowText(GetDlgItem( hwnd, IDC_LAB_BACKUP_STATUS ),
  322. "Backup thread status: Running");
  323. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_MANUAL ), FALSE );
  324. }
  325. else if (g_IsBackupSuspend)
  326. {
  327. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_SUS_RES ), TRUE );
  328. ::SetWindowText(GetDlgItem( hwnd, IDC_LAB_BACKUP_STATUS ),
  329. "Backup thread status: Suspended");
  330. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_MANUAL ), TRUE );
  331. }
  333. }
  334. else
  335. {
  336. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_SUS_RES ), FALSE );
  337. ::SetWindowText(GetDlgItem( hwnd, IDC_LAB_BACKUP_STATUS ),
  338. "Backup thread status: Stop");
  339. ::EnableWindow( GetDlgItem( hwnd, IDC_BTN_BACKUP_MANUAL ), FALSE );
  340. }
  341. }
  342. break;
  344. case WM_INITDIALOG:
  346. LoadSetting();
  348. ::SetDlgItemInt( hwnd, IDC_EDIT_PORT, g_nServerPort, FALSE );
  349. ::SetDlgItemInt( hwnd, IDC_EDIT_MAXNUM_ROLE, g_nMaxRoleCount, FALSE );
  350. ::SetDlgItemInt( hwnd, IDC_EDIT_BACKUP_SLEEP_TIME, g_BackupSleepTime, FALSE );
  352. ::SetTimer( hwnd, g_nIDEvent, g_nElapse, NULL );
  354. return TRUE;
  355. break;
  357. case WM_CLOSE:
  358. {
  359. if ( IDYES == ::MessageBox( hwnd, "Are you sure quit?", "Info", MB_YESNO | MB_ICONQUESTION ) )
  360. {
  362. DestroyDatabaseEngine();
  364. ::KillTimer( hwnd, g_nIDEvent );
  366. g_nServerPort = ::GetDlgItemInt( hwnd, IDC_EDIT_PORT, &bTranslated, FALSE );
  367. g_nMaxRoleCount = ::GetDlgItemInt( hwnd, IDC_EDIT_MAXNUM_ROLE, &bTranslated, FALSE );
  368. g_BackupSleepTime = ::GetDlgItemInt( hwnd, IDC_EDIT_BACKUP_SLEEP_TIME, &bTranslated, FALSE );
  370. SaveSetting();
  372. ::DestroyWindow( hwnd );
  373. ::PostQuitMessage( 0 );
  374. }
  375. }
  377. return TRUE;
  378. break;
  380. case WM_COMMAND:
  382. switch ( wParam )
  383. {
  384. case IDOK:
  386. ::EnableWindow( GetDlgItem( hwnd, IDOK ), FALSE );
  387. ::EnableWindow( GetDlgItem( hwnd, IDC_EDIT_PORT ), FALSE );
  388. ::EnableWindow( GetDlgItem( hwnd, IDC_EDIT_MAXNUM_ROLE ), FALSE );
  389. ::EnableWindow( GetDlgItem( hwnd, IDC_EDIT_BACKUP_SLEEP_TIME ), FALSE );
  391. ::SetWindowText( hwnd, "Goddess - [Start up...]" );
  393. ::PostMessage( hwnd, WM_CREATE_ENGINE, 0L, 0L );
  395. break;
  397. case IDCANCEL:
  399. ::PostMessage( hwnd, WM_CLOSE, 0L, 0L );
  400. break;
  402. case IDC_BTN_BACKUP_MANUAL:
  404. if(!IsBackupThreadWorking())
  405. MessageBox(hwnd, "Backup thread is not started yet.",
  406. "Information",MB_OK | MB_ICONEXCLAMATION);
  407. if(!DoManualBackup())
  408. MessageBox(hwnd, "Unable to start manual backup.nPlease try later.",
  409. "Information",MB_OK | MB_ICONEXCLAMATION);
  411. break;
  412. case IDC_RESTART:
  413. CClientNode::End();
  414. ReleaseDBInterface();
  415. InitDBInterface(g_nMaxRoleCount);
  416. CClientNode::Start( g_pServer );
  417. break;
  418. case IDC_BTN_BACKUP_SUS_RES:
  420. if(!IsBackupThreadWorking())
  421. MessageBox(hwnd, "Backup thread is not started yet.",
  422. "Information",MB_OK | MB_ICONEXCLAMATION);
  423. else if(g_IsBackupSuspend)
  424. {
  425. if(ResumeBackupTimer())
  426. {
  427. g_IsBackupSuspend = false;
  428. ::SetWindowText(GetDlgItem( hwnd, IDC_BTN_BACKUP_SUS_RES ),
  429. "Suspend");
  430. }
  431. else
  432. MessageBox(hwnd, "Unable to resume backup thread.nPlease try later.",
  433. "Information",MB_OK | MB_ICONEXCLAMATION);
  434. }
  435. else
  436. {
  437. if(SuspendBackupTimer())
  438. {
  439. g_IsBackupSuspend = true;
  440. ::SetWindowText(GetDlgItem( hwnd, IDC_BTN_BACKUP_SUS_RES ),
  441. "Resume");
  442. }
  443. else
  444. MessageBox(hwnd, "Unable to suspend backup thread.nPlease try later.",
  445. "Information",MB_OK | MB_ICONEXCLAMATION);
  446. }
  447. break;
  448. }
  450. break;
  452. case WM_CREATE_ENGINE:
  453. {
  454. g_nServerPort = ::GetDlgItemInt( hwnd, IDC_EDIT_PORT, &bTranslated, FALSE );
  455. g_nMaxRoleCount = ::GetDlgItemInt( hwnd, IDC_EDIT_MAXNUM_ROLE, &bTranslated, FALSE );
  456. g_BackupSleepTime = ::GetDlgItemInt( hwnd, IDC_EDIT_BACKUP_SLEEP_TIME, &bTranslated, FALSE );
  458. CreateDatabaseEngine( hwnd );
  459. /* DWORD dwThreadId, dwThrdParam = 1;
  460. HANDLE hMonitorThread = CreateThread(
  461. NULL, // no security attributes 
  462. 0, // use default stack size  
  463. MonitorProc, // thread function 
  464. &dwThrdParam, // argument to thread function 
  465. 0, // use default creation flags 
  466. &dwThreadId); // returns the thread identifier 
  467. */
  468. ::SetWindowText( hwnd, "Goddess - [Enable]" );
  469. }
  470. break;
  472. case WM_ADD_CLIENT:
  473. {
  474. const char *pText = g_pServer->GetClientInfo( wParam );
  476. if ( pText && pText[0] )
  477. {
  478. HWND hCtrl = ::GetDlgItem( hwnd, IDC_LIST_CLIENT );
  480. if ( hCtrl && ::IsWindow( hCtrl ) )
  481. {
  482. int nIndex = ::SendMessage( hCtrl, LB_ADDSTRING, 0, ( LPARAM )pText );
  484. if ( LB_ERR != nIndex )
  485. {
  486. ::SendMessage( hCtrl, LB_SETITEMDATA, nIndex, wParam );
  487. }
  488. }
  489. }
  490. }
  491. break;
  493. case WM_DEL_CLIENT:
  494. {
  495. HWND hCtrl = ::GetDlgItem( hwnd, IDC_LIST_CLIENT );
  497. if ( hCtrl && ::IsWindow( hCtrl ) )
  498. {
  499. int nCount = ::SendMessage( hCtrl, LB_GETCOUNT, 0, 0 );
  501. for ( int i=0; i<nCount, LB_ERR != nCount; i++ )
  502. {
  503. UINT nSearchID = 0;
  505. if ( wParam == ( nSearchID = ::SendMessage( hCtrl, LB_GETITEMDATA, i, 0 ) ) )
  506. {
  507. ::SendMessage( hCtrl, LB_DELETESTRING, i, 0 );
  509. return TRUE;
  510. }
  511. }
  512. }
  513. }
  514. break;
  516. default:
  517. break;
  518. }
  520. /*
  521.  * Clean up any unused messages by calling DefWindowProc
  522.  */
  523. return FALSE;
  524. }
  526. bool CreateDatabaseEngine( HWND hwnd )
  527. {
  528. if ( !InitDBInterface( g_nMaxRoleCount ) )
  529. {
  530. ::MessageBox( NULL, "Setup dbserver is failed!", "Warning", MB_OK | MB_ICONSTOP );
  532. return false;
  533. }
  535. StartBackupTimer(g_BackupSleepTime); //开始运行备份线程
  536. ::SetWindowText(GetDlgItem( hwnd, IDC_LAB_BACKUP_STATUS ),
  537. "Backup thread status: Running");
  538. /*
  539.  * Open this server to client
  540.  */
  541. pfnCreateServerInterface pFactroyFun = ( pfnCreateServerInterface )( g_theHeavenLibrary.GetProcAddress( "CreateInterface" ) );
  543. IServerFactory *pServerFactory = NULL;
  545. if ( pFactroyFun && SUCCEEDED( pFactroyFun( IID_IServerFactory, reinterpret_cast< void ** >( &pServerFactory ) ) ) )
  546. {
  547. pServerFactory->SetEnvironment( 10, 10, 20, 4 * 1024 * 1024 );
  549. pServerFactory->CreateServerInterface( IID_IIOCPServer, reinterpret_cast< void ** >( &g_pServer ) );
  551. pServerFactory->Release();
  552. }
  554. if ( !g_pServer )
  555. {
  556. ::MessageBox( NULL, "Initialization failed! Don't find a correct heaven.dll", "Warning", MB_OK | MB_ICONSTOP );
  558. return false;
  559. }
  561. g_pServer->Startup();
  563. g_pServer->RegisterMsgFilter( ( void * )hwnd, ServerEventNotify );
  565. if ( FAILED( g_pServer->OpenService( INADDR_ANY, g_nServerPort ) ) )
  566. {
  567. return false;
  568. }
  570. DWORD dwThreadID = 0L;
  572. IServer *pCloneServer = NULL;
  573. g_pServer->QueryInterface( IID_IIOCPServer, ( void ** )&pCloneServer );
  575. g_hThread = ::CreateThread( NULL, 0, ThreadProcess, ( void * )pCloneServer, 0, &dwThreadID );
  577. if ( !g_hThread )
  578. {
  579. return false;
  580. }
  582. CClientNode::Start( g_pServer );
  584. g_hStartWorkEvent.Set();
  586. return true;
  587. }
  589. void DestroyDatabaseEngine()
  590. {
  591. CClientNode::End();
  593. g_hQuitEvent.Set();
  595. DWORD result = ::WaitForSingleObject( g_hThread, 50000 );
  597. if ( result == WAIT_TIMEOUT )
  598. {
  599. ::TerminateThread( g_hThread, ( DWORD )( -1 ) );
  600. }
  602. SAFE_CLOSEHANDLE( g_hThread );
  604. if ( g_pServer )
  605. {
  606. g_pServer->CloseService();
  607. g_pServer = NULL;
  608. }
  610. SAFE_RELEASE( g_pServer );
  612. ReleaseDBInterface();
  613. }
  615. void LoadSetting()
  616. {
  617. CIniFile theConfigFile;
  619. _tstring sAppPath = GetAppFullPath( NULL );
  621. _tstring sConfigFile;
  623. sConfigFile = sAppPath + "Goddess.cfg";
  625. theConfigFile.SetFile( sConfigFile.c_str() );
  627. g_nServerPort = theConfigFile.ReadInteger( "Setting", "Port", 5001 );
  629. g_nMaxRoleCount = theConfigFile.ReadInteger( "Setting", "MaxRoleCount", 3 );
  631. g_BackupSleepTime = theConfigFile.ReadInteger( "Setting", "BackupSleepTime", 3 );
  632. }
  634. void SaveSetting()
  635. {
  636. CIniFile theConfigFile;
  638. _tstring sAppPath = GetAppFullPath( NULL );
  640. _tstring sConfigFile;
  642. sConfigFile = sAppPath + "Goddess.cfg";
  644. theConfigFile.SetFile( sConfigFile.c_str() );
  646. theConfigFile.WriteInteger( "Setting", "Port", g_nServerPort );
  648. theConfigFile.WriteInteger(  "Setting", "MaxRoleCount", g_nMaxRoleCount );
  650. theConfigFile.WriteInteger(  "Setting", "BackupSleepTime", g_BackupSleepTime );
  651. }