Add exception handling

[platform/core/appfw/xdgmime.git] / xdgmime / src / xdgmimemagic.c

diff --git a/xdgmime/src/xdgmimemagic.c b/xdgmime/src/xdgmimemagic.c
index a2320f5..fbc03af 100644 (file)
--- a/xdgmime/src/xdgmimemagic.c
+++ b/xdgmime/src/xdgmimemagic.c
@@ -169,6 +169,9 @@ _xdg_mime_magic_read_to_newline (FILE *magic_file,
   len = 128;
   pos = 0;
   retval = malloc (len);
+  if (retval == NULL)
+    return NULL;
+
   *end_of_file = FALSE;
 
   while (TRUE)
@@ -185,7 +188,13 @@ _xdg_mime_magic_read_to_newline (FILE *magic_file,
       if (pos % 128 == 127)
        {
          len = len + 128;
-         retval = realloc (retval, len);
+         unsigned char *tmp = realloc (retval, len);
+          if (tmp == NULL)
+           {
+              free(retval);
+              return NULL;
+            }
+         retval = tmp;
        }
     }