305 lines
7.7 KiB
C++
305 lines
7.7 KiB
C++
#include "StdAfx.h"
|
||
#include "WinService.h"
|
||
#include "DataProcess.h"
|
||
#include "SockProcess.h"
|
||
#include "SelectSockProcess.h"
|
||
#include "IOCPSockProcess.h"
|
||
#include "GateServer.h"
|
||
#include "GateServerConfig.h"
|
||
#include "FileLogger.h"
|
||
#include <string.h>
|
||
#ifndef WIN32
|
||
// #include "client/linux/handler/exception_handler.h"
|
||
#endif
|
||
#ifndef WIN32
|
||
DWORD GetTickCount()
|
||
{
|
||
struct timespec ts;
|
||
clock_gettime(CLOCK_MONOTONIC, &ts);
|
||
return (ts.tv_sec*1000 + ts.tv_nsec/1000000);
|
||
}
|
||
#endif
|
||
|
||
bool GateEngineRunning = true;
|
||
|
||
static void signal_handler(int sig_num)
|
||
{
|
||
if( sig_num == SIGHUP ||
|
||
sig_num == SIGINT ||
|
||
sig_num == SIGTERM )
|
||
{
|
||
GateEngineRunning = false;
|
||
OutputMsg( rmTip, _T("[SIGNAL] signal(%d) signal_handler... "), sig_num);
|
||
}
|
||
else
|
||
{
|
||
OutputMsg( rmError, _T("[SIGNAL] signal(%d) signal_handler... "), sig_num);
|
||
}
|
||
|
||
}
|
||
|
||
void ServerMain(int argc, char** argv);
|
||
VOID ServiceMain(int argc, char** argv);
|
||
|
||
bool CreateAndStartGateServers(CBaseList<CRunSockProcesser*> &gateList,char *pFileName=NULL);
|
||
void DestroyGateServers(CBaseList<CRunSockProcesser*> &gateList);
|
||
|
||
//服务器是否在运行
|
||
bool g_ServerIsRunning =true;
|
||
|
||
void TestMemory()
|
||
{
|
||
DWORD dwStart = GetTickCount();
|
||
for(int i = 0; i < 1000000; i++) { char *p = (char*)malloc(i%1000000); free(p); }
|
||
|
||
DWORD dwEnd = GetTickCount();
|
||
printf("***********************",dwEnd - dwStart);
|
||
printf("100wMemoryTest:%d ms\n",dwEnd - dwStart);
|
||
}
|
||
|
||
|
||
#ifdef WIN32
|
||
class CGateService : public CWinService
|
||
{
|
||
public:
|
||
CGateService(LPCSTR pServiceName, DWORD dwServiceType = SERVICE_WIN32_OWN_PROCESS):CWinService(pServiceName, dwServiceType){};
|
||
int AppMain()
|
||
{
|
||
CFileLogger flog(_T("GateServer_%s.log"), getCurrentTimeDesc());
|
||
CBaseList<CRunSockProcesser*> gateList;
|
||
SetCurrentDirectory("../"); //CWinService里会把当前目录改成exe允许的目录，所以这里再转一下
|
||
|
||
|
||
if ( CreateAndStartGateServers(gateList) )
|
||
{
|
||
in_addr ia;
|
||
ia.s_addr = RUNGATE_KRN_VERSION;
|
||
|
||
OutputMsg( rmTip, _T("-------------------------------------------") );
|
||
OutputMsg( rmTip, _T("网关服务器启动成功，核心版本号是%s"), inet_ntoa(ia) );
|
||
OutputMsg( rmTip, _T("quit命令可以停止服务并退出程序") );
|
||
OutputMsg( rmTip, _T("-------------------------------------------") );
|
||
while (!m_boServiceExit)
|
||
{
|
||
Sleep(1000);
|
||
}
|
||
}
|
||
DestroyGateServers(gateList);
|
||
Sleep(2000);//休眠一秒等待日志全部刷新到文件
|
||
return 0;
|
||
}
|
||
};
|
||
#endif
|
||
|
||
int main(int argc, char **argv)
|
||
{
|
||
//TestMemory();
|
||
#ifdef WIN32
|
||
SetUnhandledExceptionFilter( DefaultUnHandleExceptionFilter );
|
||
#endif
|
||
(void)signal(SIGHUP, SIG_IGN);
|
||
(void)signal(SIGINT, signal_handler);
|
||
(void)signal(SIGTERM, signal_handler);
|
||
InitDefMsgOut();
|
||
CTimeProfMgr::getSingleton().InitMgr();
|
||
winitseed((unsigned int)_getTickCount()); //初始化随机数函数
|
||
|
||
#ifndef _SERVICE
|
||
//if (argc == 2 && strncmp("/cmd",argv[1],4)==0)//平时调试用
|
||
{
|
||
#ifdef WIN32
|
||
SetCurrentDirectory("./");
|
||
#else
|
||
std::string filename(argv[0]);
|
||
size_t found = filename.find_last_of("/\\");
|
||
filename = filename.substr(0, found);
|
||
if( filename[0] == '.' && filename.length()==1 )
|
||
filename = "./" ;
|
||
SetCurrentDirectory(filename.c_str());
|
||
#endif
|
||
ServerMain(argc, argv);
|
||
}
|
||
#else
|
||
//else
|
||
{
|
||
ServiceMain(argc, argv);
|
||
}
|
||
#endif
|
||
|
||
UninitDefMsgOut();
|
||
|
||
#ifdef _MLIB_DUMP_MEMORY_LEAKS_
|
||
_CrtDumpMemoryLeaks();
|
||
#endif
|
||
return 0;
|
||
}
|
||
|
||
#ifdef WIN32
|
||
VOID ServiceMain(int argc, char** argv)
|
||
{
|
||
LPTSTR sCmd = NULL;
|
||
if (argc >= 2)
|
||
{
|
||
sCmd = argv[1];
|
||
}
|
||
CGateService service("GateService");
|
||
|
||
service.Run(sCmd);
|
||
}
|
||
#endif
|
||
|
||
void ServerMain(int argc, char **argv)
|
||
{
|
||
CFileLogger flog(_T("../log/GateServer_%s.log"), getCurrentTimeDesc());
|
||
CBaseList<CRunSockProcesser*> gateList;
|
||
|
||
char *pFileName =NULL;
|
||
if(argc >=2)
|
||
{
|
||
pFileName = argv[1];
|
||
}else
|
||
{
|
||
pFileName="GateServerLinux.txt";
|
||
}
|
||
|
||
if ( CreateAndStartGateServers(gateList,pFileName) )
|
||
{
|
||
TCHAR sCmdBuf[512];
|
||
in_addr ia;
|
||
ia.s_addr = RUNGATE_KRN_VERSION;
|
||
|
||
OutputMsg( rmTip, _T("-------------------------------------------") );
|
||
OutputMsg( rmTip, _T("网关服务器启动成功，核心版本号是%s"), inet_ntoa(ia) );
|
||
OutputMsg( rmTip, _T("spf命令可以查看当前的网关状态") );
|
||
OutputMsg( rmTip, _T("quit命令可以停止服务并退出程序") );
|
||
OutputMsg( rmTip, _T("-------------------------------------------") );
|
||
|
||
while (g_ServerIsRunning)
|
||
{
|
||
#ifdef WIN32
|
||
if(_kbhit())
|
||
{
|
||
_getts(sCmdBuf);
|
||
}
|
||
else
|
||
{
|
||
Sleep(100);
|
||
continue;
|
||
}
|
||
|
||
|
||
if ( strncmp(sCmdBuf, _T("\\q"), 2) == 0
|
||
|| strncmp(sCmdBuf, _T("exit"), 4) == 0
|
||
|| strncmp(sCmdBuf, _T("quit"), 4) == 0 )
|
||
{
|
||
OutputMsg( rmTip, _T("正在停止网关服务...") );
|
||
break;
|
||
}
|
||
else if ( strncmp(sCmdBuf, _T("spf"), 3) == 0 ) // 需要刷
|
||
{
|
||
for (INT_PTR i=0; i< gateList.count(); ++i)
|
||
{
|
||
CRunDataProcesser::ShowGateStatus(gateList[i]->GetDataProcesser());
|
||
}
|
||
|
||
CTimeProfMgr::getSingleton().dump();
|
||
}
|
||
else if (strncmp(sCmdBuf, _T("stst"), 4) == 0) // stst 10（ms为单位）
|
||
{
|
||
char *pParam = sCmdBuf + 4;
|
||
while (*pParam == ' ')
|
||
pParam++;
|
||
if (pParam)
|
||
{
|
||
UINT_PTR nInterval = atoi(pParam);
|
||
if (nInterval > 0)
|
||
{
|
||
CRunDataProcesser::s_nSndThreadSleepTime = (unsigned int)nInterval;
|
||
OutputMsg(rmNormal, _T("Set SndThreadSleepTime[%d ms] Succ"), nInterval);
|
||
}
|
||
}
|
||
}
|
||
else if(strncmp(sCmdBuf, _T("ignore"), 6) == 0)
|
||
{
|
||
CRunDataProcesser::s_nIgnoreDataPacket = 1;
|
||
}
|
||
else if(strncmp(sCmdBuf, _T("resume"), 6) == 0)
|
||
{
|
||
CRunDataProcesser::s_nIgnoreDataPacket = 0;
|
||
}
|
||
|
||
#else
|
||
if(!GateEngineRunning) break;
|
||
#endif
|
||
|
||
Sleep(10);
|
||
}
|
||
}
|
||
DestroyGateServers(gateList);
|
||
Sleep(2000);//休眠一秒等待日志全部刷新到文件
|
||
}
|
||
|
||
bool CreateAndStartGateServers(CBaseList<CRunSockProcesser*> &gateList,char *pConfig)
|
||
{
|
||
CGateConfigList configList;
|
||
CGateServerConfig configReader;
|
||
CRunSockProcesser *pRunSock;
|
||
CRunDataProcesser *pRunData;
|
||
|
||
if ( configReader.loadConfig(configList,pConfig) )
|
||
{
|
||
INT_PTR i, nCount = configList.count();
|
||
PGATECONFIG pConfigList = configList;
|
||
for (i=0; i<nCount; ++i)
|
||
{
|
||
//创建并配置网关后台服务数据处理器
|
||
pRunData = new CRunDataProcesser(__max(1024, pConfigList[i].GateServer.nMaxSession));
|
||
pRunData->SetSendThreadCount( __max(1,pConfigList[i].GateServer.SendThreadCount) );
|
||
pRunData->SetDataProcessType(dp_Default);
|
||
pRunData->SetServerHost(pConfigList[i].BackServer.sHost);
|
||
pRunData->SetServerPort(pConfigList[i].BackServer.nPort);
|
||
pRunData->SetName(pConfigList[i].GateServer.sName);
|
||
//创建并配置网关用户服务器处理器
|
||
pRunSock = new CSelectRunSockProcesser();
|
||
pRunSock->SetBindAddress( pConfigList[i].GateServer.sAddress );
|
||
pRunSock->SetBindPort( pConfigList[i].GateServer.nPort );
|
||
pRunSock->SetDataProcesser(pRunData);
|
||
pRunData->SetRunSockProcesser(pRunSock);
|
||
//将网关添加到列表中
|
||
gateList.add(pRunSock);
|
||
//启动网关
|
||
if ( !pRunSock->InitBase() || !pRunData->Startup() || !pRunSock->Start() )
|
||
{
|
||
OutputMsg(rmError, _T("Start GateServer[%s] Failed"), pConfigList[i].GateServer.sName);
|
||
return false;
|
||
}
|
||
OutputMsg(rmTip, _T("GateServer[%s] Started"), pConfigList[i].GateServer.sName);
|
||
}
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
void DestroyGateServers(CBaseList<CRunSockProcesser*> &gateList)
|
||
{
|
||
CRunSockProcesser *pRunSock;
|
||
CRunDataProcesser *pRunData;
|
||
INT_PTR i, nCount = gateList.count();
|
||
|
||
for (i=0; i<nCount; ++i)
|
||
{
|
||
//停止网关用户服务器处理器
|
||
pRunSock = gateList[i];
|
||
pRunSock->Stop();
|
||
//停止网关后台服务数据处理器
|
||
pRunData = pRunSock->GetDataProcesser();
|
||
pRunData->Stop();
|
||
//销毁网关对象
|
||
pRunSock->SetDataProcesser(NULL);
|
||
OutputMsg(rmTip, _T("GateServer[%s] Stoped"), pRunData->GetName());
|
||
delete pRunSock;
|
||
delete pRunData;
|
||
}
|
||
gateList.empty();
|
||
}
|