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 LIBMEMUNREACHABLE_SCOPED_ALARM_H_ 18 #define LIBMEMUNREACHABLE_SCOPED_ALARM_H_ 19 20 #include <signal.h> 21 #include <sys/time.h> 22 23 #include <chrono> 24 #include <functional> 25 26 class ScopedAlarm { 27 public: 28 ScopedAlarm(std::chrono::microseconds us, std::function<void()> func) { 29 func_ = func; 30 struct sigaction oldact{}; 31 struct sigaction act{}; 32 act.sa_handler = [](int) { 33 ScopedAlarm::func_(); 34 }; 35 sigaction(SIGALRM, &act, &oldact); 36 37 std::chrono::seconds s = std::chrono::duration_cast<std::chrono::seconds>(us); 38 itimerval t = itimerval{}; 39 t.it_value.tv_sec = s.count(); 40 t.it_value.tv_usec = (us - s).count(); 41 setitimer(ITIMER_REAL, &t, NULL); 42 } 43 ~ScopedAlarm() { 44 itimerval t = itimerval{}; 45 setitimer(ITIMER_REAL, &t, NULL); 46 struct sigaction act{}; 47 act.sa_handler = SIG_DFL; 48 sigaction(SIGALRM, &act, NULL); 49 } 50 private: 51 static std::function<void()> func_; 52 }; 53 #endif 54
