myvtm/Module/mod_UpgradeMgr/mod_UpgradeMgr.cpp

#include "stdafx.h"
#include "mod_UpgradeMgr.h"
#include "LocalMediaPlay_client_g.h"



#include "fileutil.h"
#include <regex>
#include "RVCComm.h"

#ifdef RVC_OS_WIN
#include <io.h>
#else
#include <unistd.h>
#include <dirent.h>
#include <errno.h>
#endif
#include "EventCode.h"
#include "json/json.h"

namespace Task
{
	//查询灰度控制
	struct QueryUpgradeControlTaskReq : CHTTPReq {
		string m_reqStr;
		string ToJson() {
			return m_reqStr;
		}
	};
	struct QueryUpgradeControlTaskRet : CHTTPRet {
		string m_retStr;
		bool Parse(string strData) {
			m_retStr = strData;
			return true;
		}
	};

	struct InitFSMTask : public ITaskSp {
		CUpgradeMgrEntity* Mgr;
		explicit InitFSMTask(CUpgradeMgrEntity* e) : Mgr(e) {}
		void Process() {
			LOG_FUNCTION();

			Mgr->bNewUpgradeMgr = true;//默认新模式
			CSystemStaticInfo info;
			ErrorCodeEnum Err = Mgr->GetFunction()->GetSystemStaticInfo(info);
			if (Err != Error_Succeed) {
				DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI("InitFSMTask")("load GetSystemStaticInfo fail ");
				LogWarn(Severity_Low, Error_Exception, WARN_TASK_START_NEW_MODE, "init new upgradeTaskMgr");
			}
			else {
				CSimpleStringA msgStr = CSimpleStringA::Format("init new upgradeTaskMgr, version = %s , dep = %s ", info.InstallVersion.ToString().GetData(), info.CurrDepVersion.GetData());
				LogWarn(Severity_Low, Error_Exception, WARN_TASK_START_NEW_MODE, msgStr.GetData());
			}


			if (Mgr->bNewUpgradeMgr) {
				ErrorCodeEnum rc = Mgr->m_taskFSM.Init(Mgr);//启动新状态机	
				if (rc != Error_Succeed)
				{
					LogWarn(Severity_Middle, Error_Exception, ERR_TASK_INIT_NEW_UPGRADE, "init new upgradeTaskMgr FSM fail");
					Mgr->m_testResult = Error_InvalidState;//自检失败
					return;
				}
				else {
					DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI("InitFSMTask")("init new upgradeTaskMgr FSM succ");
					Mgr->m_initSucc = true;
				}

				Mgr->m_beginSM3HashTime = CSmallDateTime::GetNow();
			}
			else {
				LogWarn(Severity_Low, Error_Exception, WARN_TASK_START_OLD_MODE, "init old upgradeTaskMgr");
			}

		}
	};

	struct RollBackTask : public ITaskSp {
		CUpgradeMgrEntity* Mgr;
		CVersion historyVersion;
		explicit RollBackTask(CUpgradeMgrEntity* e, CVersion version) : Mgr(e) {
			historyVersion = version;
		}
		void Process() {
			bool loop = true;
			while (loop) {
				//判断是否继续执行
				CSmartPointer<IConfigInfo> spConfig;
				ErrorCodeEnum ret = Mgr->GetFunction()->OpenConfig(Config_CenterSetting, spConfig);
				int rollbackTemp = 0;
				if (ret == Error_Succeed) {
					spConfig->ReadConfigValueInt("UpgradeManager", "RollbackFlag", rollbackTemp);
					DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI("RollbackUpdate")("get centerSetting RollbackFlag=%d", rollbackTemp);
				}
				else {
					DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI("RollbackUpdate")("get centerSetting RollbackFlag fail,use default value 0");
				}
				if (rollbackTemp == 0) {
					//立即回退模式
					DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Immediate mode,begin rollback version");
					ErrorCodeEnum rc = Mgr->m_taskFSM.RollBackToHistoryVersion(historyVersion);
					if (rc == Error_Succeed)
					{
						// 通过事件通知健康实体
						LogEvent(Severity_Middle, Event_Req_Framework_Rollback, "rollback upgrade succeed");
					}
					else
					{
						LogWarn(Severity_Low, rc, ERR_TASK_ROLLBACK_FAIL, "rollback upgrade fail");
					}
					loop = false;//退出
				}
				else {
					//等待首页回退模式
					CSimpleStringA strValue;
					if (Mgr->GetFunction()->GetSysVar("UIState", strValue) == Error_Succeed && strValue.Compare("M") == 0) {
						// 已经进入首页,开始回退
						DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("enter main page, begin rollback version");
						ErrorCodeEnum rc = Mgr->m_taskFSM.RollBackToHistoryVersion(historyVersion);
						if (rc == Error_Succeed)
						{
							// 通过事件通知健康实体
							LogEvent(Severity_Middle, Event_Req_Framework_Rollback, "rollback upgrade succeed");
						}
						else
						{
							LogWarn(Severity_Low, rc, ERR_TASK_ROLLBACK_FAIL, "rollback upgrade fail");
						}
						loop = false;//退出
					}
					else {
						Sleep(30 * 1000);//继续循环
						DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("enter not main page, rollback version delay 30s");
					}
				}
			}
			Mgr->m_taskFSM.m_bRollbackTask = false;//线程退出
		}
	};

	struct RollBackDepTask : public ITaskSp {
		CUpgradeMgrEntity* Mgr;
		CSimpleStringA historyVersion;
		explicit RollBackDepTask(CUpgradeMgrEntity* e, CSimpleStringA depVersion) : Mgr(e) {
			historyVersion = depVersion;
		}
		void Process() {
			bool loop = true;
			while (loop) {
				//判断是否继续执行
				CSmartPointer<IConfigInfo> spConfig;
				ErrorCodeEnum ret = Mgr->GetFunction()->OpenConfig(Config_CenterSetting, spConfig);
				int rollbackTemp = 0;
				if (ret == Error_Succeed) {
					spConfig->ReadConfigValueInt("UpgradeManager", "RollbackFlag", rollbackTemp);
					DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("get centerSetting RollbackFlag=%d", rollbackTemp);
				}
				else {
					DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("get centerSetting RollbackFlag fail");
				}
				if (rollbackTemp == 0) {
					//立即回退模式
					DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("Immediate mode,begin rollback dep version");
					ErrorCodeEnum rc = Mgr->m_taskFSM.RollBackToHistoryDepVersion(historyVersion);
					if (rc == Error_Succeed)
					{
						// 通过事件通知健康实体
						LogEvent(Severity_Middle, Event_DeviceAdapterRollBack_Framework_Restart, "rollback upgrade succeed");
					}
					else
					{
						LogWarn(Severity_Low, rc, Event_DeviceAdapterRollBack_Framework_Restart, "rollback upgrade fail");
					}
					loop = false;//退出
				}
				else {
					//等待首页回退模式
					CSimpleStringA strValue;
					if (Mgr->GetFunction()->GetSysVar("UIState", strValue) == Error_Succeed && strValue.Compare("M") == 0) {
						// 已经进入首页,开始回退
						DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("enter main page, begin rollback dep version");
						ErrorCodeEnum rc = Mgr->m_taskFSM.RollBackToHistoryDepVersion(historyVersion);
						if (rc == Error_Succeed)
						{
							// 通过事件通知健康实体
							LogEvent(Severity_Middle, Event_DeviceAdapterRollBack_Framework_Restart, "rollback upgrade succeed");
						}
						else
						{
							LogWarn(Severity_Low, rc, Event_DeviceAdapterRollBack_Framework_Restart, "rollback upgrade fail");
						}
						loop = false;//退出
					}
					else {
						Sleep(30 * 1000);//继续循环
						DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("enter not main page, rollback dep version delay 30s");
					}
				}
			}
			Mgr->m_taskFSM.m_bRollbackTask = false;//线程退出
		}
	};
}

// 升级管理 UpgradeManager	0x506
void CUpgradeMgrEntity::OnStarted()
{
	// 初始化状态机
	//m_fsm.Init(this);

	auto pFunc = GetFunction();

	// 监视准入状态
	ErrorCodeEnum rc;
	CSimpleStringA strValue;
	if (pFunc->GetSysVar("UIState", strValue) == Error_Succeed && strValue.Compare("M") ==0)
	{
		// 已经进入首页状态
		DbgWithLink(LOG_LEVEL_DEBUG,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("system page isStartup 1");
		m_bStartUp =true;
	}		
	else
	{
		// 否则启动监控
		rc= pFunc->RegistSysVarEvent("UIState", this);
		if (rc != Error_Succeed)
		{
			LogWarn(Severity_Middle, rc, ERR_WRAN_REGIST_SYS_VAR_FAIL, CSimpleStringA::Format("RegistSysVarEvent UIState  is fail,%d",(int)rc).GetData());
			m_testResult=Error_InvalidState;//自检失败
		}else{
			DbgWithLink(LOG_LEVEL_DEBUG,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RegistSysVarEvent UIState succ");
		}

	}

	GetFunction()->SubscribeLog(m_uuidHeartBeat, this, Log_Event, Severity_Middle, Error_IgnoreAll, -1, "HeartBeat", false);

	//启动灰度控制判断，看启动什么状态机
	Task::InitFSMTask* task = new Task::InitFSMTask(this);
	rc = this->GetFunction()->PostThreadPoolTask(task);
	if (rc != Error_Succeed)
	{
		LogWarn(Severity_Middle, rc, ERR_TASK_INIT_NEW_UPGRADE, CSimpleStringA::Format("Post InitFSMTask task to Thread is fail,%d",(int)rc).GetData());
		m_testResult=Error_InvalidState;//自检失败
	}
}

void CUpgradeMgrEntity::OnSelfTest(EntityTestEnum eTestType,CSmartPointer<ITransactionContext> pTransactionContext)
{
	if (Test_ShakeHand == eTestType)
	{
		//根据不同情况返回握手情况，监控实体根据不同情况杀死实体进程
		pTransactionContext->SendAnswer(m_testResult); 
	}
}

void CUpgradeMgrEntity::OnPreClose(EntityCloseCauseEnum eCloseCause,CSmartPointer<ITransactionContext> pTransactionContext)
{
	pTransactionContext->SendAnswer(Error_Succeed); 
}

void CUpgradeMgrEntity::OnSysVarEvent(const char *pszKey, const char *pszValue,const char *pszOldValue,const char *pszEntityName)
{

	if(strcmp("UIState",pszKey)==0){
		if(!m_bStartUp){
			if (strcmp(pszValue, "M") ==0){
				DbgWithLink(LOG_LEVEL_DEBUG,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("system page isStartup 2");
				m_bStartUp = true;				
			}
		}
	}
}

void CUpgradeMgrEntity::OnLog(const CAutoArray<CUUID>& SubIDs, const CUUID nLogID, const LogTypeEnum eLogType, const SeverityLevelEnum eLevel, const DWORD dwSysError, const DWORD dwUserCode, const DWORD dwEntityInstanceID, const WORD wEntityDevelID, const CAutoArray<DWORD>& Param, const char* pszEntityName, const char* pszModuleName, const char* pszMessage, const linkContext& pLinkInfo)
{
	DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM)("user code:%x,from entity:%s", dwUserCode, pszEntityName);
	switch (dwUserCode) {

	case LOG_EVT_INC_VERSION_ROLLBACK:
		{
			DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("receive heartBeat instruction version to be rollback to %s", pszMessage);
			CSimpleStringA rollbackStr = pszMessage;
			this->RollbackUpdate(rollbackStr);
		}
		break;
	default:
		return;
	}
}

char CUpgradeMgrEntity::GetInstallStateVal(const InstallStateEnum enumVal)
{
	const struct {
		int ch;
		InstallStateEnum val;
	} tbl[] = {
		{'A', Install_Active},
		{'I', Install_Pending},
		{'S', Install_SetToStart},
		{'F', Install_FailRun},
		{'R', Install_RollBack},
		{'U', Install_Upgraded},
		{'C',Install_Cancelled},
		{'W',Install_WaitConfirm},
		{'D', Install_Installed},
	};
	int i;
	for (i = 0; i < sizeof(tbl)/sizeof(tbl[0]); ++i) {
		if (tbl[i].val == enumVal) {
			return tbl[i].ch;
		}
	}

	return ' ';  // error
}

//新状态机处理
ErrorCodeEnum CUpgradeMgrEntity::RollbackUpdate(const CSimpleStringA &strVersion)
{
	if(m_initSucc){
		if(bNewUpgradeMgr){
			DbgWithLink(LOG_LEVEL_DEBUG,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackUpdate new UpgradeMgr deal with");
			if(strVersion.GetLength()==0){
				LogWarn(Severity_Low, Error_NotImpl, ERR_TASK_ROLLBACK_FAIL, "version is empty , rollback upgrade fail");
				return Error_NotImpl;
			}else{
				//判定适配器or应用心跳回退
				string failReason = "";
				CSimpleStringA strErrMsg = "";
				ErrorCodeEnum err =  this->m_taskFSM.judgeRollbackTaskClass(strVersion, failReason, strErrMsg);
				if (err != Error_Succeed) {
					LogWarn(Severity_Low, Error_Exception, ERR_TASK_ROLLBACK_FAIL, CSimpleStringA::Format("judgeRollbackTaskClass fail,err=%s", strErrMsg.GetData()).GetData());
					DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("judgeRollbackTaskClass fail,err=%s", strErrMsg.GetData());
					return err;
				}
				else {
					if (!this->m_taskFSM.m_TempDepUpgradeFlag) {

						ErrorCodeEnum rc = Error_Succeed;
						DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("try rollback to version: [%s]", (const char*)strVersion);
						int w1, w2, w3, w4;
						int ret = sscanf(strVersion, "%d.%d.%d.%d", &w1, &w2, &w3, &w4);
						if (ret < 4)
						{
							DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("version [%s] parse fail", (const char*)strVersion);
							return Error_Param;
						}
						else
						{	//回退放入线程当中
							if (!m_taskFSM.m_bRollbackTask) {
								Task::RollBackTask* rTask = new Task::RollBackTask(this, CVersion(w1, w2, w3, w4));
								ErrorCodeEnum rc = this->GetFunction()->PostThreadPoolTask(rTask);
								if (Error_Succeed == rc) {
									DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackFlag thread start succ");
									m_taskFSM.m_bRollbackTask = true;//启动线程
									return Error_Succeed;
								}
								else {
									DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackFlag thread start fail : %d", (int)rc);
									return rc;
								}
							}
							else {
								DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackFlag thread is exist");
								return Error_Succeed;
							}
						}
					}
					else {
						//回退放入线程当中
						if (!m_taskFSM.m_bRollbackTask) {
							Task::RollBackDepTask* rTask = new Task::RollBackDepTask(this, strVersion);
							ErrorCodeEnum rc = this->GetFunction()->PostThreadPoolTask(rTask);
							if (Error_Succeed == rc) {
								DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackFlag Dep thread start succ");
								m_taskFSM.m_bRollbackTask = true;//启动线程
								return Error_Succeed;
							}
							else {
								DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackFlag Dep thread start fail : %d", (int)rc);
								return rc;
							}
						}
						else {
							DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackFlag Dep thread is exist");
							return Error_Succeed;
						}
					}
				}
			}
		}else{
			DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("RollbackUpdate old UpgradeMgr deal not with");
			return Error_Succeed;
		}
	}else{
		DbgWithLink(LOG_LEVEL_INFO,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("initFSM is not end");
		return Error_Pending;//状态机还未启动
	}
}

CServerSessionBase* CUpgradeMgrEntity::OnNewSession(const char* /*pszRemoteEntityName*/, const char * /*pszParam*/)
{
	return new CUpgradeMgrSession(this);
}


//新状态机处理
ErrorCodeEnum CUpgradeMgrEntity::GetUpgradeState(bool &bInstalling, CSimpleStringA &strPackFile, CSimpleStringA &strExecID,
	char &cInstallState, bool &bSysInstall, bool &bLightPack, CSimpleStringA &strNewVersion)
{
	if(m_initSucc){
		DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("GetUpgradeState new UpgradeMgr deal with");
		return m_taskFSM.GetUpgradeState(bInstalling, strPackFile, strExecID, cInstallState, bSysInstall, bLightPack, strNewVersion);
	}else{
		DbgWithLink(LOG_LEVEL_INFO,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("initFSM is not end");
		return Error_Pending;//状态机还未启动
	}
}

ErrorCodeEnum CUpgradeMgrEntity::testActive()
{
	CSimpleStringA rootVerPath;
	ErrorCodeEnum rc = this->GetFunction()->GetPath("RootVer",rootVerPath);//获取version根路径
	if(rc!=Error_Succeed){
		LogWarn(Severity_Middle, Error_Exception, ERR_WRAN_OPEN_ACTIVE_FAIL,"testActive fail , get RootVer path is fai");
		return Error_Bug;
	}

	CSimpleStringA strActiveFile = CSimpleStringA::Format("%s" SPLIT_SLASH_STR "active.txt", rootVerPath.GetData());
	FILE* fp = fopen(strActiveFile.GetData(),"rb+");
	if(fp==NULL){
		LogWarn(Severity_Middle, Error_Exception, ERR_WRAN_OPEN_ACTIVE_FAIL, CSimpleStringA::Format("open write active.txt fail,please edit active attribute, err=%d", (int)GetLastError()).GetData());
		return Error_Unexpect;
	}
	fclose(fp);
	return Error_Succeed;
}


ErrorCodeEnum CUpgradeMgrEntity::NewStopMediaPlay()
{

	// 通知媒体停止播放广告和声音
	LocalMediaPlay::PlayService_ClientBase *pClient = new LocalMediaPlay::PlayService_ClientBase(this);
	ErrorCodeEnum rc = pClient->Connect();
	if (rc == Error_Succeed)
	{
		DbgWithLink(LOG_LEVEL_DEBUG,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("connect to entity [LocalMediaPlay] succeed, start StopMediaPlay now");
		LocalMediaPlay::PlayService_StopPlayVideo_Req req1 = {};
		LocalMediaPlay::PlayService_StopPlayVideo_Ans ans1 = {};
		req1.CfgInx = 1;
		rc = (*pClient)(EntityResource::getLink().upgradeLink())->StopPlayVideo(req1, ans1, 30000);
		if (Error_Succeed == rc)
		{
			DbgWithLink(LOG_LEVEL_DEBUG,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("StopPlayVideo success");
		}else{
			DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("StopPlayVideo fail, err=%d", (int)rc);
			pClient->GetFunction()->CloseSession();
			return rc;
		}

		LocalMediaPlay::PlayService_StopPlayAudio_Req req2 = {};
		LocalMediaPlay::PlayService_StopPlayAudio_Ans ans2 = {};
		rc = (*pClient)(EntityResource::getLink().upgradeLink())->StopPlayAudio(req2, ans2, 30000);
		if (Error_Succeed == rc)
		{
			DbgWithLink(LOG_LEVEL_DEBUG, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("StopPlayAudio success");
		}else{
			DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("StopPlayAudio fail, err=%d", (int)rc);
			pClient->GetFunction()->CloseSession();
			return rc;
		}
		pClient->GetFunction()->CloseSession();
		return rc;
	}
	else
	{
		LogWarn(Severity_Low, rc, ERR_TASK_STOP_MEDIA_FAIL, "connect to entity [LocalMediaPlay] fail");
		pClient->SafeDelete();
		return rc;
	}
}

SP_BEGIN_ENTITY_MAP()
	SP_ENTITY(CUpgradeMgrEntity)
SP_END_ENTITY_MAP()

void CUpgradeMgrEntity::HttpsLogCallBack(const char* logtxt)
{
		if(logtxt!=NULL){
			DbgWithLink(LOG_LEVEL_INFO,LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("http dbg: %s",logtxt);
		}
}