content/child/child_thread.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_CHILD_CHILD_THREAD_H_
   6 #define CONTENT_CHILD_CHILD_THREAD_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/memory/scoped_ptr.h"
  12 #include "base/memory/shared_memory.h"
  13 #include "base/memory/weak_ptr.h"
  14 #include "base/power_monitor/power_monitor.h"
  15 #include "base/tracked_objects.h"
  16 #include "content/common/content_export.h"
  17 #include "content/common/message_router.h"
  18 #include "ipc/ipc_message.h"  // For IPC_MESSAGE_LOG_ENABLED.
  19 #include "mojo/public/interfaces/service_provider/service_provider.mojom.h"
  20
  21 namespace base {
  22 class MessageLoop;
  23
  24 namespace debug {
  25 class TraceMemoryController;
  26 }  // namespace debug
  27 }  // namespace base
  28
  29 namespace IPC {
  30 class SyncChannel;
  31 class SyncMessageFilter;
  32 }  // namespace IPC
  33
  34 namespace blink {
  35 class WebFrame;
  36 }  // namespace blink
  37
  38 namespace webkit_glue {
  39 class ResourceLoaderBridge;
  40 }  // namespace webkit_glue
  41
  42 namespace content {
  43 class ChildHistogramMessageFilter;
  44 class ChildResourceMessageFilter;
  45 class ChildSharedBitmapManager;
  46 class FileSystemDispatcher;
  47 class MojoApplication;
  48 class ServiceWorkerDispatcher;
  49 class ServiceWorkerMessageFilter;
  50 class QuotaDispatcher;
  51 class QuotaMessageFilter;
  52 class ResourceDispatcher;
  53 class SocketStreamDispatcher;
  54 class ThreadSafeSender;
  55 class WebSocketDispatcher;
  56 struct RequestInfo;
  57
  58 // The main thread of a child process derives from this class.
  59 class CONTENT_EXPORT ChildThread
  60     : public IPC::Listener,
  61       public IPC::Sender,
  62       public NON_EXPORTED_BASE(mojo::ServiceProvider) {
  63  public:
  64   // Creates the thread.
  65   ChildThread();
  66   // Used for single-process mode and for in process gpu mode.
  67   explicit ChildThread(const std::string& channel_name);
  68   // ChildProcess::main_thread() is reset after Shutdown(), and before the
  69   // destructor, so any subsystem that relies on ChildProcess::main_thread()
  70   // must be terminated before Shutdown returns. In particular, if a subsystem
  71   // has a thread that post tasks to ChildProcess::main_thread(), that thread
  72   // should be joined in Shutdown().
  73   virtual ~ChildThread();
  74   virtual void Shutdown();
  75
  76   // IPC::Sender implementation:
  77   virtual bool Send(IPC::Message* msg) OVERRIDE;
  78
  79   IPC::SyncChannel* channel() { return channel_.get(); }
  80
  81   MessageRouter* GetRouter();
  82
  83   // Allocates a block of shared memory of the given size and
  84   // maps in into the address space. Returns NULL of failure.
  85   // Note: On posix, this requires a sync IPC to the browser process,
  86   // but on windows the child process directly allocates the block.
  87   base::SharedMemory* AllocateSharedMemory(size_t buf_size);
  88
  89   // A static variant that can be called on background threads provided
  90   // the |sender| passed in is safe to use on background threads.
  91   static base::SharedMemory* AllocateSharedMemory(size_t buf_size,
  92                                                   IPC::Sender* sender);
  93
  94   ChildSharedBitmapManager* shared_bitmap_manager() const {
  95     return shared_bitmap_manager_.get();
  96   }
  97
  98   ResourceDispatcher* resource_dispatcher() const {
  99     return resource_dispatcher_.get();
 100   }
 101
 102   SocketStreamDispatcher* socket_stream_dispatcher() const {
 103     return socket_stream_dispatcher_.get();
 104   }
 105
 106   WebSocketDispatcher* websocket_dispatcher() const {
 107     return websocket_dispatcher_.get();
 108   }
 109
 110   FileSystemDispatcher* file_system_dispatcher() const {
 111     return file_system_dispatcher_.get();
 112   }
 113
 114   ServiceWorkerDispatcher* service_worker_dispatcher() const {
 115     return service_worker_dispatcher_.get();
 116   }
 117
 118   QuotaDispatcher* quota_dispatcher() const {
 119     return quota_dispatcher_.get();
 120   }
 121
 122   IPC::SyncMessageFilter* sync_message_filter() const {
 123     return sync_message_filter_.get();
 124   }
 125
 126   // The getter should only be called on the main thread, however the
 127   // IPC::Sender it returns may be safely called on any thread including
 128   // the main thread.
 129   ThreadSafeSender* thread_safe_sender() const {
 130     return thread_safe_sender_.get();
 131   }
 132
 133   ChildHistogramMessageFilter* child_histogram_message_filter() const {
 134     return histogram_message_filter_.get();
 135   }
 136
 137   ServiceWorkerMessageFilter* service_worker_message_filter() const {
 138     return service_worker_message_filter_.get();
 139   }
 140
 141   QuotaMessageFilter* quota_message_filter() const {
 142     return quota_message_filter_.get();
 143   }
 144
 145   base::MessageLoop* message_loop() const { return message_loop_; }
 146
 147   // Returns the one child thread. Can only be called on the main thread.
 148   static ChildThread* current();
 149
 150 #if defined(OS_ANDROID)
 151   // Called on Android's service thread to shutdown the main thread of this
 152   // process.
 153   static void ShutdownThread();
 154 #endif
 155
 156  protected:
 157   friend class ChildProcess;
 158
 159   // Called when the process refcount is 0.
 160   void OnProcessFinalRelease();
 161
 162   virtual bool OnControlMessageReceived(const IPC::Message& msg);
 163
 164   void set_on_channel_error_called(bool on_channel_error_called) {
 165     on_channel_error_called_ = on_channel_error_called;
 166   }
 167
 168   // IPC::Listener implementation:
 169   virtual bool OnMessageReceived(const IPC::Message& msg) OVERRIDE;
 170   virtual void OnChannelConnected(int32 peer_pid) OVERRIDE;
 171   virtual void OnChannelError() OVERRIDE;
 172
 173   // mojo::ServiceProvider implementation:
 174   virtual void ConnectToService(
 175       const mojo::String& url,
 176       const mojo::String& name,
 177       mojo::ScopedMessagePipeHandle message_pipe) OVERRIDE;
 178
 179  private:
 180   class ChildThreadMessageRouter : public MessageRouter {
 181    public:
 182     // |sender| must outlive this object.
 183     explicit ChildThreadMessageRouter(IPC::Sender* sender);
 184     virtual bool Send(IPC::Message* msg) OVERRIDE;
 185
 186    private:
 187     IPC::Sender* const sender_;
 188   };
 189
 190   void Init();
 191
 192   // IPC message handlers.
 193   void OnShutdown();
 194   void OnSetProfilerStatus(tracked_objects::ThreadData::Status status);
 195   void OnGetChildProfilerData(int sequence_number);
 196   void OnDumpHandles();
 197 #if defined(OS_WIN)
 198   void OnProcessBackgrounded(bool background);
 199 #endif
 200 #ifdef IPC_MESSAGE_LOG_ENABLED
 201   void OnSetIPCLoggingEnabled(bool enable);
 202 #endif
 203 #if defined(USE_TCMALLOC)
 204   void OnGetTcmallocStats();
 205 #endif
 206
 207   void EnsureConnected();
 208
 209   scoped_ptr<MojoApplication> mojo_application_;
 210
 211   std::string channel_name_;
 212   scoped_ptr<IPC::SyncChannel> channel_;
 213
 214   // Allows threads other than the main thread to send sync messages.
 215   scoped_refptr<IPC::SyncMessageFilter> sync_message_filter_;
 216
 217   scoped_refptr<ThreadSafeSender> thread_safe_sender_;
 218
 219   // Implements message routing functionality to the consumers of ChildThread.
 220   ChildThreadMessageRouter router_;
 221
 222   // Handles resource loads for this process.
 223   scoped_ptr<ResourceDispatcher> resource_dispatcher_;
 224
 225   // Handles SocketStream for this process.
 226   scoped_ptr<SocketStreamDispatcher> socket_stream_dispatcher_;
 227
 228   scoped_ptr<WebSocketDispatcher> websocket_dispatcher_;
 229
 230   // The OnChannelError() callback was invoked - the channel is dead, don't
 231   // attempt to communicate.
 232   bool on_channel_error_called_;
 233
 234   base::MessageLoop* message_loop_;
 235
 236   scoped_ptr<FileSystemDispatcher> file_system_dispatcher_;
 237
 238   scoped_ptr<ServiceWorkerDispatcher> service_worker_dispatcher_;
 239
 240   scoped_ptr<QuotaDispatcher> quota_dispatcher_;
 241
 242   scoped_refptr<ChildHistogramMessageFilter> histogram_message_filter_;
 243
 244   scoped_refptr<ChildResourceMessageFilter> resource_message_filter_;
 245
 246   scoped_refptr<ServiceWorkerMessageFilter> service_worker_message_filter_;
 247
 248   scoped_refptr<QuotaMessageFilter> quota_message_filter_;
 249
 250   scoped_ptr<ChildSharedBitmapManager> shared_bitmap_manager_;
 251
 252   base::WeakPtrFactory<ChildThread> channel_connected_factory_;
 253
 254   // Observes the trace event system. When tracing is enabled, optionally
 255   // starts profiling the tcmalloc heap.
 256   scoped_ptr<base::debug::TraceMemoryController> trace_memory_controller_;
 257
 258   scoped_ptr<base::PowerMonitor> power_monitor_;
 259
 260   bool in_browser_process_;
 261
 262   DISALLOW_COPY_AND_ASSIGN(ChildThread);
 263 };
 264
 265 }  // namespace content
 266
 267 #endif  // CONTENT_CHILD_CHILD_THREAD_H_