Check if we run out of memory

Author: raz-adroll

c_src/zstd_nif.c

@@ -237,7 +237,11 @@ static ERL_NIF_TERM zstd_nif_flush_compression_stream(ErlNifEnv* env, int argc,
       finished = remaining == 0;
       if(!finished) {
           offset += ZSTD_CStreamOutSize();
-          enif_realloc_binary(&bin, bin.size + ZSTD_CStreamOutSize());
+          if(!enif_realloc_binary(&bin, bin.size + ZSTD_CStreamOutSize())) {
+            enif_release_binary(&bin);
+            return enif_make_tuple2(env, zstd_atom_error, zstd_atom_enomem);
+          }
+          
       }
   } while (!finished);