| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef BLIMP_NET_COMPRESSED_PACKET_READER_H_ | |
| 6 #define BLIMP_NET_COMPRESSED_PACKET_READER_H_ | |
| 7 | |
| 8 #include <memory> | |
| 9 | |
| 10 #include "base/memory/weak_ptr.h" | |
| 11 #include "blimp/net/blimp_net_export.h" | |
| 12 #include "blimp/net/packet_reader.h" | |
| 13 #include "net/base/completion_callback.h" | |
| 14 #include "net/base/io_buffer.h" | |
| 15 #include "third_party/zlib/zlib.h" | |
| 16 | |
| 17 namespace blimp { | |
| 18 | |
| 19 // Filter object which wraps a PacketReader, adding a DEFLATE decompression step | |
| 20 // to received packets. | |
| 21 // Details about the encoding format are available in the CompressedPacketWriter | |
| 22 // header. | |
| 23 class BLIMP_NET_EXPORT CompressedPacketReader : public PacketReader { | |
| 24 public: | |
| 25 // |source|: The source which from which compressed packets are read. | |
| 26 explicit CompressedPacketReader(std::unique_ptr<PacketReader> source); | |
| 27 | |
| 28 ~CompressedPacketReader() override; | |
| 29 | |
| 30 // PacketReader implementation. | |
| 31 void ReadPacket(const scoped_refptr<net::GrowableIOBuffer>& decompressed_buf, | |
| 32 const net::CompletionCallback& cb) override; | |
| 33 | |
| 34 private: | |
| 35 // Called when source->ReadPacket() has completed. | |
| 36 // |buf|: The buffer which will receive decompressed data. | |
| 37 // |cb|: Callback which is triggered after decompression has finished. | |
| 38 // |result|: The result of the read operation. | |
| 39 void OnCompressedPacketReceived( | |
| 40 const scoped_refptr<net::GrowableIOBuffer> decompressed_buf, | |
| 41 const net::CompletionCallback& cb, | |
| 42 int result); | |
| 43 | |
| 44 // Decompresses the contents of |compressed_buf_| to the buffer | |
| 45 // |decompressed_buf|. The size of |compressed_buf| is passed with the | |
| 46 // argument |size_compressed|. | |
| 47 // On success, returns a >= 0 value indicating the size of the | |
| 48 // decompressed payload. | |
| 49 // On failure, returns a negative value indicating the error code | |
| 50 // (see net_errors.h). | |
| 51 int DecompressPacket( | |
| 52 const scoped_refptr<net::GrowableIOBuffer>& decompressed_buf, | |
| 53 int size_compressed); | |
| 54 | |
| 55 std::unique_ptr<PacketReader> source_; | |
| 56 scoped_refptr<net::GrowableIOBuffer> compressed_buf_; | |
| 57 z_stream zlib_stream_; | |
| 58 | |
| 59 // Used to abandon pending ReadPacket() calls on teardown. | |
| 60 base::WeakPtrFactory<CompressedPacketReader> weak_factory_; | |
| 61 | |
| 62 DISALLOW_COPY_AND_ASSIGN(CompressedPacketReader); | |
| 63 }; | |
| 64 | |
| 65 } // namespace blimp | |
| 66 | |
| 67 #endif // BLIMP_NET_COMPRESSED_PACKET_READER_H_ | |
| OLD | NEW |