浏览代码

LibCompress: Add a utility to GZIP compress an entire file

This is copy-pasted from the gzip utility, along with its existing TODO.
This is currently only needed by that utility, but this gives us API
symmetry with GzipDecompressor, and helps ensure we won't end up in a
situation where only one utility receives optimizations that should be
received by all interested parties.
Timothy Flynn 2 年之前
父节点
当前提交
7ec91dfde7
共有 2 个文件被更改,包括 21 次插入0 次删除
  1. 20 0
      Userland/Libraries/LibCompress/Gzip.cpp
  2. 1 0
      Userland/Libraries/LibCompress/Gzip.h

+ 20 - 0
Userland/Libraries/LibCompress/Gzip.cpp

@@ -12,6 +12,8 @@
 #include <AK/MemoryStream.h>
 #include <AK/MemoryStream.h>
 #include <LibCore/DateTime.h>
 #include <LibCore/DateTime.h>
 #include <LibCore/File.h>
 #include <LibCore/File.h>
+#include <LibCore/MappedFile.h>
+#include <LibCore/System.h>
 
 
 namespace Compress {
 namespace Compress {
 
 
@@ -262,4 +264,22 @@ ErrorOr<ByteBuffer> GzipCompressor::compress_all(ReadonlyBytes bytes)
     return buffer;
     return buffer;
 }
 }
 
 
+ErrorOr<void> GzipCompressor::compress_file(StringView input_filename, NonnullOwnPtr<Stream> output_stream)
+{
+    // We map the whole file instead of streaming to reduce size overhead (gzip header) and increase the deflate block size (better compression)
+    // TODO: automatically fallback to buffered streaming for very large files
+    RefPtr<Core::MappedFile> file;
+    ReadonlyBytes input_bytes;
+
+    if (TRY(Core::System::stat(input_filename)).st_size > 0) {
+        file = TRY(Core::MappedFile::map(input_filename));
+        input_bytes = file->bytes();
+    }
+
+    auto output_bytes = TRY(Compress::GzipCompressor::compress_all(input_bytes));
+    TRY(output_stream->write_until_depleted(output_bytes));
+
+    return {};
+}
+
 }
 }

+ 1 - 0
Userland/Libraries/LibCompress/Gzip.h

@@ -93,6 +93,7 @@ public:
     virtual void close() override;
     virtual void close() override;
 
 
     static ErrorOr<ByteBuffer> compress_all(ReadonlyBytes bytes);
     static ErrorOr<ByteBuffer> compress_all(ReadonlyBytes bytes);
+    static ErrorOr<void> compress_file(StringView input_file, NonnullOwnPtr<Stream> output_stream);
 
 
 private:
 private:
     MaybeOwned<Stream> m_output_stream;
     MaybeOwned<Stream> m_output_stream;