Modify ResourceMessageFilter so that it can support attaching to PluginProcessHost.

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"
@@ skipping 12 matching lines @@
 namespace content {
 
 namespace {
 
 const size_t kMaxSocketStreamHosts = 16 * 1024;
 
 }  // namespace
 
 SocketStreamDispatcherHost::SocketStreamDispatcherHost(
     int render_process_id,
-    ResourceMessageFilter::URLRequestContextSelector* selector,
+    const GetRequestContextCallback& request_context_callback,
     ResourceContext* resource_context)
     : render_process_id_(render_process_id),
-      url_request_context_selector_(selector),
+      request_context_callback_(request_context_callback),
       resource_context_(resource_context),
       weak_ptr_factory_(this),
       on_shutdown_(false) {
-  DCHECK(selector);
   net::WebSocketJob::EnsureInit();
 }
 
 bool SocketStreamDispatcherHost::OnMessageReceived(const IPC::Message& message,
                                                    bool* message_was_ok) {
   if (on_shutdown_)
     return false;
 
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP_EX(SocketStreamDispatcherHost, message, *message_was_ok)
@@ skipping 205 matching lines @@
   SocketStreamHost* socket_stream_host = hosts_.Lookup(socket_id);
   DCHECK(socket_stream_host);
   delete socket_stream_host;
   hosts_.Remove(socket_id);
   if (!Send(new SocketStreamMsg_Closed(socket_id))) {
     DVLOG(1) << "SocketStreamMsg_Closed failed.";
   }
 }
 
 net::URLRequestContext* SocketStreamDispatcherHost::GetURLRequestContext() {
-  return url_request_context_selector_->GetRequestContext(
-      ResourceType::SUB_RESOURCE);
+  return request_context_callback_.Run(ResourceType::SUB_RESOURCE);
 }
 
 void SocketStreamDispatcherHost::Shutdown() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   // TODO(ukai): Implement IDMap::RemoveAll().
   for (IDMap<SocketStreamHost>::const_iterator iter(&hosts_);
        !iter.IsAtEnd();
        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