TAO/tests/Bug_2935_Regression/source.cpp

   1 # include "source_i.h"
   2
   3 // A ThreeTier client that calls tick and/or tock
   4
   5 const ACE_TCHAR * ior_input_file = 0;
   6
   7 char input_ior[5000];
   8
   9 void eat_args (int & argc, ACE_TCHAR *argv[], int argp, int how_many)
  10 {
  11   for (int marg = argp; marg + how_many < argc; ++marg)
  12     {
  13       argv[marg] = argv[marg + how_many];
  14     }
  15   argc -= how_many;
  16 }
  17
  18 bool parse_args (int & argc, ACE_TCHAR *argv[])
  19 {
  20   int argp = 1;
  21   while (argp < argc)
  22     {
  23       const ACE_TCHAR* arg = argv[argp];
  24       if(arg[0] == '-' && arg[1] == 'f' && argp + 1 < argc)
  25         {
  26           if (ior_input_file != 0)
  27             {
  28               ACE_ERROR ((LM_DEBUG,
  29                           "Sink (%P|%t) duplicate -f options\n"));
  30               return false;
  31             }
  32           // capture input file name
  33           // then remove it from arguemnt list
  34           ior_input_file = argv[argp + 1];
  35           eat_args (argc, argv, argp, 2);
  36         }
  37       else
  38         {
  39            argp += 1;
  40            // just ignore unknown arguments
  41         }
  42     }
  43   if (ior_input_file == 0)
  44     {
  45       ACE_ERROR ((LM_DEBUG,
  46                   "Sink (%P|%t) missing required -f option\n"));
  47       return false;
  48     }
  49   return true;
  50
  51 }
  52
  53
  54 int
  55 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  56 {
  57   int result = 0;
  58   ACE_DEBUG ((LM_DEBUG,
  59               "Source (%P|%t) started\n"));
  60   try
  61     {
  62       // Initialize the ORB.
  63       CORBA::ORB_var orb =
  64         CORBA::ORB_init (argc, argv);
  65
  66       // Initialize options based on command-line arguments.
  67       if (!parse_args (argc, argv))
  68         {
  69           return -1;
  70         }
  71
  72       FILE *input_file = ACE_OS::fopen (ior_input_file, "r");
  73       if (input_file == 0)
  74         {
  75           ACE_ERROR_RETURN ((LM_ERROR,
  76                              "Cannot open input IOR file: %s",
  77                              ior_input_file),
  78                             -1);
  79         }
  80       ACE_OS::fread (input_ior, 1, sizeof(input_ior), input_file);
  81       ACE_OS::fclose (input_file);
  82
  83       // Convert the IOR to an object reference.
  84       CORBA::Object_var object =
  85         orb->string_to_object (input_ior);
  86
  87       // narrow the object reference to a ThreeTier reference
  88       ThreeTier_var server = ThreeTier::_narrow (object.in ());
  89
  90       if (CORBA::is_nil (server.in ()))
  91         {
  92             ACE_ERROR_RETURN ((LM_ERROR,
  93                                "IOR does not refer to a ThreeTier implementation"),
  94                               -1);
  95         }
  96
  97       Source_i source (server.in ());
  98       result = source.run();
  99     }
 100   catch (const CORBA::Exception& ex)
 101     {
 102       ex._tao_print_exception ("Exception caught:");
 103       result = -1;
 104     }
 105
 106   ACE_DEBUG ((LM_DEBUG,
 107               "Source (%P|%t) exits\n"));
 108   return result;
 109 }