fixed another security issue

jkriege2 · May 21, 2024 · 8072be8 · 8072be8
1 parent 3a8d9cc
commit 8072be8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/tinymatwriter.cpp b/src/tinymatwriter.cpp
@@ -328,8 +328,11 @@ const char *TinyMATWriter_getVersion()
        else if (newsize < 1000 * 1024 * 1024) newsize = newsize * 3 / 2;
        else newsize = newsize * 6 / 5;
      }
-     file->filedata = (uint8_t*)realloc(file->filedata, newsize);
-     file->filedata_size = newsize;
+     auto newMem = (uint8_t*)realloc(file->filedata, newsize);
+     if (newMem) {
+        file->filedata=newMem;
+        file->filedata_size = newsize;
+     }
    }
 #endif
  }