Safe zlib inflation in Erlang

zlib_safe_uncompress.erl

-spec zlib_safe_uncompress(binary(), non_neg_integer()) -> binary() | no_return().
zlib_safe_uncompress(CompressedData, UncompressedSize) ->
    Z = zlib:open(),
    zlib:inflateInit(Z),
    zlib:setBufSize(Z, UncompressedSize),
    case zlib:inflateChunk(Z, CompressedData) of
        {more, _Chunk} ->
            error({badarg, uncompressed_size_mismatch});
        Data ->
            zlib:inflateEnd(Z),
            iolist_to_binary(Data)
end.