ghc/rts/Itimer.c

   1 /* -----------------------------------------------------------------------------
   2  * $Id: Itimer.c,v 1.24 2001/11/13 13:38:02 simonmar Exp $
   3  *
   4  * (c) The GHC Team, 1995-1999
   5  *
   6  * Interval timer for profiling and pre-emptive scheduling.
   7  *
   8  * ---------------------------------------------------------------------------*/
   9
  10 /*
  11  * The interval timer is used for profiling and for context switching in the
  12  * threaded build.  Though POSIX 1003.1b includes a standard interface for
  13  * such things, no one really seems to be implementing them yet.  Even
  14  * Solaris 2.3 only seems to provide support for @CLOCK_REAL@, whereas we're
  15  * keen on getting access to @CLOCK_VIRTUAL@.
  16  *
  17  * Hence, we use the old-fashioned @setitimer@ that just about everyone seems
  18  * to support.  So much for standards.
  19  */
  20
  21 /* This is not posix compliant. */
  22 /* #include "PosixSource.h" */
  23
  24 #include "Rts.h"
  25 #include "RtsFlags.h"
  26 #include "Itimer.h"
  27 #include "Proftimer.h"
  28 #include "Schedule.h"
  29
  30 /* As recommended in the autoconf manual */
  31 # ifdef TIME_WITH_SYS_TIME
  32 #  include <sys/time.h>
  33 #  include <time.h>
  34 # else
  35 #  ifdef HAVE_SYS_TIME_H
  36 #   include <sys/time.h>
  37 #  else
  38 #   include <time.h>
  39 #  endif
  40 # endif
  41
  42 #if HAVE_WINDOWS_H
  43 # include <windows.h>
  44 #endif
  45
  46 lnat total_ticks = 0;
  47
  48 /* ticks left before next pre-emptive context switch */
  49 int ticks_to_ctxt_switch = 0;
  50
  51 /* -----------------------------------------------------------------------------
  52    Tick handler
  53
  54    We use the ticker for time profiling.
  55
  56    SMP note: this signal could be delivered to *any* thread.  We have
  57    to ensure that it doesn't matter which thread actually runs the
  58    signal handler.
  59    -------------------------------------------------------------------------- */
  60
  61 static
  62 void
  63 #if defined(mingw32_TARGET_OS) || (defined(cygwin32_TARGET_OS) && !defined(HAVE_SETITIMER))
  64 CALLBACK
  65 handle_tick(UINT uID STG_UNUSED, UINT uMsg STG_UNUSED, DWORD dwUser STG_UNUSED,
  66             DWORD dw1 STG_UNUSED, DWORD d STG_UNUSED)
  67 #else
  68 handle_tick(int unused STG_UNUSED)
  69 #endif
  70 {
  71   total_ticks++;
  72
  73 #ifdef PROFILING
  74   handleProfTick();
  75 #endif
  76
  77   if (RtsFlags.ConcFlags.ctxtSwitchTicks > 0) {
  78       ticks_to_ctxt_switch--;
  79       if (ticks_to_ctxt_switch <= 0) {
  80           ticks_to_ctxt_switch = RtsFlags.ConcFlags.ctxtSwitchTicks;
  81           context_switch = 1;   /* schedule a context switch */
  82       }
  83   }
  84 }
  85
  86
  87 /*
  88  * Handling timer events under cygwin32 is not done with signal/setitimer.
  89  * Instead of the two steps of first registering a signal handler to handle
  90  * \tr{SIGVTALRM} and then start generating them via @setitimer()@, we use
  91  * the Multimedia API (MM) and its @timeSetEvent@. (Internally, the MM API
  92  * creates a separate thread that will notify the main thread of timer
  93  * expiry). -- SOF 7/96
  94  *
  95  * 11/98: if the cygwin DLL supports setitimer(), then use it instead.
  96  */
  97
  98 #if defined(mingw32_TARGET_OS) || (defined(cygwin32_TARGET_OS) && !defined(HAVE_SETITIMER))
  99
 100 LPTIMECALLBACK vtalrm_cback;
 101
 102 nat
 103 initialize_virtual_timer(nat ms)
 104 {
 105   /* On Win32 setups that don't have support for
 106      setitimer(), we use the MultiMedia API's timer
 107      support.
 108
 109      The delivery of ticks isn't free; the performance hit should be checked.
 110   */
 111   unsigned int delay;
 112   static unsigned int vtalrm_id;
 113
 114   if (ms) {
 115     delay = timeBeginPeriod(1);
 116     if (delay == TIMERR_NOCANDO) { /* error of some sort. */
 117       return delay;
 118     }
 119     vtalrm_id =
 120       timeSetEvent(ms,      /* event every `delay' milliseconds. */
 121                    1,       /* precision is within 1 ms */
 122                    vtalrm_cback,
 123                    TIME_CALLBACK_FUNCTION,     /* ordinary callback */
 124                    TIME_PERIODIC);
 125   } else {
 126     timeKillEvent(vtalrm_id);
 127     timeEndPeriod(1);
 128   }
 129
 130   return 0;
 131 }
 132
 133 #else
 134
 135 nat
 136 initialize_virtual_timer(nat ms)
 137 {
 138 # ifndef HAVE_SETITIMER
 139   /*    fprintf(stderr, "No virtual timer on this system\n"); */
 140     return -1;
 141 # else
 142     struct itimerval it;
 143
 144     timestamp = getourtimeofday();
 145
 146     it.it_value.tv_sec = ms / 1000;
 147     it.it_value.tv_usec = 1000 * (ms - (1000 * it.it_value.tv_sec));
 148     it.it_interval = it.it_value;
 149     return (setitimer(ITIMER_VIRTUAL, &it, NULL));
 150 # endif
 151 }
 152
 153 #endif /* !cygwin32_TARGET_OS */
 154
 155 # if 0
 156 /* This is a potential POSIX version */
 157 nat
 158 initialize_virtual_timer(nat ms)
 159 {
 160     struct sigevent se;
 161     struct itimerspec it;
 162     timer_t tid;
 163
 164     timestamp = getourtimeofday();
 165
 166     se.sigev_notify = SIGEV_SIGNAL;
 167     se.sigev_signo = SIGVTALRM;
 168     se.sigev_value.sival_int = SIGVTALRM;
 169     if (timer_create(CLOCK_VIRTUAL, &se, &tid)) {
 170         barf("can't create virtual timer");
 171     }
 172     it.it_value.tv_sec = ms / 1000;
 173     it.it_value.tv_nsec = 1000000 * (ms - 1000 * it.it_value.tv_sec);
 174     it.it_interval = it.it_value;
 175     timer_settime(tid, TIMER_RELTIME, &it, NULL);
 176 }
 177 # endif
 178
 179 #if defined(mingw32_TARGET_OS) || (defined(cygwin32_TARGET_OS) && !defined(HAVE_SETITIMER))
 180 int
 181 install_vtalrm_handler(void)
 182 {
 183   vtalrm_cback = handle_tick;
 184   return 0;
 185 }
 186
 187 #else
 188 int
 189 install_vtalrm_handler(void)
 190 {
 191     struct sigaction action;
 192
 193     action.sa_handler = handle_tick;
 194
 195     sigemptyset(&action.sa_mask);
 196     action.sa_flags = 0;
 197
 198     return sigaction(SIGVTALRM, &action, NULL);
 199 }
 200
 201 void
 202 block_vtalrm_signal(void)
 203 {
 204     sigset_t signals;
 205
 206     sigemptyset(&signals);
 207     sigaddset(&signals, SIGVTALRM);
 208
 209     (void) sigprocmask(SIG_BLOCK, &signals, NULL);
 210 }
 211
 212 void
 213 unblock_vtalrm_signal(void)
 214 {
 215     sigset_t signals;
 216
 217     sigemptyset(&signals);
 218     sigaddset(&signals, SIGVTALRM);
 219
 220     (void) sigprocmask(SIG_UNBLOCK, &signals, NULL);
 221 }
 222 #endif
 223
 224 /* gettimeofday() takes around 1us on our 500MHz PIII.  Since we're
 225  * only calling it 50 times/s, it shouldn't have any great impact.
 226  */
 227 #if !defined(mingw32_TARGET_OS)
 228 unsigned int
 229 getourtimeofday(void)
 230 {
 231   struct timeval tv;
 232   gettimeofday(&tv, (struct timezone *) NULL);
 233   return (tv.tv_sec * TICK_FREQUENCY +
 234           tv.tv_usec * TICK_FREQUENCY / 1000000);
 235 }
 236 #else
 237 unsigned int
 238 getourtimeofday(void)
 239 {
 240   return ((unsigned int)GetTickCount() * TICK_FREQUENCY) / 1000;
 241 }
 242 #endif