Make the AppCacheDispatcherHost be a MessageFilter so that ResourceMessageFil...

chrome/browser/appcache/appcache_dispatcher_host.h

 // 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_BROWSER_APPCACHE_APPCACHE_DISPATCHER_HOST_H_
 #define CHROME_BROWSER_APPCACHE_APPCACHE_DISPATCHER_HOST_H_
 #pragma once
 
 #include <vector>
 
 #include "base/process.h"
 #include "base/ref_counted.h"
 #include "base/scoped_ptr.h"
 #include "chrome/browser/appcache/appcache_frontend_proxy.h"
+#include "chrome/browser/browser_io_message_filter.h"
 #include "chrome/browser/renderer_host/resource_dispatcher_host.h"
 #include "ipc/ipc_message.h"
 #include "webkit/appcache/appcache_backend_impl.h"
 
 class ChromeAppCacheService;
 class URLRequestContext;
 class URLRequestContextGetter;
 
 // Handles appcache related messages sent to the main browser process from
 // its child processes. There is a distinct host for each child process.
 // Messages are handled on the IO thread. The ResourceMessageFilter and
 // WorkerProcessHost create an instance and delegates calls to it.
-class AppCacheDispatcherHost {
+class AppCacheDispatcherHost : public BrowserIOMessageFilter {
  public:
   // Constructor for use on the IO thread.
-  explicit AppCacheDispatcherHost(
-      URLRequestContext* request_context);
+  AppCacheDispatcherHost(URLRequestContext* request_context,
+                         int process_id);
 
   // Constructor for use on the UI thread.
-  explicit AppCacheDispatcherHost(
-      URLRequestContextGetter* request_context_getter);
+  AppCacheDispatcherHost(URLRequestContextGetter* request_context_getter,
+                         int process_id);
 
   ~AppCacheDispatcherHost();
 
-  void Initialize(ResourceDispatcherHost::Receiver* receiver);
-  bool OnMessageReceived(const IPC::Message& msg, bool* msg_is_ok);
-
-  int process_id() const { return backend_impl_.process_id(); }
-
-  // Note: needed to satisfy ipc message dispatching macros.
-  bool Send(IPC::Message* msg) {
-    return frontend_proxy_.sender()->Send(msg);
-  }
+  // BrowserIOMessageFilter implementation
+  virtual bool OnMessageReceived(const IPC::Message& message);
+  virtual void OnChannelConnected(int32 peer_pid);
 
  private:
-  // Ipc message handlers
+  // IPC message handlers
   void OnRegisterHost(int host_id);
   void OnUnregisterHost(int host_id);
   void OnSelectCache(int host_id, const GURL& document_url,
                      int64 cache_document_was_loaded_from,
                      const GURL& opt_manifest_url);
   void OnSelectCacheForWorker(int host_id, int parent_process_id,
                               int parent_host_id);
   void OnSelectCacheForSharedWorker(int host_id, int64 appcache_id);
   void OnMarkAsForeignEntry(int host_id, const GURL& document_url,
                             int64 cache_document_was_loaded_from);
   void OnGetStatus(int host_id, IPC::Message* reply_msg);
   void OnStartUpdate(int host_id, IPC::Message* reply_msg);
   void OnSwapCache(int host_id, IPC::Message* reply_msg);
   void OnGetResourceList(
       int host_id,
       std::vector<appcache::AppCacheResourceInfo>* resource_infos);
   void GetStatusCallback(appcache::Status status, void* param);
   void StartUpdateCallback(bool result, void* param);
   void SwapCacheCallback(bool result, void* param);
 
-  void ReceivedBadMessage(uint32 msg_type);
+  // This is only valid once Initialize() has been called.  This MUST be defined
+  // before backend_impl_ since the latter maintains a (non-refcounted) pointer
+  // to it.
+  scoped_refptr<ChromeAppCacheService> appcache_service_;
 
   AppCacheFrontendProxy frontend_proxy_;
   appcache::AppCacheBackendImpl backend_impl_;
 
-  // Temporary until Initialize() can be called from the IO thread,
+  // Temporary until OnChannelConnected() can be called from the IO thread,
   // which will extract the AppCacheService from the URLRequestContext.
   scoped_refptr<URLRequestContext> request_context_;
   scoped_refptr<URLRequestContextGetter> request_context_getter_;
 
-  // This is only valid once Initialize() has been called.
-  scoped_refptr<ChromeAppCacheService> appcache_service_;
-
   scoped_ptr<appcache::GetStatusCallback> get_status_callback_;
   scoped_ptr<appcache::StartUpdateCallback> start_update_callback_;
   scoped_ptr<appcache::SwapCacheCallback> swap_cache_callback_;
   scoped_ptr<IPC::Message> pending_reply_msg_;
 
-  ResourceDispatcherHost::Receiver* receiver_;
+  // The corresponding ChildProcessHost object's id().
+  int process_id_;
 
   DISALLOW_COPY_AND_ASSIGN(AppCacheDispatcherHost);
 };
 
 #endif  // CHROME_BROWSER_APPCACHE_APPCACHE_DISPATCHER_HOST_H_