ACE/tests/Reactor_Exceptions_Test.cpp

   1 //=============================================================================
   2 /**
   3  *  @file    Reactor_Exceptions_Test.cpp
   4  *
   5  *    This is a test that makes sure the <ACE_Reactor> works
   6  *    correctly in the face of C++ exceptions and threads.
   7  *
   8  *  @author Luca Priorelli <lucapri@mbox.vol.it> and Douglas C. Schmidt <d.schmidt@vanderbilt.edu>
   9  */
  10 //=============================================================================
  11
  12 #include "test_config.h"
  13 #include "ace/Reactor.h"
  14 #include "ace/SOCK_Dgram.h"
  15 #include "ace/INET_Addr.h"
  16 #include "ace/Thread_Manager.h"
  17 #include "ace/Select_Reactor.h"
  18
  19 // Just need a simple exception class.
  20 class Except {};
  21
  22 static void
  23 throw_exception ()
  24 {
  25   ACE_DEBUG ((LM_DEBUG,
  26               ACE_TEXT ("(%P|%t) throw exception\n")));
  27   throw Except ();
  28 }
  29
  30 class My_Handler : public ACE_Event_Handler, public ACE_SOCK_Dgram
  31 {
  32 public:
  33   My_Handler (const ACE_INET_Addr &local_addr);
  34
  35   ACE_HANDLE get_handle () const override;
  36   int handle_input (ACE_HANDLE handle) override;
  37 };
  38
  39 My_Handler::My_Handler (const ACE_INET_Addr &local_addr)
  40   : ACE_SOCK_Dgram (local_addr)
  41 {
  42 }
  43
  44 ACE_HANDLE
  45 My_Handler::get_handle () const
  46 {
  47   return ACE_SOCK_Dgram::get_handle ();
  48 }
  49
  50 int
  51 My_Handler::handle_input (ACE_HANDLE)
  52 {
  53   ACE_TCHAR buf[BUFSIZ];
  54   ACE_INET_Addr from_addr;
  55
  56   ACE_DEBUG ((LM_DEBUG,
  57               ACE_TEXT ("Activity occurred on handle %d!\n"),
  58               ACE_SOCK_Dgram::get_handle ()));
  59
  60   ssize_t n = ACE_SOCK_Dgram::recv (buf,
  61                                     sizeof buf,
  62                                     from_addr);
  63   if (n == -1)
  64     ACE_ERROR ((LM_ERROR,
  65                 ACE_TEXT ("%p\n"),
  66                 ACE_TEXT ("handle_input")));
  67   else
  68     ACE_DEBUG ((LM_DEBUG,
  69                 ACE_TEXT ("got buf = %s\n"),
  70                 buf));
  71
  72   throw_exception ();
  73   return 0;
  74 }
  75
  76 class My_Reactor : public ACE_Select_Reactor
  77 {
  78 public:
  79   int handle_events (ACE_Time_Value *max_wait_time) override
  80   {
  81     int ret = 0;
  82
  83     try
  84       {
  85         ret = ACE_Select_Reactor::handle_events (max_wait_time);
  86       }
  87     catch (...)
  88       {
  89         ACE_DEBUG ((LM_DEBUG,
  90                     ACE_TEXT (" (%t) catch exception\n")));
  91         ret = -1;
  92         // do your thing, etc.
  93       }
  94     return ret;
  95   }
  96
  97   int handle_events (ACE_Time_Value &max_wait_time) override
  98   {
  99     return this->handle_events (&max_wait_time);
 100   }
 101 };
 102
 103 #if defined (ACE_HAS_THREADS)
 104 static void*
 105 worker (void*)
 106 {
 107   ACE_Reactor::instance ()->owner (ACE_OS::thr_self ());
 108
 109   for (;;)
 110     if (ACE_Reactor::instance ()->handle_events () == -1)
 111       {
 112         ACE_DEBUG ((LM_DEBUG, ACE_TEXT (" (%t) exception return\n")));
 113         break;
 114       }
 115
 116   return 0;
 117 }
 118 #endif /* ACE_HAS_THREADS */
 119
 120 int
 121 run_main (int argc, ACE_TCHAR *argv[])
 122 {
 123   ACE_START_TEST (ACE_TEXT ("Reactor_Exceptions_Test"));
 124
 125   ACE_DEBUG ((LM_DEBUG,
 126               ACE_TEXT ("Starting tracing\n")));
 127
 128   u_short port = argc > 1 ? ACE_OS::atoi (argv[1]) : ACE_DEFAULT_SERVER_PORT;
 129
 130   ACE_INET_Addr local_addr (port);
 131   ACE_INET_Addr remote_addr (port, ACE_LOCALHOST, PF_INET);
 132   ACE_Reactor reactor (new My_Reactor, true);
 133
 134   ACE_Reactor::instance (&reactor);
 135   ACE_Thread_Manager *thr_mgr = ACE_Thread_Manager::instance ();
 136
 137   {
 138     // Make sure handler gets cleaned up before reactor by putting it in its
 139     // own scope
 140     My_Handler handler (local_addr);
 141
 142     if (ACE_Reactor::instance ()->register_handler
 143         (&handler,
 144          ACE_Event_Handler::READ_MASK) == -1)
 145       ACE_ERROR_RETURN ((LM_ERROR,
 146                          ACE_TEXT ("%p\n"),
 147                          ACE_TEXT ("register_handler")),
 148                         -1);
 149
 150 #if defined (ACE_HAS_THREADS)
 151     thr_mgr->spawn (ACE_THR_FUNC (worker));
 152 #else
 153     // Need to figure out how to implement this test.
 154     ACE_ERROR ((LM_INFO,
 155                 ACE_TEXT ("threads not supported on this platform\n")));
 156 #endif /* ACE_HAS_THREADS */
 157
 158     ACE_SOCK_Dgram dgram (ACE_sap_any_cast (ACE_INET_Addr &), PF_INET);
 159
 160     for (size_t i = 0; i < ACE_MAX_ITERATIONS; i++)
 161       dgram.send (ACE_TEXT ("Hello"),
 162                   sizeof (ACE_TEXT ("Hello")),
 163                   remote_addr);
 164     // Barrier to wait for the other thread to return.
 165     thr_mgr->wait ();
 166
 167     handler.close ();
 168     dgram.close ();
 169   }
 170
 171   ACE_DEBUG ((LM_DEBUG,
 172               ACE_TEXT (" (%t) exiting main\n")));
 173
 174   ACE_END_TEST;
 175   return 0;
 176 }