Implement WebSocketDeflater.

net/websockets/websocket_deflater.h

Index: net/websockets/websocket_deflater.h
diff --git a/net/websockets/websocket_deflater.h b/net/websockets/websocket_deflater.h
new file mode 100644
index 0000000000000000000000000000000000000000..7bc7b79fed91115b6c0dfc9b0559e7eccde2e1de
--- /dev/null
+++ b/net/websockets/websocket_deflater.h
@@ -0,0 +1,74 @@
+// Copyright (c) 2013 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.
+
+#ifndef NET_WEBSOCKETS_WEBSOCKET_DEFLATER_H_
+#define NET_WEBSOCKETS_WEBSOCKET_DEFLATER_H_
+
+#include <deque>
+#include <vector>
+
+#include "base/basictypes.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "net/base/net_export.h"
+#include "third_party/zlib/zlib.h"

[szym, 2013/09/10 06:54:07]

Replace this include with forward declaration:

struct z_stream;

[yhirano, 2013/09/10 06:58:58]

I once implemented so, but I fixed that in response to a ricea's comment at PS2.
https://codereview.chromium.org/23623008/diff/3001/net/websockets/websocket_d...

Should we have 'extern "C"' for a forward declaration of a struct?

[yhirano, 2013/09/10 07:48:40]

Done.

+
+namespace net {
+
+class IOBufferWithSize;
+
+class NET_EXPORT_PRIVATE WebSocketDeflater {
+ public:
+  enum ContextTakeOverMode {
+    DO_NOT_TAKE_OVER_CONTEXT,
+    TAKE_OVER_CONTEXT,
+  };
+
+  explicit WebSocketDeflater(ContextTakeOverMode mode);
+  ~WebSocketDeflater();
+
+  // Returns true if there is no error and false otherwise.
+  // This function must be called exactly once before calling any of
+  // following methods.
+  // |window_bits| must be between 8 and 15 (both inclusive).
+  bool Initialize(int window_bits);
+
+  // Adds bytes to |stream_|.
+  // Returns true if there is no error and false otherwise.
+  bool AddBytes(const char* data, size_t size);
+
+  // Flushes the current processing data.
+  // Returns true if there is no error and false otherwise.
+  bool Finish();
+
+  // Pushes "\x00\x00\xff\xff" to the end of the buffer.
+  void PushSyncMark();
+
+  // Returns the current deflated output.
+  // If the current output is larger than |size| bytes,
+  // returns the first |size| bytes of the current output.
+  // The returned bytes will be dropped from the current output and never be
+  // returned thereafter.
+  scoped_refptr<IOBufferWithSize> GetOutput(size_t size);
+
+  // Returns the size of the current deflated output.
+  size_t CurrentOutputSize() const { return buffer_.size(); }
+
+ private:
+  void ResetContext();
+  int Deflate(int flush);
+
+  scoped_ptr<z_stream> stream_;
+  ContextTakeOverMode mode_;
+  std::deque<char> buffer_;
+  std::vector<char> fixed_buffer_;
+  // true if bytes were added after last Finish().
+  bool are_bytes_added_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebSocketDeflater);
+};
+
+}  // namespace net
+
+#endif  // NET_WEBSOCKETS_WEBSOCKET_DEFLATER_H_