OLD | NEW |
1 // Copyright (c) 2011 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/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
49 } | 49 } |
50 | 50 |
51 class ServiceProcessTerminateMonitor | 51 class ServiceProcessTerminateMonitor |
52 : public base::win::ObjectWatcher::Delegate { | 52 : public base::win::ObjectWatcher::Delegate { |
53 public: | 53 public: |
54 explicit ServiceProcessTerminateMonitor(Task* terminate_task) | 54 explicit ServiceProcessTerminateMonitor(Task* terminate_task) |
55 : terminate_task_(terminate_task) { | 55 : terminate_task_(terminate_task) { |
56 } | 56 } |
57 void Start() { | 57 void Start() { |
58 string16 event_name = GetServiceProcessTerminateEventName(); | 58 string16 event_name = GetServiceProcessTerminateEventName(); |
59 CHECK(event_name.length() <= MAX_PATH); | 59 DCHECK(event_name.length() <= MAX_PATH); |
60 terminate_event_.Set(CreateEvent(NULL, TRUE, FALSE, event_name.c_str())); | 60 terminate_event_.Set(CreateEvent(NULL, TRUE, FALSE, event_name.c_str())); |
61 watcher_.StartWatching(terminate_event_.Get(), this); | 61 watcher_.StartWatching(terminate_event_.Get(), this); |
62 } | 62 } |
63 | 63 |
64 // base::ObjectWatcher::Delegate implementation. | 64 // base::ObjectWatcher::Delegate implementation. |
65 virtual void OnObjectSignaled(HANDLE object) { | 65 virtual void OnObjectSignaled(HANDLE object) { |
66 terminate_task_->Run(); | 66 terminate_task_->Run(); |
67 terminate_task_.reset(); | 67 terminate_task_.reset(); |
68 } | 68 } |
69 | 69 |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
104 return WaitForSingleObject(event, 0) == WAIT_OBJECT_0; | 104 return WaitForSingleObject(event, 0) == WAIT_OBJECT_0; |
105 } | 105 } |
106 | 106 |
107 struct ServiceProcessState::StateData { | 107 struct ServiceProcessState::StateData { |
108 // An event that is signaled when a service process is ready. | 108 // An event that is signaled when a service process is ready. |
109 base::win::ScopedHandle ready_event; | 109 base::win::ScopedHandle ready_event; |
110 scoped_ptr<ServiceProcessTerminateMonitor> terminate_monitor; | 110 scoped_ptr<ServiceProcessTerminateMonitor> terminate_monitor; |
111 }; | 111 }; |
112 | 112 |
113 void ServiceProcessState::CreateState() { | 113 void ServiceProcessState::CreateState() { |
114 CHECK(!state_); | 114 DCHECK(!state_); |
115 state_ = new StateData; | 115 state_ = new StateData; |
116 } | 116 } |
117 | 117 |
118 bool ServiceProcessState::TakeSingletonLock() { | 118 bool ServiceProcessState::TakeSingletonLock() { |
119 DCHECK(state_); | 119 DCHECK(state_); |
120 string16 event_name = GetServiceProcessReadyEventName(); | 120 string16 event_name = GetServiceProcessReadyEventName(); |
121 CHECK(event_name.length() <= MAX_PATH); | 121 DCHECK(event_name.length() <= MAX_PATH); |
122 base::win::ScopedHandle service_process_ready_event; | 122 base::win::ScopedHandle service_process_ready_event; |
123 service_process_ready_event.Set( | 123 service_process_ready_event.Set( |
124 CreateEvent(NULL, TRUE, FALSE, event_name.c_str())); | 124 CreateEvent(NULL, TRUE, FALSE, event_name.c_str())); |
125 DWORD error = GetLastError(); | 125 DWORD error = GetLastError(); |
126 if ((error == ERROR_ALREADY_EXISTS) || (error == ERROR_ACCESS_DENIED)) | 126 if ((error == ERROR_ALREADY_EXISTS) || (error == ERROR_ACCESS_DENIED)) |
127 return false; | 127 return false; |
128 DCHECK(service_process_ready_event.IsValid()); | 128 DCHECK(service_process_ready_event.IsValid()); |
129 state_->ready_event.Set(service_process_ready_event.Take()); | 129 state_->ready_event.Set(service_process_ready_event.Take()); |
130 return true; | 130 return true; |
131 } | 131 } |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
164 base::win::RemoveCommandFromAutoRun( | 164 base::win::RemoveCommandFromAutoRun( |
165 HKEY_CURRENT_USER, UTF8ToWide(GetObsoleteServiceProcessAutoRunKey())); | 165 HKEY_CURRENT_USER, UTF8ToWide(GetObsoleteServiceProcessAutoRunKey())); |
166 return base::win::RemoveCommandFromAutoRun( | 166 return base::win::RemoveCommandFromAutoRun( |
167 HKEY_CURRENT_USER, UTF8ToWide(GetServiceProcessAutoRunKey())); | 167 HKEY_CURRENT_USER, UTF8ToWide(GetServiceProcessAutoRunKey())); |
168 } | 168 } |
169 | 169 |
170 void ServiceProcessState::TearDownState() { | 170 void ServiceProcessState::TearDownState() { |
171 delete state_; | 171 delete state_; |
172 state_ = NULL; | 172 state_ = NULL; |
173 } | 173 } |
OLD | NEW |