225 lines
8.0 KiB
Java
225 lines
8.0 KiB
Java
/*
|
|
* Copyright 2005-2009 Sun Microsystems, Inc. All Rights Reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 only, as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
|
|
* CA 95054 USA or visit www.sun.com if you need additional information or
|
|
* have any questions.
|
|
*/
|
|
|
|
/**
|
|
* @test
|
|
* @bug 4635230 6365103 6366054 6824440
|
|
* @summary Basic unit tests for validating XML Signatures with JSR 105
|
|
* @compile -XDignore.symbol.file KeySelectors.java SignatureValidator.java
|
|
* X509KeySelector.java ValidationTests.java
|
|
* @run main ValidationTests
|
|
* @author Sean Mullan
|
|
*/
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.security.*;
|
|
import javax.xml.crypto.Data;
|
|
import javax.xml.crypto.KeySelector;
|
|
import javax.xml.crypto.OctetStreamData;
|
|
import javax.xml.crypto.URIDereferencer;
|
|
import javax.xml.crypto.URIReference;
|
|
import javax.xml.crypto.URIReferenceException;
|
|
import javax.xml.crypto.XMLCryptoContext;
|
|
import javax.xml.crypto.dsig.XMLSignatureException;
|
|
import javax.xml.crypto.dsig.XMLSignatureFactory;
|
|
|
|
/**
|
|
* This is a testcase to validate all "merlin-xmldsig-twenty-three"
|
|
* testcases from Baltimore
|
|
*/
|
|
public class ValidationTests {
|
|
|
|
private static SignatureValidator validator;
|
|
private final static String DIR = System.getProperty("test.src", ".");
|
|
private final static String DATA_DIR =
|
|
DIR + System.getProperty("file.separator") + "data";
|
|
private final static String KEYSTORE =
|
|
DATA_DIR + System.getProperty("file.separator") + "certs" +
|
|
System.getProperty("file.separator") + "xmldsig.jks";
|
|
private final static String STYLESHEET =
|
|
"http://www.w3.org/TR/xml-stylesheet";
|
|
private final static String STYLESHEET_B64 =
|
|
"http://www.w3.org/Signature/2002/04/xml-stylesheet.b64";
|
|
|
|
private final static String[] FILES = {
|
|
"signature-enveloped-dsa.xml",
|
|
"signature-enveloping-b64-dsa.xml",
|
|
"signature-enveloping-dsa.xml",
|
|
"signature-enveloping-rsa.xml",
|
|
"signature-enveloping-hmac-sha1.xml",
|
|
"signature-external-dsa.xml",
|
|
"signature-external-b64-dsa.xml",
|
|
"signature-retrievalmethod-rawx509crt.xml",
|
|
"signature-keyname.xml",
|
|
"signature-x509-crt-crl.xml",
|
|
"signature-x509-crt.xml",
|
|
"signature-x509-is.xml",
|
|
"signature-x509-ski.xml",
|
|
"signature-x509-sn.xml",
|
|
// "signature.xml",
|
|
"exc-signature.xml",
|
|
"sign-spec.xml"
|
|
};
|
|
|
|
static KeySelector skks;
|
|
static {
|
|
try {
|
|
skks =
|
|
new KeySelectors.SecretKeySelector("secret".getBytes("ASCII"));
|
|
} catch (Exception e) {
|
|
//should not occur
|
|
}
|
|
}
|
|
private final static KeySelector SKKS = skks;
|
|
private final static KeySelector KVKS =
|
|
new KeySelectors.KeyValueKeySelector();
|
|
private final static KeySelector CKS =
|
|
new KeySelectors.CollectionKeySelector(new File(DATA_DIR));
|
|
private final static KeySelector RXKS =
|
|
new KeySelectors.RawX509KeySelector();
|
|
private final static KeySelector XKS = null;
|
|
private final static KeySelector[] KEY_SELECTORS = {
|
|
KVKS,
|
|
KVKS,
|
|
KVKS,
|
|
KVKS,
|
|
SKKS,
|
|
KVKS,
|
|
KVKS,
|
|
CKS,
|
|
CKS,
|
|
RXKS,
|
|
RXKS,
|
|
CKS,
|
|
CKS,
|
|
CKS,
|
|
// XKS,
|
|
KVKS,
|
|
RXKS
|
|
};
|
|
private static URIDereferencer httpUd = null;
|
|
|
|
public static void main(String args[]) throws Exception {
|
|
httpUd = new HttpURIDereferencer();
|
|
|
|
validator = new SignatureValidator(new File(DATA_DIR));
|
|
|
|
boolean atLeastOneFailed = false;
|
|
for (int i=0; i < FILES.length; i++) {
|
|
System.out.println("Validating " + FILES[i]);
|
|
if (test_signature(FILES[i], KEY_SELECTORS[i])) {
|
|
System.out.println("PASSED");
|
|
} else {
|
|
System.out.println("FAILED");
|
|
atLeastOneFailed = true;
|
|
}
|
|
}
|
|
// test with reference caching enabled
|
|
System.out.println("Validating sign-spec.xml with caching enabled");
|
|
if (test_signature("sign-spec.xml", RXKS, true)) {
|
|
System.out.println("PASSED");
|
|
} else {
|
|
System.out.println("FAILED");
|
|
atLeastOneFailed = true;
|
|
}
|
|
|
|
System.out.println("Validating signature-enveloping-hmac-sha1-40.xml");
|
|
try {
|
|
test_signature("signature-enveloping-hmac-sha1-40.xml", SKKS, false);
|
|
System.out.println("FAILED");
|
|
atLeastOneFailed = true;
|
|
} catch (XMLSignatureException xse) {
|
|
System.out.println(xse.getMessage());
|
|
System.out.println("PASSED");
|
|
}
|
|
|
|
System.out.println("Validating signature-enveloping-hmac-sha1-trunclen-0-attack.xml");
|
|
try {
|
|
test_signature("signature-enveloping-hmac-sha1-trunclen-0-attack.xml", SKKS, false);
|
|
System.out.println("FAILED");
|
|
atLeastOneFailed = true;
|
|
} catch (XMLSignatureException xse) {
|
|
System.out.println(xse.getMessage());
|
|
System.out.println("PASSED");
|
|
}
|
|
|
|
System.out.println("Validating signature-enveloping-hmac-sha1-trunclen-8-attack.xml");
|
|
try {
|
|
test_signature("signature-enveloping-hmac-sha1-trunclen-8-attack.xml", SKKS, false);
|
|
System.out.println("FAILED");
|
|
atLeastOneFailed = true;
|
|
} catch (XMLSignatureException xse) {
|
|
System.out.println(xse.getMessage());
|
|
System.out.println("PASSED");
|
|
}
|
|
|
|
if (atLeastOneFailed) {
|
|
throw new Exception
|
|
("At least one signature did not validate as expected");
|
|
}
|
|
}
|
|
|
|
public static boolean test_signature(String file, KeySelector ks)
|
|
throws Exception {
|
|
return test_signature(file, ks, false);
|
|
}
|
|
|
|
public static boolean test_signature(String file, KeySelector ks,
|
|
boolean cache) throws Exception {
|
|
if (ks == null) {
|
|
KeyStore keystore = KeyStore.getInstance("JKS");
|
|
keystore.load
|
|
(new FileInputStream(KEYSTORE), "changeit".toCharArray());
|
|
ks = new X509KeySelector(keystore, false);
|
|
}
|
|
return validator.validate(file, ks, httpUd, cache);
|
|
}
|
|
|
|
/**
|
|
* This URIDereferencer returns locally cached copies of http content to
|
|
* avoid test failures due to network glitches, etc.
|
|
*/
|
|
private static class HttpURIDereferencer implements URIDereferencer {
|
|
private URIDereferencer defaultUd;
|
|
|
|
HttpURIDereferencer() {
|
|
defaultUd = XMLSignatureFactory.getInstance().getURIDereferencer();
|
|
}
|
|
|
|
public Data dereference(final URIReference ref, XMLCryptoContext ctx)
|
|
throws URIReferenceException {
|
|
String uri = ref.getURI();
|
|
if (uri.equals(STYLESHEET) || uri.equals(STYLESHEET_B64)) {
|
|
try {
|
|
FileInputStream fis = new FileInputStream(new File
|
|
(DATA_DIR, uri.substring(uri.lastIndexOf('/'))));
|
|
return new OctetStreamData(fis,ref.getURI(),ref.getType());
|
|
} catch (Exception e) { throw new URIReferenceException(e); }
|
|
}
|
|
|
|
// fallback on builtin deref
|
|
return defaultUd.dereference(ref, ctx);
|
|
}
|
|
}
|
|
}
|