132 lines
4.7 KiB
Java
132 lines
4.7 KiB
Java
/*
|
|
* Copyright (c) 2015, 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.security.KeyFactory;
|
|
import java.security.KeyPair;
|
|
import java.security.KeyPairGenerator;
|
|
import java.security.PrivateKey;
|
|
import java.security.PublicKey;
|
|
import java.security.interfaces.RSAKey;
|
|
import java.security.interfaces.RSAPrivateKey;
|
|
import java.security.interfaces.RSAPublicKey;
|
|
import java.security.spec.RSAPrivateKeySpec;
|
|
import java.security.spec.RSAPublicKeySpec;
|
|
|
|
/**
|
|
* @test
|
|
* @bug 8044199
|
|
* @summary test if the private and public key size are the same as what is set
|
|
* through KeyPairGenerator.
|
|
* @run main KeySizeTest 512 10
|
|
* @run main KeySizeTest 768 10
|
|
* @run main KeySizeTest 1024 10
|
|
* @run main KeySizeTest 2048 5
|
|
* @run main KeySizeTest 4096 1
|
|
*/
|
|
public class KeySizeTest {
|
|
|
|
/**
|
|
* ALGORITHM name, fixed as RSA.
|
|
*/
|
|
private static final String KEYALG = "RSA";
|
|
|
|
/**
|
|
* JDK default RSA Provider.
|
|
*/
|
|
private static final String PROVIDER_NAME =
|
|
System.getProperty("test.provider.name", "SunRsaSign");
|
|
|
|
public static void main(String[] args) throws Exception {
|
|
int iKeyPairSize = Integer.parseInt(args[0]);
|
|
int maxLoopCnt = Integer.parseInt(args[1]);
|
|
|
|
int failCount = 0;
|
|
KeyPairGenerator keyPairGen
|
|
= KeyPairGenerator.getInstance(KEYALG, PROVIDER_NAME);
|
|
keyPairGen.initialize(iKeyPairSize);
|
|
// Generate RSA keypair
|
|
KeyPair keyPair = keyPairGen.generateKeyPair();
|
|
|
|
// Get priavte and public keys
|
|
PrivateKey privateKey = keyPair.getPrivate();
|
|
PublicKey publicKey = keyPair.getPublic();
|
|
try {
|
|
if (!sizeTest(keyPair)) {
|
|
failCount++;
|
|
}
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace(System.err);
|
|
failCount++;
|
|
}
|
|
|
|
for (int iCnt = 0; iCnt < maxLoopCnt; iCnt++) {
|
|
|
|
// Get keysize (modulus) of keys
|
|
KeyFactory keyFact = KeyFactory.getInstance(KEYALG, PROVIDER_NAME);
|
|
|
|
// Comparing binary length.
|
|
RSAPrivateKeySpec privateKeySpec
|
|
= (RSAPrivateKeySpec) keyFact.getKeySpec(privateKey,
|
|
RSAPrivateKeySpec.class);
|
|
int iPrivateKeySize = privateKeySpec.getModulus().bitLength();
|
|
|
|
RSAPublicKeySpec publicKeySpec
|
|
= (RSAPublicKeySpec) keyFact.getKeySpec(publicKey,
|
|
RSAPublicKeySpec.class);
|
|
int iPublicKeySize = publicKeySpec.getModulus().bitLength();
|
|
|
|
if ((iKeyPairSize != iPublicKeySize) || (iKeyPairSize != iPrivateKeySize)) {
|
|
System.err.println("iKeyPairSize : " + iKeyPairSize);
|
|
System.err.println("Generated a " + iPrivateKeySize
|
|
+ " bit RSA private key");
|
|
System.err.println("Generated a " + iPublicKeySize
|
|
+ " bit RSA public key");
|
|
failCount++;
|
|
}
|
|
}
|
|
|
|
if (failCount > 0) {
|
|
throw new RuntimeException("There are " + failCount + " tests failed.");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param kpair test key pair.
|
|
* @return true if test passed. false if test failed.
|
|
*/
|
|
private static boolean sizeTest(KeyPair kpair) {
|
|
RSAPrivateKey priv = (RSAPrivateKey) kpair.getPrivate();
|
|
RSAPublicKey pub = (RSAPublicKey) kpair.getPublic();
|
|
|
|
// test the getModulus method
|
|
if ((priv instanceof RSAKey) && (pub instanceof RSAKey)) {
|
|
if (!priv.getModulus().equals(pub.getModulus())) {
|
|
System.err.println("priv.getModulus() = " + priv.getModulus());
|
|
System.err.println("pub.getModulus() = " + pub.getModulus());
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|