/*

 * Copyright (c) 2019, 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.

 */

/*

 * @test

 * @bug 8180392

 * @summary Ensure SunJCE provider throws exception for unsupported modes

 *          and padding combinations

 */

import java.security.*;

import java.security.spec.AlgorithmParameterSpec;

import javax.crypto.*;

public class TestNoPaddingModes {

    // SunJCE only supports NoPadding with following modes

    private static final String[] MODES = {

        "CTR", "CTS", "GCM"

    };

    private static final String[] PADDINGS = {

        "PKCS5Padding", "ISO10126Padding"

    };

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

        Provider p = Security.getProvider("SunJCE");

        String transformation;

        for (String mode : MODES) {

            for (String padding : PADDINGS) {

                transformation = "AES/" + mode + "/" + padding;

                System.out.println("Test using " + transformation);

                try {

                    Cipher c = Cipher.getInstance(transformation, "SunJCE");

                    throw new RuntimeException("=> Fail, no exception thrown");

                } catch (NoSuchAlgorithmException | NoSuchPaddingException ex) {

                    System.out.println("=> Expected ex: " + ex);

                }

                try {

                    Cipher c = Cipher.getInstance(transformation, p);

                    throw new RuntimeException("=> Fail, no exception thrown");

                } catch (NoSuchAlgorithmException | NoSuchPaddingException ex) {

                    System.out.println("=> Expected ex: " + ex);

                }

            }

        }

        System.out.println("Test Passed");

    }

    public static class MyCipImpl extends CipherSpi {

        public MyCipImpl() {

            super();

            System.out.println("MyCipImpl is created");

        }

        protected void engineSetMode(String mode)

            throws NoSuchAlgorithmException {};

        protected void engineSetPadding(String padding)

            throws NoSuchPaddingException {};

        protected int engineGetBlockSize() { return 16; }

        protected int engineGetOutputSize(int inputLen) { return 0; }

        protected byte[] engineGetIV() { return null; }

        protected AlgorithmParameters engineGetParameters() { return null; }

        protected void engineInit(int opmode, Key key, SecureRandom random)

            throws InvalidKeyException {};

        protected void engineInit(int opmode, Key key,

                                  AlgorithmParameterSpec params,

                                  SecureRandom random)

            throws InvalidKeyException, InvalidAlgorithmParameterException {};

        protected void engineInit(int opmode, Key key,

                                  AlgorithmParameters params,

                                  SecureRandom random)

            throws InvalidKeyException, InvalidAlgorithmParameterException {};

        protected byte[] engineUpdate(byte[] input, int inputOffset,

                                      int inputLen) { return null; }

        protected int engineUpdate(byte[] input, int inputOffset,

                                   int inputLen, byte[] output,

                                   int outputOffset)

            throws ShortBufferException { return 0; };

        protected byte[] engineDoFinal(byte[] input, int inputOffset,

                                       int inputLen)

            throws IllegalBlockSizeException, BadPaddingException {

            return null;

        }

        protected int engineDoFinal(byte[] input, int inputOffset,

                                    int inputLen, byte[] output,

                                    int outputOffset)

            throws ShortBufferException, IllegalBlockSizeException,

                   BadPaddingException  { return 0; }

    }

}