summaryrefslogtreecommitdiff
path: root/ACE/examples/Reactor/Dgram/Dgram.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'ACE/examples/Reactor/Dgram/Dgram.cpp')
-rw-r--r--ACE/examples/Reactor/Dgram/Dgram.cpp258
1 files changed, 258 insertions, 0 deletions
diff --git a/ACE/examples/Reactor/Dgram/Dgram.cpp b/ACE/examples/Reactor/Dgram/Dgram.cpp
new file mode 100644
index 00000000000..c4c21e84186
--- /dev/null
+++ b/ACE/examples/Reactor/Dgram/Dgram.cpp
@@ -0,0 +1,258 @@
+// $Id$
+
+// Exercise the <ACE_SOCK_Dgram> wrapper along with the <ACE_Reactor>.
+// This test simply ping-pongs datagrams back and forth between the
+// peer1 and peer2 processes. This test can be run in two ways:
+//
+// 1. Stand-alone -- e.g.,
+//
+// % ./Dgram
+//
+// which will spawn a child process and run peer1 and peer2
+// in different processes on the same machine.
+//
+// 2. Distributed -- e.g.,
+//
+// # Peer1
+// % ./Dgram 10002 tango.cs.wustl.edu 10003 peer1
+//
+// # Peer1
+// % ./Dgram 10003 tango.cs.wustl.edu 10002 peer2
+//
+// which will run peer1 and peer2 in different processes
+// on the same or different machines. Note that you MUST
+// give the name "peer1" as the final argument to one and
+// only one of the programs so that the test will work properly.
+
+#include "ace/OS_main.h"
+#include "ace/OS_NS_string.h"
+#include "ace/OS_NS_unistd.h"
+#include "ace/Reactor.h"
+#include "ace/Process.h"
+#include "ace/SOCK_Dgram.h"
+#include "ace/INET_Addr.h"
+#include "ace/Log_Msg.h"
+
+ACE_RCSID(Dgram, Dgram, "$Id$")
+
+// Port used to receive for dgrams.
+static u_short port1;
+
+class Dgram_Endpoint : public ACE_Event_Handler
+{
+public:
+ Dgram_Endpoint (const ACE_INET_Addr &local_addr);
+
+ // = Hook methods inherited from the <ACE_Event_Handler>.
+ virtual ACE_HANDLE get_handle (void) const;
+ virtual int handle_input (ACE_HANDLE handle);
+ virtual int handle_timeout (const ACE_Time_Value & tv,
+ const void *arg = 0);
+ virtual int handle_close (ACE_HANDLE handle,
+ ACE_Reactor_Mask close_mask);
+
+ int send (const char *buf, size_t len, const ACE_INET_Addr &);
+ // Send the <buf> to the peer.
+
+private:
+ ACE_SOCK_Dgram endpoint_;
+ // Wrapper for sending/receiving dgrams.
+};
+
+int
+Dgram_Endpoint::send (const char *buf,
+ size_t len,
+ const ACE_INET_Addr &addr)
+{
+ return this->endpoint_.send (buf, len, addr);
+}
+
+Dgram_Endpoint::Dgram_Endpoint (const ACE_INET_Addr &local_addr)
+ : endpoint_ (local_addr)
+{
+}
+
+ACE_HANDLE
+Dgram_Endpoint::get_handle (void) const
+{
+ return this->endpoint_.get_handle ();
+}
+
+int
+Dgram_Endpoint::handle_close (ACE_HANDLE handle,
+ ACE_Reactor_Mask)
+{
+ ACE_UNUSED_ARG (handle);
+
+ this->endpoint_.close ();
+ delete this;
+ return 0;
+}
+
+int
+Dgram_Endpoint::handle_input (ACE_HANDLE)
+{
+ char buf[BUFSIZ];
+ ACE_INET_Addr from_addr;
+
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) activity occurred on handle %d!\n",
+ this->endpoint_.get_handle ()));
+
+ ssize_t n = this->endpoint_.recv (buf,
+ sizeof buf,
+ from_addr);
+
+ if (n == -1)
+ ACE_ERROR ((LM_ERROR,
+ "%p\n",
+ "handle_input"));
+ else
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) buf of size %d = %*s\n",
+ n,
+ n,
+ buf));
+ return 0;
+}
+
+int
+Dgram_Endpoint::handle_timeout (const ACE_Time_Value &,
+ const void *)
+{
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) timed out for endpoint\n"));
+ return 0;
+}
+
+static int
+run_test (u_short localport,
+ const ACE_TCHAR *remotehost,
+ u_short remoteport,
+ const ACE_TCHAR *peer)
+{
+ ACE_INET_Addr remote_addr (remoteport,
+ remotehost);
+ ACE_INET_Addr local_addr (localport);
+
+ Dgram_Endpoint *endpoint;
+
+ ACE_NEW_RETURN (endpoint,
+ Dgram_Endpoint (local_addr),
+ -1);
+
+ // Read data from other side.
+ if (ACE_Reactor::instance ()->register_handler
+ (endpoint,
+ ACE_Event_Handler::READ_MASK) == -1)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ "ACE_Reactor::register_handler"),
+ -1);
+
+ char buf[BUFSIZ];
+ ACE_OS::strcpy (buf, "Data to transmit");
+ size_t len = ACE_OS::strlen (buf);
+
+ if (ACE_OS::strncmp (peer, ACE_TEXT("peer1"), 5) == 0)
+ {
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) sending data\n"));
+
+ for (size_t i = 0; i < 20; i++)
+ {
+ endpoint->send (buf, len, remote_addr);
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) .\n"));
+ ACE_OS::sleep (1);
+ }
+ }
+
+ for (int i = 0; i < 40; i++)
+ {
+ // Wait up to 10 seconds for data.
+ ACE_Time_Value tv (10, 0);
+
+ if (ACE_Reactor::instance ()->handle_events (tv) <= 0)
+ ACE_ERROR_RETURN ((LM_DEBUG,
+ "(%P|%t) %p\n",
+ "handle_events"),
+ -1);
+
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) return from handle events\n"));
+
+ endpoint->send (buf, len, remote_addr);
+
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) .\n"));
+ }
+
+ if (ACE_Reactor::instance ()->remove_handler
+ (endpoint,
+ ACE_Event_Handler::READ_MASK) == -1)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ "ACE_Reactor::remove_handler"),
+ -1);
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) exiting\n"));
+ return 0;
+}
+
+int
+ACE_TMAIN (int argc, ACE_TCHAR *argv[])
+{
+ // Estabish call backs and socket names.
+
+ port1 = argc > 1 ? ACE_OS::atoi (argv[1]) : ACE_DEFAULT_SERVER_PORT;
+ const ACE_TCHAR *remotehost = argc > 2 ? argv[2] : ACE_DEFAULT_SERVER_HOST;
+ const u_short port2 = argc > 3 ? ACE_OS::atoi (argv[3]) : port1 + 1;
+
+ // Providing the fourth command line argument indicate we don't want
+ // to spawn a new process. On Win32, we use this to exec the new
+ // program.
+ if (argc > 4)
+ run_test (port1, remotehost, port2, argv[4]);
+ else
+ {
+ ACE_DEBUG ((LM_DEBUG,
+ "(%P|%t) local port = %d, remote host = %s, remote port = %d\n",
+ port1,
+ remotehost,
+ port2));
+
+ ACE_Process_Options options;
+ options.command_line ("%s %d %s %d %c",
+ argv[0],
+ port1,
+ remotehost,
+ port2,
+ 'c');
+
+ // This has no effect on NT and will spawn a process that exec
+ // the above run_test function.
+ options.creation_flags (ACE_Process_Options::NO_EXEC);
+
+ ACE_Process new_process;
+ switch (new_process.spawn (options))
+ {
+ case -1:
+ return -1;
+
+ case 0:
+ run_test (port1,
+ remotehost,
+ port2,
+ ACE_TEXT("peer1"));
+ break;
+
+ default:
+ run_test (port2,
+ remotehost,
+ port1,
+ ACE_TEXT("peer2"));
+ new_process.wait ();
+ break;
+ }
+ }
+ return 0;
+}