Fixes to constants, allow sequences of condition rules, new policy rule for Delegatio...
[shibboleth/cpp-opensaml.git] / saml / saml2 / profile / impl / DelegationRestrictionRule.cpp
diff --git a/saml/saml2/profile/impl/DelegationRestrictionRule.cpp b/saml/saml2/profile/impl/DelegationRestrictionRule.cpp
new file mode 100644 (file)
index 0000000..5bc3bea
--- /dev/null
@@ -0,0 +1,171 @@
+/*
+ *  Copyright 2009 Internet2
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * DelegationRestrictionRule.cpp
+ *
+ * SAML DelegationRestriction SecurityPolicyRule
+ */
+
+#include "internal.h"
+#include "exceptions.h"
+#include "binding/SecurityPolicyRule.h"
+#include "saml2/core/Assertions.h"
+#include "util/SAMLConstants.h"
+
+#include <xmltooling/logging.h>
+
+using namespace opensaml::saml2;
+using namespace opensaml;
+using namespace xmltooling::logging;
+using namespace xmltooling;
+using namespace std;
+
+namespace opensaml {
+    namespace saml2 {
+        class SAML_DLLLOCAL DelegationRestrictionRule : public SecurityPolicyRule
+        {
+        public:
+            DelegationRestrictionRule(const DOMElement* e);
+
+            virtual ~DelegationRestrictionRule() {
+                for_each(m_delegates.begin(), m_delegates.end(), xmltooling::cleanup<Delegate>());
+            }
+            const char* getType() const {
+                return DELEGATION_POLICY_RULE;
+            }
+            bool evaluate(const XMLObject& message, const GenericRequest* request, SecurityPolicy& policy) const;
+
+        private:
+            vector<Delegate*> m_delegates;
+            enum {
+                MATCH_ANY,
+                MATCH_NEWEST,
+                MATCH_OLDEST
+            } m_match;
+        };
+
+        SecurityPolicyRule* SAML_DLLLOCAL DelegationRestrictionRuleFactory(const DOMElement* const & e)
+        {
+            return new DelegationRestrictionRule(e);
+        }
+
+        class SAML_DLLLOCAL _isSameDelegate : public binary_function<const Delegate*,const Delegate*,bool>,
+            public unary_function<const Delegate*,bool>
+        {
+            const Delegate* m_operand;
+            bool isSameFormat(const XMLCh* f1, const XMLCh* f2) const {
+                if (!f1 || !*f1)
+                    f1 = NameIDType::UNSPECIFIED;
+                if (!f2 || !*f2)
+                    f2 = NameIDType::UNSPECIFIED;
+                return XMLString::equals(f1, f2);
+            }
+            bool matches(const NameID* n1, const NameID* n2) const {
+                return (isSameFormat(n1->getFormat(), n2->getFormat()) &&
+                        XMLString::equals(n1->getName(), n2->getName()) &&
+                        XMLString::equals(n1->getNameQualifier(), n2->getNameQualifier()) &&
+                        XMLString::equals(n1->getSPNameQualifier(), n2->getSPNameQualifier()));
+            }
+        public:
+            _isSameDelegate() : m_operand(NULL) {}
+            _isSameDelegate(const Delegate* d) : m_operand(d) {}
+
+            // d1 is the input from the message, d2 is from the policy
+            bool operator()(const Delegate* d1, const Delegate* d2) const {
+                if (!d1->getNameID()) {
+                    Category::getInstance(SAML_LOGCAT".SecurityPolicyRule.DelegationRestriction").error(
+                        "rule doesn't support evaluation of BaseID or EncryptedID in a Delegate"
+                        );
+                    return false;
+                }
+                if (!d2->getConfirmationMethod() || XMLString::equals(d1->getConfirmationMethod(), d2->getConfirmationMethod())) {
+                    return matches(d1->getNameID(), d2->getNameID());
+                }
+                return false;
+            }
+
+            // d is from the policy
+            bool operator()(const Delegate* d) const {
+                return this->operator()(m_operand, d);
+            }
+        };
+
+        static XMLCh match[] =  UNICODE_LITERAL_5(m,a,t,c,h);
+        static XMLCh any[] =    UNICODE_LITERAL_8(a,n,y,O,r,d,e,r);
+        static XMLCh newest[] = UNICODE_LITERAL_6(n,e,w,e,s,t);
+        static XMLCh oldest[] = UNICODE_LITERAL_6(o,l,d,e,s,t);
+
+    }
+};
+
+DelegationRestrictionRule::DelegationRestrictionRule(const DOMElement* e) : m_match(MATCH_ANY)
+{
+    if (e) {
+        const XMLCh* m = e->getAttributeNS(NULL, match);
+        if (XMLString::equals(m, newest))
+            m_match = MATCH_NEWEST;
+        else if (XMLString::equals(m, oldest))
+            m_match = MATCH_OLDEST;
+        else if (m && *m && !XMLString::equals(m, any))
+            throw SecurityPolicyException("Invalid value for \"match\" attribute in Delegation rule.");
+
+        try {
+            DOMElement* d = XMLHelper::getFirstChildElement(e, samlconstants::SAML20_DELEGATION_CONDITION_NS, Delegate::LOCAL_NAME);
+            while (d) {
+                auto_ptr<XMLObject> wrapper(XMLObjectBuilder::buildOneFromElement(d));
+                Delegate* down = dynamic_cast<Delegate*>(wrapper.get());
+                if (down) {
+                    m_delegates.push_back(down);
+                    wrapper.release();
+                }
+                d = XMLHelper::getNextSiblingElement(d, samlconstants::SAML20_DELEGATION_CONDITION_NS, Delegate::LOCAL_NAME);
+            }
+        }
+        catch (exception&) {
+            for_each(m_delegates.begin(), m_delegates.end(), xmltooling::cleanup<Delegate>());
+            throw;
+        }
+    }
+}
+
+bool DelegationRestrictionRule::evaluate(const XMLObject& message, const GenericRequest* request, SecurityPolicy& policy) const
+{
+    const DelegationRestrictionType* drt=dynamic_cast<const DelegationRestrictionType*>(&message);
+    if (!drt)
+        return false;
+
+    // If we have no embedded Delegates, the condition evaluates to true.
+    if (m_delegates.empty())
+        return true;
+
+    const vector<Delegate*>& dels = drt->getDelegates();
+    if (m_match == MATCH_ANY) {
+        // Each Delegate in the condition MUST match an embedded Delegate.
+        for (vector<Delegate*>::const_iterator d1 = dels.begin(); d1 != dels.end(); ++d1) {
+            if (find_if(m_delegates.begin(), m_delegates.end(), _isSameDelegate(*d1)) == m_delegates.end())
+                return false;
+        }
+    }
+    else if (m_match == MATCH_OLDEST) {
+        return (search(dels.begin(), dels.end(), m_delegates.begin(), m_delegates.end(), _isSameDelegate()) == dels.begin());
+    }
+    else if (m_match == MATCH_NEWEST) {
+        return (search(dels.rbegin(), dels.rend(), m_delegates.begin(), m_delegates.end(), _isSameDelegate()) == dels.rbegin());
+    }
+
+    return true;
+}