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 */
|