SpEntityPrivilege.cpp 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326
  1. #include "stdafx.h"
  2. #include "SpBase.h"
  3. #include "SpTimer.h"
  4. #include "SpMisc.h"
  5. #include "SpEntity.h"
  6. #include "SpModule.h"
  7. #include "SpEntityPrivilege.h"
  8. #include "SpAsyncWait.h"
  9. #include "sp_iom.h"
  10. #include "sp_svc.h"
  11. #include "sp_log.h"
  12. #include "sp_var.h"
  13. #include "sp_def.h"
  14. #include "sp_cfg.h"
  15. #include "sp_env.h"
  16. #include "memutil.h"
  17. #include "strutil.h"
  18. #include "fileutil.h"
  19. #include "shm_array.h"
  20. #include "shm_mem.h"
  21. #include "iobuffer.h"
  22. #include "iniutil.h"
  23. #include "def.h"
  24. #include "dbgutil.h"
  25. #include <algorithm>
  26. #include "sp_httpDefine.h"
  27. #ifdef RVC_OS_WIN
  28. #include "CodeSignVerify.h"
  29. #include <io.h>
  30. #include "log_producer_config.h"
  31. #else
  32. #include "sp_dbg_export.h"
  33. #include <winpr/ini.h>
  34. #include <winpr/string.h>
  35. #include <winpr/crt.h>
  36. #include <winpr/file.h>
  37. #endif //RVC_OS_WIN
  38. static ErrorCodeEnum ControlEntity(
  39. SpEntity *pEntity,
  40. const char *pszEntityName,
  41. int call_type,
  42. const char *cmd_line,
  43. //don't change this type bcz: 1. ipc would not convey pointer-type, and receiver may receive as 4bytes type. [4/2/2020 11:13 Gifur]
  44. int param1, int param2,
  45. CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  46. {
  47. if (!pszEntityName) {
  48. return Error_Null;
  49. } else if (strlen(pszEntityName) == 0) {
  50. return Error_Param;
  51. }
  52. sp_env_t *env = sp_get_env();
  53. sp_entity_t *ent = sp_mod_mgr_find_entity_by_name(env->mod_mgr, pszEntityName);
  54. if (!ent)
  55. return Error_NotExist;
  56. iobuffer_t *req_pkt = iobuffer_create(-1, -1);
  57. int v = ent->cfg->idx;
  58. bool fetch_user_code = false;
  59. iobuffer_write(req_pkt, IOBUF_T_I4, &v, 0);
  60. if (call_type == SHELL_CMD_REQ_ENTITY_START)
  61. iobuffer_write(req_pkt, IOBUF_T_STR, cmd_line, -1);
  62. else if (call_type == SHELL_CMD_REQ_ENTITY_TEST) {
  63. /** prohibit user from sending test call in the absence of test mode [5/20/2020 Gifur] */
  64. if (param1 == 1/*Test_Examine*/ && !is_own_test_mode( env->cfg->args->test_mode)) {
  65. iobuffer_destroy(req_pkt);
  66. return Error_NoPrivilege;
  67. }
  68. fetch_user_code = true;
  69. iobuffer_write(req_pkt, IOBUF_T_I4, &param1, 0);
  70. }
  71. SpAsyncWaitRPC *pAsyncWait = new SpAsyncWaitRPC(pEntity, &req_pkt, call_type, fetch_user_code);
  72. ErrorCodeEnum Error = pAsyncWait->Begin();
  73. if (Error == Error_Succeed) {
  74. pAsynWaitSp.Attach(pAsyncWait, pAsyncWait->GetRefCountPtr());
  75. }
  76. pAsyncWait->DecrementRef(); // xkm@20150115
  77. if (req_pkt)
  78. iobuffer_dec_ref(req_pkt);
  79. return Error;
  80. }
  81. SpEntityPrivilege::SpEntityPrivilege(SpModule *pModule, sp_entity_t *ent, sp_cfg_shell_entity_t *cfg_ent, CEntityBase *pEntityBase)
  82. : SpEntity(pModule, ent, cfg_ent, pEntityBase), m_pEntityLifeListener(NULL)
  83. {
  84. spinlock_init(&m_maplock);
  85. m_pEntityStateListenerMap = stringmap_create(0);
  86. }
  87. SpEntityPrivilege::~SpEntityPrivilege()
  88. {
  89. UnregistEntityStateEvent(NULL);
  90. stringmap_destroy(m_pEntityStateListenerMap);
  91. }
  92. ErrorCodeEnum SpEntityPrivilege::Init()
  93. {
  94. ErrorCodeEnum Error = SpEntity::Init();
  95. return Error;
  96. }
  97. void SpEntityPrivilege::Term()
  98. {
  99. //....
  100. SpEntity::Term();
  101. UnregistEntityStateEvent(NULL); // unregister all
  102. }
  103. CSmartPointer<IEntityFunctionPrivilege> SpEntityPrivilege::GetPrivilegeFunction()
  104. {
  105. IEntityFunctionPrivilege *pPrivilege = dynamic_cast<IEntityFunctionPrivilege*>(this);
  106. return pPrivilege;
  107. }
  108. ErrorCodeEnum SpEntityPrivilege::StartEntity(const char *pszEntityName, const char *pszCmdLine,CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  109. {
  110. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_START, pszCmdLine, 0, 0, pAsynWaitSp);
  111. }
  112. /*To Close*/
  113. ErrorCodeEnum SpEntityPrivilege::StopEntity(const char *pszEntityName,CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  114. {
  115. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_STOP, NULL, 0, 0, pAsynWaitSp);
  116. }
  117. /*To Close*/
  118. ErrorCodeEnum SpEntityPrivilege::CloseEntity(const char *pszEntityName,CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  119. {
  120. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_STOP, NULL, 0, 0, pAsynWaitSp);
  121. }
  122. /*To kill*/
  123. ErrorCodeEnum SpEntityPrivilege::TerminateEntity(const char *pszEntityName,CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  124. {
  125. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_TERMINATE, NULL, 0, 0, pAsynWaitSp);
  126. }
  127. ErrorCodeEnum SpEntityPrivilege::PauseEntity(const char *pszEntityName,CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  128. {
  129. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_PAUSE, NULL, 0, 0, pAsynWaitSp);
  130. }
  131. ErrorCodeEnum SpEntityPrivilege::ContinueEntity(const char *pszEntityName,CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  132. {
  133. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_CONTINUE, NULL, 0, 0, pAsynWaitSp);
  134. }
  135. ErrorCodeEnum SpEntityPrivilege::TestEntity(const char *pszEntityName,EntityTestEnum eTestType, CSmartPointer<IAsynWaitSp> &pAsynWaitSp)
  136. {
  137. return ControlEntity(this, pszEntityName, SHELL_CMD_REQ_ENTITY_TEST, NULL, (int)eTestType, 0, pAsynWaitSp);
  138. }
  139. ErrorCodeEnum SpEntityPrivilege::RegistEntityLifeEvent(IEntityLifeListener *pListener)
  140. {
  141. if (!m_pEntityLifeListener) {
  142. sp_mod_entity_life_cb cb;
  143. cb.on_entity_close = &SpEntityPrivilege::__on_entity_close;
  144. cb.on_entity_create = &SpEntityPrivilege::__on_entity_create;
  145. cb.on_entity_exception = &SpEntityPrivilege::__on_entity_exception;
  146. cb.user_data = this;
  147. int rc = sp_mod_entity_life_listener_create(&cb, m_svc, pListener, &m_pEntityLifeListener);
  148. return SpTranslateError(rc);
  149. } else {
  150. return Error_Duplication;
  151. }
  152. }
  153. ErrorCodeEnum SpEntityPrivilege::UnregistLiftEvent()
  154. {
  155. if (!m_pEntityLifeListener) {
  156. return Error_NotInit;
  157. } else {
  158. sp_mod_entity_life_listener_destroy(m_pEntityLifeListener);
  159. m_pEntityLifeListener = NULL;
  160. return Error_Succeed;
  161. }
  162. }
  163. ErrorCodeEnum SpEntityPrivilege::RegistEntityStateEvent(const char *pszEntityName,IEntityStateListener *pListener)
  164. {
  165. sp_env_t *env = sp_get_env();
  166. int target_ent_id;
  167. if (!pListener)
  168. return Error_Null;
  169. if (pszEntityName) {
  170. sp_entity_t *ent;
  171. ent = sp_mod_mgr_find_entity_by_name(env->mod_mgr, pszEntityName);
  172. if (!ent)
  173. return Error_NotExist;
  174. target_ent_id = ent->cfg->idx;
  175. } else {
  176. LOG_TRACE("registered state event for all entities!");
  177. target_ent_id = -1; // regist for all entities
  178. pszEntityName = "*";
  179. }
  180. ErrorCodeEnum Error = Error_Succeed;
  181. spinlock_enter(&m_maplock, -1);
  182. if (stringmap_find(m_pEntityStateListenerMap, pszEntityName) ){
  183. Error = Error_AlreadyExist;
  184. } else {
  185. int rc;
  186. sp_mod_entity_state_listener_t *listener;
  187. sp_mod_entity_event_cb cb;
  188. cb.on_close_connection = &SpEntityPrivilege::__on_close_connection;
  189. cb.on_create_connection = &SpEntityPrivilege::__on_create_connection;
  190. cb.on_user_state = &SpEntityPrivilege::__on_user_state;;
  191. cb.on_entity_state = &SpEntityPrivilege::__on_entity_state;
  192. cb.user_data = this;
  193. rc = sp_mod_entity_listener_create(target_ent_id, &cb, m_svc, pListener, &listener);
  194. if (rc == 0) {
  195. stringmap_add(m_pEntityStateListenerMap, pszEntityName ? pszEntityName : "*", listener);
  196. } else {
  197. Error = SpTranslateError(rc);
  198. }
  199. }
  200. spinlock_leave(&m_maplock);
  201. return Error;
  202. }
  203. ErrorCodeEnum SpEntityPrivilege::UnregistEntityStateEvent(const char *pszEntityName)
  204. {
  205. ErrorCodeEnum Error = Error_Succeed;
  206. if (pszEntityName == NULL) {
  207. pszEntityName = "*";
  208. }
  209. spinlock_enter(&m_maplock, -1);
  210. if (pszEntityName) {
  211. stringmap_kv_pair *kvp;
  212. kvp = stringmap_find(m_pEntityStateListenerMap, pszEntityName);
  213. if (kvp) {
  214. sp_mod_entity_state_listener_t *listener;
  215. listener = (sp_mod_entity_state_listener_t *)stringmap_kv_pair_get_value(kvp);
  216. stringmap_remove(m_pEntityStateListenerMap, pszEntityName);
  217. sp_mod_entity_listener_destroy(listener);
  218. } else {
  219. Error = Error_NotExist;
  220. }
  221. } else { // unregist all
  222. stringmap_iterator *it = stringmap_iterator_create(m_pEntityStateListenerMap);
  223. while (stringmap_iterator_next(it) == 0) {
  224. const char *ent_name = (const char*)stringmap_iterator_get_key(it);
  225. sp_mod_entity_state_listener_t *listener = (sp_mod_entity_state_listener_t *)stringmap_iterator_get_value(it);
  226. stringmap_remove(m_pEntityStateListenerMap, ent_name);
  227. sp_mod_entity_listener_destroy(listener);
  228. }
  229. stringmap_iterator_destroy(it);
  230. }
  231. spinlock_leave(&m_maplock);
  232. return Error;
  233. }
  234. ErrorCodeEnum SpEntityPrivilege::GetSpecifiedEntityState(const char* pszEntityName, EntityStateEnum& eEntityState, DWORD& dwUserState)
  235. {
  236. if (pszEntityName == NULL) {
  237. return Error_Param;
  238. }
  239. sp_env_t* env = sp_get_env();
  240. sp_entity_t* ent;
  241. if (strlen(pszEntityName) == 0) {
  242. ent = m_ent;
  243. } else {
  244. ent = sp_mod_mgr_find_entity_by_name(env->mod_mgr, pszEntityName);
  245. }
  246. if (!ent) {
  247. return Error_NotExist;
  248. }
  249. eEntityState = (EntityStateEnum)ent->state;
  250. dwUserState = ent->user_state;
  251. return Error_Succeed;
  252. }
  253. ErrorCodeEnum SpEntityPrivilege::DisplayBlueScreen(const char *pszTitle)
  254. {
  255. ErrorCodeEnum Error;
  256. iobuffer_t *pkt = iobuffer_create(-1, -1);
  257. iobuffer_write(pkt, IOBUF_T_STR, pszTitle, -1);
  258. //int nBlueScreen = 1;
  259. //iobuffer_write(pkt, IOBUF_T_I4, &nBlueScreen, 0);
  260. Error = PostInfoShell(SHELL_CMD_INFO_BLUESCREEN_DISPLAY, &pkt);
  261. if (pkt)
  262. iobuffer_dec_ref(pkt);
  263. return Error;
  264. }
  265. ErrorCodeEnum SpEntityPrivilege::UndisplayBlueScreen()
  266. {
  267. ErrorCodeEnum Error;
  268. iobuffer_t *pkt = iobuffer_create(-1, -1);
  269. Error = PostInfoShell(SHELL_CMD_INFO_BLUESCREEN_UNDISPLAY, &pkt);
  270. if (pkt)
  271. iobuffer_dec_ref(pkt);
  272. return Error;
  273. }
  274. ErrorCodeEnum SpEntityPrivilege::Reboot(RebootTriggerEnum eTriggerReason,RebootWayEnum eWay)
  275. {
  276. ErrorCodeEnum Error;
  277. int v;
  278. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("call SpEntityPrivilege::Reboot, reason:%d, way:%d", eTriggerReason, eWay);
  279. iobuffer_t *pkt = iobuffer_create(-1, -1);
  280. v = eTriggerReason;
  281. iobuffer_write(pkt, IOBUF_T_I4, &v, 0);
  282. v = eWay;
  283. iobuffer_write(pkt, IOBUF_T_I4, &v, 0);
  284. Error = PostInfoShell(SHELL_CMD_INFO_MACHINE_REBOOT, &pkt);
  285. if (pkt)
  286. iobuffer_dec_ref(pkt);
  287. return Error;
  288. }
  289. bool SpEntityPrivilege::WriteInstallLog(const char *pszPackName, const char *pszLogText)
  290. {
  291. auto env = sp_get_env();
  292. // 创建安装包日志文件
  293. CSimpleStringA strInstallLogDir = CSimpleStringA::Format("%s" SPLIT_SLASH_STR "InstallLog", env->dir->root_runinfo_path);
  294. if (!ExistsDirA(strInstallLogDir))
  295. CreateDirA(strInstallLogDir, TRUE);
  296. // 写安装包日志
  297. CSimpleStringA strInstallLog = CSimpleStringA::Format("%s" SPLIT_SLASH_STR "%s.log", (const char*)strInstallLogDir, pszPackName);
  298. FILE *pInstallLog = fopen(strInstallLog, "a+");
  299. if (pInstallLog == NULL)
  300. {
  301. LogError(Severity_Low, Error_Unexpect, 0, CSimpleStringA::Format("open install log file [%s] to write fail", pszPackName));
  302. return false;
  303. }
  304. fprintf_s(pInstallLog, "[%s] %s\r\n", (const char*)CSmallDateTime::GetNow().ToTimeString(), pszLogText);
  305. fflush(pInstallLog);
  306. fclose(pInstallLog);
  307. return true;
  308. }
  309. bool SpEntityPrivilege::RecursiveCopyDir(const char *pszSourceDir, const char *pszDestDir, CSimpleStringA& strErrInfo)
  310. {
  311. array_header_t *arr;
  312. if (!ExistsDirA(pszSourceDir))
  313. {
  314. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("source dir [%s] not exists", pszSourceDir);
  315. return false;
  316. }
  317. if (!ExistsDirA(pszDestDir))
  318. {
  319. if (!CreateDirRecursiveA(pszDestDir))
  320. {
  321. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create dir [%s] fail: %d", pszDestDir, GetLastError());
  322. return false;
  323. }
  324. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create dir [%s] succ", pszDestDir);
  325. }
  326. arr = fileutil_get_sub_files_a(pszSourceDir);
  327. if (arr)
  328. {
  329. int i;
  330. for (i = 0; i < arr->nelts; ++i)
  331. {
  332. char szDestFile[256] = { 0 };
  333. char *file = ARRAY_IDX(arr, i, char*);
  334. strcpy(szDestFile, pszDestDir);
  335. if (szDestFile[strlen(szDestFile) - 1] != SPLIT_SLASH)
  336. strcat(szDestFile, SPLIT_SLASH_STR);
  337. strcat(szDestFile, strrchr(file, SPLIT_SLASH) + 1);
  338. if (!CopyFileA(file, szDestFile, FALSE))
  339. {
  340. strErrInfo = CSimpleStringA::Format("copy file [%s] fail: %d", file, GetLastError());
  341. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy file [%s] fail: %d", file, GetLastError());
  342. toolkit_array_free2(arr);
  343. return false;
  344. }
  345. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy file [%s] succ", file);
  346. }
  347. toolkit_array_free2(arr);
  348. }
  349. arr = fileutil_get_sub_dirs_a(pszSourceDir);
  350. if (arr)
  351. {
  352. int i;
  353. for (i = 0; i < arr->nelts; ++i)
  354. {
  355. char szDestSubDir[256] = { 0 };
  356. char *dir = ARRAY_IDX(arr, i, char*);
  357. strcpy(szDestSubDir, pszDestDir);
  358. if (szDestSubDir[strlen(szDestSubDir) - 1] != SPLIT_SLASH)
  359. strcat(szDestSubDir, SPLIT_SLASH_STR);
  360. strcat(szDestSubDir, strrchr(dir, SPLIT_SLASH) + 1);
  361. if (!RecursiveCopyDir(dir, szDestSubDir, strErrInfo))
  362. {
  363. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy dir [%s] fail: %d", dir, GetLastError());
  364. toolkit_array_free2(arr);
  365. return false;
  366. }
  367. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy dir [%s] succ", dir);
  368. }
  369. toolkit_array_free2(arr);
  370. }
  371. return true;
  372. }
  373. ErrorCodeEnum SpEntityPrivilege::UpdateActiveTxt(const char* strActiveFilePath, const char* toWriteVersionStr, bool toCheck)
  374. {
  375. HANDLE hFile = ::CreateFileA(strActiveFilePath, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  376. BOOL bSetSucc = FALSE;
  377. if (hFile == INVALID_HANDLE_VALUE)
  378. {
  379. LogError(Severity_Low, Error_Unexpect, 0, "open active.txt fail");
  380. return Error_Unexpect;
  381. }
  382. DWORD dwWrittenLen(0);
  383. bSetSucc = WriteFile(hFile, toWriteVersionStr, strlen(toWriteVersionStr), &dwWrittenLen, NULL);
  384. FlushFileBuffers(hFile);
  385. SetEndOfFile(hFile);
  386. CloseHandle(hFile);
  387. LOG_TRACE("write version [%s] succeed", toWriteVersionStr);
  388. //出现active文件为空的问题,此处增加写文件件后再读一次,确认是否写成功,以此来确认是否是写文件导致
  389. if (toCheck) {
  390. hFile = ::CreateFileA(strActiveFilePath, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  391. if (hFile == INVALID_HANDLE_VALUE)
  392. {
  393. LogError(Severity_Low, Error_Unexpect, 0, "open active.txt fail");
  394. return Error_Unexpect;
  395. }
  396. DWORD dwReadLen(0);
  397. char cVer[16] = { 0 };
  398. if (ReadFile(hFile, cVer, 16, &dwReadLen, NULL))
  399. {
  400. LOG_TRACE("read active.txt, version len[%d], version[%s]", dwReadLen, cVer);
  401. //如果为空,则重写
  402. if (strlen(cVer) == 0)
  403. {
  404. if (WriteFile(hFile, toWriteVersionStr, strlen(toWriteVersionStr), &dwWrittenLen, NULL))
  405. {
  406. LOG_TRACE("rewrite active.txt success, version[%s]", toWriteVersionStr);
  407. }
  408. else
  409. {
  410. LogError(Severity_Low, Error_Unexpect, 0, "rewrite active.txt fail");
  411. }
  412. }
  413. }
  414. else
  415. {
  416. LogError(Severity_Low, Error_Unexpect, 0, "Read active.txt fail");
  417. }
  418. FlushFileBuffers(hFile);
  419. SetEndOfFile(hFile);
  420. CloseHandle(hFile);
  421. }
  422. return bSetSucc ? Error_Succeed : Error_Unexpect;
  423. }
  424. ErrorCodeEnum CopyFolder(CSimpleStringA strSourcePath, CSimpleStringA strDestPath)
  425. {
  426. if (strSourcePath.IsNullOrEmpty() || strDestPath.IsNullOrEmpty())
  427. {
  428. return Error_Null;
  429. }
  430. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Start to copy dir %s to dir %s", (const char*)strSourcePath,(const char*)strDestPath);
  431. #ifdef _WIN32
  432. _finddata_t FileInfo;
  433. CSimpleStringA strfind = strSourcePath + "\\*";
  434. long Handle = _findfirst(strfind, &FileInfo);
  435. if (-1L == Handle)
  436. {
  437. _findclose(Handle);
  438. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("%s文件夹为空", strSourcePath);
  439. return Error_Succeed;
  440. }
  441. CSimpleStringA newPath;
  442. do{
  443. if (FileInfo.attrib & _A_SUBDIR)
  444. {
  445. if ((strcmp(FileInfo.name, ".") != 0) && (strcmp(FileInfo.name, "..") != 0))
  446. {
  447. newPath = strSourcePath + "\\" + FileInfo.name;
  448. CSimpleStringA newDestPath = strDestPath + "\\" + FileInfo.name;
  449. /*if (strcmp(FileInfo.name, "") == 0)
  450. {
  451. continue;
  452. }*/
  453. if (!ExistsDirA(newDestPath))
  454. {
  455. CreateDirA(newDestPath, TRUE);
  456. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Create Dir %s success", (const char*)newDestPath);
  457. }
  458. CopyFolder(newPath, newDestPath);
  459. }
  460. }
  461. else
  462. {
  463. CSimpleStringA strFindName = FileInfo.name;
  464. {
  465. CSimpleStringA strSourceFile = strSourcePath + "\\" + strFindName;
  466. CSimpleStringA strDestFile = strDestPath + "\\" + strFindName;
  467. if (ExistsFileA(strSourceFile))
  468. {
  469. // 先去除目标文件只读属性
  470. if (ExistsFileA(strDestFile))
  471. RemoveFileReadOnlyAttributeA(strDestFile);
  472. LOG_TRACE("copy file from [%s] to [%s]", (const char*)strSourceFile, (const char*)strDestFile);
  473. BOOL bRet = CopyFileA(strSourceFile, strDestFile, FALSE);
  474. // 去除只读属性
  475. if (bRet)
  476. RemoveFileReadOnlyAttributeA(strDestFile);
  477. else
  478. {
  479. // 覆盖失败,则生成.new副本
  480. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy file [%s] to [%s] fail, now save as .new file", (const char*) strSourcePath, (const char*) strDestFile);
  481. strDestFile += ".new";
  482. bRet = CopyFileA(strSourceFile, strDestFile, FALSE);
  483. if (bRet)
  484. {
  485. RemoveFileReadOnlyAttributeA(strDestPath);
  486. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy [%s] as [%s] succeed", (const char*)strSourceFile, (const char*) strDestFile);
  487. }
  488. else
  489. {
  490. LogError(Severity_Low, Error_Unexpect, 0, CSimpleStringA::Format("copy file [%s] as [%s] fail", (const char*) strSourceFile, (const char*) strDestFile));
  491. return Error_Block;
  492. }
  493. }
  494. }
  495. else
  496. {
  497. LogError(Severity_Low, Error_NotExist, 0, CSimpleStringA::Format("file [%s] to copy not exists", (const char*)strSourceFile));
  498. return Error_NotExist;
  499. }
  500. }
  501. }
  502. } while (_findnext(Handle, &FileInfo) == 0);
  503. _findclose(Handle);
  504. return Error_Succeed;
  505. #else
  506. if (CopyDirA(strSourcePath, strDestPath)) {
  507. return Error_Succeed;
  508. }
  509. return Error_Unexpect;
  510. #endif //_WIN32
  511. }
  512. CSimpleStringA GetFileDirectory(const char *pszFullPath)
  513. {
  514. int i=strlen(pszFullPath)-1;
  515. for( ; i>0 && pszFullPath[i]!=SPLIT_SLASH; i--)NULL;
  516. return CSimpleStringA(pszFullPath, i);
  517. }
  518. #ifdef _WIN32
  519. bool SpEntityPrivilege::IsWow64Process()
  520. {
  521. typedef BOOL(WINAPI *LPFN_ISWOW64PROCESS)(HANDLE, PBOOL);
  522. LPFN_ISWOW64PROCESS fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandle("kernel32"), "IsWow64Process");
  523. BOOL bIsWow64 = FALSE;
  524. if (NULL != fnIsWow64Process)
  525. {
  526. if (!fnIsWow64Process(GetCurrentProcess(), &bIsWow64))
  527. {
  528. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("call IsWow64Process error: %d", GetLastError());
  529. return false;
  530. }
  531. return bIsWow64 == TRUE;
  532. }
  533. else
  534. {
  535. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("get IsWow64Process error: %d", GetLastError());
  536. return false;
  537. }
  538. }
  539. #endif //_WIN32
  540. // result=0&msg=install ok
  541. bool SpEntityPrivilege::GetSysUpgradeResult(const char *pszResultLog, ErrorCodeEnum &eErrorCode, CSimpleStringA &strErrMsg)
  542. {
  543. // 设置初始值
  544. eErrorCode = Error_Succeed;
  545. strErrMsg = "";
  546. if (!ExistsFileA(pszResultLog))
  547. {
  548. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)(CSimpleStringA::Format("result log [%s] not exist", pszResultLog));
  549. strErrMsg = "result.log not exist";
  550. return false;
  551. }
  552. FILE *pResultLog = fopen(pszResultLog, "r");
  553. if (pResultLog == NULL)
  554. {
  555. LogError(Severity_Low, Error_Unexpect, 0, CSimpleStringA::Format("open result log [%s] fail", pszResultLog));
  556. strErrMsg = "open result.log fail";
  557. return false;
  558. }
  559. char szTmp[4096] = {};
  560. int nRead = fread_s(szTmp, sizeof(szTmp), 1, sizeof(szTmp), pResultLog);
  561. CSimpleStringA strResult = szTmp;
  562. fclose(pResultLog);
  563. pResultLog = NULL;
  564. auto arr = strResult.Split('&');
  565. for (int i = 0; i < arr.GetCount(); i++)
  566. {
  567. auto arr2 = arr[i].Split('=');
  568. if (arr2.GetCount() == 2)
  569. {
  570. if (stricmp(arr2[0], "result") == 0)
  571. eErrorCode = (ErrorCodeEnum)(DWORD)atoi(arr2[1]);
  572. else if (stricmp(arr2[0], "msg") == 0)
  573. strErrMsg = arr2[1];
  574. }
  575. }
  576. return true;
  577. }
  578. // File*A.*
  579. bool SpEntityPrivilege::IsFileMatch(const char *pszFilter, const char *pszFileName)
  580. {
  581. if (stricmp(pszFilter, "*") == 0 || stricmp(pszFilter, "*.*") == 0)
  582. return true;
  583. const char *p1 = pszFilter;
  584. const char *p2 = pszFileName;
  585. if (p1 == NULL)
  586. return true;
  587. if (p2 == NULL)
  588. return false;
  589. if (*p1 ==NULL && *p2 ==NULL)
  590. return true;
  591. else if (*p1 == NULL)
  592. return false;
  593. else if (*p2 == NULL)
  594. {
  595. // 查找*p1是否全是*
  596. while(*p1 == '*')
  597. p1++;
  598. if (*p1 == NULL)
  599. return true;
  600. else
  601. return false;
  602. }
  603. if (*p1 != '*')
  604. {
  605. if (tolower(*p1) != tolower(*p2))
  606. return false;
  607. else
  608. return IsFileMatch(p1+1, p2+1);
  609. }
  610. else
  611. {
  612. while(*++p1 == '*');
  613. if (*p1 == NULL)
  614. return true;
  615. while (*p2)
  616. {
  617. while(tolower(*p1) != tolower(*p2) && *++p2);
  618. if (*p2 == NULL)
  619. return false;
  620. if(IsFileMatch(p1+1, p2+1))
  621. return true;
  622. p2++;
  623. }
  624. return false;
  625. }
  626. }
  627. CSimpleStringA SpEntityPrivilege::GetFileDirectory(const char *pszFullPath)
  628. {
  629. int i=strlen(pszFullPath);
  630. for( ; i>0 && pszFullPath[i-1]!=SPLIT_SLASH; i--)NULL;
  631. return CSimpleStringA(pszFullPath, i);
  632. }
  633. ErrorCodeEnum SpEntityPrivilege::SetSysDebugLevel(const char *pszEntityName,DebugLevelEnum eDebugLevel,bool bPersist)
  634. {
  635. sp_env_t *env = sp_get_env();
  636. if (!pszEntityName)
  637. return Error_Null;
  638. if (stricmp(pszEntityName, "SpShell") == 0)
  639. env->cfg->shell_ini->shell_debug_level = (int)eDebugLevel;
  640. sp_cfg_shell_entity_t *cfg_ent = sp_cfg_get_entity_by_name(env->cfg, pszEntityName);
  641. if (cfg_ent) {
  642. int old = cfg_ent->debug_level;
  643. if (old != eDebugLevel) {
  644. cfg_ent->debug_level = (int)eDebugLevel;
  645. if (bPersist) {
  646. return SpTranslateError(sp_cfg_refresh_debug_level(env->cfg, cfg_ent));
  647. }
  648. }
  649. } else {
  650. return Error_NotExist;
  651. }
  652. return Error_Succeed;
  653. }
  654. ErrorCodeEnum SpEntityPrivilege::RefreshFrameworkState(FrameworkStateEnum eState)
  655. {
  656. int rc = Error_Succeed;
  657. char n[12] = { '\0' };
  658. sp_env_t* env = sp_get_env();
  659. sp_cfg_t* cfg = env->cfg;
  660. const int old_state = cfg->shell_ini->shell_state;
  661. if (old_state != (int)eState) {
  662. cfg->shell_ini->shell_state = eState;
  663. _itoa(eState, n, 10);
  664. rc = sp_var_client_set((sp_var_client_t*)m_var_client, VAR_RSERVERD_KEY_TERM_STATE, n, 0);
  665. if (rc == Error_Succeed) {
  666. iobuffer_t* pkt = iobuffer_create(-1, -1);
  667. int v1 = (int)eState;
  668. int v2 = (int)old_state;
  669. iobuffer_write(pkt, IOBUF_T_I4, &v1, 0);
  670. iobuffer_write(pkt, IOBUF_T_I4, &v2, 0);
  671. rc = PostInfoShell(SHELL_CMD_INFO_TERMINALSTAGE_CHANGE, &pkt);
  672. if (pkt) {
  673. iobuffer_dec_ref(pkt);
  674. }
  675. }
  676. }
  677. return SpTranslateError(rc);
  678. }
  679. ErrorCodeEnum SpEntityPrivilege::ShowOuputConsole()
  680. {
  681. ErrorCodeEnum Error;
  682. iobuffer_t *pkt = iobuffer_create(-1, -1);
  683. Error = PostInfoShell(SHELL_CMD_INFO_OUTPUT_CONSOLE_ON, &pkt);
  684. if (pkt)
  685. iobuffer_dec_ref(pkt);
  686. return Error;
  687. }
  688. ErrorCodeEnum SpEntityPrivilege::CloseOuputConsole()
  689. {
  690. ErrorCodeEnum Error;
  691. iobuffer_t *pkt = iobuffer_create(-1, -1);
  692. Error = PostInfoShell(SHELL_CMD_INFO_OUTPUT_CONSOLE_OFF, &pkt);
  693. if (pkt)
  694. iobuffer_dec_ref(pkt);
  695. return Error;
  696. }
  697. ErrorCodeEnum SpEntityPrivilege::BeginLogSend(const char* endpoint, const char* topicSys, const char* topicUser, const char* topicBeidou
  698. , const char* bussSys, const char* bussUser, const char* vtmWeb)
  699. {
  700. ErrorCodeEnum Error;
  701. iobuffer_t* pkt = iobuffer_create(-1, -1);
  702. iobuffer_write(pkt, IOBUF_T_STR, endpoint, -1);
  703. iobuffer_write(pkt, IOBUF_T_STR, topicSys, -1);
  704. iobuffer_write(pkt, IOBUF_T_STR, topicUser, -1);
  705. iobuffer_write(pkt, IOBUF_T_STR, topicBeidou, -1);
  706. iobuffer_write(pkt, IOBUF_T_STR, bussSys, -1);
  707. iobuffer_write(pkt, IOBUF_T_STR, bussUser, -1);
  708. iobuffer_write(pkt, IOBUF_T_STR, vtmWeb, -1);
  709. Error = PostInfoShell(SHELL_CMD_INFO_START_UPLOAD_LOG, &pkt);
  710. if (pkt)
  711. iobuffer_dec_ref(pkt);
  712. return Error;
  713. }
  714. ErrorCodeEnum SpEntityPrivilege::SetHttpToken(const char* channelId, const char* token)
  715. {
  716. ErrorCodeEnum Error;
  717. iobuffer_t* pkt = iobuffer_create(-1, -1);
  718. iobuffer_write(pkt, IOBUF_T_STR, channelId, -1);
  719. iobuffer_write(pkt, IOBUF_T_STR, token, -1);
  720. Error = PostInfoShell(SHELL_CMD_INFO_UPDATE_TOKEN, &pkt);
  721. if (pkt)
  722. iobuffer_dec_ref(pkt);
  723. return Error;
  724. }
  725. ErrorCodeEnum SpEntityPrivilege::SetEntityPriority(const char* pszEntityName, EntityPriorityEnum nPriority)
  726. {
  727. if (!pszEntityName)
  728. return Error_Null;
  729. sp_env_t* env = sp_get_env();
  730. sp_mod_mgr_t* mod_mgr = env->mod_mgr;
  731. sp_entity_t* ent = sp_mod_mgr_find_entity_by_name(mod_mgr, pszEntityName);
  732. if (!ent) {
  733. return Error_NotExist;
  734. }
  735. #ifndef _WIN32
  736. auto func = [nPriority] {
  737. switch (nPriority) {
  738. case Priority_Low:
  739. return TOOLKIT_PRIORITY_LOW;
  740. case Priority_Below_Normal:
  741. return TOOLKIT_PRIORITY_BELOW_NORMAL;
  742. case Priority_Normal:
  743. return TOOLKIT_PRIORITY_NORMAL;
  744. case Priority_Above_Normal:
  745. return TOOLKIT_PRIORITY_ABOVE_NORMAL;
  746. case Priority_High:
  747. return TOOLKIT_PRIORITY_HIGH;
  748. case Priority_Highest:
  749. return TOOLKIT_PRIORITY_HIGHEST;
  750. default:
  751. return TOOLKIT_PRIORITY_NORMAL;
  752. }
  753. };
  754. int res = toolkit_os_setpriority(ent->mod->process.pid, func());
  755. #else
  756. int priviliegeValue = TOOLKIT_PRIORITY_NORMAL;
  757. switch (nPriority) {
  758. case Priority_Low:
  759. priviliegeValue = TOOLKIT_PRIORITY_LOW;
  760. break;
  761. case Priority_Below_Normal:
  762. priviliegeValue = TOOLKIT_PRIORITY_BELOW_NORMAL;
  763. break;
  764. case Priority_Normal:
  765. priviliegeValue = TOOLKIT_PRIORITY_NORMAL;
  766. break;
  767. case Priority_Above_Normal:
  768. priviliegeValue = TOOLKIT_PRIORITY_ABOVE_NORMAL;
  769. break;
  770. case Priority_High:
  771. priviliegeValue = TOOLKIT_PRIORITY_HIGH;
  772. break;
  773. case Priority_Highest:
  774. priviliegeValue = TOOLKIT_PRIORITY_HIGHEST;
  775. break;
  776. default:
  777. priviliegeValue = TOOLKIT_PRIORITY_NORMAL;
  778. break;
  779. }
  780. int res = toolkit_os_setpriority(ent->mod->process.pid, priviliegeValue);
  781. #endif //NOT _WIN32
  782. if (res != 0) {
  783. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("toolkit_os_setpriority failed %s ", toolkit_strerror(res));
  784. return Error_Unexpect;
  785. }
  786. return Error_Succeed;
  787. }
  788. ErrorCodeEnum SpEntityPrivilege::GetEntityPriority(const char* pszEntityName, EntityPriorityEnum& nPriority)
  789. {
  790. if (!pszEntityName)
  791. return Error_Null;
  792. sp_env_t* env = sp_get_env();
  793. sp_mod_mgr_t* mod_mgr = env->mod_mgr;
  794. sp_entity_t* ent = sp_mod_mgr_find_entity_by_name(mod_mgr, pszEntityName);
  795. if (!ent) {
  796. return Error_NotExist;
  797. }
  798. int priority;
  799. int res = toolkit_os_getpriority(ent->mod->process.pid, &priority);
  800. if (res != 0) {
  801. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("toolkit_os_getpriority failed %s ", toolkit_strerror(res));
  802. return Error_Unexpect;
  803. }
  804. #if defined(_MSC_VER)
  805. switch (priority) {
  806. case TOOLKIT_PRIORITY_LOW:
  807. nPriority = Priority_Low;
  808. break;
  809. case TOOLKIT_PRIORITY_BELOW_NORMAL:
  810. nPriority = Priority_Below_Normal;
  811. break;
  812. case TOOLKIT_PRIORITY_NORMAL:
  813. nPriority = Priority_Normal;
  814. break;
  815. case TOOLKIT_PRIORITY_ABOVE_NORMAL:
  816. nPriority = Priority_Above_Normal;
  817. break;
  818. case TOOLKIT_PRIORITY_HIGH:
  819. nPriority = Priority_High;
  820. break;
  821. case TOOLKIT_PRIORITY_HIGHEST:
  822. nPriority = Priority_Highest;
  823. break;
  824. default:
  825. nPriority = Priority_Unknown;
  826. break;
  827. }
  828. #else
  829. auto func = [](int value) {
  830. switch (value) {
  831. case TOOLKIT_PRIORITY_LOW:
  832. return Priority_Low;
  833. case TOOLKIT_PRIORITY_BELOW_NORMAL:
  834. return Priority_Below_Normal;
  835. case TOOLKIT_PRIORITY_NORMAL:
  836. return Priority_Normal;
  837. case TOOLKIT_PRIORITY_ABOVE_NORMAL:
  838. return Priority_Above_Normal;
  839. case TOOLKIT_PRIORITY_HIGH:
  840. return Priority_High;
  841. case TOOLKIT_PRIORITY_HIGHEST:
  842. return Priority_Highest;
  843. default:
  844. return Priority_Unknown;
  845. }
  846. };
  847. nPriority = func(priority);
  848. #endif //_MSC_VER
  849. return Error_Succeed;
  850. }
  851. ErrorCodeEnum SpEntityPrivilege::GetToken(CSimpleString& curToken)
  852. {
  853. sp_env_t* env = sp_get_env();
  854. if (env->cfg->shell_ini->token == NULL || strlen(env->cfg->shell_ini->token) == 0)
  855. return ErrorCodeEnum::Error_NotConfig;
  856. curToken = env->cfg->shell_ini->token;
  857. return Error_Succeed;
  858. }
  859. ErrorCodeEnum SpEntityPrivilege::GetVTMErrMsgArr(CAutoArray<CSimpleStringA>& strErrorCodeArr, CAutoArray<CSimpleStringA>& strDescriptionArr,
  860. CAutoArray<CSimpleStringA>& strRemarkArr)
  861. {
  862. sp_env_t* env = sp_get_env();
  863. if(env->cfg->root_ini->vtm_err_msg_config == NULL || strlen(env->cfg->root_ini->vtm_err_msg_config) == 0)
  864. return ErrorCodeEnum::Error_NotConfig;
  865. return (ErrorCodeEnum)ConvertStrToVTMErrMsg(env->cfg->root_ini->vtm_err_msg_config, strErrorCodeArr, strDescriptionArr, strRemarkArr);
  866. }
  867. ErrorCodeEnum SpEntityPrivilege::TryUpdateToken(CSimpleString& oldToken, CSimpleString& newToken)
  868. {
  869. char t_oldToken[MAX_PATH] = "", t_newToken[MAX_PATH] = "";
  870. auto ret = sp_TryUpdateToken(t_oldToken, t_newToken);
  871. if (ret == Error_Succeed)
  872. {
  873. oldToken = t_oldToken;
  874. newToken = t_newToken;
  875. }
  876. return (ErrorCodeEnum)ret;
  877. }
  878. ErrorCodeEnum SpEntityPrivilege::InitCfgUrl(VTMInitParam& info)
  879. {
  880. auto cfg = sp_get_env()->cfg;
  881. if (cfg == NULL || cfg->shell_ini == NULL || cfg->root_ini == NULL)
  882. return Error_Null;
  883. if (info.terminalNo.GetLength() > 0)
  884. {
  885. if (cfg->root_ini->terminal_no)
  886. shm_free(cfg->root_ini->terminal_no);
  887. cfg->root_ini->terminal_no = shm_strdup(info.terminalNo.GetData());
  888. }
  889. if (info.channelId.GetLength() > 0)
  890. {
  891. if (cfg->shell_ini->channelId)
  892. shm_free(cfg->shell_ini->channelId);
  893. cfg->shell_ini->channelId = shm_strdup(info.channelId.GetData());
  894. }
  895. if (info.tokenSecret.GetLength() > 0)
  896. {
  897. if (cfg->shell_ini->tokenSecret)
  898. shm_free(cfg->shell_ini->tokenSecret);
  899. cfg->shell_ini->tokenSecret = shm_strdup(info.tokenSecret.GetData());
  900. }
  901. if (info.CommonLaunchUrl.GetLength() > 0)
  902. {
  903. if (cfg->shell_ini->CommonLaunchUrl)
  904. shm_free(cfg->shell_ini->CommonLaunchUrl);
  905. cfg->shell_ini->CommonLaunchUrl = shm_strdup(info.CommonLaunchUrl.GetData());
  906. }
  907. if (info.CenterConfigTotal.GetLength() > 0)
  908. {
  909. if (cfg->shell_ini->CenterConfigTotal)
  910. shm_free(cfg->shell_ini->CenterConfigTotal);
  911. cfg->shell_ini->CenterConfigTotal = shm_strdup(info.CenterConfigTotal.GetData());
  912. }
  913. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("TerminalNo:%s, CommonLaunchUrl:%s, CenterConfigTotal:%s"
  914. , cfg->root_ini->terminal_no == NULL ? "" : cfg->root_ini->terminal_no
  915. , cfg->shell_ini->CommonLaunchUrl == NULL ? "" : cfg->shell_ini->CommonLaunchUrl
  916. , cfg->shell_ini->CenterConfigTotal == NULL ? "" : cfg->shell_ini->CenterConfigTotal);
  917. return Error_Succeed;
  918. }
  919. ErrorCodeEnum SpEntityPrivilege::TryUpdateVTMERRMSG()
  920. {
  921. LOG_FUNCTION();
  922. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  923. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_TRY_UPDATE_VTMERRMSG);
  924. ErrorCodeEnum Error = pAsyncWait->Begin();
  925. if (Error == Error_Succeed)
  926. {
  927. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  928. if (Error == Error_Succeed)
  929. {
  930. CAutoBuffer AnsBuf;
  931. bool bEnd;
  932. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  933. if (Error == Error_Succeed)
  934. {
  935. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  936. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  937. }
  938. }
  939. }
  940. pAsyncWait->DecrementRef();
  941. if (req_pkt)
  942. iobuffer_dec_ref(req_pkt);
  943. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("TryUpdateVTMERRMSG ret:%d", Error);
  944. return Error;
  945. }
  946. ErrorCodeEnum SpEntityPrivilege::TryUpdateCfg()
  947. {
  948. LOG_FUNCTION();
  949. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  950. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_TRY_UPDATE_CFG);
  951. ErrorCodeEnum Error = pAsyncWait->Begin();
  952. if (Error == Error_Succeed)
  953. {
  954. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  955. if (Error == Error_Succeed)
  956. {
  957. CAutoBuffer AnsBuf;
  958. bool bEnd;
  959. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  960. if (Error == Error_Succeed)
  961. {
  962. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  963. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  964. }
  965. }
  966. }
  967. pAsyncWait->DecrementRef();
  968. if (req_pkt)
  969. iobuffer_dec_ref(req_pkt);
  970. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("TryUpdateCfg ret:%d", Error);
  971. return Error;
  972. }
  973. ErrorCodeEnum SpEntityPrivilege::ModifyMemCfgParam(MemCfgParam& info)
  974. {
  975. LOG_FUNCTION();
  976. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  977. iobuffer_write(req_pkt, IOBUF_T_STR, info.configType, -1);
  978. iobuffer_write(req_pkt, IOBUF_T_STR, info.module.GetData(), -1);
  979. iobuffer_write(req_pkt, IOBUF_T_STR, info.name.GetData(), -1);
  980. iobuffer_write(req_pkt, IOBUF_T_STR, info.value.GetData(), -1);
  981. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_MODIFY_MEM_CFG);
  982. ErrorCodeEnum Error = pAsyncWait->Begin();
  983. if (Error == Error_Succeed)
  984. {
  985. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  986. if (Error == Error_Succeed)
  987. {
  988. CAutoBuffer AnsBuf;
  989. bool bEnd;
  990. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  991. if (Error == Error_Succeed)
  992. {
  993. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  994. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  995. }
  996. }
  997. }
  998. pAsyncWait->DecrementRef();
  999. if (req_pkt)
  1000. iobuffer_dec_ref(req_pkt);
  1001. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("ModifyMemCfgParam ret:%d", Error);
  1002. return Error;
  1003. }
  1004. ErrorCodeEnum SpEntityPrivilege::WriteTerminalNoToRootIni(CSimpleString terminalNo)
  1005. {
  1006. LOG_FUNCTION();
  1007. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  1008. iobuffer_write(req_pkt, IOBUF_T_STR, terminalNo.GetData(), -1);
  1009. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_WRITE_TERMINALNO);
  1010. ErrorCodeEnum Error = pAsyncWait->Begin();
  1011. if (Error == Error_Succeed)
  1012. {
  1013. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  1014. if (Error == Error_Succeed)
  1015. {
  1016. CAutoBuffer AnsBuf;
  1017. bool bEnd;
  1018. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  1019. if (Error == Error_Succeed)
  1020. {
  1021. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  1022. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  1023. }
  1024. }
  1025. }
  1026. pAsyncWait->DecrementRef();
  1027. if (req_pkt)
  1028. iobuffer_dec_ref(req_pkt);
  1029. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("WriteTerminalNoToRootIni ret:%d", Error);
  1030. return Error;
  1031. }
  1032. ErrorCodeEnum SpEntityPrivilege::TryUpdateCenterCfg(bool& isUpdate, bool& isReset, CSimpleString& version)
  1033. {
  1034. char t_version[MAX_PATH] = "";
  1035. int t_isUpdate = 0, t_isReset = 0;
  1036. auto ret = sp_TryUpdateCenterCfg(&t_isUpdate, &t_isReset, t_version);
  1037. if (ret == Error_Succeed)
  1038. {
  1039. isUpdate = t_isUpdate;
  1040. isReset = t_isReset;
  1041. version = t_version;
  1042. }
  1043. return (ErrorCodeEnum)ret;
  1044. }
  1045. void SpEntityPrivilege::GetSendLogInfo(unsigned long* t_upload_TerminalSys_Suc, unsigned long* t_upload_TerminalUser_Suc,
  1046. unsigned long* t_upload_BussinessSys_Suc, unsigned long* t_upload_BussinessUser_Suc, unsigned long* t_upload_beidou_Suc,
  1047. unsigned long* t_upload_TerminalSys_Err, unsigned long* t_upload_TerminalUser_Err,
  1048. unsigned long* t_upload_BussinessSys_Err, unsigned long* t_upload_BussinessUser_Err, unsigned long* t_upload_beidou_Err)
  1049. {
  1050. #if defined(_MSC_VER)
  1051. log_producer_config_get_upload_info(t_upload_TerminalSys_Suc, t_upload_TerminalUser_Suc, t_upload_BussinessSys_Suc, t_upload_BussinessUser_Suc, t_upload_beidou_Suc,
  1052. t_upload_TerminalSys_Err, t_upload_TerminalUser_Err, t_upload_BussinessSys_Err, t_upload_BussinessUser_Err, t_upload_beidou_Err);
  1053. #else
  1054. ///*TODO(80374374@3/9/2023): */
  1055. #endif //_MSC_VER
  1056. }
  1057. void SpEntityPrivilege::on_entity_create(int entity_id, int trigger_entity_id)
  1058. {
  1059. if (m_pEntityLifeListener) {
  1060. sp_env_t *env = sp_get_env();
  1061. IEntityLifeListener *pCallback = (IEntityLifeListener *)sp_mod_entity_life_listener_get_tag(m_pEntityLifeListener);
  1062. sp_mod_mgr_t *mod_mgr = env->mod_mgr;
  1063. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, entity_id);
  1064. sp_entity_t *trigger_ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, trigger_entity_id);
  1065. pCallback->OnCreated(ent->cfg->name, Error_Succeed, trigger_ent->cfg->name);
  1066. }
  1067. }
  1068. void SpEntityPrivilege::on_entity_close(int entity_id, int trigger_entity_id, int cause_code)
  1069. {
  1070. if (m_pEntityLifeListener) {
  1071. sp_env_t *env = sp_get_env();
  1072. IEntityLifeListener *pCallback = (IEntityLifeListener *)sp_mod_entity_life_listener_get_tag(m_pEntityLifeListener);
  1073. sp_mod_mgr_t *mod_mgr = env->mod_mgr;
  1074. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, entity_id);
  1075. sp_entity_t *trigger_ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, trigger_entity_id);
  1076. pCallback->OnClosed(ent->cfg->name, (EntityCloseCauseEnum)cause_code, Error_Succeed, trigger_ent->cfg->name);
  1077. }
  1078. }
  1079. void SpEntityPrivilege::on_entity_exception(int entity_id, int error, int entity_state)
  1080. {
  1081. if (m_pEntityLifeListener) {
  1082. sp_env_t *env = sp_get_env();
  1083. IEntityLifeListener *pCallback = (IEntityLifeListener *)sp_mod_entity_life_listener_get_tag(m_pEntityLifeListener);
  1084. sp_mod_mgr_t *mod_mgr = env->mod_mgr;
  1085. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, entity_id);
  1086. pCallback->OnException(ent->cfg->name, "", (EntityStateEnum)entity_state, (EntityStateEnum)entity_state, (ErrorCodeEnum)error);
  1087. }
  1088. }
  1089. void SpEntityPrivilege::__on_entity_create(sp_mod_entity_life_listener_t *listener, int entity_id, int trigger_entity_id, void *user_data)
  1090. {
  1091. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1092. pThis->on_entity_create(entity_id, trigger_entity_id);
  1093. }
  1094. void SpEntityPrivilege::__on_entity_close(sp_mod_entity_life_listener_t *listener, int entity_id, int trigger_entity_id, int cause_code, void *user_data)
  1095. {
  1096. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1097. pThis->on_entity_close(entity_id, trigger_entity_id, cause_code);
  1098. }
  1099. void SpEntityPrivilege::__on_entity_exception(sp_mod_entity_life_listener_t *listener, int entity_id, int error, int entity_state, void *user_data)
  1100. {
  1101. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1102. pThis->on_entity_exception(entity_id, error, entity_state);
  1103. }
  1104. void SpEntityPrivilege::on_user_state(sp_mod_entity_state_listener_t *listener, int entity_id, int last_state, int curr_state)
  1105. {
  1106. sp_env_t *env = sp_get_env();
  1107. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1108. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, entity_id);
  1109. pListener->OnUserStateHook(ent->cfg->name, (DWORD)curr_state, (DWORD)last_state);
  1110. }
  1111. void SpEntityPrivilege::on_entity_state(sp_mod_entity_state_listener_t *listener, int entity_id, int trigger_entity_id, int last_state, int curr_state)
  1112. {
  1113. sp_env_t *env = sp_get_env();
  1114. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1115. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, entity_id);
  1116. sp_entity_t *trigger_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, trigger_entity_id);
  1117. pListener->OnEntityStateHook(ent->cfg->name, trigger_ent->cfg->name, (EntityStateEnum)curr_state, (EntityStateEnum)last_state);
  1118. }
  1119. void SpEntityPrivilege::on_create_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id)
  1120. {
  1121. sp_env_t *env = sp_get_env();
  1122. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1123. sp_entity_t *src_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, src_entity_id);
  1124. sp_entity_t *dst_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, dst_entity_id);
  1125. pListener->OnCeateConnection(src_ent->cfg->name, dst_ent->cfg->name);
  1126. }
  1127. void SpEntityPrivilege::on_close_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id)
  1128. {
  1129. sp_env_t *env = sp_get_env();
  1130. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1131. sp_entity_t *src_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, src_entity_id);
  1132. sp_entity_t *dst_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, dst_entity_id);
  1133. pListener->OnCloseConnection(src_ent->cfg->name, dst_ent->cfg->name);
  1134. }
  1135. void SpEntityPrivilege::__on_user_state(sp_mod_entity_state_listener_t *listener, int entity_id, int last_state, int curr_state, void *user_data)
  1136. {
  1137. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1138. pThis->on_user_state(listener, entity_id, last_state, curr_state);
  1139. }
  1140. void SpEntityPrivilege::__on_entity_state(sp_mod_entity_state_listener_t *listener, int entity_id, int trigger_entity_id, int last_state, int curr_state, void *user_data)
  1141. {
  1142. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1143. pThis->on_entity_state(listener, entity_id, trigger_entity_id, last_state, curr_state);
  1144. }
  1145. void SpEntityPrivilege::__on_create_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id, void *user_data)
  1146. {
  1147. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1148. pThis->on_create_connection(listener, src_entity_id, dst_entity_id);
  1149. }
  1150. void SpEntityPrivilege::__on_close_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id, void *user_data)
  1151. {
  1152. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1153. pThis->on_close_connection(listener, src_entity_id, dst_entity_id);
  1154. }