Add exception handling

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

diff --git a/xdgmime/src/xdgmime.c b/xdgmime/src/xdgmime.c
index 3ec0969..7f1dc9c 100644 (file)
--- a/xdgmime/src/xdgmime.c
+++ b/xdgmime/src/xdgmime.c
@@ -146,6 +146,9 @@ xdg_mime_init_from_directory (const char *directory)
   assert (directory != NULL);
 
   file_name = malloc (strlen (directory) + strlen ("/mime/mime.cache") + 1);
+  if (file_name == NULL)
+      return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/mime.cache");
   if (stat (file_name, &st) == 0)
     {
@@ -155,17 +158,24 @@ xdg_mime_init_from_directory (const char *directory)
        {
          xdg_dir_time_list_add (file_name, st.st_mtime);
 
-         _caches = realloc (_caches, sizeof (XdgMimeCache *) * (n_caches + 2));
-         _caches[n_caches] = cache;
-          _caches[n_caches + 1] = NULL;
-         n_caches++;
+         XdgMimeCache **tmp_caches = realloc (_caches, sizeof (XdgMimeCache *) * (n_caches + 2));
+          if (tmp_caches != NULL)
+            {
+              _caches = tmp_caches;
+              _caches[n_caches] = cache;
+              _caches[n_caches + 1] = NULL;
+              n_caches++;
 
-         return FALSE;
-       }
+              return FALSE;
+            }
+        }
     }
   free (file_name);
 
   file_name = malloc (strlen (directory) + strlen ("/mime/globs2") + 1);
+  if (file_name == NULL)
+    return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/globs2");
   if (stat (file_name, &st) == 0)
     {
@@ -176,6 +186,9 @@ xdg_mime_init_from_directory (const char *directory)
     {
       free (file_name);
       file_name = malloc (strlen (directory) + strlen ("/mime/globs") + 1);
+      if (file_name == NULL)
+        return TRUE;
+
       strcpy (file_name, directory); strcat (file_name, "/mime/globs");
       if (stat (file_name, &st) == 0)
         {
@@ -189,6 +202,9 @@ xdg_mime_init_from_directory (const char *directory)
     }
 
   file_name = malloc (strlen (directory) + strlen ("/mime/magic") + 1);
+  if (file_name == NULL)
+    return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/magic");
   if (stat (file_name, &st) == 0)
     {
@@ -201,21 +217,33 @@ xdg_mime_init_from_directory (const char *directory)
     }
 
   file_name = malloc (strlen (directory) + strlen ("/mime/aliases") + 1);
+  if (file_name == NULL)
+    return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/aliases");
   _xdg_mime_alias_read_from_file (alias_list, file_name);
   free (file_name);
 
   file_name = malloc (strlen (directory) + strlen ("/mime/subclasses") + 1);
+  if (file_name == NULL)
+    return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/subclasses");
   _xdg_mime_parent_read_from_file (parent_list, file_name);
   free (file_name);
 
   file_name = malloc (strlen (directory) + strlen ("/mime/icons") + 1);
+  if (file_name == NULL)
+    return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/icons");
   _xdg_mime_icon_read_from_file (icon_list, file_name);
   free (file_name);
 
   file_name = malloc (strlen (directory) + strlen ("/mime/generic-icons") + 1);
+  if (file_name == NULL)
+    return TRUE;
+
   strcpy (file_name, directory); strcat (file_name, "/mime/generic-icons");
   _xdg_mime_icon_read_from_file (generic_icon_list, file_name);
   free (file_name);