TAO/tests/GIOP_Fragments/PMB_With_Fragments/server.cpp

   1 #include "Payload_Receiver.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/OS_NS_stdio.h"
   4
   5 const ACE_TCHAR *ior_output_file = ACE_TEXT("server.ior");
   6
   7 int
   8 parse_args (int argc, ACE_TCHAR *argv[])
   9 {
  10   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  11   int c;
  12
  13   while ((c = get_opts ()) != -1)
  14     switch (c)
  15       {
  16       case 'o':
  17         ior_output_file = get_opts.opt_arg ();
  18         break;
  19       case '?':
  20       default:
  21         ACE_ERROR_RETURN ((LM_ERROR,
  22                            "usage:  %s "
  23                            "-o <iorfile>"
  24                            "\n",
  25                            argv [0]),
  26                           -1);
  27       }
  28   // Indicates successful parsing of the command line
  29   return 0;
  30 }
  31
  32 int
  33 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  34 {
  35   int status = 0;
  36   try
  37     {
  38       CORBA::ORB_var orb =
  39         CORBA::ORB_init (argc, argv);
  40
  41       CORBA::Object_var poa_object =
  42         orb->resolve_initial_references("RootPOA");
  43
  44       if (CORBA::is_nil (poa_object.in ()))
  45         ACE_ERROR_RETURN ((LM_ERROR,
  46                            " (%P|%t) Unable to initialize the POA.\n"),
  47                           1);
  48
  49       PortableServer::POA_var root_poa =
  50         PortableServer::POA::_narrow (poa_object.in ());
  51
  52       PortableServer::POAManager_var poa_manager =
  53         root_poa->the_POAManager ();
  54
  55       if (parse_args (argc, argv) != 0)
  56         return 1;
  57
  58       PortableServer::LifespanPolicy_var lifespan =
  59         root_poa->create_lifespan_policy (PortableServer::PERSISTENT);
  60       CORBA::PolicyList policy_list;
  61       policy_list.length (1);
  62       policy_list[0] = PortableServer::LifespanPolicy::_duplicate (
  63                                                          lifespan.in ());
  64       PortableServer::POA_var persistent_poa =
  65         root_poa->create_POA ("PersistentPOA", poa_manager.in (),
  66                               policy_list);
  67       lifespan->destroy ();
  68
  69       Payload_Receiver *payload_receiver_impl = 0;
  70       ACE_NEW_RETURN (payload_receiver_impl,
  71                       Payload_Receiver (orb.in ()),
  72                       1);
  73       PortableServer::ServantBase_var receiver_owner_transfer(payload_receiver_impl);
  74
  75       PortableServer::ObjectId_var id =
  76         persistent_poa->activate_object (payload_receiver_impl);
  77
  78       CORBA::Object_var obj =
  79         persistent_poa->id_to_reference (id.in ());
  80
  81       CORBA::String_var ior =
  82         orb->object_to_string (obj.in ());
  83
  84       // If the ior_output_file exists, output the ior to it
  85       FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
  86       if (output_file == 0)
  87         ACE_ERROR_RETURN ((LM_ERROR,
  88                            "Cannot open output file for writing IOR: %s",
  89                            ior_output_file),
  90                               1);
  91       ACE_OS::fprintf (output_file, "%s", ior.in ());
  92       ACE_OS::fclose (output_file);
  93
  94       poa_manager->activate ();
  95
  96       orb->run ();
  97
  98       if (payload_receiver_impl->get_count () != 2)
  99         {
 100           ACE_ERROR ((LM_ERROR,
 101                       "ERROR: %d is not the correct "
 102                       "number of calls\n",
 103                       payload_receiver_impl->get_count ()));
 104           ++status;
 105         }
 106
 107       root_poa->destroy (true, true);
 108
 109       orb->destroy ();
 110     }
 111   catch (const CORBA::Exception& ex)
 112     {
 113       ex._tao_print_exception ("Exception caught:");
 114       ++status;
 115     }
 116
 117   return status;
 118 }