diff options
Diffstat (limited to 'ACE/examples/QOS/Simple/sender.cpp')
| -rw-r--r-- | ACE/examples/QOS/Simple/sender.cpp | 315 |
1 files changed, 315 insertions, 0 deletions
diff --git a/ACE/examples/QOS/Simple/sender.cpp b/ACE/examples/QOS/Simple/sender.cpp new file mode 100644 index 00000000000..97e6aafa407 --- /dev/null +++ b/ACE/examples/QOS/Simple/sender.cpp @@ -0,0 +1,315 @@ +/* -*- C++ -*- */ +// $Id$ + +// ============================================================================ +// +// = LIBRARY +// ACE_wrappers/examples/QOS +// +// = FILENAME +// client.cpp +// +// = AUTHOR +// Vishal Kachroo <vishal@cs.wustl.edu> +// +// ============================================================================ + + +#include "ace/QoS/QoS_Session.h" +#include "ace/QoS/QoS_Session_Factory.h" +#include "ace/QoS/QoS_Session_Impl.h" +#include "ace/QoS/QoS_Decorator.h" +#include "ace/QoS/SOCK_Dgram_Mcast_QoS.h" + +#include "QoS_Util.h" +#include "Fill_ACE_QoS.h" +#include "QoS_Signal_Handler.h" +#include "Sender_QoS_Event_Handler.h" + +// To open QOS sockets administrative access is required on the +// machine. Fill in default values for QoS structure. The default +// values were simply choosen from existing QOS templates available +// via WSAGetQosByName. Notice that ProviderSpecific settings are +// being allowed when picking the "default" template but not for +// "well-known" QOS templates. Also notice that since data is only +// flowing from sender to receiver, different flowspecs are filled in +// depending upon whether this application is acting as a sender or +// receiver. + +// This function fills up the ACE_QoS_Params with the supplied iovec and ACE_QoS. + +int +FillQoSParams (ACE_QoS_Params &qos_params, + iovec* iov, + ACE_QoS* qos) +{ + qos_params.callee_data (iov); + qos_params.caller_data (0); + qos_params.socket_qos (qos); + qos_params.group_socket_qos (0); + qos_params.flags (ACE_JL_BOTH); + + return 0; +} + +int +ACE_TMAIN (int argc, ACE_TCHAR * argv[]) +{ + + ACE_DEBUG ((LM_DEBUG, + "Sender\n")); + + QoS_Util qos_util(argc, argv); + + if (qos_util.parse_args () == -1) + ACE_ERROR_RETURN ((LM_ERROR, + "Error in parsing args\n"), + -1); + + // This is a multicast application. + if (qos_util.multicast_flag ()) + { + Fill_ACE_QoS fill_ace_qos; + + // The application adds the flow specs that it wants into the + // Fill_ACE_QoS. The Fill_ACE_QoS indexes the flow specs by the flow + // spec names. Here the new flowspec being added is g_711. + ACE_CString g_711 ("g_711"); + + switch (fill_ace_qos.map ().bind (g_711, + new ACE_Flow_Spec (9200, + 708, + 18400, + 0, + 0, + ACE_SERVICETYPE_CONTROLLEDLOAD, + 368, + 368, + 25, + 1))) + { + case 1 : + ACE_ERROR_RETURN ((LM_ERROR, + "Unable to bind the new flow spec\n" + "The Flow Spec name already exists\n"), + -1); + break; + case -1 : + ACE_ERROR_RETURN ((LM_ERROR, + "Unable to bind the new flow spec\n"), + -1); + break; + } + + ACE_DEBUG ((LM_DEBUG, + "g_711 Flow Spec bound successfully\n")); + + // This is a sender. So we fill in the sending QoS parameters. + ACE_QoS ace_qos_sender; + + if (fill_ace_qos.fill_simplex_sender_qos (ace_qos_sender, + g_711) !=0) + ACE_ERROR_RETURN ((LM_ERROR, + "Unable to fill simplex sender qos\n"), + -1); + else + ACE_DEBUG ((LM_DEBUG, + "Filled up the Sender QoS parameters\n")); + + // Opening a new Multicast Datagram. It is absolutely necessary that + // the sender and the receiver subscribe to the same multicast + // addresses to make sure the "multicast sessions" for the two are + // the same. This is used to match the RESV<->PATH states. + ACE_SOCK_Dgram_Mcast_QoS dgram_mcast_qos; + + // Multicast Session Address specified by user at command line. + // If this address is not specified, + // <localhost:ACE_DEFAULT_MULTICAST_PORT> is assumed. + ACE_INET_Addr mult_addr (*(qos_util.mult_session_addr ())); + + // Fill the ACE_QoS_Params to be passed to the <ACE_OS::join_leaf> + // through subscribe. + + ACE_QoS_Params qos_params; + FillQoSParams (qos_params, 0, &ace_qos_sender); + + // Create a QoS Session Factory. + ACE_QoS_Session_Factory session_factory; + + // Ask the factory to create a QoS session. + ACE_QoS_Session *qos_session = + session_factory.create_session (); + + // Create a destination address for the QoS session. The same + // address should be used for the subscribe call later. A copy is + // made below only to distinguish the two usages of the dest + // address. + + ACE_INET_Addr dest_addr (mult_addr); + + // A QoS session is defined by the 3-tuple [DestAddr, DestPort, + // Protocol]. Initialize the QoS session. + if (qos_session->open (mult_addr, + IPPROTO_UDP) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + "Error in opening the QoS session\n"), + -1); + else + ACE_DEBUG ((LM_DEBUG, + "QoS session opened successfully\n")); + + // The following call opens the Dgram_Mcast and calls the + // <ACE_OS::join_leaf> with the qos_params supplied here. Note the + // QoS session object is passed into this call. This subscribes the + // underlying socket to the passed in QoS session. For joining + // multiple multicast sessions, the following subscribe call should + // be made with different multicast addresses and a new QoS session + // object should be passed in for each such call. The QoS session + // objects can be created only through the session factory. Care + // should be taken that the mult_addr for the subscribe() call + // matches the dest_addr of the QoS session object. If this is not + // done, the subscribe call will fail. A more abstract version of + // subscribe will be added that constrains the various features of + // GQoS like different flags etc. + + if (dgram_mcast_qos.subscribe (mult_addr, + qos_params, + 1, + 0, + AF_INET, + // ACE_FROM_PROTOCOL_INFO, + 0, + 0, // ACE_Protocol_Info, + 0, + ACE_OVERLAPPED_SOCKET_FLAG + | ACE_FLAG_MULTIPOINT_C_LEAF + | ACE_FLAG_MULTIPOINT_D_LEAF, + qos_session) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + "Error in subscribe\n"), + -1); + else + ACE_DEBUG ((LM_DEBUG, + "Dgram_Mcast subscribe succeeds \n")); + + int nIP_TTL = 25; + char achInBuf [BUFSIZ]; + u_long dwBytes; + + // Should this be abstracted into QoS objects ?? Doesnt seem to have + // to do anything directly with QoS. + if (ACE_OS::ioctl (dgram_mcast_qos.get_handle (), // Socket. + ACE_SIO_MULTICAST_SCOPE, // IO control code. + &nIP_TTL, // In buffer. + sizeof (nIP_TTL), // Length of in buffer. + achInBuf, // Out buffer. + BUFSIZ, // Length of Out buffer. + &dwBytes, // bytes returned. + 0, // Overlapped. + 0) == -1) // Func. + ACE_ERROR ((LM_ERROR, + "Error in Multicast scope ACE_OS::ioctl() \n")); + else + ACE_DEBUG ((LM_DEBUG, + "Setting TTL with Multicast scope ACE_OS::ioctl call succeeds \n")); + + int bFlag = 0; + + // Should this be abstracted into QoS objects ?? Doesnt seem to have + // to do anything directly with QoS. + if (ACE_OS::ioctl (dgram_mcast_qos.get_handle (), // Socket. + ACE_SIO_MULTIPOINT_LOOPBACK, // IO control code. + &bFlag, // In buffer. + sizeof (bFlag), // Length of in buffer. + achInBuf, // Out buffer. + BUFSIZ, // Length of Out buffer. + &dwBytes, // bytes returned. + 0, // Overlapped. + 0) == -1) // Func. + ACE_ERROR ((LM_ERROR, + "Error in Loopback ACE_OS::ioctl() \n")); + else + ACE_DEBUG ((LM_DEBUG, + "Disable Loopback with ACE_OS::ioctl call succeeds \n")); + + // This is a sender. + qos_session->flags (ACE_QoS_Session::ACE_QOS_SENDER); + + ACE_QoS_Manager qos_manager = dgram_mcast_qos.qos_manager (); + + // Since we are using RSVP, it is imperative that the client + // application have the option of supplying the source sender + // port for the RSVP messages. A default will be chosen by the + // ACE API if this is not done. + qos_session->source_port (qos_util.source_port ()); + + // Set the QoS for the session. Replaces the ioctl () call that + // was being made previously. + if (qos_session->qos (&dgram_mcast_qos, + &qos_manager, + ace_qos_sender) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + "Unable to set QoS\n"), + -1); + else + ACE_DEBUG ((LM_DEBUG, + "Setting QOS succeeds.\n")); + + // Register a signal handler that helps to gracefully close the open + // QoS sessions. + QoS_Signal_Handler qos_signal_handler (qos_session); + + // Register the usual SIGINT signal handler with the Reactor for + // the application to gracefully release the QoS session and + // shutdown. + if (ACE_Reactor::instance ()->register_handler + (SIGINT, &qos_signal_handler) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + "Error in registering the Signal Handler.\n"), + -1); + + // Handler to process QoS and Data events for the reciever. + Sender_QoS_Event_Handler qos_event_handler (dgram_mcast_qos, + qos_session); + + // Decorate the above handler with QoS functionality. + ACE_QoS_Decorator qos_decorator (&qos_event_handler, + qos_session); + + // Initialize the Decorator. + if (qos_decorator.init () != 0) + ACE_ERROR_RETURN ((LM_ERROR, + "QoS Decorator init () failed.\n"), + -1); + + // Register the decorated Event Handler with the Reactor. + if (ACE_Reactor::instance ()->register_handler (&qos_decorator, + ACE_Event_Handler::QOS_MASK | + ACE_Event_Handler::READ_MASK) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + "Error in registering the Decorator with the Reactor\n"), + -1); + + + + // Start the event loop. + ACE_DEBUG ((LM_DEBUG, + "Running the Event Loop ... \n")); + + ACE_Reactor::instance ()->run_event_loop (); + + ACE_DEBUG ((LM_DEBUG, + "(%P|%t) shutting down server logging daemon\n")); + } + else + ACE_DEBUG ((LM_DEBUG, + "Specify a -m option for multicast application\n")); + return 0; +} + + + + + + + |