Simplify how Content*Client interfaces are created. Instead of depending on the embedder to know wh…

content/app/content_main_runner.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/public/app/content_main_runner.h"
 
 #include "base/allocator/allocator_extension.h"
 #include "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/debug/debugger.h"
 #include "base/debug/trace_event.h"
 #include "base/file_path.h"
 #include "base/i18n/icu_util.h"
+#include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/metrics/stats_table.h"
 #include "base/path_service.h"
 #include "base/process_util.h"
 #include "base/stringprintf.h"
 #include "base/string_number_conversions.h"
 #include "content/browser/browser_main.h"
 #include "content/common/set_process_title.h"
 #include "content/common/url_schemes.h"
 #include "content/public/app/content_main_delegate.h"
 #include "content/public/app/startup_helper_win.h"
+#include "content/public/browser/content_browser_client.h"
 #include "content/public/common/content_client.h"
 #include "content/public/common/content_constants.h"
 #include "content/public/common/content_paths.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/main_function_params.h"
 #include "content/public/common/sandbox_init.h"
+#include "content/public/plugin/content_plugin_client.h"
+#include "content/public/renderer/content_renderer_client.h"
+#include "content/public/utility/content_utility_client.h"
 #include "crypto/nss_util.h"
 #include "ipc/ipc_switches.h"
 #include "media/base/media.h"
 #include "sandbox/src/sandbox_types.h"
 #include "ui/base/ui_base_switches.h"
 #include "ui/base/ui_base_paths.h"
 #include "ui/base/win/dpi.h"
 #include "webkit/glue/webkit_glue.h"
 
 #if defined(USE_TCMALLOC)
@@ skipping 32 matching lines @@
 
 extern int GpuMain(const content::MainFunctionParams&);
 extern int PluginMain(const content::MainFunctionParams&);
 extern int PpapiPluginMain(const content::MainFunctionParams&);
 extern int PpapiBrokerMain(const content::MainFunctionParams&);
 extern int RendererMain(const content::MainFunctionParams&);
 extern int WorkerMain(const content::MainFunctionParams&);
 extern int UtilityMain(const content::MainFunctionParams&);
 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_ANDROID)
 namespace content {
-extern int ZygoteMain(const content::MainFunctionParams&,
-                      content::ZygoteForkDelegate* forkdelegate);
+extern int ZygoteMain(const MainFunctionParams&,
+                      ZygoteForkDelegate* forkdelegate);
 }  // namespace content
 #endif
 
-namespace {
+namespace content {
+
+base::LazyInstance<ContentBrowserClient>
+    g_empty_content_browser_client = LAZY_INSTANCE_INITIALIZER;
+base::LazyInstance<ContentPluginClient>
+    g_empty_content_plugin_client = LAZY_INSTANCE_INITIALIZER;
+base::LazyInstance<ContentRendererClient>
+    g_empty_content_renderer_client = LAZY_INSTANCE_INITIALIZER;
+base::LazyInstance<ContentUtilityClient>
+    g_empty_content_utility_client = LAZY_INSTANCE_INITIALIZER;
 
 #if defined(OS_WIN)
 
 static CAppModule _Module;
 
 #elif defined(OS_MACOSX)
 
 // Completes the Mach IPC handshake by sending this process' task port to the
 // parent process.  The parent is listening on the Mach port given by
 // |GetMachPortName()|.  The task port is used by the parent to get CPU/memory
@@ skipping 99 matching lines @@
 static void InitializeStatsTable(const CommandLine& command_line) {
   // Initialize the Stats Counters table.  With this initialized,
   // the StatsViewer can be utilized to read counters outside of
   // Chrome.  These lines can be commented out to effectively turn
   // counters 'off'.  The table is created and exists for the life
   // of the process.  It is not cleaned up.
   if (command_line.HasSwitch(switches::kEnableStatsTable)) {
     // NOTIMPLEMENTED: we probably need to shut this down correctly to avoid
     // leaking shared memory regions on posix platforms.
     std::string statsfile =
-      base::StringPrintf(
-          "%s-%u", content::kStatsFilename,
+      base::StringPrintf("%s-%u", kStatsFilename,
           static_cast<unsigned int>(GetBrowserPid(command_line)));
     base::StatsTable* stats_table = new base::StatsTable(statsfile,
-        content::kStatsMaxThreads, content::kStatsMaxCounters);
+        kStatsMaxThreads, kStatsMaxCounters);
     base::StatsTable::set_current(stats_table);
   }
 }
 
+class ContentClientInitializer {
+ public:
+  static void Set(const std::string& process_type,
+                  ContentMainDelegate* delegate) {
+    ContentClient* content_client = GetContentClient();
+    if (process_type.empty()) {
+      if (delegate)
+        content_client->browser_ = delegate->CreateContentBrowserClient();
+      if (!content_client->browser_)
+        content_client->browser_ = &g_empty_content_browser_client.Get();
+    }
+
+    if (process_type == switches::kPluginProcess) {
+      if (delegate)
+        content_client->plugin_ = delegate->CreateContentPluginClient();
+      if (!content_client->plugin_)
+        content_client->plugin_ = &g_empty_content_plugin_client.Get();
+    } else if (process_type == switches::kRendererProcess) {
+      if (delegate)
+        content_client->renderer_ = delegate->CreateContentRendererClient();
+      if (!content_client->renderer_)
+        content_client->renderer_ = &g_empty_content_renderer_client.Get();
+    } else if (process_type == switches::kUtilityProcess) {
+      if (delegate)
+        content_client->utility_ = delegate->CreateContentUtilityClient();
+      if (!content_client->utility_)
+        content_client->utility_ = &g_empty_content_utility_client.Get();
+    }
+  }
+};
+
 // We dispatch to a process-type-specific FooMain() based on a command-line
 // flag.  This struct is used to build a table of (flag, main function) pairs.
 struct MainFunction {
   const char* name;
-  int (*function)(const content::MainFunctionParams&);
+  int (*function)(const MainFunctionParams&);
 };
 
 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_ANDROID)
 // On platforms that use the zygote, we have a special subset of
 // subprocesses that are launched via the zygote.  This function
 // fills in some process-launching bits around ZygoteMain().
 // Returns the exit code of the subprocess.
-int RunZygote(const content::MainFunctionParams& main_function_params,
-              content::ContentMainDelegate* delegate) {
+int RunZygote(const MainFunctionParams& main_function_params,
+              ContentMainDelegate* delegate) {
   static const MainFunction kMainFunctions[] = {
     { switches::kRendererProcess,    RendererMain },
     { switches::kWorkerProcess,      WorkerMain },
     { switches::kPpapiPluginProcess, PpapiPluginMain },
     { switches::kUtilityProcess,     UtilityMain },
   };
 
-  scoped_ptr<content::ZygoteForkDelegate> zygote_fork_delegate;
+  scoped_ptr<ZygoteForkDelegate> zygote_fork_delegate;
   if (delegate) {
     zygote_fork_delegate.reset(delegate->ZygoteStarting());
     // Each Renderer we spawn will re-attempt initialization of the media
     // libraries, at which point failure will be detected and handled, so
     // we do not need to cope with initialization failures here.
     FilePath media_path;
-    if (PathService::Get(content::DIR_MEDIA_LIBS, &media_path))
+    if (PathService::Get(DIR_MEDIA_LIBS, &media_path))
       media::InitializeMediaLibrary(media_path);
   }
 
   // This function call can return multiple times, once per fork().
-  if (!content::ZygoteMain(main_function_params, zygote_fork_delegate.get()))
+  if (!ZygoteMain(main_function_params, zygote_fork_delegate.get()))
     return 1;
 
   if (delegate) delegate->ZygoteForked();
 
   // Zygote::HandleForkRequest may have reallocated the command
   // line so update it here with the new version.
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
+  std::string process_type =
+      command_line.GetSwitchValueASCII(switches::kProcessType);
+  ContentClientInitializer::Set(process_type, delegate);
 
   // If a custom user agent was passed on the command line, we need
   // to (re)set it now, rather than using the default one the zygote
   // initialized.
   if (command_line.HasSwitch(switches::kUserAgent)) {
     webkit_glue::SetUserAgent(
         command_line.GetSwitchValueASCII(switches::kUserAgent), true);
   }
 
   // The StatsTable must be initialized in each process; we already
   // initialized for the browser process, now we need to initialize
   // within the new processes as well.
   InitializeStatsTable(command_line);
 
-  content::MainFunctionParams main_params(command_line);
-
-  // Get the new process type from the new command line.
-  std::string process_type =
-      command_line.GetSwitchValueASCII(switches::kProcessType);
+  MainFunctionParams main_params(command_line);
 
   for (size_t i = 0; i < arraysize(kMainFunctions); ++i) {
     if (process_type == kMainFunctions[i].name)
       return kMainFunctions[i].function(main_params);
   }
 
   if (delegate)
     return delegate->RunProcess(process_type, main_params);
 
   NOTREACHED() << "Unknown zygote process type: " << process_type;
   return 1;
 }
 #endif  // defined(OS_POSIX) && !defined(OS_MACOSX)
 
 // Run the FooMain() for a given process type.
 // If |process_type| is empty, runs BrowserMain().
 // Returns the exit code for this process.
 int RunNamedProcessTypeMain(
     const std::string& process_type,
-    const content::MainFunctionParams& main_function_params,
-    content::ContentMainDelegate* delegate) {
+    const MainFunctionParams& main_function_params,
+    ContentMainDelegate* delegate) {
   static const MainFunction kMainFunctions[] = {
     { "",                            BrowserMain },
     { switches::kRendererProcess,    RendererMain },
     { switches::kPluginProcess,      PluginMain },
     { switches::kWorkerProcess,      WorkerMain },
     { switches::kPpapiPluginProcess, PpapiPluginMain },
     { switches::kPpapiBrokerProcess, PpapiBrokerMain },
     { switches::kUtilityProcess,     UtilityMain },
     { switches::kGpuProcess,         GpuMain },
   };
@@ skipping 18 matching lines @@
 #endif
 
   // If it's a process we don't know about, the embedder should know.
   if (delegate)
     return delegate->RunProcess(process_type, main_function_params);
 
   NOTREACHED() << "Unknown process type: " << process_type;
   return 1;
 }
 
-class ContentMainRunnerImpl : public content::ContentMainRunner {
+class ContentMainRunnerImpl : public ContentMainRunner {
  public:
   ContentMainRunnerImpl()
       : is_initialized_(false),
         is_shutdown_(false),
         completed_basic_startup_(false) {
   }
 
   ~ContentMainRunnerImpl() {
     if (is_initialized_ && !is_shutdown_)
       Shutdown();
   }
 
 #if defined(USE_TCMALLOC)
 static void GetStatsThunk(char* buffer, int buffer_length) {
   MallocExtension::instance()->GetStats(buffer, buffer_length);
 }
 
 static void ReleaseFreeMemoryThunk() {
   MallocExtension::instance()->ReleaseFreeMemory();
 }
 #endif
 
 
 #if defined(OS_WIN)
   virtual int Initialize(HINSTANCE instance,
                          sandbox::SandboxInterfaceInfo* sandbox_info,
-                         content::ContentMainDelegate* delegate) OVERRIDE {
+                         ContentMainDelegate* delegate) OVERRIDE {
     // argc/argv are ignored on Windows; see command_line.h for details.
     int argc = 0;
     char** argv = NULL;
 
-    content::RegisterInvalidParamHandler();
+    RegisterInvalidParamHandler();
     _Module.Init(NULL, static_cast<HINSTANCE>(instance));
 
     if (sandbox_info)
       sandbox_info_ = *sandbox_info;
     else
       memset(&sandbox_info_, 0, sizeof(sandbox_info_));
 
 #else  // !OS_WIN
   virtual int Initialize(int argc,
                          const char** argv,
-                         content::ContentMainDelegate* delegate) OVERRIDE {
+                         ContentMainDelegate* delegate) OVERRIDE {
 
     // NOTE(willchan): One might ask why this call is done here rather than in
     // process_util_linux.cc with the definition of
     // EnableTerminationOnOutOfMemory().  That's because base shouldn't have a
     // dependency on TCMalloc.  Really, we ought to have our allocator shim code
     // implement this EnableTerminationOnOutOfMemory() function.  Whateverz.
     // This works for now.
 #if !defined(OS_MACOSX) && defined(USE_TCMALLOC)
     // For tcmalloc, we need to tell it to behave like new.
     tc_set_new_mode(1);
@@ skipping 50 matching lines @@
 
     // On Android, the command line is initialized when library is loaded.
     // (But *is* initialized here for content shell bringup)
 #if !defined(OS_ANDROID) || defined(ANDROID_UPSTREAM_BRINGUP)
     CommandLine::Init(argc, argv);
 #endif
 
     int exit_code;
     if (delegate && delegate->BasicStartupComplete(&exit_code))
       return exit_code;
-    DCHECK(!delegate || content::GetContentClient()) <<
-        "BasicStartupComplete didn't set the content client";
 
     completed_basic_startup_ = true;
 
     const CommandLine& command_line = *CommandLine::ForCurrentProcess();
     std::string process_type =
-          command_line.GetSwitchValueASCII(switches::kProcessType);
+        command_line.GetSwitchValueASCII(switches::kProcessType);
+
+    if (!GetContentClient())
+      SetContentClient(&empty_content_client_);
+    ContentClientInitializer::Set(process_type, delegate_);
 
     // Enable startup tracing asap to avoid early TRACE_EVENT calls being
     // ignored.
     if (command_line.HasSwitch(switches::kTraceStartup)) {
       base::debug::TraceLog::GetInstance()->SetEnabled(
           command_line.GetSwitchValueASCII(switches::kTraceStartup));
     }
 
 #if defined(OS_MACOSX)
     // We need to allocate the IO Ports before the Sandbox is initialized or
@@ skipping 11 matching lines @@
     }
 
     if (!process_type.empty() &&
         (!delegate || delegate->ShouldSendMachPort(process_type))) {
       SendTaskPortToParentProcess();
     }
 #elif defined(OS_WIN)
 #if defined(ENABLE_HIDPI)
     ui::EnableHighDPISupport();
 #endif
-    content::SetupCRT(command_line);
+    SetupCRT(command_line);
 #endif
 
 #if defined(OS_POSIX)
     if (!process_type.empty()) {
       // When you hit Ctrl-C in a terminal running the browser
       // process, a SIGINT is delivered to the entire process group.
       // When debugging the browser process via gdb, gdb catches the
       // SIGINT for the browser process (and dumps you back to the gdb
       // console) but doesn't for the child processes, killing them.
       // The fix is to have child processes ignore SIGINT; they'll die
       // on their own when the browser process goes away.
       //
       // Note that we *can't* rely on BeingDebugged to catch this case because
       // we are the child process, which is not being debugged.
       // TODO(evanm): move this to some shared subprocess-init function.
       if (!base::debug::BeingDebugged())
         signal(SIGINT, SIG_IGN);
     }
 #endif
 
 #if defined(USE_NSS)
     crypto::EarlySetupForNSSInit();
 #endif
 
     ui::RegisterPathProvider();
-    content::RegisterPathProvider();
-    content::RegisterContentSchemes(true);
+    RegisterPathProvider();
+    RegisterContentSchemes(true);
 
     CHECK(icu_util::Initialize());
 
     InitializeStatsTable(command_line);
 
     if (delegate)
       delegate->PreSandboxStartup();
 
     // Set any custom user agent passed on the command line now so the string
     // doesn't change between calls to webkit_glue::GetUserAgent(), otherwise it
     // defaults to the user agent set during SetContentClient().
     if (command_line.HasSwitch(switches::kUserAgent)) {
       webkit_glue::SetUserAgent(
           command_line.GetSwitchValueASCII(switches::kUserAgent), true);
     }
 
     if (!process_type.empty())
       CommonSubprocessInit(process_type);
 
 #if defined(OS_WIN)
-    CHECK(content::InitializeSandbox(sandbox_info));
+    CHECK(InitializeSandbox(sandbox_info));
 #elif defined(OS_MACOSX)
     if (process_type == switches::kRendererProcess ||
         process_type == switches::kPpapiPluginProcess ||
         (delegate && delegate->DelaySandboxInitialization(process_type))) {
       // On OS X the renderer sandbox needs to be initialized later in the
       // startup sequence in RendererMainPlatformDelegate::EnableSandbox().
     } else {
-      CHECK(content::InitializeSandbox());
+      CHECK(InitializeSandbox());
     }
 #endif
 
     if (delegate)
       delegate->SandboxInitialized(process_type);
 
 #if defined(OS_POSIX)
     SetProcessTitleFromCommandLine(argv);
 #endif
 
     // Return -1 to indicate no early termination.
     return -1;
   }
 
   virtual int Run() OVERRIDE {
     DCHECK(is_initialized_);
     DCHECK(!is_shutdown_);
     const CommandLine& command_line = *CommandLine::ForCurrentProcess();
     std::string process_type =
           command_line.GetSwitchValueASCII(switches::kProcessType);
 
-    content::MainFunctionParams main_params(command_line);
+    MainFunctionParams main_params(command_line);
 #if defined(OS_WIN)
     main_params.sandbox_info = &sandbox_info_;
 #elif defined(OS_MACOSX)
     main_params.autorelease_pool = autorelease_pool_.get();
 #endif
 
     return RunNamedProcessTypeMain(process_type, main_params, delegate_);
   }
 
   virtual void Shutdown() OVERRIDE {
@@ skipping 19 matching lines @@
 #if defined(OS_MACOSX)
     autorelease_pool_.reset(NULL);
 #endif
 
     exit_manager_.reset(NULL);
 
     delegate_ = NULL;
     is_shutdown_ = true;
   }
 
- protected:
+ private:
   // True if the runner has been initialized.
   bool is_initialized_;
 
   // True if the runner has been shut down.
   bool is_shutdown_;
 
   // True if basic startup was completed.
   bool completed_basic_startup_;
 
+  // Used if the embedder doesn't set one.
+  ContentClient empty_content_client_;
+
   // The delegate will outlive this object.
-  content::ContentMainDelegate* delegate_;
+  ContentMainDelegate* delegate_;
 
   scoped_ptr<base::AtExitManager> exit_manager_;
 #if defined(OS_WIN)
   sandbox::SandboxInterfaceInfo sandbox_info_;
 #elif defined(OS_MACOSX)
   scoped_ptr<base::mac::ScopedNSAutoreleasePool> autorelease_pool_;
 #endif
 
   DISALLOW_COPY_AND_ASSIGN(ContentMainRunnerImpl);
 };
 
-}  // namespace
-
-namespace content {
-
 // static
 ContentMainRunner* ContentMainRunner::Create() {
   return new ContentMainRunnerImpl();
 }
 
 }  // namespace content