WebSocket support in chromium.

Run with --enable-web-sockets enables WebSocket features.

BUG=12497
TEST=none

Review URL: http://codereview.chromium.org/292044

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@30440 0039d316-1c4b-4281-b951-d872f2087c98

5 files changed, 266 insertions, 4 deletions

diff --git a/chrome/renderer/render_thread.cc b/chrome/renderer/render_thread.cc
index 998a2fa..33702fc 100644
--- a/chrome/renderer/render_thread.cc
+++ b/chrome/renderer/render_thread.cc
@@ -44,6 +44,7 @@
 #include "chrome/renderer/render_view.h"
 #include "chrome/renderer/renderer_webkitclient_impl.h"
 #include "chrome/renderer/renderer_web_database_observer.h"
+#include "chrome/renderer/socket_stream_dispatcher.h"
 #include "chrome/renderer/user_script_slave.h"
 #include "ipc/ipc_message.h"
 #include "webkit/api/public/WebCache.h"
@@ -139,6 +140,7 @@ void RenderThread::Init() {
   dns_master_.reset(new RenderDnsMaster());
   histogram_snapshots_.reset(new RendererHistogramSnapshots());
   appcache_dispatcher_.reset(new AppCacheDispatcher(this));
+  socket_stream_dispatcher_.reset(new SocketStreamDispatcher());
   devtools_agent_filter_ = new DevToolsAgentFilter();
   AddFilter(devtools_agent_filter_.get());
   db_message_filter_ = new DBMessageFilter();
@@ -277,6 +279,8 @@ void RenderThread::OnControlMessageReceived(const IPC::Message& msg) {
   // App cache messages are handled by a delegate.
   if (appcache_dispatcher_->OnMessageReceived(msg))
     return;
+  if (socket_stream_dispatcher_->OnMessageReceived(msg))
+    return;
 
   IPC_BEGIN_MESSAGE_MAP(RenderThread, msg)
     IPC_MESSAGE_HANDLER(ViewMsg_VisitedLink_NewTable, OnUpdateVisitedLinks)
diff --git a/chrome/renderer/render_thread.h b/chrome/renderer/render_thread.h
index 3c3bc78..5dfb1c9 100644
--- a/chrome/renderer/render_thread.h
+++ b/chrome/renderer/render_thread.h
@@ -29,6 +29,7 @@ class RendererHistogram;
 class RendererWebDatabaseObserver;
 class RendererWebKitClientImpl;
 class SkBitmap;
+class SocketStreamDispatcher;
 class UserScriptSlave;
 class URLPattern;
 
@@ -115,6 +116,10 @@ class RenderThread : public RenderThreadBase,
     return appcache_dispatcher_.get();
   }
 
+  SocketStreamDispatcher* socket_stream_dispatcher() const {
+    return socket_stream_dispatcher_.get();
+  }
+
   bool plugin_refresh_allowed() const { return plugin_refresh_allowed_; }
 
   // Do DNS prefetch resolution of a hostname.
@@ -202,6 +207,7 @@ class RenderThread : public RenderThreadBase,
   scoped_ptr<RendererHistogramSnapshots> histogram_snapshots_;
   scoped_ptr<RendererWebKitClientImpl> webkit_client_;
   scoped_ptr<WebKit::WebStorageEventDispatcher> dom_storage_event_dispatcher_;
+  scoped_ptr<SocketStreamDispatcher> socket_stream_dispatcher_;
   scoped_ptr<RendererWebDatabaseObserver> renderer_web_database_observer_;
 
   // Used on the renderer and IPC threads.
diff --git a/chrome/renderer/renderer_glue.cc b/chrome/renderer/renderer_glue.cc
index 43012e9..2dab562 100644
--- a/chrome/renderer/renderer_glue.cc
+++ b/chrome/renderer/renderer_glue.cc
@@ -24,6 +24,7 @@
 #include "chrome/renderer/net/render_dns_master.h"
 #include "chrome/renderer/render_process.h"
 #include "chrome/renderer/render_thread.h"
+#include "chrome/renderer/socket_stream_dispatcher.h"
 #include "googleurl/src/url_util.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "webkit/api/public/WebKit.h"
@@ -245,14 +246,13 @@ ResourceLoaderBridge* ResourceLoaderBridge::Create(
                                 appcache_host_id, routing_id);
 }
 
-
 // static factory function
 WebSocketStreamHandleBridge* WebSocketStreamHandleBridge::Create(
     WebKit::WebSocketStreamHandle* handle,
     WebSocketStreamHandleDelegate* delegate) {
-  // TODO(ukai): implement dispathcer class.
-  NOTREACHED();
-  return NULL;
+  SocketStreamDispatcher* dispatcher =
+      RenderThread::current()->socket_stream_dispatcher();
+  return dispatcher->CreateBridge(handle, delegate);
 }
 
 void NotifyCacheStats() {
diff --git a/chrome/renderer/socket_stream_dispatcher.cc b/chrome/renderer/socket_stream_dispatcher.cc
new file mode 100644
index 0000000..aa9caec
--- /dev/null
+++ b/chrome/renderer/socket_stream_dispatcher.cc
@@ -0,0 +1,214 @@
+// Copyright (c) 2009 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.
+
+#include "chrome/renderer/socket_stream_dispatcher.h"
+
+#include <vector>
+
+#include "base/id_map.h"
+#include "base/ref_counted.h"
+#include "chrome/common/render_messages.h"
+#include "chrome/common/net/socket_stream.h"
+#include "chrome/renderer/render_thread.h"
+#include "googleurl/src/gurl.h"
+#include "webkit/glue/websocketstreamhandle_bridge.h"
+#include "webkit/glue/websocketstreamhandle_delegate.h"
+
+// IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle.
+// It communicates with the main browser process via SocketStreamDispatcher.
+class IPCWebSocketStreamHandleBridge
+    : public webkit_glue::WebSocketStreamHandleBridge {
+ public:
+  IPCWebSocketStreamHandleBridge(
+      IPC::Message::Sender* sender,
+      WebKit::WebSocketStreamHandle* handle,
+      webkit_glue::WebSocketStreamHandleDelegate* delegate)
+      : socket_id_(chrome_common_net::kNoSocketId),
+        sender_(sender),
+        handle_(handle),
+        delegate_(delegate) {}
+
+  // Returns the handle having given id or NULL if there is no such handle.
+  static IPCWebSocketStreamHandleBridge* FromSocketId(int id);
+
+  // webkit_glue::WebSocketStreamHandleBridge methods.
+  virtual void Connect(const GURL& url);
+  virtual bool Send(const std::vector<char>& data);
+  virtual void Close();
+
+  // Called by SocketStreamDispatcher.
+  void OnConnected(int max_amount_send_allowed);
+  void OnSentData(int amount_sent);
+  void OnReceivedData(const std::vector<char>& data);
+  void OnClosed();
+
+ private:
+  virtual ~IPCWebSocketStreamHandleBridge();
+
+  void DoConnect(const GURL& url);
+  int socket_id_;
+
+  IPC::Message::Sender* sender_;
+  WebKit::WebSocketStreamHandle* handle_;
+  webkit_glue::WebSocketStreamHandleDelegate* delegate_;
+
+  static IDMap<IPCWebSocketStreamHandleBridge> all_bridges;
+};
+
+IDMap<IPCWebSocketStreamHandleBridge>
+IPCWebSocketStreamHandleBridge::all_bridges;
+
+/* static */
+IPCWebSocketStreamHandleBridge* IPCWebSocketStreamHandleBridge::FromSocketId(
+    int id) {
+  return all_bridges.Lookup(id);
+}
+
+IPCWebSocketStreamHandleBridge::~IPCWebSocketStreamHandleBridge() {
+  DLOG(INFO) << "IPCWebSocketStreamHandleBridge destructor socket_id="
+             << socket_id_;
+  if (socket_id_ != chrome_common_net::kNoSocketId) {
+    sender_->Send(new ViewHostMsg_Close(socket_id_));
+    socket_id_ = chrome_common_net::kNoSocketId;
+  }
+}
+
+void IPCWebSocketStreamHandleBridge::Connect(const GURL& url) {
+  DCHECK(sender_);
+  DLOG(INFO) << "Connect url=" << url;
+  MessageLoop::current()->PostTask(
+      FROM_HERE,
+      NewRunnableMethod(this, &IPCWebSocketStreamHandleBridge::DoConnect,
+                        url));
+}
+
+bool IPCWebSocketStreamHandleBridge::Send(
+    const std::vector<char>& data) {
+  DLOG(INFO) << "Send data.size=" << data.size();
+  if (sender_->Send(new ViewHostMsg_SocketStream_SendData(socket_id_, data))) {
+    if (delegate_)
+      delegate_->WillSendData(handle_, &data[0], data.size());
+    return true;
+  }
+  return false;
+}
+
+void IPCWebSocketStreamHandleBridge::Close() {
+  DLOG(INFO) << "Close socket_id" << socket_id_;
+  sender_->Send(new ViewHostMsg_SocketStream_Close(socket_id_));
+}
+
+void IPCWebSocketStreamHandleBridge::OnConnected(int max_pending_send_allowed) {
+  DLOG(INFO) << "IPCWebSocketStreamHandleBridge::OnConnected socket_id="
+             << socket_id_;
+  if (delegate_)
+    delegate_->DidOpenStream(handle_, max_pending_send_allowed);
+}
+
+void IPCWebSocketStreamHandleBridge::OnSentData(int amount_sent) {
+  if (delegate_)
+    delegate_->DidSendData(handle_, amount_sent);
+}
+
+void IPCWebSocketStreamHandleBridge::OnReceivedData(
+    const std::vector<char>& data) {
+  if (delegate_)
+    delegate_->DidReceiveData(handle_, &data[0], data.size());
+}
+
+void IPCWebSocketStreamHandleBridge::OnClosed() {
+  DLOG(INFO) << "IPCWebSocketStreamHandleBridge::OnClosed";
+  if (socket_id_ != chrome_common_net::kNoSocketId) {
+    all_bridges.Remove(socket_id_);
+    socket_id_ = chrome_common_net::kNoSocketId;
+  }
+  if (delegate_) {
+    delegate_->DidClose(handle_);
+  }
+  delegate_ = NULL;
+  Release();
+}
+
+void IPCWebSocketStreamHandleBridge::DoConnect(const GURL& url) {
+  DCHECK(sender_);
+  DCHECK_EQ(socket_id_, chrome_common_net::kNoSocketId);
+  if (delegate_)
+    delegate_->WillOpenStream(handle_, url);
+
+  socket_id_ = all_bridges.Add(this);
+  DCHECK_NE(socket_id_, chrome_common_net::kNoSocketId);
+  if (sender_->Send(new ViewHostMsg_SocketStream_Connect(url, socket_id_))) {
+    DLOG(INFO) << "Connect socket_id=" << socket_id_;
+    AddRef();  // Released in OnClosed().
+    // TODO(ukai): timeout to OnConnected.
+  } else {
+    LOG(ERROR) << "IPC SocketStream_Connect failed.";
+    OnClosed();
+  }
+}
+
+SocketStreamDispatcher::SocketStreamDispatcher() {
+}
+
+/* static */
+webkit_glue::WebSocketStreamHandleBridge*
+SocketStreamDispatcher::CreateBridge(
+    WebKit::WebSocketStreamHandle* handle,
+    webkit_glue::WebSocketStreamHandleDelegate* delegate) {
+  return new IPCWebSocketStreamHandleBridge(
+      ChildThread::current(), handle, delegate);
+}
+
+bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) {
+  DLOG(INFO) << "SocketStreamDispatcher::OnMessageReceived";
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg)
+    IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Connected, OnConnected)
+    IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_SentData, OnSentData)
+    IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_ReceivedData, OnReceivedData)
+    IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Closed, OnClosed)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+void SocketStreamDispatcher::OnConnected(int socket_id,
+                                         int max_pending_send_allowed) {
+  DLOG(INFO) << "SocketStreamDispatcher::OnConnected socket_id=" << socket_id
+             << " max_pending_send_allowed=" << max_pending_send_allowed;
+  IPCWebSocketStreamHandleBridge* bridge =
+      IPCWebSocketStreamHandleBridge::FromSocketId(socket_id);
+  if (bridge)
+    bridge->OnConnected(max_pending_send_allowed);
+  else
+    DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id;
+}
+
+void SocketStreamDispatcher::OnSentData(int socket_id, int amount_sent) {
+  IPCWebSocketStreamHandleBridge* bridge =
+      IPCWebSocketStreamHandleBridge::FromSocketId(socket_id);
+  if (bridge)
+    bridge->OnSentData(amount_sent);
+  else
+    DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id;
+}
+
+void SocketStreamDispatcher::OnReceivedData(
+    int socket_id, const std::vector<char>& data) {
+  IPCWebSocketStreamHandleBridge* bridge =
+      IPCWebSocketStreamHandleBridge::FromSocketId(socket_id);
+  if (bridge)
+    bridge->OnReceivedData(data);
+  else
+    DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id;
+}
+
+void SocketStreamDispatcher::OnClosed(int socket_id) {
+  IPCWebSocketStreamHandleBridge* bridge =
+      IPCWebSocketStreamHandleBridge::FromSocketId(socket_id);
+  if (bridge)
+    bridge->OnClosed();
+  else
+    DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id;
+}
diff --git a/chrome/renderer/socket_stream_dispatcher.h b/chrome/renderer/socket_stream_dispatcher.h
new file mode 100644
index 0000000..5399030
--- /dev/null
+++ b/chrome/renderer/socket_stream_dispatcher.h
@@ -0,0 +1,38 @@
+// Copyright (c) 2009 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 CHROME_RENDERER_SOCKET_STREAM_DISPATCHER_H_
+#define CHROME_RENDERER_SOCKET_STREAM_DISPATCHER_H_
+
+#include <vector>
+
+#include "base/basictypes.h"
+#include "ipc/ipc_channel.h"
+#include "ipc/ipc_message.h"
+#include "webkit/glue/websocketstreamhandle_bridge.h"
+
+// Dispatches socket stream related messages sent to a child process from the
+// main browser process.  There is one instance per child process.  Messages
+// are dispatched on the main child thread.  The RenderThread class
+// creates an instance of SocketStreamDispatcher and delegates calls to it.
+class SocketStreamDispatcher {
+ public:
+  SocketStreamDispatcher();
+  ~SocketStreamDispatcher() {}
+
+  static webkit_glue::WebSocketStreamHandleBridge* CreateBridge(
+      WebKit::WebSocketStreamHandle* handle,
+      webkit_glue::WebSocketStreamHandleDelegate* delegate);
+  bool OnMessageReceived(const IPC::Message& msg);
+
+ private:
+  void OnConnected(int socket_id, int max_amount_send_allowed);
+  void OnSentData(int socket_id, int amount_sent);
+  void OnReceivedData(int socket_id, const std::vector<char>& data);
+  void OnClosed(int socket_id);
+
+  DISALLOW_COPY_AND_ASSIGN(SocketStreamDispatcher);
+};
+
+#endif  // CHROME_RENDERER_SOCKET_STREAM_DISPATCHER_H_