blob: c984b17c2ccf22ab7be045c969d20b4fcf7c9c54 [file] [log] [blame]
Dave Allisonf4b80bc2014-05-14 15:41:25 -07001/*
2 * Copyright (C) 2014 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Andreas Gampec60e1b72015-07-30 08:57:50 -070017#ifdef __ANDROID__
Dave Allisonf4b80bc2014-05-14 15:41:25 -070018#include <android/log.h>
Dave Allison69dfe512014-07-11 17:11:58 +000019#else
20#include <stdarg.h>
21#include <iostream>
22#endif
23
Dave Allisonf4b80bc2014-05-14 15:41:25 -070024#include <dlfcn.h>
25#include <signal.h>
26#include <stdio.h>
27#include <stdlib.h>
28
Dave Allisoncefcea82014-09-16 10:01:01 -070029#include "sigchain.h"
30
Dave Allison69dfe512014-07-11 17:11:58 +000031#if defined(__APPLE__)
32#define _NSIG NSIG
Dave Allison38680092014-08-29 12:29:34 -070033#define sighandler_t sig_t
Dave Allison69dfe512014-07-11 17:11:58 +000034#endif
35
Dave Allisonf4b80bc2014-05-14 15:41:25 -070036namespace art {
37
Mathieu Chartierd0004802014-10-15 16:59:47 -070038typedef int (*SigActionFnPtr)(int, const struct sigaction*, struct sigaction*);
39
Dave Allisonf4b80bc2014-05-14 15:41:25 -070040class SignalAction {
41 public:
Andreas Gampe03c2cc82015-05-22 18:31:50 -070042 SignalAction() : claimed_(false), uses_old_style_(false), special_handler_(nullptr) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -070043 }
44
45 // Claim the signal and keep the action specified.
46 void Claim(const struct sigaction& action) {
47 action_ = action;
48 claimed_ = true;
49 }
50
51 // Unclaim the signal and restore the old action.
52 void Unclaim(int signal) {
Dave Allison1f8ef6f2014-08-20 17:38:41 -070053 claimed_ = false;
Mathieu Chartier2cebb242015-04-21 16:50:40 -070054 sigaction(signal, &action_, nullptr); // Restore old action.
Dave Allisonf4b80bc2014-05-14 15:41:25 -070055 }
56
57 // Get the action associated with this signal.
58 const struct sigaction& GetAction() const {
59 return action_;
60 }
61
62 // Is the signal claimed?
63 bool IsClaimed() const {
64 return claimed_;
65 }
66
67 // Change the recorded action to that specified.
Dave Allison91a83662014-08-28 16:12:40 -070068 // If oldstyle is true then this action is from an older style signal()
69 // call as opposed to sigaction(). In this case the sa_handler is
70 // used when invoking the user's handler.
71 void SetAction(const struct sigaction& action, bool oldstyle) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -070072 action_ = action;
Dave Allison91a83662014-08-28 16:12:40 -070073 uses_old_style_ = oldstyle;
74 }
75
76 bool OldStyle() const {
77 return uses_old_style_;
Dave Allisonf4b80bc2014-05-14 15:41:25 -070078 }
79
Andreas Gampe03c2cc82015-05-22 18:31:50 -070080 void SetSpecialHandler(SpecialSignalHandlerFn fn) {
81 special_handler_ = fn;
82 }
83
84 SpecialSignalHandlerFn GetSpecialHandler() {
85 return special_handler_;
86 }
87
Dave Allisonf4b80bc2014-05-14 15:41:25 -070088 private:
Andreas Gampe03c2cc82015-05-22 18:31:50 -070089 struct sigaction action_; // Action to be performed.
90 bool claimed_; // Whether signal is claimed or not.
91 bool uses_old_style_; // Action is created using signal(). Use sa_handler.
92 SpecialSignalHandlerFn special_handler_; // A special handler executed before user handlers.
Dave Allisonf4b80bc2014-05-14 15:41:25 -070093};
94
95// User's signal handlers
96static SignalAction user_sigactions[_NSIG];
Dave Allisoncefcea82014-09-16 10:01:01 -070097static bool initialized;
98static void* linked_sigaction_sym;
99static void* linked_sigprocmask_sym;
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700100
101static void log(const char* format, ...) {
102 char buf[256];
103 va_list ap;
104 va_start(ap, format);
105 vsnprintf(buf, sizeof(buf), format, ap);
Andreas Gampec60e1b72015-07-30 08:57:50 -0700106#ifdef __ANDROID__
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700107 __android_log_write(ANDROID_LOG_ERROR, "libsigchain", buf);
Dave Allison69dfe512014-07-11 17:11:58 +0000108#else
109 std::cout << buf << "\n";
110#endif
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700111 va_end(ap);
112}
113
114static void CheckSignalValid(int signal) {
115 if (signal <= 0 || signal >= _NSIG) {
116 log("Invalid signal %d", signal);
117 abort();
118 }
119}
120
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700121// Sigchainlib's own handler so we can ensure a managed handler is called first even if nobody
122// claimed a chain. Simply forward to InvokeUserSignalHandler.
123static void sigchainlib_managed_handler_sigaction(int sig, siginfo_t* info, void* context) {
124 InvokeUserSignalHandler(sig, info, context);
125}
126
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700127// Claim a signal chain for a particular signal.
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700128extern "C" void ClaimSignalChain(int signal, struct sigaction* oldaction) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700129 CheckSignalValid(signal);
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700130
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700131 user_sigactions[signal].Claim(*oldaction);
132}
133
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700134extern "C" void UnclaimSignalChain(int signal) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700135 CheckSignalValid(signal);
136
137 user_sigactions[signal].Unclaim(signal);
138}
139
140// Invoke the user's signal handler.
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700141extern "C" void InvokeUserSignalHandler(int sig, siginfo_t* info, void* context) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700142 // Check the arguments.
143 CheckSignalValid(sig);
144
145 // The signal must have been claimed in order to get here. Check it.
146 if (!user_sigactions[sig].IsClaimed()) {
147 abort();
148 }
149
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700150 // Do we have a managed handler? If so, run it first.
151 SpecialSignalHandlerFn managed = user_sigactions[sig].GetSpecialHandler();
152 if (managed != nullptr) {
153 // Call the handler. If it succeeds, we're done.
154 if (managed(sig, info, context)) {
155 return;
156 }
157 }
158
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700159 const struct sigaction& action = user_sigactions[sig].GetAction();
Dave Allison91a83662014-08-28 16:12:40 -0700160 if (user_sigactions[sig].OldStyle()) {
Mathieu Chartier2cebb242015-04-21 16:50:40 -0700161 if (action.sa_handler != nullptr) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700162 action.sa_handler(sig);
Dave Allison69dfe512014-07-11 17:11:58 +0000163 } else {
Dmitriy Ivanova3164b92015-04-01 11:08:45 -0700164 signal(sig, SIG_DFL);
165 raise(sig);
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700166 }
167 } else {
Mathieu Chartier2cebb242015-04-21 16:50:40 -0700168 if (action.sa_sigaction != nullptr) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700169 action.sa_sigaction(sig, info, context);
Dave Allison69dfe512014-07-11 17:11:58 +0000170 } else {
Dmitriy Ivanova3164b92015-04-01 11:08:45 -0700171 signal(sig, SIG_DFL);
172 raise(sig);
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700173 }
174 }
175}
176
Mathieu Chartierd0004802014-10-15 16:59:47 -0700177extern "C" void EnsureFrontOfChain(int signal, struct sigaction* expected_action) {
178 CheckSignalValid(signal);
179 // Read the current action without looking at the chain, it should be the expected action.
180 SigActionFnPtr linked_sigaction = reinterpret_cast<SigActionFnPtr>(linked_sigaction_sym);
181 struct sigaction current_action;
182 linked_sigaction(signal, nullptr, &current_action);
183 // If the sigactions don't match then we put the current action on the chain and make ourself as
184 // the main action.
185 if (current_action.sa_sigaction != expected_action->sa_sigaction) {
186 log("Warning: Unexpected sigaction action found %p\n", current_action.sa_sigaction);
187 user_sigactions[signal].Claim(current_action);
188 linked_sigaction(signal, expected_action, nullptr);
189 }
190}
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700191
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700192extern "C" int sigaction(int signal, const struct sigaction* new_action, struct sigaction* old_action) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700193 // If this signal has been claimed as a signal chain, record the user's
194 // action but don't pass it on to the kernel.
195 // Note that we check that the signal number is in range here. An out of range signal
196 // number should behave exactly as the libc sigaction.
Dmitriy Ivanov34a0c202015-04-02 18:30:22 -0700197 if (signal > 0 && signal < _NSIG && user_sigactions[signal].IsClaimed() &&
198 (new_action == nullptr || new_action->sa_handler != SIG_DFL)) {
Dmitriy Ivanovc01683b2015-01-06 14:55:26 -0800199 struct sigaction saved_action = user_sigactions[signal].GetAction();
Mathieu Chartier2cebb242015-04-21 16:50:40 -0700200 if (new_action != nullptr) {
Dave Allison91a83662014-08-28 16:12:40 -0700201 user_sigactions[signal].SetAction(*new_action, false);
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700202 }
Mathieu Chartier2cebb242015-04-21 16:50:40 -0700203 if (old_action != nullptr) {
Dmitriy Ivanovc01683b2015-01-06 14:55:26 -0800204 *old_action = saved_action;
205 }
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700206 return 0;
207 }
208
209 // Will only get here if the signal chain has not been claimed. We want
210 // to pass the sigaction on to the kernel via the real sigaction in libc.
211
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700212 if (linked_sigaction_sym == nullptr) {
Dave Allisoncefcea82014-09-16 10:01:01 -0700213 // Perform lazy initialization.
214 // This will only occur outside of a signal context since we have
215 // not been initialized and therefore cannot be within the ART
216 // runtime.
217 InitializeSignalChain();
218 }
219
220 if (linked_sigaction_sym == nullptr) {
221 log("Unable to find next sigaction in signal chain");
222 abort();
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700223 }
Mathieu Chartierd0004802014-10-15 16:59:47 -0700224 SigActionFnPtr linked_sigaction = reinterpret_cast<SigActionFnPtr>(linked_sigaction_sym);
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700225 return linked_sigaction(signal, new_action, old_action);
226}
227
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700228extern "C" sighandler_t signal(int signal, sighandler_t handler) {
Dave Allison91a83662014-08-28 16:12:40 -0700229 struct sigaction sa;
230 sigemptyset(&sa.sa_mask);
231 sa.sa_handler = handler;
232 sa.sa_flags = SA_RESTART;
233 sighandler_t oldhandler;
234
235 // If this signal has been claimed as a signal chain, record the user's
236 // action but don't pass it on to the kernel.
237 // Note that we check that the signal number is in range here. An out of range signal
238 // number should behave exactly as the libc sigaction.
Dmitriy Ivanov34a0c202015-04-02 18:30:22 -0700239 if (signal > 0 && signal < _NSIG && user_sigactions[signal].IsClaimed() && handler != SIG_DFL) {
Dave Allison91a83662014-08-28 16:12:40 -0700240 oldhandler = reinterpret_cast<sighandler_t>(user_sigactions[signal].GetAction().sa_handler);
241 user_sigactions[signal].SetAction(sa, true);
242 return oldhandler;
243 }
244
245 // Will only get here if the signal chain has not been claimed. We want
246 // to pass the sigaction on to the kernel via the real sigaction in libc.
247
Dave Allison91a83662014-08-28 16:12:40 -0700248 if (linked_sigaction_sym == nullptr) {
Dave Allisoncefcea82014-09-16 10:01:01 -0700249 // Perform lazy initialization.
250 InitializeSignalChain();
251 }
252
253 if (linked_sigaction_sym == nullptr) {
254 log("Unable to find next sigaction in signal chain");
255 abort();
Dave Allison91a83662014-08-28 16:12:40 -0700256 }
257
258 typedef int (*SigAction)(int, const struct sigaction*, struct sigaction*);
259 SigAction linked_sigaction = reinterpret_cast<SigAction>(linked_sigaction_sym);
260 if (linked_sigaction(signal, &sa, &sa) == -1) {
261 return SIG_ERR;
262 }
263
264 return reinterpret_cast<sighandler_t>(sa.sa_handler);
265}
266
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700267extern "C" int sigprocmask(int how, const sigset_t* bionic_new_set, sigset_t* bionic_old_set) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700268 const sigset_t* new_set_ptr = bionic_new_set;
269 sigset_t tmpset;
Mathieu Chartier2cebb242015-04-21 16:50:40 -0700270 if (bionic_new_set != nullptr) {
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700271 tmpset = *bionic_new_set;
272
273 if (how == SIG_BLOCK) {
274 // Don't allow claimed signals in the mask. If a signal chain has been claimed
275 // we can't allow the user to block that signal.
276 for (int i = 0 ; i < _NSIG; ++i) {
277 if (user_sigactions[i].IsClaimed() && sigismember(&tmpset, i)) {
Dmitriy Ivanova3164b92015-04-01 11:08:45 -0700278 sigdelset(&tmpset, i);
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700279 }
280 }
281 }
282 new_set_ptr = &tmpset;
283 }
284
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700285 if (linked_sigprocmask_sym == nullptr) {
Dave Allisoncefcea82014-09-16 10:01:01 -0700286 // Perform lazy initialization.
287 InitializeSignalChain();
288 }
289
290 if (linked_sigprocmask_sym == nullptr) {
291 log("Unable to find next sigprocmask in signal chain");
292 abort();
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700293 }
294
295 typedef int (*SigProcMask)(int how, const sigset_t*, sigset_t*);
296 SigProcMask linked_sigprocmask= reinterpret_cast<SigProcMask>(linked_sigprocmask_sym);
297 return linked_sigprocmask(how, new_set_ptr, bionic_old_set);
298}
Dave Allisoncefcea82014-09-16 10:01:01 -0700299
Dmitriy Ivanovf57874d2014-10-07 13:43:23 -0700300extern "C" void InitializeSignalChain() {
Dave Allisoncefcea82014-09-16 10:01:01 -0700301 // Warning.
302 // Don't call this from within a signal context as it makes calls to
303 // dlsym. Calling into the dynamic linker will result in locks being
304 // taken and if it so happens that a signal occurs while one of these
305 // locks is already taken, dlsym will block trying to reenter a
306 // mutex and we will never get out of it.
307 if (initialized) {
308 // Don't initialize twice.
309 return;
310 }
311 linked_sigaction_sym = dlsym(RTLD_NEXT, "sigaction");
312 if (linked_sigaction_sym == nullptr) {
313 linked_sigaction_sym = dlsym(RTLD_DEFAULT, "sigaction");
314 if (linked_sigaction_sym == nullptr ||
Dmitriy Ivanova3164b92015-04-01 11:08:45 -0700315 linked_sigaction_sym == reinterpret_cast<void*>(sigaction)) {
316 linked_sigaction_sym = nullptr;
Dave Allisoncefcea82014-09-16 10:01:01 -0700317 }
318 }
319
320 linked_sigprocmask_sym = dlsym(RTLD_NEXT, "sigprocmask");
321 if (linked_sigprocmask_sym == nullptr) {
322 linked_sigprocmask_sym = dlsym(RTLD_DEFAULT, "sigprocmask");
323 if (linked_sigprocmask_sym == nullptr ||
324 linked_sigprocmask_sym == reinterpret_cast<void*>(sigprocmask)) {
Dmitriy Ivanova3164b92015-04-01 11:08:45 -0700325 linked_sigprocmask_sym = nullptr;
Dave Allisoncefcea82014-09-16 10:01:01 -0700326 }
327 }
328 initialized = true;
329}
Mathieu Chartierd0004802014-10-15 16:59:47 -0700330
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700331extern "C" void SetSpecialSignalHandlerFn(int signal, SpecialSignalHandlerFn fn) {
332 CheckSignalValid(signal);
333
334 // Set the managed_handler.
335 user_sigactions[signal].SetSpecialHandler(fn);
336
337 // In case the chain isn't claimed, claim it for ourself so we can ensure the managed handler
338 // goes first.
339 if (!user_sigactions[signal].IsClaimed()) {
jgu210c7c55c2015-07-24 13:40:33 +0800340 struct sigaction act, old_act;
341 act.sa_sigaction = sigchainlib_managed_handler_sigaction;
342 sigemptyset(&act.sa_mask);
343 act.sa_flags = SA_SIGINFO | SA_ONSTACK;
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700344#if !defined(__APPLE__) && !defined(__mips__)
jgu210c7c55c2015-07-24 13:40:33 +0800345 act.sa_restorer = nullptr;
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700346#endif
jgu210c7c55c2015-07-24 13:40:33 +0800347 if (sigaction(signal, &act, &old_act) != -1) {
348 user_sigactions[signal].Claim(old_act);
349 }
Andreas Gampe03c2cc82015-05-22 18:31:50 -0700350 }
351}
352
Dave Allisonf4b80bc2014-05-14 15:41:25 -0700353} // namespace art
354