TAO/tests/NestedUpcall/Simple/server.cpp

   1 #include "server_i.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/Task.h"
   4
   5 static int quiet = 0;
   6 // The test is quiet...
   7
   8 static const ACE_TCHAR *ior_file = ACE_TEXT ("test.ior");
   9 // File of file to which the ior is written
  10
  11 static int number_of_threads = 1;
  12 // Number of server threads.
  13
  14 class Server_Task : public ACE_Task_Base
  15 {
  16 public:
  17   Server_Task (CORBA::ORB_ptr orb);
  18   int svc ();
  19
  20 private:
  21   CORBA::ORB_var orb_;
  22 };
  23
  24 Server_Task::Server_Task (CORBA::ORB_ptr orb)
  25   : orb_ (CORBA::ORB::_duplicate (orb))
  26 {
  27 }
  28
  29 int
  30 Server_Task::svc ()
  31 {
  32   try
  33     {
  34       this->orb_->run ();
  35     }
  36   catch (const CORBA::Exception&)
  37     {
  38       return -1;
  39     }
  40   return 0;
  41 }
  42
  43 static int
  44 write_ior_to_file (const char *ior)
  45 {
  46   FILE *output_file = ACE_OS::fopen (ior_file, "w");
  47
  48   if (output_file == 0)
  49     ACE_ERROR_RETURN ((LM_ERROR, "Cannot open output files for writing IOR: %s\n",
  50                        ior_file),
  51                       -1);
  52
  53   int result = 0;
  54
  55   result = ACE_OS::fprintf (output_file,
  56                             "%s",
  57                             ior);
  58   if (result < 0 ||
  59       static_cast<size_t> (result) != ACE_OS::strlen (ior))
  60     ACE_ERROR_RETURN ((LM_ERROR,
  61                        "ACE_OS::fprintf failed while writing %C to %s\n",
  62                        ior,
  63                        ior_file),
  64                       -1);
  65
  66   ACE_OS::fclose (output_file);
  67
  68   return 0;
  69 }
  70
  71 static int
  72 parse_args (int argc,
  73             ACE_TCHAR **argv)
  74 {
  75   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("q:f:t:"));
  76   int c;
  77
  78   while ((c = get_opts ()) != -1)
  79     switch (c)
  80       {
  81       case 'q':
  82         quiet = 1;
  83         break;
  84
  85       case 't':
  86         number_of_threads = ACE_OS::atoi (get_opts.opt_arg ());
  87         break;
  88
  89       case 'f':
  90         ior_file = get_opts.opt_arg ();
  91         break;
  92
  93       case '?':
  94       default:
  95         ACE_ERROR_RETURN ((LM_ERROR,
  96                            "usage:  %s"
  97                            " [-t number of threads]"
  98                            " [-f ior file]"
  99                            " [-q (quite)]"
 100                            "\n",
 101                            argv[0]),
 102                           -1);
 103       }
 104
 105   return 0;
 106 }
 107
 108 int
 109 ACE_TMAIN (int argc, ACE_TCHAR **argv)
 110 {
 111   try
 112     {
 113       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 114
 115       int result = parse_args (argc, argv);
 116       if (result != 0)
 117         return result;
 118
 119       CORBA::Object_var object = orb->resolve_initial_references ("RootPOA");
 120
 121       PortableServer::POA_var root_poa =
 122         PortableServer::POA::_narrow (object.in ());
 123
 124       PortableServer::POAManager_var poa_manager =
 125         root_poa->the_POAManager ();
 126
 127       poa_manager->activate ();
 128
 129       server_i server_servant (quiet,
 130                                orb.in ());
 131
 132       PortableServer::ObjectId_var id =
 133         root_poa->activate_object (&server_servant);
 134
 135       CORBA::Object_var object_act = root_poa->id_to_reference (id.in ());
 136
 137       server_var server_object = server::_narrow (object_act.in ());
 138
 139       CORBA::String_var ior = orb->object_to_string (server_object.in ());
 140
 141       result = write_ior_to_file (ior.in ());
 142       if (result != 0)
 143         return result;
 144
 145       Server_Task server_tasks (orb.in ());
 146
 147       result = server_tasks.activate (THR_BOUND,
 148                                       number_of_threads);
 149
 150       if (result != 0)
 151         return result;
 152
 153       result = ACE_Thread_Manager::instance ()->wait ();
 154
 155       if (result != 0)
 156         return result;
 157
 158       root_poa->destroy (true, true);
 159     }
 160   catch (const CORBA::Exception& ex)
 161     {
 162       ex._tao_print_exception ("server::main");
 163       return -1;
 164     }
 165
 166   return 0;
 167 }