浏览代码

!2 change add firewall wait time to ms

chenliangyu 3 月之前
父节点
当前提交
34b03d528f
共有 4 个文件被更改,包括 6 次插入5 次删除
  1. 1 1
      Framework/Common/SpBase.h
  2. 3 3
      Framework/spbase/SpEntity.cpp
  3. 1 1
      Framework/spbase/SpEntity.h
  4. 1 0
      Framework/spshell/spshell.cpp

+ 1 - 1
Framework/Common/SpBase.h

@@ -1060,7 +1060,7 @@ struct IEntityFunction
 
 
 	virtual ErrorCodeEnum GetToken(CSimpleString& channelId, CSimpleString& token) = 0;
 	virtual ErrorCodeEnum GetToken(CSimpleString& channelId, CSimpleString& token) = 0;
 
 
-	virtual ErrorCodeEnum SetNewPathToFirewall(CSimpleString& pszPath, int maxWaitTime = 30) = 0;
+	virtual ErrorCodeEnum SetNewPathToFirewall(CSimpleString& pszPath, int maxWaitMs = 30000) = 0;
 	/* GetRunningVersion
 	/* GetRunningVersion
 	currently return version in active.txt, may be it will return the version from micro sever latter.
 	currently return version in active.txt, may be it will return the version from micro sever latter.
 	*/
 	*/

+ 3 - 3
Framework/spbase/SpEntity.cpp

@@ -1748,7 +1748,7 @@ ErrorCodeEnum SpEntity::GetToken(CSimpleString &channelId, CSimpleString &token)
 		return ErrorCodeEnum::Error_Succeed;
 		return ErrorCodeEnum::Error_Succeed;
 }
 }
 
 
-ErrorCodeEnum SpEntity::SetNewPathToFirewall(CSimpleString &pszPath, int maxWaitTime)
+ErrorCodeEnum SpEntity::SetNewPathToFirewall(CSimpleString &pszPath, int maxWaitMs)
 {
 {
 	if (pszPath.GetLength() == 0) {
 	if (pszPath.GetLength() == 0) {
 		DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("SetNewPathToFirewall param error");
 		DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("SetNewPathToFirewall param error");
@@ -1764,7 +1764,7 @@ ErrorCodeEnum SpEntity::SetNewPathToFirewall(CSimpleString &pszPath, int maxWait
 
 
 	std::future<ErrorCodeEnum> result = std::async(std::launch::async, addFirewall_worker, pszPath.GetData());
 	std::future<ErrorCodeEnum> result = std::async(std::launch::async, addFirewall_worker, pszPath.GetData());
 
 
-	auto status = result.wait_for(std::chrono::seconds(maxWaitTime));
+	auto status = result.wait_for(std::chrono::milliseconds(maxWaitMs));
 	if (status == std::future_status::ready)
 	if (status == std::future_status::ready)
 	{
 	{
 		auto ret = result.get();
 		auto ret = result.get();
@@ -1773,7 +1773,7 @@ ErrorCodeEnum SpEntity::SetNewPathToFirewall(CSimpleString &pszPath, int maxWait
 	}
 	}
 	else
 	else
 	{
 	{
-		DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("SetNewPathToFirewall run over %d seconds", maxWaitTime);
+		DbgWithLink(LOG_LEVEL_INFO, LOG_TYPE_SYSTEM)("SetNewPathToFirewall run over %d milliseconds", maxWaitMs);
 		return Error_TimeOut;
 		return Error_TimeOut;
 	}
 	}
 
 

+ 1 - 1
Framework/spbase/SpEntity.h

@@ -125,7 +125,7 @@ public:
 	virtual ErrorCodeEnum GetEntityLogPath(const char* pszEntityName, CEntityLogInfo& Info);
 	virtual ErrorCodeEnum GetEntityLogPath(const char* pszEntityName, CEntityLogInfo& Info);
 	virtual WORD GetDevID();
 	virtual WORD GetDevID();
 	virtual ErrorCodeEnum GetToken(CSimpleString &channelId, CSimpleString &token);
 	virtual ErrorCodeEnum GetToken(CSimpleString &channelId, CSimpleString &token);
-	virtual ErrorCodeEnum SetNewPathToFirewall(CSimpleString &pszPath, int maxWaitTime = 30); // Input path:D:\\Runxxx\\version\\7.1.1.1
+	virtual ErrorCodeEnum SetNewPathToFirewall(CSimpleString &pszPath, int maxWaitMs = 30000); // Input path:D:\\Runxxx\\version\\7.1.1.1
 
 
 	// connect
 	// connect
 	virtual ErrorCodeEnum ConnectRemoteEntity(CClientSessionBase *pClientSession, const char *pszRemoteEntity,const char *pszParam,CSmartPointer<IAsynWaitSp> &pAsynWaitSp);
 	virtual ErrorCodeEnum ConnectRemoteEntity(CClientSessionBase *pClientSession, const char *pszRemoteEntity,const char *pszParam,CSmartPointer<IAsynWaitSp> &pAsynWaitSp);

+ 1 - 0
Framework/spshell/spshell.cpp

@@ -302,6 +302,7 @@ static bool AddFirewallRules()
 
 
 int AddFireAddFirewallRulesThread(void* param)
 int AddFireAddFirewallRulesThread(void* param)
 {
 {
+	LOG_FUNCTION();
 	if (!sp_CheckAllRules())
 	if (!sp_CheckAllRules())
 	{
 	{
 		DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("设置Windows防火墙策略失败!!!");
 		DbgWithLink(LOG_LEVEL_WARN, LOG_TYPE_SYSTEM).setAPI(__FUNCTION__)("设置Windows防火墙策略失败!!!");