OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CONTENT_BROWSER_MACH_BROKER_MAC_H_ | 5 #ifndef CONTENT_BROWSER_MACH_BROKER_MAC_H_ |
6 #define CONTENT_BROWSER_MACH_BROKER_MAC_H_ | 6 #define CONTENT_BROWSER_MACH_BROKER_MAC_H_ |
7 | 7 |
8 #include <mach/mach.h> | |
9 | |
10 #include <map> | 8 #include <map> |
11 #include <string> | 9 #include <string> |
12 | 10 |
13 #include "base/mac/dispatch_source_mach.h" | 11 #include "base/mac/mach_port_broker.h" |
14 #include "base/mac/scoped_mach_port.h" | |
15 #include "base/macros.h" | 12 #include "base/macros.h" |
16 #include "base/memory/scoped_ptr.h" | |
17 #include "base/memory/singleton.h" | 13 #include "base/memory/singleton.h" |
18 #include "base/process/port_provider_mac.h" | 14 #include "base/process/port_provider_mac.h" |
19 #include "base/process/process_handle.h" | 15 #include "base/process/process_handle.h" |
20 #include "base/synchronization/lock.h" | 16 #include "base/synchronization/lock.h" |
21 #include "content/public/browser/browser_child_process_observer.h" | 17 #include "content/public/browser/browser_child_process_observer.h" |
22 #include "content/public/browser/notification_observer.h" | 18 #include "content/public/browser/notification_observer.h" |
23 #include "content/public/browser/notification_registrar.h" | 19 #include "content/public/browser/notification_registrar.h" |
24 | 20 |
25 namespace content { | 21 namespace content { |
26 | 22 |
27 // On OS X, the task port of a process is required to collect metrics about the | 23 // A global |MachBroker| singleton is used by content embedders to provide |
28 // process, and to insert Mach ports into the process. Running |task_for_pid()| | 24 // access to mach task ports for content child processes. |
29 // is only allowed for privileged code. However, a process has port rights to | |
30 // all its subprocesses, so let the browser's child processes send their Mach | |
31 // port to the browser over IPC. | |
32 // | |
33 // Mach ports can only be sent over Mach IPC, not over the |socketpair()| that | |
34 // the regular IPC system uses. Hence, the child processes open a Mach | |
35 // connection shortly after launching and ipc their mach data to the browser | |
36 // process. This data is kept in a global |MachBroker| object. | |
37 // | |
38 // Since this data arrives over a separate channel, it is not available | |
39 // immediately after a child process has been started. | |
40 class CONTENT_EXPORT MachBroker : public base::PortProvider, | 25 class CONTENT_EXPORT MachBroker : public base::PortProvider, |
41 public BrowserChildProcessObserver, | 26 public BrowserChildProcessObserver, |
42 public NotificationObserver { | 27 public NotificationObserver { |
43 public: | 28 public: |
44 // For use in child processes. This will send the task port of the current | 29 // For use in child processes. This will send the task port of the current |
45 // process over Mach IPC to the port registered by name (via this class) in | 30 // process over Mach IPC to the port registered by name (via this class) in |
46 // the parent process. Returns true if the message was sent successfully | 31 // the parent process. Returns true if the message was sent successfully |
47 // and false if otherwise. | 32 // and false if otherwise. |
48 static bool ChildSendTaskPortToParent(); | 33 static bool ChildSendTaskPortToParent(); |
49 | 34 |
50 // Returns the global MachBroker. | 35 // Returns the global MachBroker. |
51 static MachBroker* GetInstance(); | 36 static MachBroker* GetInstance(); |
52 | 37 |
53 // The lock that protects this MachBroker object. Clients MUST acquire and | 38 // The lock that protects this MachBroker object. Clients MUST acquire and |
54 // release this lock around calls to EnsureRunning(), PlaceholderForPid(), | 39 // release this lock around calls to EnsureRunning() and PlaceholderForPid(). |
55 // and FinalizePid(). | |
56 base::Lock& GetLock(); | 40 base::Lock& GetLock(); |
57 | 41 |
58 // Performs any necessary setup that cannot happen in the constructor. | 42 // Performs any necessary setup that cannot happen in the constructor. |
59 // Callers MUST acquire the lock given by GetLock() before calling this | 43 // Callers MUST acquire the lock given by GetLock() before calling this |
60 // method (and release the lock afterwards). | 44 // method (and release the lock afterwards). |
61 void EnsureRunning(); | 45 void EnsureRunning(); |
62 | 46 |
63 // Adds a placeholder to the map for the given pid with MACH_PORT_NULL. | 47 // Adds a placeholder to the map for the given pid with MACH_PORT_NULL. |
64 // Callers are expected to later update the port with FinalizePid(). Callers | 48 // Callers MUST acquire the lock given by GetLock() before calling this method |
65 // MUST acquire the lock given by GetLock() before calling this method (and | 49 // (and release the lock afterwards). |
66 // release the lock afterwards). | |
67 void AddPlaceholderForPid(base::ProcessHandle pid, int child_process_id); | 50 void AddPlaceholderForPid(base::ProcessHandle pid, int child_process_id); |
68 | 51 |
69 // Implement |base::PortProvider|. | 52 // Implement |base::PortProvider|. |
70 mach_port_t TaskForPid(base::ProcessHandle process) const override; | 53 mach_port_t TaskForPid(base::ProcessHandle process) const override; |
71 | 54 |
72 // Implement |BrowserChildProcessObserver|. | 55 // Implement |BrowserChildProcessObserver|. |
73 void BrowserChildProcessHostDisconnected( | 56 void BrowserChildProcessHostDisconnected( |
74 const ChildProcessData& data) override; | 57 const ChildProcessData& data) override; |
75 void BrowserChildProcessCrashed(const ChildProcessData& data, | 58 void BrowserChildProcessCrashed(const ChildProcessData& data, |
76 int exit_code) override; | 59 int exit_code) override; |
77 | 60 |
78 // Implement |NotificationObserver|. | 61 // Implement |NotificationObserver|. |
79 void Observe(int type, | 62 void Observe(int type, |
80 const NotificationSource& source, | 63 const NotificationSource& source, |
81 const NotificationDetails& details) override; | 64 const NotificationDetails& details) override; |
82 | 65 |
83 // Returns the Mach port name to use when sending or receiving messages. | 66 // Returns the Mach port name to use when sending or receiving messages. |
84 // Does the Right Thing in the browser and in child processes. | 67 // Does the Right Thing in the browser and in child processes. |
85 static std::string GetMachPortName(); | 68 static std::string GetMachPortName(); |
86 | 69 |
87 private: | 70 private: |
88 friend class MachBrokerTest; | 71 friend class MachBrokerTest; |
89 friend struct base::DefaultSingletonTraits<MachBroker>; | 72 friend struct base::DefaultSingletonTraits<MachBroker>; |
90 | 73 |
91 MachBroker(); | 74 MachBroker(); |
92 ~MachBroker() override; | 75 ~MachBroker() override; |
93 | 76 |
94 // Performs any initialization work. | |
95 bool Init(); | |
96 | |
97 // Message handler that is invoked on |dispatch_source_| when an | |
98 // incoming message needs to be received. | |
99 void HandleRequest(); | |
100 | |
101 // Updates the mapping for |pid| to include the given |mach_info|. Does | |
102 // nothing if PlaceholderForPid() has not already been called for the given | |
103 // |pid|. Callers MUST acquire the lock given by GetLock() before calling | |
104 // this method (and release the lock afterwards). | |
105 void FinalizePid(base::ProcessHandle pid, mach_port_t task_port); | |
106 | |
107 // Removes all mappings belonging to |child_process_id| from the broker. | 77 // Removes all mappings belonging to |child_process_id| from the broker. |
108 void InvalidateChildProcessId(int child_process_id); | 78 void InvalidateChildProcessId(int child_process_id); |
109 | 79 |
110 // Callback used to register notifications on the UI thread. | 80 // Callback used to register notifications on the UI thread. |
111 void RegisterNotifications(); | 81 void RegisterNotifications(); |
112 | 82 |
113 // Whether or not the class has been initialized. | 83 // Whether or not the class has been initialized. |
114 bool initialized_; | 84 bool initialized_; |
115 | 85 |
116 // Used to register for notifications received by NotificationObserver. | 86 // Used to register for notifications received by NotificationObserver. |
117 // Accessed only on the UI thread. | 87 // Accessed only on the UI thread. |
118 NotificationRegistrar registrar_; | 88 NotificationRegistrar registrar_; |
119 | 89 |
120 // The Mach port on which the server listens. | |
121 base::mac::ScopedMachReceiveRight server_port_; | |
122 | |
123 // The dispatch source and queue on which Mach messages will be received. | |
124 scoped_ptr<base::DispatchSourceMach> dispatch_source_; | |
125 | |
126 // Stores mach info for every process in the broker. | |
127 typedef std::map<base::ProcessHandle, mach_port_t> MachMap; | |
128 MachMap mach_map_; | |
129 | |
130 // Stores the Child process unique id (RenderProcessHost ID) for every | 90 // Stores the Child process unique id (RenderProcessHost ID) for every |
131 // process. | 91 // process. Protected by base::MachPortBroker::GetLock(). |
132 typedef std::map<int, base::ProcessHandle> ChildProcessIdMap; | 92 typedef std::map<int, base::ProcessHandle> ChildProcessIdMap; |
133 ChildProcessIdMap child_process_id_map_; | 93 ChildProcessIdMap child_process_id_map_; |
134 | 94 |
135 // Mutex that guards |mach_map_| and |child_process_id_map_|. | 95 // Underlying port broker that receives and manages mach ports. |
136 mutable base::Lock lock_; | 96 base::MachPortBroker broker_; |
137 | 97 |
138 DISALLOW_COPY_AND_ASSIGN(MachBroker); | 98 DISALLOW_COPY_AND_ASSIGN(MachBroker); |
139 }; | 99 }; |
140 | 100 |
141 } // namespace content | 101 } // namespace content |
142 | 102 |
143 #endif // CONTENT_BROWSER_MACH_BROKER_MAC_H_ | 103 #endif // CONTENT_BROWSER_MACH_BROKER_MAC_H_ |
OLD | NEW |