StoreSecretKeyTest.java revision 6507:5bf1c9e6be60 
1/*
2 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test
26 * @bug 8005408
27 * @summary KeyStore API enhancements
28 */
29
30import java.io.*;
31import java.security.*;
32import java.util.*;
33import javax.crypto.*;
34import javax.crypto.spec.*;
35
36// Store a secret key in a keystore and retrieve it again.
37
38public class StoreSecretKeyTest {
39    private final static String DIR = System.getProperty("test.src", ".");
40    private static final char[] PASSWORD = "passphrase".toCharArray();
41    private static final String KEYSTORE = "keystore.p12";
42    private static final String ALIAS = "my secret key";
43
44    public static void main(String[] args) throws Exception {
45
46        // Skip test if AES is unavailable
47        try {
48            SecretKeyFactory.getInstance("AES");
49        } catch (NoSuchAlgorithmException nsae) {
50            System.out.println("AES is unavailable. Skipping test...");
51            return;
52        }
53
54        new File(KEYSTORE).delete();
55
56        KeyStore keystore = KeyStore.getInstance("PKCS12");
57        keystore.load(null, null);
58
59        // Set entry
60        keystore.setEntry(ALIAS,
61            new KeyStore.SecretKeyEntry(generateSecretKey("AES", 128)),
62                new KeyStore.PasswordProtection(PASSWORD));
63
64        try (FileOutputStream outStream = new FileOutputStream(KEYSTORE)) {
65            System.out.println("Storing keystore to: " + KEYSTORE);
66            keystore.store(outStream, PASSWORD);
67        }
68
69        try (FileInputStream inStream = new FileInputStream(KEYSTORE)) {
70            System.out.println("Loading keystore from: " + KEYSTORE);
71            keystore.load(inStream, PASSWORD);
72            System.out.println("Loaded keystore with " + keystore.size() +
73                " entries");
74        }
75
76        KeyStore.Entry entry = keystore.getEntry(ALIAS,
77            new KeyStore.PasswordProtection(PASSWORD));
78        System.out.println("Retrieved entry: " + entry);
79
80        if (entry instanceof KeyStore.SecretKeyEntry) {
81            System.out.println("Retrieved secret key entry: " + entry);
82        } else {
83            throw new Exception("Not a secret key entry");
84        }
85    }
86
87    private static SecretKey generateSecretKey(String algorithm, int size)
88        throws NoSuchAlgorithmException {
89
90        // Failover to DES if the requested secret key factory is unavailable
91        SecretKeyFactory keyFactory;
92        try {
93            keyFactory = SecretKeyFactory.getInstance(algorithm);
94        } catch (NoSuchAlgorithmException nsae) {
95            keyFactory = SecretKeyFactory.getInstance("DES");
96            algorithm = "DES";
97            size = 56;
98        }
99
100        KeyGenerator generator = KeyGenerator.getInstance(algorithm);
101        generator.init(size);
102        return generator.generateKey();
103    }
104}
105