Allow MessageFilters to restrict listening to specific message classes

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;
 
+const uint32 kFilteredMessageClasses[] = {
+  SocketStreamMsgStart,
+};
+
 }  // namespace
 
 SocketStreamDispatcherHost::SocketStreamDispatcherHost(
     int render_process_id,
     const GetRequestContextCallback& request_context_callback,
     ResourceContext* resource_context)
-    : render_process_id_(render_process_id),
+    : BrowserMessageFilter(
+          kFilteredMessageClasses, arraysize(kFilteredMessageClasses)),
+      render_process_id_(render_process_id),
       request_context_callback_(request_context_callback),
       resource_context_(resource_context),
       weak_ptr_factory_(this),
       on_shutdown_(false) {
   net::WebSocketJob::EnsureInit();
 }
 
 bool SocketStreamDispatcherHost::OnMessageReceived(const IPC::Message& message,
                                                    bool* message_was_ok) {
   if (on_shutdown_)
@@ skipping 227 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