Skip to content

Commit

Permalink
8272908: Missing coverage for certain classes in com.sun.org.apache.x…
Browse files Browse the repository at this point in the history
…ml.internal.security

Reviewed-by: rhalade
  • Loading branch information
fguallini authored and rhalade committed Oct 14, 2021
1 parent ede3f4e commit a16f2d0
Show file tree
Hide file tree
Showing 3 changed files with 253 additions and 7 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,216 @@
/*
* Copyright (c) 2021, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/

/**
* @test
* @bug 8272908
* @summary Verify signature KeyInfo
* @library /test/lib
* @modules java.xml.crypto/com.sun.org.apache.xml.internal.security
* java.xml.crypto/com.sun.org.apache.xml.internal.security.c14n
* java.xml.crypto/com.sun.org.apache.xml.internal.security.signature
* java.xml.crypto/com.sun.org.apache.xml.internal.security.utils
* java.xml.crypto/com.sun.org.apache.xml.internal.security.keys
* java.xml.crypto/com.sun.org.apache.xml.internal.security.keys.content.keyvalues
* java.xml.crypto/com.sun.org.apache.xml.internal.security.keys.content
* java.xml.crypto/com.sun.org.apache.xml.internal.security.exceptions
* @run main/othervm SignatureKeyInfo
*/

import com.sun.org.apache.xml.internal.security.Init;
import com.sun.org.apache.xml.internal.security.exceptions.XMLSecurityException;
import com.sun.org.apache.xml.internal.security.keys.KeyInfo;
import com.sun.org.apache.xml.internal.security.keys.content.PGPData;
import com.sun.org.apache.xml.internal.security.keys.content.RetrievalMethod;
import com.sun.org.apache.xml.internal.security.keys.content.SPKIData;
import com.sun.org.apache.xml.internal.security.signature.XMLSignature;
import com.sun.org.apache.xml.internal.security.utils.Constants;
import com.sun.org.apache.xml.internal.security.utils.XMLUtils;
import com.sun.org.apache.xml.internal.security.utils.ElementProxy;
import com.sun.org.apache.xml.internal.security.keys.content.keyvalues.RSAKeyValue;
import com.sun.org.apache.xml.internal.security.keys.content.keyvalues.DSAKeyValue;

import jdk.test.lib.Asserts;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;


import javax.xml.crypto.dsig.CanonicalizationMethod;
import javax.xml.crypto.dsig.SignatureMethod;
import javax.xml.parsers.DocumentBuilderFactory;
import java.io.File;
import java.math.BigInteger;
import java.security.*;

import static jdk.test.lib.Asserts.assertEquals;

public class SignatureKeyInfo {

private final static String DIR = System.getProperty("test.src", ".");
private static DocumentBuilderFactory dbf = null;
private static Document doc;

private static final String NAME = "testName";
private static final String TEXT = "testText";
private static final String NS = Constants.SignatureSpecNS;
private static final String RSA = "RSA";
private static final String DSA = "DSA";
private static final String FILE_TO_SIGN = "signature-enveloping-hmac-sha1.xml";
private static final String FILE_TO_VERIFY = "signature-enveloping-hmac-sha1-keyinfo.xml";
private static final int FIRST_EL = 0;

public static void main(String[] args) throws Exception {

Init.init();
dbf = DocumentBuilderFactory.newInstance();
dbf.setNamespaceAware(true);
dbf.setValidating(false);
verifyXmlKeyInfo();
sign(RSA);
sign(DSA);
}

private static void sign(String algorithm) throws Exception {
File file = new File(DIR, FILE_TO_SIGN);

doc = dbf.newDocumentBuilder().parse(file);

KeyPair kp = getKeyPair(algorithm);

String signMethod = RSA.equals(algorithm) ? SignatureMethod.RSA_SHA256
: SignatureMethod.DSA_SHA256;

XMLSignature signature = new XMLSignature(doc, null,
signMethod, CanonicalizationMethod.INCLUSIVE);

signature.addKeyInfo(kp.getPublic());
KeyInfo keyInfo = signature.getKeyInfo();
addKeyInfoData(keyInfo, algorithm);
signature.sign(kp.getPrivate());
}

private static Element getSignElement() {
NodeList nl =
doc.getElementsByTagNameNS(NS, "Signature");
if (nl.getLength() == 0) {
throw new RuntimeException("Could not find signature Element");
}

return (Element) nl.item(FIRST_EL);
}

private static void addKeyInfoData(KeyInfo keyInfo, String algorithm) throws Exception {
KeyPair keyPair = getKeyPair(algorithm);

if (algorithm.equals(RSA)) {
RSAKeyValue rsaKeyValue = new RSAKeyValue(doc, keyPair.getPublic());
keyInfo.add(rsaKeyValue);
} else {
DSAKeyValue dsaKeyValue = new DSAKeyValue(doc, keyPair.getPublic());
keyInfo.add(dsaKeyValue);
}

Element elpgp= doc.createElementNS(NS, Constants._TAG_PGPDATA);
Element elrm= doc.createElementNS(NS, Constants._TAG_RETRIEVALMETHOD);
Element elspki= doc.createElementNS(NS, Constants._TAG_SPKIDATA);
keyInfo.add(new PGPData(elpgp, NS));
keyInfo.add(new RetrievalMethod(elrm, NS));
keyInfo.add(new SPKIData(elspki, NS));

keyInfo.setId(TEXT);
keyInfo.addKeyName(TEXT);
keyInfo.add(keyPair.getPublic());
keyInfo.addKeyValue(keyPair.getPublic());
keyInfo.addDEREncodedKeyValue(keyPair.getPublic());
keyInfo.addKeyInfoReference(NS);
keyInfo.addMgmtData(TEXT);

Element e = XMLUtils.createElementInSignatureSpace(doc, NAME);
keyInfo.addKeyValue(e);
keyInfo.addUnknownElement(e);
keyInfo.addText(TEXT);
keyInfo.addTextElement(TEXT, NAME);
keyInfo.addBigIntegerElement(BigInteger.valueOf(12345), NAME);
keyInfo.addBase64Text(TEXT.getBytes());
keyInfo.addBase64Element(TEXT.getBytes(), NAME);

verifyKeyInfoData(keyInfo, algorithm);
}

private static KeyPair getKeyPair(String algorithm) throws NoSuchAlgorithmException {
KeyPairGenerator keyGen = KeyPairGenerator.getInstance(algorithm);
keyGen.initialize(2048);

return keyGen.genKeyPair();
}

private static void verifyKeyInfoData(KeyInfo keyInfo, String algorithm)
throws XMLSecurityException {
Asserts.assertTrue(keyInfo.containsKeyName());
verifyElementText(keyInfo.itemKeyName(FIRST_EL));
Asserts.assertTrue(keyInfo.containsKeyValue());
verifyElementNS(keyInfo.itemKeyValue(FIRST_EL).getBaseNamespace());

Asserts.assertTrue(keyInfo.containsKeyInfoReference());
verifyElementNS(keyInfo.itemKeyInfoReference(FIRST_EL).getURI());
Asserts.assertTrue(keyInfo.containsDEREncodedKeyValue());
Asserts.assertTrue(keyInfo.containsMgmtData());
verifyElementText(keyInfo.itemMgmtData(FIRST_EL));
Asserts.assertEquals(TEXT, keyInfo.getId());

Asserts.assertTrue(keyInfo.containsPGPData());
verifyElementNS(keyInfo.itemPGPData(FIRST_EL).getBaseNamespace());

Asserts.assertTrue(keyInfo.containsRetrievalMethod());
verifyElementNS(keyInfo.itemRetrievalMethod(FIRST_EL).getBaseNamespace());
Asserts.assertTrue(keyInfo.containsSPKIData());
verifyElementNS(keyInfo.itemSPKIData(FIRST_EL).getBaseNamespace());

Asserts.assertTrue(keyInfo.containsUnknownElement());
Asserts.assertEquals(NAME, keyInfo.itemUnknownElement(13).getLocalName());

Asserts.assertFalse(keyInfo.isEmpty());
Asserts.assertEquals(algorithm, keyInfo.getPublicKey().getAlgorithm());
}

private static void verifyXmlKeyInfo() throws Exception {
File file = new File(DIR, FILE_TO_VERIFY);

doc = dbf.newDocumentBuilder().parse(file);
Element sigElement = getSignElement();
XMLSignature signature = new XMLSignature
(sigElement, file.toURI().toString());

KeyInfo keyInfo = signature.getKeyInfo();
assertEquals(TEXT, keyInfo.itemMgmtData(FIRST_EL).getMgmtData());
}

private static void verifyElementText(ElementProxy elementProxy) {
Asserts.assertEquals(TEXT, elementProxy.getTextFromTextChild());
}

private static void verifyElementNS(String actualNs) {
Asserts.assertEquals(NS, actualNs);
}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2009, 2013, Oracle and/or its affiliates. All rights reserved.
* Copyright (c) 2009, 2021, Oracle and/or its affiliates. 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
Expand Down Expand Up @@ -47,6 +47,7 @@
import com.sun.org.apache.xml.internal.security.signature.XMLSignatureException;
import com.sun.org.apache.xml.internal.security.utils.Constants;

import static java.nio.charset.StandardCharsets.US_ASCII;

public class TruncateHMAC {

Expand All @@ -64,7 +65,11 @@ public static void main(String[] args) throws Exception {
validate("signature-enveloping-hmac-sha1-trunclen-8-attack.xml", false);
// this one should pass
validate("signature-enveloping-hmac-sha1.xml", true);
generate_hmac_sha1_40();

// There are multiple validations regarding hmac min output length, therefore
// checking different values will exercise multiple code blocks
generate_hmac_sha1(40);
generate_hmac_sha1(128);

if (atLeastOneFailed) {
throw new Exception
Expand All @@ -86,7 +91,7 @@ private static void validate(String data, boolean pass) throws Exception {
try {
XMLSignature signature = new XMLSignature
(sigElement, file.toURI().toString());
SecretKey sk = signature.createSecretKey("secret".getBytes("ASCII"));
SecretKey sk = signature.createSecretKey("secret".getBytes(US_ASCII));
System.out.println
("Validation status: " + signature.checkSignatureValue(sk));
if (!pass) {
Expand All @@ -106,15 +111,15 @@ private static void validate(String data, boolean pass) throws Exception {
}
}

private static void generate_hmac_sha1_40() throws Exception {
System.out.println("Generating ");
private static void generate_hmac_sha1(int hmacOutputLength) throws Exception {
System.out.println("Generating " + hmacOutputLength);

Document doc = dbf.newDocumentBuilder().newDocument();
try {
XMLSignature sig = new XMLSignature
(doc, null, XMLSignature.ALGO_ID_MAC_HMAC_SHA1, 40,
(doc, null, XMLSignature.ALGO_ID_MAC_HMAC_SHA1, hmacOutputLength,
Canonicalizer.ALGO_ID_C14N_OMIT_COMMENTS);
sig.sign(getSecretKey("secret".getBytes("ASCII")));
sig.sign(getSecretKey("secret".getBytes(US_ASCII)));
System.out.println("FAILED");
atLeastOneFailed = true;
} catch (XMLSignatureException xse) {
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
<?xml version="1.0" encoding="UTF-8"?>
<Signature xmlns="http://www.w3.org/2000/09/xmldsig#">
<SignedInfo>
<CanonicalizationMethod Algorithm="http://www.w3.org/TR/2001/REC-xml-c14n-20010315" />
<SignatureMethod Algorithm="http://www.w3.org/2000/09/xmldsig#hmac-sha1" />
<Reference URI="#object">
<DigestMethod Algorithm="http://www.w3.org/2000/09/xmldsig#sha1" />
<DigestValue>7/XTsHaBSOnJ/jXD5v0zL6VKYsk=</DigestValue>
</Reference>
</SignedInfo>
<SignatureValue>
JElPttIT4Am7Q+MNoMyv+WDfAZw=
</SignatureValue>
<KeyInfo>
<MgmtData>testText</MgmtData>
<KeyValue>
<ECKeyValue xmlns="http://www.w3.org/2009/xmldsig11#">
<NamedCurve URI="urn:oid:1.2.840.10045.3.1.7"/>
<PublicKey>BAds672US3sCYunM2k2bEQLbuRxdQlNTvq+5fitOpDMe0mBdZV4J3yZaG0taziYIuAT9GJGfds+q
xtXOCNWe/60=</PublicKey>
</ECKeyValue>
</KeyValue>
</KeyInfo>
<Object Id="object">some text</Object>
</Signature>

5 comments on commit a16f2d0

@openjdk-notifier
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@TheRealMDoerr
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

/backport jdk17u-dev

@openjdk
Copy link

@openjdk openjdk bot commented on a16f2d0 Mar 30, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@TheRealMDoerr the backport was successfully created on the branch TheRealMDoerr-backport-a16f2d0a in my personal fork of openjdk/jdk17u-dev. To create a pull request with this backport targeting openjdk/jdk17u-dev:master, just click the following link:

➡️ Create pull request

The title of the pull request is automatically filled in correctly and below you find a suggestion for the pull request body:

Hi all,

This pull request contains a backport of commit a16f2d0a from the openjdk/jdk repository.

The commit being backported was authored by Fernando Guallini on 14 Oct 2021 and was reviewed by Rajan Halade.

Thanks!

If you need to update the source branch of the pull then run the following commands in a local clone of your personal fork of openjdk/jdk17u-dev:

$ git fetch https://github.com/openjdk-bots/jdk17u-dev TheRealMDoerr-backport-a16f2d0a:TheRealMDoerr-backport-a16f2d0a
$ git checkout TheRealMDoerr-backport-a16f2d0a
# make changes
$ git add paths/to/changed/files
$ git commit --message 'Describe additional changes made'
$ git push https://github.com/openjdk-bots/jdk17u-dev TheRealMDoerr-backport-a16f2d0a

@GoeLin
Copy link
Member

@GoeLin GoeLin commented on a16f2d0 Apr 1, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

/backport jdk11u-dev

@openjdk
Copy link

@openjdk openjdk bot commented on a16f2d0 Apr 1, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@GoeLin the backport was successfully created on the branch GoeLin-backport-a16f2d0a in my personal fork of openjdk/jdk11u-dev. To create a pull request with this backport targeting openjdk/jdk11u-dev:master, just click the following link:

➡️ Create pull request

The title of the pull request is automatically filled in correctly and below you find a suggestion for the pull request body:

Hi all,

This pull request contains a backport of commit a16f2d0a from the openjdk/jdk repository.

The commit being backported was authored by Fernando Guallini on 14 Oct 2021 and was reviewed by Rajan Halade.

Thanks!

If you need to update the source branch of the pull then run the following commands in a local clone of your personal fork of openjdk/jdk11u-dev:

$ git fetch https://github.com/openjdk-bots/jdk11u-dev GoeLin-backport-a16f2d0a:GoeLin-backport-a16f2d0a
$ git checkout GoeLin-backport-a16f2d0a
# make changes
$ git add paths/to/changed/files
$ git commit --message 'Describe additional changes made'
$ git push https://github.com/openjdk-bots/jdk11u-dev GoeLin-backport-a16f2d0a

Please sign in to comment.