/*

 * Copyright (c) 1999, 2021, 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.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * 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.

 */

package com.sun.crypto.provider;

import jdk.internal.access.SharedSecrets;

import java.security.Key;

import java.security.PublicKey;

import java.security.PrivateKey;

import java.security.KeyFactory;

import java.security.InvalidKeyException;

import java.security.NoSuchAlgorithmException;

import java.security.spec.PKCS8EncodedKeySpec;

import java.security.spec.X509EncodedKeySpec;

import java.security.spec.InvalidKeySpecException;

import java.util.Arrays;

import javax.crypto.SecretKey;

import javax.crypto.Cipher;

import javax.crypto.spec.SecretKeySpec;

/**

 * This class is a helper class which construct key objects

 * from encoded keys.

 *

 * @author Sharon Liu

 *

 */

final class ConstructKeys {

    private static final PublicKey constructPublicKey(byte[] encodedKey,

            int ofs, int len, String encodedKeyAlgorithm)

            throws InvalidKeyException, NoSuchAlgorithmException {

        PublicKey key = null;

        byte[] keyBytes = (ofs == 0 && encodedKey.length == len)

                ? encodedKey : Arrays.copyOfRange(encodedKey, ofs, ofs + len);

        X509EncodedKeySpec keySpec = new X509EncodedKeySpec(keyBytes);

        try {

            KeyFactory keyFactory =

                KeyFactory.getInstance(encodedKeyAlgorithm,

                    SunJCE.getInstance());

            key = keyFactory.generatePublic(keySpec);

        } catch (NoSuchAlgorithmException nsae) {

            // Try to see whether there is another

            // provider which supports this algorithm

            try {

                KeyFactory keyFactory =

                    KeyFactory.getInstance(encodedKeyAlgorithm);

                key = keyFactory.generatePublic(keySpec);

            } catch (NoSuchAlgorithmException nsae2) {

                throw new NoSuchAlgorithmException("No installed providers " +

                                                   "can create keys for the " +

                                                   encodedKeyAlgorithm +

                                                   "algorithm");

            } catch (InvalidKeySpecException ikse2) {

                InvalidKeyException ike =

                    new InvalidKeyException("Cannot construct public key");

                ike.initCause(ikse2);

                throw ike;

            }

        } catch (InvalidKeySpecException ikse) {

            InvalidKeyException ike =

                new InvalidKeyException("Cannot construct public key");

            ike.initCause(ikse);

            throw ike;

        }

        return key;

    }

    private static final PrivateKey constructPrivateKey(byte[] encodedKey,

            int ofs, int len, String encodedKeyAlgorithm)

            throws InvalidKeyException, NoSuchAlgorithmException {

        PrivateKey key = null;

        byte[] keyBytes = (ofs == 0 && encodedKey.length == len)

                ? encodedKey : Arrays.copyOfRange(encodedKey, ofs, ofs + len);

        PKCS8EncodedKeySpec keySpec = new PKCS8EncodedKeySpec(keyBytes);

        try {

            KeyFactory keyFactory =

                KeyFactory.getInstance(encodedKeyAlgorithm,

                    SunJCE.getInstance());

            return keyFactory.generatePrivate(keySpec);

        } catch (NoSuchAlgorithmException nsae) {

            // Try to see whether there is another

            // provider which supports this algorithm

            try {

                KeyFactory keyFactory =

                    KeyFactory.getInstance(encodedKeyAlgorithm);

                key = keyFactory.generatePrivate(keySpec);

            } catch (NoSuchAlgorithmException nsae2) {

                throw new NoSuchAlgorithmException("No installed providers " +

                                                   "can create keys for the " +

                                                   encodedKeyAlgorithm +

                                                   "algorithm");

            } catch (InvalidKeySpecException ikse2) {

                InvalidKeyException ike =

                    new InvalidKeyException("Cannot construct private key");

                ike.initCause(ikse2);

                throw ike;

            }

        } catch (InvalidKeySpecException ikse) {

            InvalidKeyException ike =

                new InvalidKeyException("Cannot construct private key");

            ike.initCause(ikse);

            throw ike;

        } finally {

            SharedSecrets.getJavaSecuritySpecAccess().clearEncodedKeySpec(keySpec);

            if (keyBytes != encodedKey) {

                Arrays.fill(keyBytes, (byte)0);

            }

        }

        return key;

    }

    private static final SecretKey constructSecretKey(byte[] encodedKey,

            int ofs, int len, String encodedKeyAlgorithm) {

        return (new SecretKeySpec(encodedKey, ofs, len, encodedKeyAlgorithm));

    }

    static final Key constructKey(byte[] encoding, String keyAlgorithm,

            int keyType) throws InvalidKeyException, NoSuchAlgorithmException {

        return constructKey(encoding, 0, encoding.length, keyAlgorithm,

                keyType);

    }

    static final Key constructKey(byte[] encoding, int ofs, int len,

            String keyAlgorithm, int keyType)

            throws InvalidKeyException, NoSuchAlgorithmException {

        return switch (keyType) {

            case Cipher.SECRET_KEY -> ConstructKeys.constructSecretKey(

                    encoding, ofs, len, keyAlgorithm);

            case Cipher.PRIVATE_KEY -> ConstructKeys.constructPrivateKey(

                    encoding, ofs, len, keyAlgorithm);

            case Cipher.PUBLIC_KEY -> ConstructKeys.constructPublicKey(

                    encoding, ofs, len, keyAlgorithm);

            default -> throw new NoSuchAlgorithmException("Unsupported key type");

        };

    }

}