/*

 * Copyright 2002-2003 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

/*

 * @test

 * @bug 4750141 4895631

 * @summary Check enabled and supported ciphersuites are correct

 */

import java.util.*;

import javax.net.ssl.*;

import javax.crypto.Cipher;

import javax.crypto.spec.*;

public class CheckCipherSuites {

    private final static String[] ENABLED_DEFAULT = {

        "SSL_RSA_WITH_RC4_128_MD5",

        "SSL_RSA_WITH_RC4_128_SHA",

        "TLS_RSA_WITH_AES_128_CBC_SHA",

        "TLS_DHE_RSA_WITH_AES_128_CBC_SHA",

        "TLS_DHE_DSS_WITH_AES_128_CBC_SHA",

        "SSL_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_DSS_WITH_3DES_EDE_CBC_SHA",

        "SSL_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_DSS_WITH_DES_CBC_SHA",

        "SSL_RSA_EXPORT_WITH_RC4_40_MD5",

        "SSL_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_DSS_EXPORT_WITH_DES40_CBC_SHA",

    };

    private final static String[] ENABLED_UNLIMITED = {

        "SSL_RSA_WITH_RC4_128_MD5",

        "SSL_RSA_WITH_RC4_128_SHA",

        "TLS_RSA_WITH_AES_128_CBC_SHA",

        "TLS_RSA_WITH_AES_256_CBC_SHA",

        "TLS_DHE_RSA_WITH_AES_128_CBC_SHA",

        "TLS_DHE_RSA_WITH_AES_256_CBC_SHA",

        "TLS_DHE_DSS_WITH_AES_128_CBC_SHA",

        "TLS_DHE_DSS_WITH_AES_256_CBC_SHA",

        "SSL_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_DSS_WITH_3DES_EDE_CBC_SHA",

        "SSL_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_DSS_WITH_DES_CBC_SHA",

        "SSL_RSA_EXPORT_WITH_RC4_40_MD5",

        "SSL_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_DSS_EXPORT_WITH_DES40_CBC_SHA",

    };

    // supported ciphersuites using default JCE policy jurisdiction files

    // AES/256 unavailable

    private final static String[] SUPPORTED_DEFAULT = {

        "SSL_RSA_WITH_RC4_128_MD5",

        "SSL_RSA_WITH_RC4_128_SHA",

        "TLS_RSA_WITH_AES_128_CBC_SHA",

        "TLS_DHE_RSA_WITH_AES_128_CBC_SHA",

        "TLS_DHE_DSS_WITH_AES_128_CBC_SHA",

        "SSL_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_DSS_WITH_3DES_EDE_CBC_SHA",

        "SSL_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_DSS_WITH_DES_CBC_SHA",

        "SSL_RSA_EXPORT_WITH_RC4_40_MD5",

        "SSL_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_DSS_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_RSA_WITH_NULL_MD5",

        "SSL_RSA_WITH_NULL_SHA",

        "SSL_DH_anon_WITH_RC4_128_MD5",

        "TLS_DH_anon_WITH_AES_128_CBC_SHA",

        "SSL_DH_anon_WITH_3DES_EDE_CBC_SHA",

        "SSL_DH_anon_WITH_DES_CBC_SHA",

        "SSL_DH_anon_EXPORT_WITH_RC4_40_MD5",

        "SSL_DH_anon_EXPORT_WITH_DES40_CBC_SHA",

        "TLS_KRB5_WITH_RC4_128_SHA",

        "TLS_KRB5_WITH_RC4_128_MD5",

        "TLS_KRB5_WITH_3DES_EDE_CBC_SHA",

        "TLS_KRB5_WITH_3DES_EDE_CBC_MD5",

        "TLS_KRB5_WITH_DES_CBC_SHA",

        "TLS_KRB5_WITH_DES_CBC_MD5",

        "TLS_KRB5_EXPORT_WITH_RC4_40_SHA",

        "TLS_KRB5_EXPORT_WITH_RC4_40_MD5",

        "TLS_KRB5_EXPORT_WITH_DES_CBC_40_SHA",

        "TLS_KRB5_EXPORT_WITH_DES_CBC_40_MD5"

    };

    // supported ciphersuites using unlimited JCE policy jurisdiction files

    // AES/256 available

    private final static String[] SUPPORTED_UNLIMITED = {

        "SSL_RSA_WITH_RC4_128_MD5",

        "SSL_RSA_WITH_RC4_128_SHA",

        "TLS_RSA_WITH_AES_128_CBC_SHA",

        "TLS_RSA_WITH_AES_256_CBC_SHA",

        "TLS_DHE_RSA_WITH_AES_128_CBC_SHA",

        "TLS_DHE_RSA_WITH_AES_256_CBC_SHA",

        "TLS_DHE_DSS_WITH_AES_128_CBC_SHA",

        "TLS_DHE_DSS_WITH_AES_256_CBC_SHA",

        "SSL_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_RSA_WITH_3DES_EDE_CBC_SHA",

        "SSL_DHE_DSS_WITH_3DES_EDE_CBC_SHA",

        "SSL_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_RSA_WITH_DES_CBC_SHA",

        "SSL_DHE_DSS_WITH_DES_CBC_SHA",

        "SSL_RSA_EXPORT_WITH_RC4_40_MD5",

        "SSL_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_RSA_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_DHE_DSS_EXPORT_WITH_DES40_CBC_SHA",

        "SSL_RSA_WITH_NULL_MD5",

        "SSL_RSA_WITH_NULL_SHA",

        "SSL_DH_anon_WITH_RC4_128_MD5",

        "TLS_DH_anon_WITH_AES_128_CBC_SHA",

        "TLS_DH_anon_WITH_AES_256_CBC_SHA",

        "SSL_DH_anon_WITH_3DES_EDE_CBC_SHA",

        "SSL_DH_anon_WITH_DES_CBC_SHA",

        "SSL_DH_anon_EXPORT_WITH_RC4_40_MD5",

        "SSL_DH_anon_EXPORT_WITH_DES40_CBC_SHA",

        "TLS_KRB5_WITH_RC4_128_SHA",

        "TLS_KRB5_WITH_RC4_128_MD5",

        "TLS_KRB5_WITH_3DES_EDE_CBC_SHA",

        "TLS_KRB5_WITH_3DES_EDE_CBC_MD5",

        "TLS_KRB5_WITH_DES_CBC_SHA",

        "TLS_KRB5_WITH_DES_CBC_MD5",

        "TLS_KRB5_EXPORT_WITH_RC4_40_SHA",

        "TLS_KRB5_EXPORT_WITH_RC4_40_MD5",

        "TLS_KRB5_EXPORT_WITH_DES_CBC_40_SHA",

        "TLS_KRB5_EXPORT_WITH_DES_CBC_40_MD5"

    };

    private static void showSuites(String[] suites) {

        if ((suites == null) || (suites.length == 0)) {

            System.out.println("<none>");

        }

        for (int i = 0; i < suites.length; i++) {

            System.out.println("  " + suites[i]);

        }

    }

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

        long start = System.currentTimeMillis();

        String[] ENABLED;

        String[] SUPPORTED;

        try {

            Cipher c = Cipher.getInstance("AES/CBC/NoPadding");

            SecretKeySpec key = new SecretKeySpec(new byte[32], "AES");

            c.init(Cipher.ENCRYPT_MODE, key);

            System.out.println("AES/256 is available");

            ENABLED = ENABLED_UNLIMITED;

            SUPPORTED = SUPPORTED_UNLIMITED;

        } catch (Exception e) {

            System.out.println("AES/256 is NOT available (" + e + ")");

            ENABLED = ENABLED_DEFAULT;

            SUPPORTED = SUPPORTED_DEFAULT;

        }

        SSLSocketFactory factory = (SSLSocketFactory)SSLSocketFactory.getDefault();

        SSLSocket socket = (SSLSocket)factory.createSocket();

        String[] enabled = socket.getEnabledCipherSuites();

        System.out.println("Default enabled ciphersuites:");

        showSuites(enabled);

        if (Arrays.equals(ENABLED, enabled) == false) {

            System.out.println("*** MISMATCH, should be ***");

            showSuites(ENABLED);

            throw new Exception("Enabled ciphersuite mismatch");

        }

        System.out.println("OK");

        System.out.println();

        String[] supported = socket.getSupportedCipherSuites();

        System.out.println("Supported ciphersuites:");

        showSuites(supported);

        if (Arrays.equals(SUPPORTED, supported) == false) {

            System.out.println("*** MISMATCH, should be ***");

            showSuites(SUPPORTED);

            throw new Exception("Supported ciphersuite mismatch");

        }

        System.out.println("OK");

        long end = System.currentTimeMillis();

        System.out.println("Done (" + (end - start) + " ms).");

    }

}