xmllite/writer: Handle possible memory allocation failures.

Signed-off-by: David Kahurani <k.kahurani@gmail.com>

xmllite/writer: Handle possible memory allocation failures.
098f3c89 · David Kahurani · Alexandre Julliard · 630c38b2 · 098f3c89
Commit 098f3c89 authored Jun 20, 2022 by David Kahurani Committed by Alexandre Julliard Jun 20, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 1 deletion

writer.c dlls/xmllite/writer.c +9 -1

No files found.
--- a/dlls/xmllite/writer.c
+++ b/dlls/xmllite/writer.c
@@ -180,6 +180,13 @@ static struct element *alloc_element(xmlwriter *writer, const WCHAR *prefix, con
    len += lstrlenW(local);

    ret->qname = writer_alloc(writer, (len + 1)*sizeof(WCHAR));
+
+    if (!ret->qname)
+    {
+        writer_free(writer, ret);
+        return NULL;
+    }
+
    ret->len = len;
    if (prefix)
    {
@@ -249,7 +256,8 @@ static WCHAR *writer_strndupW(const xmlwriter *writer, const WCHAR *str, int len

    size = (len + 1) * sizeof(WCHAR);
    ret = writer_alloc(writer, size);
-    memcpy(ret, str, size);
+    if (ret) memcpy(ret, str, size);
+
    return ret;
 }