TAO/tests/Bug_2494_Regression/client.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/Task.h"
   3 #include "ace/OS_NS_string.h"
   4 #include "testC.h"
   5
   6 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   7 int nthreads = 5;
   8 int do_shutdown = 0;
   9
  10 int
  11 parse_args (int argc, ACE_TCHAR *argv[])
  12 {
  13   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("xk:n:"));
  14   int c;
  15
  16   while ((c = get_opts ()) != -1)
  17     switch (c)
  18       {
  19       case 'x':
  20         do_shutdown = 1;
  21         break;
  22
  23       case 'k':
  24         ior = get_opts.opt_arg ();
  25         break;
  26
  27       case 'n':
  28         nthreads = ACE_OS::atoi (get_opts.opt_arg ());
  29         break;
  30
  31       case '?':
  32       default:
  33         ACE_ERROR_RETURN ((LM_ERROR,
  34                            "usage: %s "
  35                            "-k <ior> "
  36                            "[-n <nthreads> | -x]"
  37                            "\n",
  38                            argv [0]),
  39                           -1);
  40       }
  41   // Indicates successful parsing of the command line
  42   return 0;
  43 }
  44
  45 namespace
  46 {
  47   const char *twohundredbytes =
  48   "12345678901234567890123456789012345678901234567890"
  49   "12345678901234567890123456789012345678901234567890"
  50   "12345678901234567890123456789012345678901234567890"
  51   "12345678901234567890123456789012345678901234567890"
  52   ;
  53 }
  54
  55 struct Worker : ACE_Task_Base
  56 {
  57
  58   Worker (Simple_Server_ptr srv)
  59     : srv_(Simple_Server::_duplicate(srv))
  60   {
  61   }
  62
  63   Simple_Server_var srv_;
  64
  65   int svc ()
  66   {
  67     CORBA::String_var str = CORBA::string_alloc (200*2000 + 1);
  68     if (str.in () == 0) return 1;
  69     str.inout ()[0] = CORBA::Char ('\0');
  70     for (int i=0; i < 2000; ++i)
  71       {
  72         ACE_OS::strcat (str.inout (), twohundredbytes);
  73       }
  74
  75     while (1)
  76       {
  77         try
  78           {
  79             CORBA::String_var ret = srv_->test_method (str.in ());
  80             if (0 != ACE_OS::strcmp (str.in (), ret.in ())) return 1;
  81           }
  82         catch (const CORBA::Exception& ex)
  83           {
  84             ACE_DEBUG ((LM_ERROR,
  85                         ACE_TEXT ("(%P|%t) Exception caught: \n%C\n"),
  86                         ex._info ().c_str ()));
  87             return 1;
  88           }
  89       }
  90     return 0;
  91   }
  92
  93 };
  94
  95 int
  96 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  97 {
  98   try
  99     {
 100       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 101
 102       if (parse_args (argc, argv) != 0)
 103         return 1;
 104
 105       CORBA::Object_var object = orb->string_to_object (ior);
 106
 107       Simple_Server_var server = Simple_Server::_narrow (object.in ());
 108
 109       if (CORBA::is_nil (server.in ()))
 110         {
 111           ACE_ERROR_RETURN ((LM_ERROR,
 112                              "Object reference <%s> is nil.\n",
 113                              ior), 1);
 114         }
 115
 116       if (do_shutdown)
 117         {
 118           server->shutdown ();
 119         }
 120       else
 121         {
 122           Worker wrk (server.in ());
 123           wrk.activate (THR_NEW_LWP|THR_JOINABLE|THR_INHERIT_SCHED, nthreads);
 124           wrk.thr_mgr ()->wait ();
 125         }
 126     }
 127   catch (const CORBA::Exception& ex)
 128     {
 129       ACE_DEBUG ((LM_ERROR, ACE_TEXT ("(%P|%t) ERROR: Exception caught: \n%C\n"),
 130                   ex._info ().c_str ()));
 131       return 1;
 132     }
 133
 134   return 0;
 135 }