Reduce the complexity of WebSocketThrottle's wakeup code

content/browser/renderer_host/socket_stream_dispatcher_host.cc

 // Copyright (c) 2012 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 "content/browser/renderer_host/socket_stream_dispatcher_host.h"
 
 #include <string>
 
 #include "base/logging.h"
 #include "content/browser/renderer_host/socket_stream_host.h"
 #include "content/browser/ssl/ssl_manager.h"
 #include "content/common/resource_messages.h"
 #include "content/common/socket_stream.h"
 #include "content/common/socket_stream_messages.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/global_request_id.h"
+#include "net/base/net_errors.h"
 #include "net/cookies/canonical_cookie.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/websockets/websocket_job.h"
 #include "net/websockets/websocket_throttle.h"
 
 namespace content {
 
+namespace {
+
+const size_t kMaxSocketStreamHosts = 16 * 1024;
+
+}  // namespace
+
 SocketStreamDispatcherHost::SocketStreamDispatcherHost(
     int render_process_id,
     ResourceMessageFilter::URLRequestContextSelector* selector,
     ResourceContext* resource_context)
     : render_process_id_(render_process_id),
       url_request_context_selector_(selector),
       resource_context_(resource_context),
       weak_ptr_factory_(this),
       on_shutdown_(false) {
   DCHECK(selector);
@@ skipping 156 matching lines @@
 
 // Message handlers called by OnMessageReceived.
 void SocketStreamDispatcherHost::OnConnect(int render_view_id,
                                            const GURL& url,
                                            int socket_id) {
   DVLOG(2) << "SocketStreamDispatcherHost::OnConnect"
            << " render_view_id=" << render_view_id
            << " url=" << url
            << " socket_id=" << socket_id;
   DCHECK_NE(kNoSocketId, socket_id);
+
+  if (hosts_.size() >= kMaxSocketStreamHosts) {
+    if (!Send(new SocketStreamMsg_Failed(socket_id,
+                                         net::ERR_TOO_MANY_SOCKET_STREAMS))) {
+      DVLOG(1) << "SocketStreamMsg_Failed failed.";
+    }
+    if (!Send(new SocketStreamMsg_Closed(socket_id))) {
+      DVLOG(1) << "SocketStreamMsg_Closed failed.";
+    }
+    return;
+  }
+
   if (hosts_.Lookup(socket_id)) {
     DVLOG(1) << "socket_id=" << socket_id << " already registered.";
     return;
   }
 
   // Note that the SocketStreamHost is responsible for checking that |url|
   // is valid.
   SocketStreamHost* socket_stream_host =
       new SocketStreamHost(this, render_view_id, socket_id);
   hosts_.AddWithID(socket_stream_host, socket_id);
@@ skipping 46 matching lines @@
        iter.Advance()) {
     int socket_id = iter.GetCurrentKey();
     const SocketStreamHost* socket_stream_host = iter.GetCurrentValue();
     delete socket_stream_host;
     hosts_.Remove(socket_id);
   }
   on_shutdown_ = true;
 }
 
 }  // namespace content