Index: components/metrics/net/compression_utils.cc |
=================================================================== |
--- components/metrics/net/compression_utils.cc (revision 0) |
+++ components/metrics/net/compression_utils.cc (revision 0) |
@@ -0,0 +1,96 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "components/metrics/net/compression_utils.h" |
+ |
+#include <vector> |
+ |
+#include "base/basictypes.h" |
+#include "third_party/zlib/zlib.h" |
+ |
+namespace { |
+ |
+// The difference in bytes between a zlib header and a gzip header. |
+const size_t kGzipZlibHeaderDifferenceBytes = 16; |
+ |
+// Pass an integer greater than the following get a gzip header instead of a |
+// zlib header when calling deflateInit2_. |
+const int kWindowBitsToGetGzipHeader = 16; |
+ |
+// This describes the amount of memory zlib uses to compress data. It can go |
+// from 1 to 9, with 8 being the default. For details, see: |
+// http://www.zlib.net/manual.html (search for memLevel). |
+const int kZlibMemoryLevel = 8; |
+ |
+// This code is taken almost verbatim from third_party/zlib/compress.c. The only |
+// difference is deflateInit2_ is called which sets the window bits to be > 16. |
+// That causes a gzip header to be emitted rather than a zlib header. |
+int GzipCompressHelper(Bytef* dest, |
+ uLongf* dest_length, |
+ const Bytef* source, |
+ uLong source_length) { |
+ z_stream stream; |
+ |
+ stream.next_in = bit_cast<Bytef*>(source); |
+ stream.avail_in = static_cast<uInt>(source_length); |
+ stream.next_out = dest; |
+ stream.avail_out = static_cast<uInt>(*dest_length); |
+ if (static_cast<uLong>(stream.avail_out) != *dest_length) |
+ return Z_BUF_ERROR; |
+ |
+ stream.zalloc = static_cast<alloc_func>(0); |
+ stream.zfree = static_cast<free_func>(0); |
+ stream.opaque = static_cast<voidpf>(0); |
+ |
+ gz_header gzip_header; |
+ memset(&gzip_header, 0, sizeof(gzip_header)); |
+ int err = deflateInit2_(&stream, |
+ Z_DEFAULT_COMPRESSION, |
+ Z_DEFLATED, |
+ MAX_WBITS + kWindowBitsToGetGzipHeader, |
+ kZlibMemoryLevel, |
+ Z_DEFAULT_STRATEGY, |
+ ZLIB_VERSION, |
+ sizeof(z_stream)); |
+ if (err != Z_OK) |
+ return err; |
+ |
+ err = deflateSetHeader(&stream, &gzip_header); |
+ if (err != Z_OK) |
+ return err; |
+ |
+ err = deflate(&stream, Z_FINISH); |
+ if (err != Z_STREAM_END) { |
+ deflateEnd(&stream); |
+ return err == Z_OK ? Z_BUF_ERROR : err; |
+ } |
+ *dest_length = stream.total_out; |
+ |
+ err = deflateEnd(&stream); |
+ return err; |
+} |
+ |
+} // namespace |
+ |
+namespace metrics { |
+ |
+bool GzipCompress(const std::string& input, std::string* output) { |
+ const uLongf input_size = static_cast<uLongf>(input.size()); |
+ std::vector<Bytef> compressed_data(kGzipZlibHeaderDifferenceBytes + |
+ compressBound(input_size)); |
+ |
+ uLongf compressed_size = static_cast<uLongf>(compressed_data.size()); |
+ if (GzipCompressHelper(&compressed_data.front(), |
+ &compressed_size, |
+ bit_cast<const Bytef*>(input.data()), |
+ input_size) != Z_OK) { |
+ return false; |
+ } |
+ |
+ compressed_data.resize(compressed_size); |
+ output->assign(compressed_data.begin(), compressed_data.end()); |
+ return true; |
+} |
+ |
+} // namespace metrics |