WebSocket Pepper API: GetBufferedAmount() implementation

webkit/plugins/ppapi/ppb_websocket_impl.h

 // Copyright (c) 2011 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 WEBKIT_PLUGINS_PPAPI_PPB_WEBSOCKET_IMPL_H_
 #define WEBKIT_PLUGINS_PPAPI_PPB_WEBSOCKET_IMPL_H_
 
 #include <queue>
 
 #include "base/memory/scoped_ptr.h"
@@ skipping 48 matching lines @@
   virtual PP_WebSocketReadyState_Dev GetReadyState() OVERRIDE;
   virtual PP_Var GetURL() OVERRIDE;
 
   // WebSocketClient implementation.
   virtual void didConnect() OVERRIDE;
   virtual void didReceiveMessage(const WebKit::WebString& message) OVERRIDE;
   virtual void didReceiveBinaryData(
       const WebKit::WebData& binaryData) OVERRIDE;
   virtual void didReceiveMessageError() OVERRIDE;
   // TODO(toyoshim): Add OVERRIDE after landing WebKit side change.
-  virtual void didUpdateBufferedAmount(unsigned long bufferedAmount);
+  virtual void didUpdateBufferedAmount(unsigned long buffered_amount);
   virtual void didStartClosingHandshake() OVERRIDE;
-  virtual void didClose(unsigned long bufferedAmount,
+  virtual void didClose(unsigned long buffered_amount,
                         ClosingHandshakeCompletionStatus status,
                         unsigned short code,
                         const WebKit::WebString& reason) OVERRIDE;
  private:
   int32_t DoReceive();
+  uint64_t GetFrameSize(uint64_t payload_size);
+  uint64_t SaturateAdd(uint64_t a, uint64_t b);
 
   scoped_ptr<WebKit::WebSocket> websocket_;
   PP_WebSocketReadyState_Dev state_;
 
   PP_CompletionCallback connect_callback_;
 
   PP_CompletionCallback receive_callback_;
   PP_Var* receive_callback_var_;
   bool wait_for_receive_;
   // TODO(toyoshim): Use std::queue<Var> when it supports binary.
   std::queue<PP_Var> received_messages_;
 
   PP_CompletionCallback close_callback_;
   uint16_t close_code_;
   scoped_refptr< ::ppapi::StringVar> close_reason_;
   PP_Bool close_was_clean_;
 
   scoped_refptr< ::ppapi::StringVar> empty_string_;
   scoped_refptr< ::ppapi::StringVar> extensions_;
   scoped_refptr< ::ppapi::StringVar> protocol_;
   scoped_refptr< ::ppapi::StringVar> url_;
 
+  uint64_t buffered_amount_;
+  uint64_t buffered_amount_after_close_;
+
   DISALLOW_COPY_AND_ASSIGN(PPB_WebSocket_Impl);
 };
 
 }  // namespace ppapi
 }  // namespace webkit
 
 #endif  // WEBKIT_PLUGINS_PPAPI_PPB_WEBSOCKET_IMPL_H_