1 /*
2  * Copyright (C) 2012 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 
17 #ifndef _BIONIC_TESTS_SCOPED_SIGNAL_HANDLER_H
18 #define _BIONIC_TESTS_SCOPED_SIGNAL_HANDLER_H
19 
20 #include <signal.h>
21 #include <string.h>
22 
23 class ScopedSignalHandler {
24  public:
25   ScopedSignalHandler(int signal_number, void (*handler)(int), int sa_flags = 0)
signal_number_(signal_number)26       : signal_number_(signal_number) {
27     memset(&action_, 0, sizeof(action_));
28     action_.sa_flags = sa_flags;
29     action_.sa_handler = handler;
30     sigaction(signal_number_, &action_, &old_action_);
31   }
32 
33   ScopedSignalHandler(int signal_number, void (*action)(int, siginfo_t*, void*),
34                       int sa_flags = SA_SIGINFO)
signal_number_(signal_number)35       : signal_number_(signal_number) {
36     memset(&action_, 0, sizeof(action_));
37     action_.sa_flags = sa_flags;
38     action_.sa_sigaction = action;
39     sigaction(signal_number_, &action_, &old_action_);
40   }
41 
ScopedSignalHandler(int signal_number)42   ScopedSignalHandler(int signal_number) : signal_number_(signal_number) {
43     sigaction(signal_number, nullptr, &old_action_);
44   }
45 
~ScopedSignalHandler()46   ~ScopedSignalHandler() {
47     sigaction(signal_number_, &old_action_, NULL);
48   }
49 
50  private:
51   struct sigaction action_;
52   struct sigaction old_action_;
53   const int signal_number_;
54 };
55 
56 class ScopedSignalMask {
57  public:
ScopedSignalMask()58   ScopedSignalMask() {
59     sigprocmask(SIG_SETMASK, nullptr, &old_mask_);
60   }
61 
~ScopedSignalMask()62   ~ScopedSignalMask() {
63     sigprocmask(SIG_SETMASK, &old_mask_, nullptr);
64   }
65 
66  private:
67   sigset_t old_mask_;
68 };
69 
70 #endif // _BIONIC_TESTS_SCOPED_SIGNAL_HANDLER_H
71