fallin demo: no write to failed alloc

dankamongmen · Jun 24, 2021 · ce073d2 · ce073d2
1 parent 4064ede
commit ce073d2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/demo/fallin.c b/src/demo/fallin.c
@@ -97,6 +97,9 @@ int fallin_demo(struct notcurses* nc){
   struct ncplane* stdn = notcurses_stddim_yx(nc, &dimy, &dimx);
   size_t usesize = sizeof(bool) * dimy * dimx;
   bool* usemap = malloc(usesize);
+  if(usemap == NULL){
+    return -1;
+  }
   memset(usemap, 0, usesize);
   // brick size is relative to the screen size
   const int maxx = dimx > 39 ? dimx / 10 : 2;