654 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			654 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			C++
		
	
	
	
/* timerfd.cc: timerfd helper classes
 | 
						|
 | 
						|
This file is part of Cygwin.
 | 
						|
 | 
						|
This software is a copyrighted work licensed under the terms of the
 | 
						|
Cygwin license.  Please consult the file "CYGWIN_LICENSE" for
 | 
						|
details. */
 | 
						|
 | 
						|
#include "winsup.h"
 | 
						|
#include "path.h"
 | 
						|
#include "fhandler.h"
 | 
						|
#include "pinfo.h"
 | 
						|
#include "dtable.h"
 | 
						|
#include "cygheap.h"
 | 
						|
#include "cygerrno.h"
 | 
						|
#include <sys/timerfd.h>
 | 
						|
#include "timerfd.h"
 | 
						|
 | 
						|
#define TFD_CANCEL_FLAGS (TFD_TIMER_ABSTIME | TFD_TIMER_CANCEL_ON_SET)
 | 
						|
 | 
						|
/* Unfortunately MsgWaitForMultipleObjectsEx does not receive WM_TIMECHANGED
 | 
						|
   messages without a window defined in this process.  Create a hidden window
 | 
						|
   for that purpose. */
 | 
						|
 | 
						|
void
 | 
						|
timerfd_tracker::create_timechange_window ()
 | 
						|
{
 | 
						|
  WNDCLASSW wclass = { 0 };
 | 
						|
  WCHAR cname[NAME_MAX];
 | 
						|
 | 
						|
  __small_swprintf (cname, L"Cygwin.timerfd.%p", this);
 | 
						|
  wclass.lpfnWndProc = DefWindowProcW;
 | 
						|
  wclass.hInstance = user_data->hmodule;
 | 
						|
  wclass.lpszClassName = cname;
 | 
						|
  /* This sleep is required on Windows 10 64 bit only, and only when running
 | 
						|
     under strace.  One of the child processes inheriting the timerfd
 | 
						|
     descriptor will get a STATUS_FLOAT_INEXACT_RESULT exception inside of
 | 
						|
     msvcrt.dll.  While this is completely crazy in itself, it's apparently
 | 
						|
     some timing problem.  It occurs in 4 out of 5 runs under strace only.
 | 
						|
     The sleep is required before calling RegisterClassW.  Moving it before
 | 
						|
     CreateWindowExW does not work.  What the heck? */
 | 
						|
  if (being_debugged ())
 | 
						|
    Sleep (1L);
 | 
						|
  atom = RegisterClassW (&wclass);
 | 
						|
  if (!atom)
 | 
						|
    debug_printf ("RegisterClass %E");
 | 
						|
  else
 | 
						|
    {
 | 
						|
      window = CreateWindowExW (0, cname, cname, WS_POPUP, 0, 0, 0, 0,
 | 
						|
				NULL, NULL, user_data->hmodule, NULL);
 | 
						|
      if (!window)
 | 
						|
	debug_printf ("CreateWindowEx %E");
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
timerfd_tracker::delete_timechange_window ()
 | 
						|
{
 | 
						|
  if (window)
 | 
						|
    DestroyWindow (window);
 | 
						|
  if (atom)
 | 
						|
    UnregisterClassW ((LPWSTR) (uintptr_t) atom, user_data->hmodule);
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
timerfd_tracker::handle_timechange_window ()
 | 
						|
{
 | 
						|
  MSG msg;
 | 
						|
 | 
						|
  while (PeekMessageW (&msg, NULL, 0, 0, PM_REMOVE | PM_QS_POSTMESSAGE)
 | 
						|
	 && msg.message != WM_QUIT)
 | 
						|
    {
 | 
						|
      DispatchMessageW (&msg);
 | 
						|
      if (msg.message == WM_TIMECHANGE
 | 
						|
	  && get_clockid () == CLOCK_REALTIME
 | 
						|
	  && (get_flags () & TFD_CANCEL_FLAGS) == TFD_CANCEL_FLAGS
 | 
						|
	  && enter_critical_section ())
 | 
						|
	{
 | 
						|
	  /* make sure to handle each WM_TIMECHANGE only once! */
 | 
						|
	  if (msg.time != tc_time ())
 | 
						|
	    {
 | 
						|
	      set_expiration_count (-1LL);
 | 
						|
	      disarm_timer ();
 | 
						|
	      timer_expired ();
 | 
						|
	      set_tc_time (msg.time);
 | 
						|
	    }
 | 
						|
	  leave_critical_section ();
 | 
						|
	}
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
/* Like enter_critical_section, but returns -1 on a cancel event. */
 | 
						|
int
 | 
						|
timerfd_tracker::enter_critical_section_cancelable ()
 | 
						|
{
 | 
						|
  HANDLE w[2] = { cancel_evt, _access_mtx };
 | 
						|
  DWORD waitret = WaitForMultipleObjects (2, w, FALSE, INFINITE);
 | 
						|
 | 
						|
  switch (waitret)
 | 
						|
    {
 | 
						|
    case WAIT_OBJECT_0:
 | 
						|
      return -1;
 | 
						|
    case WAIT_OBJECT_0 + 1:
 | 
						|
    case WAIT_ABANDONED_0 + 1:
 | 
						|
      return 1;
 | 
						|
    default:
 | 
						|
      return 0;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
DWORD
 | 
						|
timerfd_tracker::thread_func ()
 | 
						|
{
 | 
						|
  /* Outer loop: Is the timer armed?  If not, wait for it. */
 | 
						|
  HANDLE armed[2] = { arm_evt (),
 | 
						|
		      cancel_evt };
 | 
						|
 | 
						|
  create_timechange_window ();
 | 
						|
  while (1)
 | 
						|
    {
 | 
						|
      switch (MsgWaitForMultipleObjectsEx (2, armed, INFINITE, QS_POSTMESSAGE,
 | 
						|
					   MWMO_INPUTAVAILABLE))
 | 
						|
	{
 | 
						|
	case WAIT_OBJECT_0:
 | 
						|
	  break;
 | 
						|
	case WAIT_OBJECT_0 + 1:
 | 
						|
	  goto canceled;
 | 
						|
	case WAIT_OBJECT_0 + 2:
 | 
						|
	  handle_timechange_window ();
 | 
						|
	  continue;
 | 
						|
	default:
 | 
						|
	  continue;
 | 
						|
	}
 | 
						|
 | 
						|
      /* Inner loop: Timer expired?  If not, wait for it. */
 | 
						|
      HANDLE expired[3] = { timer (),
 | 
						|
			    disarm_evt (),
 | 
						|
			    cancel_evt };
 | 
						|
 | 
						|
      while (1)
 | 
						|
	{
 | 
						|
	  switch (MsgWaitForMultipleObjectsEx (3, expired, INFINITE,
 | 
						|
					       QS_POSTMESSAGE,
 | 
						|
					       MWMO_INPUTAVAILABLE))
 | 
						|
	    {
 | 
						|
	    case WAIT_OBJECT_0:
 | 
						|
	      break;
 | 
						|
	    case WAIT_OBJECT_0 + 1:
 | 
						|
	      goto disarmed;
 | 
						|
	    case WAIT_OBJECT_0 + 2:
 | 
						|
	      goto canceled;
 | 
						|
	    case WAIT_OBJECT_0 + 3:
 | 
						|
	      handle_timechange_window ();
 | 
						|
	      continue;
 | 
						|
	    default:
 | 
						|
	      continue;
 | 
						|
	    }
 | 
						|
 | 
						|
	  int ec = enter_critical_section_cancelable ();
 | 
						|
	  if (ec < 0)
 | 
						|
	    goto canceled;
 | 
						|
	  else if (!ec)
 | 
						|
	    continue;
 | 
						|
	  /* Make sure we haven't been abandoned and/or disarmed
 | 
						|
	     in the meantime */
 | 
						|
	  if (expiration_count () == -1LL
 | 
						|
	      || IsEventSignalled (disarm_evt ()))
 | 
						|
	    {
 | 
						|
	      leave_critical_section ();
 | 
						|
	      goto disarmed;
 | 
						|
	    }
 | 
						|
	  /* One-shot timer? */
 | 
						|
	  if (!get_interval ())
 | 
						|
	    {
 | 
						|
	      /* Set expiration count, disarm timer */
 | 
						|
	      increment_expiration_count (1);
 | 
						|
	      disarm_timer ();
 | 
						|
	    }
 | 
						|
	  else
 | 
						|
	    {
 | 
						|
	      /* Compute expiration count. */
 | 
						|
	      LONG64 now = get_clock_now ();
 | 
						|
	      LONG64 ts = get_exp_ts ();
 | 
						|
	      LONG64 exp_cnt;
 | 
						|
 | 
						|
	      /* Make concessions for unexact realtime clock */
 | 
						|
	      if (ts > now)
 | 
						|
		ts = now - 1;
 | 
						|
	      exp_cnt = (now - ts + get_interval () - 1) / get_interval ();
 | 
						|
	      increment_expiration_count (exp_cnt);
 | 
						|
	      ts += get_interval () * exp_cnt;
 | 
						|
	      /* Set exp_ts to current timestamp.  Make sure exp_ts ends up
 | 
						|
		 bigger than "now" and fix expiration count as required */
 | 
						|
	      while (ts <= (now = get_clock_now ()))
 | 
						|
		{
 | 
						|
		  exp_cnt = (now - ts + get_interval () - 1) / get_interval ();
 | 
						|
		  increment_expiration_count (exp_cnt);
 | 
						|
		  ts += get_interval () * exp_cnt;
 | 
						|
		}
 | 
						|
	      set_exp_ts (ts);
 | 
						|
	      /* NtSetTimer allows periods of up to 24 days only.  If the time
 | 
						|
		 is longer, we set the timer up as one-shot timer for each
 | 
						|
		 interval.  Restart timer here with new due time. */
 | 
						|
	      if (get_interval () > INT_MAX * (NS100PERSEC / MSPERSEC))
 | 
						|
		{
 | 
						|
		  BOOLEAN Resume = (get_clockid () == CLOCK_REALTIME_ALARM
 | 
						|
				    || get_clockid () == CLOCK_BOOTTIME_ALARM);
 | 
						|
		  LARGE_INTEGER DueTime = { QuadPart: -get_interval () };
 | 
						|
 | 
						|
		  NtSetTimer (timer (), &DueTime, NULL, NULL, Resume, 0, NULL);
 | 
						|
		}
 | 
						|
	    }
 | 
						|
	  /* Arm the expiry object */
 | 
						|
	  timer_expired ();
 | 
						|
	  leave_critical_section ();
 | 
						|
	}
 | 
						|
disarmed:
 | 
						|
      ;
 | 
						|
    }
 | 
						|
 | 
						|
canceled:
 | 
						|
  delete_timechange_window ();
 | 
						|
  /* automatically return the cygthread to the cygthread pool */
 | 
						|
  _my_tls._ctinfo->auto_release ();
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
 | 
						|
static DWORD WINAPI
 | 
						|
timerfd_thread (VOID *arg)
 | 
						|
{
 | 
						|
  timerfd_tracker *tt = ((timerfd_tracker *) arg);
 | 
						|
  return tt->thread_func ();
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
timerfd_tracker::create (clockid_t clock_id)
 | 
						|
{
 | 
						|
  int ret;
 | 
						|
  NTSTATUS status;
 | 
						|
  OBJECT_ATTRIBUTES attr;
 | 
						|
 | 
						|
  const ACCESS_MASK access = STANDARD_RIGHTS_REQUIRED
 | 
						|
			     | SECTION_MAP_READ | SECTION_MAP_WRITE;
 | 
						|
  SIZE_T vsize = wincap.page_size ();
 | 
						|
  LARGE_INTEGER sectionsize = { QuadPart: (LONGLONG) wincap.page_size () };
 | 
						|
 | 
						|
  /* Valid clock? */
 | 
						|
  if (!get_clock (clock_id))
 | 
						|
    {
 | 
						|
      ret = -EINVAL;
 | 
						|
      goto err;
 | 
						|
    }
 | 
						|
 | 
						|
  /* Create shared objects */
 | 
						|
  InitializeObjectAttributes (&attr, NULL, OBJ_INHERIT, NULL, NULL);
 | 
						|
  /* Create shared section */
 | 
						|
  status = NtCreateSection (&tfd_shared_hdl, access, &attr, §ionsize,
 | 
						|
			    PAGE_READWRITE, SEC_COMMIT, NULL);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err;
 | 
						|
    }
 | 
						|
  /* Create access mutex */
 | 
						|
  status = NtCreateMutant (&_access_mtx, MUTEX_ALL_ACCESS, &attr, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_tfd_shared_hdl;
 | 
						|
    }
 | 
						|
  /* Create "timer is armed" event, set to "Unsignaled" at creation time */
 | 
						|
  status = NtCreateEvent (&_arm_evt, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_access_mtx;
 | 
						|
    }
 | 
						|
  /* Create "timer is disarmed" event, set to "Signaled" at creation time */
 | 
						|
  status = NtCreateEvent (&_disarm_evt, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, TRUE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_arm_evt;
 | 
						|
    }
 | 
						|
  /* Create timer */
 | 
						|
  status = NtCreateTimer (&_timer, TIMER_ALL_ACCESS, &attr,
 | 
						|
			  SynchronizationTimer);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_disarm_evt;
 | 
						|
    }
 | 
						|
  /* Create "timer expired" semaphore */
 | 
						|
  status = NtCreateEvent (&_expired_evt, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_timer;
 | 
						|
    }
 | 
						|
  /* Create process-local cancel event for this processes timer thread
 | 
						|
     (has to be recreated after fork/exec)*/
 | 
						|
  InitializeObjectAttributes (&attr, NULL, 0, NULL, NULL);
 | 
						|
  status = NtCreateEvent (&cancel_evt, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_expired_evt;
 | 
						|
    }
 | 
						|
  /* Create sync event for this processes timer thread */
 | 
						|
  status = NtCreateEvent (&sync_thr, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_cancel_evt;
 | 
						|
    }
 | 
						|
  /* Create section mapping (has to be recreated after fork/exec) */
 | 
						|
  tfd_shared = NULL;
 | 
						|
  status = NtMapViewOfSection (tfd_shared_hdl, NtCurrentProcess (),
 | 
						|
			       (void **) &tfd_shared, 0, vsize, NULL,
 | 
						|
			       &vsize, ViewShare, 0, PAGE_READWRITE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      ret = -geterrno_from_nt_status (status);
 | 
						|
      goto err_close_sync_thr;
 | 
						|
    }
 | 
						|
  /* Initialize clock id */
 | 
						|
  set_clockid (clock_id);
 | 
						|
  /* Set our winpid for fixup_after_fork_exec */
 | 
						|
  winpid = GetCurrentProcessId ();
 | 
						|
  /* Start timerfd thread */
 | 
						|
  new cygthread (timerfd_thread, this, "timerfd", sync_thr);
 | 
						|
  return 0;
 | 
						|
 | 
						|
err_close_sync_thr:
 | 
						|
  NtClose (sync_thr);
 | 
						|
err_close_cancel_evt:
 | 
						|
  NtClose (cancel_evt);
 | 
						|
err_close_expired_evt:
 | 
						|
  NtClose (_expired_evt);
 | 
						|
err_close_timer:
 | 
						|
  NtClose (_timer);
 | 
						|
err_close_disarm_evt:
 | 
						|
  NtClose (_disarm_evt);
 | 
						|
err_close_arm_evt:
 | 
						|
  NtClose (_arm_evt);
 | 
						|
err_close_access_mtx:
 | 
						|
  NtClose (_access_mtx);
 | 
						|
err_close_tfd_shared_hdl:
 | 
						|
  NtClose (tfd_shared_hdl);
 | 
						|
err:
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
/* Return true if this was the last instance of a timerfd, process-wide,
 | 
						|
   false otherwise.  Basically this is a destructor, but one which may
 | 
						|
   notify the caller NOT to deleted the object. */
 | 
						|
bool
 | 
						|
timerfd_tracker::dtor ()
 | 
						|
{
 | 
						|
  if (!enter_critical_section ())
 | 
						|
    return false;
 | 
						|
  if (decrement_instances () > 0)
 | 
						|
    {
 | 
						|
      leave_critical_section ();
 | 
						|
      return false;
 | 
						|
    }
 | 
						|
  if (cancel_evt)
 | 
						|
    SetEvent (cancel_evt);
 | 
						|
  if (sync_thr)
 | 
						|
    {
 | 
						|
      WaitForSingleObject (sync_thr, INFINITE);
 | 
						|
      NtClose (sync_thr);
 | 
						|
    }
 | 
						|
  leave_critical_section ();
 | 
						|
  if (tfd_shared)
 | 
						|
    NtUnmapViewOfSection (NtCurrentProcess (), tfd_shared);
 | 
						|
  if (cancel_evt)
 | 
						|
    NtClose (cancel_evt);
 | 
						|
  NtClose (tfd_shared_hdl);
 | 
						|
  NtClose (_expired_evt);
 | 
						|
  NtClose (_timer);
 | 
						|
  NtClose (_disarm_evt);
 | 
						|
  NtClose (_arm_evt);
 | 
						|
  NtClose (_access_mtx);
 | 
						|
  return true;
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
timerfd_tracker::dtor (timerfd_tracker *tfd)
 | 
						|
{
 | 
						|
  if (tfd->dtor ())
 | 
						|
    cfree (tfd);
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
timerfd_tracker::ioctl_set_ticks (uint64_t new_exp_cnt)
 | 
						|
{
 | 
						|
  LONG64 exp_cnt = (LONG64) new_exp_cnt;
 | 
						|
  if (exp_cnt == 0 || exp_cnt == -1LL)
 | 
						|
    return -EINVAL;
 | 
						|
  if (!enter_critical_section ())
 | 
						|
    return -EBADF;
 | 
						|
  set_expiration_count (exp_cnt);
 | 
						|
  timer_expired ();
 | 
						|
  leave_critical_section ();
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
timerfd_tracker::init_fixup_after_fork_exec ()
 | 
						|
{
 | 
						|
  /* Run this only if this is the first call, or all previous calls
 | 
						|
     came from close_on_exec descriptors */
 | 
						|
  if (winpid == GetCurrentProcessId ())
 | 
						|
    return;
 | 
						|
  tfd_shared = NULL;
 | 
						|
  cancel_evt = NULL;
 | 
						|
  sync_thr = NULL;
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
timerfd_tracker::fixup_after_fork_exec (bool execing)
 | 
						|
{
 | 
						|
  NTSTATUS status;
 | 
						|
  OBJECT_ATTRIBUTES attr;
 | 
						|
  SIZE_T vsize = wincap.page_size ();
 | 
						|
 | 
						|
  /* Run this only once per process */
 | 
						|
  if (winpid == GetCurrentProcessId ())
 | 
						|
    return;
 | 
						|
  /* Recreate shared section mapping */
 | 
						|
  tfd_shared = NULL;
 | 
						|
  status = NtMapViewOfSection (tfd_shared_hdl, NtCurrentProcess (),
 | 
						|
			       (PVOID *) &tfd_shared, 0, vsize, NULL,
 | 
						|
			       &vsize, ViewShare, 0, PAGE_READWRITE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    api_fatal ("Can't recreate shared timerfd section during %s, status %y!",
 | 
						|
	       execing ? "execve" : "fork", status);
 | 
						|
  /* Create cancel event for this processes timer thread */
 | 
						|
  InitializeObjectAttributes (&attr, NULL, 0, NULL, NULL);
 | 
						|
  status = NtCreateEvent (&cancel_evt, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    api_fatal ("Can't recreate timerfd cancel event during %s, status %y!",
 | 
						|
	       execing ? "execve" : "fork", status);
 | 
						|
  /* Create sync event for this processes timer thread */
 | 
						|
  InitializeObjectAttributes (&attr, NULL, 0, NULL, NULL);
 | 
						|
  status = NtCreateEvent (&sync_thr, EVENT_ALL_ACCESS, &attr,
 | 
						|
			  NotificationEvent, FALSE);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    api_fatal ("Can't recreate timerfd sync event during %s, status %y!",
 | 
						|
	       execing ? "execve" : "fork", status);
 | 
						|
  /* Set winpid so we don't run this twice */
 | 
						|
  winpid = GetCurrentProcessId ();
 | 
						|
  new cygthread (timerfd_thread, this, "timerfd", sync_thr);
 | 
						|
}
 | 
						|
 | 
						|
LONG64
 | 
						|
timerfd_tracker::wait (bool nonblocking)
 | 
						|
{
 | 
						|
  HANDLE w4[2] = { get_timerfd_handle (), NULL };
 | 
						|
  LONG64 ret;
 | 
						|
 | 
						|
  wait_signal_arrived here (w4[1]);
 | 
						|
repeat:
 | 
						|
  switch (WaitForMultipleObjects (2, w4, FALSE, nonblocking ? 0 : INFINITE))
 | 
						|
    {
 | 
						|
    case WAIT_OBJECT_0:		/* timer event */
 | 
						|
      if (!enter_critical_section ())
 | 
						|
	ret = -EIO;
 | 
						|
      else
 | 
						|
	{
 | 
						|
	  ret = read_and_reset_expiration_count ();
 | 
						|
	  leave_critical_section ();
 | 
						|
	  switch (ret)
 | 
						|
	    {
 | 
						|
	    case -1:	/* TFD_TIMER_CANCEL_ON_SET */
 | 
						|
	      ret = -ECANCELED;
 | 
						|
	      break;
 | 
						|
	    case 0:	/* Another read was quicker. */
 | 
						|
	      if (!nonblocking)
 | 
						|
		goto repeat;
 | 
						|
	      ret = -EAGAIN;
 | 
						|
	      break;
 | 
						|
	    default:	/* Return (positive) expiration count. */
 | 
						|
	      if (ret < 0)
 | 
						|
		ret = INT64_MAX;
 | 
						|
	      break;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
      break;
 | 
						|
    case WAIT_OBJECT_0 + 1:	/* signal */
 | 
						|
      if (_my_tls.call_signal_handler ())
 | 
						|
        goto repeat;
 | 
						|
      ret = -EINTR;
 | 
						|
      break;
 | 
						|
    case WAIT_TIMEOUT:
 | 
						|
      ret = -EAGAIN;
 | 
						|
      break;
 | 
						|
    default:
 | 
						|
      ret = -geterrno_from_win_error ();
 | 
						|
      break;
 | 
						|
    }
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
timerfd_tracker::gettime (struct itimerspec *curr_value)
 | 
						|
{
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  __try
 | 
						|
    {
 | 
						|
      if (!enter_critical_section ())
 | 
						|
	{
 | 
						|
	  ret = -EBADF;
 | 
						|
	  __leave;
 | 
						|
	}
 | 
						|
    }
 | 
						|
  __except (NO_ERROR)
 | 
						|
    {
 | 
						|
      return -EFAULT;
 | 
						|
    }
 | 
						|
  __endtry
 | 
						|
 | 
						|
  __try
 | 
						|
    {
 | 
						|
      if (IsEventSignalled (disarm_evt ()))
 | 
						|
	*curr_value = time_spec ();
 | 
						|
      else
 | 
						|
	{
 | 
						|
	  LONG64 next_relative_exp = get_exp_ts () - get_clock_now ();
 | 
						|
	  curr_value->it_value.tv_sec = next_relative_exp / NS100PERSEC;
 | 
						|
	  next_relative_exp -= curr_value->it_value.tv_sec * NS100PERSEC;
 | 
						|
	  curr_value->it_value.tv_nsec = next_relative_exp
 | 
						|
					 * (NSPERSEC / NS100PERSEC);
 | 
						|
	  curr_value->it_interval = time_spec ().it_interval;
 | 
						|
	}
 | 
						|
      ret = 0;
 | 
						|
    }
 | 
						|
  __except (NO_ERROR)
 | 
						|
    {
 | 
						|
      ret = -EFAULT;
 | 
						|
    }
 | 
						|
  __endtry
 | 
						|
  leave_critical_section ();
 | 
						|
  return ret;
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
timerfd_tracker::arm_timer (int flags, const struct itimerspec *new_value)
 | 
						|
{
 | 
						|
  LONG64 interval;
 | 
						|
  LONG64 ts;
 | 
						|
  NTSTATUS status;
 | 
						|
  LARGE_INTEGER DueTime;
 | 
						|
  BOOLEAN Resume;
 | 
						|
  LONG Period;
 | 
						|
 | 
						|
  ResetEvent (disarm_evt ());
 | 
						|
 | 
						|
  /* Convert incoming itimerspec into 100ns interval and timestamp */
 | 
						|
  interval = new_value->it_interval.tv_sec * NS100PERSEC
 | 
						|
	     + (new_value->it_interval.tv_nsec + (NSPERSEC / NS100PERSEC) - 1)
 | 
						|
	       / (NSPERSEC / NS100PERSEC);
 | 
						|
  ts = new_value->it_value.tv_sec * NS100PERSEC
 | 
						|
	    + (new_value->it_value.tv_nsec + (NSPERSEC / NS100PERSEC) - 1)
 | 
						|
	      / (NSPERSEC / NS100PERSEC);
 | 
						|
  set_flags (flags);
 | 
						|
  if (flags & TFD_TIMER_ABSTIME)
 | 
						|
    {
 | 
						|
      if (get_clockid () == CLOCK_REALTIME)
 | 
						|
	DueTime.QuadPart = ts + FACTOR;
 | 
						|
      else /* non-REALTIME clocks require relative DueTime. */
 | 
						|
	{
 | 
						|
	  DueTime.QuadPart = get_clock_now () - ts;
 | 
						|
	  /* If the timestamp was earlier than now, compute number
 | 
						|
	     of expirations and offset DueTime to expire immediately. */
 | 
						|
	  if (DueTime.QuadPart >= 0)
 | 
						|
	    DueTime.QuadPart = -1LL;
 | 
						|
	}
 | 
						|
    }
 | 
						|
  else
 | 
						|
    {
 | 
						|
      /* Keep relative timestamps relative for the timer, but store the
 | 
						|
         expiry timestamp absolute for the timer thread. */
 | 
						|
      DueTime.QuadPart = -ts;
 | 
						|
      ts += get_clock_now ();
 | 
						|
    }
 | 
						|
  time_spec () = *new_value;
 | 
						|
  set_exp_ts (ts);
 | 
						|
  set_interval (interval);
 | 
						|
  read_and_reset_expiration_count ();
 | 
						|
  /* Note: Advanced Power Settings -> Sleep -> Allow Wake Timers
 | 
						|
	   since W10 1709 */
 | 
						|
  Resume = (get_clockid () == CLOCK_REALTIME_ALARM
 | 
						|
	    || get_clockid () == CLOCK_BOOTTIME_ALARM);
 | 
						|
  if (interval > INT_MAX * (NS100PERSEC / MSPERSEC))
 | 
						|
    Period = 0;
 | 
						|
  else
 | 
						|
    Period = (interval + (NS100PERSEC / MSPERSEC) - 1)
 | 
						|
	     / (NS100PERSEC / MSPERSEC);
 | 
						|
  status = NtSetTimer (timer (), &DueTime, NULL, NULL, Resume, Period, NULL);
 | 
						|
  if (!NT_SUCCESS (status))
 | 
						|
    {
 | 
						|
      disarm_timer ();
 | 
						|
      return -geterrno_from_nt_status (status);
 | 
						|
    }
 | 
						|
 | 
						|
  SetEvent (arm_evt ());
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
timerfd_tracker::settime (int flags, const struct itimerspec *new_value,
 | 
						|
			  struct itimerspec *old_value)
 | 
						|
{
 | 
						|
  int ret = 0;
 | 
						|
 | 
						|
  __try
 | 
						|
    {
 | 
						|
      if (!valid_timespec (new_value->it_value)
 | 
						|
          || !valid_timespec (new_value->it_interval))
 | 
						|
	{
 | 
						|
	  ret = -EINVAL;
 | 
						|
	  __leave;
 | 
						|
	}
 | 
						|
      if (!enter_critical_section ())
 | 
						|
	{
 | 
						|
	  ret = -EBADF;
 | 
						|
	  __leave;
 | 
						|
	}
 | 
						|
      if (old_value && (ret = gettime (old_value)) < 0)
 | 
						|
	__leave;
 | 
						|
      if (new_value->it_value.tv_sec == 0 && new_value->it_value.tv_nsec == 0)
 | 
						|
	ret = disarm_timer ();
 | 
						|
      else
 | 
						|
	ret = arm_timer (flags, new_value);
 | 
						|
      leave_critical_section ();
 | 
						|
      ret = 0;
 | 
						|
    }
 | 
						|
  __except (NO_ERROR)
 | 
						|
    {
 | 
						|
      ret = -EFAULT;
 | 
						|
    }
 | 
						|
  __endtry
 | 
						|
  return ret;
 | 
						|
}
 |