jdk-24/test/jdk/sun/security/rsa/TestSigGen15.java
2024-10-30 18:51:02 +00:00

128 lines
4.5 KiB
Java

/*
* Copyright (c) 2018, 2024, 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.
*/
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.security.*;
import java.security.spec.*;
import java.security.interfaces.*;
import java.util.ArrayList;
import java.util.HexFormat;
import java.util.List;
/*
* @test
* @bug 8146293
* @summary Known Answer Tests based on NIST 186-3 at:
* @compile SigRecord.java
* @run main/othervm TestSigGen15
*/
public class TestSigGen15 {
private static final String[] testFiles = {
"SigGen15_186-3.txt", "SigGen15_186-3_TruncatedSHAs.txt"
};
public static void main(String[] args) throws Exception {
boolean success = true;
for (String f : testFiles) {
System.out.println("[INPUT FILE " + f + "]");
try {
success &= runTest(SigRecord.read(f));
} catch (IOException e) {
System.out.println("Unexpected exception: " + e);
e.printStackTrace(System.out);
success = false;
}
}
if (!success) {
throw new RuntimeException("One or more test failed");
}
System.out.println("Test passed");
}
/*
* Run all the tests in the data list with specified algorithm
*/
static boolean runTest(List<SigRecord> records) throws Exception {
boolean success = true;
//for (Provider provider : Security.getProviders()) {
Provider p = Security.getProvider(
System.getProperty("test.provider.name","SunRsaSign"));
KeyFactory kf = KeyFactory.getInstance("RSA", p);
for (SigRecord sr : records) {
System.out.println("==Testing Record : " + sr + "==");
PrivateKey privKey = kf.generatePrivate(sr.privKeySpec);
PublicKey pubKey = kf.generatePublic(sr.pubKeySpec);
success &= check(privKey, pubKey, sr.testVectors, p);
System.out.println("==Done==");
}
return success;
}
/*
* Generate the signature, check against known values and verify.
*/
static boolean check(PrivateKey privKey, PublicKey pubKey,
List<SigRecord.SigVector> vectors, Provider p) throws Exception {
boolean success = true;
for (SigRecord.SigVector v : vectors) {
System.out.println("\tAgainst " + v.mdAlg);
String sigAlgo = v.mdAlg + "withRSA";
Signature sig;
try {
sig = Signature.getInstance(sigAlgo, p);
} catch (NoSuchAlgorithmException e) {
System.out.println("\tSkip " + sigAlgo +
" due to no support");
continue;
}
byte[] msgBytes = HexFormat.of().parseHex(v.msg);
byte[] expSigBytes = HexFormat.of().parseHex(v.sig);
sig.initSign(privKey);
sig.update(msgBytes);
byte[] actualSigBytes = sig.sign();
success &= MessageDigest.isEqual(actualSigBytes, expSigBytes);
if (!success) {
System.out.println("\tFailed:");
System.out.println("\tSHAALG = " + v.mdAlg);
System.out.println("\tMsg = " + v.msg);
System.out.println("\tExpected Sig = " + v.sig);
System.out.println("\tActual Sig = " + HexFormat.of().formatHex(actualSigBytes));
} else {
System.out.println("\t" + v.mdAlg + " Test Vector Passed");
}
}
return success;
}
}