efl_ui_win: add error hanling code for malloc fail 04/142004/1
authorJiyoun Park <jy0703.park@samsung.com>
Wed, 2 Aug 2017 07:48:27 +0000 (16:48 +0900)
committerJiyoun Park <jy0703.park@samsung.com>
Wed, 2 Aug 2017 07:48:27 +0000 (16:48 +0900)
Change-Id: I57db76192233afcc95df22bef8eff824611228d7

src/lib/elm_win.c

index 6302ab117c87777b47544afef2ff527300871620..8e8134314b47dfc789044ac6c19efbfa9b004175 100644 (file)
@@ -664,11 +664,14 @@ _shot_file_get(Elm_Win_Data *sd)
                     {
                        size_t size = sizeof(char) * (strlen(tmp) + 16);
                        repname = malloc(size);
-                       strncpy(repname, tmp, dotptr - tmp);
-                       snprintf(repname + (dotptr - tmp), size -
-                                (dotptr - tmp), "%03i",
-                                sd->shot.shot_counter + 1);
-                       strcat(repname, dotptr);
+                       if (repname)
+                         {
+                            strncpy(repname, tmp, dotptr - tmp);
+                            snprintf(repname + (dotptr - tmp), size -
+                                     (dotptr - tmp), "%03i",
+                                     sd->shot.shot_counter + 1);
+                            strcat(repname, dotptr);
+                         }
                        free(tmp);
                        return repname;
                     }
@@ -679,6 +682,7 @@ _shot_file_get(Elm_Win_Data *sd)
    if (!sd->shot.repeat_count) return strdup("out.png");
 
    repname = malloc(sizeof(char) * 24);
+   if (!repname) return NULL;
    snprintf(repname, sizeof(char) * 24, "out%03i.png",
             sd->shot.shot_counter + 1);