TAO/tests/Bug_3567_Regression/server.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    server.cpp
   5  *
   6  *  Implementation of the server.
   7  *
   8  *  @author Alexander Babu Arulanthu <alex@cs.wustl.edu>
   9  *  @author Michael Kircher <Michael.Kircher@mchp.siemens.de>
  10  */
  11 //=============================================================================
  12
  13 #include "ami_test_i.h"
  14 #include "tao/debug.h"
  15 #include "ace/OS_NS_stdio.h"
  16 #include "ace/Get_Opt.h"
  17 #include "ace/Task.h"
  18
  19 const ACE_TCHAR *ior_output_file = 0;
  20 int nthreads = 5;
  21
  22 /**
  23  * Run a server thread
  24  *
  25  * Use the ACE_Task_Base class to run server threads
  26  */
  27 class Worker : public ACE_Task_Base
  28 {
  29 public:
  30   // ctor
  31   Worker (CORBA::ORB_ptr orb);
  32
  33   // The thread entry point.
  34   virtual int svc ();
  35
  36 private:
  37   // The orb
  38   CORBA::ORB_var orb_;
  39 };
  40
  41 int
  42 parse_args (int argc, ACE_TCHAR *argv[])
  43 {
  44   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:d"));
  45   int c;
  46
  47   while ((c = get_opts ()) != -1)
  48     switch (c)
  49       {
  50       case 'o':
  51         ior_output_file = get_opts.opt_arg ();
  52         break;
  53       case 'd':
  54         ++TAO_debug_level;
  55         break;
  56       case '?':
  57       default:
  58         ACE_ERROR_RETURN ((LM_ERROR,
  59                            "usage:  %s "
  60                            "-o <iorfile>"
  61                            "\n",
  62                            argv [0]),
  63                           -1);
  64       }
  65   // Indicates successful parsing of the command line
  66   return 0;
  67 }
  68
  69 int
  70 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  71 {
  72   try
  73     {
  74       CORBA::ORB_var orb =
  75         CORBA::ORB_init (argc, argv);
  76
  77       CORBA::Object_var poa_object =
  78         orb->resolve_initial_references("RootPOA");
  79
  80       if (CORBA::is_nil (poa_object.in ()))
  81         ACE_ERROR_RETURN ((LM_ERROR,
  82                            " (%P|%t) Unable to initialize the POA.\n"),
  83                           1);
  84
  85       PortableServer::POA_var root_poa =
  86         PortableServer::POA::_narrow (poa_object.in ());
  87
  88       PortableServer::POAManager_var poa_manager =
  89         root_poa->the_POAManager ();
  90
  91       if (parse_args (argc, argv) != 0)
  92         return 1;
  93
  94       AMI_Test_i ami_test_i (orb.in ());
  95
  96       PortableServer::ObjectId_var id =
  97         root_poa->activate_object (&ami_test_i);
  98
  99       CORBA::Object_var object = root_poa->id_to_reference (id.in ());
 100
 101       A::AMI_Test_var ami_test_var =
 102         A::AMI_Test::_narrow (object.in ());
 103
 104       CORBA::String_var ior =
 105         orb->object_to_string (ami_test_var.in ());
 106
 107       ACE_DEBUG ((LM_DEBUG, "Activated as <%C>\n", ior.in ()));
 108
 109       // If the ior_output_file exists, output the ior to it
 110       if (ior_output_file != 0)
 111         {
 112           FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
 113           if (output_file == 0)
 114             ACE_ERROR_RETURN ((LM_ERROR,
 115                                "Cannot open output file for writing IOR: %s",
 116                                ior_output_file),
 117                               1);
 118           ACE_OS::fprintf (output_file, "%s", ior.in ());
 119           ACE_OS::fclose (output_file);
 120         }
 121
 122       poa_manager->activate ();
 123
 124       Worker worker (orb.in ());
 125       if (worker.activate (THR_NEW_LWP | THR_JOINABLE,
 126                            nthreads) != 0)
 127         ACE_ERROR_RETURN ((LM_ERROR,
 128                            "Cannot activate client threads\n"),
 129                           1);
 130
 131       worker.thr_mgr ()->wait ();
 132
 133       root_poa->destroy (true,  // ethernalize objects
 134                          false); // wait for completion
 135
 136       orb->destroy ();
 137
 138       ACE_DEBUG ((LM_DEBUG, "event loop finished\n"));
 139     }
 140   catch (const CORBA::Exception& ex)
 141     {
 142       ex._tao_print_exception ("Caught exception:");
 143       return 1;
 144     }
 145
 146   return 0;
 147 }
 148
 149 Worker::Worker (CORBA::ORB_ptr orb)
 150   :  orb_ (CORBA::ORB::_duplicate (orb))
 151 {
 152 }
 153
 154 int
 155 Worker::svc ()
 156 {
 157   try
 158     {
 159       this->orb_->run ();
 160     }
 161   catch (const CORBA::Exception&)
 162     {
 163     }
 164   return 0;
 165 }