TAO/tests/Bug_3068_Regression/server.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/OS_NS_stdio.h"
   3 #include "test_i.h"
   4 #include "ace/Task.h"
   5
   6 #include "tao/Messaging/Messaging.h"
   7 #include "tao/BiDir_GIOP/BiDirGIOP.h"
   8
   9 const ACE_TCHAR *ior_output_file = 0;
  10 int iterations = 10;
  11
  12 int
  13 parse_args (int argc, ACE_TCHAR *argv[])
  14 {
  15   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT ("o:i:"));
  16   int c;
  17
  18   while ((c = get_opts ()) != -1)
  19     switch (c)
  20       {
  21       case 'o':
  22         ior_output_file = get_opts.opt_arg ();
  23         break;
  24       case 'i':
  25         iterations = ACE_OS::atoi (get_opts.opt_arg ());
  26         break;
  27       case '?':
  28       default:
  29         ACE_ERROR_RETURN ((LM_ERROR, ACE_TEXT ("usage: %s -o <iorfile>")
  30                            ACE_TEXT ("-i <iterations>\n"), argv [0]),
  31                           -1);
  32       }
  33   // Indicates successful parsing of the command line
  34   return 0;
  35 }
  36
  37 int
  38 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  39 {
  40   try
  41     {
  42       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  43
  44       CORBA::Object_var object =
  45         orb->resolve_initial_references ("ORBPolicyManager");
  46
  47       CORBA::PolicyManager_var policy_mgr =
  48         CORBA::PolicyManager::_narrow (object.in ());
  49       if (CORBA::is_nil (policy_mgr.in ()))
  50         ACE_ERROR_RETURN ((LM_ERROR, "failed to get policy manager\n"), 1);
  51       CORBA::String_var ref_id = policy_mgr->_interface_repository_id ();
  52       ACE_DEBUG ((LM_DEBUG, "Policy MGR intid = %C\n", ref_id.in ()));
  53
  54       CORBA::Object_var poa_object =
  55         orb->resolve_initial_references ("RootPOA");
  56
  57       if (CORBA::is_nil (poa_object.in ()))
  58         ACE_ERROR_RETURN ((LM_ERROR,
  59                            " (%P|%t) Unable to initialize the POA.\n"),
  60                           1);
  61
  62       PortableServer::POA_var root_poa =
  63         PortableServer::POA::_narrow (poa_object.in ());
  64       PortableServer::POAManager_var poa_manager = root_poa->the_POAManager ();
  65
  66       // Policies for the childPOA to be created.
  67       CORBA::PolicyList policies (1);
  68       policies.length (1);
  69
  70       CORBA::Any pol;
  71       pol <<= BiDirPolicy::BOTH;
  72       policies[0] =
  73         orb->create_policy (BiDirPolicy::BIDIRECTIONAL_POLICY_TYPE, pol);
  74
  75       // Create POA as child of RootPOA with the above policies.  This POA
  76       // will receive request in the same connection in which it sent
  77       // the request
  78       PortableServer::POA_var child_poa =
  79         root_poa->create_POA ("child", poa_manager.in (), policies);
  80
  81       // Creation of childPOA is over. Destroy the Policy objects.
  82       for (CORBA::ULong i = 0; i < policies.length (); ++i)
  83         {
  84           policies[i]->destroy ();
  85         }
  86
  87       poa_manager->activate ();
  88
  89       if (parse_args (argc, argv) != 0)
  90         return 1;
  91
  92       Simple_Server_i server_impl (orb.in (), iterations);
  93
  94       PortableServer::ObjectId_var id =
  95         PortableServer::string_to_ObjectId ("simple_server");
  96
  97       child_poa->activate_object_with_id (id.in (), &server_impl);
  98
  99       CORBA::Object_var obj = child_poa->id_to_reference (id.in ());
 100
 101       CORBA::String_var ior = orb->object_to_string (obj.in ());
 102
 103       ACE_DEBUG ((LM_DEBUG, "(%P|%t) Server activated as <%C>\n", ior.in ()));
 104
 105       // If the ior_output_file exists, output the ior to it
 106       if (ior_output_file != 0)
 107         {
 108           FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
 109           if (output_file == 0)
 110             ACE_ERROR_RETURN ((LM_ERROR,
 111                                "Cannot open output file for writing IOR: %C",
 112                                ior_output_file),
 113                               1);
 114           ACE_OS::fprintf (output_file, "%s", ior.in ());
 115           ACE_OS::fclose (output_file);
 116         }
 117
 118       orb->run();
 119
 120       ACE_Thread_Manager::instance ()->wait ();
 121
 122       root_poa->destroy (1, 1);
 123     }
 124   catch (const CORBA::Exception &excep)
 125     {
 126       excep._tao_print_exception ("Caught exception:");
 127       return 1;
 128     }
 129
 130   return 0;
 131 }