用MASM32编程截取屏幕

;<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
; 功能: 把屏幕内容拷贝到剪贴板并显示在程序窗口上
;作者: Purple Endurer
;环境: Windows 2000 PRO + MASM32 V8
;
; 日志
; ---------------------
; 2005.12.24 创建!
;<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
.386
.model flat, stdcall
option casemap:none
include /masm32/include/windows.inc
include /masm32/include/gdi32.inc
include /masm32/include/kernel32.inc
include /masm32/include/user32.inc

includelib /masm32/lib/gdi32.lib
includelib /masm32/lib/kernel32.lib
includelib /masm32/lib/user32.lib

DlgProc PROTO :DWORD, :DWORD, :DWORD, :DWORD
CopyScreenToBitmap PROTO :DWORD
CopyBMPtoClipboard PROTO :HANDLE, :HWND
DrawBMP PROTO :HWND, :HANDLE

m_m2m MACRO d1, d2
push d2
pop d1
ENDM

IDD_dlgCopyScr equ 8000
IDC_btnCopyScr equ 1008

.data?
g_hInstance HINSTANCE ?
g_ScrRect RECT <>

.code
start:
invoke GetModuleHandle, NULL
mov g_hInstance, eax

;m_m2m g_ScrRect.top, 0
;m_m2m g_ScrRect.left, 0
;m_m2m g_ScrRect.bottom, 200
;m_m2m g_ScrRect.right, 100

invoke DialogBoxParam, g_hInstance, IDD_dlgCopyScr, NULL, OFFSET DlgProc, NULL
invoke ExitProcess, eax

DlgProc proc hWnd: HWND, uMsg: UINT, wParam: WPARAM, lParam: LPARAM
.IF uMsg == WM_CLOSE
invoke EndDialog, hWnd, NULL

;.ELSEIF uMsg == WM_INITDIALOG

.ELSEIF (uMsg == WM_COMMAND)
mov eax, wParam
mov edx, eax
shr edx, 16
.if (dx == BN_CLICKED) ;If button is clicked
.IF (eax == IDC_btnCopyScr)
invoke GetDesktopWindow
invoke GetWindowRect, eax, ADDR g_ScrRect
invoke CopyScreenToBitmap, ADDR g_ScrRect
push eax
invoke CopyBMPtoClipboard, eax, hWnd
pop eax
invoke DrawBMP, hWnd, eax
.ENDIF
.endif

.ELSE
mov eax, FALSE
ret
.ENDIF

mov eax, TRUE
ret
DlgProc endp

; lpRect 为指向要拷贝的屏幕区域的坐标结构的指针
CopyScreenToBitmap proc lpRect: DWORD
LOCAL hScrDC, hMemDC: HDC ;屏幕和内存设备描述表
LOCAL hBitmap, hOldBitmap: HBITMAP ;位图句柄
LOCAL nX, nY, nX2, nY2: DWORD ;选定区域坐标
LOCAL nWidth, nHeight: DWORD ;位图宽度和高度
LOCAL xScrn, yScrn: DWORD ;屏幕分辨率

;判断选定区域是否为空矩形
invoke IsRectEmpty, lpRect
or eax, eax
jz @F
;m_m2m eax, 1 ;是空矩形
ret
@@:
;为屏幕创建设备描述表
xor eax, eax
jz @F
szDisplay db "DISPLAY", 0
@@:
invoke CreateDC, ADDR szDisplay, NULL, NULL, NULL
cmp eax, NULL
jne @F
;m_m2m eax, 2 ;不能创建DC
ret
@@:
mov hScrDC, eax

;为屏幕设备描述表创建兼容的内存设备描述表
invoke CreateCompatibleDC, hScrDC
cmp eax, NULL
jne @F
;m_m2m eax, 3 ;不能创建兼容的内存设备描述表
ret
@@:
mov hMemDC, eax

;获得选定区域坐标
mov eax, lpRect
assume eax: ptr RECT
m_m2m nX, [eax].left
m_m2m nY, [eax].top
m_m2m nX2, [eax].right
m_m2m nY2, [eax].bottom
assume eax: nothing

;获得屏幕分辨率
invoke GetDeviceCaps, hScrDC, HORZRES
mov xScrn, eax
invoke GetDeviceCaps, hScrDC, VERTRES
mov yScrn, eax

;确保选定区域是可见的
.if (nX < 0)
m_m2m nX, 0
.endif

.if (nY < 0)
m_m2m nY, 0
.endif

mov eax, nX2
.if (eax > xScrn)
m_m2m nX2, xScrn
.endif

mov eax, nY2
.if (eax > yScrn)
m_m2m nY2, yScrn
.endif

mov eax, nX2
sub eax, nX
mov nWidth, eax

mov eax, nY2
sub eax, nY
mov nHeight, eax

;创建一个与屏幕设备描述表兼容的位图
invoke CreateCompatibleBitmap, hScrDC, nWidth, nHeight
cmp eax, NULL
jne @F
;m_m2m eax, 4 ;不能创建屏幕设备描述表兼容的位图
ret
@@:
mov hBitmap, eax

;把新位图选到内存设备描述表中
invoke SelectObject, hMemDC, hBitmap
cmp eax, NULL
jne @F
;m_m2m eax, 5 ;an error occurs and the selected object is not a region
ret
@@:
cmp eax, GDI_ERROR
jne @F
;m_m2m eax, 6
ret
@@:
mov hOldBitmap, eax

;把屏幕设备描述表拷贝到内存设备描述表中
invoke BitBlt, hMemDC, 0, 0, nWidth, nHeight, hScrDC, nX, nY, SRCCOPY

;得到屏幕位图的句柄
invoke SelectObject, hMemDC, hOldBitmap
push eax ;mov hBitmap, eax
;清除
invoke DeleteDC, hScrDC
invoke DeleteDC, hMemDC

;返回位图句柄
pop eax ;mov eax, hBitmap

ret
CopyScreenToBitmap endp

;得到屏幕位图句柄以后,我们可以把屏幕内容粘贴到剪贴板上
;hBitmap: 位图句柄
;hWnd: 程序窗口句柄
CopyBMPtoClipboard proc hBitmap: HANDLE, hWnd: HWND
invoke OpenClipboard, hWnd ;
or eax, eax
jnz @F
m_m2m eax, 1
ret
@@:
;清空剪贴板
invoke EmptyClipboard
or eax, eax
jnz @F
m_m2m eax, 2
ret
@@:
;把位图内容粘贴到剪贴板上
invoke SetClipboardData, CF_BITMAP, hBitmap
or eax, eax
jnz @F
m_m2m eax, 3
ret
@@:
;关闭剪贴板
invoke CloseClipboard
or eax, eax
jnz @F
m_m2m eax, 4
ret
@@:
xor eax, eax
ret
CopyBMPtoClipboard endp

;在指定窗口上绘制位图
DrawBMP proc hWnd: HWND, hBmp: HANDLE
LOCAL hDC: HDC
LOCAL hOld: DWORD
LOCAL memDC :DWORD

invoke GetDC, hWnd
mov hDC, eax
invoke CreateCompatibleDC, eax
mov memDC, eax

invoke SelectObject, memDC, hBmp
mov hOld, eax

invoke BitBlt, hDC, 120, 0, 1000, 1000, memDC, 0, 0, SRCCOPY

invoke SelectObject, hDC, hOld
invoke DeleteDC, memDC
invoke DeleteDC, hDC
;xor eax, eax
ret
DrawBMP endp

end start

;<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
;资源文件rsrc.rc
;<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
#include "/masm32/include/resource.h"

#define IDD_dlgcopyScr 8000
#define IDC_btnCopyScr 1008

IDD_dlgcopyScr DIALOG 20, 20, 295, 165
STYLE 0x0004 | DS_CENTER | WS_CAPTION | DS_SYSMODAL | WS_SYSMENU | WS_MINIMIZEBOX
| WS_VISIBLE | WS_OVERLAPPED | DS_MODALFRAME | DS_3DLOOK
CAPTION "拷贝屏幕"
{
PUSHBUTTON "&C 拷贝屏幕", IDC_btnCopyScr, 5, 5, 52, 52, WS_BORDER
}