From 0aaf9921425f32e2fe86bd1851335c78a0a4e0b0 Mon Sep 17 00:00:00 2001 From: wangxx1809 Date: Fri, 1 Mar 2024 16:33:16 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=99=BB=E5=BD=95=E5=92=8C?= =?UTF-8?q?=E6=B3=A8=E9=94=80=E5=88=A4=E5=AE=9A=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Packet.cpp | 35 ++++++++++++++++++++++++++--------- Packet.h | 7 +++++-- 2 files changed, 31 insertions(+), 11 deletions(-) diff --git a/Packet.cpp b/Packet.cpp index fb5ae2c..2cbb5c0 100644 --- a/Packet.cpp +++ b/Packet.cpp @@ -31,7 +31,8 @@ Packet::Packet() , m_intervalTime(0) , m_adhandle(nullptr) , m_quitFlag(false) //退出标记 - , m_isShowToast(false) { + , m_isShowToast(false) + , m_specialPackCount(0){ } @@ -52,7 +53,7 @@ void Packet::Init() { m_hbPort = 1433; //心跳端口 m_startTime = time(nullptr); //当前时间 m_serverIp = "139.159.230.183"; //服务器ip - m_intervalTime = 900; //时间间隔5分钟 + m_intervalTime = 180; //时间间隔5分钟 } @@ -224,11 +225,11 @@ void Packet::PacketHandler(u_char* param, const struct pcap_pkthdr* header, cons //printf(result); - pthis->CheckTimeOut(sport,dport,result); + pthis->CheckTimeOut(sport,dport,result, header->len); } -bool IsProcessRunning(const std::wstring& processName) { +bool Packet::IsProcessRunning(const std::wstring& processName) { HANDLE hProcessSnap; PROCESSENTRY32 pe32; @@ -255,7 +256,7 @@ bool IsProcessRunning(const std::wstring& processName) { } -void Packet::CheckTimeOut(u_short sport, u_short dport,const std::string& tcpInfoStr) { +void Packet::CheckTimeOut(u_short sport, u_short dport,const std::string& tcpInfoStr,int len) { bool isRun = IsProcessRunning(L"EdmServer.exe"); if (!isRun) return; //plm进程不存在，就不打印日志 @@ -263,8 +264,10 @@ void Packet::CheckTimeOut(u_short sport, u_short dport,const std::string& tcpInf if (dport == m_hbPort || sport == m_hbPort) { //心跳数据 int timeDiff = static_cast(time(nullptr) - GetStartTime()); if (timeDiff > m_intervalTime) { //超时，调用注销脚本 - LogoutPLM(); - LOG(DEBUG) << "Logout PLM end..."; + if (m_specialPackCount > 0) { + LogoutPLM(); + LOG(DEBUG) << "Logout PLM end..."; + } UpdateStartTime(); } else { @@ -275,6 +278,18 @@ void Packet::CheckTimeOut(u_short sport, u_short dport,const std::string& tcpInf hbCount = 0; } } + + //1.时间在弹窗前的最后两分钟内 m_intervalTime - timeDiff<120s + //2.服务端 接收端口1433 长度54 + if (m_intervalTime - timeDiff <= 120 ) { + if (dport == 1433 && 54 == len) { + ++m_specialPackCount; + LOG(DEBUG) << "m_specialPackCount:" << m_specialPackCount; + } + } + else { + m_specialPackCount = 0; + } } else { UpdateStartTime(); @@ -345,7 +360,9 @@ void Packet::UserHandleProc(){ m_checkThread = std::thread([this, uiPath]() { while (!m_quitFlag) { int timeDiff = static_cast(time(nullptr) - GetStartTime()); - if (IsProcessRunning(L"EdmServer.exe") && timeDiff >= m_intervalTime - 10 && timeDiff <= m_intervalTime - 9) { //超时，调用注销脚本 + //m_specialPackCount>0表示登录状态，超时，调用注销脚本 + if (m_specialPackCount>0 && IsProcessRunning(L"EdmServer.exe") + && timeDiff >= m_intervalTime - 10 && timeDiff <= m_intervalTime - 9) { LOG(DEBUG) << "pop ui begin..."; DWORD exit_code = -1; if (WaitUserHandle((LPWSTR)uiPath.data(), &exit_code)) { @@ -605,7 +622,7 @@ BOOL Packet::WaitUserHandle(LPWSTR command_line, DWORD* exit_code) GetExitCodeProcess(pi.hProcess, &exitCode); *exit_code = *exit_code== CANCEL ? *exit_code:exitCode; - LOG(DEBUG)<<"子进程ui退出码:"<< *exit_code; + LOG(DEBUG)<<"子进程ui退出码(1:确定，2:取消):"<< *exit_code; // 终止子进程 TerminateProcess(pi.hProcess, 0); diff --git a/Packet.h b/Packet.h index 8c469db..e9b4f8b 100644 --- a/Packet.h +++ b/Packet.h @@ -89,11 +89,14 @@ public: private: bool CheckKeepAlive(u_char* option); //无用 - void CheckTimeOut(u_short sport, u_short dport, const std::string& tcpInfoStr); //检测是否超时 + void CheckTimeOut(u_short sport, u_short dport, const std::string& tcpInfoStr,int len); //检测是否超时 unsigned long ConvertMaskToULong(const std::string& maskString); INT64 PopToast(); BOOL WaitUserHandle(LPWSTR command_line, DWORD* exit_code); //创建ui进程 void UserHandleProc(); + + + static bool IsProcessRunning(const std::wstring& processName); //进程是否在运行 static void FtpPacketHandler(const struct pcap_pkthdr* header, const u_char* pkt_data); static void PacketHandler(u_char* param, const struct pcap_pkthdr* header, const u_char* pkt_data); @@ -112,7 +115,7 @@ private: std::thread m_checkThread; //检测线程 bool m_isShowToast; //是否弹出了toast; - + int m_specialPackCount; //累计特殊包次数（弹窗前最后两分钟内），注销后为0，登录后大于0 }; class MyToastHandler :public WinToastLib::IWinToastHandler {