/*
 * Copyright (c) 2008, 2014, 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 static java.lang.System.out;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.security.Key;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.security.UnrecoverableKeyException;
import java.security.cert.Certificate;
import java.util.Arrays;
import java.util.Base64;
import java.util.Enumeration;

/*
 * @test
 * @bug 8048619
 * @author  Bill Situ
 * @summary Test converting keystore from jceks to P12 and from P12 to other
 *  (jceks,jks). including following test cases:
 * Read jceks key store and convert to the p12 key store, then compare entries
 *  in the two key stores.
 * Read p12 key store and convert to the jceks key store, then compare entries
 *  in the two key stores.
 * Read p12 key store (contains only private key and a self-signed certificate)
 *  and convert to the jceks key store, then compare entries of two key stores.
 * Read p12 key store (contains 2 entries) and convert to the jceks key store,
 *  then compare entries in the two key stores.
 * Read p12 key store (entry password and key store password are different) and
 *  convert to the jceks key store, then compare entries in the two key stores.
 * Read p12 key store and convert to the jks key store, then compare entries
 *  in the two key stores.
 * Read p12 key store (contains only private key and a self-signed certificate)
 *  and convert to the jks key store, then compare entries in the two key stores.
 * Read p12 key store (contains 2 entries) and convert to the jks key store,
 *  then compare entries in the two key stores.
 * Read p12 key store (entry password and key store password are different) and
 * convert to the jks key store, then compare entries in the two key stores.
 */

public class ConvertP12Test {

    private static final String SUN_JSSE = "SunJSSE";
    private static final String SUN_JCE = "SunJCE";
    private static final String SUN = "SUN";
    private static final String PKCS12 = "pkcs12";
    private static final String JCE_KS = "JceKS";
    private static final String JKS = "JKS";

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

        ConvertP12Test jstest = new ConvertP12Test();

        jstest.driver("JceksToP12", "keystoreCA.jceks.data", JCE_KS, SUN_JCE,
                "storepass", "keypass", PKCS12, SUN_JSSE);

        jstest.driver("P12ToJceks_Chain", "ie_jceks_chain.pfx.data", PKCS12,
                SUN_JSSE, "pass", "pass", JCE_KS, SUN_JCE);

        jstest.driver("P12ToJceks_SelfSigned", "jdk_jceks_selfsigned.p12.data",
                PKCS12, SUN_JSSE, "pass", "pass", JCE_KS, SUN_JCE);

        jstest.driver("P12ToJceks_TwoEntry", "jdk_jceks_twoentry.p12.data",
                PKCS12, SUN_JSSE, "pass", "pass", JCE_KS, SUN_JCE);

        jstest.driver("P12ToJceks_TwoPass", "jdk_jceks_twopass.p12.data",
                PKCS12, SUN_JSSE, "storepass", "keypass", JCE_KS, SUN_JCE);

        jstest.driver("P12ToJks_Chain", "ie_jks_chain.pfx.data", PKCS12,
                SUN_JSSE, "pass", "pass", JKS, SUN);

        jstest.driver("P12ToJks_SelfSigned", "jdk_jks_selfsigned.p12.data",
                PKCS12, SUN_JSSE, "pass", "pass", JKS, SUN);

        jstest.driver("P12ToJks_TwoEntry", "jdk_jks_twoentry.p12.data", PKCS12,
                SUN_JSSE, "pass", "pass", JKS, SUN);

        jstest.driver("P12ToJks_TwoPass", "jdk_jks_twopass.p12.data", PKCS12,
                SUN_JSSE, "storepass", "keypass", JKS, SUN);

    }

    private void driver(String testCase, String inKeyStore,
            String inKeyStoreType, String inKeyStoreTypePrv,
            String inStorePass, String inKeyPass, String outKeyStoreType,
            String outKeyStorePrv) throws Exception {

        String outStorePass = "pass";
        String outKeyPass = "pass";
        KeyStore inputKeyStore, outputKeyStore;

        out.println("Testing " + testCase);
        String keystorePath = System.getProperty("test.src", ".")
                + File.separator + "certs" + File.separator + "convertP12";
        out.println("Output KeyStore : " + inKeyStore + ".out");
        String outKeyStoreName = inKeyStore + ".out";
        try (FileOutputStream fout = new FileOutputStream(outKeyStoreName);) {
            inputKeyStore = KeyStore.getInstance(inKeyStoreType,
                    inKeyStoreTypePrv);

            // KeyStore have encoded by Base64.getMimeEncoder().encode(),need
            // decode first.
            byte[] input = Files.readAllBytes(Paths.get(keystorePath,
                    inKeyStore));
            ByteArrayInputStream arrayIn = new ByteArrayInputStream(Base64
                    .getMimeDecoder().decode(input));

            out.println("Input KeyStore : " + inKeyStore);

            inputKeyStore.load(arrayIn, inStorePass.toCharArray());

            outputKeyStore = KeyStore.getInstance(outKeyStoreType,
                    outKeyStorePrv);
            outputKeyStore.load(null, null);

            run(inputKeyStore, outputKeyStore, inKeyPass, outKeyPass);

            outputKeyStore.store(fout, outStorePass.toCharArray());

            // for P12ToJks_TwoEntry test case will test includes each other,
            // others just test compareKeystore
            if (testCase.contains("TwoEntry")) {

                compareKeyStore(inputKeyStore, outputKeyStore, inKeyPass,
                        outKeyPass, 2);
                compareKeyStore(outputKeyStore, inputKeyStore, outKeyPass,
                        inKeyPass, 2);
            } else {
                compareKeyStore(inputKeyStore, outputKeyStore, inKeyPass,
                        outKeyPass, 1);
            }
            out.println("Test " + testCase + " STATUS: Pass!!");
        } catch (Exception ex) {
            out.println("Test " + testCase + " STATUS: failed with exception: "
                    + ex.getMessage());
            throw ex;
        }
    }

    private void run(KeyStore inputKeyStore, KeyStore outputKeyStore,
            String inKeyPass, String outKeyPass) throws Exception {
        Enumeration<String> e = inputKeyStore.aliases();
        String alias;
        while (e.hasMoreElements()) {
            alias = e.nextElement();
            Certificate[] certs = inputKeyStore.getCertificateChain(alias);

            boolean isCertEntry = inputKeyStore.isCertificateEntry(alias);
            // Test KeyStore only contain key pair entries.
            if (isCertEntry == true) {
                throw new RuntimeException(
                        "inputKeystore should not be certEntry because test"
                                + " keystore only contain key pair entries"
                                + " for alias:" + alias);
            }

            boolean isKeyEntry = inputKeyStore.isKeyEntry(alias);
            Key key = null;
            if (isKeyEntry) {
                key = inputKeyStore.getKey(alias, inKeyPass.toCharArray());
            } else {
                throw new RuntimeException("Entry type unknown for alias:"
                        + alias);
            }
            outputKeyStore.setKeyEntry(alias, key, outKeyPass.toCharArray(),
                    certs);
        }
    }

    private void compareKeyStore(KeyStore a, KeyStore b, String inKeyPass,
            String outKeyPass, int keyStoreSize) throws Exception {
        if (a.size() != keyStoreSize || b.size() != keyStoreSize) {
            throw new RuntimeException("size not match or size not equal to "
                    + keyStoreSize);
        }

        Enumeration<String> eA = a.aliases();
        while (eA.hasMoreElements()) {
            String aliasA = eA.nextElement();

            if (!b.containsAlias(aliasA)) {
                throw new RuntimeException("alias not match for alias:"
                        + aliasA);
            }

            compareKeyEntry(a, b, inKeyPass, outKeyPass, aliasA);
        }
    }

    private void compareKeyEntry(KeyStore a, KeyStore b, String aPass,
            String bPass, String alias) throws KeyStoreException,
            UnrecoverableKeyException, NoSuchAlgorithmException {
        Certificate[] certsA = a.getCertificateChain(alias);
        Certificate[] certsB = b.getCertificateChain(alias);

        if (!Arrays.equals(certsA, certsB)) {
            throw new RuntimeException("Certs don't match for alias:" + alias);
        }

        Key keyA = a.getKey(alias, aPass.toCharArray());
        Key keyB = b.getKey(alias, bPass.toCharArray());

        if (!keyA.equals(keyB)) {
            throw new RuntimeException(
                    "Key don't match for alias:" + alias);
        }
    }
}