TAO/tests/TransportCurrent/lib/Client_Request_Interceptor.cpp

   1 // -*- C++ -*-
   2 #include "ace/Log_Msg.h"
   3
   4 #include "Current_TestC.h"
   5 #include "Client_Request_Interceptor.h"
   6 #include "Client_ORBInitializer.h"
   7
   8 namespace Test
   9 {
  10   Client_Request_Interceptor::Client_Request_Interceptor (const char *orb_id,
  11                                                           TEST test)
  12     : orb_id_ (CORBA::string_dup (orb_id))
  13     , request_count_ (0)
  14     , requestID_ (1)
  15     , test_ (test)
  16   {
  17   }
  18
  19   CORBA::Long
  20   Client_Request_Interceptor::interceptions ()
  21   {
  22     return this->request_count_;
  23   }
  24
  25   char *
  26   Client_Request_Interceptor::name ()
  27   {
  28     return CORBA::string_dup ("Client_Request_Interceptor");
  29   }
  30
  31   void
  32   Client_Request_Interceptor::test_transport_current (const ACE_TCHAR* amethod)
  33   {
  34     ACE_DEBUG ((LM_DEBUG,
  35                 ACE_TEXT ("CRI    (%P|%t) Test accessing Transport Current from %s\n"),
  36                 amethod));
  37
  38     ++this->request_count_;
  39
  40     int tmp = 0;
  41     ACE_TCHAR **argv = 0;
  42     CORBA::ORB_var orb = CORBA::ORB_init (tmp,
  43                                           argv,
  44                                           orb_id_.in ());
  45
  46     // Call the test function, which will throw an exception on
  47     // failure
  48     (*this->test_) (orb.in ());
  49
  50     ACE_DEBUG ((LM_DEBUG,
  51                 ACE_TEXT ("CRI    (%P|%t) Successfully tested Transport Current from %s\n"),
  52                 amethod));
  53   }
  54
  55   void
  56   Client_Request_Interceptor::destroy ()
  57   {
  58   }
  59
  60   void
  61   Client_Request_Interceptor::send_request (PortableInterceptor::ClientRequestInfo_ptr ri)
  62   {
  63     // Test TC
  64     test_transport_current (ACE_TEXT ("send_request"));
  65
  66     CORBA::Boolean const response_expected =
  67       ri->response_expected ();
  68
  69     // Oneway?
  70     if (response_expected)
  71       ACE_DEBUG ((LM_DEBUG, ACE_TEXT("CRI    (%P|%t) Sending a two-way\n")));
  72     else
  73       ACE_DEBUG ((LM_DEBUG, ACE_TEXT("CRI    (%P|%t) Sending a one-way\n")));
  74
  75     // Make the context to send the context to the target
  76     IOP::ServiceContext sc;
  77     sc.context_id = Test::Transport::CurrentTest::ContextTag;
  78
  79     // How long can a number really get?
  80     char temp[32];
  81     {
  82       ACE_GUARD (TAO_SYNCH_MUTEX, monitor, this->lock_);
  83
  84       ACE_OS::sprintf (temp, "%ld", this->requestID_);
  85       ++this->requestID_;
  86     }
  87
  88     CORBA::ULong string_len = ACE_OS::strlen (temp) + 1;
  89     CORBA::Octet *buf = CORBA::OctetSeq::allocbuf (string_len);
  90     ACE_OS::strcpy (reinterpret_cast <char *> (buf), temp);
  91
  92     sc.context_data.replace (string_len, string_len, buf, 1);
  93
  94     // Add this context to the service context list.
  95     ri->add_request_service_context (sc, 0);
  96   }
  97
  98   void
  99   Client_Request_Interceptor::send_poll (PortableInterceptor::ClientRequestInfo_ptr)
 100   {
 101     test_transport_current (ACE_TEXT ("send_poll"));
 102   }
 103
 104   void
 105   Client_Request_Interceptor::receive_reply (PortableInterceptor::ClientRequestInfo_ptr)
 106   {
 107     test_transport_current (ACE_TEXT ("receive_reply"));
 108   }
 109
 110   void
 111   Client_Request_Interceptor::receive_exception (PortableInterceptor::ClientRequestInfo_ptr)
 112   {
 113     test_transport_current (ACE_TEXT ("receive_exception"));
 114   }
 115
 116   void
 117   Client_Request_Interceptor::receive_other (PortableInterceptor::ClientRequestInfo_ptr)
 118   {
 119     test_transport_current (ACE_TEXT ("receive_other"));
 120   }
 121 }