[RTL] Handle RtlReAllocateHeap failure in com_class_add_progid. CID 1223161

diff --git a/sdk/lib/rtl/actctx.c b/sdk/lib/rtl/actctx.c
index 4cf9bba..6ce1734 100644 (file)
--- a/sdk/lib/rtl/actctx.c
+++ b/sdk/lib/rtl/actctx.c
@@ -1504,8 +1504,11 @@ static BOOL com_class_add_progid(const xmlstr_t *progid, struct entity *entity)
 
     if (progids->allocated == progids->num)
     {
 
     if (progids->allocated == progids->num)
     {

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

         progids->allocated *= 2;
         progids->allocated *= 2;

-        progids->progids = RtlReAllocateHeap(RtlGetProcessHeap(), 0, progids->progids, progids->allocated * sizeof(WCHAR*));
+        progids->progids = new_progids;

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