Merge pull request DreamAndDead#44 from ByteMansion/patch-1

memset cause core dump
gbrhomecom · Jul 1, 2019 · 3eeb67a · 3eeb67a
2 parents 4aa18f4 + e7ade10
commit 3eeb67a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chapter2/code/calloc.c b/chapter2/code/calloc.c
@@ -16,7 +16,9 @@ void* another_calloc(size_t nmemb, size_t size) {
   /* a good way to check overflow or not */
   if (nmemb == buf_size / size) {
     void* ptr = malloc(buf_size);
-    memset(ptr, 0, buf_size);
+    if(ptr != NULL) {
+      memset(ptr, 0, buf_size);
+    }
     return ptr;
   }
   return NULL;