SimpleRemoter/client/ScreenSpy.h

// ScreenSpy.h: interface for the CScreenSpy class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_SCREENSPY_H__5F74528D_9ABD_404E_84D2_06C96A0615F4__INCLUDED_)
#define AFX_SCREENSPY_H__5F74528D_9ABD_404E_84D2_06C96A0615F4__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#define COPY_ALL 1	// 拷贝全部屏幕，不分块拷贝（added by yuanyuanxiang 2019-1-7）
#include "CursorInfo.h"
#include "ScreenCapture.h"


class EnumHwndsPrintData
{
public:
    EnumHwndsPrintData()
    {
        memset(this, 0, sizeof(EnumHwndsPrintData));
    }
    void Create(HDC desktop, int _x, int _y, int w, int h)
    {
        x = _x;
        y = _y;
        width = GetSystemMetrics(SM_CXVIRTUALSCREEN);
        height = GetSystemMetrics(SM_CYVIRTUALSCREEN);
        hDcWindow = CreateCompatibleDC(desktop);
        hBmpWindow = CreateCompatibleBitmap(desktop, width, height);
    }
    EnumHwndsPrintData& SetScreenDC(HDC dc)
    {
        hDcScreen = dc;
        return *this;
    }
    ~EnumHwndsPrintData()
    {
        if (hDcWindow) DeleteDC(hDcWindow);
        hDcWindow = nullptr;
        if (hBmpWindow)DeleteObject(hBmpWindow);
        hBmpWindow = nullptr;
    }
    HDC GetWindowDC() const
    {
        return hDcWindow;
    }
    HDC GetScreenDC() const
    {
        return hDcScreen;
    }
    HBITMAP GetWindowBmp() const
    {
        return hBmpWindow;
    }
    int X() const
    {
        return x;
    }
    int Y() const
    {
        return y;
    }
    int Width()const
    {
        return width;
    }
    int Height()const
    {
        return height;
    }
private:
    HDC hDcWindow;
    HDC hDcScreen;
    HBITMAP hBmpWindow;
    int x;
    int y;
    int width;
    int height;
};

class CScreenSpy  : public ScreenCapture
{
protected:
    HDC              m_hDeskTopDC;          // 屏幕上下文
    HDC              m_hFullMemDC;          // 上一个上下文
    HDC              m_hDiffMemDC;          // 差异上下文
    HBITMAP	         m_BitmapHandle;        // 上一帧位图
    HBITMAP	         m_DiffBitmapHandle;    // 差异帧位图
    PVOID            m_BitmapData_Full;     // 当前位图数据
    PVOID            m_DiffBitmapData_Full; // 差异位图数据

    BOOL					m_bVirtualPaint;// 是否虚拟绘制
    EnumHwndsPrintData		m_data;

public:
    CScreenSpy(ULONG ulbiBitCount, BYTE algo, BOOL vDesk = FALSE, int gop = DEFAULT_GOP, BOOL all = FALSE);

    virtual ~CScreenSpy();

    int GetWidth()const
    {
        return m_ulFullWidth;
    }

    int GetHeight() const
    {
        return m_ulFullHeight;
    }

    bool IsZoomed() const
    {
        return m_bZoomed;
    }

    double GetWZoom() const
    {
        return m_wZoom;
    }

    double GetHZoom() const
    {
        return m_hZoom;
    }

    const LPBITMAPINFO& GetBIData() const
    {
        return m_BitmapInfor_Full;
    }

    ULONG GetFirstScreenLength() const
    {
        return m_BitmapInfor_Full->bmiHeader.biSizeImage;
    }

    static BOOL PaintWindow(HWND hWnd, EnumHwndsPrintData* data)
    {
        if (!IsWindowVisible(hWnd) || IsIconic(hWnd))
            return TRUE;

        RECT rect;
        if (!GetWindowRect(hWnd, &rect))
            return FALSE;

        HDC hDcWindow = data->GetWindowDC();
        HBITMAP hOldBmp = (HBITMAP)SelectObject(hDcWindow, data->GetWindowBmp());
        BOOL ret = FALSE;
        if (PrintWindow(hWnd, hDcWindow, PW_RENDERFULLCONTENT) || SendMessageTimeout(hWnd, WM_PRINT,
                (WPARAM)hDcWindow, PRF_CLIENT | PRF_NONCLIENT, SMTO_BLOCK, 50, NULL)) {
            BitBlt(data->GetScreenDC(), rect.left - data->X(), rect.top - data->Y(),
                   rect.right - rect.left, rect.bottom - rect.top, hDcWindow, 0, 0, SRCCOPY);

            ret = TRUE;
        }
        SelectObject(hDcWindow, hOldBmp);

        return ret;
    }

    static int EnumWindowsTopToDown(HWND owner, WNDENUMPROC proc, LPARAM param)
    {
        HWND currentWindow = GetTopWindow(owner);
        if (currentWindow == NULL)
            return -1;
        if ((currentWindow = GetWindow(currentWindow, GW_HWNDLAST)) == NULL)
            return -2;
        while (proc(currentWindow, param) && (currentWindow = GetWindow(currentWindow, GW_HWNDPREV)) != NULL);
        return 0;
    }

    static BOOL CALLBACK EnumHwndsPrint(HWND hWnd, LPARAM lParam)
    {
        AUTO_TICK_C(100);
        if (FALSE == PaintWindow(hWnd, (EnumHwndsPrintData*)lParam)) {
            char text[_MAX_PATH] = {};
            GetWindowText(hWnd, text, sizeof(text));
            Mprintf("PaintWindow %s[%p] failed!!!\n", text, hWnd);
        }

        static OSVERSIONINFOA versionInfo = { sizeof(OSVERSIONINFOA) };
        static BOOL result = GetVersionExA(&versionInfo);
        if (versionInfo.dwMajorVersion < 6) {
            DWORD style = GetWindowLongA(hWnd, GWL_EXSTYLE);
            SetWindowLongA(hWnd, GWL_EXSTYLE, style | WS_EX_COMPOSITED);
            EnumWindowsTopToDown(hWnd, EnumHwndsPrint, lParam);
        }

        return TRUE;
    }

    virtual LPBYTE GetFirstScreenData(ULONG* ulFirstScreenLength);

    virtual LPBYTE ScanNextScreen()
    {
        ScanScreen(m_hDiffMemDC, m_hDeskTopDC, m_ulFullWidth, m_ulFullHeight);
        return (LPBYTE)m_DiffBitmapData_Full;
    }

    VOID ScanScreen(HDC hdcDest, HDC hdcSour, ULONG ulWidth, ULONG ulHeight);
};

#endif // !defined(AFX_SCREENSPY_H__5F74528D_9ABD_404E_84D2_06C96A0615F4__INCLUDED_)