#pragma once #include #include #include "IOCPServer.h" #include "..\..\client\CursorInfo.h" #include "VideoDlg.h" #include "ToolbarDlg.h" #include "2015RemoteDlg.h" extern "C" { #include "libavcodec\avcodec.h" #include "libavutil\avutil.h" #include "libyuv\libyuv.h" } #ifndef _WIN64 // https://github.com/Terodee/FFMpeg-windows-static-build/releases #pragma comment(lib,"ffmpeg/libavcodec.lib") #pragma comment(lib,"ffmpeg/libavutil.lib") #pragma comment(lib,"ffmpeg/libswresample.lib") #pragma comment(lib,"libyuv/libyuv.lib") #else #pragma comment(lib,"x264/libx264_x64.lib") #pragma comment(lib,"libyuv/libyuv_x64.lib") // https://github.com/ShiftMediaProject/FFmpeg #pragma comment(lib,"ffmpeg/libavcodec_x64.lib") #pragma comment(lib,"ffmpeg/libavutil_x64.lib") #pragma comment(lib,"ffmpeg/libswresample_x64.lib") #endif #pragma comment(lib, "Mfplat.lib") #pragma comment(lib, "Mfuuid.lib") #pragma comment(lib, "Bcrypt.lib") #pragma comment(lib, "Strmiids.lib") // 文件接收消息（用于将工作线程的文件数据转发到主线程处理） #define WM_RECVFILEV2_CHUNK (WM_USER + 0x200) #define WM_RECVFILEV2_COMPLETE (WM_USER + 0x201) // ScreenSpyDlg 系统菜单命令 ID enum { IDM_CONTROL = 0x1010, IDM_FULLSCREEN, IDM_SEND_CTRL_ALT_DEL, IDM_TRACE_CURSOR, // 跟踪显示远程鼠标 IDM_BLOCK_INPUT, // 锁定远程计算机输入 IDM_SAVEDIB, // 保存图片 IDM_GET_CLIPBOARD, // 获取剪贴板 IDM_SET_CLIPBOARD, // 设置剪贴板 IDM_ADAPTIVE_SIZE, IDM_SAVEAVI, IDM_SAVEAVI_H264, IDM_SWITCHSCREEN, IDM_MULTITHREAD_COMPRESS, IDM_FPS_10, IDM_FPS_15, IDM_FPS_20, IDM_FPS_25, IDM_FPS_30, IDM_FPS_UNLIMITED, IDM_ORIGINAL_SIZE, IDM_SCREEN_1080P, IDM_REMOTE_CURSOR, IDM_SCROLL_DETECT_OFF, // 滚动检测：关闭（局域网） IDM_SCROLL_DETECT_2, // 滚动检测：跨网推荐 IDM_SCROLL_DETECT_4, // 滚动检测：标准模式 IDM_SCROLL_DETECT_8, // 滚动检测：省CPU模式 IDM_QUALITY_OFF, // 关闭质量控制（使用原有算法） IDM_ADAPTIVE_QUALITY, // 自适应质量 IDM_QUALITY_ULTRA, // 手动质量：Ultra IDM_QUALITY_HIGH, // 手动质量：High IDM_QUALITY_GOOD, // 手动质量：Good IDM_QUALITY_MEDIUM, // 手动质量：Medium IDM_QUALITY_LOW, // 手动质量：Low IDM_QUALITY_MINIMAL, // 手动质量：Minimal IDM_ENABLE_SSE2, IDM_FAST_STRETCH, // 快速缩放模式（降低CPU占用） IDM_CUSTOM_CURSOR, // 使用自定义光标 IDM_RESTORE_CONSOLE, // RDP会话归位 IDM_RESET_VIRTUAL_DESKTOP, // 重置虚拟桌面 IDM_AUDIO_TOGGLE, // 音频开关 }; // 状态信息窗口 - 全屏时显示帧率/速度/质量 class CStatusInfoWnd : public CWnd { public: CStatusInfoWnd() : m_nOpacityLevel(0), m_bVisible(false), m_bDragging(false) {} BOOL Create(CWnd* pParent); void UpdateInfo(double fps, double kbps, const CString& quality); void Show(); void Hide(); void SetOpacityLevel(int level); void UpdatePosition(const RECT& rcMonitor); void LoadSettings(); void SaveSettings(); bool IsVisible() const { return m_bVisible; } bool IsParentInControlMode(); // 检查父窗口是否处于控制模式 protected: afx_msg void OnPaint(); afx_msg BOOL OnEraseBkgnd(CDC* pDC); afx_msg void OnLButtonDown(UINT nFlags, CPoint point); afx_msg void OnLButtonUp(UINT nFlags, CPoint point); afx_msg void OnMouseMove(UINT nFlags, CPoint point); DECLARE_MESSAGE_MAP() private: CString m_strInfo; int m_nOpacityLevel; bool m_bVisible; // 拖动支持 bool m_bDragging; CPoint m_ptDragStart; // 位置保存 double m_dOffsetXRatio = 0.5; int m_nOffsetY = 50; bool m_bHasCustomPosition = false; }; // CScreenSpyDlg 对话框 class CScreenSpyDlg : public DialogBase { DECLARE_DYNAMIC(CScreenSpyDlg) CToolbarDlg* m_pToolbar = nullptr; CMy2015RemoteDlg* m_pParent = nullptr; public: CStatusInfoWnd* m_pStatusInfoWnd = nullptr; // MaxFPS=20, ScrollDetectInterval=2, Reserved={}, Capabilities=0 // MaxFPS=20, CompressThread=0, ScreenStrategy=0, ScreenWidth=0, ScreenHeight=0, // FullScreen=0, RemoteCursor=0, ScrollDetectInterval=2, QualityLevel=-1, // CpuSpeedup=0, ScreenType=0, AudioEnabled=0, Reserved={}, Capabilities=0 ScreenSettings m_Settings = { 20, 0, 0, 0, 0, 0, 0, 2, -1, 0, 0, 0, {}, 0 }; public: // 快速缩放模式（全局配置，所有实例共享） static int s_nFastStretch; // -1=未初始化, 0=关闭, 1=开启 static bool GetFastStretchMode(); static void SetFastStretchMode(bool bFast); CScreenSpyDlg(CMy2015RemoteDlg* Parent, Server* IOCPServer=NULL, CONTEXT_OBJECT *ContextObject=NULL); virtual ~CScreenSpyDlg(); virtual BOOL ShouldReconnect() { return TRUE; } VOID SendNext(void); VOID OnReceiveComplete(); HDC m_hFullDC; HDC m_hFullMemDC; HBITMAP m_BitmapHandle; PVOID m_BitmapData_Full; LPBITMAPINFO m_BitmapInfor_Full; VOID DrawFirstScreen(void); VOID DrawNextScreenDiff(bool keyFrame); VOID DrawScrollFrame(void); BOOL m_bIsFirst; bool m_bQualitySwitch = false; // 质量切换中，不显示"请等待" ULONG m_ulHScrollPos; ULONG m_ulVScrollPos; // fillMode: 0=不填充, 1=全黑, 2=半透明 VOID DrawTipString(CString strString, int fillMode=1); POINT m_ClientCursorPos; BYTE m_bCursorIndex; BOOL m_bIsTraceCursor; CCursorInfo m_CursorInfo; //自定义的一个系统的光标类 VOID SendCommand(const MYMSG* Msg); void SendScaledMouseMessage(MSG* pMsg, bool makeLP); VOID UpdateServerClipboard(char *szBuffer,ULONG ulLength); VOID SendServerClipboard(void); BOOL m_bIsCtrl; LPBYTE m_szData; BOOL m_bSend; ULONG m_ulMsgCount; int m_FrameID; HIMC m_hOldIMC = NULL; // 保存原始 IME 上下文，控制模式切换时使用 bool m_bHide = false; std::string m_strSaveNotice; // 截图保存路径提示 ULONGLONG m_nSaveNoticeTime = 0; // 截图提示开始时间 BOOL m_bUsingFRP = FALSE; // 文件接收进度对话框（用于 Linux Ctrl+C -> 服务端 Ctrl+V） // 按 transferID 管理多个并发传输 std::map m_FileRecvDlgs; void SaveSnapshot(void); // 对话框数据 enum { IDD = IDD_DIALOG_SCREEN_SPY }; WINDOWPLACEMENT m_struOldWndpl; const AVCodec* m_pCodec; AVCodecContext* m_pCodecContext; AVPacket m_AVPacket; AVFrame m_AVFrame; clock_t m_lastMouseMove; // 鼠标移动时间 POINT m_lastMousePoint;// 上次鼠标位置 BOOL m_bAdaptiveSize = TRUE; HCURSOR m_hRemoteCursor = NULL; HCURSOR m_hCustomCursor = NULL; // 缓存的自定义光标 DWORD m_dwCustomCursorHash = 0; // 当前自定义光标哈希 BOOL m_bUseCustomCursor = TRUE; // 是否使用自定义光标 CRect m_CRect; double m_wZoom=1, m_hZoom=1; bool m_bMouseTracking = false; CString m_aviFile; CBmpToAvi m_aviStream; // 传输速率统计 ULONG m_ulBytesThisSecond = 0; // 本秒累计字节 double m_dTransferRate = 0; // 当前速率 (KB/s) // 帧率统计 (使用EMA平滑) ULONG m_ulFramesThisSecond = 0; // 本秒累计帧数 double m_dFrameRate = 0; // 平滑后的帧率 (FPS) // 自适应质量 struct { bool enabled = false; // 是否启用自适应 (默认关闭) int currentLevel = QUALITY_HIGH; // 当前质量等级 int currentMaxWidth = 0; // 当前分辨率限制 (0=原始) int lastRTT = 0; // 上次RTT值 ULONGLONG lastChangeTime = 0; // 上次切换时间 ULONGLONG lastResChangeTime = 0; // 上次分辨率变化时间 ULONGLONG startTime = 0; // 启动时间 (用于延迟启动自适应) int stableCount = 0; // 稳定计数 (用于防抖) } m_AdaptiveQuality; volatile bool m_bResolutionChanging = false; // 分辨率切换中，阻止解码 // ========== 音频播放 ========== // m_Settings.AudioEnabled 表示是否启用音频（与客户端同步） BOOL m_bAudioPlaying = FALSE; // 音频是否正在播放 HWAVEOUT m_hWaveOut = NULL; // 波形输出设备句柄 WAVEFORMATEX m_AudioFormat = {}; // 音频格式 static const int AUDIO_BUFFER_COUNT = 8; // 缓冲区数量（增加到8个） WAVEHDR m_WaveHdr[AUDIO_BUFFER_COUNT] = {}; // 波形头 LPBYTE m_pAudioBuf[AUDIO_BUFFER_COUNT] = {};// 音频缓冲区 int m_nAudioBufIndex = 0; // 当前缓冲区索引 static const DWORD AUDIO_BUF_SIZE = 8192; // 8KB 每个缓冲区（更小更频繁） // 环形缓冲区（吸收网络抖动） static const DWORD RING_BUF_SIZE = 65536; // 64KB 环形缓冲区 BYTE* m_pRingBuf = nullptr; // 环形缓冲区 DWORD m_nRingHead = 0; // 写入位置 DWORD m_nRingTail = 0; // 读取位置 DWORD m_nRingDataLen = 0; // 缓冲数据量 int m_nPrebufferCount = 0; // 预缓冲计数 static const int PREBUFFER_TARGET = 5; // 预缓冲目标（积累5个包再开始播放，约50ms） BYTE m_nAudioCompression = 0; // 音频压缩类型 (AudioCompression) #if USING_OPUS void* m_pOpusDecoder = nullptr; // Opus 解码器 short* m_pOpusDecodeBuffer = nullptr; // Opus 解码输出缓冲区 #endif void OnAudioData(BYTE* pData, UINT32 len); // 处理音频数据 BOOL InitAudioPlayback(const AudioFormat* fmt); // 初始化音频播放 void StopAudioPlayback(); // 停止音频播放 void SendAudioCtrl(BYTE enable, BYTE persist); // 发送音频控制命令 void FeedAudioBuffers(); // 填充音频缓冲区 int GetClientRTT(); // 获取客户端RTT(ms) void EvaluateQuality(); // 评估并调整质量 void ApplyQualityLevel(int level, bool persist = false); // 应用质量等级 const char* GetQualityName(int level); // 获取质量等级名称 void UpdateQualityMenuCheck(CMenu* SysMenu = nullptr); // 更新质量菜单勾选状态 void OnTimer(UINT_PTR nIDEvent); void UpdateWindowTitle(); bool Decode(LPBYTE Buffer, int size); void EnterFullScreen(); bool LeaveFullScreen(); void UpdateCtrlStatus(BOOL ctrl); void OnDropFiles(HDROP hDropInfo); afx_msg void OnHScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar); afx_msg void OnVScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar); afx_msg void OnLButtonDown(UINT nFlags, CPoint point); afx_msg void OnLButtonUp(UINT nFlags, CPoint point); afx_msg BOOL OnMouseWheel(UINT nFlags, short zDelta, CPoint pt); afx_msg void OnMouseMove(UINT nFlags, CPoint point); afx_msg void OnMouseLeave(); afx_msg void OnKillFocus(CWnd* pNewWnd); afx_msg void OnSize(UINT nType, int cx, int cy); afx_msg void OnActivate(UINT nState, CWnd* pWndOther, BOOL bMinimized); afx_msg LRESULT OnDisconnect(WPARAM wParam, LPARAM lParam); afx_msg LRESULT OnWaveOutDone(WPARAM wParam, LPARAM lParam); afx_msg LRESULT OnRecvFileV2Chunk(WPARAM wParam, LPARAM lParam); afx_msg LRESULT OnRecvFileV2Complete(WPARAM wParam, LPARAM lParam); afx_msg void OnExitFullscreen() { BYTE cmd[4] = { CMD_FULL_SCREEN, m_Settings.FullScreen = FALSE }; m_ContextObject->Send2Client(cmd, sizeof(cmd)); LeaveFullScreen(); } afx_msg void OnShowStatusInfo() { if (m_pStatusInfoWnd) m_pStatusInfoWnd->Show(); } afx_msg void OnHideStatusInfo() { if (m_pStatusInfoWnd) m_pStatusInfoWnd->Hide(); } protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 DECLARE_MESSAGE_MAP() public: void PrepareDrawing(const LPBITMAPINFO bmp); virtual BOOL OnInitDialog(); afx_msg void OnClose(); afx_msg void OnPaint(); BOOL OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message); afx_msg void OnSysCommand(UINT nID, LPARAM lParam); virtual BOOL PreTranslateMessage(MSG* pMsg); void OnLButtonDblClk(UINT nFlags, CPoint point); };