saml/encryption/EncryptedKeyResolver.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  * EncryptedKeyResolver.cpp
  23  *
  24  * SAML-specific encrypted key resolver.
  25  */
  26
  27 #include "internal.h"
  28 #include "encryption/EncryptedKeyResolver.h"
  29 #include "saml2/core/Assertions.h"
  30
  31 #include <boost/lambda/bind.hpp>
  32 #include <boost/lambda/lambda.hpp>
  33
  34 using namespace xmlencryption;
  35 using opensaml::saml2::EncryptedElementType;
  36 using namespace boost::lambda;
  37 using namespace boost;
  38 using namespace std;
  39
  40 opensaml::EncryptedKeyResolver::EncryptedKeyResolver(const EncryptedElementType& ref) : m_ref(ref)
  41 {
  42 }
  43
  44 opensaml::EncryptedKeyResolver::~EncryptedKeyResolver()
  45 {
  46 }
  47
  48 const EncryptedKey* opensaml::EncryptedKeyResolver::resolveKey(const EncryptedData& encryptedData, const XMLCh* recipient) const
  49 {
  50     const EncryptedKey* base = xmlencryption::EncryptedKeyResolver::resolveKey(encryptedData, recipient);
  51     if (base)
  52         return base;
  53
  54     static bool (*equal_fn)(const XMLCh*, const XMLCh*) = &XMLString::equals;
  55
  56     // Look for first match that has no Recipient attribute, or matches the input recipient.
  57     // Using XMLString::equals allows for both to be NULL and still match.
  58     vector<EncryptedKey*>::const_iterator k = find_if(
  59         m_ref.getEncryptedKeys().begin(), m_ref.getEncryptedKeys().end(),
  60         (lambda::bind(&EncryptedKey::getRecipient, _1) == nullptr ||
  61             lambda::bind(equal_fn, recipient, lambda::bind(&EncryptedKey::getRecipient, _1)))
  62         );
  63     return (k != m_ref.getEncryptedKeys().end()) ? (*k) : nullptr;
  64 }