ChangeLogTag: Mon May 7 11:46:47 UTC 2007 Chad Elliott <elliott_c@ociweb.com>

4 files changed, 485 insertions, 0 deletions

diff --git a/TAO/tao/IORManipulation/IORManip_Filter.cpp b/TAO/tao/IORManipulation/IORManip_Filter.cpp
new file mode 100644
index 00000000000..3f70b119f08
--- /dev/null
+++ b/TAO/tao/IORManipulation/IORManip_Filter.cpp
@@ -0,0 +1,82 @@
+// $Id$
+
+#include "tao/IORManipulation/IORManip_Filter.h"
+#include "tao/IORManipulation/IORManip_Loader.h"
+#include "tao/MProfile.h"
+#include "tao/ORB_Core.h"
+#include "tao/Stub.h"
+
+ACE_RCSID (IORManip_Filter, IORManip_Filter, "$Id$")
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+TAO_IORManip_Filter::TAO_IORManip_Filter (void)
+{
+}
+
+
+TAO_IORManip_Filter::~TAO_IORManip_Filter (void)
+{
+}
+
+
+CORBA::Object_ptr
+TAO_IORManip_Filter::sanitize_profiles (CORBA::Object_ptr object,
+                                        TAO_Profile* profile)
+{
+  return this->sanitize (object, profile);
+}
+
+
+CORBA::Object_ptr
+TAO_IORManip_Filter::sanitize (CORBA::Object_ptr object,
+                               TAO_Profile* guideline)
+{
+  TAO_MProfile profiles = object->_stubobj ()->base_profiles ();
+  TAO_MProfile new_profiles (profiles.profile_count ());
+  TAO_Profile* profile = 0;
+
+  while ((profile = profiles.get_next ()) != 0)
+    {
+      // Call the filter implementation
+      this->filter_and_add (profile, new_profiles, guideline);
+    }
+
+  // The remainder of this code has been lifted from IORManipulation.cpp
+  CORBA::String_var id =
+      CORBA::string_dup (object->_stubobj ()->type_id.in ());
+
+  TAO_ORB_Core *orb_core = object->_stubobj ()->orb_core ();
+  if (orb_core == 0)
+    orb_core = TAO_ORB_Core_instance ();
+
+  TAO_Stub *stub = orb_core->create_stub (id.in (),  // give the id string to stub
+                                          new_profiles);
+
+  // Make the stub memory allocation exception safe for the duration
+  // of this method.
+  TAO_Stub_Auto_Ptr safe_stub (stub);
+
+  // Create the CORBA level proxy
+  CORBA::Object_ptr temp_obj = CORBA::Object::_nil ();
+  ACE_NEW_THROW_EX (temp_obj,
+                    CORBA::Object (safe_stub.get ()),
+                    CORBA::NO_MEMORY ());
+
+  CORBA::Object_var new_obj = temp_obj;
+
+
+  // Clean up in case of errors.
+  if (CORBA::is_nil (new_obj.in ()))
+    {
+      throw TAO_IOP::Invalid_IOR ();
+    }
+
+  // Release ownership of the pointers protected by the auto_ptrs since
+  // they no longer need to be protected by this point.
+  stub = safe_stub.release ();
+
+  return new_obj._retn ();
+}
+
+TAO_END_VERSIONED_NAMESPACE_DECL
diff --git a/TAO/tao/IORManipulation/IORManip_Filter.h b/TAO/tao/IORManipulation/IORManip_Filter.h
new file mode 100644
index 00000000000..e5b98736a97
--- /dev/null
+++ b/TAO/tao/IORManipulation/IORManip_Filter.h
@@ -0,0 +1,68 @@
+// $Id$
+
+// =========================================================================
+//
+// = LIBRARY
+//    TAO
+//
+// = FILENAME
+//    IORManip_Filter.h
+//
+// = AUTHOR
+//    Chad Elliott <elliott_c@ociweb.com>
+//
+// =========================================================================
+
+#ifndef TAO_IORMANIP_FILTER_H
+#define TAO_IORMANIP_FILTER_H
+#include /**/ "ace/pre.h"
+
+#include "tao/GIOP_Message_State.h"
+
+#if !defined (ACE_LACKS_PRAGMA_ONCE)
+# pragma once
+#endif /* ACE_LACKS_PRAGMA_ONCE */
+
+#include "tao/corbafwd.h"
+#include "tao/IORManipulation/ior_manip_export.h"
+#include "tao/Environment.h"
+#include "ace/CORBA_macros.h"
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+class TAO_Profile;
+class TAO_MProfile;
+
+class TAO_IORManip_Export TAO_IORManip_Filter
+{
+public:
+  /// Constructor.
+  TAO_IORManip_Filter (void);
+
+  /// Destructor.
+  virtual ~TAO_IORManip_Filter (void);
+
+  /// Perform filtering using the profile passed in as a guide.
+  /// If no profile is provided, filter using the profile_matches() method.
+  CORBA::Object_ptr sanitize_profiles (CORBA::Object_ptr object,
+                                       TAO_Profile* profile = 0);
+
+protected:
+
+  /// This will be the bulk of the filtering code.
+  virtual void filter_and_add (TAO_Profile* profile,
+                               TAO_MProfile& profiles,
+                               TAO_Profile* guideline = 0) = 0;
+
+
+private:
+
+  /// The sanitize_profiles() methods call this to do the work.
+  CORBA::Object_ptr sanitize (CORBA::Object_ptr object,
+                              TAO_Profile* profile);
+};
+
+TAO_END_VERSIONED_NAMESPACE_DECL
+
+#include /**/ "ace/post.h"
+#endif /* TAO_IORMANIP_FILTER_H */
diff --git a/TAO/tao/IORManipulation/IORManip_IIOP_Filter.cpp b/TAO/tao/IORManipulation/IORManip_IIOP_Filter.cpp
new file mode 100644
index 00000000000..d159130192b
--- /dev/null
+++ b/TAO/tao/IORManipulation/IORManip_IIOP_Filter.cpp
@@ -0,0 +1,247 @@
+// $Id$
+
+#include "tao/IORManipulation/IORManip_IIOP_Filter.h"
+#include "tao/IORManipulation/IORManip_Loader.h"
+#include "tao/IIOP_Profile.h"
+#include "tao/MProfile.h"
+
+ACE_RCSID (IORManip_IIOP_Filter, IORManip_IIOP_Filter, "$Id$")
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+TAO_IORManip_IIOP_Filter::TAO_IORManip_IIOP_Filter (void)
+{
+}
+
+
+TAO_IORManip_IIOP_Filter::~TAO_IORManip_IIOP_Filter (void)
+{
+}
+
+
+CORBA::Boolean
+TAO_IORManip_IIOP_Filter::compare_profile_info (
+                        const TAO_IORManip_IIOP_Filter::Profile_Info& left,
+                        const TAO_IORManip_IIOP_Filter::Profile_Info& right)
+{
+  return (left.version_.major == right.version_.major &&
+          left.version_.minor == right.version_.minor &&
+          left.port_          == right.port_          &&
+          left.host_name_     == right.host_name_);
+}
+
+
+CORBA::Boolean
+TAO_IORManip_IIOP_Filter::profile_info_matches (
+                            const TAO_IORManip_IIOP_Filter::Profile_Info&)
+{
+  return true;
+}
+
+
+void
+TAO_IORManip_IIOP_Filter::filter_and_add (TAO_Profile* profile,
+                                          TAO_MProfile& new_profiles,
+                                          TAO_Profile* guideline)
+{
+  TAO_IORManip_IIOP_Filter::Profile_Info ginfo;
+  TAO_IORManip_IIOP_Filter::Profile_Info pinfo;
+  TAO::IIOPEndpointSequence endpoints;
+
+  this->fill_profile_info (guideline, ginfo);
+  this->get_endpoints (profile, endpoints);
+  if (endpoints.length () == 0)
+    {
+      CORBA::Boolean matches = false;
+      this->fill_profile_info (profile, pinfo);
+
+      if (guideline == 0)
+        {
+          matches = this->profile_info_matches (pinfo);
+        }
+      else
+        {
+          // Compare the current profile with the guideline profile
+          matches = this->compare_profile_info (pinfo, ginfo);
+        }
+
+      if (matches)
+        {
+          if (new_profiles.add_profile (profile) == -1)
+            {
+              throw CORBA::NO_MEMORY ();
+            }
+        }
+    }
+  else
+    {
+      // Create a new profile with just the
+      // components we are looking for
+      TAO_IIOP_Profile* new_profile =
+          this->create_profile (profile);
+
+      // Set pinfo to match the current profile
+      this->fill_profile_info (profile, pinfo);
+
+      // Add each endpoint if it matches.  Begin from the end
+      // of the sequence to preserve endpoint order, since <add_endpoint>
+      // method reverses the order of endpoints in the list.
+      for (CORBA::Long i = endpoints.length () - 1; i >= 0; i--) {
+        // Set pinfo host name to the current endpoint host name
+        pinfo.host_name_ = endpoints[i].host.in ();
+        pinfo.port_ = endpoints[i].port;
+
+        CORBA::Boolean matches = false;
+        if (guideline == 0)
+          {
+            matches = this->profile_info_matches (pinfo);
+          }
+        else
+          {
+            // Compare the current profile with the guideline profile
+            matches = this->compare_profile_info (pinfo, ginfo);
+          }
+
+        if (matches)
+          {
+            // Set the main endpoint on the profile
+            if (i == 0)
+              {
+                TAO_IIOP_Endpoint* ep = dynamic_cast<TAO_IIOP_Endpoint*> (
+                                          new_profile->endpoint ());
+                if (ep == 0)
+                  {
+                    new_profile->_decr_refcnt ();
+                    return;
+                  }
+                else
+                  {
+                    ep->host (CORBA::string_dup (endpoints[i].host));
+                    ep->port (endpoints[i].port);
+                    ep->priority (endpoints[i].priority);
+                  }
+              }
+            else
+              {
+                TAO_IIOP_Endpoint *endpoint = 0;
+                ACE_NEW (endpoint,
+                         TAO_IIOP_Endpoint (endpoints[i].host,
+                                            endpoints[i].port,
+                                            endpoints[i].priority));
+                if (endpoint == 0)
+                  {
+                    new_profile->_decr_refcnt ();
+                    return;
+                  }
+
+                new_profile->add_endpoint (endpoint);
+              }
+          }
+      }
+
+      if (new_profiles.add_profile (new_profile) == -1)
+        {
+          throw CORBA::NO_MEMORY ();
+        }
+
+      new_profile->encode_endpoints ();
+
+      // After adding the profile to the MProfile,
+      // we must decrement the reference to avoid a memory leak
+      new_profile->_decr_refcnt ();
+    }
+}
+
+
+int
+TAO_IORManip_IIOP_Filter::fill_profile_info (
+                             TAO_Profile* profile,
+                             TAO_IORManip_IIOP_Filter::Profile_Info& pinfo)
+{
+  static const int host_length = 384;
+  int status = 0;
+  if (profile != 0)
+    {
+      char host[host_length] = "";
+      if (profile->endpoint ()->addr_to_string (host, host_length) != -1)
+        {
+          char* delim = ACE_OS::strchr (host, ':');
+          if (delim != 0)
+            {
+              *delim = '\0';
+              pinfo.port_ = ACE_OS::atoi (delim + 1);
+              status = 1;
+            }
+        }
+
+      pinfo.host_name_ = host;
+      pinfo.version_ = profile->version ();
+    }
+
+  return status;
+}
+
+
+int
+TAO_IORManip_IIOP_Filter::get_endpoints (TAO_Profile* profile,
+                                         TAO::IIOPEndpointSequence& endpoints)
+{
+  // Reset the endpoints
+  endpoints.length (0);
+
+  // Get the endpoints tagged component
+  const TAO_Tagged_Components& comps = profile->tagged_components ();
+  IOP::TaggedComponent tagged_component;
+  tagged_component.tag = TAO_TAG_ENDPOINTS;
+  comps.get_component (tagged_component);
+
+  // Prepare the CDR for endpoint extraction
+  const CORBA::Octet *buf =
+    tagged_component.component_data.get_buffer ();
+
+  TAO_InputCDR in_cdr (reinterpret_cast<const char*> (buf),
+                       tagged_component.component_data.length ());
+
+  // Extract the Byte Order.
+  CORBA::Boolean byte_order;
+  if ((in_cdr >> ACE_InputCDR::to_boolean (byte_order)) == 0)
+    return 0;
+  in_cdr.reset_byte_order (static_cast<int> (byte_order));
+
+  // Extract endpoints sequence.
+  if ((in_cdr >> endpoints) == 0)
+    return 0;
+
+  return 1;
+}
+
+
+TAO_IIOP_Profile*
+TAO_IORManip_IIOP_Filter::create_profile (TAO_Profile* profile)
+{
+  ACE_INET_Addr addr;
+  TAO_IIOP_Profile* new_profile = 0;
+  ACE_NEW_THROW_EX (new_profile,
+                   TAO_IIOP_Profile (addr,
+                                     profile->object_key (),
+                                     profile->version (),
+                                     profile->orb_core ()),
+                   CORBA::NO_MEMORY (
+                     CORBA::SystemException::_tao_minor_code (
+                       0,
+                       ENOMEM),
+                     CORBA::COMPLETED_NO));
+
+  // Copy all of the tagged components
+  const TAO_Tagged_Components& comps = profile->tagged_components ();
+  new_profile->tagged_components () = comps;
+
+  // Set the endpoints component to an empty list
+  IOP::TaggedComponent tagged_component;
+  tagged_component.tag = TAO_TAG_ENDPOINTS;
+  new_profile->tagged_components ().set_component (tagged_component);
+
+  return new_profile;
+}
+
+TAO_END_VERSIONED_NAMESPACE_DECL
diff --git a/TAO/tao/IORManipulation/IORManip_IIOP_Filter.h b/TAO/tao/IORManipulation/IORManip_IIOP_Filter.h
new file mode 100644
index 00000000000..06e602b1e8c
--- /dev/null
+++ b/TAO/tao/IORManipulation/IORManip_IIOP_Filter.h
@@ -0,0 +1,88 @@
+// $Id$
+
+// =========================================================================
+//
+// = LIBRARY
+//    TAO
+//
+// = FILENAME
+//    IORManip_IIOP_Filter.h
+//
+// = AUTHOR
+//    Chad Elliott <elliott_c@ociweb.com>
+//
+// =========================================================================
+
+#ifndef TAO_IORMANIP_IIOP_FILTER_H
+#define TAO_IORMANIP_IIOP_FILTER_H
+#include /**/ "ace/pre.h"
+
+#include "IORManip_Filter.h"
+
+#if !defined (ACE_LACKS_PRAGMA_ONCE)
+# pragma once
+#endif /* ACE_LACKS_PRAGMA_ONCE */
+
+#include "tao/IIOP_EndpointsC.h"
+#include "ace/SString.h"
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+class TAO_IIOP_Profile;
+namespace TAO
+{
+  class IIOPEndpointSequence;
+}
+
+class TAO_IORManip_Export TAO_IORManip_IIOP_Filter: public TAO_IORManip_Filter
+{
+public:
+  struct Profile_Info
+  {
+    ACE_CString host_name_;
+    TAO_GIOP_Message_Version version_;
+    CORBA::UShort port_;
+  };
+
+  /// Constructor.
+  TAO_IORManip_IIOP_Filter (void);
+
+  /// Destructor.
+  virtual ~TAO_IORManip_IIOP_Filter (void);
+
+  /// Compares the profile to the profile info.
+  virtual CORBA::Boolean compare_profile_info (
+                         const TAO_IORManip_IIOP_Filter::Profile_Info& left,
+                         const TAO_IORManip_IIOP_Filter::Profile_Info& right);
+
+  /// Empty virtual method to match on the profile info.
+  /// Users must provide an implementation to use the first
+  /// form of sanitize_profiles().
+  virtual CORBA::Boolean profile_info_matches (
+                         const TAO_IORManip_IIOP_Filter::Profile_Info& pinfo);
+
+protected:
+
+  /// This is the bulk of the filtering code.
+  virtual void filter_and_add (TAO_Profile* profile,
+                               TAO_MProfile& profiles,
+                               TAO_Profile* guideline = 0);
+
+private:
+
+  /// Fill in the Profile_Info with information from the profile.
+  int fill_profile_info (TAO_Profile* profile,
+                         TAO_IORManip_IIOP_Filter::Profile_Info& pinfo);
+
+  /// Get the endpoint sequence from the profile.
+  int get_endpoints (TAO_Profile* profile,
+                     TAO::IIOPEndpointSequence& endpoints);
+
+  /// Allocate a new IIOP Profile based on the profile passed in.
+  TAO_IIOP_Profile* create_profile (TAO_Profile* profile);
+};
+
+TAO_END_VERSIONED_NAMESPACE_DECL
+
+#include /**/ "ace/post.h"
+#endif /* TAO_IORMANIP_IIOP_FILTER_H */