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 int
  54 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  55 {
  56   int result = 0;
  57   ACE_DEBUG ((LM_DEBUG,
  58               "Source (%P|%t) started\n"));
  59   try
  60     {
  61       // Initialize the ORB.
  62       CORBA::ORB_var orb =
  63         CORBA::ORB_init (argc, argv);
  64
  65       // Initialize options based on command-line arguments.
  66       if (!parse_args (argc, argv))
  67         {
  68           return -1;
  69         }
  70
  71       FILE *input_file = ACE_OS::fopen (ior_input_file, "r");
  72       if (input_file == 0)
  73         {
  74           ACE_ERROR_RETURN ((LM_ERROR,
  75                              "Cannot open input IOR file: %s",
  76                              ior_input_file),
  77                             -1);
  78         }
  79       ACE_OS::fread (input_ior, 1, sizeof(input_ior), input_file);
  80       ACE_OS::fclose (input_file);
  81
  82       // Convert the IOR to an object reference.
  83       CORBA::Object_var object =
  84         orb->string_to_object (input_ior);
  85
  86       // narrow the object reference to a ThreeTier reference
  87       ThreeTier_var server = ThreeTier::_narrow (object.in ());
  88
  89       if (CORBA::is_nil (server.in ()))
  90         {
  91             ACE_ERROR_RETURN ((LM_ERROR,
  92                                "IOR does not refer to a ThreeTier implementation"),
  93                               -1);
  94         }
  95
  96       Source_i source (server.in ());
  97       result = source.run();
  98     }
  99   catch (const CORBA::Exception& ex)
 100     {
 101       ex._tao_print_exception ("Exception caught:");
 102       result = -1;
 103     }
 104
 105   ACE_DEBUG ((LM_DEBUG,
 106               "Source (%P|%t) exits\n"));
 107   return result;
 108 }