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