Implement content::WebSocketBlobSender

content/browser/renderer_host/websocket_host.h

 // Copyright 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 CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
 #define CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
 
 #include <stdint.h>
 
 #include <string>
@@ skipping 16 matching lines @@
 class WebSocketChannel;
 class URLRequestContext;
 }  // namespace net
 
 namespace IPC {
 class Message;
 }  // namespace IPC
 
 namespace content {
 
+class WebSocketBlobSender;
 class WebSocketDispatcherHost;
 
 // Host of net::WebSocketChannel. The lifetime of an instance of this class is
 // completely controlled by the WebSocketDispatcherHost object.
 class CONTENT_EXPORT WebSocketHost {
  public:
   WebSocketHost(int routing_id,
                 WebSocketDispatcherHost* dispatcher,
                 net::URLRequestContext* url_request_context,
                 base::TimeDelta delay);
   virtual ~WebSocketHost();
 
   // The renderer process is going away.
   // This function is virtual for testing.
   virtual void GoAway();
 
   // General message dispatch. WebSocketDispatcherHost::OnMessageReceived
   // delegates to this method after looking up the |routing_id|.
   virtual bool OnMessageReceived(const IPC::Message& message);
 
   int routing_id() const { return routing_id_; }
 
   bool handshake_succeeded() const { return handshake_succeeded_; }
   void OnHandshakeSucceeded() { handshake_succeeded_ = true; }
 
  private:
+  class WebSocketEventHandler;
+
   // Handlers for each message type, dispatched by OnMessageReceived(), as
   // defined in content/common/websocket_messages.h
 
   void OnAddChannelRequest(const GURL& socket_url,
                            const std::vector<std::string>& requested_protocols,
                            const url::Origin& origin,
                            int render_frame_id);
 
   void AddChannel(const GURL& socket_url,
                   const std::vector<std::string>& requested_protocols,
                   const url::Origin& origin,
                   int render_frame_id);
 
+  void OnSendBlob(const std::string& uuid, uint64_t expected_size);
+
   void OnSendFrame(bool fin,
                    WebSocketMessageType type,
                    const std::vector<char>& data);
 
   void OnFlowControl(int64_t quota);
 
   void OnDropChannel(bool was_clean, uint16_t code, const std::string& reason);
 
+  void BlobSendComplete(int result);
+
+  // non-NULL if and only if this object is currently in "blob sending mode".
+  scoped_ptr<WebSocketBlobSender> blob_sender_;
+
   // The channel we use to send events to the network.
   scoped_ptr<net::WebSocketChannel> channel_;
 
   // The WebSocketHostDispatcher that created this object.
   WebSocketDispatcherHost* const dispatcher_;
 
   // The URL request context for the channel.
   net::URLRequestContext* const url_request_context_;
 
   // The ID used to route messages.
@@ skipping 12 matching lines @@
   bool handshake_succeeded_;
 
   base::WeakPtrFactory<WebSocketHost> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(WebSocketHost);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_