截图WinAPI的 - 可用内存
我已经问了一个非常类似的问题一两天以前,但我的问题是不是很清楚,所以我在这里试着重新制定它:截图WinAPI的 - 可用内存
我想要做我的屏幕截图要做到这一点我搜索,发现这个代码:
#include <windows.h>
#include <stdio.h>
inline int GetFilePointer(HANDLE FileHandle)
{
return SetFilePointer(FileHandle, 0, 0, FILE_CURRENT);
}
extern _Bool SaveBMPFile(char* filePath, HBITMAP bitmap, HDC bitmapDC, int width, int height)
{
_Bool Success = 0;
HDC SurfDC = NULL; // GDI-compatible device context for the surface
HBITMAP OffscrBmp = NULL; // bitmap that is converted to a DIB
HDC OffscrDC = NULL; // offscreen DC that we can select OffscrBmp into
LPBITMAPINFO lpbi = NULL; // bitmap format info; used by GetDIBits
LPVOID lpvBits = NULL; // pointer to bitmap bits array
HANDLE BmpFile = INVALID_HANDLE_VALUE; // destination .bmp file
BITMAPFILEHEADER bmfh; // .bmp file header
// We need an HBITMAP to convert it to a DIB:
if ((OffscrBmp = CreateCompatibleBitmap(bitmapDC, width, height)) == NULL)
return 0;
// The bitmap is empty, so let's copy the contents of the surface to it.
// For that we need to select it into a device context. We create one.
if ((OffscrDC = CreateCompatibleDC(bitmapDC)) == NULL)
return 0;
// Select OffscrBmp into OffscrDC:
HBITMAP OldBmp = (HBITMAP)SelectObject(OffscrDC, OffscrBmp);
// Now we can copy the contents of the surface to the offscreen bitmap:
BitBlt(OffscrDC, 0, 0, width, height, bitmapDC, 0, 0, SRCCOPY);
// GetDIBits requires format info about the bitmap. We can have GetDIBits
// fill a structure with that info if we pass a NULL pointer for lpvBits:
// Reserve memory for bitmap info (BITMAPINFOHEADER + largest possible
// palette):
if ((lpbi = (LPBITMAPINFO)malloc(sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD))) == NULL)
return 0;
ZeroMemory(&lpbi->bmiHeader, sizeof(BITMAPINFOHEADER));
lpbi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
// Get info but first de-select OffscrBmp because GetDIBits requires it:
SelectObject(OffscrDC, OldBmp);
if (!GetDIBits(OffscrDC, OffscrBmp, 0, height, NULL, lpbi, DIB_RGB_COLORS))
return 0;
// Reserve memory for bitmap bits:
if ((lpvBits = malloc(lpbi->bmiHeader.biSizeImage)) == NULL)
return 0;
// Have GetDIBits convert OffscrBmp to a DIB (device-independent bitmap):
if (!GetDIBits(OffscrDC, OffscrBmp, 0, height, lpvBits, lpbi, DIB_RGB_COLORS))
return 0;
//ANSI->Unicode
LPCSTR szAnsi = filePath;
int Size = MultiByteToWideChar(CP_ACP, 0, szAnsi, -1, NULL, 0);
LPWSTR filename = malloc(sizeof(LPWSTR) * Size);
MultiByteToWideChar(CP_ACP, 0, szAnsi, -1, filename, Size);
// Create a file to save the DIB to:
if ((BmpFile = CreateFile(filename,
GENERIC_WRITE,
0, NULL,
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL)) == INVALID_HANDLE_VALUE)
return 0;
DWORD Written; // number of bytes written by WriteFile
// Write a file header to the file:
bmfh.bfType = 19778; // 'BM'
// bmfh.bfSize = ??? // we'll write that later
bmfh.bfReserved1 = bmfh.bfReserved2 = 0;
// bmfh.bfOffBits = ??? // we'll write that later
if (!WriteFile(BmpFile, &bmfh, sizeof(bmfh), &Written, NULL))
return 0;
if (Written < sizeof(bmfh))
return 0;
// Write BITMAPINFOHEADER to the file:
if (!WriteFile(BmpFile, &lpbi->bmiHeader, sizeof(BITMAPINFOHEADER), &Written, NULL))
return 0;
if (Written < sizeof(BITMAPINFOHEADER))
return 0;
// Calculate size of palette:
int PalEntries;
// 16-bit or 32-bit bitmaps require bit masks:
if (lpbi->bmiHeader.biCompression == BI_BITFIELDS)
PalEntries = 3;
else
// bitmap is palettized?
PalEntries = (lpbi->bmiHeader.biBitCount <= 8) ?
// 2^biBitCount palette entries max.:
(int)(1 << lpbi->bmiHeader.biBitCount)
// bitmap is TrueColor -> no palette:
: 0;
// If biClrUsed use only biClrUsed palette entries:
if (lpbi->bmiHeader.biClrUsed)
PalEntries = lpbi->bmiHeader.biClrUsed;
// Write palette to the file:
if (PalEntries) {
if (!WriteFile(BmpFile, &lpbi->bmiColors, PalEntries * sizeof(RGBQUAD), &Written, NULL))
return 0;
if (Written < PalEntries * sizeof(RGBQUAD))
return 0;
}
// The current position in the file (at the beginning of the bitmap bits)
// will be saved to the BITMAPFILEHEADER:
bmfh.bfOffBits = GetFilePointer(BmpFile);
// Write bitmap bits to the file:
if (!WriteFile(BmpFile, lpvBits, lpbi->bmiHeader.biSizeImage, &Written, NULL))
return 0;
if (Written < lpbi->bmiHeader.biSizeImage)
return 0;
// The current pos. in the file is the final file size and will be saved:
bmfh.bfSize = GetFilePointer(BmpFile);
// We have all the info for the file header. Save the updated version:
SetFilePointer(BmpFile, 0, 0, FILE_BEGIN);
if (!WriteFile(BmpFile, &bmfh, sizeof(bmfh), &Written, NULL))
return 0;
if (Written < sizeof(bmfh))
return 0;
return 1;
}
_Bool ScreenCapture(char* filePath, int xStart, int yStart, int width, int height)
{
// get a DC compat. w/ the screen
HDC hDc = CreateCompatibleDC(0);
// make a bmp in memory to store the capture in
HBITMAP hBmp = CreateCompatibleBitmap(GetDC(0), width, height);
// join em up
SelectObject(hDc, hBmp);
// copy from the screen to my bitmap
BitBlt(hDc, 0, 0, width, height, GetDC(0), xStart, yStart, SRCCOPY);
// save my bitmap
_Bool ret = SaveBMPFile(filePath, hBmp, hDc, width, height);
// free the bitmap memory
DeleteObject(hBmp);
return ret;
}
main()
{
ScreenCapture("screenshot.bmp", 0, 0, 1920, 1080);
FILE* Screen = NULL;
Screen = fopen("screenshot.bmp", "r"); //Error, the image is "used" somewhere...
return 0;
}
我不知道WinAPI的一个字,但我在Viual工作室看到,在每次调用抓屏(),有非释放的RAM,这就是一个问题。
如果你们当中有些人知道哪来的错误...
赞赏帮助,谢谢:)
下面是最后的代码,没有问题的工作,没有内存泄漏的工作:
#include <windows.h>
#include <stdio.h>
// Helper function to retrieve current position of file pointer:
inline int GetFilePointer(HANDLE FileHandle)
{
return SetFilePointer(FileHandle, 0, 0, FILE_CURRENT);
}
//---------------------------------------------------------------------------
// Screenshot
// -> FileName: Name of file to save screenshot to
// -> lpDDS: DirectDraw surface to capture
// <- Result: Success
//
extern _Bool SaveBMPFile(char* filePath, HBITMAP bitmap, HDC bitmapDC, int width, int height)
{
_Bool Success = 0;
HBITMAP OffscrBmp = NULL; // bitmap that is converted to a DIB
HDC OffscrDC = NULL; // offscreen DC that we can select OffscrBmp into
LPBITMAPINFO lpbi = NULL; // bitmap format info; used by GetDIBits
LPVOID lpvBits = NULL; // pointer to bitmap bits array
HANDLE BmpFile = INVALID_HANDLE_VALUE; // destination .bmp file
BITMAPFILEHEADER bmfh; // .bmp file header
// We need an HBITMAP to convert it to a DIB:
if ((OffscrBmp = CreateCompatibleBitmap(bitmapDC, width, height)) == NULL)
return 0;
// The bitmap is empty, so let's copy the contents of the surface to it.
// For that we need to select it into a device context. We create one.
if ((OffscrDC = CreateCompatibleDC(bitmapDC)) == NULL)
return 0;
// Select OffscrBmp into OffscrDC:
HBITMAP OldBmp = (HBITMAP)SelectObject(OffscrDC, OffscrBmp);
// Now we can copy the contents of the surface to the offscreen bitmap:
BitBlt(OffscrDC, 0, 0, width, height, bitmapDC, 0, 0, SRCCOPY);
// GetDIBits requires format info about the bitmap. We can have GetDIBits
// fill a structure with that info if we pass a NULL pointer for lpvBits:
// Reserve memory for bitmap info (BITMAPINFOHEADER + largest possible
// palette):
if ((lpbi = (LPBITMAPINFO)malloc(sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD))) == NULL)
return 0;
ZeroMemory(&lpbi->bmiHeader, sizeof(BITMAPINFOHEADER));
lpbi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
// Get info but first de-select OffscrBmp because GetDIBits requires it:
SelectObject(OffscrDC, OldBmp);
if (!GetDIBits(OffscrDC, OffscrBmp, 0, height, NULL, lpbi, DIB_RGB_COLORS))
return 0;
// Reserve memory for bitmap bits:
if ((lpvBits = malloc(lpbi->bmiHeader.biSizeImage)) == NULL)
return 0;
// Have GetDIBits convert OffscrBmp to a DIB (device-independent bitmap):
if (!GetDIBits(OffscrDC, OffscrBmp, 0, height, lpvBits, lpbi, DIB_RGB_COLORS))
return 0;
//ANSI->Unicode
LPCSTR szAnsi = filePath;
int Size = MultiByteToWideChar(CP_ACP, 0, szAnsi, -1, NULL, 0);
LPWSTR filename = malloc(sizeof(LPWSTR) * Size);
MultiByteToWideChar(CP_ACP, 0, szAnsi, -1, filename, Size);
// Create a file to save the DIB to:
if ((BmpFile = CreateFile(filename,
GENERIC_WRITE,
0, NULL,
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL)) == INVALID_HANDLE_VALUE)
return 0;
DWORD Written; // number of bytes written by WriteFile
// Write a file header to the file:
bmfh.bfType = 19778; // 'BM'
// bmfh.bfSize = ??? // we'll write that later
bmfh.bfReserved1 = bmfh.bfReserved2 = 0;
// bmfh.bfOffBits = ??? // we'll write that later
if (!WriteFile(BmpFile, &bmfh, sizeof(bmfh), &Written, NULL))
return 0;
if (Written < sizeof(bmfh))
return 0;
// Write BITMAPINFOHEADER to the file:
if (!WriteFile(BmpFile, &lpbi->bmiHeader, sizeof(BITMAPINFOHEADER), &Written, NULL))
return 0;
if (Written < sizeof(BITMAPINFOHEADER))
return 0;
// Calculate size of palette:
int PalEntries;
// 16-bit or 32-bit bitmaps require bit masks:
if (lpbi->bmiHeader.biCompression == BI_BITFIELDS)
PalEntries = 3;
else
// bitmap is palettized?
PalEntries = (lpbi->bmiHeader.biBitCount <= 8) ?
// 2^biBitCount palette entries max.:
(int)(1 << lpbi->bmiHeader.biBitCount)
// bitmap is TrueColor -> no palette:
: 0;
// If biClrUsed use only biClrUsed palette entries:
if (lpbi->bmiHeader.biClrUsed)
PalEntries = lpbi->bmiHeader.biClrUsed;
// Write palette to the file:
if (PalEntries) {
if (!WriteFile(BmpFile, &lpbi->bmiColors, PalEntries * sizeof(RGBQUAD), &Written, NULL))
return 0;
if (Written < PalEntries * sizeof(RGBQUAD))
return 0;
}
// The current position in the file (at the beginning of the bitmap bits)
// will be saved to the BITMAPFILEHEADER:
bmfh.bfOffBits = GetFilePointer(BmpFile);
// Write bitmap bits to the file:
if (!WriteFile(BmpFile, lpvBits, lpbi->bmiHeader.biSizeImage, &Written, NULL))
return 0;
if (Written < lpbi->bmiHeader.biSizeImage)
return 0;
// The current pos. in the file is the final file size and will be saved:
bmfh.bfSize = GetFilePointer(BmpFile);
// We have all the info for the file header. Save the updated version:
SetFilePointer(BmpFile, 0, 0, FILE_BEGIN);
if (!WriteFile(BmpFile, &bmfh, sizeof(bmfh), &Written, NULL))
return 0;
if (Written < sizeof(bmfh))
return 0;
DeleteObject(OffscrBmp);
DeleteObject(OldBmp);
DeleteDC(OffscrDC);
CloseHandle(BmpFile);
free(lpbi);
free(lpvBits);
free(filename);
return 1;
}
_Bool ScreenCapture(char* filePath, int xStart, int yStart, int width, int height)
{
// get a DC compat. w/ the screen
HDC hDc = CreateCompatibleDC(0);
// make a bmp in memory to store the capture in
HBITMAP hBmp = CreateCompatibleBitmap(GetDC(0), width, height);
// join em up
SelectObject(hDc, hBmp);
// copy from the screen to my bitmap
BitBlt(hDc, 0, 0, width, height, GetDC(0), xStart, yStart, SRCCOPY);
// save my bitmap
_Bool ret = SaveBMPFile(filePath, hBmp, hDc, width, height);
// free the bitmap memory
DeleteObject(hBmp);
DeleteDC(hDc);
return ret;
}
main()
{
ScreenCapture("screenshot.png", 0, 0, 1920, 1080);
return 0;
}
无论如何你的代码会被误认为是错误的,并且你将在任何错误上失败免费资源 – RbMm
请在文本中描述错误是什么。 –
每次调用ScreenCapture时都有内存泄漏,不再有内存泄漏。也许还有一些错误,但我无法修复代码...不知道WinAPI –
HBITMAP hBmp = CreateCompatibleBitmap(GetDC(0), width, height);
必须使用ReleasedDC
清理。当你用这种方式写GetDC(0)
时,就不可能清理。您的程序只有10,000个GDI句柄可用,所以此资源泄漏可能是一个严重问题。
你应该这样做,而不是:
HDC hdc_desktop = GetDC(HWND_DESKTOP);
HBITMAP hBmp = CreateCompatibleBitmap(hdc_desktop, width, height);
...
ReleaseDC(HWND_DESKTOP, hdc_desktop);
HWND_DESKTOP
只是0
,我用它的清晰度。 GetDC
返回一个句柄hdc_desktop
,最后可以清理它。
DeleteObject(hBmp)
是一个小错误,因为hBmp
目前在hDc
中被选中,无法删除。很多程序员犯这个错误,所以更新的Windows版本(我认为自XP以来)已经期待这个,所以Windows修复了这个错误(至少在这种情况下),但是您应该真正保存旧位图的句柄,恢复旧的位图,然后删除新的位图。
还有一个小错误GetDIBits
。文档说,调用该函数时,hbitmap不能选择为直流。但是Windows又习惯看到这个错误并修复它。可以简化你的SaveBMPFile
。您不保存调色板位图,您可以完全忽略调色板。
Unreleated问题:您正在使用Unicode的功能,让您UTF-16字符串文字:中L"screenshot.bmp"
代替"screenshot.bmp"
extern _Bool SaveBMPFile(const wchar_t* filePath,
HDC memdc, HBITMAP hbitmap, int width, int height)
{
_Bool success = 0;
WORD bpp = 24; //or 32 for 32-bit bitmap
DWORD size = ((width * bpp + 31)/32) * 4 * height;
BITMAPFILEHEADER filehdr = { 0 };
filehdr.bfType = 19778;
filehdr.bfSize = 54 + size;
filehdr.bfOffBits = 54;
//54 = 14 + 40, sizeof BITMAPFILEHEADER & BITMAPINFOHEADER
BITMAPINFOHEADER infohdr = { sizeof(infohdr) };
infohdr.biWidth = width;
infohdr.biHeight = height;
infohdr.biPlanes = 1;
infohdr.biBitCount = bpp;
BYTE *bits = malloc(size);
GetDIBits(memdc, hbitmap, 0, height, bits, (BITMAPINFO*)&infohdr, DIB_RGB_COLORS);
HANDLE hfile = CreateFileW(filePath,
GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if(hfile != INVALID_HANDLE_VALUE)
{
DWORD temp;
WriteFile(hfile, &filehdr, 14, &temp, NULL);
WriteFile(hfile, &infohdr, 40, &temp, NULL);
WriteFile(hfile, bits, size, &temp, NULL);
CloseHandle(hfile);
success = 1;
}
free(bits);
return success;
}
_Bool ScreenCapture(const wchar_t* filePath, int x, int y, int width, int height)
{
HDC hdc = GetDC(HWND_DESKTOP);
HDC memdc = CreateCompatibleDC(hdc);
HBITMAP hbitmap = CreateCompatibleBitmap(hdc, width, height);
HBITMAP oldbitmap = SelectObject(memdc, hbitmap);
BitBlt(memdc, 0, 0, width, height, hdc, x, y, SRCCOPY);
SelectObject(memdc, oldbitmap);
_Bool ret = SaveBMPFile(filePath, memdc, hbitmap, width, height);
DeleteObject(hbitmap);
DeleteDC(memdc);
ReleaseDC(HWND_DESKTOP, hdc);
return ret;
}
int main(void)
{
ScreenCapture(L"screenshot.bmp", 0, 0,
GetSystemMetrics(SM_CXFULLSCREEN), GetSystemMetrics(SM_CYFULLSCREEN));
return 0;
}
您可以在一个循环中运行这个,看GDI处理在任务管理器,以确保有没有泄漏。
您不关闭'BmpFile'。在从'SaveBMPFile'返回之前调用'CloseHandle(BmpFile);' –
是的,所有*你创建的对象需要被释放/销毁。每个创建功能都有相应的功能。你会发现它在创建函数的MSDN文档中列出。 –
谢谢!这样做,没有更多的问题:) –