diff options
author | Luke Granger-Brown <git@lukegb.com> | 2021-04-22 02:33:59 +0000 |
---|---|---|
committer | Luke Granger-Brown <git@lukegb.com> | 2021-04-22 02:33:59 +0000 |
commit | 97dde3cdd983b2fef9e45dd0ca8d62716a62417d (patch) | |
tree | e20c5813ba043d2199a3b46835f74d7f98f0bc49 /src | |
parent | 8d651a1f68c018b8a10dd37da81e9d3612073656 (diff) |
libutil: allow decompression with none/empty method
The S3 store relies on the ability to be able to decompress things with
an empty method, because it just passes the value of the Content-Encoding
directly to decompress.
If the file is not compressed, then this will cause the compression
routine to get confused.
This caused NixOS/nixpkgs#120120.
Diffstat (limited to 'src')
-rw-r--r-- | src/libutil/compression.cc | 4 | ||||
-rw-r--r-- | src/libutil/tests/compression.cc | 18 |
2 files changed, 21 insertions, 1 deletions
diff --git a/src/libutil/compression.cc b/src/libutil/compression.cc index 08812ef57..c5dae42fa 100644 --- a/src/libutil/compression.cc +++ b/src/libutil/compression.cc @@ -186,7 +186,9 @@ struct BrotliDecompressionSink : ChunkedCompressionSink ref<std::string> decompress(const std::string & method, const std::string & in) { - if (method == "br") { + if (method == "none" || method == "") + return make_ref<std::string>(in); + else if (method == "br") { StringSink ssink; auto sink = makeDecompressionSink(method, ssink); (*sink)(in); diff --git a/src/libutil/tests/compression.cc b/src/libutil/tests/compression.cc index 5b7a2c5b9..2efa3266b 100644 --- a/src/libutil/tests/compression.cc +++ b/src/libutil/tests/compression.cc @@ -17,6 +17,24 @@ namespace nix { ASSERT_EQ(*o, "this-is-a-test"); } + TEST(decompress, decompressNoneCompressed) { + auto method = "none"; + auto str = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf"; + ref<std::string> o = decompress(method, str); + + ASSERT_EQ(*o, str); + } + + TEST(decompress, decompressEmptyCompressed) { + // Empty-method decompression used e.g. by S3 store + // (Content-Encoding == ""). + auto method = ""; + auto str = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf"; + ref<std::string> o = decompress(method, str); + + ASSERT_EQ(*o, str); + } + TEST(decompress, decompressXzCompressed) { auto method = "xz"; auto str = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf"; |