fix many crashes when out of memory

1 files changed, 37 insertions, 1 deletions

diff --git a/src/save.c b/src/save.c
index 8f4b83b..499a8cc 100644
--- a/src/save.c
+++ b/src/save.c
@@ -494,6 +494,12 @@ void *build_save_OPS(int *size, int orig_x0, int orig_y0, int orig_w, int orig_h
 	wallDataLen = blockW*blockH;
 	fanData = malloc((blockW*blockH)*2);
 	fanDataLen = 0;
+	if (!wallData || !fanData)
+	{
+		puts("Save Error, out of memory\n");
+		outputData = NULL;
+		goto fin;
+	}
 	for(x = blockX; x < blockX+blockW; x++)
 	{
 		for(y = blockY; y < blockY+blockH; y++)
@@ -534,6 +540,12 @@ void *build_save_OPS(int *size, int orig_x0, int orig_y0, int orig_w, int orig_h
 	partsPosLastMap = calloc(fullW*fullH, sizeof(unsigned));
 	partsPosCount = calloc(fullW*fullH, sizeof(unsigned));
 	partsPosLink = calloc(NPART, sizeof(unsigned));
+	if (!partsPosFirstMap || !partsPosLastMap || !partsPosCount || !partsPosLink)
+	{
+		puts("Save Error, out of memory\n");
+		outputData = NULL;
+		goto fin;
+	}
 	for(i = 0; i < NPART; i++)
 	{
 		if(partsptr[i].type)
@@ -565,6 +577,12 @@ void *build_save_OPS(int *size, int orig_x0, int orig_y0, int orig_w, int orig_h
 	//Store number of particles in each position
 	partsPosData = malloc(fullW*fullH*3);
 	partsPosDataLen = 0;
+	if (!partsPosData)
+	{
+		puts("Save Error, out of memory\n");
+		outputData = NULL;
+		goto fin;
+	}
 	for (y=0;y<fullH;y++)
 	{
 		for (x=0;x<fullW;x++)
@@ -586,6 +604,12 @@ void *build_save_OPS(int *size, int orig_x0, int orig_y0, int orig_w, int orig_h
 	partsDataLen = 0;
 	partsSaveIndex = calloc(NPART, sizeof(unsigned));
 	partsCount = 0;
+	if (!partsData || !partsSaveIndex)
+	{
+		puts("Save Error, out of memory\n");
+		outputData = NULL;
+		goto fin;
+	}
 	for (y=0;y<fullH;y++)
 	{
 		for (x=0;x<fullW;x++)
@@ -726,6 +750,12 @@ void *build_save_OPS(int *size, int orig_x0, int orig_y0, int orig_w, int orig_h
 
 	soapLinkData = malloc(3*elementCount[PT_SOAP]);
 	soapLinkDataLen = 0;
+	if (!soapLinkData)
+	{
+		puts("Save Error, out of memory\n");
+		outputData = NULL;
+		goto fin;
+	}
 	//Iterate through particles in the same order that they were saved
 	for (y=0;y<fullH;y++)
 	{
@@ -822,7 +852,13 @@ void *build_save_OPS(int *size, int orig_x0, int orig_y0, int orig_w, int orig_h
 	finalData = bson_data(&b);
 	finalDataLen = bson_size(&b);
 	outputDataLen = finalDataLen*2+12;
-	outputData = malloc(outputDataLen);
+	outputData = (unsigned char*)malloc(outputDataLen);
+	if (!outputData)
+	{
+		puts("Save Error, out of memory\n");
+		outputData = NULL;
+		goto fin;
+	}
 
 	outputData[0] = 'O';
 	outputData[1] = 'P';