TAO/tests/Portable_Interceptors/AMI/Client_Interceptor.cpp

   1 /**
   2  * @file Client_Interceptor.cpp
   3  *
   4  * @author Carlos O'Ryan <coryan@atdesk.com>
   5  */
   6
   7 #include "Client_Interceptor.h"
   8 #include "Shared_Interceptor.h"
   9 #include "tao/OctetSeqC.h"
  10
  11 #include "ace/Log_Msg.h"
  12 #include "ace/OS_NS_string.h"
  13
  14 unsigned long Echo_Client_Request_Interceptor::request_count = 0;
  15 unsigned long Echo_Client_Request_Interceptor::reply_count = 0;
  16 unsigned long Echo_Client_Request_Interceptor::other_count = 0;
  17 unsigned long Echo_Client_Request_Interceptor::exception_count = 0;
  18
  19 Echo_Client_Request_Interceptor::
  20 Echo_Client_Request_Interceptor ()
  21 {
  22 }
  23
  24 char *
  25 Echo_Client_Request_Interceptor::name ()
  26 {
  27   return CORBA::string_dup ("Echo_Client_Interceptor");
  28 }
  29
  30 void
  31 Echo_Client_Request_Interceptor::destroy ()
  32 {
  33 }
  34
  35 void
  36 Echo_Client_Request_Interceptor::send_poll (
  37     PortableInterceptor::ClientRequestInfo_ptr)
  38 {
  39   ACE_ERROR((LM_ERROR,
  40              "ERROR, unexpected interception point called send_poll()\n"));
  41   throw CORBA::BAD_PARAM ();
  42 }
  43
  44 void
  45 Echo_Client_Request_Interceptor::send_request (
  46     PortableInterceptor::ClientRequestInfo_ptr ri)
  47 {
  48   IOP::ServiceContext sc;
  49   sc.context_id = ::service_id;
  50
  51   CORBA::Octet *buf = CORBA::OctetSeq::allocbuf(magic_cookie_len);
  52   ACE_OS::memcpy(buf, magic_cookie, magic_cookie_len);
  53   sc.context_data.replace (magic_cookie_len, magic_cookie_len, buf, 1);
  54
  55   // Add this context to the service context list.
  56   ri->add_request_service_context (sc,
  57                                    0);
  58
  59   // Check that the request service context can be retrieved.
  60   IOP::ServiceContext_var rc =
  61     ri->get_request_service_context (::service_id);
  62
  63   if (rc->context_data.length() != magic_cookie_len
  64       || ACE_OS::memcmp(
  65              magic_cookie, rc->context_data.get_buffer(),
  66              magic_cookie_len) != 0
  67       )
  68     {
  69       throw CORBA::BAD_PARAM();
  70     }
  71
  72   Echo_Client_Request_Interceptor::request_count++;
  73 }
  74
  75 void
  76 Echo_Client_Request_Interceptor::receive_reply (
  77     PortableInterceptor::ClientRequestInfo_ptr ri)
  78 {
  79   // Check that the request service context can be retrieved.
  80   IOP::ServiceContext_var rc =
  81     ri->get_request_service_context (::service_id);
  82
  83   if (rc->context_data.length() != magic_cookie_len
  84       || ACE_OS::memcmp(
  85              magic_cookie, rc->context_data.get_buffer(),
  86              magic_cookie_len) != 0
  87       )
  88     {
  89       throw CORBA::BAD_PARAM();
  90     }
  91
  92   ++Echo_Client_Request_Interceptor::reply_count;
  93 }
  94
  95 void
  96 Echo_Client_Request_Interceptor::receive_other (
  97     PortableInterceptor::ClientRequestInfo_ptr ri)
  98 {
  99   // Check that the request service context can be retrieved.
 100   IOP::ServiceContext_var rc =
 101     ri->get_request_service_context (::service_id);
 102
 103   if (rc->context_data.length() != magic_cookie_len
 104       || ACE_OS::memcmp(
 105              magic_cookie, rc->context_data.get_buffer(),
 106              magic_cookie_len) != 0
 107       )
 108     {
 109       throw CORBA::BAD_PARAM ();
 110     }
 111
 112   Echo_Client_Request_Interceptor::other_count++;
 113 }
 114
 115 void
 116 Echo_Client_Request_Interceptor::receive_exception (
 117     PortableInterceptor::ClientRequestInfo_ptr)
 118 {
 119   Echo_Client_Request_Interceptor::exception_count++;
 120 }