TAO/orbsvcs/tests/Security/MT_IIOP_SSL/server.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "test_i.h"
   3 #include "Server_Worker.h"
   4
   5 const ACE_TCHAR *ior_output_file = 0;
   6 int nthreads = 4;
   7
   8 int
   9 parse_args (int argc, ACE_TCHAR *argv[])
  10 {
  11   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:n:"));
  12   int c;
  13
  14   while ((c = get_opts ()) != -1)
  15     switch (c)
  16       {
  17       case 'o':
  18         ior_output_file = get_opts.opt_arg ();
  19         break;
  20
  21       case 'n':
  22         nthreads = ACE_OS::atoi (get_opts.opt_arg ());
  23         break;
  24
  25       case '?':
  26       default:
  27         ACE_ERROR_RETURN ((LM_ERROR,
  28                            "usage:  %s "
  29                            "-o <iorfile>"
  30                            "\n",
  31                            argv [0]),
  32                           -1);
  33       }
  34   // Indicates successful parsing of the command line
  35   return 0;
  36 }
  37
  38
  39 int
  40 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  41 {
  42   try
  43     {
  44       // Initialize the ORB
  45       CORBA::ORB_var orb =
  46         CORBA::ORB_init (argc, argv);
  47
  48       CORBA::Object_var poa_object =
  49         orb->resolve_initial_references("RootPOA");
  50
  51       if (CORBA::is_nil (poa_object.in ()))
  52         ACE_ERROR_RETURN ((LM_ERROR,
  53                            " (%P|%t) Unable to initialize the POA.\n"),
  54                            1);
  55
  56       // Get a Root POA reference
  57       PortableServer::POA_var root_poa =
  58         PortableServer::POA::_narrow (poa_object.in ());
  59
  60       PortableServer::POAManager_var poa_manager =
  61         root_poa->the_POAManager ();
  62
  63       if (parse_args (argc, argv) != 0)
  64         return 1;
  65
  66       Simple_Server_i server_impl (orb.in ());
  67
  68       Simple_Server_var server =
  69         server_impl._this ();
  70
  71       CORBA::String_var ior =
  72         orb->object_to_string (server.in ());
  73
  74       ACE_DEBUG ((LM_DEBUG, "Activated as <%C>\n", ior.in ()));
  75
  76       // If the ior_output_file exists, output the ior to it
  77       if (ior_output_file != 0)
  78         {
  79           FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
  80           if (output_file == 0)
  81             ACE_ERROR_RETURN ((LM_ERROR,
  82                                "Cannot open output file for writing IOR: %s",
  83                                ior_output_file),
  84                               1);
  85           ACE_OS::fprintf (output_file, "%s", ior.in ());
  86           ACE_OS::fclose (output_file);
  87         }
  88
  89       poa_manager->activate ();
  90
  91       Server_Worker worker (orb.in ());
  92       if (worker.activate (THR_NEW_LWP | THR_JOINABLE,
  93                            nthreads) != 0)
  94         ACE_ERROR_RETURN ((LM_ERROR,
  95                            "Cannot activate client threads\n"),
  96                            1);
  97
  98       worker.thr_mgr ()->wait ();
  99
 100       ACE_DEBUG ((LM_DEBUG, "event loop finished\n"));
 101
 102       orb->destroy ();
 103     }
 104   catch (const CORBA::Exception& ex)
 105     {
 106       ex._tao_print_exception ("Exception caught:");
 107       return 1;
 108     }
 109
 110   return 0;
 111 }