simpleperf/scoped_signal_handler.h

   1 /*
   2  * Copyright (C) 2016 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 SIMPLE_PERF_SCOPED_SIGNAL_HANDLER_H_
  18 #define SIMPLE_PERF_SCOPED_SIGNAL_HANDLER_H_
  19
  20 #include <signal.h>
  21
  22 #include <vector>
  23
  24 class ScopedSignalHandler {
  25  public:
  26   ScopedSignalHandler(const std::vector<int>& signums, void (*handler)(int)) {
  27     for (auto& sig : signums) {
  28       sig_t old_handler = signal(sig, handler);
  29       saved_signal_handlers_.push_back(std::make_pair(sig, old_handler));
  30     }
  31   }
  32
  33   ~ScopedSignalHandler() {
  34     for (auto& pair : saved_signal_handlers_) {
  35       signal(pair.first, pair.second);
  36     }
  37   }
  38
  39  private:
  40   std::vector<std::pair<int, sig_t>> saved_signal_handlers_;
  41 };
  42
  43 #endif  // SIMPLE_PERF_SCOPED_SIGNAL_HANDLER_H_