1 files changed, 290 insertions, 0 deletions
diff --git a/TAO/orbsvcs/orbsvcs/PortableGroup/UIPMC_Connection_Handler.cpp b/TAO/orbsvcs/orbsvcs/PortableGroup/UIPMC_Connection_Handler.cpp
new file mode 100644
index 00000000000..d01a0248fbd
--- /dev/null
+++ b/TAO/orbsvcs/orbsvcs/PortableGroup/UIPMC_Connection_Handler.cpp
@@ -0,0 +1,290 @@
+// This may look like C, but it's really -*- C++ -*-
+//
+// $Id$
+
+
+#include "orbsvcs/PortableGroup/UIPMC_Connection_Handler.h"
+#include "orbsvcs/PortableGroup/UIPMC_Transport.h"
+#include "orbsvcs/PortableGroup/UIPMC_Endpoint.h"
+
+#include "tao/Timeprobe.h"
+#include "tao/debug.h"
+#include "tao/ORB_Core.h"
+#include "tao/ORB.h"
+#include "tao/CDR.h"
+#include "tao/Server_Strategy_Factory.h"
+#include "tao/Transport_Cache_Manager.h"
+#include "tao/Thread_Lane_Resources.h"
+#include "tao/Base_Transport_Property.h"
+#include "tao/Resume_Handle.h"
+
+
+ACE_RCSID(PortableGroup,
+          UIPMC_Connection_Handler,
+          "$Id$")
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+TAO_UIPMC_Connection_Handler::TAO_UIPMC_Connection_Handler (ACE_Thread_Manager *t)
+  : TAO_UIPMC_SVC_HANDLER (t, 0 , 0),
+    TAO_Connection_Handler (0),
+    udp_socket_ (ACE_sap_any_cast (ACE_INET_Addr &)),
+    mcast_socket_ (),
+    using_mcast_ (0)
+{
+  // This constructor should *never* get called, it is just here to
+  // make the compiler happy: the default implementation of the
+  // Creation_Strategy requires a constructor with that signature, we
+  // don't use that implementation, but some (most?) compilers
+  // instantiate it anyway.
+  ACE_ASSERT (0);
+}
+
+
+TAO_UIPMC_Connection_Handler::TAO_UIPMC_Connection_Handler (TAO_ORB_Core *orb_core)
+  : TAO_UIPMC_SVC_HANDLER (orb_core->thr_mgr (), 0, 0),
+    TAO_Connection_Handler (orb_core),
+    udp_socket_ (ACE_sap_any_cast (ACE_INET_Addr &)),
+    mcast_socket_ (),
+    using_mcast_ (0)
+{
+  TAO_UIPMC_Transport* specific_transport = 0;
+  ACE_NEW(specific_transport,
+          TAO_UIPMC_Transport(this, orb_core, 0));
+
+  // store this pointer (indirectly increment ref count)
+  this->transport (specific_transport);
+}
+
+
+TAO_UIPMC_Connection_Handler::~TAO_UIPMC_Connection_Handler (void)
+{
+  if (this->using_mcast_)
+    {
+      // Closing a multicast socket automatically unsubscribes us from
+      // the multicast group.
+      this->mcast_socket_.close ();
+    }
+  else
+    this->udp_socket_.close ();
+
+  delete this->transport ();
+  int const result =
+    this->release_os_resources ();
+
+  if (result == -1 && TAO_debug_level)
+    {
+      ACE_ERROR ((LM_ERROR,
+                  ACE_TEXT("TAO (%P|%t) - UMPIC_Connection_Handler::")
+                  ACE_TEXT("~UMPIC_Connection_Handler, ")
+                  ACE_TEXT("release_os_resources() failed %m\n")));
+    }
+}
+
+ACE_HANDLE
+TAO_UIPMC_Connection_Handler::get_handle (void) const
+{
+  if (this->using_mcast_)
+    return this->mcast_socket_.get_handle ();
+  else
+    return this->udp_socket_.get_handle ();
+}
+
+const ACE_INET_Addr &
+TAO_UIPMC_Connection_Handler::addr (void)
+{
+  return this->addr_;
+}
+
+
+void
+TAO_UIPMC_Connection_Handler::addr (const ACE_INET_Addr &addr)
+{
+  this->addr_ = addr;
+}
+
+
+const ACE_INET_Addr &
+TAO_UIPMC_Connection_Handler::local_addr (void)
+{
+  return local_addr_;
+}
+
+
+void
+TAO_UIPMC_Connection_Handler::local_addr (const ACE_INET_Addr &addr)
+{
+  local_addr_ = addr;
+}
+
+
+const ACE_SOCK_Dgram &
+TAO_UIPMC_Connection_Handler::dgram (void)
+{
+  return this->udp_socket_;
+}
+
+const ACE_SOCK_Dgram_Mcast &
+TAO_UIPMC_Connection_Handler::mcast_dgram (void)
+{
+  return this->mcast_socket_;
+}
+
+
+int
+TAO_UIPMC_Connection_Handler::open_handler (void *v)
+{
+  return this->open (v);
+}
+
+int
+TAO_UIPMC_Connection_Handler::open (void*)
+{
+  this->udp_socket_.open (this->local_addr_);
+
+  if(TAO_debug_level > 5)
+  {
+     ACE_DEBUG ((LM_DEBUG,
+                 ACE_TEXT("TAO (%P|%t) TAO_UIPMC_Connection_Handler::open, ")
+                 ACE_TEXT("listening on: <%s:%u>\n"),
+                 this->local_addr_.get_host_addr (),
+                 this->local_addr_.get_port_number ()));
+  }
+
+  this->using_mcast_ = 0;
+
+  // Set that the transport is now connected, if fails we return -1
+  // Use C-style cast b/c otherwise we get warnings on lots of
+  // compilers
+  if (!this->transport ()->post_open ((size_t) this->udp_socket_.get_handle ()))
+    return -1;
+
+  return 0;
+}
+
+int
+TAO_UIPMC_Connection_Handler::open_server (void)
+{
+  this->mcast_socket_.join (this->local_addr_);
+  if( TAO_debug_level > 5)
+  {
+     ACE_DEBUG ((LM_DEBUG,
+                 ACE_TEXT("TAO (%P|%t) TAO_UIPMC_Connection_Handler::open_server, ")
+                 ACE_TEXT("subcribed to multicast group at %s:%d\n"),
+                 this->local_addr_.get_host_addr (),
+                 this->local_addr_.get_port_number ()
+               ));
+  }
+
+  this->transport ()->id ((size_t) this->mcast_socket_.get_handle ());
+  this->using_mcast_ = 1;
+
+  return 0;
+}
+
+int
+TAO_UIPMC_Connection_Handler::resume_handler (void)
+{
+  return ACE_Event_Handler::ACE_APPLICATION_RESUMES_HANDLER;
+}
+
+int
+TAO_UIPMC_Connection_Handler::close_connection (void)
+{
+  return this->close_connection_eh (this);
+}
+
+int
+TAO_UIPMC_Connection_Handler::handle_input (ACE_HANDLE h)
+{
+  return this->handle_input_eh (h, this);
+}
+
+int
+TAO_UIPMC_Connection_Handler::handle_output (ACE_HANDLE handle)
+{
+  int result =
+    this->handle_output_eh (handle, this);
+
+  if (result == -1)
+    {
+      this->close_connection ();
+      return 0;
+    }
+
+  return result;
+}
+
+int
+TAO_UIPMC_Connection_Handler::handle_timeout (const ACE_Time_Value &,
+                                              const void *)
+{
+  // Using this to ensure this instance will be deleted (if necessary)
+  // only after reset_state(). Without this, when this refcount==1 -
+  // the call to close() will cause a call to remove_reference() which
+  // will delete this. At that point this->reset_state() is in no
+  // man's territory and that causes SEGV on some platforms (Windows!)
+
+  TAO_Auto_Reference<TAO_UIPMC_Connection_Handler> safeguard (*this);
+
+  // NOTE: Perhaps not the best solution, as it feels like the upper
+  // layers should be responsible for this?
+
+  // We don't use this upcall for I/O.  This is only used by the
+  // Connector to indicate that the connection timedout.  Therefore,
+  // we should call close().
+  int ret = this->close ();
+  this->reset_state (TAO_LF_Event::LFS_TIMEOUT);
+  return ret;
+}
+
+int
+TAO_UIPMC_Connection_Handler::handle_close (ACE_HANDLE,
+                                            ACE_Reactor_Mask)
+{
+  // No asserts here since the handler is registered with the Reactor
+  // and the handler ownership is given to the Reactor.  When the
+  // Reactor closes, it will call handle_close() on the handler.  It
+  // is however important to overwrite handle_close() to do nothing
+  // since the base class does too much.
+  return 0;
+}
+
+int
+TAO_UIPMC_Connection_Handler::close (u_long)
+{
+  this->state_changed (TAO_LF_Event::LFS_CONNECTION_CLOSED,
+           this->orb_core ()->leader_follower ());
+  this->transport ()->remove_reference ();
+  return 0;
+}
+
+int
+TAO_UIPMC_Connection_Handler::release_os_resources (void)
+{
+  return this->peer().close ();
+}
+
+int
+TAO_UIPMC_Connection_Handler::add_transport_to_cache (void)
+{
+  ACE_INET_Addr addr;
+
+  // This function is called by the acceptor to add this
+  // transport to the transport cache.  This is really
+  // important for proper shutdown.  The address used
+  // is irrelevent, since UIPMC is connectionless.
+
+  // Construct a UIPMC_Endpoint object.
+  TAO_UIPMC_Endpoint endpoint (addr);
+
+  // Construct a property object
+  TAO_Base_Transport_Property prop (&endpoint);
+
+  // Add the handler to Cache
+  return this->orb_core ()->lane_resources ()
+          .transport_cache ().cache_transport (&prop,
+                                               this->transport ());
+}
+
+TAO_END_VERSIONED_NAMESPACE_DECL