ntdll: Handle RtlReAllocateHeap failure in com_class_add_progid (Coverity).

Signed-off-by: Thomas Faber <thomas.faber@reactos.org> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

ntdll: Handle RtlReAllocateHeap failure in com_class_add_progid (Coverity).
42f742e1 · Thomas Faber · Alexandre Julliard · cf629f7e · 42f742e1
Commit 42f742e1 authored Jan 16, 2018 by Thomas Faber Committed by Alexandre Julliard Jan 18, 2018
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

actctx.c dlls/ntdll/actctx.c +4 -1

No files found.
--- a/dlls/ntdll/actctx.c
+++ b/dlls/ntdll/actctx.c
@@ -1453,8 +1453,11 @@ static BOOL com_class_add_progid(const xmlstr_t *progid, struct entity *entity)

    if (progids->allocated == progids->num)
    {
+        WCHAR **new_progids = RtlReAllocateHeap(GetProcessHeap(), 0, progids->progids,
+                                                2 * progids->allocated * sizeof(WCHAR*));
+        if (!new_progids) return FALSE;
        progids->allocated *= 2;
-        progids->progids = RtlReAllocateHeap(GetProcessHeap(), 0, progids->progids, progids->allocated * sizeof(WCHAR*));
+        progids->progids = new_progids;
    }

    if (!(progids->progids[progids->num] = xmlstrdupW(progid))) return FALSE;