Add null checks in case of bad allocation

Author: Lomholy

mcstas-comps/union/Union_logger_2DQ.comp

@@ -397,6 +397,12 @@ double** allocate2Ddouble(int count_x, int count_y) {
 
     // create array or pointers to first elem in each 2D row
     double **ptr_array = malloc(sizeof(double*) * count_x);
+    if (data == NULL || ptr_array == NULL){
+      free(data);
+      free(ptr_array);
+      printf("\nERROR: ptr array or data not allocated in Union_logger_2DQ\n");
+      exit(1);
+    }
     for (i = 0; i < count_x; i++) {
         ptr_array[i] = data + (i*count_y);
     }

mcstas-comps/union/Union_logger_2D_space.comp

@@ -436,6 +436,12 @@ double** allocate2Ddouble_2DS(int count_x, int count_y) {
 
     // create array or pointers to first elem in each 2D row
     double **ptr_array = malloc(sizeof(double*) * count_x);
+    if (data == NULL || ptr_array == NULL){
+      free(data);
+      free(ptr_array);
+      printf("\nERROR: ptr array or data not allocated in Union_logger_2D_space\n");
+      exit(1);
+    }
     for (i = 0; i < count_x; i++) {
         ptr_array[i] = data + (i*count_y);
     }

mcstas-comps/union/Union_logger_3D_space.comp

@@ -457,12 +457,26 @@ void manual_linking_function_logger_processes(char *input_string, struct physics
 double*** allocate3Ddouble_3DS(int count_x, int count_y, int count_z, double *storage) {
    //double *storage = malloc(count_x * count_y * count_z * sizeof(double));
    storage = malloc(count_x * count_y * count_z * sizeof(double));
+   if (storage == NULL){
+    free(storage);
+    printf("\nERROR: storage array not allocated in Union_logger_3D_space\n");
+   }
    double *alloc = storage;
    double ***x;
    int i,j;
    x = malloc(count_x * sizeof(*x));
+   if ( x == NULL){
+      free(x);
+      printf("\nERROR: x array not allocated in Union_logger_3D_space\n");
+      exit(1);
+    }
    for (i = 0; i < count_x; i++) {
      x[i] = malloc(count_y * sizeof(**x));
+     if ( x[i] == NULL){
+      free(x[i]);
+      printf("\nERROR: x[i] array not allocated in Union_logger_3D_space\n");
+      exit(1);
+    }
      for (j = 0; j < count_y; j++) {
        x[i][j] = alloc;
        alloc += count_z;