TAO/tests/Crash_On_Write/server.cpp

   1 #include "Oneway_Receiver.h"
   2 #include "Crash_Task.h"
   3 #include "ace/Get_Opt.h"
   4
   5 const ACE_TCHAR *ior_output_file = ACE_TEXT("test.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
  20       case '?':
  21       default:
  22         ACE_ERROR_RETURN ((LM_ERROR,
  23                            "usage:  %s "
  24                            "-o <iorfile>"
  25                            "\n",
  26                            argv [0]),
  27                           -1);
  28       }
  29   // Indicates successful parsing of the command line
  30   return 0;
  31 }
  32
  33 int
  34 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  35 {
  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       PortableServer::POA_var root_poa =
  45         PortableServer::POA::_narrow (poa_object.in ());
  46
  47       if (CORBA::is_nil (root_poa.in ()))
  48         ACE_ERROR_RETURN ((LM_ERROR,
  49                            " (%P|%t) Panic: nil RootPOA\n"),
  50                           1);
  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       Oneway_Receiver *oneway_receiver_impl;
  59       ACE_NEW_RETURN (oneway_receiver_impl,
  60                       Oneway_Receiver,
  61                       1);
  62       PortableServer::ServantBase_var owner_transfer(oneway_receiver_impl);
  63
  64       PortableServer::ObjectId_var id =
  65         root_poa->activate_object (oneway_receiver_impl);
  66
  67       CORBA::Object_var object = root_poa->id_to_reference (id.in ());
  68
  69       Test::Oneway_Receiver_var oneway_receiver =
  70         Test::Oneway_Receiver::_narrow (object.in ());
  71
  72       CORBA::String_var ior =
  73         orb->object_to_string (oneway_receiver.in ());
  74
  75       // If the ior_output_file exists, output the ior to it
  76       FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
  77       if (output_file == 0)
  78         ACE_ERROR_RETURN ((LM_ERROR,
  79                            "Cannot open output file for writing IOR: %s",
  80                            ior_output_file),
  81                           1);
  82       ACE_OS::fprintf (output_file, "%s", ior.in ());
  83       ACE_OS::fclose (output_file);
  84
  85       poa_manager->activate ();
  86
  87       // Create a separate thread to crash the server.
  88       Crash_Task crash_task (ACE_Thread_Manager::instance (), ACE_Time_Value(10));
  89
  90       if (crash_task.activate () == -1)
  91         {
  92           ACE_ERROR ((LM_ERROR, "Error activating crash task\n"));
  93           return 1;
  94         }
  95
  96       orb->run ();
  97
  98       /*NOTREACHED*/
  99       ACE_DEBUG ((LM_DEBUG, "(%P|%t) server - event loop finished\n"));
 100
 101       root_poa->destroy (1, 1);
 102
 103       orb->destroy ();
 104     }
 105   catch (const CORBA::Exception& ex)
 106     {
 107       ex._tao_print_exception ("Exception caught:");
 108       return 1;
 109     }
 110
 111   return 0;
 112 }