mod_UpgradeMgr.h 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. #ifndef RVC_MOD_UPGRADEMGR_H_
  2. #define RVC_MOD_UPGRADEMGR_H_
  3. #include "stdafx.h"
  4. #include "SpBase.h"
  5. #include "SimpleString.h"
  6. #include "UpgradeManager_server_g.h"
  7. #ifdef RVC_OS_WIN
  8. #else
  9. #include "SpTest.h"
  10. #include "modVer.h"
  11. #endif
  12. #include "UpgradeTaskFSM.h"
  13. using namespace UpgradeManager;
  14. #define UPGRADE_MGR_NOT_IMPLEMENT 0xF0000000 // 新升级模式未实现
  15. // 升级管理 UpgradeManager 0x506
  16. class CUpgradeMgrEntity : public CEntityBase, public ISysVarListener
  17. {
  18. public:
  19. CUpgradeMgrEntity(){
  20. m_IsSM3listSucc=0;
  21. m_beginSM3HashTime=0;
  22. m_strSM3Hash="";
  23. m_bStartUp=false;//首页是否进入
  24. m_initSucc=false;//状态机是否启动成功
  25. m_bSystemStartUp=false;//框架是否启动成功
  26. m_testResult=Error_Succeed;//默认是成功的
  27. m_iBlackFileSum=0;
  28. }
  29. virtual ~CUpgradeMgrEntity() {}
  30. virtual const char *GetEntityName() const { return "UpgradeManager"; }
  31. virtual bool IsService()const{return true;}
  32. #ifdef RVC_OS_WIN
  33. #else
  34. const char* GetEntityVersion() const override
  35. {
  36. return MODULE_VERSION_FULL;
  37. }
  38. ON_ENTITYT_TEST()
  39. #endif
  40. virtual void OnStarted();
  41. virtual void OnSelfTest(EntityTestEnum eTestType,CSmartPointer<ITransactionContext> pTransactionContext);
  42. virtual void OnPreClose(EntityCloseCauseEnum eCloseCause,CSmartPointer<ITransactionContext> pTransactionContext);
  43. virtual CServerSessionBase *OnNewSession(const char* /*pszRemoteEntityName*/, const char * /*pszParam*/);
  44. virtual void OnSysVarEvent(const char *pszKey, const char *pszValue,const char *pszOldValue,const char *pszEntityName);
  45. static char GetInstallStateVal(const InstallStateEnum enumVal);
  46. ErrorCodeEnum RegistLocalPack(const CSimpleStringA &strPackFile);
  47. DWORD RegistManualPack(const CSimpleStringA &strPackFile);
  48. DWORD GetManualPacks(CSimpleStringA &strManualPacks);
  49. ErrorCodeEnum CancelUpdate(const CSimpleStringA &strPackName);
  50. ErrorCodeEnum RollbackUpdate(const CSimpleStringA &strVersion);
  51. ErrorCodeEnum SwitchUpgrade(const CSimpleStringA &strPack);
  52. ErrorCodeEnum GetUpgradeState(bool &bInstalling, CSimpleStringA &strPackFile, CSimpleStringA &strExecID,
  53. char &nInstallState, bool &bSysInstall, bool &bLightPack, CSimpleStringA &strNewVersion);
  54. //检测active文件是否被锁
  55. ErrorCodeEnum testActive();
  56. //modify by lwt 新状态机加入方法
  57. ErrorCodeEnum NewStopMediaPlay();
  58. static void HttpsLogCallBack(const char* logtxt);
  59. public:
  60. map<CSimpleStringA, CSimpleStringA> m_FileHashMap; // 版本文件hash值集合
  61. CSimpleStringA m_strMD5List;
  62. CSimpleStringA m_strSM3Hash;//SM3 32位hash
  63. int m_IsSM3listSucc;//获取SM3哈希list是否成功,0:还未计算完毕,1:表示成功 ,-1表示计算失败
  64. DWORD m_beginSM3HashTime;//开始计算sm3hahs时间
  65. bool m_bStartUp;//是否系统首页启动成功
  66. bool m_bSystemStartUp;//框架启动成功
  67. CUpgradeTaskFSM m_taskFSM;
  68. bool bNewUpgradeMgr;//是否启用新状态机(很重要的变量)
  69. ErrorCodeEnum m_testResult;//实体级别握手变量
  70. CSimpleStringA m_grayLaunchUrl;//灰度发布url地址
  71. int m_CenterModel;//集中配置模式 0:旧模式 1:新旧并存模式 2:新模式
  72. bool m_initSucc;//状态机是否启动成功
  73. int m_iBlackFileSum;//黑名单文件数量
  74. };
  75. class CUpgradeMgrSession : public UpgradeMgrService_ServerSessionBase
  76. {
  77. public:
  78. CUpgradeMgrSession(CUpgradeMgrEntity *pEntity) : m_pEntity(pEntity) {}
  79. virtual ~CUpgradeMgrSession() {}
  80. virtual void Handle_RegistLocalPack(SpReqAnsContext<UpgradeMgrService_RegistLocalPack_Req, UpgradeMgrService_RegistLocalPack_Ans>::Pointer ctx)
  81. {
  82. DbgToBeidou(ctx->link, __FUNCTION__)();
  83. ErrorCodeEnum rc = m_pEntity->RegistLocalPack(ctx->Req.strPackFile);
  84. ctx->Answer(rc);
  85. }
  86. virtual void Handle_RegistManualPack(SpReqAnsContext<UpgradeMgrService_RegistManualPack_Req, UpgradeMgrService_RegistManualPack_Ans>::Pointer ctx)
  87. {
  88. LOG_FUNCTION();
  89. DbgToBeidou(ctx->link, __FUNCTION__)();
  90. DbgWithLink(LOG_LEVEL_INFO, ctx->link.checkEmpty() ? LOG_TYPE_SYSTEM : LOG_TYPE_USER).setAPI("Handle_IsDownloading")("RegistManualPack received!");
  91. DWORD dwRet = 0;
  92. CSimpleStringA strErrmsg="RegistManualPack new UpgradeMgr is not deal with";
  93. dwRet = m_pEntity->RegistManualPack(ctx->Req.strPackFile);
  94. ctx->Ans.Errcode = dwRet;
  95. ctx->Ans.ErrMsg = strErrmsg;
  96. ctx->Answer(Error_Succeed);
  97. }
  98. virtual void Handle_CancelUpdate(SpReqAnsContext<UpgradeMgrService_CancelUpdate_Req, UpgradeMgrService_CancelUpdate_Ans>::Pointer ctx)
  99. {
  100. DbgToBeidou(ctx->link, __FUNCTION__)();
  101. DbgWithLink(LOG_LEVEL_INFO, ctx->link.checkEmpty() ? LOG_TYPE_SYSTEM : LOG_TYPE_USER).setAPI("Handle_CancelUpdate")("CancelUpdate received!");
  102. ErrorCodeEnum rc = m_pEntity->CancelUpdate(ctx->Req.strPackFile);
  103. ctx->Answer(rc);
  104. }
  105. virtual void Handle_RollbackUpdate(SpReqAnsContext<UpgradeMgrService_RollbackUpdate_Req, UpgradeMgrService_RollbackUpdate_Ans>::Pointer ctx)
  106. {
  107. DbgToBeidou(ctx->link, __FUNCTION__)();
  108. DbgWithLink(LOG_LEVEL_INFO, ctx->link.checkEmpty() ? LOG_TYPE_SYSTEM : LOG_TYPE_USER).setAPI("Handle_RollbackUpdate")("RollbackUpdate received!");
  109. auto rc = m_pEntity->RollbackUpdate(ctx->Req.strVersion);
  110. ctx->Answer(rc);
  111. }
  112. virtual void Handle_GetUpgradeState(SpReqAnsContext<UpgradeMgrService_GetUpgradeState_Req, UpgradeMgrService_GetUpgradeState_Ans>::Pointer ctx)
  113. {
  114. DbgToBeidou(ctx->link, __FUNCTION__)();
  115. DbgWithLink(LOG_LEVEL_INFO, ctx->link.checkEmpty() ? LOG_TYPE_SYSTEM : LOG_TYPE_USER).setAPI("Handle_GetUpgradeState")("GetUpgradeState received!");
  116. auto rc = m_pEntity->GetUpgradeState(ctx->Ans.bInstalling, ctx->Ans.strPackFile, ctx->Ans.strExecID,
  117. ctx->Ans.cInstallState, ctx->Ans.bSysInstall, ctx->Ans.bLightPack, ctx->Ans.strNewVersion);
  118. ctx->Answer(rc);
  119. }
  120. virtual void Handle_SwitchUpgrade(SpReqAnsContext<UpgradeMgrService_SwitchUpgrade_Req, UpgradeMgrService_SwitchUpgrade_Ans>::Pointer ctx)
  121. {
  122. DbgToBeidou(ctx->link, __FUNCTION__)();
  123. DbgWithLink(LOG_LEVEL_INFO, ctx->link.checkEmpty() ? LOG_TYPE_SYSTEM : LOG_TYPE_USER).setAPI("Handle_SwitchUpgrade")("SwitchUpgrade received!");
  124. auto rc = m_pEntity->SwitchUpgrade(ctx->Req.strPackFile);
  125. ctx->Answer(rc);
  126. }
  127. virtual void Handle_GetManualPacks(SpReqAnsContext<UpgradeMgrService_GetManualPacks_Req, UpgradeMgrService_GetManualPacks_Ans>::Pointer ctx)
  128. {
  129. LOG_FUNCTION();
  130. DbgToBeidou(ctx->link, __FUNCTION__)();
  131. DbgWithLink(LOG_LEVEL_INFO, ctx->link.checkEmpty() ? LOG_TYPE_SYSTEM : LOG_TYPE_USER).setAPI("Handle_GetManualPacks")("GetManualPacks received!");
  132. DWORD dwRet = 0;
  133. CSimpleStringA strErrmsg= "GetManualPacks new UpgradeMgr is not deal with";
  134. CSimpleStringA strPackList="";
  135. dwRet = m_pEntity->GetManualPacks(strPackList);
  136. ctx->Ans.Errcode = dwRet;
  137. ctx->Ans.ErrMsg = strErrmsg;
  138. ctx->Answer(Error_Succeed);
  139. }
  140. private:
  141. CUpgradeMgrEntity *m_pEntity;
  142. };
  143. #endif //RVC_MOD_UPGRADEMGR_H_