mshtml: Handle memory allocation failure when creating IOleUndoManager.

Signed-off-by: Dmitry Timoshkov <dmitry@baikal.ru> Signed-off-by: Jacek Caban <jacek@codeweavers.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

mshtml: Handle memory allocation failure when creating IOleUndoManager.
e8b9bda5 · Dmitry Timoshkov · Alexandre Julliard · 49bdb815 · e8b9bda5
Commit e8b9bda5 authored Oct 12, 2015 by Dmitry Timoshkov Committed by Alexandre Julliard Oct 12, 2015
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

service.c dlls/mshtml/service.c +5 -0

No files found.
--- a/dlls/mshtml/service.c
+++ b/dlls/mshtml/service.c
@@ -197,6 +197,8 @@ static IOleUndoManager *create_undomgr(void)
 {
    UndoManager *ret = heap_alloc(sizeof(UndoManager));

+    if (!ret) return NULL;
+
    ret->IOleUndoManager_iface.lpVtbl = &OleUndoManagerVtbl;
    ret->ref = 1;

@@ -246,6 +248,9 @@ static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface, REFG
        if(!This->doc_obj->undomgr)
            This->doc_obj->undomgr = create_undomgr();

+        if (!This->doc_obj->undomgr)
+            return E_OUTOFMEMORY;
+
        return IOleUndoManager_QueryInterface(This->doc_obj->undomgr, riid, ppv);
    }