external/bsd/bind/dist/unit/atf-src/atf-run/timer.cpp

   1 //
   2 // Automated Testing Framework (atf)
   3 //
   4 // Copyright (c) 2010 The NetBSD Foundation, Inc.
   5 // All rights reserved.
   6 //
   7 // Redistribution and use in source and binary forms, with or without
   8 // modification, are permitted provided that the following conditions
   9 // are met:
  10 // 1. Redistributions of source code must retain the above copyright
  11 //    notice, this list of conditions and the following disclaimer.
  12 // 2. Redistributions in binary form must reproduce the above copyright
  13 //    notice, this list of conditions and the following disclaimer in the
  14 //    documentation and/or other materials provided with the distribution.
  15 //
  16 // THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
  17 // CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
  18 // INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  19 // MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  20 // IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
  21 // DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22 // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  23 // GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  25 // IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  26 // OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
  27 // IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28 //
  29
  30 #if defined(HAVE_CONFIG_H)
  31 #   include <bconfig.h>
  32 #endif
  33
  34 extern "C" {
  35 #include <sys/time.h>
  36 }
  37
  38 #include <cerrno>
  39 #include <csignal>
  40 #include <ctime>
  41
  42 extern "C" {
  43 #include "atf-c/defs.h"
  44 }
  45
  46 #include "atf-c++/detail/exceptions.hpp"
  47 #include "atf-c++/detail/sanity.hpp"
  48
  49 #include "signals.hpp"
  50 #include "timer.hpp"
  51
  52 namespace impl = atf::atf_run;
  53 #define IMPL_NAME "atf::atf_run"
  54
  55 #if !defined(HAVE_TIMER_T)
  56 static impl::timer* compat_handle;
  57 #endif
  58
  59 // ------------------------------------------------------------------------
  60 // Auxiliary functions.
  61 // ------------------------------------------------------------------------
  62
  63 #if defined(HAVE_TIMER_T)
  64 static
  65 void
  66 handler(const int signo ATF_DEFS_ATTRIBUTE_UNUSED, siginfo_t* si,
  67         void* uc ATF_DEFS_ATTRIBUTE_UNUSED)
  68 {
  69     impl::timer* timer = static_cast< impl::timer* >(si->si_value.sival_ptr);
  70     timer->set_fired();
  71     timer->timeout_callback();
  72 }
  73 #else
  74 static
  75 void
  76 handler(const int signo ATF_DEFS_ATTRIBUTE_UNUSED,
  77         siginfo_t* si ATF_DEFS_ATTRIBUTE_UNUSED,
  78         void* uc ATF_DEFS_ATTRIBUTE_UNUSED)
  79 {
  80     compat_handle->set_fired();
  81     compat_handle->timeout_callback();
  82 }
  83 #endif
  84
  85 // ------------------------------------------------------------------------
  86 // The "timer" class.
  87 // ------------------------------------------------------------------------
  88
  89 struct impl::timer::impl {
  90 #if defined(HAVE_TIMER_T)
  91     ::timer_t m_timer;
  92     ::itimerspec m_old_it;
  93 #else
  94     ::itimerval m_old_it;
  95 #endif
  96
  97     struct ::sigaction m_old_sa;
  98     volatile bool m_fired;
  99
 100     impl(void) : m_fired(false)
 101     {
 102     }
 103 };
 104
 105 impl::timer::timer(const unsigned int seconds) :
 106     m_pimpl(new impl())
 107 {
 108     struct ::sigaction sa;
 109     sigemptyset(&sa.sa_mask);
 110     sa.sa_flags = SA_SIGINFO;
 111     sa.sa_sigaction = ::handler;
 112     if (::sigaction(SIGALRM, &sa, &m_pimpl->m_old_sa) == -1)
 113         throw system_error(IMPL_NAME "::timer::timer",
 114                            "Failed to set signal handler", errno);
 115
 116 #if defined(HAVE_TIMER_T)
 117     struct ::sigevent se;
 118     se.sigev_notify = SIGEV_SIGNAL;
 119     se.sigev_signo = SIGALRM;
 120     se.sigev_value.sival_ptr = static_cast< void* >(this);
 121     se.sigev_notify_function = NULL;
 122     se.sigev_notify_attributes = NULL;
 123     if (::timer_create(CLOCK_MONOTONIC, &se, &m_pimpl->m_timer) == -1) {
 124         ::sigaction(SIGALRM, &m_pimpl->m_old_sa, NULL);
 125         throw system_error(IMPL_NAME "::timer::timer",
 126                            "Failed to create timer", errno);
 127     }
 128
 129     struct ::itimerspec it;
 130     it.it_interval.tv_sec = 0;
 131     it.it_interval.tv_nsec = 0;
 132     it.it_value.tv_sec = seconds;
 133     it.it_value.tv_nsec = 0;
 134     if (::timer_settime(m_pimpl->m_timer, 0, &it, &m_pimpl->m_old_it) == -1) {
 135        ::sigaction(SIGALRM, &m_pimpl->m_old_sa, NULL);
 136        ::timer_delete(m_pimpl->m_timer);
 137         throw system_error(IMPL_NAME "::timer::timer",
 138                            "Failed to program timer", errno);
 139     }
 140 #else
 141     ::itimerval it;
 142     it.it_interval.tv_sec = 0;
 143     it.it_interval.tv_usec = 0;
 144     it.it_value.tv_sec = seconds;
 145     it.it_value.tv_usec = 0;
 146     if (::setitimer(ITIMER_REAL, &it, &m_pimpl->m_old_it) == -1) {
 147         ::sigaction(SIGALRM, &m_pimpl->m_old_sa, NULL);
 148         throw system_error(IMPL_NAME "::timer::timer",
 149                            "Failed to program timer", errno);
 150     }
 151     INV(compat_handle == NULL);
 152     compat_handle = this;
 153 #endif
 154 }
 155
 156 impl::timer::~timer(void)
 157 {
 158 #if defined(HAVE_TIMER_T)
 159     {
 160         const int ret = ::timer_delete(m_pimpl->m_timer);
 161         INV(ret != -1);
 162     }
 163 #else
 164     {
 165         const int ret = ::setitimer(ITIMER_REAL, &m_pimpl->m_old_it, NULL);
 166         INV(ret != -1);
 167     }
 168 #endif
 169     const int ret = ::sigaction(SIGALRM, &m_pimpl->m_old_sa, NULL);
 170     INV(ret != -1);
 171
 172 #if !defined(HAVE_TIMER_T)
 173     compat_handle = NULL;
 174 #endif
 175 }
 176
 177 bool
 178 impl::timer::fired(void)
 179     const
 180 {
 181     return m_pimpl->m_fired;
 182 }
 183
 184 void
 185 impl::timer::set_fired(void)
 186 {
 187     m_pimpl->m_fired = true;
 188 }
 189
 190 // ------------------------------------------------------------------------
 191 // The "child_timer" class.
 192 // ------------------------------------------------------------------------
 193
 194 impl::child_timer::child_timer(const unsigned int seconds, const pid_t pid,
 195                                volatile bool& terminate) :
 196     timer(seconds),
 197     m_pid(pid),
 198     m_terminate(terminate)
 199 {
 200 }
 201
 202 impl::child_timer::~child_timer(void)
 203 {
 204 }
 205
 206 void
 207 impl::child_timer::timeout_callback(void)
 208 {
 209     static const timespec ts = { 1, 0 };
 210     m_terminate = true;
 211     ::kill(-m_pid, SIGTERM);
 212     ::nanosleep(&ts, NULL);
 213     if (::kill(-m_pid, 0) != -1)
 214        ::kill(-m_pid, SIGKILL);
 215 }