OLD | NEW |
| (Empty) |
1 // Copyright 2013 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 NET_WEBSOCKETS_WEBSOCKET_DEFLATER_H_ | |
6 #define NET_WEBSOCKETS_WEBSOCKET_DEFLATER_H_ | |
7 | |
8 #include <deque> | |
9 #include <vector> | |
10 | |
11 #include "base/basictypes.h" | |
12 #include "base/memory/ref_counted.h" | |
13 #include "base/memory/scoped_ptr.h" | |
14 #include "net/base/net_export.h" | |
15 | |
16 extern "C" struct z_stream_s; | |
17 | |
18 namespace net { | |
19 | |
20 class IOBufferWithSize; | |
21 | |
22 class NET_EXPORT_PRIVATE WebSocketDeflater { | |
23 public: | |
24 // Do not reorder or remove entries of this enum. The values of them are used | |
25 // in UMA. | |
26 enum ContextTakeOverMode { | |
27 DO_NOT_TAKE_OVER_CONTEXT, | |
28 TAKE_OVER_CONTEXT, | |
29 NUM_CONTEXT_TAKEOVER_MODE_TYPES, | |
30 }; | |
31 | |
32 explicit WebSocketDeflater(ContextTakeOverMode mode); | |
33 ~WebSocketDeflater(); | |
34 | |
35 // Returns true if there is no error and false otherwise. | |
36 // This function must be called exactly once before calling any of | |
37 // following methods. | |
38 // |window_bits| must be between 8 and 15 (both inclusive). | |
39 bool Initialize(int window_bits); | |
40 | |
41 // Adds bytes to |stream_|. | |
42 // Returns true if there is no error and false otherwise. | |
43 bool AddBytes(const char* data, size_t size); | |
44 | |
45 // Flushes the current processing data. | |
46 // Returns true if there is no error and false otherwise. | |
47 bool Finish(); | |
48 | |
49 // Pushes "\x00\x00\xff\xff" to the end of the buffer. | |
50 void PushSyncMark(); | |
51 | |
52 // Returns the current deflated output. | |
53 // If the current output is larger than |size| bytes, | |
54 // returns the first |size| bytes of the current output. | |
55 // The returned bytes will be dropped from the current output and never be | |
56 // returned thereafter. | |
57 scoped_refptr<IOBufferWithSize> GetOutput(size_t size); | |
58 | |
59 // Returns the size of the current deflated output. | |
60 size_t CurrentOutputSize() const { return buffer_.size(); } | |
61 | |
62 private: | |
63 void ResetContext(); | |
64 int Deflate(int flush); | |
65 | |
66 scoped_ptr<z_stream_s> stream_; | |
67 ContextTakeOverMode mode_; | |
68 std::deque<char> buffer_; | |
69 std::vector<char> fixed_buffer_; | |
70 // true if bytes were added after last Finish(). | |
71 bool are_bytes_added_; | |
72 | |
73 DISALLOW_COPY_AND_ASSIGN(WebSocketDeflater); | |
74 }; | |
75 | |
76 } // namespace net | |
77 | |
78 #endif // NET_WEBSOCKETS_WEBSOCKET_DEFLATER_H_ | |
OLD | NEW |