vkd3d: Check extension_count for NULL before dereferencing it.

Signed-off-by: Krzysztof Bogacki <[email protected]>
Heroic-Games-Launcher · Jun 17, 2024 · 7e99b34 · 7e99b34
1 parent de32a38
commit 7e99b34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/vkd3d/device_vkd3d_ext.c b/libs/vkd3d/device_vkd3d_ext.c
@@ -326,7 +326,7 @@ static HRESULT STDMETHODCALLTYPE d3d12_dxvk_interop_device_GetInstanceExtensions
 
     TRACE("iface %p, extension_count %p, extensions %p.\n", iface, extension_count, extensions);
 
-    if (extensions && (*extension_count < instance->vk_info.extension_count))
+    if (!extension_count || (extensions && (*extension_count < instance->vk_info.extension_count)))
         return E_INVALIDARG;
 
     *extension_count = instance->vk_info.extension_count;
@@ -345,7 +345,7 @@ static HRESULT STDMETHODCALLTYPE d3d12_dxvk_interop_device_GetDeviceExtensions(I
 
     TRACE("iface %p, extension_count %p, extensions %p.\n", iface, extension_count, extensions);
 
-    if (extensions && (*extension_count < device->vk_info.extension_count))
+    if (!extension_count || (extensions && (*extension_count < device->vk_info.extension_count)))
         return E_INVALIDARG;
 
     *extension_count = device->vk_info.extension_count;