TAO/tests/OBV/TC_Alignment/client.cpp

   1 #include "testC.h"
   2 #include "ace/Get_Opt.h"
   3
   4 static const ACE_TCHAR *ior_input_file = ACE_TEXT("file://test.ior");
   5
   6 static obvmod::tcOffset1 *vt1 = 0;
   7 static obvmod::tcOffset003 *vt3 = 0;
   8 static obvmod::tcOffset00005 *vt5 = 0;
   9 static obvmod::tcOffset0000007 *vt7 = 0;
  10
  11 int debug = 0;
  12 int do_shutdown = 0;
  13
  14 int
  15 register_factories (CORBA::ORB_ptr orb)
  16 {
  17   // Create and register factories.
  18
  19   obvmod::tcOffset1_init *factory1 = 0;
  20   ACE_NEW_RETURN (factory1,
  21                   obvmod::tcOffset1_init,
  22                   -1);
  23
  24   orb->register_value_factory (factory1->tao_repository_id (),
  25                                factory1);
  26
  27   factory1->_remove_ref ();
  28
  29   obvmod::tcOffset003_init *factory3 = 0;
  30   ACE_NEW_RETURN (factory3,
  31                   obvmod::tcOffset003_init,
  32                   -1);
  33
  34   orb->register_value_factory (factory3->tao_repository_id (),
  35                                factory3);
  36
  37   factory3->_remove_ref ();
  38
  39   obvmod::tcOffset00005_init *factory5 = 0;
  40   ACE_NEW_RETURN (factory5,
  41                   obvmod::tcOffset00005_init,
  42                   -1);
  43
  44   orb->register_value_factory (factory5->tao_repository_id (),
  45                                factory5);
  46
  47   factory5->_remove_ref ();
  48
  49   obvmod::tcOffset0000007_init *factory7 = 0;
  50   ACE_NEW_RETURN (factory7,
  51                   obvmod::tcOffset0000007_init,
  52                   -1);
  53
  54   orb->register_value_factory (factory7->tao_repository_id (),
  55                                factory7);
  56
  57   factory7->_remove_ref ();
  58   return 0;
  59 }
  60
  61 int
  62 run_test (test_ptr objref,
  63           CORBA::Long offset)
  64 {
  65   CORBA::Any_var result = objref->get_value (offset);
  66
  67   CORBA::Boolean status = 0;
  68   CORBA::Long member_value = 0;
  69
  70   switch (offset)
  71   {
  72     case 1:
  73       status = (result.in () >>= vt1);
  74       member_value = vt1->mem ();
  75       break;
  76     case 3:
  77       status = (result.in () >>= vt3);
  78       member_value = vt3->mem ();
  79       break;
  80     case 5:
  81       status = (result.in () >>= vt5);
  82       member_value = vt5->mem ();
  83       break;
  84     case 7:
  85       status = (result.in () >>= vt7);
  86       member_value = vt7->mem ();
  87       break;
  88     default:
  89       return -1;
  90   }
  91
  92   if (status == 0)
  93     {
  94       if (debug)
  95         {
  96           ACE_DEBUG ((LM_DEBUG,
  97                       "Any extraction failed for repo id length offset"
  98                       " of %d from 8-byte boundary\n",
  99                       offset));
 100         }
 101
 102       return -1;
 103     }
 104
 105   return (member_value == offset);
 106 }
 107
 108 int
 109 parse_args (int argc, ACE_TCHAR *argv[])
 110 {
 111   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("dk:x"));
 112   int c;
 113
 114   while ((c = get_opts ()) != -1)
 115     switch (c)
 116       {
 117       case 'd':
 118         debug = 1;
 119         break;
 120
 121       case 'k':
 122         ior_input_file = get_opts.optarg;
 123         break;
 124
 125       case 'x':
 126         do_shutdown = 1;
 127         break;
 128
 129       case '?':
 130       default:
 131         ACE_ERROR_RETURN ((LM_ERROR,
 132                            "usage:  %s "
 133                            "-d "
 134                            "-k <ior> "
 135                            "-x "
 136                            "\n",
 137                            argv [0]),
 138                           -1);
 139       }
 140   // Indicates successful parsing of the command line
 141   return 0;
 142 }
 143
 144 int
 145 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 146 {
 147   try
 148     {
 149       CORBA::ORB_var orb = CORBA::ORB_init (argc,
 150                                             argv);
 151
 152       if (parse_args (argc, argv) != 0)
 153         {
 154           return 1;
 155         }
 156
 157       CORBA::Object_var obj =
 158         orb->string_to_object (ior_input_file);
 159
 160       int status = register_factories (orb.in ());
 161
 162       if (status != 0)
 163         {
 164           if (debug)
 165             {
 166               ACE_DEBUG ((LM_DEBUG,
 167                           "OBV factory registration failed\n"));
 168             }
 169
 170           return -1;
 171         }
 172
 173       test_var objref = test::_narrow (obj.in ());
 174
 175       for (CORBA::Long i = 1; i < 8; i += 2)
 176         {
 177           status = run_test (objref.in (), i);
 178
 179           if (status == 0)
 180             {
 181               if (debug)
 182                 {
 183                   ACE_DEBUG ((LM_DEBUG,
 184                               "Incorrect member value - expected %d\n",
 185                               i));
 186                 }
 187
 188               return -1;
 189             }
 190         }
 191
 192       if (do_shutdown)
 193         {
 194           objref->shutdown ();
 195         }
 196
 197       orb->destroy ();
 198     }
 199   catch (const CORBA::Exception& ex)
 200     {
 201       ex._tao_print_exception ("exception:");
 202       return 1;
 203     }
 204
 205   return 0;
 206 }