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_BROWSER_UTILITY_PROCESS_HOST_IMPL_H_ 6 #define CONTENT_BROWSER_UTILITY_PROCESS_HOST_IMPL_H_ 7 8 #include <string> 9 #include <vector> 10 11 #include "base/basictypes.h" 12 #include "base/compiler_specific.h" 13 #include "base/files/file_path.h" 14 #include "base/memory/ref_counted.h" 15 #include "base/memory/scoped_ptr.h" 16 #include "base/memory/weak_ptr.h" 17 #include "content/public/browser/browser_child_process_host_delegate.h" 18 #include "content/public/browser/utility_process_host.h" 19 20 namespace base { 21 class SequencedTaskRunner; 22 class Thread; 23 } 24 25 namespace content { 26 class BrowserChildProcessHostImpl; 27 28 typedef base::Thread* (*UtilityMainThreadFactoryFunction)( 29 const std::string& id); 30 31 class CONTENT_EXPORT UtilityProcessHostImpl NON_EXPORTED_BASE(UtilityProcessHost)32 : public NON_EXPORTED_BASE(UtilityProcessHost), 33 public BrowserChildProcessHostDelegate { 34 public: 35 static void RegisterUtilityMainThreadFactory( 36 UtilityMainThreadFactoryFunction create); 37 38 UtilityProcessHostImpl( 39 const scoped_refptr<UtilityProcessHostClient>& client, 40 const scoped_refptr<base::SequencedTaskRunner>& client_task_runner); 41 virtual ~UtilityProcessHostImpl(); 42 43 // UtilityProcessHost implementation: 44 virtual bool Send(IPC::Message* message) OVERRIDE; 45 virtual bool StartBatchMode() OVERRIDE; 46 virtual void EndBatchMode() OVERRIDE; 47 virtual void SetExposedDir(const base::FilePath& dir) OVERRIDE; 48 virtual void EnableMDns() OVERRIDE; 49 virtual void DisableSandbox() OVERRIDE; 50 #if defined(OS_WIN) 51 virtual void ElevatePrivileges() OVERRIDE; 52 #endif 53 virtual const ChildProcessData& GetData() OVERRIDE; 54 #if defined(OS_POSIX) 55 virtual void SetEnv(const base::EnvironmentMap& env) OVERRIDE; 56 #endif 57 58 void set_child_flags(int flags) { child_flags_ = flags; } 59 60 private: 61 // Starts a process if necessary. Returns true if it succeeded or a process 62 // has already been started via StartBatchMode(). 63 bool StartProcess(); 64 65 // BrowserChildProcessHost: 66 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; 67 virtual void OnProcessLaunchFailed() OVERRIDE; 68 virtual void OnProcessCrashed(int exit_code) OVERRIDE; 69 70 // A pointer to our client interface, who will be informed of progress. 71 scoped_refptr<UtilityProcessHostClient> client_; 72 scoped_refptr<base::SequencedTaskRunner> client_task_runner_; 73 // True when running in batch mode, i.e., StartBatchMode() has been called 74 // and the utility process will run until EndBatchMode(). 75 bool is_batch_mode_; 76 77 base::FilePath exposed_dir_; 78 79 // Whether the utility process needs to perform presandbox initialization 80 // for mDNS. 81 bool is_mdns_enabled_; 82 83 // Whether to pass switches::kNoSandbox to the child. 84 bool no_sandbox_; 85 86 // Whether to launch the process with elevated privileges. 87 bool run_elevated_; 88 89 // Flags defined in ChildProcessHost with which to start the process. 90 int child_flags_; 91 92 base::EnvironmentMap env_; 93 94 bool started_; 95 96 scoped_ptr<BrowserChildProcessHostImpl> process_; 97 98 // Used in single-process mode instead of process_. 99 scoped_ptr<base::Thread> in_process_thread_; 100 101 DISALLOW_COPY_AND_ASSIGN(UtilityProcessHostImpl); 102 }; 103 104 } // namespace content 105 106 #endif // CONTENT_BROWSER_UTILITY_PROCESS_HOST_IMPL_H_ 107