| Index: net/filter/gzip_stream_source.h
|
| diff --git a/net/filter/gzip_stream_source.h b/net/filter/gzip_stream_source.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f52dc1822d571ce7c176716bf4ff52382298937c
|
| --- /dev/null
|
| +++ b/net/filter/gzip_stream_source.h
|
| @@ -0,0 +1,89 @@
|
| +// Copyright 2016 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 "base/memory/scoped_ptr.h"
|
| +#include "net/filter/gzip_header.h"
|
| +#include "net/filter/stream_source.h"
|
| +
|
| +typedef struct z_stream_s z_stream;
|
| +
|
| +namespace net {
|
| +
|
| +class BlockBuffer;
|
| +
|
| +class GzipStreamSource : public StreamSource {
|
| + public:
|
| + enum GzipStreamSourceMode {
|
| + GZIP_STREAM_SOURCE_DEFLATE,
|
| + GZIP_STREAM_SOURCE_GZIP,
|
| + };
|
| +
|
| + GzipStreamSource(scoped_ptr<StreamSource> previous);
|
| + ~GzipStreamSource() override;
|
| +
|
| + bool Init(GzipStreamSourceMode mode, bool gzip_fallback);
|
| +
|
| + // StreamSource implementation
|
| + net::Error Read(IOBuffer* dest_buffer,
|
| + size_t buffer_size,
|
| + size_t* bytes_read,
|
| + const OnReadCompleteCallback& callback) override;
|
| + size_t GetBytesOutput() const override;
|
| +
|
| + private:
|
| + enum GzipStreamState {
|
| + GZIP_STREAM_ERROR,
|
| + GZIP_STREAM_MORE_INPUT,
|
| + GZIP_STREAM_MORE_OUTPUT_SPACE,
|
| + };
|
| +
|
| + GzipStreamState Decompress(IOBuffer* dest_buffer,
|
| + size_t buffer_size,
|
| + size_t* bytes_read);
|
| +
|
| + GzipStreamState Passthrough(IOBuffer* dest_buffer,
|
| + size_t buffer_size,
|
| + size_t* bytes_read);
|
| +
|
| + void OnReadComplete(const OnReadCompleteCallback& callback,
|
| + IOBuffer* dest_buffer,
|
| + size_t dest_buffer_size,
|
| + Error error,
|
| + size_t bytes_read);
|
| +
|
| + bool InsertZlibHeader();
|
| +
|
| + // Returns whether an invalid GZip header has been observed. This method
|
| + // returns true as soon as an invalid GZip header is observed; it returns
|
| + // false if a complete, valid header has been observed, or not enough bytes
|
| + // have been seen to be sure. This method pulls its input from |buffer_|.
|
| + bool IsGzipHeaderInvalid();
|
| +
|
| + // Returns whether this stream looks like it could be plain text (ie, not
|
| + // actually gzipped). Right now this uses an extremely simple heuristic; see
|
| + // the source for details. This method checks the input in |buffer_|, but does
|
| + // not drain from it.
|
| + bool ShouldFallbackToPlain();
|
| +
|
| + // Skips gzip footer bytes if necessary. Might drain input from |buffer_| if
|
| + // there are still footer bytes to read.
|
| + void SkipGzipFooterIfNeeded();
|
| +
|
| + scoped_refptr<IOBuffer> pending_read_buffer_;
|
| +
|
| + scoped_ptr<BlockBuffer> buffer_;
|
| + scoped_ptr<StreamSource> previous_;
|
| + scoped_ptr<z_stream> zlib_stream_;
|
| + bool zlib_eof_;
|
| + bool zlib_header_added_;
|
| +
|
| + GZipHeader gzip_header_;
|
| + bool gzip_header_unchecked_;
|
| + bool gzip_fallback_unchecked_;
|
| + size_t gzip_footer_bytes_left_;
|
| +
|
| + size_t total_bytes_output_;
|
| +};
|
| +
|
| +} // namespace net
|
|
|