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 using namespace xmlencryption;
  32 using opensaml::saml2::EncryptedElementType;
  33 using namespace std;
  34
  35 opensaml::EncryptedKeyResolver::EncryptedKeyResolver(const EncryptedElementType& ref) : m_ref(ref)
  36 {
  37 }
  38
  39 opensaml::EncryptedKeyResolver::~EncryptedKeyResolver()
  40 {
  41 }
  42
  43 const EncryptedKey* opensaml::EncryptedKeyResolver::resolveKey(const EncryptedData& encryptedData, const XMLCh* recipient) const
  44 {
  45     const EncryptedKey* base = xmlencryption::EncryptedKeyResolver::resolveKey(encryptedData, recipient);
  46     if (base)
  47         return base;
  48
  49     const vector<EncryptedKey*>& keys=m_ref.getEncryptedKeys();
  50     for (vector<EncryptedKey*>::const_iterator i=keys.begin(); i!=keys.end(); i++) {
  51         if ((*i)->getRecipient() == nullptr)
  52             return (*i);
  53         else if (XMLString::equals(recipient,(*i)->getRecipient()))
  54             return (*i);
  55     }
  56
  57     return nullptr;
  58 }