1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
|
/*
* Copyright 2001-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.
*/
/**
* CommonDomainCookie.cpp
*
* Helper class for maintaining discovery cookie.
*/
#include "internal.h"
#include "util/CommonDomainCookie.h"
#include <xercesc/util/Base64.hpp>
#include <xsec/framework/XSECDefs.hpp>
#include <xmltooling/XMLToolingConfig.h>
#include <xmltooling/util/URLEncoder.h>
using namespace opensaml;
using namespace xmltooling;
using namespace std;
const char CommonDomainCookie::CDCName[] = "_saml_idp";
CommonDomainCookie::CommonDomainCookie(const char* cookie)
{
if (!cookie)
return;
// Copy it so we can URL-decode it.
char* b64=strdup(cookie);
XMLToolingConfig::getConfig().getURLEncoder()->decode(b64);
// Chop it up and save off elements.
vector<string> templist;
char* ptr=b64;
while (*ptr) {
while (*ptr && isspace(*ptr)) ptr++;
char* end=ptr;
while (*end && !isspace(*end)) end++;
templist.push_back(string(ptr,end-ptr));
ptr=end;
}
free(b64);
// Now Base64 decode the list.
xsecsize_t len;
for (vector<string>::iterator i=templist.begin(); i!=templist.end(); ++i) {
XMLByte* decoded=Base64::decode(reinterpret_cast<const XMLByte*>(i->c_str()),&len);
if (decoded && *decoded) {
m_list.push_back(reinterpret_cast<char*>(decoded));
#ifdef OPENSAML_XERCESC_HAS_XMLBYTE_RELEASE
XMLString::release(&decoded);
#else
XMLString::release((char**)&decoded);
#endif
}
}
}
CommonDomainCookie::~CommonDomainCookie()
{
}
const vector<string>& CommonDomainCookie::get() const
{
return m_list;
}
const char* CommonDomainCookie::set(const char* entityID)
{
// First scan the list for this IdP.
for (vector<string>::iterator i=m_list.begin(); i!=m_list.end(); i++) {
if (*i == entityID) {
m_list.erase(i);
break;
}
}
// Append it to the end.
m_list.push_back(entityID);
// Now rebuild the delimited list.
xsecsize_t len;
string delimited;
for (vector<string>::const_iterator j=m_list.begin(); j!=m_list.end(); j++) {
if (!delimited.empty()) delimited += ' ';
XMLByte* b64=Base64::encode(reinterpret_cast<const XMLByte*>(j->c_str()),j->length(),&len);
XMLByte *pos, *pos2;
for (pos=b64, pos2=b64; *pos2; pos2++)
if (isgraph(*pos2))
*pos++=*pos2;
*pos=0;
delimited += reinterpret_cast<char*>(b64);
#ifdef OPENSAML_XERCESC_HAS_XMLBYTE_RELEASE
XMLString::release(&b64);
#else
XMLString::release((char**)&b64);
#endif
}
m_encoded=XMLToolingConfig::getConfig().getURLEncoder()->encode(delimited.c_str());
return m_encoded.c_str();
}
|