guardian.cpp 37 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432
  1. // guardian.cpp : Defines the entry point for the console application.
  2. //
  3. //#include "stdafx.h"
  4. #ifdef linux
  5. #include <sys/types.h>
  6. #include <sys/stat.h>
  7. #include <sys/socket.h>
  8. #include <sys/epoll.h>
  9. #include <netinet/in.h>
  10. #include <arpa/inet.h>
  11. #include <dirent.h>
  12. #include <fcntl.h>
  13. #include <unistd.h>
  14. #include <stdio.h>
  15. #include <errno.h>
  16. #include <string.h>
  17. #include <syslog.h>
  18. #include <signal.h>
  19. #include <pthread.h>
  20. int epfd;
  21. struct epoll_event ev;
  22. #define MAXLINE 128
  23. #define OPEN_MAX 100
  24. #define LISTENQ 20
  25. #define INFTIM 1000
  26. #else
  27. #undef UNICODE
  28. #pragma comment( linker, "/subsystem:\"windows\" /entry:\"mainCRTStartup\"" )
  29. #define WIN32_LEAN_AND_MEAN
  30. #include <windows.h>
  31. #include <winsock2.h>
  32. #include <ws2tcpip.h>
  33. #include <TlHelp32.h>
  34. #include <direct.h>
  35. #include <ctime>
  36. #endif //linux
  37. #include <iostream>
  38. #include <fstream>
  39. #include <string>
  40. #include "GuardianBase.h"
  41. #include "guardian.h"
  42. using namespace std;
  43. #define NET_TYPE_WIRELESS 1
  44. #define NET_TYPE_ETHERNET 2
  45. #define NET_TYPE_3G 3
  46. #define NET_TYPE_4G 4
  47. #define NET_TYPE_BLUETOOTH 5
  48. #define DLL_PATH "C:\\RVC\\SystemInit\\CmbPadDll.dll"
  49. typedef DWORD(*pWlanConnect)(BOOL bConnect, int nType);
  50. typedef DWORD (*pDisableBluetooth)(BOOL bDisable);
  51. #ifdef linux
  52. int g_sListen;
  53. pthread_mutex_t g_cs, g_cs_event, g_cs_log;
  54. #define EnterCriticalSectionRVC(xType) pthread_mutex_lock(&xType)
  55. #define LeaveCriticalSectionRVC(xType) pthread_mutex_unlock(&xType)
  56. #else
  57. SOCKET g_sListen = INVALID_SOCKET;
  58. #pragma comment (lib, "Ws2_32.lib")
  59. unsigned int __stdcall DoNetControl(void *pData);
  60. CRITICAL_SECTION g_cs,g_cs_event,g_cs_log;
  61. #define EnterCriticalSectionRVC(xType) EnterCriticalSection(&xType)
  62. #define LeaveCriticalSectionRVC(xType) LeaveCriticalSection(&xType)
  63. #endif //linux
  64. //unsigned int __stdcall DoLog(void *pData);
  65. const int DEFAULT_BUFLEN = 512;
  66. const int SHAKEHAND_BUFZIE = 32;
  67. #define DEFAULT_PORT "30005"
  68. const int default_port = 30005;
  69. int g_needToRollBack = 0;
  70. bool g_bFrameQuit = false,g_bFrameOnline = false,g_bLogFileOpen = false,g_bAuthSuc = false,g_bInUpgrade = false;
  71. ULONGLONG g_dwTimeBegin = 0;
  72. ULONGLONG g_dwUpgradeRestartTimeBegin = 0;
  73. int cnt = 0;
  74. ofstream logFile;
  75. char currDirBuf[256];
  76. char chDisk[2];
  77. void NetControl();
  78. const int MAX_WAIT_TIME_TO_RESTART = 600000;
  79. const int MAX_WAIT_TIME_TO_RESTART_UPGRADE = 110000;
  80. int g_startDay = 0;
  81. //HANDLE g_logHandle;
  82. char g_allMsg[1024];
  83. #ifdef linux
  84. extern "C"
  85. {
  86. void* DoWorkLinux(void* arg);
  87. void* DoNetControlLinux(void* arg);
  88. }
  89. #define FUNCTION_STDCALL
  90. typedef unsigned short WORD;
  91. typedef struct _SYSTEMTIME {
  92. WORD wYear;
  93. WORD wMonth;
  94. WORD wDayOfWeek;
  95. WORD wDay;
  96. WORD wHour;
  97. WORD wMinute;
  98. WORD wSecond;
  99. WORD wMilliseconds;
  100. } SYSTEMTIME;
  101. #else
  102. typedef struct
  103. {
  104. OVERLAPPED Overlapped;
  105. WSABUF DataBuf;
  106. CHAR Buffer[DATA_BUFSIZE];
  107. } PER_IO_OPERATION_DATA, * LPPER_IO_OPERATION_DATA;
  108. typedef struct
  109. {
  110. SOCKET Socket;
  111. } PER_HANDLE_DATA, * LPPER_HANDLE_DATA;
  112. unsigned int __stdcall DoWork(void* pData);
  113. #define FUNCTION_STDCALL __stdcall
  114. #endif
  115. unsigned long long GetTickCountRVC() {
  116. #ifdef linux
  117. struct timespec ts;
  118. clock_gettime(CLOCK_MONOTONIC, &ts);
  119. return (ts.tv_sec * 1000 + ts.tv_nsec / 1000000);
  120. #else
  121. return GetTickCount64();
  122. #endif //RVC_OS_WIN
  123. }
  124. void GetLocalTimeRVC(SYSTEMTIME& stTime)
  125. {
  126. #ifdef linux
  127. struct tm* pst = NULL;
  128. time_t t = time(NULL);
  129. pst = localtime(&t);
  130. stTime.wYear = pst->tm_year + 1900;
  131. stTime.wMonth = pst->tm_mon + 1;//because the range of tm_mon is:0-11
  132. stTime.wDay = pst->tm_mday;
  133. stTime.wHour = pst->tm_hour;
  134. stTime.wMinute = pst->tm_min;
  135. stTime.wDayOfWeek = pst->tm_wday;
  136. stTime.wSecond = pst->tm_sec;
  137. stTime.wMilliseconds = 0;
  138. //("oiltest,time:%d,%d,%d,%d", stTime.wYear, stTime.wHour, stTime.wMinute, stTime.wDayOfWeek);
  139. #else
  140. GetLocalTime(stTime);
  141. #endif
  142. }
  143. int BeginThreadRVC(void* (*pFuncLinux)(void*), unsigned int(FUNCTION_STDCALL* pFuncWin)(void*))
  144. {
  145. #ifdef linux
  146. pthread_t tidp;
  147. if ((pthread_create(&tidp, NULL, pFuncLinux, NULL) == -1))
  148. {
  149. //printf("create error!\n");
  150. return 1;
  151. }
  152. cout << "after pthread_create of linux thread" << endl;
  153. return 0;
  154. #else
  155. _beginthreadex(NULL, 0, pFuncWin, NULL, 0, NULL);
  156. return 0;
  157. #endif //linux
  158. }
  159. void EndThreadRVC()
  160. {
  161. #ifdef linux
  162. cout << "thread to exit" << endl;
  163. pthread_exit(0);
  164. cout << "return of EndThreadRVC" << endl;
  165. #else
  166. _endthreadex(0);
  167. #endif //linux
  168. }
  169. void StartDoNetControlRVC()
  170. {
  171. #ifdef linux
  172. BeginThreadRVC(DoNetControlLinux, NULL);
  173. cout << "after DoNetControlLinux thread created." << endl;
  174. #else
  175. BeginThreadRVC(NULL, DoNetControl);
  176. #endif //linux
  177. }
  178. void StartDoWorkRVC()
  179. {
  180. #ifdef linux
  181. BeginThreadRVC(DoWorkLinux, NULL);
  182. #else
  183. BeginThreadRVC(NULL, DoWork);
  184. #endif //linux
  185. }
  186. void ServerReportEvent(const char* szFunction,const char* szName="guardian")
  187. {
  188. #ifdef linux
  189. return;//oiltest
  190. cout << "oiltest(SPE):" << szFunction << endl;
  191. openlog("guardian", LOG_CONS | LOG_PID, LOG_LOCAL0);
  192. syslog(LOG_ERR, szFunction);
  193. closelog();
  194. #else
  195. //LogToFile(false,false,false,szFunction);
  196. HANDLE hEventSource;
  197. LPCTSTR lpszStrings[2];
  198. int len = sizeof(szFunction);
  199. //char *Buffer = new char[len];
  200. char Buffer[1024] = {0};
  201. hEventSource = RegisterEventSource(NULL, szName);
  202. if( NULL != hEventSource )
  203. {
  204. //StringCchPrintf(Buffer, 80, TEXT("%s failed with %d"), szFunction, GetLastError());
  205. strcpy_s(Buffer,1024,szFunction);
  206. lpszStrings[0] = szName;
  207. lpszStrings[1] = Buffer;
  208. ReportEvent(hEventSource, // event log handle
  209. EVENTLOG_ERROR_TYPE, // event type
  210. 0, // event category
  211. 0xe0000001, // event identifier
  212. NULL, // no security identifier
  213. 2, // size of lpszStrings array
  214. 0, // no binary data
  215. lpszStrings, // array of strings
  216. NULL); // no binary data
  217. DeregisterEventSource(hEventSource);
  218. }
  219. #endif //linux
  220. }
  221. bool VersionRollback()
  222. {
  223. const int VersionLenMax = 64;
  224. fstream verFile,verBak;//to confirm
  225. string strActiveTxtPath, strVersionDatPath;
  226. #ifdef linux
  227. //todo oiltestlinux
  228. strActiveTxtPath = "/opt/run/version/active.txt";
  229. strVersionDatPath = "/opt/run/runinfo/runcfg/version.dat";
  230. #else
  231. string strDisk(chDisk);
  232. strActiveTxtPath = strDisk + ":\\Run\\version\\active.txt";
  233. strVersionDatPath = strDisk + ":\\Run\\runinfo\\runcfg\\version.dat";
  234. #endif //linux
  235. verFile.open(strActiveTxtPath,std::fstream::in|std::fstream::out|std::fstream::binary);
  236. verBak.open(strVersionDatPath,std::fstream::in|std::fstream::out|std::fstream::binary);
  237. if (!verFile.is_open() || !verBak.is_open())
  238. return false;
  239. char *pVerBak,*pNULL,*pCurrVer;
  240. pVerBak = new char[VersionLenMax];
  241. pNULL = new char[VersionLenMax];
  242. pCurrVer = new char[VersionLenMax];
  243. memset(pVerBak,0,VersionLenMax);
  244. memset(pNULL,0,VersionLenMax);
  245. memset(pCurrVer,0,VersionLenMax);
  246. verBak.seekg(0,verBak.end);
  247. int lenBak = verBak.tellg();
  248. verBak.seekg(0,verBak.beg);
  249. verBak.read(pVerBak,lenBak);
  250. //need to consider atomic op? oilyang
  251. verFile.seekg(0,verFile.end);
  252. int len = verFile.tellg();
  253. verFile.seekg(0,verFile.beg);
  254. verFile.read(pCurrVer,len);
  255. LogSingleMsg(pVerBak);
  256. LogSingleMsg(pCurrVer);
  257. LogSingleMsg(pNULL);
  258. if (strcmp(pVerBak,pCurrVer) == 0)
  259. {
  260. ServerReportEvent(pVerBak);
  261. verFile.close();
  262. verBak.close();
  263. delete []pVerBak;
  264. delete []pNULL;
  265. return false;
  266. }
  267. //verBak.write(pNULL,len);
  268. //verFile.write(pNULL,len);
  269. //how foolish?
  270. verFile.close();
  271. verFile.open(strActiveTxtPath,std::fstream::in|std::fstream::out|std::fstream::binary|std::fstream::trunc);
  272. verFile.seekg(0,verFile.beg);
  273. verFile.write(pVerBak,lenBak);
  274. LogSingleMsg("version will rollback to ");
  275. LogSingleMsg(pVerBak);
  276. verFile.close();
  277. verBak.close();
  278. delete []pVerBak;
  279. delete []pNULL;
  280. return true;
  281. }
  282. int FrameworkShutdown(bool bUpgrade=false,bool bRestart = true)
  283. {
  284. cout << "in FrameworkShutdown,bUpgrade:" << bUpgrade << ",bRestart:" << bRestart << endl;
  285. if (!bUpgrade)
  286. g_dwTimeBegin = GetTickCountRVC();
  287. NetControl();
  288. #ifdef linux
  289. //todo oiltestlinux
  290. LogSingleMsg("to kill spshell");
  291. ServerReportEvent("FrameworkShutdown linux");
  292. system("killall -9 spshell");
  293. system("killall -9 sphost");
  294. sleep(2);
  295. system("/opt/run/version/spexplorer.sh");
  296. LogSingleMsg("after kill spshell");
  297. return 0;
  298. #else
  299. STARTUPINFO si;
  300. PROCESS_INFORMATION pi;
  301. ZeroMemory( &si, sizeof(si) );
  302. si.cb = sizeof(si);
  303. ZeroMemory( &pi, sizeof(pi) );
  304. // Start the child process.
  305. CSimpleStringA csRestart,csVerPath,csAll,csSep("\""),csBlank(" "),csScript("wscript.exe"),csReFlag("r");
  306. csRestart = "sprestart.exe ";
  307. csVerPath = chDisk;
  308. csVerPath += ":\\Run\\version";
  309. csVerPath +="\\spexplorerfast.vbs";
  310. if (!bRestart)
  311. csReFlag = "n";
  312. csAll = csSep + csRestart + csSep + csBlank + csSep + csScript+csBlank + csVerPath + csSep
  313. + csBlank + csSep + csReFlag + csSep;
  314. LPTSTR szCmdline = _strdup(csAll);
  315. LogSingleMsg(szCmdline);
  316. //MessageBoxA(0,0,0,0);
  317. if( !CreateProcess( NULL,szCmdline,NULL,NULL,FALSE,CREATE_NEW_CONSOLE,NULL,NULL,&si,&pi))
  318. {
  319. //Dbg("CreateProcess failed (%d).\n", GetLastError());
  320. return -1;
  321. }
  322. DWORD dwErr = GetLastError();
  323. // Wait until child process exits.
  324. WaitForSingleObject( pi.hProcess, INFINITE );
  325. // Close process and thread handles.
  326. CloseHandle( pi.hProcess );
  327. CloseHandle( pi.hThread );
  328. return 0;
  329. #endif //linux
  330. }
  331. bool FrameworkRollBack()
  332. {
  333. bool bVerRollback = VersionRollback();
  334. LogToFile(false,false,false,"version rollback ");
  335. if (true)
  336. FrameworkShutdown();
  337. return true;
  338. }
  339. bool ReadRuninfoContent(string &strData)
  340. {
  341. fstream runinfo;
  342. string strGdRuninfoPath;
  343. #ifdef linux
  344. //todo oiltestlinux
  345. strGdRuninfoPath = "/opt/run/runinfo/runcfg/gdruninfo";
  346. #else
  347. string strDisk(chDisk);
  348. strGdRuninfoPath = strDisk + ":\\Run\\runinfo\\runcfg\\gdruninfo";
  349. #endif //linux
  350. runinfo.open(strGdRuninfoPath,std::fstream::in|std::fstream::out|std::fstream::binary);
  351. if (!runinfo.is_open())
  352. {
  353. LogToFile(true, true,false, "open gdruninfo(read) failed.",strerror(errno), errno);
  354. return false;
  355. }
  356. runinfo.seekg(0,ios::end);
  357. int size = runinfo.tellg();
  358. if (size <= 0)
  359. return false;
  360. char *pData = new char[size+1];
  361. memset(pData,0,size+1);
  362. runinfo.seekg(0,ios::beg);
  363. runinfo.read(pData,size);
  364. strData = string(pData);
  365. LogToFile(true, false,false, "read runinfo ", strData.c_str());
  366. return true;
  367. }
  368. bool WriteRunInfoContent(const char* pData)
  369. {
  370. ofstream runinfo;
  371. string strGdRuninfoPath;
  372. #ifdef linux
  373. //todo oiltestlinux
  374. strGdRuninfoPath = "/opt/run/runinfo/runcfg/gdruninfo";
  375. #else
  376. string strDisk(chDisk);
  377. strGdRuninfoPath = strDisk + ":\\Run\\runinfo\\runcfg\\gdruninfo";
  378. #endif //linux
  379. runinfo.open(strGdRuninfoPath,std::ofstream::in|std::ofstream::binary|std::ofstream::trunc);
  380. if (!runinfo.is_open())
  381. {
  382. ServerReportEvent("open gdruninfo(write) failed.");
  383. return false;
  384. }
  385. runinfo.write(pData,strlen(pData));
  386. runinfo.close();
  387. return true;
  388. }
  389. bool IsStartTimeFileExist()
  390. {
  391. ofstream runinfo;
  392. string strStartTimePath;
  393. #ifdef linux
  394. //todo oiltestlinux
  395. strStartTimePath = "/opt/run/runinfo/runcfg/starttime.dat";
  396. #else
  397. string strDisk(chDisk);
  398. strStartTimePath = strDisk + ":\\Run\\runinfo\\runcfg\\starttime.dat";
  399. #endif //linux
  400. runinfo.open(strStartTimePath, std::fstream::in | std::fstream::out | std::fstream::binary);
  401. if (!runinfo.is_open())
  402. return false;
  403. else
  404. {
  405. runinfo.close();
  406. return true;
  407. }
  408. }
  409. #ifdef linux
  410. void CheckDoWork(int sig)
  411. {
  412. cout << "CheckDoWork" << endl;
  413. if (SIGALRM == sig)
  414. {
  415. LogSingleMsg("to check value(linux)");
  416. EnterCriticalSectionRVC(g_cs);
  417. if (!g_bFrameQuit)
  418. {
  419. ULONGLONG dwTmpBegin = g_dwTimeBegin;
  420. ULONGLONG dwUpgradeTmpBegin = g_dwUpgradeRestartTimeBegin;
  421. LeaveCriticalSectionRVC(g_cs);
  422. ULONGLONG dwTimeEnd = GetTickCountRVC();
  423. //oilyang@20190828 add
  424. //升级重启后,在10分钟内,只要离最后一次交互时间大于2分钟,重启框架
  425. if ((g_bInUpgrade && ((dwTimeEnd - dwUpgradeTmpBegin) < MAX_WAIT_TIME_TO_RESTART))
  426. && (!g_bAuthSuc && ((dwTimeEnd - dwTmpBegin) > MAX_WAIT_TIME_TO_RESTART_UPGRADE)))
  427. {
  428. LogSingleMsg("**in upgrade restart**,to restart framework(linux).");
  429. cout << "oiltest 0127" << endl;
  430. FrameworkShutdown(true, true);
  431. }
  432. if ((dwTimeEnd - dwTmpBegin) > MAX_WAIT_TIME_TO_RESTART)
  433. {
  434. LogSingleMsg("framework maybe down.");
  435. string strRunInfo;
  436. if (ReadRuninfoContent(strRunInfo))
  437. {
  438. LogSingleMsg("to check if need rollback.");
  439. LogSingleMsg((char*)strRunInfo.c_str());
  440. if (strRunInfo.compare("111") == 0 && IsStartTimeFileExist())
  441. {
  442. EnterCriticalSectionRVC(g_cs);
  443. g_needToRollBack = 1;
  444. LeaveCriticalSectionRVC(g_cs);
  445. }
  446. }
  447. else
  448. LogSingleMsg("read run info failed.");
  449. if (g_needToRollBack)
  450. {
  451. LogSingleMsg("after upgrade,time elapse,but can't wait the shake hands,so rollback.");
  452. FrameworkRollBack();
  453. }
  454. else
  455. {
  456. //framework is down,to restart it! oilyang 20150413
  457. FrameworkShutdown();
  458. }
  459. }
  460. }
  461. else
  462. LeaveCriticalSectionRVC(g_cs);
  463. alarm(2 * 60); //we contimue set the timer
  464. }
  465. return;
  466. }
  467. void* DoWorkLinux(void* arg)
  468. {
  469. //LogSingleMsg("to wait.");
  470. cout << "DoWorkLinux" << endl;
  471. signal(SIGALRM, CheckDoWork); //relate the signal and function
  472. alarm(2 * 60); //trigger the timer
  473. EndThreadRVC();
  474. //return 0;
  475. }
  476. void* DoNetControlLinux(void* arg)
  477. {
  478. cout << "to end DoNetControlLinux" << endl;
  479. EndThreadRVC();
  480. cout << "after end DoNetControlLinux" << endl;
  481. }
  482. void DataProcessLinux(int socket,const char*data)
  483. {
  484. GuardianInfo* pInfo = (GuardianInfo*)data;
  485. LogToFile(false, true, false, "eType:", "", pInfo->eType);
  486. cout << "eType:" << pInfo->eType << ",p1:" << pInfo->dwParam1 << ",p2:" << pInfo->dwParam2 << ",dwSize:" << pInfo->dwSize << endl;
  487. //LogToFile(false,true,false,"clientsocket:", "", pPerHandleData->Socket);
  488. switch (pInfo->eType)
  489. {
  490. case GdOpShakeHand:
  491. {
  492. LogToFile(false, false, false, "Shake hand.");
  493. if (!g_bFrameOnline)
  494. {
  495. g_bFrameOnline = true;
  496. StartDoNetControlRVC();
  497. }
  498. int rc, err;
  499. EnterCriticalSectionRVC(g_cs);
  500. g_bFrameQuit = false;
  501. g_dwTimeBegin = GetTickCountRVC();
  502. LeaveCriticalSectionRVC(g_cs);
  503. char* pBuf = new char[8];
  504. memset(pBuf, 0, 8);
  505. //get guardian state
  506. //ex:u(update),g(guardian),r(reboot),b(rollback)
  507. if (cnt % 2 == 0)
  508. memcpy(pBuf, "g", 1);
  509. else
  510. memcpy(pBuf, "u", 1);
  511. cnt++;
  512. rc = send(socket, pBuf, 8, 0);
  513. delete[]pBuf;
  514. close(socket);
  515. }
  516. break;
  517. case GdOpUpdateTask:
  518. {
  519. if (!g_bFrameOnline)
  520. {
  521. g_bFrameOnline = true;
  522. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  523. StartDoNetControlRVC();
  524. }
  525. }
  526. break;
  527. case GdOpQueryInstall:
  528. {
  529. if (!g_bFrameOnline)
  530. {
  531. g_bFrameOnline = true;
  532. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  533. StartDoNetControlRVC();
  534. }
  535. int rc, err;
  536. char* pBuf = new char[8];
  537. memset(pBuf, 0, 8);
  538. if (cnt % 2 == 0)
  539. memcpy(pBuf, "y", 1);
  540. else
  541. memcpy(pBuf, "n", 1);
  542. cnt++;
  543. rc = send(socket, pBuf, 8, 0);
  544. delete[]pBuf;
  545. close(socket);
  546. }
  547. break;
  548. case GdOpUpgradeRestart:
  549. {
  550. if (pInfo->dwParam1 == 4)//just know framework is starting...
  551. {
  552. LogSingleMsg("framework is starting...");
  553. cout << "framework is starting..." << endl;
  554. g_bFrameOnline = true;
  555. g_bAuthSuc = false;
  556. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  557. StartDoNetControlRVC();
  558. cout << "to break upgrade..." << endl;
  559. close(socket);
  560. break;
  561. }
  562. if (pInfo->dwParam1 == 5)
  563. {
  564. LogSingleMsg("framework auth ok.");
  565. g_bInUpgrade = false;
  566. g_bAuthSuc = true;
  567. }
  568. if (!g_bFrameOnline)
  569. {
  570. g_bFrameOnline = true;
  571. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  572. StartDoNetControlRVC();
  573. }
  574. if (pInfo->dwParam1 == 1 || pInfo->dwParam1 == 3)
  575. g_dwTimeBegin = GetTickCountRVC();
  576. ServerReportEvent("upgrade restart");
  577. LogToFile(false, true, false, "upgrade restart ", "", pInfo->dwParam1);
  578. if (pInfo->dwParam1 == 1)//framework to restart after upgrade
  579. {
  580. g_bInUpgrade = true;
  581. WriteRunInfoContent("111");
  582. g_dwUpgradeRestartTimeBegin = GetTickCountRVC();
  583. }
  584. else if (pInfo->dwParam1 == 2)//need rollback after upgrade
  585. {
  586. LogSingleMsg("upgrade restart 222");
  587. WriteRunInfoContent("222");
  588. //rollback
  589. EnterCriticalSectionRVC(g_cs);
  590. g_needToRollBack = 0;
  591. LeaveCriticalSectionRVC(g_cs);
  592. LogSingleMsg("healthmanager said to rollback");
  593. FrameworkRollBack();
  594. }
  595. else if (pInfo->dwParam1 == 3)//upgrade succeeded
  596. {
  597. LogSingleMsg("upgrade restart ok");
  598. WriteRunInfoContent("333");
  599. EnterCriticalSectionRVC(g_cs);
  600. g_needToRollBack = 0;
  601. LeaveCriticalSectionRVC(g_cs);
  602. }
  603. //runinfo.close();
  604. close(socket);
  605. }
  606. break;
  607. case GdOpFrameQuit:
  608. if (!g_bFrameOnline)
  609. {
  610. g_bFrameOnline = true;
  611. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  612. StartDoNetControlRVC();
  613. }
  614. EnterCriticalSectionRVC(g_cs);
  615. g_bFrameQuit = true;
  616. LeaveCriticalSectionRVC(g_cs);
  617. close(socket);
  618. break;
  619. default:
  620. break;
  621. }
  622. cout << "quit of DataProcessLinux" << endl;
  623. }
  624. #else
  625. void DataProcess(LPPER_HANDLE_DATA pPerHandleData,LPPER_IO_OPERATION_DATA pPerIoData)
  626. {
  627. GuardianInfo* pInfo = (GuardianInfo*)pPerIoData->Buffer;
  628. LogToFile(false, true, false,"eType:", "", pInfo->eType);
  629. //LogToFile(false,true,false,"clientsocket:", "", pPerHandleData->Socket);
  630. switch(pInfo->eType)
  631. {
  632. case GdOpShakeHand:
  633. {
  634. LogToFile(false, false,false, "Shake hand.");
  635. if (!g_bFrameOnline)
  636. {
  637. g_bFrameOnline = true;
  638. StartDoNetControlRVC();
  639. }
  640. int rc,err;
  641. EnterCriticalSectionRVC(g_cs);
  642. g_bFrameQuit = false;
  643. g_dwTimeBegin = GetTickCountRVC();
  644. LeaveCriticalSectionRVC(g_cs);
  645. char* pBuf = new char[8];
  646. memset(pBuf,0,8);
  647. //get guardian state
  648. //ex:u(update),g(guardian),r(reboot),b(rollback)
  649. if (cnt%2 == 0)
  650. memcpy(pBuf,"g",1);
  651. else
  652. memcpy(pBuf,"u",1);
  653. cnt++;
  654. rc = send(pPerHandleData->Socket,pBuf,8,0);
  655. delete []pBuf;
  656. closesocket(pPerHandleData->Socket);
  657. }
  658. break;
  659. case GdOpUpdateTask:
  660. {
  661. if (!g_bFrameOnline)
  662. {
  663. g_bFrameOnline = true;
  664. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  665. StartDoNetControlRVC();
  666. }
  667. }
  668. break;
  669. case GdOpQueryInstall:
  670. {
  671. if (!g_bFrameOnline)
  672. {
  673. g_bFrameOnline = true;
  674. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  675. StartDoNetControlRVC();
  676. }
  677. int rc,err;
  678. char* pBuf = new char[8];
  679. memset(pBuf,0,8);
  680. if (cnt%2 == 0)
  681. memcpy(pBuf,"y",1);
  682. else
  683. memcpy(pBuf,"n",1);
  684. cnt++;
  685. rc = send(pPerHandleData->Socket,pBuf,8,0);
  686. delete []pBuf;
  687. closesocket(pPerHandleData->Socket);
  688. }
  689. break;
  690. case GdOpUpgradeRestart:
  691. {
  692. if (pInfo->dwParam1 == 4)//just know framework is starting...
  693. {
  694. LogSingleMsg("framework is starting...");
  695. g_bFrameOnline = true;
  696. g_bAuthSuc = false;
  697. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  698. StartDoNetControlRVC();
  699. break;
  700. }
  701. if (pInfo->dwParam1 == 5)
  702. {
  703. LogSingleMsg("framework auth ok.");
  704. g_bInUpgrade = false;
  705. g_bAuthSuc = true;
  706. }
  707. if (!g_bFrameOnline)
  708. {
  709. g_bFrameOnline = true;
  710. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  711. StartDoNetControlRVC();
  712. }
  713. if (pInfo->dwParam1 == 1 || pInfo->dwParam1 == 3)
  714. g_dwTimeBegin = GetTickCountRVC();
  715. ServerReportEvent("upgrade restart");
  716. LogToFile(false, true,false, "upgrade restart ", "", pInfo->dwParam1);
  717. if (pInfo->dwParam1 == 1)//framework to restart after upgrade
  718. {
  719. g_bInUpgrade = true;
  720. WriteRunInfoContent("111");
  721. g_dwUpgradeRestartTimeBegin = GetTickCountRVC();
  722. }
  723. else if (pInfo->dwParam1 == 2)//need rollback after upgrade
  724. {
  725. LogSingleMsg("upgrade restart 222");
  726. WriteRunInfoContent("222");
  727. //rollback
  728. EnterCriticalSectionRVC(g_cs);
  729. g_needToRollBack = 0;
  730. LeaveCriticalSectionRVC(g_cs);
  731. LogSingleMsg("healthmanager said to rollback");
  732. FrameworkRollBack();
  733. }else if (pInfo->dwParam1 == 3)//upgrade succeeded
  734. {
  735. LogSingleMsg("upgrade restart ok");
  736. WriteRunInfoContent("333");
  737. EnterCriticalSectionRVC(g_cs);
  738. g_needToRollBack = 0;
  739. LeaveCriticalSectionRVC(g_cs);
  740. }
  741. //runinfo.close();
  742. closesocket(pPerHandleData->Socket);
  743. }
  744. break;
  745. case GdOpFrameQuit:
  746. if (!g_bFrameOnline)
  747. {
  748. g_bFrameOnline = true;
  749. //_beginthreadex(NULL,0,DoNetControl,NULL,0,NULL);
  750. StartDoNetControlRVC();
  751. }
  752. EnterCriticalSectionRVC(g_cs);
  753. g_bFrameQuit = true;
  754. LeaveCriticalSectionRVC(g_cs);
  755. closesocket(pPerHandleData->Socket);
  756. break;
  757. default:
  758. break;
  759. }
  760. }
  761. DWORD WINAPI ProcessIO(LPVOID lpParam)
  762. {
  763. HANDLE CompletionPort = (HANDLE)lpParam;
  764. DWORD BytesTransferred;
  765. LPPER_HANDLE_DATA PerHandleData;
  766. LPPER_IO_OPERATION_DATA PerIoData;
  767. while(true)
  768. {
  769. if(0 == GetQueuedCompletionStatus(CompletionPort, &BytesTransferred, (LPDWORD)&PerHandleData, (LPOVERLAPPED*)&PerIoData, INFINITE))
  770. {
  771. if( (GetLastError() == WAIT_TIMEOUT) || (GetLastError() == ERROR_NETNAME_DELETED) )
  772. {
  773. LogToFile(false, true,false, "closing socket(1)", "", PerHandleData->Socket);
  774. closesocket(PerHandleData->Socket);
  775. delete PerIoData;
  776. delete PerHandleData;
  777. continue;
  778. }
  779. else
  780. {
  781. //OutErr("GetQueuedCompletionStatus failed!");
  782. }
  783. return 0;
  784. }
  785. // client quit
  786. if(BytesTransferred == 0)
  787. {
  788. LogToFile(false, true,false, "closing socket(2)", "", PerHandleData->Socket);
  789. closesocket(PerHandleData->Socket);
  790. delete PerIoData;
  791. delete PerHandleData;
  792. continue;
  793. }
  794. //MessageBoxA(0,0,0,0);
  795. //LogToFile(false, true,false, "socket", "", PerHandleData->Socket);
  796. // receiving data process
  797. DataProcess(PerHandleData,PerIoData);
  798. // socket WSARecv
  799. DWORD Flags = 0;
  800. DWORD dwRecv = 0;
  801. ZeroMemory(PerIoData, sizeof(PER_IO_OPERATION_DATA));
  802. PerIoData->DataBuf.buf = PerIoData->Buffer;
  803. PerIoData->DataBuf.len = DATA_BUFSIZE;
  804. WSARecv(PerHandleData->Socket, &PerIoData->DataBuf, 1, &dwRecv, &Flags, &PerIoData->Overlapped, NULL);
  805. }
  806. return 0;
  807. }
  808. unsigned int __stdcall DoWork(void* pData)
  809. {
  810. LogSingleMsg("to wait.");
  811. HANDLE hTimer;
  812. LARGE_INTEGER li;
  813. hTimer = CreateWaitableTimer(NULL,FALSE,NULL);
  814. const int nTimerUnitsPerSecond = 10000000;
  815. li.QuadPart = -(1*60*nTimerUnitsPerSecond);//oiltmp 1 minute
  816. SetWaitableTimer(hTimer,&li,2*60*1000,NULL,NULL,FALSE);
  817. while(1)
  818. {
  819. WaitForSingleObject(hTimer,INFINITE);
  820. LogSingleMsg("to check value");
  821. EnterCriticalSectionRVC(g_cs);
  822. if (!g_bFrameQuit)
  823. {
  824. ULONGLONG dwTmpBegin = g_dwTimeBegin;
  825. ULONGLONG dwUpgradeTmpBegin = g_dwUpgradeRestartTimeBegin;
  826. LeaveCriticalSectionRVC(g_cs);
  827. ULONGLONG dwTimeEnd = GetTickCountRVC();
  828. //oilyang@20190828 add
  829. //升级重启后,在10分钟内,只要离最后一次交互时间大于2分钟,重启框架
  830. if ((g_bInUpgrade && ((dwTimeEnd - dwUpgradeTmpBegin) < MAX_WAIT_TIME_TO_RESTART))
  831. && (!g_bAuthSuc && ((dwTimeEnd - dwTmpBegin) > MAX_WAIT_TIME_TO_RESTART_UPGRADE)))
  832. {
  833. LogSingleMsg("**in upgrade restart**,to restart framework.");
  834. FrameworkShutdown(true,true);
  835. }
  836. if ((dwTimeEnd-dwTmpBegin) > MAX_WAIT_TIME_TO_RESTART)
  837. {
  838. LogSingleMsg("framework maybe down.");
  839. string strRunInfo;
  840. if (ReadRuninfoContent(strRunInfo))
  841. {
  842. LogSingleMsg("to check if need rollback.");
  843. LogSingleMsg((char*)strRunInfo.c_str());
  844. if (strRunInfo.compare("111") == 0 && IsStartTimeFileExist())
  845. {
  846. EnterCriticalSectionRVC(g_cs);
  847. g_needToRollBack = 1;
  848. LeaveCriticalSectionRVC(g_cs);
  849. }
  850. }
  851. else
  852. LogSingleMsg("read run info failed.");
  853. if (g_needToRollBack)
  854. {
  855. LogSingleMsg("after upgrade,time elapse,but can't wait the shake hands,so rollback.");
  856. FrameworkRollBack();
  857. }
  858. else
  859. {
  860. //framework is down,to restart it! oilyang 20150413
  861. FrameworkShutdown();
  862. }
  863. }
  864. }
  865. else
  866. LeaveCriticalSectionRVC(g_cs);
  867. }
  868. //_endthreadex(0);
  869. EndThreadRVC();
  870. return 0;
  871. }
  872. unsigned int __stdcall DoNetControl(void *pData)
  873. {
  874. ServerReportEvent("DoNetControl start");
  875. HANDLE hSnapshot;
  876. //find spshell.exe
  877. hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  878. if (hSnapshot)
  879. {
  880. PROCESSENTRY32 pe;
  881. pe.dwSize = sizeof(pe);
  882. if (Process32First(hSnapshot, &pe))
  883. {
  884. do {
  885. if (_stricmp(&pe.szExeFile[0], "spshell.exe") == 0)
  886. {
  887. //ServerReportEvent("find spshell.exe");
  888. LogToFile(false, false, false, "find spshell.exe");
  889. //DWORD dwExit = 0;
  890. //do
  891. //{
  892. // GetExitCodeProcess(&pe.th32ProcessID,&dwExit);
  893. // if (dwExit != STILL_ACTIVE)
  894. // break;
  895. //}while(1);
  896. HANDLE hP = OpenProcess( PROCESS_ALL_ACCESS, FALSE, pe.th32ProcessID );
  897. WaitForSingleObject(hP, INFINITE );
  898. LogToFile(false, false, false, "spshell.exe quit");
  899. //ServerReportEvent("spshell.exe quit");
  900. NetControl();
  901. break;
  902. }
  903. } while (Process32Next(hSnapshot, &pe));
  904. }
  905. CloseHandle(hSnapshot);
  906. }
  907. else
  908. ServerReportEvent("create snapshot failed.");
  909. //_endthreadex(0);
  910. EndThreadRVC();
  911. return 0;
  912. }
  913. #endif //linux
  914. unsigned int InitListenSocketRVC()
  915. {
  916. #ifdef linux
  917. socklen_t clilen;
  918. epfd = epoll_create(256);
  919. struct sockaddr_in clientaddr;
  920. struct sockaddr_in serveraddr;
  921. g_sListen = socket(AF_INET, SOCK_STREAM, 0);
  922. //setnonblocking(listenfd);
  923. ev.data.fd = g_sListen;
  924. ev.events = EPOLLIN | EPOLLET;
  925. epoll_ctl(epfd, EPOLL_CTL_ADD, g_sListen, &ev);
  926. memset(&serveraddr, 0, sizeof(serveraddr));
  927. serveraddr.sin_family = AF_INET;
  928. char* local_addr = "127.0.0.1";
  929. inet_aton(local_addr, &(serveraddr.sin_addr));
  930. serveraddr.sin_port = htons(default_port);
  931. bind(g_sListen, (sockaddr*)& serveraddr, sizeof(serveraddr));
  932. listen(g_sListen, LISTENQ);
  933. cout << "listen on:" << g_sListen << endl;
  934. return g_sListen;
  935. #else
  936. WSADATA wsaData;
  937. int iResult;
  938. struct addrinfo* result = NULL;
  939. struct addrinfo hints;
  940. int iSendResult;
  941. char recvbuf[DEFAULT_BUFLEN];
  942. int recvbuflen = DEFAULT_BUFLEN;
  943. // Initialize Winsock
  944. iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
  945. if (iResult != 0) {
  946. LogToFile(false, true, false, "WSAStartup failed with error:", "", iResult);
  947. return 1;
  948. }
  949. ZeroMemory(&hints, sizeof(hints));
  950. hints.ai_family = AF_INET;
  951. hints.ai_socktype = SOCK_STREAM;
  952. hints.ai_protocol = IPPROTO_TCP;
  953. hints.ai_flags = AI_PASSIVE;
  954. // Resolve the server address and port
  955. iResult = getaddrinfo(NULL, DEFAULT_PORT, &hints, &result);
  956. if (iResult != 0) {
  957. LogToFile(false, true, false, "getaddrinfo failed with error:", "", iResult);
  958. WSACleanup();
  959. return 1;
  960. }
  961. // get system info to create work thread
  962. SYSTEM_INFO SystemInfo;
  963. GetSystemInfo(&SystemInfo);
  964. // Create a SOCKET for connecting to server
  965. g_sListen = socket(result->ai_family, result->ai_socktype, result->ai_protocol);
  966. if (g_sListen == INVALID_SOCKET) {
  967. LogToFile(false, true, false, "socket failed with error:", "", WSAGetLastError());
  968. freeaddrinfo(result);
  969. WSACleanup();
  970. return 1;
  971. }
  972. // Setup the TCP listening socket
  973. struct sockaddr_in* sockaddr_ipv4 = (struct sockaddr_in*) result->ai_addr;
  974. LogToFile(true, false, false, "ip:", inet_ntoa(sockaddr_ipv4->sin_addr));
  975. iResult = bind(g_sListen, result->ai_addr, (int)result->ai_addrlen);
  976. if (iResult == SOCKET_ERROR) {
  977. printf("bind failed with error: %d\n", WSAGetLastError());
  978. freeaddrinfo(result);
  979. closesocket(g_sListen);
  980. WSACleanup();
  981. return 1;
  982. }
  983. freeaddrinfo(result);
  984. LogToFile(false, false, false, "to listen.");
  985. iResult = listen(g_sListen, SOMAXCONN);
  986. if (iResult == SOCKET_ERROR) {
  987. printf("listen failed with error: %d\n", WSAGetLastError());
  988. closesocket(g_sListen);
  989. WSACleanup();
  990. return 1;
  991. }
  992. LogToFile(false, false, false, "listen ok.");
  993. return g_sListen;
  994. #endif //linux
  995. }
  996. void AcceptReqRVC()
  997. {
  998. #ifdef linux
  999. StartDoWorkRVC();
  1000. cout << "AcceptReqRVC:after StartDoWorkRVC" << endl;
  1001. int maxi, connfd, sockfd, nfds;
  1002. struct epoll_event events[20];
  1003. struct sockaddr_in clientaddr;
  1004. socklen_t clilen = sizeof(struct sockaddr);
  1005. ssize_t n;
  1006. maxi = 0;
  1007. char line[MAXLINE];
  1008. for (; ; ) {
  1009. //等待epoll事件的发生
  1010. nfds = epoll_wait(epfd, events, 20, 500);
  1011. //处理所发生的所有事件
  1012. for (int i = 0; i < nfds; ++i)
  1013. {
  1014. if (events[i].data.fd == g_sListen)//new connection
  1015. {
  1016. connfd = accept(g_sListen, (sockaddr*)& clientaddr, &clilen);
  1017. if (connfd < 0) {
  1018. perror("connfd<0");
  1019. exit(1);
  1020. }
  1021. //setnonblocking(connfd);
  1022. char* str = inet_ntoa(clientaddr.sin_addr);
  1023. cout << "accapt a connection from " << str << endl;
  1024. ev.data.fd = connfd;
  1025. ev.events = EPOLLIN | EPOLLET;
  1026. //ev.events=EPOLLIN;
  1027. epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
  1028. }
  1029. else if (events[i].events & EPOLLIN)
  1030. {
  1031. cout << "EPOLLIN,i:" << i << ",sock:" << events[i].data.fd << endl;
  1032. if ((sockfd = events[i].data.fd) < 0)
  1033. continue;
  1034. if ((n = read(sockfd, line, MAXLINE)) < 0) {
  1035. if (errno == ECONNRESET) {
  1036. close(sockfd);
  1037. events[i].data.fd = -1;
  1038. }
  1039. else
  1040. std::cout << "readline error" << strerror(errno) << std::endl;
  1041. }
  1042. else if (n == 0) {
  1043. close(sockfd);
  1044. events[i].data.fd = -1;
  1045. }
  1046. DataProcessLinux(sockfd,line);
  1047. line[n] = '\0';
  1048. cout << "read " << line << endl;
  1049. ev.data.fd = sockfd;
  1050. ev.events = EPOLLOUT | EPOLLET;
  1051. epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
  1052. }
  1053. else if (events[i].events & EPOLLOUT)
  1054. {
  1055. cout << "epoll out" << endl;
  1056. sockfd = events[i].data.fd;
  1057. write(sockfd, line, n);
  1058. ev.data.fd = sockfd;
  1059. ev.events = EPOLLIN | EPOLLET;
  1060. epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev);
  1061. }
  1062. }
  1063. }
  1064. #else
  1065. HANDLE hCompletionPort;
  1066. hCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
  1067. // get system info to create work thread
  1068. SYSTEM_INFO SystemInfo;
  1069. GetSystemInfo(&SystemInfo);
  1070. for (int i = 0; i < SystemInfo.dwNumberOfProcessors * 2; i++)
  1071. {
  1072. HANDLE hProcessIO = CreateThread(NULL, 0, ProcessIO, hCompletionPort, 0, NULL);
  1073. if (hProcessIO)
  1074. CloseHandle(hProcessIO);
  1075. }
  1076. //waiting thread
  1077. //_beginthreadex(NULL,0,DoWork,NULL,0,NULL);
  1078. StartDoWorkRVC();
  1079. SOCKET sClient;
  1080. LPPER_HANDLE_DATA PerHandleData;
  1081. LPPER_IO_OPERATION_DATA PerIoData;
  1082. while (true)
  1083. {
  1084. //sClient = WSAAccept(sListen, NULL, NULL, NULL, 0);
  1085. sClient = accept(g_sListen, NULL, NULL);
  1086. //cout << "Socket " << sClient << "connect" << endl;
  1087. PerHandleData = new PER_HANDLE_DATA();
  1088. PerHandleData->Socket = sClient;
  1089. // client completion port
  1090. CreateIoCompletionPort((HANDLE)sClient, hCompletionPort, (DWORD)PerHandleData, 0);
  1091. //
  1092. PerIoData = new PER_IO_OPERATION_DATA();
  1093. ZeroMemory(PerIoData, sizeof(PER_IO_OPERATION_DATA));
  1094. PerIoData->DataBuf.buf = PerIoData->Buffer;
  1095. PerIoData->DataBuf.len = DATA_BUFSIZE;
  1096. // WSARecv
  1097. DWORD Flags = 0;
  1098. DWORD dwRecv = 0;
  1099. WSARecv(sClient, &PerIoData->DataBuf, 1, &dwRecv, &Flags, &PerIoData->Overlapped, NULL);
  1100. LogToFile(false, false, false, "hold");
  1101. }
  1102. DWORD dwByteTrans;
  1103. PostQueuedCompletionStatus(hCompletionPort, dwByteTrans, 0, 0);
  1104. closesocket(g_sListen);
  1105. #endif //linux
  1106. }
  1107. void CheckGuardianDbgDirAndCreateDbgFile(bool bChangeDay = false)
  1108. {
  1109. #ifdef linux
  1110. //todo oiltestlinux
  1111. string guardianLogDir("/opt/rvc/dbg/guardian");
  1112. if (opendir(guardianLogDir.c_str()) == NULL)
  1113. {
  1114. ServerReportEvent("Can't find guardian dbg dir,to create it.");
  1115. mkdir(guardianLogDir.c_str(), 0775);
  1116. }
  1117. #else
  1118. ZeroMemory(currDirBuf, sizeof(currDirBuf));
  1119. ZeroMemory(chDisk, sizeof(chDisk));
  1120. GetCurrentDirectory(256, currDirBuf);
  1121. chDisk[0] = currDirBuf[0];
  1122. // C:\Run\version\1.7.0.1
  1123. string strDir(chDisk), guardianLogDir(""), strGuardian("\\rvc\\dbg\\guardian");
  1124. guardianLogDir = strDir + ":" + strGuardian;
  1125. WIN32_FIND_DATA wfd;
  1126. bool rValue = false;
  1127. HANDLE hFind = FindFirstFile(guardianLogDir.c_str(), &wfd);
  1128. if ((hFind != INVALID_HANDLE_VALUE) && (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
  1129. {
  1130. ServerReportEvent("find guardian log dir.");
  1131. }
  1132. else
  1133. {
  1134. ServerReportEvent("can't find guardian log dir:");
  1135. int ret = _mkdir(guardianLogDir.c_str());
  1136. if (ret != 0)
  1137. {
  1138. ServerReportEvent("Create guardian log dir failed.");
  1139. ServerReportEvent(guardianLogDir.c_str());
  1140. return;
  1141. }
  1142. else
  1143. ServerReportEvent("Create guardian log dir suc.");
  1144. }
  1145. FindClose(hFind);
  1146. #endif //linux
  1147. SYSTEMTIME st;
  1148. GetLocalTimeRVC(st);
  1149. g_startDay = st.wDay;
  1150. char date[12];
  1151. memset(date,0,12);
  1152. sprintf(date, "%.4d%.2d%.2d", st.wYear, st.wMonth, st.wDay);
  1153. string strLogFile(""), strDate(date);
  1154. strLogFile = guardianLogDir + "/" + date + ".log";
  1155. if (!bChangeDay)
  1156. LogSingleMsg(strLogFile.c_str());
  1157. logFile.open(strLogFile.c_str(), std::ofstream::binary | ios::app);
  1158. if (logFile.is_open())
  1159. {
  1160. if (!bChangeDay)
  1161. LogSingleMsg("open guardian dbg file suc.");
  1162. g_bLogFileOpen = true;
  1163. if (!bChangeDay){
  1164. LogToFile(false, false, false, "==============guardian start==============");
  1165. LogToFile(false, false, false, strLogFile.c_str());
  1166. }
  1167. }
  1168. else
  1169. {
  1170. if (!bChangeDay){
  1171. LogSingleMsg("open guardian dbg file failed.",errno);
  1172. LogSingleMsg(strerror(errno));
  1173. }
  1174. }
  1175. }
  1176. int main()
  1177. {
  1178. //attach file descriptors 0,1,2 to /dev/null(for deamon app)
  1179. int fd0, fd1, fd2;
  1180. fd0 = open("/dev/null", O_RDWR);
  1181. fd1 = dup(0);
  1182. fd2 = dup(0);
  1183. g_bFrameOnline = true;
  1184. #ifdef linux
  1185. pthread_mutex_init(&g_cs_event,NULL);
  1186. pthread_mutex_init(&g_cs_log,NULL);
  1187. pthread_mutex_init(&g_cs,NULL);
  1188. #else
  1189. InitializeCriticalSectionAndSpinCount(&g_cs_event, 100);
  1190. InitializeCriticalSectionAndSpinCount(&g_cs_log, 100);
  1191. InitializeCriticalSectionAndSpinCount(&g_cs, 100);
  1192. g_logHandle = CreateEventA(NULL, FALSE, FALSE, NULL);
  1193. #endif
  1194. g_dwTimeBegin = GetTickCountRVC();
  1195. int i=1000;
  1196. CheckGuardianDbgDirAndCreateDbgFile();
  1197. StartDoNetControlRVC();
  1198. LogSingleMsg(currDirBuf);
  1199. InitListenSocketRVC();
  1200. AcceptReqRVC();
  1201. }
  1202. void NetControl()
  1203. {
  1204. return;//oiltmp 20150724 for zl don't forbid net for now
  1205. #ifdef linux
  1206. //todo oiltestlinux
  1207. #else
  1208. HINSTANCE hmod = ::LoadLibrary(DLL_PATH);
  1209. if(hmod != NULL)
  1210. {
  1211. LogSingleMsg("NetControl start");
  1212. pWlanConnect WlanConnect = (pWlanConnect)GetProcAddress(hmod, "WlanConnect");
  1213. pDisableBluetooth BluetoothConnect = (pDisableBluetooth)GetProcAddress(hmod, "DisableBluetooth");
  1214. if (WlanConnect != NULL && BluetoothConnect != NULL)
  1215. {
  1216. DWORD dwRetWire,dwRetEther,dwRet3G,dwRet4G,dwRetBT;
  1217. dwRetWire = dwRetEther = dwRet3G = dwRet4G = dwRetBT = 0;
  1218. dwRetWire = WlanConnect(FALSE, NET_TYPE_WIRELESS);
  1219. LogSingleMsg("after wireless");
  1220. dwRetEther = WlanConnect(FALSE, NET_TYPE_ETHERNET);
  1221. LogSingleMsg("after ether");
  1222. dwRet3G = WlanConnect(FALSE, NET_TYPE_3G);
  1223. LogSingleMsg("after 3g");
  1224. dwRet4G = WlanConnect(FALSE, NET_TYPE_4G);
  1225. LogSingleMsg("after 4g");
  1226. dwRetBT = BluetoothConnect(TRUE);
  1227. LogSingleMsg("after bluetooth");
  1228. char buf[128];
  1229. ZeroMemory(buf,sizeof(buf));
  1230. sprintf(buf,"Wire Net open state:Wireless[%d],Ethernet[%d],3G[%d],4G[%d],Blue tooth[%d]",dwRetWire,dwRetEther,dwRet3G,dwRet4G,dwRetBT);
  1231. LogSingleMsg(buf);
  1232. }
  1233. else
  1234. LogSingleMsg("Get function from CmbPadDll failed!");
  1235. }
  1236. else
  1237. LogSingleMsg("load CmbPadDll failed!");
  1238. #endif //linux
  1239. }
  1240. void LogDebug(bool bRevMsg, bool bCode1,bool bCode2,const char *msg, const char *revmsg, int code1, int code2)
  1241. {
  1242. SYSTEMTIME st;
  1243. GetLocalTimeRVC(st);
  1244. memset(g_allMsg,0,1024);
  1245. if (bRevMsg)
  1246. {
  1247. if (bCode1)
  1248. {
  1249. if (bCode2)
  1250. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s,%s,%d,%d\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, revmsg, code1, code2);
  1251. else
  1252. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s,%s %d\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, revmsg, code1);
  1253. }
  1254. else
  1255. {
  1256. if (bCode2)
  1257. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s,%s %d\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, revmsg,code2);
  1258. else
  1259. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s,%s\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, revmsg);
  1260. }
  1261. }
  1262. else
  1263. {
  1264. if (bCode1)
  1265. {
  1266. if (bCode2)
  1267. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s,%d,%d\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, code1, code2);
  1268. else
  1269. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s %d\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, code1);
  1270. }
  1271. else
  1272. {
  1273. if (bCode2)
  1274. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s %d\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg, code2);
  1275. else
  1276. sprintf(g_allMsg, "[%.2d:%.2d:%.2d.%.3d][DEBUG]%s\r\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, msg);
  1277. }
  1278. }
  1279. if (g_bLogFileOpen)
  1280. {
  1281. logFile.write(g_allMsg, strlen(g_allMsg));
  1282. logFile.flush();
  1283. }
  1284. }
  1285. void LogSingleMsg(const char *msg, int type)
  1286. {
  1287. #ifdef linux
  1288. SYSTEMTIME st;
  1289. GetLocalTimeRVC(st);
  1290. #endif
  1291. LogToFile(false, false, false, msg,"",0,0,type);
  1292. }
  1293. void LogToFile(bool bRevMsg, bool bCode1,bool bCode2,const char *msg, const char *revmsg, int code1, int code2, int type)
  1294. {
  1295. SYSTEMTIME st;
  1296. GetLocalTimeRVC(st);
  1297. EnterCriticalSectionRVC(g_cs_event);
  1298. //LogDebug(true, true, true, "oil", "test", g_startDay, st.wDay);
  1299. if (g_startDay != st.wDay)//switch to new day log
  1300. {
  1301. ServerReportEvent("change date");
  1302. logFile.close();
  1303. CheckGuardianDbgDirAndCreateDbgFile(true);
  1304. }
  1305. switch (type)
  1306. {
  1307. case 0:
  1308. default:
  1309. LogDebug(bRevMsg, bCode1,bCode2,msg, revmsg, code1, code2);
  1310. break;
  1311. }
  1312. LeaveCriticalSectionRVC(g_cs_event);
  1313. }
  1314. //unsigned int __stdcall DoLog(void *pData)
  1315. //{
  1316. // while (1)
  1317. // {
  1318. // cout << "wait..." << endl;
  1319. // WaitForSingleObject(g_logHandle, INFINITE);
  1320. // if (g_bLogFileOpen)
  1321. // {
  1322. // EnterCriticalSection(&g_cs_log);
  1323. // logFile.write(g_allMsg, strlen(g_allMsg));
  1324. // logFile.flush();
  1325. // LeaveCriticalSection(&g_cs_log);
  1326. // }
  1327. // }
  1328. //
  1329. // //_endthreadex(0);
  1330. // EndThreadRVC();
  1331. // return 0;
  1332. //}