check return value from vasprintf

bestpika · Dec 19, 2011 · 2b18840 · 2b18840
1 parent 8c34597
commit 2b18840
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/xml_sax_parser.cc b/src/xml_sax_parser.cc
@@ -455,11 +455,10 @@ SaxParserCallback::warning(void* context,
 
   va_list args;
   va_start(args, msg);
-  vasprintf(&message, msg, args);
+  if (vasprintf(&message, msg, args) >= 0)
+      parser->warning(message);
   va_end(args);
 
-  parser->warning(message);
-
   free(message);
 }
 
@@ -473,11 +472,10 @@ SaxParserCallback::error(void* context,
 
   va_list args;
   va_start(args, msg);
-  vasprintf(&message, msg, args);
+  if (vasprintf(&message, msg, args) >= 0)
+      parser->error(message);
   va_end(args);
 
-  parser->error(message);
-
   free(message);
 }