diff options
Diffstat (limited to 'ACE/TAO/orbsvcs/tests/Security/Big_Request/server.cpp')
-rw-r--r-- | ACE/TAO/orbsvcs/tests/Security/Big_Request/server.cpp | 146 |
1 files changed, 146 insertions, 0 deletions
diff --git a/ACE/TAO/orbsvcs/tests/Security/Big_Request/server.cpp b/ACE/TAO/orbsvcs/tests/Security/Big_Request/server.cpp new file mode 100644 index 00000000000..a535b6b2e50 --- /dev/null +++ b/ACE/TAO/orbsvcs/tests/Security/Big_Request/server.cpp @@ -0,0 +1,146 @@ + +#include "TX_Object_i.h" +#include "ace/Get_Opt.h" +#include "ace/OS_NS_stdio.h" +#include "ace/OS_NS_string.h" +#include "ace/SString.h" +#include "ace/Task.h" + +ACE_RCSID (Big_Request, + server, + "$Id$") + +const ACE_TCHAR *ior_output_file = 0; +const char *cert_file = "cacert.pem"; + +class OrbTask : public ACE_Task_Base +{ +public: + OrbTask(const CORBA::ORB_ptr orb) + : orb_(CORBA::ORB::_duplicate(orb)) + { + } + + virtual int svc() + { + try + { + this->orb_->run (); + } + catch (const CORBA::Exception&) + { + } + return 0; + } + +private: + CORBA::ORB_var orb_; +}; + +static int n_threads = 1; + +int +parse_args (int argc, ACE_TCHAR *argv[]) +{ + ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:t:")); + int c; + + while ((c = get_opts ()) != -1) + switch (c) + { + case 'o': + ior_output_file = get_opts.opt_arg (); + break; + case 't': + n_threads = ACE_OS::atoi(get_opts.opt_arg()); + break; + case '?': + default: + ACE_ERROR_RETURN ((LM_ERROR, + "Usage: %s " + "-o <iorfile>" + "-t <thread count>" + "\n", + argv [0]), + -1); + } + + // Indicates sucessful parsing of the command line + return 0; +} + + +int +ACE_TMAIN(int argc, ACE_TCHAR *argv[]) +{ + try + { + ACE_CString env ("SSL_CERT_FILE="); + env += cert_file; + ACE_OS::putenv (env.c_str ()); + + CORBA::ORB_var orb = CORBA::ORB_init (argc, argv); + + CORBA::Object_var poaObj = + orb->resolve_initial_references ("RootPOA"); + + PortableServer::POA_var rootPoa = + PortableServer::POA::_narrow (poaObj.in ()); + + PortableServer::POAManager_var poa_manager = + rootPoa->the_POAManager (); + + poa_manager->activate (); + + if (::parse_args (argc, argv) != 0) + return 1; + + TX_Object_i implObject (orb.in ()); + + TX_Object_var txObject = implObject._this (); + + CORBA::String_var ior = + orb->object_to_string (txObject.in ()); + + ACE_DEBUG ((LM_DEBUG, "Activated as <%C>\n", ior.in ())); + + // If the ior_output_file exists, output the ior to it. + if (ior_output_file != 0) + { + FILE *output_file= ACE_OS::fopen (ior_output_file, "w"); + if (output_file == 0) + ACE_ERROR_RETURN ((LM_ERROR, + "Cannot open output file for writing IOR: %s\n", + ior_output_file), + 1); + ACE_OS::fprintf (output_file, "%s", ior.in ()); + ACE_OS::fclose (output_file); + } + + OrbTask task(orb.in()); + + if (task.activate (THR_NEW_LWP | THR_JOINABLE, + n_threads) != 0) + ACE_ERROR_RETURN ((LM_ERROR, + "Cannot activate threads\n"), + 1); + task.wait(); + + ACE_DEBUG ((LM_DEBUG, + "\n" + "Event loop finished.\n")); + + rootPoa->destroy (1, 1); + + orb->destroy (); + } + catch (const CORBA::Exception& ex) + { + ex._tao_print_exception ("ERROR"); + + return -1; + } + + + return 0; +} |