/*

 * 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.

 */

package compiler.codegen.aes;

import jdk.test.lib.Utils;

import javax.crypto.Cipher;

import javax.crypto.SecretKey;

import javax.crypto.spec.GCMParameterSpec;

import javax.crypto.spec.IvParameterSpec;

import javax.crypto.spec.SecretKeySpec;

import java.security.AlgorithmParameters;

import java.util.Random;

/**

 * @author Tom Deneau

 */

public abstract class TestAESBase {

    int msgSize = Integer.getInteger("msgSize", 646);

    boolean checkOutput = Boolean.getBoolean("checkOutput");

    boolean noReinit = Boolean.getBoolean("noReinit");

    boolean testingMisalignment;

    private static final int ALIGN = 8;

    int encInputOffset = Integer.getInteger("encInputOffset", 0) % ALIGN;

    int encOutputOffset = Integer.getInteger("encOutputOffset", 0) % ALIGN;

    int decOutputOffset = Integer.getInteger("decOutputOffset", 0) % ALIGN;

    int lastChunkSize = Integer.getInteger("lastChunkSize", 32);

    int keySize = Integer.getInteger("keySize", 128);

    int inputLength;

    int encodeLength;

    int decodeLength;

    int decodeMsgSize;

    String algorithm = System.getProperty("algorithm", "AES");

    String mode = System.getProperty("mode", "CBC");

    byte[] input;

    byte[] encode;

    byte[] expectedEncode;

    byte[] decode;

    byte[] expectedDecode;

    final Random random = Utils.getRandomInstance();

    Cipher cipher;

    Cipher dCipher;

    AlgorithmParameters algParams = null;

    SecretKey key;

    GCMParameterSpec gcm_spec;

    byte[] aad = {0x11, 0x22, 0x33, 0x44, 0x55};

    int tlen = 12;

    byte[] iv = new byte[16];

    static int numThreads = 0;

    int threadId;

    static synchronized int getThreadId() {

        int id = numThreads;

        numThreads++;

        return id;

    }

    abstract public void run();

    public void prepare() {

        try {

            System.out.println("\nalgorithm=" + algorithm + ", mode=" + mode + ", paddingStr=" + paddingStr +

                    ", msgSize=" + msgSize + ", keySize=" + keySize + ", noReinit=" + noReinit +

                    ", checkOutput=" + checkOutput + ", encInputOffset=" + encInputOffset + ", encOutputOffset=" +

                    encOutputOffset + ", decOutputOffset=" + decOutputOffset + ", lastChunkSize=" + lastChunkSize);

            if (encInputOffset % ALIGN != 0 || encOutputOffset % ALIGN != 0 || decOutputOffset % ALIGN != 0)

                testingMisalignment = true;

            int keyLenBytes = (keySize == 0 ? 16 : keySize / 8);

            byte keyBytes[] = new byte[keyLenBytes];

            if (keySize == 128)

                keyBytes = new byte[]{-8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7};

            else

                random.nextBytes(keyBytes);

            key = new SecretKeySpec(keyBytes, algorithm);

            if (threadId == 0) {

                System.out.println("Algorithm: " + key.getAlgorithm() + "("

                        + key.getEncoded().length * 8 + "bit)");

            }

            cipher = Cipher.getInstance(algorithm + "/" + mode + "/" + paddingStr, "SunJCE");

            dCipher = Cipher.getInstance(algorithm + "/" + mode + "/" + paddingStr, "SunJCE");

            // CBC or CTR init

            if (mode.equals("CBC") || mode.equals("CTR")) {

                IvParameterSpec initVector = new IvParameterSpec(iv);

                cipher.init(Cipher.ENCRYPT_MODE, key, initVector);

                algParams = cipher.getParameters();

                dCipher.init(Cipher.DECRYPT_MODE, key, initVector);

                // GCM init

            } else if (mode.equals("GCM")) {

                gcm_init(true);

                gcm_init(false);

                // ECB init

            } else {

                cipher.init(Cipher.ENCRYPT_MODE, key, algParams);

                dCipher.init(Cipher.DECRYPT_MODE, key, algParams);

            }

            if (threadId == 0) {

                childShowCipher();

            }

            inputLength = msgSize + encInputOffset;

            if (testingMisalignment) {

                encodeLength = cipher.getOutputSize(msgSize - lastChunkSize) + encOutputOffset;

                encodeLength += cipher.getOutputSize(lastChunkSize);

                decodeLength = dCipher.getOutputSize(encodeLength - lastChunkSize) + decOutputOffset;

                decodeLength += dCipher.getOutputSize(lastChunkSize);

            } else {

                encodeLength = cipher.getOutputSize(msgSize) + encOutputOffset;

                decodeLength = dCipher.getOutputSize(encodeLength) + decOutputOffset;

            }

            input = new byte[inputLength];

            for (int i = encInputOffset, j = 0; i < inputLength; i++, j++) {

                input[i] = (byte) (j & 0xff);

            }

            // do one encode and decode in preparation

            encode = new byte[encodeLength];

            decode = new byte[decodeLength];

            if (testingMisalignment) {

                decodeMsgSize = cipher.update(input, encInputOffset, (msgSize - lastChunkSize), encode, encOutputOffset);

                decodeMsgSize += cipher.doFinal(input, (encInputOffset + msgSize - lastChunkSize), lastChunkSize, encode, (encOutputOffset + decodeMsgSize));

                int tempSize = dCipher.update(encode, encOutputOffset, (decodeMsgSize - lastChunkSize), decode, decOutputOffset);

                dCipher.doFinal(encode, (encOutputOffset + decodeMsgSize - lastChunkSize), lastChunkSize, decode, (decOutputOffset + tempSize));

            } else {

                decodeMsgSize = cipher.doFinal(input, encInputOffset, msgSize, encode, encOutputOffset);

                dCipher.doFinal(encode, encOutputOffset, decodeMsgSize, decode, decOutputOffset);

            }

            if (checkOutput) {

                expectedEncode = (byte[]) encode.clone();

                expectedDecode = (byte[]) decode.clone();

                showArray(key.getEncoded(), "key:    ");

                showArray(input, "input:  ");

                showArray(encode, "encode: ");

                showArray(decode, "decode: ");

            }

        } catch (Exception e) {

            e.printStackTrace();

            System.exit(1);

        }

    }

    void showArray(byte b[], String name) {

        System.out.format("%s [%d]: ", name, b.length);

        for (int i = 0; i < Math.min(b.length, 32); i++) {

            System.out.format("%02x ", b[i] & 0xff);

        }

        System.out.println();

    }

    void compareArrays(byte b[], byte exp[]) {

        if (b.length != exp.length) {

            System.out.format("different lengths for actual and expected output arrays\n");

            showArray(b, "test: ");

            showArray(exp, "exp : ");

            System.exit(1);

        }

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

            if (b[i] != exp[i]) {

                System.out.format("output error at index %d: got %02x, expected %02x\n", i, b[i] & 0xff, exp[i] & 0xff);

                showArray(b, "test: ");

                showArray(exp, "exp : ");

                System.exit(1);

            }

        }

    }

    void showCipher(Cipher c, String kind) {

        System.out.println(kind + " cipher provider: " + cipher.getProvider());

        System.out.println(kind + " cipher algorithm: " + cipher.getAlgorithm());

    }

    abstract void childShowCipher();

    void gcm_init(boolean encrypt) throws Exception {

        gcm_spec = new GCMParameterSpec(tlen * 8, iv);

        if (encrypt) {

            // Get a new instance everytime because of reuse IV restrictions

            cipher = Cipher.getInstance(algorithm + "/" + mode + "/" + paddingStr, "SunJCE");

            cipher.init(Cipher.ENCRYPT_MODE, key, gcm_spec);

            cipher.updateAAD(aad);

        } else {

            dCipher.init(Cipher.DECRYPT_MODE, key, gcm_spec);

            dCipher.updateAAD(aad);

        }

    }

}