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_posix.h" | 5 #include "chrome/common/service_process_util_posix.h" |
6 | 6 |
7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
8 #include "base/message_loop_proxy.h" | 8 #include "base/message_loop_proxy.h" |
9 | 9 |
10 namespace { | 10 namespace { |
11 int g_signal_socket = -1; | 11 int g_signal_socket = -1; |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
48 // down by an appropriate process. | 48 // down by an appropriate process. |
49 int message = ServiceProcessShutdownMonitor::kShutDownMessage; | 49 int message = ServiceProcessShutdownMonitor::kShutDownMessage; |
50 if (write(g_signal_socket, &message, sizeof(message)) < 0) { | 50 if (write(g_signal_socket, &message, sizeof(message)) < 0) { |
51 PLOG(ERROR) << "write"; | 51 PLOG(ERROR) << "write"; |
52 } | 52 } |
53 } | 53 } |
54 | 54 |
55 ServiceProcessState::StateData::StateData() {} | 55 ServiceProcessState::StateData::StateData() {} |
56 | 56 |
57 void ServiceProcessState::StateData::SignalReady() { | 57 void ServiceProcessState::StateData::SignalReady() { |
| 58 CHECK_EQ(g_signal_socket, -1); |
58 CHECK(MessageLoopForIO::current()->WatchFileDescriptor( | 59 CHECK(MessageLoopForIO::current()->WatchFileDescriptor( |
59 sockets_[0], true, MessageLoopForIO::WATCH_READ, | 60 sockets_[0], true, MessageLoopForIO::WATCH_READ, |
60 &watcher_, shut_down_monitor_.get())); | 61 &watcher_, shut_down_monitor_.get())); |
61 g_signal_socket = sockets_[1]; | 62 g_signal_socket = sockets_[1]; |
62 | 63 |
63 // Set up signal handler for SIGTERM. | 64 // Set up signal handler for SIGTERM. |
64 struct sigaction action; | 65 struct sigaction action; |
65 action.sa_sigaction = SigTermHandler; | 66 action.sa_sigaction = SigTermHandler; |
66 sigemptyset(&action.sa_mask); | 67 sigemptyset(&action.sa_mask); |
67 action.sa_flags = SA_SIGINFO; | 68 action.sa_flags = SA_SIGINFO; |
(...skipping 19 matching lines...) Expand all Loading... |
87 state_->AddRef(); | 88 state_->AddRef(); |
88 state_->sockets_[0] = -1; | 89 state_->sockets_[0] = -1; |
89 state_->sockets_[1] = -1; | 90 state_->sockets_[1] = -1; |
90 state_->set_action_ = false; | 91 state_->set_action_ = false; |
91 return true; | 92 return true; |
92 } | 93 } |
93 | 94 |
94 bool ServiceProcessState::SignalReady( | 95 bool ServiceProcessState::SignalReady( |
95 base::MessageLoopProxy* message_loop_proxy, Task* shutdown_task) { | 96 base::MessageLoopProxy* message_loop_proxy, Task* shutdown_task) { |
96 CHECK(state_); | 97 CHECK(state_); |
97 CHECK_EQ(g_signal_socket, -1); | 98 scoped_ptr<Task> scoped_shutdown_task(shutdown_task); |
98 | |
99 #if defined(OS_LINUX) | 99 #if defined(OS_LINUX) |
100 state_->running_lock_.reset(TakeServiceRunningLock(true)); | 100 state_->running_lock_.reset(TakeServiceRunningLock(true)); |
101 if (state_->running_lock_.get() == NULL) { | 101 if (state_->running_lock_.get() == NULL) { |
102 return false; | 102 return false; |
103 } | 103 } |
104 #endif // OS_LINUX | 104 #endif // OS_LINUX |
105 state_->shut_down_monitor_.reset( | 105 state_->shut_down_monitor_.reset( |
106 new ServiceProcessShutdownMonitor(shutdown_task)); | 106 new ServiceProcessShutdownMonitor(scoped_shutdown_task.release())); |
107 if (pipe(state_->sockets_) < 0) { | 107 if (pipe(state_->sockets_) < 0) { |
108 PLOG(ERROR) << "pipe"; | 108 PLOG(ERROR) << "pipe"; |
109 return false; | 109 return false; |
110 } | 110 } |
| 111 #if defined(OS_MACOSX) |
| 112 state_->state_ = this; |
| 113 message_loop_proxy->PostTask(FROM_HERE, |
| 114 NewRunnableMethod(state_, |
| 115 &ServiceProcessState::StateData::WatchExecutable)); |
| 116 #endif // OS_MACOSX |
111 message_loop_proxy->PostTask(FROM_HERE, | 117 message_loop_proxy->PostTask(FROM_HERE, |
112 NewRunnableMethod(state_, &ServiceProcessState::StateData::SignalReady)); | 118 NewRunnableMethod(state_, &ServiceProcessState::StateData::SignalReady)); |
113 return true; | 119 return true; |
114 } | 120 } |
115 | 121 |
116 void ServiceProcessState::TearDownState() { | 122 void ServiceProcessState::TearDownState() { |
117 g_signal_socket = -1; | |
118 if (state_) { | 123 if (state_) { |
119 if (state_->sockets_[0] != -1) { | 124 if (state_->sockets_[0] != -1) { |
120 close(state_->sockets_[0]); | 125 close(state_->sockets_[0]); |
121 } | 126 } |
122 if (state_->sockets_[1] != -1) { | 127 if (state_->sockets_[1] != -1) { |
123 close(state_->sockets_[1]); | 128 close(state_->sockets_[1]); |
124 } | 129 } |
125 if (state_->set_action_) { | 130 if (state_->set_action_) { |
126 if (sigaction(SIGTERM, &state_->old_action_, NULL) < 0) { | 131 if (sigaction(SIGTERM, &state_->old_action_, NULL) < 0) { |
127 PLOG(ERROR) << "sigaction"; | 132 PLOG(ERROR) << "sigaction"; |
128 } | 133 } |
129 } | 134 } |
| 135 g_signal_socket = -1; |
130 state_->Release(); | 136 state_->Release(); |
131 state_ = NULL; | 137 state_ = NULL; |
132 } | 138 } |
133 } | 139 } |
OLD | NEW |