[PATCH v2 04/10] quartz/vmr7: Reject unsupported FOURCC formats.

[Henri Verbeet]

From: Henri Verbeet

---
 dlls/quartz/vmr7_presenter.c | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/dlls/quartz/vmr7_presenter.c b/dlls/quartz/vmr7_presenter.c
index 1abee7b957d..29e9f31c49a 100644
--- a/dlls/quartz/vmr7_presenter.c
+++ b/dlls/quartz/vmr7_presenter.c
@@ -171,6 +171,37 @@ static ULONG WINAPI surface_allocator_Release(IVMRSurfaceAllocator *iface)
     return IVMRImagePresenter_Release(&presenter->IVMRImagePresenter_iface);
 }

+static BOOL fourcc_is_supported(IDirectDraw7 *ddraw, DWORD fourcc)
+{
+    DWORD *codes, count, i;
+    HRESULT hr;
+
+    if (FAILED(hr = IDirectDraw7_GetFourCCCodes(ddraw, &count, NULL)))
+    {
+        ERR("Failed to get FOURCC code count, hr %#lx.\n", hr);
+        return FALSE;
+    }
+
+    if (!count || !(codes = calloc(count, sizeof(*codes))))
+        return FALSE;
+
+    if (FAILED(hr = IDirectDraw7_GetFourCCCodes(ddraw, &count, codes)))
+    {
+        ERR("Failed to get FOURCC codes, hr %#lx.\n", hr);
+        free(codes);
+        return FALSE;
+    }
+
+    for (i = 0; i < count; ++i)
+    {
+        if (codes[i] == fourcc)
+            break;
+    }
+    free(codes);
+
+    return i < count;
+}
+
 static HRESULT WINAPI surface_allocator_AllocateSurface(IVMRSurfaceAllocator *iface,
         DWORD_PTR id, VMRALLOCATIONINFO *info, DWORD *count, IDirectDrawSurface7 **surface)
 {
@@ -224,6 +255,9 @@ static HRESULT WINAPI surface_allocator_AllocateSurface(IVMRSurfaceAllocator *if
     }
     else
     {
+        if (!fourcc_is_supported(presenter->ddraw, info->lpHdr->biCompression))
+            return VFW_E_DDRAW_CAPS_NOT_SUITABLE;
+
         surface_desc.ddpfPixelFormat.dwFlags = DDPF_FOURCC;
         surface_desc.ddpfPixelFormat.dwFourCC = info->lpHdr->biCompression;
     }