added malloc/realloc call checks

proh14 · Apr 21, 2024 · 42370e1 · 42370e1
1 parent 8b14fa5
commit 42370e1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/date/date.c b/src/date/date.c
@@ -109,9 +109,19 @@ int main(int argc, char **argv) {
 
   size_t cap = INITIAL_BUF_SIZE;
   char *buf = malloc(cap);
+  if (!buf) {
+    fprintf(stderr, "date: malloc failed: %s\n", strerror(errno));
+    return EXIT_FAILURE;
+  }
   while (!strftime(buf, cap, &operand[1], tm)) {
     cap *= 2;
-    buf = realloc(buf, cap);
+    char *new_buf = realloc(buf, cap);
+    if (!new_buf) {
+      free(buf);
+      fprintf(stderr, "date: realloc failed: %s\n", strerror(errno));
+      return EXIT_FAILURE;
+    }
+    buf = new_buf;
   }
   printf("%s\n", buf);
   free(buf);