Marcus Meissner : mshtml: Check for NULL return from allocations.
Alexandre Julliard
julliard at winehq.org
Wed Dec 2 10:22:13 CST 2009
Module: wine
Branch: master
Commit: 5013c4d96e24456091d0d89faf39fd60f0de601a
URL: http://source.winehq.org/git/wine.git/?a=commit;h=5013c4d96e24456091d0d89faf39fd60f0de601a
Author: Marcus Meissner <marcus at jet.franken.de>
Date: Tue Dec 1 23:28:50 2009 +0100
mshtml: Check for NULL return from allocations.
---
dlls/mshtml/txtrange.c | 14 +++++++++-----
1 files changed, 9 insertions(+), 5 deletions(-)
diff --git a/dlls/mshtml/txtrange.c b/dlls/mshtml/txtrange.c
index c068c06..0ece6ec 100644
--- a/dlls/mshtml/txtrange.c
+++ b/dlls/mshtml/txtrange.c
@@ -177,12 +177,14 @@ static BOOL is_space_elem(nsIDOMNode *node)
return ret;
}
-static inline void wstrbuf_init(wstrbuf_t *buf)
+static inline BOOL wstrbuf_init(wstrbuf_t *buf)
{
buf->len = 0;
buf->size = 16;
buf->buf = heap_alloc(buf->size * sizeof(WCHAR));
+ if (!buf->buf) return FALSE;
*buf->buf = 0;
+ return TRUE;
}
static inline void wstrbuf_finish(wstrbuf_t *buf)
@@ -549,13 +551,14 @@ HRESULT get_node_text(HTMLDOMNode *node, BSTR *ret)
wstrbuf_t buf;
HRESULT hres = S_OK;
- wstrbuf_init(&buf);
+ if (!wstrbuf_init(&buf))
+ return E_OUTOFMEMORY;
wstrbuf_append_node_rec(&buf, node->nsnode);
if(buf.buf) {
*ret = SysAllocString(buf.buf);
if(!*ret)
hres = E_OUTOFMEMORY;
- }else {
+ } else {
*ret = NULL;
}
wstrbuf_finish(&buf);
@@ -1171,9 +1174,10 @@ static HRESULT WINAPI HTMLTxtRange_get_text(IHTMLTxtRange *iface, BSTR *p)
if(!This->nsrange)
return S_OK;
- wstrbuf_init(&buf);
+ if (!wstrbuf_init(&buf))
+ return E_OUTOFMEMORY;
range_to_string(This, &buf);
- if(buf.buf)
+ if (buf.buf)
*p = SysAllocString(buf.buf);
wstrbuf_finish(&buf);
More information about the wine-cvs
mailing list