/*

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

 */

/*

 * @test

 * @bug 4898484 6604496 8001284

 * @summary basic test for symmetric ciphers with no padding

 * @author Valerie Peng

 * @library ..

 * @key randomness

 * @run main/othervm TestSymmCiphersNoPad

 * @run main/othervm TestSymmCiphersNoPad sm

 */

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.InputStream;

import java.nio.ByteBuffer;

import java.security.AlgorithmParameters;

import java.security.NoSuchAlgorithmException;

import java.security.Provider;

import java.util.Random;

import javax.crypto.Cipher;

import javax.crypto.CipherInputStream;

import javax.crypto.KeyGenerator;

import javax.crypto.SecretKey;

public class TestSymmCiphersNoPad extends PKCS11Test {

    private static class CI { // class for holding Cipher Information

        String transformation;

        String keyAlgo;

        int dataSize;

        CI(String transformation, String keyAlgo, int dataSize) {

            this.transformation = transformation;

            this.keyAlgo = keyAlgo;

            this.dataSize = dataSize;

        }

    }

    private static final CI TEST_LIST[] = {

        new CI("ARCFOUR", "ARCFOUR", 400),

        new CI("RC4", "RC4", 401),

        new CI("DES/CBC/NoPadding", "DES", 400),

        new CI("DESede/CBC/NoPadding", "DESede", 160),

        new CI("AES/CBC/NoPadding", "AES", 4800),

        new CI("Blowfish/CBC/NoPadding", "Blowfish", 24),

        new CI("AES/CTR/NoPadding", "AES", 1600),

        new CI("AES/CTR/NoPadding", "AES", 65)

    };

    private static StringBuffer debugBuf;

    @Override

    public void main(Provider p) throws Exception {

        boolean status = true;

        Random random = new Random();

        try {

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

                CI currTest = TEST_LIST[i];

                System.out.println("===" + currTest.transformation + "===");

                try {

                    KeyGenerator kg =

                        KeyGenerator.getInstance(currTest.keyAlgo, p);

                    SecretKey key = kg.generateKey();

                    Cipher c1 = Cipher.getInstance(currTest.transformation, p);

                    Cipher c2 = Cipher.getInstance(currTest.transformation,

                                                   "SunJCE");

                    byte[] plainTxt = new byte[currTest.dataSize];

                    random.nextBytes(plainTxt);

                    System.out.println("Testing inLen = " + plainTxt.length);

                    c2.init(Cipher.ENCRYPT_MODE, key);

                    AlgorithmParameters params = c2.getParameters();

                    byte[] answer = c2.doFinal(plainTxt);

                    test(c1, Cipher.ENCRYPT_MODE, key, params,

                         plainTxt, answer);

                    System.out.println("Encryption tests: DONE");

                    c2.init(Cipher.DECRYPT_MODE, key, params);

                    byte[] answer2 = c2.doFinal(answer);

                    test(c1, Cipher.DECRYPT_MODE, key, params,

                         answer, answer2);

                    System.out.println("Decryption tests: DONE");

                } catch (NoSuchAlgorithmException nsae) {

                    System.out.println("Skipping unsupported algorithm: " +

                                       nsae);

                }

            }

        } catch (Exception ex) {

            // print out debug info when exception is encountered

            if (debugBuf != null) {

                System.out.println(debugBuf.toString());

            }

            throw ex;

        }

    }

    private static void test(Cipher cipher, int mode, SecretKey key,

                             AlgorithmParameters params,

                             byte[] in, byte[] answer) throws Exception {

        // test setup

        debugBuf = new StringBuffer();

        cipher.init(mode, key, params);

        int outLen = cipher.getOutputSize(in.length);

        debugBuf.append("Estimated output size = " + outLen + "\n");

        // test data preparation

        ByteBuffer inBuf = ByteBuffer.allocate(in.length);

        inBuf.put(in);

        inBuf.position(0);

        ByteBuffer inDirectBuf = ByteBuffer.allocateDirect(in.length);

        inDirectBuf.put(in);

        inDirectBuf.position(0);

        ByteBuffer outBuf = ByteBuffer.allocate(outLen);

        ByteBuffer outDirectBuf = ByteBuffer.allocateDirect(outLen);

        // test#1: byte[] in + byte[] out

        debugBuf.append("Test#1:\n");

        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        byte[] testOut1 = cipher.update(in, 0, 16);

        if (testOut1 != null) baos.write(testOut1, 0, testOut1.length);

        testOut1 = cipher.doFinal(in, 16, in.length-16);

        if (testOut1 != null) baos.write(testOut1, 0, testOut1.length);

        testOut1 = baos.toByteArray();

        match(testOut1, answer);

        // test#2: Non-direct Buffer in + non-direct Buffer out

        debugBuf.append("Test#2:\n");

        debugBuf.append("inputBuf: " + inBuf + "\n");

        debugBuf.append("outputBuf: " + outBuf + "\n");

        cipher.update(inBuf, outBuf);

        cipher.doFinal(inBuf, outBuf);

        match(outBuf, answer);

        // test#3: Direct Buffer in + direc Buffer out

        debugBuf.append("Test#3:\n");

        debugBuf.append("(pre) inputBuf: " + inDirectBuf + "\n");

        debugBuf.append("(pre) outputBuf: " + outDirectBuf + "\n");

        cipher.update(inDirectBuf, outDirectBuf);

        cipher.doFinal(inDirectBuf, outDirectBuf);

        debugBuf.append("(post) inputBuf: " + inDirectBuf + "\n");

        debugBuf.append("(post) outputBuf: " + outDirectBuf + "\n");

        match(outDirectBuf, answer);

        // test#4: Direct Buffer in + non-direct Buffer out

        debugBuf.append("Test#4:\n");

        inDirectBuf.position(0);

        outBuf.position(0);

        debugBuf.append("inputBuf: " + inDirectBuf + "\n");

        debugBuf.append("outputBuf: " + outBuf + "\n");

        cipher.update(inDirectBuf, outBuf);

        cipher.doFinal(inDirectBuf, outBuf);

        match(outBuf, answer);

        // test#5: Non-direct Buffer in + direct Buffer out

        debugBuf.append("Test#5:\n");

        inBuf.position(0);

        outDirectBuf.position(0);

        debugBuf.append("(pre) inputBuf: " + inBuf + "\n");

        debugBuf.append("(pre) outputBuf: " + outDirectBuf + "\n");

        cipher.update(inBuf, outDirectBuf);

        cipher.doFinal(inBuf, outDirectBuf);

        debugBuf.append("(post) inputBuf: " + inBuf + "\n");

        debugBuf.append("(post) outputBuf: " + outDirectBuf + "\n");

        match(outDirectBuf, answer);

        // test#6: Streams

        debugBuf.append("Test#6: Streaming\n");

        outBuf.position(0);

        InputStream stream =

            new CipherInputStream(new ByteArrayInputStream(in), cipher);

        byte[] data = new byte[1024];

        int bytesRead = 0;

        try {

            while (bytesRead >= 0) {

                bytesRead = stream.read(data);

                if (bytesRead == -1)

                    break;

                debugBuf.append("bytesRead: " + bytesRead);

                debugBuf.append("\toutBuf.position(): " + outBuf.position() +

                    "\n");

                outBuf.put(data, 0 , bytesRead);

            }

        } catch (Exception ex) {

            debugBuf.append("Caught Exception during stream reading\n");

            throw ex;

        }

        match(outBuf, answer);

        debugBuf = null;

    }

    private static void match(byte[] b1, byte[] b2) throws Exception {

        if (b1.length != b2.length) {

            debugBuf.append("got len   : " + b1.length + "\n");

            debugBuf.append("expect len: " + b2.length + "\n");

            throw new Exception("mismatch - different length!\n");

        } else {

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

                if (b1[i] != b2[i]) {

                    debugBuf.append("got   : " + toString(b1) + "\n");

                    debugBuf.append("expect: " + toString(b2) + "\n");

                    throw new Exception("mismatch");

                }

            }

        }

    }

    private static void match(ByteBuffer bb, byte[] answer) throws Exception {

        byte[] bbTemp = new byte[bb.position()];

        bb.position(0);

        bb.get(bbTemp, 0, bbTemp.length);

        match(bbTemp, answer);

    }

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

        main(new TestSymmCiphersNoPad(), args);

    }

}