SpEntityPrivilege.cpp 41 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346
  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::RecursiveCopyDir(const char *pszSourceDir, const char *pszDestDir, CSimpleStringA& strErrInfo)
  290. {
  291. array_header_t *arr;
  292. if (!ExistsDirA(pszSourceDir))
  293. {
  294. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("source dir [%s] not exists", pszSourceDir);
  295. return false;
  296. }
  297. if (!ExistsDirA(pszDestDir))
  298. {
  299. if (!CreateDirRecursiveA(pszDestDir))
  300. {
  301. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create dir [%s] fail: %d", pszDestDir, GetLastError());
  302. return false;
  303. }
  304. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("create dir [%s] succ", pszDestDir);
  305. }
  306. arr = fileutil_get_sub_files_a(pszSourceDir);
  307. if (arr)
  308. {
  309. int i;
  310. for (i = 0; i < arr->nelts; ++i)
  311. {
  312. char szDestFile[256] = { 0 };
  313. char *file = ARRAY_IDX(arr, i, char*);
  314. strcpy(szDestFile, pszDestDir);
  315. if (szDestFile[strlen(szDestFile) - 1] != SPLIT_SLASH)
  316. strcat(szDestFile, SPLIT_SLASH_STR);
  317. strcat(szDestFile, strrchr(file, SPLIT_SLASH) + 1);
  318. if (!CopyFileA(file, szDestFile, FALSE))
  319. {
  320. strErrInfo = CSimpleStringA::Format("copy file [%s] fail: %d", file, GetLastError());
  321. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy file [%s] fail: %d", file, GetLastError());
  322. toolkit_array_free2(arr);
  323. return false;
  324. }
  325. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy file [%s] succ", file);
  326. }
  327. toolkit_array_free2(arr);
  328. }
  329. arr = fileutil_get_sub_dirs_a(pszSourceDir);
  330. if (arr)
  331. {
  332. int i;
  333. for (i = 0; i < arr->nelts; ++i)
  334. {
  335. char szDestSubDir[256] = { 0 };
  336. char *dir = ARRAY_IDX(arr, i, char*);
  337. strcpy(szDestSubDir, pszDestDir);
  338. if (szDestSubDir[strlen(szDestSubDir) - 1] != SPLIT_SLASH)
  339. strcat(szDestSubDir, SPLIT_SLASH_STR);
  340. strcat(szDestSubDir, strrchr(dir, SPLIT_SLASH) + 1);
  341. if (!RecursiveCopyDir(dir, szDestSubDir, strErrInfo))
  342. {
  343. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy dir [%s] fail: %d", dir, GetLastError());
  344. toolkit_array_free2(arr);
  345. return false;
  346. }
  347. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy dir [%s] succ", dir);
  348. }
  349. toolkit_array_free2(arr);
  350. }
  351. return true;
  352. }
  353. ErrorCodeEnum SpEntityPrivilege::UpdateVerTxt(const char* strActiveFilePath, const char* toWriteVersionStr, bool toCheck)
  354. {
  355. HANDLE hFile = ::CreateFileA(strActiveFilePath, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  356. BOOL bSetSucc = FALSE;
  357. if (hFile == INVALID_HANDLE_VALUE)
  358. {
  359. LogError(Severity_Low, Error_Unexpect, 0, "open active.txt fail");
  360. return Error_Unexpect;
  361. }
  362. DWORD dwWrittenLen(0);
  363. bSetSucc = WriteFile(hFile, toWriteVersionStr, strlen(toWriteVersionStr), &dwWrittenLen, NULL);
  364. FlushFileBuffers(hFile);
  365. SetEndOfFile(hFile);
  366. CloseHandle(hFile);
  367. LOG_TRACE("write version [%s] succeed", toWriteVersionStr);
  368. //出现active文件为空的问题,此处增加写文件件后再读一次,确认是否写成功,以此来确认是否是写文件导致
  369. if (toCheck) {
  370. hFile = ::CreateFileA(strActiveFilePath, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  371. if (hFile == INVALID_HANDLE_VALUE)
  372. {
  373. LogError(Severity_Low, Error_Unexpect, 0, "open active.txt fail");
  374. return Error_Unexpect;
  375. }
  376. DWORD dwReadLen(0);
  377. char cVer[16] = { 0 };
  378. if (ReadFile(hFile, cVer, 16, &dwReadLen, NULL))
  379. {
  380. LOG_TRACE("read active.txt, version len[%d], version[%s]", dwReadLen, cVer);
  381. //如果为空,则重写
  382. if (strlen(cVer) == 0)
  383. {
  384. if (WriteFile(hFile, toWriteVersionStr, strlen(toWriteVersionStr), &dwWrittenLen, NULL))
  385. {
  386. LOG_TRACE("rewrite active.txt success, version[%s]", toWriteVersionStr);
  387. }
  388. else
  389. {
  390. LogError(Severity_Low, Error_Unexpect, 0, "rewrite active.txt fail");
  391. }
  392. }
  393. }
  394. else
  395. {
  396. LogError(Severity_Low, Error_Unexpect, 0, "Read active.txt fail");
  397. }
  398. FlushFileBuffers(hFile);
  399. SetEndOfFile(hFile);
  400. CloseHandle(hFile);
  401. }
  402. return bSetSucc ? Error_Succeed : Error_Unexpect;
  403. }
  404. ErrorCodeEnum CopyFolder(CSimpleStringA strSourcePath, CSimpleStringA strDestPath)
  405. {
  406. if (strSourcePath.IsNullOrEmpty() || strDestPath.IsNullOrEmpty())
  407. {
  408. return Error_Null;
  409. }
  410. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Start to copy dir %s to dir %s", (const char*)strSourcePath,(const char*)strDestPath);
  411. #ifdef _WIN32
  412. _finddata_t FileInfo;
  413. CSimpleStringA strfind = strSourcePath + "\\*";
  414. long Handle = _findfirst(strfind, &FileInfo);
  415. if (-1L == Handle)
  416. {
  417. _findclose(Handle);
  418. DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("%s文件夹为空", strSourcePath);
  419. return Error_Succeed;
  420. }
  421. CSimpleStringA newPath;
  422. do{
  423. if (FileInfo.attrib & _A_SUBDIR)
  424. {
  425. if ((strcmp(FileInfo.name, ".") != 0) && (strcmp(FileInfo.name, "..") != 0))
  426. {
  427. newPath = strSourcePath + "\\" + FileInfo.name;
  428. CSimpleStringA newDestPath = strDestPath + "\\" + FileInfo.name;
  429. /*if (strcmp(FileInfo.name, "") == 0)
  430. {
  431. continue;
  432. }*/
  433. if (!ExistsDirA(newDestPath))
  434. {
  435. CreateDirA(newDestPath, TRUE);
  436. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("Create Dir %s success", (const char*)newDestPath);
  437. }
  438. CopyFolder(newPath, newDestPath);
  439. }
  440. }
  441. else
  442. {
  443. CSimpleStringA strFindName = FileInfo.name;
  444. {
  445. CSimpleStringA strSourceFile = strSourcePath + "\\" + strFindName;
  446. CSimpleStringA strDestFile = strDestPath + "\\" + strFindName;
  447. if (ExistsFileA(strSourceFile))
  448. {
  449. // 先去除目标文件只读属性
  450. if (ExistsFileA(strDestFile))
  451. RemoveFileReadOnlyAttributeA(strDestFile);
  452. LOG_TRACE("copy file from [%s] to [%s]", (const char*)strSourceFile, (const char*)strDestFile);
  453. BOOL bRet = CopyFileA(strSourceFile, strDestFile, FALSE);
  454. // 去除只读属性
  455. if (bRet)
  456. RemoveFileReadOnlyAttributeA(strDestFile);
  457. else
  458. {
  459. // 覆盖失败,则生成.new副本
  460. 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);
  461. strDestFile += ".new";
  462. bRet = CopyFileA(strSourceFile, strDestFile, FALSE);
  463. if (bRet)
  464. {
  465. RemoveFileReadOnlyAttributeA(strDestPath);
  466. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("copy [%s] as [%s] succeed", (const char*)strSourceFile, (const char*) strDestFile);
  467. }
  468. else
  469. {
  470. LogError(Severity_Low, Error_Unexpect, 0, CSimpleStringA::Format("copy file [%s] as [%s] fail", (const char*) strSourceFile, (const char*) strDestFile));
  471. return Error_Block;
  472. }
  473. }
  474. }
  475. else
  476. {
  477. LogError(Severity_Low, Error_NotExist, 0, CSimpleStringA::Format("file [%s] to copy not exists", (const char*)strSourceFile));
  478. return Error_NotExist;
  479. }
  480. }
  481. }
  482. } while (_findnext(Handle, &FileInfo) == 0);
  483. _findclose(Handle);
  484. return Error_Succeed;
  485. #else
  486. if (CopyDirA(strSourcePath, strDestPath)) {
  487. return Error_Succeed;
  488. }
  489. return Error_Unexpect;
  490. #endif //_WIN32
  491. }
  492. CSimpleStringA GetFileDirectory(const char *pszFullPath)
  493. {
  494. int i=strlen(pszFullPath)-1;
  495. for( ; i>0 && pszFullPath[i]!=SPLIT_SLASH; i--)NULL;
  496. return CSimpleStringA(pszFullPath, i);
  497. }
  498. #ifdef _WIN32
  499. bool SpEntityPrivilege::IsWow64Process()
  500. {
  501. typedef BOOL(WINAPI *LPFN_ISWOW64PROCESS)(HANDLE, PBOOL);
  502. LPFN_ISWOW64PROCESS fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress(GetModuleHandle("kernel32"), "IsWow64Process");
  503. BOOL bIsWow64 = FALSE;
  504. if (NULL != fnIsWow64Process)
  505. {
  506. if (!fnIsWow64Process(GetCurrentProcess(), &bIsWow64))
  507. {
  508. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("call IsWow64Process error: %d", GetLastError());
  509. return false;
  510. }
  511. return bIsWow64 == TRUE;
  512. }
  513. else
  514. {
  515. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("get IsWow64Process error: %d", GetLastError());
  516. return false;
  517. }
  518. }
  519. #endif //_WIN32
  520. // result=0&msg=install ok
  521. bool SpEntityPrivilege::GetSysUpgradeResult(const char *pszResultLog, ErrorCodeEnum &eErrorCode, CSimpleStringA &strErrMsg)
  522. {
  523. // 设置初始值
  524. eErrorCode = Error_Succeed;
  525. strErrMsg = "";
  526. if (!ExistsFileA(pszResultLog))
  527. {
  528. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)(CSimpleStringA::Format("result log [%s] not exist", pszResultLog));
  529. strErrMsg = "result.log not exist";
  530. return false;
  531. }
  532. FILE *pResultLog = fopen(pszResultLog, "r");
  533. if (pResultLog == NULL)
  534. {
  535. LogError(Severity_Low, Error_Unexpect, 0, CSimpleStringA::Format("open result log [%s] fail", pszResultLog));
  536. strErrMsg = "open result.log fail";
  537. return false;
  538. }
  539. char szTmp[4096] = {};
  540. int nRead = fread_s(szTmp, sizeof(szTmp), 1, sizeof(szTmp), pResultLog);
  541. CSimpleStringA strResult = szTmp;
  542. fclose(pResultLog);
  543. pResultLog = NULL;
  544. auto arr = strResult.Split('&');
  545. for (int i = 0; i < arr.GetCount(); i++)
  546. {
  547. auto arr2 = arr[i].Split('=');
  548. if (arr2.GetCount() == 2)
  549. {
  550. if (stricmp(arr2[0], "result") == 0)
  551. eErrorCode = (ErrorCodeEnum)(DWORD)atoi(arr2[1]);
  552. else if (stricmp(arr2[0], "msg") == 0)
  553. strErrMsg = arr2[1];
  554. }
  555. }
  556. return true;
  557. }
  558. // File*A.*
  559. bool SpEntityPrivilege::IsFileMatch(const char *pszFilter, const char *pszFileName)
  560. {
  561. if (stricmp(pszFilter, "*") == 0 || stricmp(pszFilter, "*.*") == 0)
  562. return true;
  563. const char *p1 = pszFilter;
  564. const char *p2 = pszFileName;
  565. if (p1 == NULL)
  566. return true;
  567. if (p2 == NULL)
  568. return false;
  569. if (*p1 ==NULL && *p2 ==NULL)
  570. return true;
  571. else if (*p1 == NULL)
  572. return false;
  573. else if (*p2 == NULL)
  574. {
  575. // 查找*p1是否全是*
  576. while(*p1 == '*')
  577. p1++;
  578. if (*p1 == NULL)
  579. return true;
  580. else
  581. return false;
  582. }
  583. if (*p1 != '*')
  584. {
  585. if (tolower(*p1) != tolower(*p2))
  586. return false;
  587. else
  588. return IsFileMatch(p1+1, p2+1);
  589. }
  590. else
  591. {
  592. while(*++p1 == '*');
  593. if (*p1 == NULL)
  594. return true;
  595. while (*p2)
  596. {
  597. while(tolower(*p1) != tolower(*p2) && *++p2);
  598. if (*p2 == NULL)
  599. return false;
  600. if(IsFileMatch(p1+1, p2+1))
  601. return true;
  602. p2++;
  603. }
  604. return false;
  605. }
  606. }
  607. CSimpleStringA SpEntityPrivilege::GetFileDirectory(const char *pszFullPath)
  608. {
  609. int i=strlen(pszFullPath);
  610. for( ; i>0 && pszFullPath[i-1]!=SPLIT_SLASH; i--)NULL;
  611. return CSimpleStringA(pszFullPath, i);
  612. }
  613. ErrorCodeEnum SpEntityPrivilege::SetSysDebugLevel(const char *pszEntityName,DebugLevelEnum eDebugLevel,bool bPersist)
  614. {
  615. sp_env_t *env = sp_get_env();
  616. if (!pszEntityName)
  617. return Error_Null;
  618. if (stricmp(pszEntityName, "SpShell") == 0)
  619. env->cfg->shell_ini->shell_debug_level = (int)eDebugLevel;
  620. sp_cfg_shell_entity_t *cfg_ent = sp_cfg_get_entity_by_name(env->cfg, pszEntityName);
  621. if (cfg_ent) {
  622. int old = cfg_ent->debug_level;
  623. if (old != eDebugLevel) {
  624. cfg_ent->debug_level = (int)eDebugLevel;
  625. if (bPersist) {
  626. return SpTranslateError(sp_cfg_refresh_debug_level(env->cfg, cfg_ent));
  627. }
  628. }
  629. } else {
  630. return Error_NotExist;
  631. }
  632. return Error_Succeed;
  633. }
  634. ErrorCodeEnum SpEntityPrivilege::RefreshFrameworkState(FrameworkStateEnum eState)
  635. {
  636. int rc = Error_Succeed;
  637. char n[12] = { '\0' };
  638. sp_env_t* env = sp_get_env();
  639. sp_cfg_t* cfg = env->cfg;
  640. const int old_state = cfg->shell_ini->shell_state;
  641. if (old_state != (int)eState) {
  642. cfg->shell_ini->shell_state = eState;
  643. _itoa(eState, n, 10);
  644. rc = sp_var_client_set((sp_var_client_t*)m_var_client, VAR_RSERVERD_KEY_TERM_STATE, n, 0);
  645. if (rc == Error_Succeed) {
  646. iobuffer_t* pkt = iobuffer_create(-1, -1);
  647. int v1 = (int)eState;
  648. int v2 = (int)old_state;
  649. iobuffer_write(pkt, IOBUF_T_I4, &v1, 0);
  650. iobuffer_write(pkt, IOBUF_T_I4, &v2, 0);
  651. rc = PostInfoShell(SHELL_CMD_INFO_TERMINALSTAGE_CHANGE, &pkt);
  652. if (pkt) {
  653. iobuffer_dec_ref(pkt);
  654. }
  655. }
  656. }
  657. return SpTranslateError(rc);
  658. }
  659. ErrorCodeEnum SpEntityPrivilege::ShowOuputConsole()
  660. {
  661. ErrorCodeEnum Error;
  662. iobuffer_t *pkt = iobuffer_create(-1, -1);
  663. Error = PostInfoShell(SHELL_CMD_INFO_OUTPUT_CONSOLE_ON, &pkt);
  664. if (pkt)
  665. iobuffer_dec_ref(pkt);
  666. return Error;
  667. }
  668. ErrorCodeEnum SpEntityPrivilege::CloseOuputConsole()
  669. {
  670. ErrorCodeEnum Error;
  671. iobuffer_t *pkt = iobuffer_create(-1, -1);
  672. Error = PostInfoShell(SHELL_CMD_INFO_OUTPUT_CONSOLE_OFF, &pkt);
  673. if (pkt)
  674. iobuffer_dec_ref(pkt);
  675. return Error;
  676. }
  677. ErrorCodeEnum SpEntityPrivilege::BeginLogSend(const char* endpoint, const char* topicSys, const char* topicUser, const char* topicBeidou
  678. , const char* bussSys, const char* bussUser, const char* vtmWeb)
  679. {
  680. ErrorCodeEnum Error;
  681. iobuffer_t* pkt = iobuffer_create(-1, -1);
  682. iobuffer_write(pkt, IOBUF_T_STR, endpoint, -1);
  683. iobuffer_write(pkt, IOBUF_T_STR, topicSys, -1);
  684. iobuffer_write(pkt, IOBUF_T_STR, topicUser, -1);
  685. iobuffer_write(pkt, IOBUF_T_STR, topicBeidou, -1);
  686. iobuffer_write(pkt, IOBUF_T_STR, bussSys, -1);
  687. iobuffer_write(pkt, IOBUF_T_STR, bussUser, -1);
  688. iobuffer_write(pkt, IOBUF_T_STR, vtmWeb, -1);
  689. Error = PostInfoShell(SHELL_CMD_INFO_START_UPLOAD_LOG, &pkt);
  690. if (pkt)
  691. iobuffer_dec_ref(pkt);
  692. return Error;
  693. }
  694. ErrorCodeEnum SpEntityPrivilege::SetHttpToken(const char* channelId, const char* token)
  695. {
  696. ErrorCodeEnum Error;
  697. iobuffer_t* pkt = iobuffer_create(-1, -1);
  698. iobuffer_write(pkt, IOBUF_T_STR, channelId, -1);
  699. iobuffer_write(pkt, IOBUF_T_STR, token, -1);
  700. Error = PostInfoShell(SHELL_CMD_INFO_UPDATE_TOKEN, &pkt);
  701. if (pkt)
  702. iobuffer_dec_ref(pkt);
  703. return Error;
  704. }
  705. ErrorCodeEnum SpEntityPrivilege::SetEntityPriority(const char* pszEntityName, EntityPriorityEnum nPriority)
  706. {
  707. if (!pszEntityName)
  708. return Error_Null;
  709. sp_env_t* env = sp_get_env();
  710. sp_mod_mgr_t* mod_mgr = env->mod_mgr;
  711. sp_entity_t* ent = sp_mod_mgr_find_entity_by_name(mod_mgr, pszEntityName);
  712. if (!ent) {
  713. return Error_NotExist;
  714. }
  715. #ifndef _WIN32
  716. auto func = [nPriority] {
  717. switch (nPriority) {
  718. case Priority_Low:
  719. return TOOLKIT_PRIORITY_LOW;
  720. case Priority_Below_Normal:
  721. return TOOLKIT_PRIORITY_BELOW_NORMAL;
  722. case Priority_Normal:
  723. return TOOLKIT_PRIORITY_NORMAL;
  724. case Priority_Above_Normal:
  725. return TOOLKIT_PRIORITY_ABOVE_NORMAL;
  726. case Priority_High:
  727. return TOOLKIT_PRIORITY_HIGH;
  728. case Priority_Highest:
  729. return TOOLKIT_PRIORITY_HIGHEST;
  730. default:
  731. return TOOLKIT_PRIORITY_NORMAL;
  732. }
  733. };
  734. int res = toolkit_os_setpriority(ent->mod->process.pid, func());
  735. #else
  736. int priviliegeValue = TOOLKIT_PRIORITY_NORMAL;
  737. switch (nPriority) {
  738. case Priority_Low:
  739. priviliegeValue = TOOLKIT_PRIORITY_LOW;
  740. break;
  741. case Priority_Below_Normal:
  742. priviliegeValue = TOOLKIT_PRIORITY_BELOW_NORMAL;
  743. break;
  744. case Priority_Normal:
  745. priviliegeValue = TOOLKIT_PRIORITY_NORMAL;
  746. break;
  747. case Priority_Above_Normal:
  748. priviliegeValue = TOOLKIT_PRIORITY_ABOVE_NORMAL;
  749. break;
  750. case Priority_High:
  751. priviliegeValue = TOOLKIT_PRIORITY_HIGH;
  752. break;
  753. case Priority_Highest:
  754. priviliegeValue = TOOLKIT_PRIORITY_HIGHEST;
  755. break;
  756. default:
  757. priviliegeValue = TOOLKIT_PRIORITY_NORMAL;
  758. break;
  759. }
  760. int res = toolkit_os_setpriority(ent->mod->process.pid, priviliegeValue);
  761. #endif //NOT _WIN32
  762. if (res != 0) {
  763. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("toolkit_os_setpriority failed %s ", toolkit_strerror(res));
  764. return Error_Unexpect;
  765. }
  766. return Error_Succeed;
  767. }
  768. ErrorCodeEnum SpEntityPrivilege::GetEntityPriority(const char* pszEntityName, EntityPriorityEnum& nPriority)
  769. {
  770. if (!pszEntityName)
  771. return Error_Null;
  772. sp_env_t* env = sp_get_env();
  773. sp_mod_mgr_t* mod_mgr = env->mod_mgr;
  774. sp_entity_t* ent = sp_mod_mgr_find_entity_by_name(mod_mgr, pszEntityName);
  775. if (!ent) {
  776. return Error_NotExist;
  777. }
  778. int priority;
  779. int res = toolkit_os_getpriority(ent->mod->process.pid, &priority);
  780. if (res != 0) {
  781. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM)("toolkit_os_getpriority failed %s ", toolkit_strerror(res));
  782. return Error_Unexpect;
  783. }
  784. #if defined(_MSC_VER)
  785. switch (priority) {
  786. case TOOLKIT_PRIORITY_LOW:
  787. nPriority = Priority_Low;
  788. break;
  789. case TOOLKIT_PRIORITY_BELOW_NORMAL:
  790. nPriority = Priority_Below_Normal;
  791. break;
  792. case TOOLKIT_PRIORITY_NORMAL:
  793. nPriority = Priority_Normal;
  794. break;
  795. case TOOLKIT_PRIORITY_ABOVE_NORMAL:
  796. nPriority = Priority_Above_Normal;
  797. break;
  798. case TOOLKIT_PRIORITY_HIGH:
  799. nPriority = Priority_High;
  800. break;
  801. case TOOLKIT_PRIORITY_HIGHEST:
  802. nPriority = Priority_Highest;
  803. break;
  804. default:
  805. nPriority = Priority_Unknown;
  806. break;
  807. }
  808. #else
  809. auto func = [](int value) {
  810. switch (value) {
  811. case TOOLKIT_PRIORITY_LOW:
  812. return Priority_Low;
  813. case TOOLKIT_PRIORITY_BELOW_NORMAL:
  814. return Priority_Below_Normal;
  815. case TOOLKIT_PRIORITY_NORMAL:
  816. return Priority_Normal;
  817. case TOOLKIT_PRIORITY_ABOVE_NORMAL:
  818. return Priority_Above_Normal;
  819. case TOOLKIT_PRIORITY_HIGH:
  820. return Priority_High;
  821. case TOOLKIT_PRIORITY_HIGHEST:
  822. return Priority_Highest;
  823. default:
  824. return Priority_Unknown;
  825. }
  826. };
  827. nPriority = func(priority);
  828. #endif //_MSC_VER
  829. return Error_Succeed;
  830. }
  831. ErrorCodeEnum SpEntityPrivilege::GetToken(CSimpleString& curToken)
  832. {
  833. sp_env_t* env = sp_get_env();
  834. if (env->cfg->shell_ini->token == NULL || strlen(env->cfg->shell_ini->token) == 0)
  835. return ErrorCodeEnum::Error_NotConfig;
  836. curToken = env->cfg->shell_ini->token;
  837. return Error_Succeed;
  838. }
  839. ErrorCodeEnum SpEntityPrivilege::GetVTMErrMsgArr(CAutoArray<CSimpleStringA>& strErrorCodeArr, CAutoArray<CSimpleStringA>& strDescriptionArr,
  840. CAutoArray<CSimpleStringA>& strRemarkArr)
  841. {
  842. sp_env_t* env = sp_get_env();
  843. if(env->cfg->root_ini->vtm_err_msg_config == NULL || strlen(env->cfg->root_ini->vtm_err_msg_config) == 0)
  844. return ErrorCodeEnum::Error_NotConfig;
  845. return (ErrorCodeEnum)ConvertStrToVTMErrMsg(env->cfg->root_ini->vtm_err_msg_config, strErrorCodeArr, strDescriptionArr, strRemarkArr);
  846. }
  847. ErrorCodeEnum SpEntityPrivilege::RewriteDepVersion(const CSimpleStringA& strVersion, bool bRefresh)
  848. {
  849. ErrorCodeEnum rc(Error_Succeed);
  850. auto dir = sp_get_env()->dir;
  851. auto cfg = sp_get_env()->cfg;
  852. if (dir == NULL)
  853. return Error_Null;
  854. if (strVersion.IsNullOrEmpty())
  855. return Error_Param;
  856. CSimpleStringA strOldVer(true);
  857. rc = GetRunningDepVersion(strOldVer);
  858. if (rc == Error_Succeed && !strOldVer.IsNullOrEmpty() && strOldVer.Compare(strVersion) == 0) {
  859. return Error_AlreadyExist;
  860. }
  861. char path[MAX_PATH] = {'\0'};
  862. rc = (ErrorCodeEnum)sp_dir_get_path(dir, SP_DIR_DEPVER_TXT, NULL, path, MAX_PATH);
  863. if (rc != 0) {
  864. DbgWithLink(LOG_LEVEL_ERROR, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("get dep ver txt failed!");
  865. return rc;
  866. }
  867. else {
  868. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("path:%s", path);
  869. }
  870. rc = UpdateVerTxt(path, strVersion, true);
  871. if (rc == Error_Succeed && bRefresh) {
  872. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("to refresh...");
  873. rc = (ErrorCodeEnum)sp_dir_refresh_dep_path(dir);
  874. if (rc == Error_Succeed) {
  875. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("to send fresh post");
  876. iobuffer_t* pkt = iobuffer_create(-1, -1);
  877. rc = PostInfoShell(SHELL_CMD_INFO_REFRESH_ENV, &pkt);
  878. if (pkt)
  879. iobuffer_dec_ref(pkt);
  880. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("send fresh post return %s", SpStrError(rc));
  881. }
  882. }
  883. return rc;
  884. }
  885. ErrorCodeEnum SpEntityPrivilege::TryUpdateToken(CSimpleString& oldToken, CSimpleString& newToken)
  886. {
  887. char t_oldToken[MAX_PATH] = "", t_newToken[MAX_PATH] = "";
  888. auto ret = sp_TryUpdateToken(t_oldToken, t_newToken);
  889. if (ret == Error_Succeed)
  890. {
  891. oldToken = t_oldToken;
  892. newToken = t_newToken;
  893. }
  894. return (ErrorCodeEnum)ret;
  895. }
  896. ErrorCodeEnum SpEntityPrivilege::InitCfgUrl(VTMInitParam& info)
  897. {
  898. auto cfg = sp_get_env()->cfg;
  899. if (cfg == NULL || cfg->shell_ini == NULL || cfg->root_ini == NULL)
  900. return Error_Null;
  901. if (info.terminalNo.GetLength() > 0)
  902. {
  903. if (cfg->root_ini->terminal_no)
  904. shm_free(cfg->root_ini->terminal_no);
  905. cfg->root_ini->terminal_no = shm_strdup(info.terminalNo.GetData());
  906. }
  907. if (info.channelId.GetLength() > 0)
  908. {
  909. if (cfg->shell_ini->channelId)
  910. shm_free(cfg->shell_ini->channelId);
  911. cfg->shell_ini->channelId = shm_strdup(info.channelId.GetData());
  912. }
  913. if (info.tokenSecret.GetLength() > 0)
  914. {
  915. if (cfg->shell_ini->tokenSecret)
  916. shm_free(cfg->shell_ini->tokenSecret);
  917. cfg->shell_ini->tokenSecret = shm_strdup(info.tokenSecret.GetData());
  918. }
  919. if (info.CommonLaunchUrl.GetLength() > 0)
  920. {
  921. if (cfg->shell_ini->CommonLaunchUrl)
  922. shm_free(cfg->shell_ini->CommonLaunchUrl);
  923. cfg->shell_ini->CommonLaunchUrl = shm_strdup(info.CommonLaunchUrl.GetData());
  924. }
  925. if (info.CenterConfigTotal.GetLength() > 0)
  926. {
  927. if (cfg->shell_ini->CenterConfigTotal)
  928. shm_free(cfg->shell_ini->CenterConfigTotal);
  929. cfg->shell_ini->CenterConfigTotal = shm_strdup(info.CenterConfigTotal.GetData());
  930. }
  931. DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("TerminalNo:%s, CommonLaunchUrl:%s, CenterConfigTotal:%s"
  932. , cfg->root_ini->terminal_no == NULL ? "" : cfg->root_ini->terminal_no
  933. , cfg->shell_ini->CommonLaunchUrl == NULL ? "" : cfg->shell_ini->CommonLaunchUrl
  934. , cfg->shell_ini->CenterConfigTotal == NULL ? "" : cfg->shell_ini->CenterConfigTotal);
  935. return Error_Succeed;
  936. }
  937. ErrorCodeEnum SpEntityPrivilege::TryUpdateVTMERRMSG()
  938. {
  939. LOG_FUNCTION();
  940. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  941. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_TRY_UPDATE_VTMERRMSG);
  942. ErrorCodeEnum Error = pAsyncWait->Begin();
  943. if (Error == Error_Succeed)
  944. {
  945. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  946. if (Error == Error_Succeed)
  947. {
  948. CAutoBuffer AnsBuf;
  949. bool bEnd;
  950. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  951. if (Error == Error_Succeed)
  952. {
  953. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  954. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  955. }
  956. }
  957. }
  958. pAsyncWait->DecrementRef();
  959. if (req_pkt)
  960. iobuffer_dec_ref(req_pkt);
  961. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("TryUpdateVTMERRMSG ret:%d", Error);
  962. return Error;
  963. }
  964. ErrorCodeEnum SpEntityPrivilege::TryUpdateCfg()
  965. {
  966. LOG_FUNCTION();
  967. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  968. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_TRY_UPDATE_CFG);
  969. ErrorCodeEnum Error = pAsyncWait->Begin();
  970. if (Error == Error_Succeed)
  971. {
  972. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  973. if (Error == Error_Succeed)
  974. {
  975. CAutoBuffer AnsBuf;
  976. bool bEnd;
  977. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  978. if (Error == Error_Succeed)
  979. {
  980. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  981. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  982. }
  983. }
  984. }
  985. pAsyncWait->DecrementRef();
  986. if (req_pkt)
  987. iobuffer_dec_ref(req_pkt);
  988. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("TryUpdateCfg ret:%d", Error);
  989. return Error;
  990. }
  991. ErrorCodeEnum SpEntityPrivilege::ModifyMemCfgParam(MemCfgParam& info)
  992. {
  993. LOG_FUNCTION();
  994. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  995. iobuffer_write(req_pkt, IOBUF_T_STR, info.configType, -1);
  996. iobuffer_write(req_pkt, IOBUF_T_STR, info.module.GetData(), -1);
  997. iobuffer_write(req_pkt, IOBUF_T_STR, info.name.GetData(), -1);
  998. iobuffer_write(req_pkt, IOBUF_T_STR, info.value.GetData(), -1);
  999. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_MODIFY_MEM_CFG);
  1000. ErrorCodeEnum Error = pAsyncWait->Begin();
  1001. if (Error == Error_Succeed)
  1002. {
  1003. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  1004. if (Error == Error_Succeed)
  1005. {
  1006. CAutoBuffer AnsBuf;
  1007. bool bEnd;
  1008. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  1009. if (Error == Error_Succeed)
  1010. {
  1011. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  1012. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  1013. }
  1014. }
  1015. }
  1016. pAsyncWait->DecrementRef();
  1017. if (req_pkt)
  1018. iobuffer_dec_ref(req_pkt);
  1019. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("ModifyMemCfgParam ret:%d", Error);
  1020. return Error;
  1021. }
  1022. ErrorCodeEnum SpEntityPrivilege::WriteTerminalNoToRootIni(CSimpleString terminalNo)
  1023. {
  1024. LOG_FUNCTION();
  1025. iobuffer_t* req_pkt = iobuffer_create(-1, -1);
  1026. iobuffer_write(req_pkt, IOBUF_T_STR, terminalNo.GetData(), -1);
  1027. SpAsyncWaitRPC* pAsyncWait = new SpAsyncWaitRPC(this, &req_pkt, SHELL_CMD_REQ_WRITE_TERMINALNO);
  1028. ErrorCodeEnum Error = pAsyncWait->Begin();
  1029. if (Error == Error_Succeed)
  1030. {
  1031. Error = pAsyncWait->WaitAnswer(30000);//WaitAnswer如果设定为INFINE,则最大10s,在http访问中会触发超时
  1032. if (Error == Error_Succeed)
  1033. {
  1034. CAutoBuffer AnsBuf;
  1035. bool bEnd;
  1036. Error = pAsyncWait->AsyncGetAnswer(AnsBuf, bEnd);
  1037. if (Error == Error_Succeed)
  1038. {
  1039. iobuffer_t* pkt = iobuffer_create(-1, AnsBuf.GetCount());
  1040. iobuffer_write(pkt, IOBUF_T_BUF, &AnsBuf[0], AnsBuf.GetCount());
  1041. }
  1042. }
  1043. }
  1044. pAsyncWait->DecrementRef();
  1045. if (req_pkt)
  1046. iobuffer_dec_ref(req_pkt);
  1047. DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("WriteTerminalNoToRootIni ret:%d", Error);
  1048. return Error;
  1049. }
  1050. ErrorCodeEnum SpEntityPrivilege::TryUpdateCenterCfg(bool& isUpdate, bool& isReset, CSimpleString& version)
  1051. {
  1052. char t_version[MAX_PATH] = "";
  1053. int t_isUpdate = 0, t_isReset = 0;
  1054. auto ret = sp_TryUpdateCenterCfg(&t_isUpdate, &t_isReset, t_version);
  1055. if (ret == Error_Succeed)
  1056. {
  1057. isUpdate = t_isUpdate;
  1058. isReset = t_isReset;
  1059. version = t_version;
  1060. }
  1061. return (ErrorCodeEnum)ret;
  1062. }
  1063. void SpEntityPrivilege::GetSendLogInfo(unsigned long* t_upload_TerminalSys_Suc, unsigned long* t_upload_TerminalUser_Suc,
  1064. unsigned long* t_upload_BussinessSys_Suc, unsigned long* t_upload_BussinessUser_Suc, unsigned long* t_upload_beidou_Suc,
  1065. unsigned long* t_upload_TerminalSys_Err, unsigned long* t_upload_TerminalUser_Err,
  1066. unsigned long* t_upload_BussinessSys_Err, unsigned long* t_upload_BussinessUser_Err, unsigned long* t_upload_beidou_Err)
  1067. {
  1068. #if defined(_MSC_VER)
  1069. 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,
  1070. t_upload_TerminalSys_Err, t_upload_TerminalUser_Err, t_upload_BussinessSys_Err, t_upload_BussinessUser_Err, t_upload_beidou_Err);
  1071. #else
  1072. ///*TODO(80374374@3/9/2023): */
  1073. #endif //_MSC_VER
  1074. }
  1075. void SpEntityPrivilege::on_entity_create(int entity_id, int trigger_entity_id)
  1076. {
  1077. if (m_pEntityLifeListener) {
  1078. sp_env_t *env = sp_get_env();
  1079. IEntityLifeListener *pCallback = (IEntityLifeListener *)sp_mod_entity_life_listener_get_tag(m_pEntityLifeListener);
  1080. sp_mod_mgr_t *mod_mgr = env->mod_mgr;
  1081. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, entity_id);
  1082. sp_entity_t *trigger_ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, trigger_entity_id);
  1083. pCallback->OnCreated(ent->cfg->name, Error_Succeed, trigger_ent->cfg->name);
  1084. }
  1085. }
  1086. void SpEntityPrivilege::on_entity_close(int entity_id, int trigger_entity_id, int cause_code)
  1087. {
  1088. if (m_pEntityLifeListener) {
  1089. sp_env_t *env = sp_get_env();
  1090. IEntityLifeListener *pCallback = (IEntityLifeListener *)sp_mod_entity_life_listener_get_tag(m_pEntityLifeListener);
  1091. sp_mod_mgr_t *mod_mgr = env->mod_mgr;
  1092. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, entity_id);
  1093. sp_entity_t *trigger_ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, trigger_entity_id);
  1094. pCallback->OnClosed(ent->cfg->name, (EntityCloseCauseEnum)cause_code, Error_Succeed, trigger_ent->cfg->name);
  1095. }
  1096. }
  1097. void SpEntityPrivilege::on_entity_exception(int entity_id, int error, int entity_state)
  1098. {
  1099. if (m_pEntityLifeListener) {
  1100. sp_env_t *env = sp_get_env();
  1101. IEntityLifeListener *pCallback = (IEntityLifeListener *)sp_mod_entity_life_listener_get_tag(m_pEntityLifeListener);
  1102. sp_mod_mgr_t *mod_mgr = env->mod_mgr;
  1103. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(mod_mgr, entity_id);
  1104. pCallback->OnException(ent->cfg->name, "", (EntityStateEnum)entity_state, (EntityStateEnum)entity_state, (ErrorCodeEnum)error);
  1105. }
  1106. }
  1107. void SpEntityPrivilege::__on_entity_create(sp_mod_entity_life_listener_t *listener, int entity_id, int trigger_entity_id, void *user_data)
  1108. {
  1109. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1110. pThis->on_entity_create(entity_id, trigger_entity_id);
  1111. }
  1112. 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)
  1113. {
  1114. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1115. pThis->on_entity_close(entity_id, trigger_entity_id, cause_code);
  1116. }
  1117. void SpEntityPrivilege::__on_entity_exception(sp_mod_entity_life_listener_t *listener, int entity_id, int error, int entity_state, void *user_data)
  1118. {
  1119. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1120. pThis->on_entity_exception(entity_id, error, entity_state);
  1121. }
  1122. void SpEntityPrivilege::on_user_state(sp_mod_entity_state_listener_t *listener, int entity_id, int last_state, int curr_state)
  1123. {
  1124. sp_env_t *env = sp_get_env();
  1125. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1126. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, entity_id);
  1127. pListener->OnUserStateHook(ent->cfg->name, (DWORD)curr_state, (DWORD)last_state);
  1128. }
  1129. 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)
  1130. {
  1131. sp_env_t *env = sp_get_env();
  1132. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1133. sp_entity_t *ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, entity_id);
  1134. sp_entity_t *trigger_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, trigger_entity_id);
  1135. pListener->OnEntityStateHook(ent->cfg->name, trigger_ent->cfg->name, (EntityStateEnum)curr_state, (EntityStateEnum)last_state);
  1136. }
  1137. void SpEntityPrivilege::on_create_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id)
  1138. {
  1139. sp_env_t *env = sp_get_env();
  1140. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1141. sp_entity_t *src_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, src_entity_id);
  1142. sp_entity_t *dst_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, dst_entity_id);
  1143. pListener->OnCeateConnection(src_ent->cfg->name, dst_ent->cfg->name);
  1144. }
  1145. void SpEntityPrivilege::on_close_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id)
  1146. {
  1147. sp_env_t *env = sp_get_env();
  1148. IEntityStateListener *pListener = (IEntityStateListener *)sp_mod_entity_listener_get_tag(listener);
  1149. sp_entity_t *src_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, src_entity_id);
  1150. sp_entity_t *dst_ent = sp_mod_mgr_find_entity_by_idx(env->mod_mgr, dst_entity_id);
  1151. pListener->OnCloseConnection(src_ent->cfg->name, dst_ent->cfg->name);
  1152. }
  1153. void SpEntityPrivilege::__on_user_state(sp_mod_entity_state_listener_t *listener, int entity_id, int last_state, int curr_state, void *user_data)
  1154. {
  1155. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1156. pThis->on_user_state(listener, entity_id, last_state, curr_state);
  1157. }
  1158. 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)
  1159. {
  1160. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1161. pThis->on_entity_state(listener, entity_id, trigger_entity_id, last_state, curr_state);
  1162. }
  1163. void SpEntityPrivilege::__on_create_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id, void *user_data)
  1164. {
  1165. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1166. pThis->on_create_connection(listener, src_entity_id, dst_entity_id);
  1167. }
  1168. void SpEntityPrivilege::__on_close_connection(sp_mod_entity_state_listener_t *listener, int src_entity_id, int dst_entity_id, void *user_data)
  1169. {
  1170. SpEntityPrivilege *pThis = static_cast<SpEntityPrivilege*>(user_data);
  1171. pThis->on_close_connection(listener, src_entity_id, dst_entity_id);
  1172. }