/*

 * Copyright (c) 2002, 2017, 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 sun.security.ssl;

import java.util.*;

import java.security.NoSuchAlgorithmException;

import java.security.InvalidKeyException;

import java.security.SecureRandom;

import java.security.KeyManagementException;

import javax.crypto.Cipher;

import javax.crypto.SecretKey;

import javax.crypto.spec.IvParameterSpec;

import javax.crypto.spec.SecretKeySpec;

import static sun.security.ssl.CipherSuite.KeyExchange.*;

import static sun.security.ssl.CipherSuite.PRF.*;

import static sun.security.ssl.CipherSuite.CipherType.*;

import static sun.security.ssl.CipherSuite.MacAlg.*;

import static sun.security.ssl.CipherSuite.BulkCipher.*;

import static sun.security.ssl.JsseJce.*;

import static sun.security.ssl.NamedGroupType.*;

/**

 * An SSL/TLS CipherSuite. Constants for the standard key exchange, cipher,

 * and mac algorithms are also defined in this class.

 *

 * The CipherSuite class and the inner classes defined in this file roughly

 * follow the type safe enum pattern described in Effective Java. This means:

 *

 *  . instances are immutable, classes are final

 *

 *  . there is a unique instance of every value, i.e. there are never two

 *    instances representing the same CipherSuite, etc. This means equality

 *    tests can be performed using == instead of equals() (although that works

 *    as well). [A minor exception are *unsupported* CipherSuites read from a

 *    handshake message, but this is usually irrelevant]

 *

 *  . instances are obtained using the static valueOf() factory methods.

 *

 *  . properties are defined as final variables and made available as

 *    package private variables without method accessors

 *

 *  . if the member variable allowed is false, the given algorithm is either

 *    unavailable or disabled at compile time

 *

 */

final class CipherSuite implements Comparable<CipherSuite> {

    // minimum priority for supported CipherSuites

    static final int SUPPORTED_SUITES_PRIORITY = 1;

    // minimum priority for default enabled CipherSuites

    static final int DEFAULT_SUITES_PRIORITY = 300;

    private static final boolean ALLOW_ECC = Debug.getBooleanProperty

        ("com.sun.net.ssl.enableECC", true);

    // Map Integer(id) -> CipherSuite

    // contains all known CipherSuites

    private static final Map<Integer,CipherSuite> idMap;

    // Map String(name) -> CipherSuite

    // contains only supported CipherSuites (i.e. allowed == true)

    private static final Map<String,CipherSuite> nameMap;

    // Protocol defined CipherSuite name, e.g. SSL_RSA_WITH_RC4_128_MD5

    // we use TLS_* only for new CipherSuites, still SSL_* for old ones

    final String name;

    // id in 16 bit MSB format, i.e. 0x0004 for SSL_RSA_WITH_RC4_128_MD5

    final int id;

    // priority for the internal default preference order. the higher the

    // better. Each supported CipherSuite *must* have a unique priority.

    // Ciphersuites with priority >= DEFAULT_SUITES_PRIORITY are enabled

    // by default

    final int priority;

    // key exchange, bulk cipher, mac and prf algorithms. See those

    // classes below.

    final KeyExchange keyExchange;

    final BulkCipher cipher;

    final MacAlg macAlg;

    final PRF prfAlg;

    // whether a CipherSuite qualifies as exportable under 512/40 bit rules.

    // TLS 1.1+ (RFC 4346) must not negotiate to these suites.

    final boolean exportable;

    // true iff implemented and enabled at compile time

    final boolean allowed;

    // obsoleted since protocol version

    //

    // TLS version is used.  If checking DTLS versions, please map to

    // TLS version firstly.  See ProtocolVersion.mapToTLSProtocol().

    final int obsoleted;

    // supported since protocol version (TLS version is used)

    //

    // TLS version is used.  If checking DTLS versions, please map to

    // TLS version firstly.  See ProtocolVersion.mapToTLSProtocol().

    final int supported;

    /**

     * Constructor for implemented CipherSuites.

     */

    private CipherSuite(String name, int id, int priority,

            KeyExchange keyExchange, BulkCipher cipher, MacAlg mac,

            boolean allowed, int obsoleted, int supported, PRF prfAlg) {

        this.name = name;

        this.id = id;

        this.priority = priority;

        this.keyExchange = keyExchange;

        this.cipher = cipher;

        this.macAlg = mac;

        this.exportable = cipher.exportable;

        allowed &= keyExchange.allowed;

        allowed &= cipher.allowed;

        this.allowed = allowed;

        this.obsoleted = obsoleted;

        this.supported = supported;

        this.prfAlg = prfAlg;

    }

    /**

     * Constructor for unimplemented CipherSuites.

     */

    private CipherSuite(String name, int id) {

        this.name = name;

        this.id = id;

        this.allowed = false;

        this.priority = 0;

        this.keyExchange = null;

        this.cipher = null;

        this.macAlg = null;

        this.exportable = false;

        this.obsoleted = ProtocolVersion.LIMIT_MAX_VALUE;

        this.supported = ProtocolVersion.LIMIT_MIN_VALUE;

        this.prfAlg = P_NONE;

    }

    /**

     * Return whether this CipherSuite is available for use. A

     * CipherSuite may be unavailable even if it is supported

     * (i.e. allowed == true) if the required JCE cipher is not installed.

     */

    boolean isAvailable() {

        return allowed && keyExchange.isAvailable() && cipher.isAvailable();

    }

    boolean isNegotiable() {

        return this != C_SCSV && isAvailable();

    }

    // See also CipherBox.calculatePacketSize().

    int calculatePacketSize(int fragmentSize,

            ProtocolVersion protocolVersion, boolean isDTLS) {

        int packetSize = fragmentSize;

        if (cipher != B_NULL) {

            int blockSize = cipher.ivSize;

            switch (cipher.cipherType) {

                case BLOCK_CIPHER:

                    packetSize += macAlg.size;

                    packetSize += 1;        // 1 byte padding length field

                    packetSize +=           // use the minimal padding

                            (blockSize - (packetSize % blockSize)) % blockSize;

                    if (protocolVersion.useTLS11PlusSpec()) {

                        packetSize += blockSize;        // explicit IV

                    }

                    break;

            case AEAD_CIPHER:

                packetSize += cipher.ivSize - cipher.fixedIvSize;   // record IV

                packetSize += cipher.tagSize;

                break;

            default:    // NULL_CIPHER or STREAM_CIPHER

                packetSize += macAlg.size;

            }

        }

        return packetSize +

            (isDTLS ? DTLSRecord.headerSize : SSLRecord.headerSize);

    }

    // See also CipherBox.calculateFragmentSize().

    int calculateFragSize(int packetLimit,

            ProtocolVersion protocolVersion, boolean isDTLS) {

        int fragSize = packetLimit -

                (isDTLS ? DTLSRecord.headerSize : SSLRecord.headerSize);

        if (cipher != B_NULL) {

            int blockSize = cipher.ivSize;

            switch (cipher.cipherType) {

            case BLOCK_CIPHER:

                if (protocolVersion.useTLS11PlusSpec()) {

                    fragSize -= blockSize;              // explicit IV

                }

                fragSize -= (fragSize % blockSize);     // cannot hold a block

                // No padding for a maximum fragment.

                fragSize -= 1;        // 1 byte padding length field: 0x00

                fragSize -= macAlg.size;

                break;

            case AEAD_CIPHER:

                fragSize -= cipher.tagSize;

                fragSize -= cipher.ivSize - cipher.fixedIvSize;     // record IV

                break;

            default:    // NULL_CIPHER or STREAM_CIPHER

                fragSize -= macAlg.size;

            }

        }

        return fragSize;

    }

    /**

     * Compares CipherSuites based on their priority. Has the effect of

     * sorting CipherSuites when put in a sorted collection, which is

     * used by CipherSuiteList. Follows standard Comparable contract.

     *

     * Note that for unsupported CipherSuites parsed from a handshake

     * message we violate the equals() contract.

     */

    @Override

    public int compareTo(CipherSuite o) {

        return o.priority - priority;

    }

    /**

     * Returns this.name.

     */

    @Override

    public String toString() {

        return name;

    }

    /**

     * Return a CipherSuite for the given name. The returned CipherSuite

     * is supported by this implementation but may not actually be

     * currently useable. See isAvailable().

     *

     * @exception IllegalArgumentException if the CipherSuite is unknown or

     * unsupported.

     */

    static CipherSuite valueOf(String s) {

        if (s == null) {

            throw new IllegalArgumentException("Name must not be null");

        }

        CipherSuite c = nameMap.get(s);

        if ((c == null) || (c.allowed == false)) {

            throw new IllegalArgumentException("Unsupported ciphersuite " + s);

        }

        return c;

    }

    /**

     * Return a CipherSuite with the given ID. A temporary object is

     * constructed if the ID is unknown. Use isAvailable() to verify that

     * the CipherSuite can actually be used.

     */

    static CipherSuite valueOf(int id1, int id2) {

        id1 &= 0xff;

        id2 &= 0xff;

        int id = (id1 << 8) | id2;

        CipherSuite c = idMap.get(id);

        if (c == null) {

            String h1 = Integer.toString(id1, 16);

            String h2 = Integer.toString(id2, 16);

            c = new CipherSuite("Unknown 0x" + h1 + ":0x" + h2, id);

        }

        return c;

    }

    // for use by SSLContextImpl only

    static Collection<CipherSuite> allowedCipherSuites() {

        return nameMap.values();

    }

    /*

     * Use this method when all of the values need to be specified.

     * This is primarily used when defining a new ciphersuite for

     * TLS 1.2+ that doesn't use the "default" PRF.

     */

    private static void add(String name, int id, int priority,

            KeyExchange keyExchange, BulkCipher cipher, MacAlg mac,

            boolean allowed, int obsoleted, int supported, PRF prf) {

        CipherSuite c = new CipherSuite(name, id, priority, keyExchange,

            cipher, mac, allowed, obsoleted, supported, prf);

        if (idMap.put(id, c) != null) {

            throw new RuntimeException("Duplicate ciphersuite definition: "

                                        + id + ", " + name);

        }

        if (c.allowed) {

            if (nameMap.put(name, c) != null) {

                throw new RuntimeException("Duplicate ciphersuite definition: "

                                            + id + ", " + name);

            }

        }

    }

    /*

     * Use this method when there is no lower protocol limit where this

     * suite can be used, and the PRF is P_SHA256.  That is, the

     * existing ciphersuites.  From RFC 5246:

     *

     *     All cipher suites in this document use P_SHA256.

     */

    private static void add(String name, int id, int priority,

            KeyExchange keyExchange, BulkCipher cipher, MacAlg mac,

            boolean allowed, int obsoleted) {

        PRF prf = obsoleted < ProtocolVersion.TLS12.v ? P_NONE : P_SHA256;

        add(name, id, priority, keyExchange, cipher, mac, allowed, obsoleted,

            ProtocolVersion.LIMIT_MIN_VALUE, prf);

    }

    /*

     * Use this method when there is no upper protocol limit.  That is,

     * suites which have not been obsoleted.

     */

    private static void add(String name, int id, int priority,

            KeyExchange keyExchange, BulkCipher cipher, MacAlg mac,

            boolean allowed) {

        add(name, id, priority, keyExchange, cipher, mac, allowed,

                ProtocolVersion.LIMIT_MAX_VALUE);

    }

    /*

     * Use this method to define an unimplemented suite.  This provides

     * a number<->name mapping that can be used for debugging.

     */

    private static void add(String name, int id) {

        CipherSuite c = new CipherSuite(name, id);

        if (idMap.put(id, c) != null) {

            throw new RuntimeException("Duplicate ciphersuite definition: "

                                        + id + ", " + name);

        }

    }

    /**

     * An SSL/TLS key exchange algorithm.

     */

    static enum KeyExchange {

        // key exchange algorithms

        K_NULL       ("NULL",       false,      NAMED_GROUP_NONE),

        K_RSA        ("RSA",        true,       NAMED_GROUP_NONE),

        K_RSA_EXPORT ("RSA_EXPORT", true,       NAMED_GROUP_NONE),

        K_DH_RSA     ("DH_RSA",     false,      NAMED_GROUP_NONE),

        K_DH_DSS     ("DH_DSS",     false,      NAMED_GROUP_NONE),

        K_DHE_DSS    ("DHE_DSS",    true,       NAMED_GROUP_FFDHE),

        K_DHE_RSA    ("DHE_RSA",    true,       NAMED_GROUP_FFDHE),

        K_DH_ANON    ("DH_anon",    true,       NAMED_GROUP_FFDHE),

        K_ECDH_ECDSA ("ECDH_ECDSA",  ALLOW_ECC, NAMED_GROUP_ECDHE),

        K_ECDH_RSA   ("ECDH_RSA",    ALLOW_ECC, NAMED_GROUP_ECDHE),

        K_ECDHE_ECDSA("ECDHE_ECDSA", ALLOW_ECC, NAMED_GROUP_ECDHE),

        K_ECDHE_RSA  ("ECDHE_RSA",   ALLOW_ECC, NAMED_GROUP_ECDHE),

        K_ECDH_ANON  ("ECDH_anon",   ALLOW_ECC, NAMED_GROUP_ECDHE),

        // Kerberos cipher suites

        K_KRB5       ("KRB5", true,             NAMED_GROUP_NONE),

        K_KRB5_EXPORT("KRB5_EXPORT", true,      NAMED_GROUP_NONE),

        // renegotiation protection request signaling cipher suite

        K_SCSV       ("SCSV",        true,      NAMED_GROUP_NONE);

        // name of the key exchange algorithm, e.g. DHE_DSS

        final String name;

        final boolean allowed;

        final NamedGroupType groupType;

        private final boolean alwaysAvailable;

        KeyExchange(String name, boolean allowed, NamedGroupType groupType) {

            this.name = name;

            this.allowed = allowed;

            this.groupType = groupType;

            this.alwaysAvailable = allowed &&

                (!name.startsWith("EC")) && (!name.startsWith("KRB"));

        }

        boolean isAvailable() {

            if (alwaysAvailable) {

                return true;

            }

            if (groupType == NAMED_GROUP_ECDHE) {

                return (allowed && JsseJce.isEcAvailable());

            } else if (name.startsWith("KRB")) {

                return (allowed && JsseJce.isKerberosAvailable());

            } else {

                return allowed;

            }

        }

        @Override

        public String toString() {

            return name;

        }

    }

    static enum CipherType {

        NULL_CIPHER,           // null cipher

        STREAM_CIPHER,         // stream cipher

        BLOCK_CIPHER,          // block cipher in CBC mode

        AEAD_CIPHER            // AEAD cipher

    }

    /**

     * An SSL/TLS bulk cipher algorithm. One instance per combination of

     * cipher and key length.

     *

     * Also contains a factory method to obtain in initialized CipherBox

     * for this algorithm.

     */

    static enum BulkCipher {

        // export strength ciphers

        B_NULL("NULL", NULL_CIPHER, 0, 0, 0, 0, true),

        B_RC4_40(CIPHER_RC4, STREAM_CIPHER, 5, 16, 0, 0, true),

        B_RC2_40("RC2", BLOCK_CIPHER, 5, 16, 8, 0, false),

        B_DES_40(CIPHER_DES,  BLOCK_CIPHER, 5, 8, 8, 0, true),

        // domestic strength ciphers

        B_RC4_128(CIPHER_RC4, STREAM_CIPHER, 16, 0, 0, true),

        B_DES(CIPHER_DES, BLOCK_CIPHER, 8, 8, 0, true),

        B_3DES(CIPHER_3DES, BLOCK_CIPHER, 24, 8, 0, true),

        B_IDEA("IDEA", BLOCK_CIPHER, 16, 8, 0, false),

        B_AES_128(CIPHER_AES, BLOCK_CIPHER, 16, 16, 0, true),

        B_AES_256(CIPHER_AES, BLOCK_CIPHER, 32, 16, 0, true),

        B_AES_128_GCM(CIPHER_AES_GCM, AEAD_CIPHER, 16, 12, 4, true),

        B_AES_256_GCM(CIPHER_AES_GCM, AEAD_CIPHER, 32, 12, 4, true);

        // descriptive name including key size, e.g. AES/128

        final String description;

        // JCE cipher transformation string, e.g. AES/CBC/NoPadding

        final String transformation;

        // algorithm name, e.g. AES

        final String algorithm;

        // supported and compile time enabled. Also see isAvailable()

        final boolean allowed;

        // number of bytes of entropy in the key

        final int keySize;

        // length of the actual cipher key in bytes.

        // for non-exportable ciphers, this is the same as keySize

        final int expandedKeySize;

        // size of the IV

        final int ivSize;

        // size of fixed IV

        //

        // record_iv_length = ivSize - fixedIvSize

        final int fixedIvSize;

        // exportable under 512/40 bit rules

        final boolean exportable;

        // Is the cipher algorithm of Cipher Block Chaining (CBC) mode?

        final CipherType cipherType;

        // size of the authentication tag, only applicable to cipher suites in

        // Galois Counter Mode (GCM)

        //

        // As far as we know, all supported GCM cipher suites use 128-bits