senf/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_CREATE
  32 #include TIMERFD_H_PATH
  33 #endif
  34 #include "senf/Utils/IgnoreValue.hh"
  35
  36 //#include "TimerSource.mpp"
  37 #define prefix_
  38 //-/////////////////////////////////////////////////////////////////////////////////////////////////
  39
  40 //-/////////////////////////////////////////////////////////////////////////////////////////////////
  41 // senf::scheduler::detail::TimerSource
  42
  43 prefix_ senf::scheduler::detail::TimerSource::~TimerSource()
  44 {}
  45
  46 //-/////////////////////////////////////////////////////////////////////////////////////////////////
  47 // senf::scheduler::detail::POSIXTimerSource
  48
  49 prefix_ senf::scheduler::detail::POSIXTimerSource::POSIXTimerSource()
  50     : timeoutEnabled_ (false), timeout_ (0), signalEnabled_ (false)
  51 {
  52     if (pipe(timerPipe_) < 0)
  53         SENF_THROW_SYSTEM_EXCEPTION("pipe()");
  54     FdManager::instance().set( timerPipe_[0], 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     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     // If the write fails there's not much we can do anyways ...
 130     senf::IGNORE( write(static_cast<POSIXTimerSource*>(siginfo->si_value.sival_ptr)->timerPipe_[1],
 131                         &data, sizeof(data)) );
 132 }
 133
 134 prefix_ void senf::scheduler::detail::POSIXTimerSource::signal(int events)
 135 {
 136     char data;
 137     // This should never fail since we are reading a single character from a signaled
 138     // filedescriptor
 139     senf::IGNORE( read(timerPipe_[0], &data, sizeof(data)) );
 140     timeoutEnabled_ = false;
 141 }
 142
 143 prefix_ void senf::scheduler::detail::POSIXTimerSource::reschedule()
 144 {
 145     struct itimerspec timer;
 146     memset(&timer, 0, sizeof(timer));
 147     if (timeoutEnabled_) {
 148         timer.it_value.tv_sec = ClockService::in_seconds(timeout_);
 149         timer.it_value.tv_nsec = ClockService::in_nanoseconds(
 150             timeout_ - ClockService::seconds(timer.it_value.tv_sec));
 151     }
 152     if (timer_settime(timerId_, TIMER_ABSTIME, &timer, 0)<0)
 153         SENF_THROW_SYSTEM_EXCEPTION("timer_settime()");
 154 }
 155
 156 //-/////////////////////////////////////////////////////////////////////////////////////////////////
 157 // senf::scheduler::detail::PollTimerSource
 158
 159 prefix_ void senf::scheduler::detail::PollTimerSource::timeout(ClockService::clock_type timeout)
 160 {
 161     ClockService::clock_type now (ClockService::now());
 162     int delay (ClockService::in_milliseconds(timeout-now)+1);
 163     IdleEventDispatcher::instance().timeout(delay<0?0:delay);
 164 }
 165
 166 prefix_ void senf::scheduler::detail::PollTimerSource::notimeout()
 167 {
 168     IdleEventDispatcher::instance().timeout(-1);
 169 }
 170
 171 prefix_ void senf::scheduler::detail::PollTimerSource::enable()
 172 {}
 173
 174 prefix_ void senf::scheduler::detail::PollTimerSource::disable()
 175 {}
 176
 177 //-/////////////////////////////////////////////////////////////////////////////////////////////////
 178 // senf::scheduler::detail::TimerFDTimerSource
 179
 180 #ifdef HAVE_TIMERFD_CREATE
 181 prefix_ senf::scheduler::detail::TimerFDTimerSource::TimerFDTimerSource()
 182     : timerfd_ (-1), timeoutEnabled_ (false), timeout_ (0)
 183 {
 184     timerfd_ = timerfd_create(CLOCK_MONOTONIC, 0);
 185     if (timerfd_ < 0)
 186         SENF_THROW_SYSTEM_EXCEPTION("timerfd_create()");
 187     FdManager::instance().set( timerfd_, FdManager::EV_READ, this);
 188 }
 189
 190 prefix_ senf::scheduler::detail::TimerFDTimerSource::~TimerFDTimerSource()
 191 {
 192     FdManager::instance().remove(timerfd_);
 193     close(timerfd_);
 194 }
 195
 196 prefix_ void
 197 senf::scheduler::detail::TimerFDTimerSource::timeout(ClockService::clock_type timeout)
 198 {
 199     if (!timeoutEnabled_ || timeout_ != timeout) {
 200         timeout_ = timeout;
 201         if (timeout_ <= 0)
 202             timeout_ = 1;
 203         timeoutEnabled_ = true;
 204         reschedule();
 205     }
 206 }
 207
 208 prefix_ void senf::scheduler::detail::TimerFDTimerSource::notimeout()
 209 {
 210     if (timeoutEnabled_) {
 211         timeoutEnabled_ = false;
 212         reschedule();
 213     }
 214 }
 215
 216 prefix_ void senf::scheduler::detail::TimerFDTimerSource::enable()
 217 {}
 218
 219 prefix_ void senf::scheduler::detail::TimerFDTimerSource::disable()
 220 {}
 221
 222 namespace {
 223
 224     struct TimerFdCheck
 225     {
 226         TimerFdCheck();
 227         bool timerFdOk;
 228     };
 229
 230     TimerFdCheck::TimerFdCheck()
 231         : timerFdOk (false)
 232     {
 233         int fd (timerfd_create(CLOCK_MONOTONIC, 0));
 234         if (fd == -1) {
 235             if (errno != EINVAL)
 236                 SENF_THROW_SYSTEM_EXCEPTION("timerfd_create()");
 237         }
 238         else {
 239             timerFdOk = true;
 240             close(fd);
 241         }
 242     }
 243
 244 }
 245 prefix_ bool senf::scheduler::detail::TimerFDTimerSource::haveTimerFD()
 246 {
 247     static TimerFdCheck check;
 248     return check.timerFdOk;
 249 }
 250
 251 prefix_ void senf::scheduler::detail::TimerFDTimerSource::signal(int events)
 252 {
 253     uint64_t expirations (0);
 254     // We ignore the return value since we ignore the value read anyways
 255     senf::IGNORE( read(timerfd_, &expirations, sizeof(expirations)) );
 256 }
 257
 258 prefix_ void senf::scheduler::detail::TimerFDTimerSource::reschedule()
 259 {
 260     struct itimerspec timer;
 261     memset(&timer, 0, sizeof(timer));
 262     if (timeoutEnabled_) {
 263         timer.it_value.tv_sec = ClockService::in_seconds(timeout_);
 264         timer.it_value.tv_nsec = ClockService::in_nanoseconds(
 265             timeout_ - ClockService::seconds(timer.it_value.tv_sec));
 266     }
 267     if (timerfd_settime(timerfd_, TFD_TIMER_ABSTIME, &timer, 0)<0)
 268         SENF_THROW_SYSTEM_EXCEPTION("timerfd_settime()");
 269 }
 270 #endif
 271
 272 //-/////////////////////////////////////////////////////////////////////////////////////////////////
 273 #undef prefix_
 274 //#include "TimerSource.mpp"
 275
 276
 277 // Local Variables:
 278 // mode: c++
 279 // fill-column: 100
 280 // comment-column: 40
 281 // c-file-style: "senf"
 282 // indent-tabs-mode: nil
 283 // ispell-local-dictionary: "american"
 284 // compile-command: "scons -u test"
 285 // End: