Check that malloc succeeded when building cache

haarg-contrib · Jan 15, 2017 · 55b6e29 · 55b6e29
1 parent 1b43c8c
commit 55b6e29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/cache.xs b/cache.xs
@@ -55,6 +55,9 @@ CODE:
         size = CACHE_DEFAULT_SIZE;
     }
     RETVAL = cache_build(aTHX_ size);
+    if (!RETVAL) {
+        croak("could not create cache");
+    }
 }
 OUTPUT: RETVAL
 

diff --git a/lru.c b/lru.c
@@ -7,9 +7,10 @@ Cache* cache_build(pTHX_ int size)
 {
     Cache* cache;
     GMEM_NEW(cache, Cache*, sizeof(Cache));
-    cache->size = size;
-    cache->data = 0;
-    /* fprintf(stderr, "LOG building cache for %d elements\n", cache->size); */
+    if (cache) {
+        cache->size = size;
+        cache->data = 0;
+    }
     return cache;
 }