TAO/tests/CollocationLockup/SimpleNamingService.cpp

   1 /**
   2  * SimpleNamingService.cpp
   3  * This is part of the regression test against bug #2130.
   4  * See CollocationLockup.cpp for a description of the test.
   5  */
   6
   7 #include "SimpleNamingServiceS.h"
   8
   9 #include "ace/String_Base.h"
  10 #include "ace/streams.h"
  11 #include "ace/Get_Opt.h"
  12
  13 #include "ace/Event_Handler.h"
  14 #include "ace/Sig_Handler.h"
  15
  16 class TestShutdown : public ACE_Event_Handler
  17 {
  18 public:
  19   TestShutdown (CORBA::ORB_ptr orb)
  20     : orb_(CORBA::ORB::_duplicate (orb))
  21   {
  22 #if !defined(ACE_LACKS_UNIX_SIGNALS)
  23     this->shutdown_.register_handler (SIGTERM, this);
  24     this->shutdown_.register_handler (SIGINT, this);
  25 #elif defined(ACE_WIN32)
  26     this->shutdown_.register_handler (SIGINT, this);
  27 #endif
  28   }
  29
  30   ~TestShutdown ()
  31   {
  32 #if !defined(ACE_LACKS_UNIX_SIGNALS)
  33     this->shutdown_.remove_handler (SIGTERM);
  34     this->shutdown_.remove_handler (SIGINT);
  35 #elif defined(ACE_WIN32)
  36     this->shutdown_.remove_handler (SIGINT);
  37 #endif
  38   }
  39
  40   virtual int handle_signal (int, siginfo_t*, ucontext_t*)
  41   {
  42     this->orb_->shutdown ();
  43     return 0;
  44   }
  45
  46 private:
  47   CORBA::ORB_var orb_;
  48
  49   ACE_Sig_Handler shutdown_;
  50 };
  51
  52 namespace
  53 {
  54   const ACE_TCHAR* iorFileName = ACE_TEXT("SimpleNamingService.ior");
  55 }
  56
  57 class SimpleNamingService_i : public virtual POA_SimpleNamingService
  58 {
  59 public:
  60   //FUZZ: disable check_for_lack_ACE_OS
  61   virtual void bind (CORBA::Object_ptr obj)
  62   {
  63     obj_ = CORBA::Object::_duplicate (obj);
  64   }
  65   //FUZZ: enable check_for_lack_ACE_OS
  66
  67
  68   virtual CORBA::Object_ptr resolve ()
  69   {
  70     return CORBA::Object::_duplicate (obj_.in ());
  71   }
  72
  73 private:
  74   CORBA::Object_var obj_;
  75 };
  76
  77 int
  78 parse_args (int argc, ACE_TCHAR *argv[])
  79 {
  80   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  81   int c;
  82
  83   while ((c = get_opts ()) != -1)
  84     switch (c)
  85       {
  86       case 'o':
  87         iorFileName = get_opts.opt_arg ();
  88         break;
  89
  90       case '?':
  91       default:
  92         ACE_ERROR_RETURN ((LM_ERROR,
  93                            "usage:  %s "
  94                            "-o <iorfile>"
  95                            "\n",
  96                            argv [0]),
  97                           -1);
  98       }
  99   // Indicates successful parsing of the command line
 100   return 0;
 101 }
 102
 103
 104 int
 105 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 106 {
 107   try
 108     {
 109       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 110
 111       TestShutdown killer (orb.in ());
 112
 113       if (parse_args (argc, argv) != 0)
 114         return 1;
 115
 116       CORBA::Object_var pPoaObj =
 117         orb->resolve_initial_references ("RootPOA");
 118       PortableServer::POA_var poa =
 119         PortableServer::POA::_narrow (pPoaObj.in ());
 120       PortableServer::POAManager_var pMgr = poa->the_POAManager ();
 121       pMgr->activate ();
 122
 123       SimpleNamingService_i* servant = new SimpleNamingService_i;
 124       PortableServer::ObjectId_var oid = poa->activate_object (servant);
 125       CORBA::Object_var obj = poa->id_to_reference (oid.in ());
 126       CORBA::String_var str = orb->object_to_string (obj.in ());
 127
 128       FILE *output_file= ACE_OS::fopen (iorFileName, "w");
 129       if (output_file == 0)
 130         ACE_ERROR_RETURN ((LM_ERROR,
 131                            "Cannot open output file for writing IOR: %s\n",
 132                            iorFileName),
 133                            1);
 134       ACE_OS::fprintf (output_file, "%s", str.in ());
 135       ACE_OS::fclose (output_file);
 136
 137       orb->run ();
 138     }
 139   catch (const CORBA::Exception& ex)
 140     {
 141       ACE_DEBUG ((LM_ERROR, "Corba Exception: %s\n", ex._info ().c_str ()));
 142       return 1;
 143     }
 144
 145   return 0;
 146 }