使用CopyPixels从IWICBitmap提取RGB矩阵时程序崩溃的问题排查

你的代码存在几个关键错误，直接导致了崩溃：

• 错误使用空位图：你创建了一个全新的空IWICBitmap（ipBitMap），但实际的截图数据是函数传入的pWICBitmapSource，你应该从这个有效数据源提取像素，而非空位图。
• 内存分配与Stride参数完全错误：
  • GUID_WICPixelFormat32bppRGBA表示每个像素占4字节（R、G、B、A各1字节），总内存应为宽度×高度×4，你错误写成了×4×3，冗余分配了3倍内存。
  • Stride（每行像素的字节数）应为宽度×4，你写的1920*(4*3)远大于实际所需，导致CopyPixels写入超出缓冲区范围，触发内存越界崩溃。
• 不安全的指针访问：直接调用ipBitMap.p->访问CComPtr的内部指针不符合规范，应直接使用ipBitMap->。

正确的RGB数组提取实现 #

直接基于传入的pWICBitmapSource提取像素，步骤如下：

1. 获取位图实际宽高（避免硬编码分辨率）；
2. 格式转换（将Desktop Duplication返回的常见格式转为32bppRGBA）；
3. 计算正确的内存大小与Stride；
4. 提取像素数据。

替换你那段代码的正确实现：

// START of corrected code
unsigned int uiWidth = 0;
unsigned int uiHeight = 0;
// 获取截图的实际宽高
ipWICBitmapSource->GetSize(&uiWidth, &uiHeight);

// 格式转换器：将源位图转换为32bppRGBA格式
CComPtr<IWICFormatConverter> pFormatConverter;
ipWICImagingFactory->CreateFormatConverter(&pFormatConverter);
HRESULT convertHr = pFormatConverter->Initialize(
    ipWICBitmapSource,
    GUID_WICPixelFormat32bppRGBA,
    WICBitmapDitherTypeNone,
    nullptr,
    0.0f,
    WICBitmapPaletteTypeCustom
);

if (SUCCEEDED(convertHr))
{
    // 计算参数：32bppRGBA每个像素4字节
    const UINT stride = uiWidth * 4;
    const UINT totalBytes = stride * uiHeight;
    BYTE* screenBuffer = (BYTE*)malloc(totalBytes);
    
    if (screenBuffer)
    {
        // 提取整个位图的像素数据
        HRESULT copyHr = pFormatConverter->CopyPixels(nullptr, stride, totalBytes, screenBuffer);
        if (SUCCEEDED(copyHr))
        {
            // 示例：读取第一个像素的R、G、B值（RGBA顺序）
            printf("%d-%d-%d\n", screenBuffer[0], screenBuffer[1], screenBuffer[2]);
            // 此处可将screenBuffer作为RGB矩阵用于后续图像处理
        }
        else
        {
            printf("CopyPixels failed, HRESULT: 0x%X\n", copyHr);
        }
        // 用完后释放内存
        free(screenBuffer);
    }
}
else
{
    printf("Format conversion failed, HRESULT: 0x%X\n", convertHr);
}
// END of corrected code

额外优化提示 #

• 如果源位图格式是GUID_WICPixelFormat32bppBGRA（Desktop Duplication的常见输出格式），可以跳过格式转换，直接提取后手动交换R和B通道，减少性能开销；
• 始终通过IWICBitmapSource->GetSize()获取分辨率，适配不同屏幕；
• 确保使用完缓冲区后调用free()释放内存，避免泄漏。

内容的提问来源于stack exchange，提问作者StefanoN