76 lines
2.8 KiB
C
76 lines
2.8 KiB
C
#ifndef SERVER_SESSION_MONITOR_H
|
||
#define SERVER_SESSION_MONITOR_H
|
||
|
||
#include <windows.h>
|
||
#include <wtsapi32.h>
|
||
|
||
#ifdef __cplusplus
|
||
extern "C" {
|
||
#endif
|
||
|
||
#pragma comment(lib, "wtsapi32.lib")
|
||
|
||
// GUI进程信息
|
||
typedef struct ServerAgentProcessInfo {
|
||
DWORD processId;
|
||
DWORD sessionId;
|
||
HANDLE hProcess;
|
||
ULONGLONG launchTime; // 启动时间 (GetTickCount64)
|
||
} ServerAgentProcessInfo;
|
||
|
||
// GUI进程数组（动态数组）
|
||
typedef struct ServerAgentProcessArray {
|
||
ServerAgentProcessInfo* items;
|
||
size_t count;
|
||
size_t capacity;
|
||
} ServerAgentProcessArray;
|
||
|
||
// 崩溃保护参数
|
||
#define CRASH_WINDOW_MS (5 * 60 * 1000) // 5分钟窗口
|
||
#define CRASH_THRESHOLD 3 // 3次崩溃触发保护
|
||
#define FAST_CRASH_TIME_MS 30000 // 30秒内退出视为快速崩溃
|
||
|
||
// 回调函数类型
|
||
typedef void (*AgentStartCallback)(DWORD processId, DWORD sessionId); // 代理启动回调
|
||
typedef void (*AgentExitCallback)(DWORD exitCode, ULONGLONG runtimeMs); // 代理退出回调（每次退出，包括正常和崩溃）
|
||
typedef void (*CrashCallback)(DWORD exitCode, ULONGLONG runtimeMs); // 崩溃回调，带退出代码和运行时间
|
||
typedef void (*CrashWindowCallback)(int crashCount, ULONGLONG firstCrashTime); // 崩溃窗口状态变化回调
|
||
typedef void (*CrashProtectionCallback)(void); // 崩溃保护回调
|
||
|
||
// 会话监控器结构
|
||
typedef struct ServerSessionMonitor {
|
||
HANDLE monitorThread;
|
||
BOOL running;
|
||
BOOL runAsUser; // TRUE: 以用户身份运行代理, FALSE: 以SYSTEM身份运行代理
|
||
CRITICAL_SECTION csProcessList;
|
||
ServerAgentProcessArray agentProcesses;
|
||
// 崩溃保护
|
||
int crashCount; // 窗口期内崩溃次数
|
||
ULONGLONG firstCrashTime; // 第一次崩溃时间（窗口起点）
|
||
BOOL crashProtected; // 是否已触发崩溃保护
|
||
// 回调函数
|
||
AgentStartCallback onAgentStart; // 代理启动时的回调（用于统计启动次数）
|
||
AgentExitCallback onAgentExit; // 代理退出时的回调（用于统计运行时间）
|
||
CrashCallback onCrash; // 每次崩溃时的回调（用于统计）
|
||
CrashWindowCallback onCrashWindowChange; // 崩溃窗口状态变化时的回调（用于持久化）
|
||
CrashProtectionCallback onCrashProtection; // 崩溃保护触发时的回调
|
||
} ServerSessionMonitor;
|
||
|
||
// 初始化会话监控器
|
||
void ServerSessionMonitor_Init(ServerSessionMonitor* self);
|
||
|
||
// 清理会话监控器资源
|
||
void ServerSessionMonitor_Cleanup(ServerSessionMonitor* self);
|
||
|
||
// 启动会话监控
|
||
BOOL ServerSessionMonitor_Start(ServerSessionMonitor* self);
|
||
|
||
// 停止会话监控
|
||
void ServerSessionMonitor_Stop(ServerSessionMonitor* self);
|
||
|
||
#ifdef __cplusplus
|
||
}
|
||
#endif
|
||
|
||
#endif /* SERVER_SESSION_MONITOR_H */
|