TAO/tests/Exposed_Policies/Policy_Verifier.cpp

   1 #include "Policy_Verifier.h"
   2 #include "ace/OS_NS_string.h"
   3
   4 Policy_Verifier::Policy_Verifier ()
   5   : rt_object_properties_ (0)
   6   , rt_poa_properties_ (0)
   7   , priority_bands_ (0)
   8 {
   9   ACE_OS::strcpy (this->base_object_ref_,
  10                   ACE_TEXT("file://default.ior"));
  11   ACE_OS::strcpy (this->overridden_object_ref_,
  12                   ACE_TEXT("file://overridden.ior"));
  13 }
  14
  15 Policy_Verifier::~Policy_Verifier ()
  16 {
  17   delete rt_object_properties_;
  18   delete rt_poa_properties_;
  19 }
  20
  21 bool
  22 Policy_Verifier::init (int argc, ACE_TCHAR *argv[])
  23 {
  24   this->orb_ =
  25     CORBA::ORB_init (argc, argv);
  26
  27   ACE_Arg_Shifter arg_shifter (argc, argv);
  28
  29   while (arg_shifter.is_anything_left ())
  30     {
  31       const ACE_TCHAR *arg = 0;
  32       // IOR File Name Option.
  33       if (0 != (arg = arg_shifter.get_the_parameter (ACE_TEXT("-POAConfigFile"))))
  34         {
  35           this->rt_poa_properties_ =
  36             RT_Properties::read_from (arg);
  37           this->priority_bands_ =
  38             this->rt_poa_properties_->priority_bands ().length ();
  39         }
  40       else if (0 != (arg = arg_shifter.get_the_parameter (ACE_TEXT("-ObjectConfigFile"))))
  41         {
  42           this->rt_object_properties_ =
  43             RT_Properties::read_from (arg);
  44         }
  45       else if (0 != (arg = arg_shifter.get_the_parameter (ACE_TEXT("-BaseObjectIOR"))))
  46         {
  47           if (this->rt_poa_properties_ == 0)
  48             {
  49               ACE_NEW_THROW_EX (this->rt_poa_properties_,
  50                                 RT_Properties,
  51                                 CORBA::NO_MEMORY (TAO::VMCID,
  52                                                   CORBA::COMPLETED_NO));
  53             }
  54           this->rt_poa_properties_->ior_source (arg);
  55           ACE_OS::strcpy (this->base_object_ref_, ACE_TEXT("file://"));
  56           ACE_OS::strcat (this->base_object_ref_,
  57                           this->rt_poa_properties_->ior_source ());
  58         }
  59       else if (0 != (arg = arg_shifter.get_the_parameter (ACE_TEXT("-OverriddenIOR"))))
  60         {
  61           if (this->rt_object_properties_ == 0)
  62             {
  63               ACE_NEW_THROW_EX (this->rt_object_properties_,
  64                                 RT_Properties,
  65                                 CORBA::NO_MEMORY (TAO::VMCID,
  66                                                   CORBA::COMPLETED_NO));
  67             }
  68           this->rt_object_properties_->ior_source (arg);
  69           ACE_OS::strcpy (this->overridden_object_ref_, ACE_TEXT("file://"));
  70           ACE_OS::strcat (this->overridden_object_ref_,
  71                           this->rt_object_properties_->ior_source ());
  72         }
  73       else
  74         {
  75           arg_shifter.consume_arg ();
  76         }
  77     }
  78
  79   if ((this->rt_poa_properties_ == 0) || (this->rt_object_properties_ == 0))
  80     {
  81       ACE_DEBUG ((LM_DEBUG,
  82                   ACE_TEXT ("Configuration file missing!\n")));
  83       return false;
  84     }
  85
  86   // Get the Object references.
  87   CORBA::Object_var object = this->orb_->string_to_object (this->base_object_ref_);
  88
  89   if (!Policy_Verifier::check_reference (object.in (), "Invalid IOR file!\n"))
  90     return false;
  91
  92   this->base_object_ = Counter::_narrow (object.in ());
  93
  94   if (!Policy_Verifier::check_reference (this->base_object_.in (),
  95                                          "Unable to convert the IOR to the proper object reference.\n"))
  96     return false;
  97
  98   object = this->orb_->string_to_object (this->overridden_object_ref_);
  99
 100   if (!Policy_Verifier::check_reference (object.in (), "Invalid IOR file!\n"))
 101     return false;
 102
 103   this->overridden_object_ = Counter::_narrow (object.in ());
 104
 105   if (!Policy_Verifier::check_reference (this->overridden_object_.in (),
 106                                          "Unable to convert the IOR to the proper object reference.\n"))
 107     return false;
 108
 109   return true;
 110 }
 111
 112 void
 113 Policy_Verifier::run ()
 114 {
 115   this->verify_reference (this->base_object_.in (),
 116                           this->rt_poa_properties_);
 117
 118   this->verify_reference (this->overridden_object_.in (),
 119                           this->rt_object_properties_);
 120 }
 121
 122 void
 123 Policy_Verifier::verify_reference (Counter_ptr object,
 124                                    RT_Properties *rt_properties)
 125 {
 126   try
 127     {
 128       CORBA::Policy_var policy_var =
 129         object->_get_policy (RTCORBA::PRIORITY_MODEL_POLICY_TYPE);
 130
 131       if (Policy_Verifier::check_reference (policy_var.in (), "Unable to get Priority Policy.\n"))
 132         {
 133           RTCORBA::PriorityModelPolicy_var priority_policy =
 134             RTCORBA::PriorityModelPolicy::_narrow (policy_var.in ());
 135
 136           RTCORBA::PriorityModel priority_model =
 137             priority_policy->priority_model ();
 138
 139           RTCORBA::Priority priority =
 140             priority_policy->server_priority ();
 141
 142           if (priority_model == RTCORBA::SERVER_DECLARED)
 143             ACE_DEBUG ((LM_DEBUG,
 144                         ACE_TEXT ("\n\nPriority Model: RTCORBA::SERVER_DECLARED\n")
 145                         ));
 146
 147           ACE_DEBUG ((LM_DEBUG,
 148                       ACE_TEXT ("Priority Model: %d\nCORBA Priority: %d\n\n"),
 149                       priority_model,
 150                       priority
 151                       ));
 152
 153
 154           if (priority != rt_properties->priority ())
 155             ACE_DEBUG ((LM_DEBUG,
 156                         ACE_TEXT ("Priority Value Mismatch.\n")));
 157         }
 158
 159       policy_var = object->_get_policy (RTCORBA::PRIORITY_BANDED_CONNECTION_POLICY_TYPE);
 160
 161       if (Policy_Verifier::check_reference (policy_var.in (),
 162                                           "Unable to get Priority Banded Policy\n"))
 163         {
 164           RTCORBA::PriorityBandedConnectionPolicy_var priority_banded_policy =
 165             RTCORBA::PriorityBandedConnectionPolicy::_narrow (policy_var.in ());
 166
 167
 168           if (Policy_Verifier::check_reference (priority_banded_policy.in (),
 169                                               "Unable to get Priority Banded Policy\n"))
 170             {
 171               // Here we have a priority banded connection policy.
 172
 173               RTCORBA::PriorityBands_var pb =
 174                 priority_banded_policy->priority_bands ();
 175               unsigned int band_num = pb->length ();
 176               if (band_num != this->priority_bands_)
 177                 ACE_DEBUG ((LM_DEBUG,
 178                             ACE_TEXT ("Mismatching Number of Priority Bands!\n")));
 179
 180               for (unsigned int i = 0; i < band_num; ++i)
 181                 {
 182                   ACE_DEBUG ((LM_DEBUG,
 183                               ACE_TEXT ("Priority Band <%d>: (%d, %d)\n"),
 184                               i,
 185                               pb[i].low,
 186                               pb[i].high
 187                               ));
 188
 189
 190                   if ((band_num == rt_properties->priority_bands ().length ()) &&
 191                       ((pb[i].low !=  rt_properties->priority_bands ()[i].low) ||
 192                        (pb[i].high !=  rt_properties->priority_bands ()[i].high)))
 193                     ACE_DEBUG ((LM_DEBUG,
 194                                 ACE_TEXT ("Mismatching Priority Band Range!\n")));
 195                 }
 196             }
 197         }
 198       policy_var = object->_get_policy (RTCORBA::CLIENT_PROTOCOL_POLICY_TYPE);
 199
 200       if (Policy_Verifier::check_reference (policy_var.in (),
 201                                             "Unable to get Client Protocol Policy\n"))
 202         {
 203           RTCORBA::ClientProtocolPolicy_var client_protocol_policy =
 204             RTCORBA::ClientProtocolPolicy::_narrow (policy_var.in ());
 205
 206           RTCORBA::ProtocolList_var protocol_list =
 207             client_protocol_policy->protocols ();
 208
 209           for (unsigned int  i = 0; i < protocol_list->length (); i++)
 210             ACE_DEBUG ((LM_DEBUG,
 211                         ACE_TEXT ("\nThe Client Protocol Type: %d\n"),
 212                         protocol_list[i].protocol_type));
 213         }
 214
 215     }
 216   catch (const CORBA::Exception& ex)
 217     {
 218       ex._tao_print_exception ("CORBA Exception Raised");
 219     }
 220
 221 }
 222
 223 CORBA::Boolean
 224 Policy_Verifier::check_reference (CORBA::Object_ptr object,
 225                                   const char *msg)
 226 {
 227   if (CORBA::is_nil (object))
 228     {
 229       ACE_DEBUG ((LM_DEBUG, msg));
 230       return 0;
 231     }
 232   return 1;
 233 }