ACE/examples/Service_Configurator/IPC-tests/server/Handle_R_Stream.inl

   1 /* -*- C++ -*- */
   2 #include "ace/Get_Opt.h"
   3 #include "ace/WFMO_Reactor.h"
   4 #include "ace/OS_NS_stdio.h"
   5 #include "ace/OS_NS_string.h"
   6 #include "ace/OS_NS_time.h"
   7 #include "ace/OS_NS_unistd.h"
   8 #include "ace/Truncate.h"
   9
  10 ACE_INLINE
  11 Handle_R_Stream::Handle_R_Stream ()
  12 {
  13   if (Handle_R_Stream::login_name == 0) {
  14 #if !defined(ACE_LACKS_CUSERID)
  15     Handle_R_Stream::login_name = ACE_OS::cuserid (Handle_R_Stream::login);
  16 #else
  17     Handle_R_Stream::login[0] = '.';
  18     Handle_R_Stream::login[1] = '\0';
  19     Handle_R_Stream::login_name = Handle_R_Stream::login;
  20 #endif
  21   }
  22
  23 }
  24
  25 ACE_INLINE int
  26 Handle_R_Stream::open (const ACE_INET_Addr &sia, int async)
  27 {
  28   if (this->ACE_SOCK_Acceptor::open (sia) == -1)
  29     return -1;
  30   else if (async && this->ACE_SOCK_Acceptor::enable (ACE_SIGIO) == -1)
  31     return -1;
  32   else
  33     return 0;
  34 }
  35
  36 ACE_INLINE int
  37 Handle_R_Stream::info (ACE_TCHAR **strp, size_t length) const
  38 {
  39   ACE_TCHAR buf[BUFSIZ];
  40   ACE_INET_Addr sa;
  41
  42   if (this->get_local_addr (sa) == -1)
  43     {
  44       return -1;
  45     }
  46
  47   ACE_OS::sprintf (buf,
  48                    ACE_TEXT("%d/%s %s"),
  49                    sa.get_port_number (),
  50                    ACE_TEXT("tcp"),
  51                    ACE_TEXT("# tests remote stream\n"));
  52
  53   if (*strp == 0 && (*strp = ACE_OS::strdup (buf)) == 0)
  54     {
  55       return -1;
  56     }
  57   else
  58     {
  59       ACE_OS::strncpy (*strp, buf, length);
  60     }
  61
  62   return ACE_Utils::truncate_cast<int> (ACE_OS::strlen (buf));
  63 }
  64
  65 ACE_INLINE int
  66 Handle_R_Stream::init (int argc, ACE_TCHAR *argv[])
  67 {
  68   ACE_INET_Addr sis (Handle_R_Stream::DEFAULT_PORT);
  69   ACE_Get_Opt   get_opt (argc, argv, ACE_TEXT("p:"), 0);
  70
  71   for (int c; (c = get_opt ()) != -1; )
  72      switch (c)
  73        {
  74        case 'p':
  75          sis.set (ACE_OS::atoi (get_opt.opt_arg ()));
  76          break;
  77        default:
  78          break;
  79        }
  80
  81   if (this->open (sis) == -1)
  82     ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "open"), -1);
  83
  84   else if (ACE_Reactor::instance ()->register_handler
  85            (this, ACE_Event_Handler::ACCEPT_MASK) == -1)
  86     ACE_ERROR_RETURN ((LM_ERROR,
  87                        "registering service with ACE_Reactor\n"),
  88                       -1);
  89   return 0;
  90 }
  91
  92 ACE_INLINE int
  93 Handle_R_Stream::fini ()
  94 {
  95   return ACE_Reactor::instance ()->remove_handler
  96     (this, ACE_Event_Handler::ACCEPT_MASK);
  97 }
  98
  99 ACE_INLINE ACE_HANDLE
 100 Handle_R_Stream::get_handle () const
 101 {
 102   return ACE_SOCK_Acceptor::get_handle ();
 103 }
 104
 105 ACE_INLINE int
 106 Handle_R_Stream::handle_input (ACE_HANDLE)
 107 {
 108   char buf[BUFSIZ];
 109   int  bytes;
 110
 111   // Try to find out if the implementation of the reactor that we are
 112   // using requires us to reset the event association for the newly
 113   // created handle. This is because the newly created handle will
 114   // inherit the properties of the listen handle, including its event
 115   // associations.
 116   int reset_new_handle =
 117     ACE_Reactor::instance ()->uses_event_associations ();
 118
 119   if (this->accept (this->new_remote_stream, // stream
 120                     0, // remote address
 121                     0, // timeout
 122                     1, // restart
 123                     reset_new_handle  // reset new handler
 124                     ) == -1)
 125     return -1;
 126   else
 127     ACE_DEBUG ((LM_INFO, "new_remote_stream fd = %d\n",
 128                this->new_remote_stream.get_handle ()));
 129
 130   ACE_INET_Addr sa;
 131
 132   if (this->new_remote_stream.get_remote_addr (sa) == -1)
 133     return -1;
 134
 135   ACE_DEBUG ((LM_INFO,
 136               "accepted from host %s at port %d\n",
 137               sa.get_host_name (),
 138               sa.get_port_number ()));
 139
 140   ACE_OS::puts (ACE_TEXT ("----------------------------------------"));
 141
 142   while ((bytes = this->new_remote_stream.recv (buf, sizeof buf)) > 0)
 143     ACE_OS::write (ACE_STDOUT, buf, bytes);
 144
 145   ACE_OS::puts (ACE_TEXT ("----------------------------------------"));
 146
 147   time_t t = ACE_OS::time (0L);
 148   ACE_TCHAR *cs = ACE_OS::ctime (&t);
 149
 150   if (this->new_remote_stream.send (4,
 151                                     Handle_R_Stream::login_name,
 152                                     ACE_OS::strlen (Handle_R_Stream::login_name),
 153                                     cs,
 154                                     ACE_OS::strlen (cs)) == -1)
 155     return -1;
 156
 157   if (this->new_remote_stream.close () == -1)
 158     return -1;
 159
 160   return 0;
 161 }
 162
 163 ACE_INLINE int
 164 Handle_R_Stream::handle_close (ACE_HANDLE, ACE_Reactor_Mask)
 165 {
 166   return this->ACE_SOCK_Acceptor::close ();
 167 }