1berry/openjdk
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

8183348: Better cleanup for jdk/test/sun/security/pkcs12/P12SecretKey.java

Browse Source 
Reviewed-by: rhalade, weijun
This commit is contained in:
Mikhail Yankelevich 2025-05-02 23:15:26 +00:00 committed by Weijun Wang
parent 4d2d12987f
commit 03f0ec4a35
1 changed files with 27 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
test/jdk/sun/security/pkcs12/P12SecretKey.java
View File

@ -1,5 +1,5 @@
/*
* Copyright (c) 2016, 2024, Oracle and/or its affiliates. All rights reserved.
* Copyright (c) 2016, 2025, 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
@ -25,6 +25,7 @@
* @test
* @bug 8149411 8007632
* @summary Get AES key from keystore (uses SecretKeySpec not SecretKeyFactory)
* @library /test/lib
* @run main P12SecretKey pkcs12 AES 128
* @run main P12SecretKey pkcs12 DES 56
* @run main P12SecretKey pkcs12 DESede 168
@ -33,14 +34,14 @@
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.nio.file.Files;
import java.security.KeyStore;
import java.security.cert.CertificateException;
import java.util.Arrays;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import static jdk.test.lib.Utils.createTempFile;
public class P12SecretKey {
private static final String ALIAS = "alias";
@ -66,30 +67,32 @@ public class P12SecretKey {
KeyStore.ProtectionParameter kspp = new KeyStore.PasswordProtection(pw);
ks.setEntry(ALIAS, ske, kspp);
File ksFile = File.createTempFile("test", ".test");
// temporary files are created in scratch directory
final File ksFile = createTempFile(
String.format("%s-%s-%d-",
keystoreType,
algName,
keySize),
".ks").toFile();
try {
try (FileOutputStream fos = new FileOutputStream(ksFile)) {
ks.store(fos, pw);
fos.flush();
}
try (FileOutputStream fos = new FileOutputStream(ksFile)) {
ks.store(fos, pw);
fos.flush();
}
// now see if we can get it back
try (FileInputStream fis = new FileInputStream(ksFile)) {
KeyStore ks2 = KeyStore.getInstance(keystoreType);
ks2.load(fis, pw);
KeyStore.Entry entry = ks2.getEntry(ALIAS, kspp);
SecretKey keyIn = ((KeyStore.SecretKeyEntry) entry).getSecretKey();
if (Arrays.equals(key.getEncoded(), keyIn.getEncoded())) {
System.err.println("OK: worked just fine with " + keystoreType +
" keystore");
} else {
System.err.println("ERROR: keys are NOT equal after storing in "
+ keystoreType + " keystore");
}
// now see if we can get it back
try (FileInputStream fis = new FileInputStream(ksFile)) {
KeyStore ks2 = KeyStore.getInstance(keystoreType);
ks2.load(fis, pw);
KeyStore.Entry entry = ks2.getEntry(ALIAS, kspp);
SecretKey keyIn = ((KeyStore.SecretKeyEntry) entry).getSecretKey();
if (Arrays.equals(key.getEncoded(), keyIn.getEncoded())) {
System.err.println("OK: worked just fine with " + keystoreType +
" keystore");
} else {
throw new RuntimeException("ERROR: keys are NOT equal after storing in "
+ keystoreType + " keystore");
}
} finally {
Files.deleteIfExists(ksFile.toPath());
}
}
}