blob: 287f479a923c50fec170e683fa266d3637156b2a [file] [log] [blame]
Colin Crossbcb4ed32016-01-14 15:35:40 -08001/*
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>
Colin Cross54a16102016-03-02 17:52:56 -080021#include <sys/time.h>
Colin Crossbcb4ed32016-01-14 15:35:40 -080022
23#include <chrono>
24#include <functional>
25
26class 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