/*

 * Copyright (c) 2014, 2015, 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 8029849 8132082 8133535

 * @summary Make sure signing via encrypt and verifying via decrypt are not

 * supported by OracleUcrypto provider.

 * @author Anthony Scarpino

 * @key randomness

 */

import java.util.Random;

import java.security.*;

import java.security.interfaces.*;

import java.security.spec.RSAPrivateKeySpec;

import javax.crypto.Cipher;

public class CipherSignNotSupported extends UcryptoTest {

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

        main(new CipherSignNotSupported(), null);

    }

    public void doTest(Provider p) throws Exception {

        Cipher c = null;

        Random random = new Random();

        byte[] pt = new byte[117];

        byte[] ct = new byte[200];

        random.nextBytes(pt);

        try {

            c = Cipher.getInstance("RSA/ECB/PKCS1Padding", p);

        } catch (NoSuchAlgorithmException e) {

            if (System.getProperty("os.version").compareTo("5.10") == 0) {

                System.out.println("RSA not supported in S10");

                return;

            }

            throw e;

        }

        KeyPairGenerator kpg = KeyPairGenerator.getInstance("RSA");

        kpg.initialize(1024);

        KeyPair kp = kpg.generateKeyPair();

        // Encryption

        c.init(Cipher.ENCRYPT_MODE, kp.getPublic());

        ct = c.doFinal(pt);

        // Decryption

        PrivateKey[] privKeys = new PrivateKey[2];

        privKeys[0] = kp.getPrivate();

        if (privKeys[0] instanceof RSAPrivateCrtKey) {

            RSAPrivateCrtKey k = (RSAPrivateCrtKey) privKeys[0];

            KeyFactory kf = KeyFactory.getInstance("RSA");

            privKeys[1] = kf.generatePrivate

                (new RSAPrivateKeySpec(k.getModulus(), k.getPrivateExponent()));

        } else {

            privKeys = new PrivateKey[] {privKeys[0]};

        }

        for (PrivateKey pk : privKeys) {

            System.out.println("Testing " + pk);

            c.init(Cipher.DECRYPT_MODE, pk);

            c.doFinal(ct);

            // Sign

            try {

                c.init(Cipher.ENCRYPT_MODE, pk);

                ct = c.doFinal(pt);

                throw new RuntimeException("Encrypt operation should have failed.");

            } catch (InvalidKeyException e) {

                if (!e.getMessage().contains("RSAPublicKey required for encryption")) {

                    System.out.println("Wrong exception thrown.");

                    throw e;

                }

            }

        }

        // Verify

        try {

            c.init(Cipher.DECRYPT_MODE, kp.getPublic());

            c.doFinal(ct);

            throw new RuntimeException("Decrypt operation should have failed.");

        } catch (InvalidKeyException e) {

            if (!e.getMessage().contains("RSAPrivateKey required for decryption")) {

                System.out.println("Wrong exception thrown.");

                throw e;

            }

        }

        System.out.println("Pass");

    }

}