https://issues.shibboleth.net/jira/browse/SSPCPP-342
[shibboleth/cpp-sp.git] / shibsp / attribute / filtering / impl / AttributeIssuerEntityAttributeFunctor.cpp
1 /**
2  * Licensed to the University Corporation for Advanced Internet
3  * Development, Inc. (UCAID) under one or more contributor license
4  * agreements. See the NOTICE file distributed with this work for
5  * additional information regarding copyright ownership.
6  *
7  * UCAID licenses this file to you under the Apache License,
8  * Version 2.0 (the "License"); you may not use this file except
9  * in compliance with the License. You may obtain a copy of the
10  * License at
11  *
12  * http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing,
15  * software distributed under the License is distributed on an
16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
17  * either express or implied. See the License for the specific
18  * language governing permissions and limitations under the License.
19  */
20
21 /**
22  * AttributeIssuerEntityAttributeFunctor.cpp
23  * 
24  * A match function that checks if the attribute issuer contains an entity attribute with the
25  * specified value or regex.
26  */
27
28 #include "internal.h"
29 #include "exceptions.h"
30 #include "attribute/filtering/FilteringContext.h"
31 #include "attribute/filtering/FilterPolicyContext.h"
32 #include "attribute/filtering/MatchFunctor.h"
33
34 #include <saml/SAMLConfig.h>
35 #include <saml/saml2/metadata/EntityMatcher.h>
36 #include <saml/saml2/metadata/Metadata.h>
37
38 using namespace opensaml::saml2md;
39 using opensaml::SAMLConfig;
40
41 namespace shibsp {
42
43     /**
44      * A match function that checks if the attribute issuer contains an entity attribute with the
45      * specified value or regex.
46      */
47     class SHIBSP_DLLLOCAL AttributeIssuerEntityAttributeFunctor : public MatchFunctor
48     {
49         boost::scoped_ptr<EntityMatcher> m_matcher;
50     public:
51         AttributeIssuerEntityAttributeFunctor(const DOMElement* e)
52             : m_matcher(SAMLConfig::getConfig().EntityMatcherManager.newPlugin(ENTITYATTR_ENTITY_MATCHER, e)) {
53         }
54
55         bool evaluatePolicyRequirement(const FilteringContext& filterContext) const {
56             if (filterContext.getAttributeIssuerMetadata()) {
57                 const EntityDescriptor* entity = dynamic_cast<EntityDescriptor*>(filterContext.getAttributeIssuerMetadata()->getParent());
58                 if (entity)
59                     return m_matcher->matches(*entity);
60             }
61             return false;
62         }
63
64         bool evaluatePermitValue(const FilteringContext& filterContext, const Attribute& attribute, size_t index) const {
65             return evaluatePolicyRequirement(filterContext);
66         }
67     };
68
69     MatchFunctor* SHIBSP_DLLLOCAL AttributeIssuerEntityAttributeExactMatchFactory(const std::pair<const FilterPolicyContext*,const DOMElement*>& p)
70     {
71         return new AttributeIssuerEntityAttributeFunctor(p.second);
72     }
73
74     MatchFunctor* SHIBSP_DLLLOCAL AttributeIssuerEntityAttributeRegexMatchFactory(const std::pair<const FilterPolicyContext*,const DOMElement*>& p)
75     {
76         return new AttributeIssuerEntityAttributeFunctor(p.second);
77     }
78
79 };