diff options
Diffstat (limited to 'TAO/tests/AMH_Exceptions/server.cpp')
-rw-r--r-- | TAO/tests/AMH_Exceptions/server.cpp | 251 |
1 files changed, 251 insertions, 0 deletions
diff --git a/TAO/tests/AMH_Exceptions/server.cpp b/TAO/tests/AMH_Exceptions/server.cpp new file mode 100644 index 00000000000..4bc543e5537 --- /dev/null +++ b/TAO/tests/AMH_Exceptions/server.cpp @@ -0,0 +1,251 @@ + +// $Id$ + +#include "ace/OS_NS_stdio.h" +#include "TestS.h" +#include "tao/ORB_Core.h" + +/***************************/ +/*** Servant Declaration ***/ + +class ST_AMH_Servant + : public virtual POA_Test::AMH_Roundtrip +{ +public: + ST_AMH_Servant (CORBA::ORB_ptr orb); + + void test_method (Test::AMH_RoundtripResponseHandler_ptr _tao_rh, + Test::Timestamp send_time + ACE_ENV_ARG_DECL) + ACE_THROW_SPEC ((CORBA::SystemException)); + + void shutdown (Test::AMH_RoundtripResponseHandler_ptr _tao_rh ACE_ENV_ARG_DECL) + ACE_THROW_SPEC ((CORBA::SystemException)); + +protected: + CORBA::ORB_ptr orb_; +}; + + +/***************************/ +/*** Servant Definition ***/ + +ST_AMH_Servant::ST_AMH_Servant (CORBA::ORB_ptr orb) + : orb_ (CORBA::ORB::_duplicate (orb)) +{ +} + +void +ST_AMH_Servant::test_method (Test::AMH_RoundtripResponseHandler_ptr _tao_rh, + Test::Timestamp send_time + ACE_ENV_ARG_DECL) + ACE_THROW_SPEC ((CORBA::SystemException)) +{ + // Throw an overload exception + + Test::ServerOverload *ts = new Test::ServerOverload; + + // Calee owns the memory now. Need not delete 'ts' + Test::AMH_RoundtripExceptionHolder holder (ts); + + ACE_TRY + { + _tao_rh->test_method_excep (&holder ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + } + ACE_CATCHALL + {} + ACE_ENDTRY; + + ACE_UNUSED_ARG (send_time); +} + +void +ST_AMH_Servant::shutdown (Test::AMH_RoundtripResponseHandler_ptr /*_tao_rh*/ ACE_ENV_ARG_DECL) + ACE_THROW_SPEC ((CORBA::SystemException)) +{ + this->orb_->shutdown (0 ACE_ENV_ARG_PARAMETER); +} + +/*** Server Declaration ***/ + +/** + Class that performs all 'dirty' initialisation work that is common to + all the AMH servers and 'hides' all the common ORB functions. + */ +class ST_AMH_Server +{ +public: + ST_AMH_Server (int *argc, char **argv); + virtual ~ST_AMH_Server (); + + /// ORB inititalisation stuff + int start_orb_and_poa (void); + + /// register the servant with the poa + virtual void register_servant (ST_AMH_Servant *servant); + + /// orb-perform_work () abstraction + virtual void run_event_loop (); + +public: + /// Accesor method (for servants) to the initialised ORB + CORBA::ORB_ptr orb () { return this->orb_.in (); } + +protected: + int *argc_; + char **argv_; + char *ior_output_file_; + CORBA::ORB_var orb_; + PortableServer::POA_var root_poa_; + +private: + /// Write servant IOR to file specified with the '-o' option + int write_ior_to_file (CORBA::String_var ior); +}; + + + +/*** Server Declaration ***/ + +ST_AMH_Server::ST_AMH_Server (int* argc, char **argv) + : argc_ (argc) + , argv_ (argv) +{ + this->ior_output_file_ = const_cast<char*> ("test.ior"); +} + +ST_AMH_Server::~ST_AMH_Server () +{ + ACE_TRY_NEW_ENV + { + this->root_poa_->destroy (1, 1 ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + this->orb_->destroy (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, "Exception caught:"); + } + ACE_ENDTRY; + +} + +int +ST_AMH_Server::start_orb_and_poa (void) +{ + ACE_TRY_NEW_ENV + { + this->orb_ = CORBA::ORB_init (*(this->argc_), + this->argv_, + "" ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + CORBA::Object_var poa_object = + this->orb_->resolve_initial_references("RootPOA" ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + if (CORBA::is_nil (poa_object.in ())) + ACE_ERROR_RETURN ((LM_ERROR, + " (%P|%t) Unable to initialize the POA.\n"), + 1); + + this->root_poa_ = PortableServer::POA::_narrow (poa_object.in () + ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + PortableServer::POAManager_var poa_manager = + this->root_poa_->the_POAManager (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + poa_manager->activate (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, "Exception caught:"); + return -1; + } + ACE_ENDTRY; + + return 0; +} + +void +ST_AMH_Server::register_servant (ST_AMH_Servant *servant) +{ + ACE_TRY_NEW_ENV + { + Test::Roundtrip_var roundtrip = + servant->_this (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + CORBA::String_var ior = + this->orb_->object_to_string (roundtrip.in () ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + (void) this->write_ior_to_file (ior); + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, "Exception caught:"); + } + ACE_ENDTRY; +} + +void +ST_AMH_Server::run_event_loop () +{ + ACE_DECLARE_NEW_CORBA_ENV; + ACE_TRY + { + ACE_Time_Value period (0, 11000); + while (!this->orb_->orb_core ()->has_shutdown ()) + { + this->orb_->perform_work (&period); + ACE_TRY_CHECK; + } + ACE_TRY_CHECK; + } + ACE_CATCHANY + {} + ACE_ENDTRY; +} + +int +ST_AMH_Server::write_ior_to_file (CORBA::String_var ior) +{ + // If the ior_output_file exists, output the ior to it + FILE *output_file= ACE_OS::fopen (ST_AMH_Server::ior_output_file_, + "w"); + if (output_file == 0) + { + ACE_ERROR ((LM_ERROR, + "Cannot open output file for writing IOR: %s", + ST_AMH_Server::ior_output_file_)); + return -1; + } + + ACE_OS::fprintf (output_file, "%s", ior.in ()); + ACE_OS::fclose (output_file); + return 0; +} + + +int +main (int argc, char *argv[]) +{ + ST_AMH_Server amh_server (&argc, argv); + + amh_server.start_orb_and_poa (); + + ST_AMH_Servant servant (amh_server.orb ()); + + amh_server.register_servant (&servant); + + amh_server.run_event_loop (); + + return 0; +} |