/*

 * Copyright (c) 2015, 2016, 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 java.security.AlgorithmParameterGenerator;

import java.security.AlgorithmParameters;

import java.security.InvalidAlgorithmParameterException;

import java.security.InvalidParameterException;

import java.security.KeyPairGenerator;

import java.security.NoSuchAlgorithmException;

import java.security.NoSuchProviderException;

import java.security.spec.DSAGenParameterSpec;

import java.security.spec.DSAParameterSpec;

import java.security.spec.InvalidParameterSpecException;

import java.util.Arrays;

import java.util.List;

/*

 * @test

 * @bug 8075286

 * @key intermittent

 * @summary Verify that DSAGenParameterSpec can and can only be used to generate

 *          DSA within some certain range of key sizes as described in the class

 *          specification (L, N) as (1024, 160), (2048, 224), (2048, 256) and

 */

public class TestDSAGenParameterSpec {

    private static final String ALGORITHM_NAME = "DSA";

    private static final String PROVIDER_NAME = "SUN";

    private static void testDSAGenParameterSpec(DataTuple dataTuple)

            throws NoSuchAlgorithmException, NoSuchProviderException,

            InvalidParameterSpecException, InvalidAlgorithmParameterException {

        System.out.printf("Test case: primePLen=%d, " + "subprimeQLen=%d%n",

                dataTuple.primePLen, dataTuple.subprimeQLen);

        AlgorithmParameterGenerator apg =

                AlgorithmParameterGenerator.getInstance(ALGORITHM_NAME,

                        PROVIDER_NAME);

        DSAGenParameterSpec genParamSpec = createGenParameterSpec(dataTuple);

        // genParamSpec will be null if IllegalAE is thrown when expected.

        if (genParamSpec == null) {

            return;

        }

        try {

            apg.init(genParamSpec, null);

            AlgorithmParameters param = apg.generateParameters();

            checkParam(param, genParamSpec);

            System.out.println("Test case passed");

        } catch (InvalidParameterException ipe) {

        }

    }

    private static void checkParam(AlgorithmParameters param,

            DSAGenParameterSpec genParam) throws InvalidParameterSpecException,

                    NoSuchAlgorithmException, NoSuchProviderException,

                    InvalidAlgorithmParameterException {

        String algorithm = param.getAlgorithm();

        if (!algorithm.equalsIgnoreCase(ALGORITHM_NAME)) {

            throw new RuntimeException(

                    "Unexpected type of parameters: " + algorithm);

        }

        DSAParameterSpec spec = param.getParameterSpec(DSAParameterSpec.class);

        int valueL = spec.getP().bitLength();

        int strengthP = genParam.getPrimePLength();

        if (strengthP != valueL) {

            System.out.printf("P: Expected %d but actual %d%n", strengthP,

                    valueL);

            throw new RuntimeException("Wrong P strength");

        }

        int valueN = spec.getQ().bitLength();

        int strengthQ = genParam.getSubprimeQLength();

        if (strengthQ != valueN) {

            System.out.printf("Q: Expected %d but actual %d%n", strengthQ,

                    valueN);

            throw new RuntimeException("Wrong Q strength");

        }

        if (genParam.getSubprimeQLength() != genParam.getSeedLength()) {

            System.out.println("Defaut seed length should be the same as Q.");

            throw new RuntimeException("Wrong seed length");

        }

        KeyPairGenerator keyGen = KeyPairGenerator.getInstance(ALGORITHM_NAME,

                PROVIDER_NAME);

        keyGen.initialize(spec);

    }

    private static DSAGenParameterSpec createGenParameterSpec(

            DataTuple dataTuple) {

        DSAGenParameterSpec genParamSpec = null;

        try {

            genParamSpec = new DSAGenParameterSpec(dataTuple.primePLen,

                    dataTuple.subprimeQLen);

            if (!dataTuple.isDSASpecSupported) {

                throw new RuntimeException(

                        "Test case failed: the key length must not supported");

            }

        } catch (IllegalArgumentException e) {

            if (!dataTuple.isDSASpecSupported) {

                System.out.println("Test case passed: expected "

                        + "IllegalArgumentException is caught");

            } else {

                throw new RuntimeException("Test case failed: unexpected "

                        + "IllegalArgumentException is thrown", e);

            }

        }

        return genParamSpec;

    }

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

            testDSAGenParameterSpec(dataTuple);

        }

    }

    private static class DataTuple {

        private int primePLen;

        private int subprimeQLen;

        private boolean isDSASpecSupported;

        private DataTuple(int primePLen, int subprimeQLen,

            this.primePLen = primePLen;

            this.subprimeQLen = subprimeQLen;

            this.isDSASpecSupported = isDSASpecSupported;

        }

    }

}