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
|
/**
* Licensed to the University Corporation for Advanced Internet
* Development, Inc. (UCAID) under one or more contributor license
* agreements. See the NOTICE file distributed with this work for
* additional information regarding copyright ownership.
*
* UCAID licenses this file to you 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.
*/
#include "internal.h"
#include <saml/SAMLConfig.h>
#include <saml/saml1/binding/SAMLArtifactType0001.h>
#include <saml/saml1/binding/SAMLArtifactType0002.h>
#include <saml/saml2/binding/SAML2ArtifactType0004.h>
#include <xmltooling/security/SecurityHelper.h>
using namespace opensaml::saml1p;
using namespace opensaml::saml2p;
using namespace opensaml;
using namespace std;
class SAMLArtifactCreationTest : public CxxTest::TestSuite
{
public:
string providerIdStr;
string handle;
void setUp() {
if (handle.empty()) {
providerIdStr = "https://idp.org/SAML";
SAMLConfig::getConfig().generateRandomBytes(handle,SAMLArtifactType0001::HANDLE_LENGTH);
}
}
void testSAMLArtifactType0001(void) {
SAMLConfig& conf=SAMLConfig::getConfig();
string sourceId;
conf.generateRandomBytes(sourceId,SAMLArtifactType0001::SOURCEID_LENGTH);
SAMLArtifactType0001 artifact1(sourceId,handle);
//printResults(artifact1);
SAMLArtifactType0001 artifact2(
SecurityHelper::doHash("SHA1", providerIdStr.data(), providerIdStr.length(), false), handle
);
//printResults(artifact2,providerIdStr.c_str());
}
void testSAMLArtifactType0002(void) {
SAMLArtifactType0002 artifact(providerIdStr,handle);
//printResults(artifact,providerIdStr.c_str());
}
void testSAMLArtifactType0004(void) {
SAML2ArtifactType0004 artifact(
SecurityHelper::doHash("SHA1", providerIdStr.data(), providerIdStr.length(), false), 666, handle
);
//printResults(artifact,providerIdStr.c_str());
}
void printResults(SAMLArtifact& artifact, const char* str=nullptr) {
// print heading:
cout << "Artifact Type " << SAMLArtifact::toHex(artifact.getTypeCode());
cout << " (size = " << artifact.getBytes().size() << ")" << endl;
// print URI:
if (str) {
cout << "URI: " << str << endl;
}
else {
cout << "URI: NONE" << endl;
}
// print hex-encoded artifact:
cout << "Hex: " << SAMLArtifact::toHex(artifact.getBytes()) << endl;
// print base64-encoded artifact:
cout << "Base64: " << artifact.encode() << endl;
// print ruler:
cout << " ----------------------------------------------------------------------" << endl;
cout << " 1234567890123456789012345678901234567890123456789012345678901234567890" << endl;
cout << " 1 2 3 4 5 6 7" << endl;
cout << " ----------------------------------------------------------------------" << endl;
}
};
|