ACE/examples/Reactor/WFMO_Reactor/APC.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    APC.cpp
   5  *
   6  *  Tests the WFMO_Reactor's ability to handle regular APC
   7  *  notifications.
   8  *
   9  *  @author Irfan Pyarali <irfan@cs.wustl.edu>
  10  */
  11 //=============================================================================
  12
  13 #include "ace/OS_main.h"
  14
  15 #if defined (ACE_WIN32) && _WIN32_WINNT >= 0x400
  16
  17 #include "ace/Reactor.h"
  18 #include "ace/Auto_Event.h"
  19 #include "ace/Log_Msg.h"
  20
  21 class Event_Handler : public ACE_Event_Handler
  22 {
  23 public:
  24   int handle_signal (int signum, siginfo_t * = 0, ucontext_t * = 0);
  25
  26   int handle_timeout (const ACE_Time_Value &tv, const void *arg = 0);
  27
  28   ACE_Auto_Event handle_;
  29   int iterations_;
  30 };
  31
  32 static Event_Handler *global_event_handler;
  33
  34 static void WINAPI
  35 apc_callback (DWORD)
  36 {
  37   ACE_DEBUG ((LM_DEBUG,
  38               "(%t) apc occurred @ %T\n"));
  39
  40   global_event_handler->handle_.signal ();
  41 }
  42
  43 void
  44 queue_apc ()
  45 {
  46   DWORD result = ::QueueUserAPC (reinterpret_cast<PAPCFUNC> (&apc_callback),
  47                                                          // pointer to APC function
  48                                  ::GetCurrentThread (),  // handle to the thread
  49                                  0);                     // argument for the APC function
  50   if (result == FALSE)
  51     ACE_OS::exit (-1);
  52 }
  53
  54 int
  55 Event_Handler::handle_signal (int, siginfo_t *, ucontext_t *)
  56 {
  57   --this->iterations_;
  58
  59   if (this->iterations_ == 0)
  60     {
  61       ACE_Reactor::instance ()->remove_handler (this->handle_.handle (),
  62                                                 ACE_Event_Handler::DONT_CALL);
  63       ACE_Reactor::end_event_loop ();
  64     }
  65
  66   return 0;
  67 }
  68
  69 int
  70 Event_Handler::handle_timeout (const ACE_Time_Value &,
  71                                const void *)
  72 {
  73   ACE_DEBUG ((LM_DEBUG,
  74               "(%t) timeout occurred @ %T\n"));
  75   queue_apc ();
  76   return 0;
  77 }
  78
  79 int
  80 ACE_TMAIN (int, ACE_TCHAR *[])
  81 {
  82   Event_Handler event_handler;
  83   event_handler.iterations_ = 5;
  84   global_event_handler = &event_handler;
  85
  86   int result = ACE_Reactor::instance ()->register_handler (&event_handler,
  87                                                            event_handler.handle_.handle ());
  88   ACE_TEST_ASSERT (result == 0);
  89
  90   ACE_Time_Value timeout (2);
  91   result = ACE_Reactor::instance ()->schedule_timer (&event_handler,
  92                                                      0,
  93                                                      timeout,
  94                                                      timeout);
  95   ACE_TEST_ASSERT (result != -1);
  96
  97   ACE_Reactor::run_alertable_event_loop ();
  98
  99   ACE_Reactor::instance ()->cancel_timer(&event_handler);
 100
 101   return 0;
 102 }
 103 #else /* !ACE_WIN32 */
 104 int
 105 ACE_TMAIN (int, ACE_TCHAR *[])
 106 {
 107   return 0;
 108 }
 109 #endif /* ACE_WIN32 */