Allow MessageFilters to restrict listening to specific message classes

content/browser/renderer_host/websocket_dispatcher_host.cc

 // 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.
 
 #include "content/browser/renderer_host/websocket_dispatcher_host.h"
 
 #include <string>
 
 #include "base/callback.h"
 #include "base/logging.h"
 #include "base/stl_util.h"
 #include "content/browser/renderer_host/websocket_host.h"
 #include "content/common/websocket_messages.h"
 
 namespace content {
 
 namespace {
 
 // Many methods defined in this file return a WebSocketHostState enum
 // value. Make WebSocketHostState visible at file scope so it doesn't have to be
 // fully-qualified every time.
 typedef WebSocketDispatcherHost::WebSocketHostState WebSocketHostState;
 
+const uint32 kFilteredMessageClasses[] = {
+  WebSocketMsgStart,
+};
+
 }  // namespace
 
 WebSocketDispatcherHost::WebSocketDispatcherHost(
     const GetRequestContextCallback& get_context_callback)
-    : get_context_callback_(get_context_callback),
+    : BrowserMessageFilter(
+          kFilteredMessageClasses, arraysize(kFilteredMessageClasses)),
+      get_context_callback_(get_context_callback),
       websocket_host_factory_(
           base::Bind(&WebSocketDispatcherHost::CreateWebSocketHost,
                      base::Unretained(this))) {}
 
 WebSocketDispatcherHost::WebSocketDispatcherHost(
     const GetRequestContextCallback& get_context_callback,
     const WebSocketHostFactory& websocket_host_factory)
-    : get_context_callback_(get_context_callback),
+    : BrowserMessageFilter(
+          kFilteredMessageClasses, arraysize(kFilteredMessageClasses)),
+      get_context_callback_(get_context_callback),
       websocket_host_factory_(websocket_host_factory) {}
 
 WebSocketHost* WebSocketDispatcherHost::CreateWebSocketHost(int routing_id) {
   return new WebSocketHost(routing_id, this, get_context_callback_.Run());
 }
 
 bool WebSocketDispatcherHost::OnMessageReceived(const IPC::Message& message,
                                                 bool* message_was_ok) {
   switch (message.type()) {
     case WebSocketHostMsg_AddChannelRequest::ID:
@@ skipping 123 matching lines @@
 }
 
 void WebSocketDispatcherHost::DeleteWebSocketHost(int routing_id) {
   WebSocketHostTable::iterator it = hosts_.find(routing_id);
   DCHECK(it != hosts_.end());
   delete it->second;
   hosts_.erase(it);
 }
 
 }  // namespace content