/*

 * Copyright (c) 2004, 2007, 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 4953556

 * @summary ensure that IllegalStateException is thrown if the

 * Cipher object is initialized with a wrong mode, e.g. WRAP_MODE

 * for update()/doFinal() calls.

 * @author Valerie Peng

 */

import java.security.*;

import java.security.spec.*;

import javax.crypto.*;

import javax.crypto.spec.SecretKeySpec;

public class TestCipherMode {

    private static final String ALGO = "DES";

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

        TestCipherMode test = new TestCipherMode();

        System.out.println("Testing ENCRYPT_MODE...");

        test.checkMode(Cipher.ENCRYPT_MODE);

        System.out.println("Testing DECRYPT_MODE...");

        test.checkMode(Cipher.DECRYPT_MODE);

        System.out.println("Testing WRAP_MODE...");

        test.checkMode(Cipher.WRAP_MODE);

        System.out.println("Testing UNWRAP_MODE...");

        test.checkMode(Cipher.UNWRAP_MODE);

        System.out.println("All Tests Passed");

   }

    private Cipher c = null;

    private SecretKey key = null;

    private TestCipherMode() throws NoSuchAlgorithmException,

    NoSuchProviderException, NoSuchPaddingException {

        c = Cipher.getInstance(ALGO + "/ECB/PKCS5Padding", "SunJCE");

        key = new SecretKeySpec(new byte[8], ALGO);

    }

    private void checkMode(int mode) throws Exception {

        c.init(mode, key);

        switch (mode) {

        case Cipher.ENCRYPT_MODE:

        case Cipher.DECRYPT_MODE:

            // call wrap()/unwrap() and see if ISE is thrown.

            try {

                c.wrap(key);

                throw new Exception("ERROR: should throw ISE for wrap()");

            } catch (IllegalStateException ise) {

                System.out.println("expected ISE is thrown for wrap()");

            }

            try {

                c.unwrap(new byte[16], ALGO, Cipher.SECRET_KEY);

                throw new Exception("ERROR: should throw ISE for unwrap()");

            } catch (IllegalStateException ise) {

                System.out.println("expected ISE is thrown for unwrap()");

            }

            break;

        case Cipher.WRAP_MODE:

        case Cipher.UNWRAP_MODE:

            try {

                c.update(new byte[16]);

                throw new Exception("ERROR: should throw ISE for update()");

            } catch (IllegalStateException ise) {

                System.out.println("expected ISE is thrown for update()");

            }

            try {

                c.doFinal();

                throw new Exception("ERROR: should throw ISE for doFinal()");

            } catch (IllegalStateException ise) {

                System.out.println("expected ISE is thrown for doFinal()");

            }

            break;

        }

    }

}