OLD | NEW |
---|---|
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #include "chrome/common/service_process_util.h" | 5 #include "chrome/common/service_process_util.h" |
6 | 6 |
7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
8 #include "base/file_util.h" | 8 #include "base/file_util.h" |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/path_service.h" | 10 #include "base/path_service.h" |
11 #include "base/scoped_handle_win.h" | |
12 #include "base/string16.h" | 11 #include "base/string16.h" |
13 #include "base/utf_string_conversions.h" | 12 #include "base/utf_string_conversions.h" |
14 #include "base/win/object_watcher.h" | 13 #include "base/win/object_watcher.h" |
14 #include "base/win/scoped_handle.h" | |
brettw
2011/01/07 18:03:46
Got some extra indentation.
tfarina
2011/01/07 22:23:08
Done.
| |
15 #include "base/win/win_util.h" | 15 #include "base/win/win_util.h" |
16 #include "chrome/common/chrome_switches.h" | 16 #include "chrome/common/chrome_switches.h" |
17 | 17 |
18 namespace { | 18 namespace { |
19 | 19 |
20 string16 GetServiceProcessReadyEventName() { | 20 string16 GetServiceProcessReadyEventName() { |
21 return UTF8ToWide( | 21 return UTF8ToWide( |
22 GetServiceProcessScopedVersionedName("_service_ready")); | 22 GetServiceProcessScopedVersionedName("_service_ready")); |
23 } | 23 } |
24 | 24 |
(...skipping 15 matching lines...) Expand all Loading... | |
40 watcher_.StartWatching(shutdown_event_.Get(), this); | 40 watcher_.StartWatching(shutdown_event_.Get(), this); |
41 } | 41 } |
42 | 42 |
43 // base::ObjectWatcher::Delegate implementation. | 43 // base::ObjectWatcher::Delegate implementation. |
44 virtual void OnObjectSignaled(HANDLE object) { | 44 virtual void OnObjectSignaled(HANDLE object) { |
45 shutdown_task_->Run(); | 45 shutdown_task_->Run(); |
46 shutdown_task_.reset(); | 46 shutdown_task_.reset(); |
47 } | 47 } |
48 | 48 |
49 private: | 49 private: |
50 ScopedHandle shutdown_event_; | 50 base::win::ScopedHandle shutdown_event_; |
51 base::win::ObjectWatcher watcher_; | 51 base::win::ObjectWatcher watcher_; |
52 scoped_ptr<Task> shutdown_task_; | 52 scoped_ptr<Task> shutdown_task_; |
53 }; | 53 }; |
54 | 54 |
55 } // namespace | 55 } // namespace |
56 | 56 |
57 bool ForceServiceProcessShutdown(const std::string& version) { | 57 bool ForceServiceProcessShutdown(const std::string& version) { |
58 ScopedHandle shutdown_event; | 58 base::win::ScopedHandle shutdown_event; |
59 std::string versioned_name = version; | 59 std::string versioned_name = version; |
60 versioned_name.append("_service_shutdown_evt"); | 60 versioned_name.append("_service_shutdown_evt"); |
61 string16 event_name = | 61 string16 event_name = |
62 UTF8ToWide(GetServiceProcessScopedName(versioned_name)); | 62 UTF8ToWide(GetServiceProcessScopedName(versioned_name)); |
63 shutdown_event.Set(OpenEvent(EVENT_MODIFY_STATE, FALSE, event_name.c_str())); | 63 shutdown_event.Set(OpenEvent(EVENT_MODIFY_STATE, FALSE, event_name.c_str())); |
64 if (!shutdown_event.IsValid()) | 64 if (!shutdown_event.IsValid()) |
65 return false; | 65 return false; |
66 SetEvent(shutdown_event.Get()); | 66 SetEvent(shutdown_event.Get()); |
67 return true; | 67 return true; |
68 } | 68 } |
69 | 69 |
70 bool CheckServiceProcessReady() { | 70 bool CheckServiceProcessReady() { |
71 string16 event_name = GetServiceProcessReadyEventName(); | 71 string16 event_name = GetServiceProcessReadyEventName(); |
72 ScopedHandle event( | 72 base::win::ScopedHandle event( |
73 OpenEvent(SYNCHRONIZE | READ_CONTROL, false, event_name.c_str())); | 73 OpenEvent(SYNCHRONIZE | READ_CONTROL, false, event_name.c_str())); |
74 if (!event.IsValid()) | 74 if (!event.IsValid()) |
75 return false; | 75 return false; |
76 // Check if the event is signaled. | 76 // Check if the event is signaled. |
77 return WaitForSingleObject(event, 0) == WAIT_OBJECT_0; | 77 return WaitForSingleObject(event, 0) == WAIT_OBJECT_0; |
78 } | 78 } |
79 | 79 |
80 struct ServiceProcessState::StateData { | 80 struct ServiceProcessState::StateData { |
81 // An event that is signaled when a service process is ready. | 81 // An event that is signaled when a service process is ready. |
82 ScopedHandle ready_event; | 82 base::win::ScopedHandle ready_event; |
83 scoped_ptr<ServiceProcessShutdownMonitor> shutdown_monitor; | 83 scoped_ptr<ServiceProcessShutdownMonitor> shutdown_monitor; |
84 }; | 84 }; |
85 | 85 |
86 bool ServiceProcessState::TakeSingletonLock() { | 86 bool ServiceProcessState::TakeSingletonLock() { |
87 DCHECK(!state_); | 87 DCHECK(!state_); |
88 string16 event_name = GetServiceProcessReadyEventName(); | 88 string16 event_name = GetServiceProcessReadyEventName(); |
89 CHECK(event_name.length() <= MAX_PATH); | 89 CHECK(event_name.length() <= MAX_PATH); |
90 ScopedHandle service_process_ready_event; | 90 base::win::ScopedHandle service_process_ready_event; |
91 service_process_ready_event.Set( | 91 service_process_ready_event.Set( |
92 CreateEvent(NULL, TRUE, FALSE, event_name.c_str())); | 92 CreateEvent(NULL, TRUE, FALSE, event_name.c_str())); |
93 DWORD error = GetLastError(); | 93 DWORD error = GetLastError(); |
94 if ((error == ERROR_ALREADY_EXISTS) || (error == ERROR_ACCESS_DENIED)) | 94 if ((error == ERROR_ALREADY_EXISTS) || (error == ERROR_ACCESS_DENIED)) |
95 return false; | 95 return false; |
96 DCHECK(service_process_ready_event.IsValid()); | 96 DCHECK(service_process_ready_event.IsValid()); |
97 state_ = new StateData; | 97 state_ = new StateData; |
98 state_->ready_event.Set(service_process_ready_event.Take()); | 98 state_->ready_event.Set(service_process_ready_event.Take()); |
99 return true; | 99 return true; |
100 } | 100 } |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
137 } | 137 } |
138 | 138 |
139 void ServiceProcessState::TearDownState() { | 139 void ServiceProcessState::TearDownState() { |
140 delete state_; | 140 delete state_; |
141 state_ = NULL; | 141 state_ = NULL; |
142 } | 142 } |
143 | 143 |
144 bool ServiceProcessState::ShouldHandleOtherVersion() { | 144 bool ServiceProcessState::ShouldHandleOtherVersion() { |
145 return true; | 145 return true; |
146 } | 146 } |
OLD | NEW |