TAO/tests/SHMIOP/shmiop_test.cpp

   1 #include "TestS.h"
   2
   3 #include "tao/Strategies/advanced_resource.h"
   4
   5 #include "ace/Get_Opt.h"
   6 #include "ace/OS_NS_stdio.h"
   7 #include "ace/SString.h"
   8 #include "ace/Task.h"
   9
  10 struct Servant : virtual POA_Test::Hello
  11 {
  12   explicit Servant (const CORBA::ORB_var &orb)
  13     : orb_ (orb)
  14   {}
  15
  16   char *get_string ()
  17   {
  18     CORBA::String_var str ("hello");
  19     return str._retn ();
  20   }
  21
  22   void shutdown ()
  23   {
  24     this->orb_->shutdown (0);
  25   }
  26
  27   CORBA::ORB_var orb_;
  28 };
  29
  30 struct ORBTask : ACE_Task_Base
  31 {
  32   explicit ORBTask (const CORBA::ORB_var &orb)
  33     : orb_ (orb)
  34   {}
  35
  36   int svc ()
  37   {
  38     try
  39       {
  40         this->orb_->run ();
  41         return 0;
  42       }
  43     catch (const CORBA::Exception &e)
  44       {
  45         e._tao_print_exception ("Exception caught from ORB::run:");
  46       }
  47     catch (...)
  48       {
  49         ACE_ERROR ((LM_ERROR, ACE_TEXT ("ERROR unknown exception caught ")
  50                               ACE_TEXT ("from ORB::run\n")));
  51       }
  52     return 1;
  53   }
  54
  55   CORBA::ORB_var orb_;
  56 };
  57
  58 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  59 {
  60   try
  61     {
  62       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  63
  64       ACE_Get_Opt opts (argc, argv, ACE_TEXT ("s:c:"));
  65       const ACE_TCHAR *server = 0, *client = 0;
  66       for (int o; (o = opts ()) != -1;)
  67         {
  68           switch (o)
  69             {
  70             case 's':
  71               server = opts.opt_arg ();
  72               break;
  73             case 'c':
  74               client = opts.opt_arg ();
  75               break;
  76             }
  77         }
  78
  79       Servant srv (orb);
  80       ORBTask task (orb);
  81
  82       if (server)
  83         {
  84           CORBA::Object_var obj = orb->resolve_initial_references ("RootPOA");
  85           PortableServer::POA_var poa = PortableServer::POA::_narrow (obj);
  86           PortableServer::POAManager_var pm = poa->the_POAManager ();
  87           pm->activate ();
  88           Test::Hello_var srv_obj = srv._this ();
  89           CORBA::String_var srv_str = orb->object_to_string (srv_obj);
  90           FILE *f = ACE_OS::fopen (server, "w");
  91           ACE_OS::fputs (srv_str, f);
  92           ACE_OS::fclose (f);
  93           if (!client)
  94             task.activate();
  95         }
  96
  97       if (client)
  98         {
  99           ACE_CString ior ("file://");
 100           ior += ACE_TEXT_ALWAYS_CHAR (client);
 101           CORBA::Object_var obj = orb->string_to_object (ior.c_str ());
 102           Test::Hello_var hello = Test::Hello::_narrow (obj);
 103           CORBA::String_var str = hello->get_string ();
 104           ACE_DEBUG ((LM_DEBUG, ACE_TEXT ("(%P|%t) - returned <%C>\n"),
 105                       str.in ()));
 106           hello->shutdown ();
 107         }
 108       else if (server)
 109         {
 110           task.wait ();
 111         }
 112
 113       orb->destroy ();
 114       return 0;
 115     }
 116   catch (const CORBA::Exception &e)
 117     {
 118       e._tao_print_exception ("Exception caught:");
 119     }
 120   catch (...)
 121     {
 122       ACE_ERROR ((LM_ERROR, ACE_TEXT ("ERROR unknown exception ")
 123                             ACE_TEXT ("caught in main\n")));
 124     }
 125   return 1;
 126 }