Scheduler/TimerSource.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2009
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief TimerSource non-inline non-template implementation */
  25
  26 #include "TimerSource.hh"
  27 //#include "TimerSource.ih"
  28
  29 // Custom includes
  30 #include "IdleEvent.hh"
  31 #ifdef HAVE_TIMERFD
  32 #include <sys/timerfd.h>
  33 #endif
  34
  35 //#include "TimerSource.mpp"
  36 #define prefix_
  37 ///////////////////////////////cc.p////////////////////////////////////////
  38
  39 ///////////////////////////////////////////////////////////////////////////
  40 // senf::scheduler::detail::TimerSource
  41
  42 prefix_ senf::scheduler::detail::TimerSource::~TimerSource()
  43 {}
  44
  45 ///////////////////////////////////////////////////////////////////////////
  46 // senf::scheduler::detail::POSIXTimerSource
  47
  48 prefix_ senf::scheduler::detail::POSIXTimerSource::POSIXTimerSource()
  49     : timeoutEnabled_ (false), timeout_ (0), signalEnabled_ (false)
  50 {
  51     if (pipe(timerPipe_) < 0)
  52         SENF_THROW_SYSTEM_EXCEPTION("pipe()");
  53     senf::scheduler::detail::FdManager::instance().set(
  54         timerPipe_[0], detail::FdManager::EV_READ, this);
  55
  56     sigemptyset(&sigSet_);
  57     sigaddset(&sigSet_, SIGALRM);
  58     sigprocmask(SIG_BLOCK, &sigSet_, 0);
  59
  60     struct sigaction act;
  61     act.sa_sigaction = &sigHandler;
  62     act.sa_mask = sigSet_;
  63     act.sa_flags = SA_SIGINFO | SA_RESTART;
  64     if (sigaction(SIGALRM, &act, 0) < 0)
  65         SENF_THROW_SYSTEM_EXCEPTION("sigaction()");
  66
  67     struct sigevent ev;
  68     ::memset(&ev, 0, sizeof(ev));
  69     ev.sigev_notify = SIGEV_SIGNAL;
  70     ev.sigev_signo = SIGALRM;
  71     ev.sigev_value.sival_ptr = this;
  72     if (timer_create(CLOCK_MONOTONIC, &ev, &timerId_) < 0)
  73         SENF_THROW_SYSTEM_EXCEPTION("timer_create()");
  74 }
  75
  76 prefix_ senf::scheduler::detail::POSIXTimerSource::~POSIXTimerSource()
  77 {
  78     timer_delete(timerId_);
  79     ::signal(SIGALRM, SIG_IGN);
  80     sigprocmask(SIG_UNBLOCK, &sigSet_, 0);
  81     senf::scheduler::detail::FdManager::instance().remove(timerPipe_[0]);
  82     close(timerPipe_[0]);
  83     close(timerPipe_[1]);
  84 }
  85
  86 prefix_ void
  87 senf::scheduler::detail::POSIXTimerSource::timeout(ClockService::clock_type timeout)
  88 {
  89     if (! timeoutEnabled_ || timeout_ != timeout) {
  90         timeout_ = timeout;
  91         if (timeout_ <= 0)
  92             timeout_ = 1;
  93         timeoutEnabled_ = true;
  94         reschedule();
  95     }
  96 }
  97
  98 prefix_ void senf::scheduler::detail::POSIXTimerSource::notimeout()
  99 {
 100     if (timeoutEnabled_) {
 101         timeoutEnabled_ = false;
 102         reschedule();
 103     }
 104 }
 105
 106 prefix_ void senf::scheduler::detail::POSIXTimerSource::enable()
 107 {
 108     if (! signalEnabled_) {
 109         signalEnabled_ = true;
 110         sigprocmask(SIG_UNBLOCK, &sigSet_, 0);
 111     }
 112 }
 113
 114 prefix_ void senf::scheduler::detail::POSIXTimerSource::disable()
 115 {
 116     if (signalEnabled_) {
 117         signalEnabled_ = false;
 118         sigprocmask(SIG_BLOCK, &sigSet_, 0);
 119     }
 120 }
 121
 122 prefix_ void senf::scheduler::detail::POSIXTimerSource::sigHandler(int,
 123                                                                    ::siginfo_t * siginfo,
 124                                                                    void *)
 125 {
 126     if (siginfo->si_value.sival_ptr == 0)
 127         return;
 128     static char data = '\xD0';
 129     write(static_cast<POSIXTimerSource*>(siginfo->si_value.sival_ptr)->timerPipe_[1],
 130           &data, sizeof(data));
 131 }
 132
 133 prefix_ void senf::scheduler::detail::POSIXTimerSource::signal(int events)
 134 {
 135     char data;
 136     read(timerPipe_[0], &data, sizeof(data));
 137     timeoutEnabled_ = false;
 138 }
 139
 140 prefix_ void senf::scheduler::detail::POSIXTimerSource::reschedule()
 141 {
 142     struct itimerspec timer;
 143     memset(&timer, 0, sizeof(timer));
 144     if (timeoutEnabled_) {
 145         timer.it_value.tv_sec = ClockService::in_seconds(timeout_);
 146         timer.it_value.tv_nsec = ClockService::in_nanoseconds(
 147             timeout_ - ClockService::seconds(timer.it_value.tv_sec));
 148     }
 149     if (timer_settime(timerId_, TIMER_ABSTIME, &timer, 0)<0)
 150         SENF_THROW_SYSTEM_EXCEPTION("timer_settime()");
 151 }
 152
 153 ///////////////////////////////////////////////////////////////////////////
 154 // senf::scheduler::detail::PollTimerSource
 155
 156 prefix_ void senf::scheduler::detail::PollTimerSource::timeout(ClockService::clock_type timeout)
 157 {
 158     ClockService::clock_type now (ClockService::now());
 159     int delay (ClockService::in_milliseconds(timeout-now)+1);
 160     IdleEventDispatcher::instance().timeout(delay<0?0:delay);
 161 }
 162
 163 prefix_ void senf::scheduler::detail::PollTimerSource::notimeout()
 164 {
 165     IdleEventDispatcher::instance().timeout(-1);
 166 }
 167
 168 prefix_ void senf::scheduler::detail::PollTimerSource::enable()
 169 {}
 170
 171 prefix_ void senf::scheduler::detail::PollTimerSource::disable()
 172 {}
 173
 174 ///////////////////////////////////////////////////////////////////////////
 175 // senf::scheduler::detail::TimerFDTimerSource
 176
 177 #ifdef HAVE_TIMERFD
 178 prefix_ senf::scheduler::detail::TimerFDTimerSource::TimerFDTimerSource()
 179     : timerfd_ (-1), timeoutEnabled_ (false), timeout_ (0)
 180 {
 181     timerfd_ = timerfd_create(CLOCK_MONOTONIC, 0);
 182     if (timerfd_ < 0)
 183         SENF_THROW_SYSTEM_EXCEPTION("timerfd_create()");
 184     senf::scheduler::detail::FdManager::instance().set(
 185         timerfd_, detail::FdManager::EV_READ, this);
 186 }
 187
 188 prefix_ senf::scheduler::detail::TimerFDTimerSource::~TimerFDTimerSource()
 189 {
 190     senf::scheduler::detail::FdManager::instance().remove(timerfd_);
 191     close(timerfd_);
 192 }
 193
 194 prefix_ void
 195 senf::scheduler::detail::TimerFDTimerSource::timeout(ClockService::clock_type timeout)
 196 {
 197     if (!timeoutEnabled_ || timeout_ != timeout) {
 198         timeout_ = timeout;
 199         if (timeout_ <= 0)
 200             timeout_ = 1;
 201         timeoutEnabled_ = true;
 202         reschedule();
 203     }
 204 }
 205
 206 prefix_ void senf::scheduler::detail::TimerFDTimerSource::notimeout()
 207 {
 208     if (timeoutEnabled_) {
 209         timeoutEnabled_ = false;
 210         reschedule();
 211     }
 212 }
 213
 214 prefix_ void senf::scheduler::detail::TimerFDTimerSource::enable()
 215 {}
 216
 217 prefix_ void senf::scheduler::detail::TimerFDTimerSource::disable()
 218 {}
 219
 220 namespace {
 221
 222     struct TimerFdCheck
 223     {
 224         TimerFdCheck();
 225         bool timerFdOk;
 226     };
 227
 228     TimerFdCheck::TimerFdCheck()
 229         : timerFdOk (false)
 230     {
 231         int fd (timerfd_create(CLOCK_MONOTONIC, 0));
 232         if (fd == -1) {
 233             if (errno != EINVAL)
 234                 SENF_THROW_SYSTEM_EXCEPTION("timerfd_create()");
 235         }
 236         else {
 237             timerFdOk = true;
 238             close(fd);
 239         }
 240     }
 241
 242 }
 243 prefix_ bool senf::scheduler::detail::TimerFDTimerSource::haveTimerFD()
 244 {
 245     static TimerFdCheck check;
 246     return check.timerFdOk;
 247 }
 248
 249 prefix_ void senf::scheduler::detail::TimerFDTimerSource::signal(int events)
 250 {
 251     uint64_t expirations (0);
 252     read(timerfd_, &expirations, sizeof(expirations));
 253 }
 254
 255 prefix_ void senf::scheduler::detail::TimerFDTimerSource::reschedule()
 256 {
 257     struct itimerspec timer;
 258     memset(&timer, 0, sizeof(timer));
 259     if (timeoutEnabled_) {
 260         timer.it_value.tv_sec = ClockService::in_seconds(timeout_);
 261         timer.it_value.tv_nsec = ClockService::in_nanoseconds(
 262             timeout_ - ClockService::seconds(timer.it_value.tv_sec));
 263     }
 264     if (timerfd_settime(timerfd_, TFD_TIMER_ABSTIME, &timer, 0)<0)
 265         SENF_THROW_SYSTEM_EXCEPTION("timerfd_settime()");
 266 }
 267 #endif
 268
 269 ///////////////////////////////cc.e////////////////////////////////////////
 270 #undef prefix_
 271 //#include "TimerSource.mpp"
 272
 273 \f
 274 // Local Variables:
 275 // mode: c++
 276 // fill-column: 100
 277 // comment-column: 40
 278 // c-file-style: "senf"
 279 // indent-tabs-mode: nil
 280 // ispell-local-dictionary: "american"
 281 // compile-command: "scons -u test"
 282 // End: