Handle unknown content size

Author: cody-friedrichsen

c_src/zstd_nif.c

@@ -37,12 +37,17 @@ static ERL_NIF_TERM zstd_nif_decompress(ErlNifEnv* env, int argc, const ERL_NIF_
   if(!enif_inspect_binary(env, argv[0], &bin))
     return enif_make_badarg(env);
 
-  uncompressed_size = ZSTD_getDecompressedSize(bin.data, bin.size);
+  uncompressed_size = ZSTD_getFrameContentSize(bin.data, bin.size);
+
+  if (uncompressed_size==ZSTD_CONTENTSIZE_ERROR) {
+      return enif_make_badarg(env);
+  } else if (uncompressed_size==ZSTD_CONTENTSIZE_UNKNOWN) {
+      uncompressed_size = 10000000;
+  }
 
   outp = enif_make_new_binary(env, uncompressed_size, &out);
 
-  if(ZSTD_decompress(outp, uncompressed_size, bin.data, bin.size) != uncompressed_size)
-    return enif_make_atom(env, "error");
+  ZSTD_decompress(outp, uncompressed_size, bin.data, bin.size);
 
   return out;
 }