/*
 * Copyright 1997-2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/*
 * @test
 * @bug 0000000 6296075
 * @summary PaddingTest
 * @author Jan Luehe
 */
import java.io.*;
import java.security.*;
import java.security.spec.*;
import javax.crypto.*;
import javax.crypto.spec.*;
import com.sun.crypto.provider.*;

public class PaddingTest {

    Cipher cipher;
    IvParameterSpec params = null;
    SecretKey cipherKey = null;
    String pinfile = null;
    String cfile = null;
    String poutfile = null;

    public static byte[] key = {
        (byte)0x01,(byte)0x23,(byte)0x45,(byte)0x67,
        (byte)0x89,(byte)0xab,(byte)0xcd,(byte)0xef
    };

    public static byte[] key3 = {
        (byte)0x01,(byte)0x23,(byte)0x45,(byte)0x67,
        (byte)0x89,(byte)0xab,(byte)0xcd,(byte)0xef,
        (byte)0xf0,(byte)0xe1,(byte)0xd2,(byte)0xc3,
        (byte)0xb4,(byte)0xa5,(byte)0x96,(byte)0x87,
        (byte)0xfe,(byte)0xdc,(byte)0xba,(byte)0x98,
        (byte)0x76,(byte)0x54,(byte)0x32,(byte)0x10};

    public static byte[] iv  = {
        (byte)0xfe,(byte)0xdc,(byte)0xba,(byte)0x98,
        (byte)0x76,(byte)0x54,(byte)0x32,(byte)0x10};

    static String[] crypts = {"DES", "DESede"};
    static String[] modes = {"ECB", "CBC", "CFB", "OFB", "PCBC"};
    static String[] paddings = {"PKCS5Padding", "NoPadding"};
    static int numFiles = 11;
    static final String currDir = System.getProperty("test.src", ".");
    static String dataDir = currDir + "/inputData/";

    private String padding = null;

    public static void main(String argv[]) throws Exception {
        PaddingTest pt = new PaddingTest();
        pt.run();
    }

    public PaddingTest() {
    }

    public void run() throws Exception {

        for (int l=0; l<numFiles; l++) {
            pinfile = new String(dataDir + "plain" + l + ".txt");
            for (int i=0; i<crypts.length; i++) {
                for (int j=0; j<modes.length; j++) {
                    for (int k=0; k<paddings.length; k++) {
                        System.out.println
                            ("===============================");
                        System.out.println
                            (crypts[i]+" "+modes[j]+" " + paddings[k]+ " " +
                             "plain" + l + " test");
                        cfile = new String
                            ("c" + l + "_" +
                             crypts[i] + "_" +
                             modes[j] + "_" +
                             paddings[k] + ".bin");
                        poutfile = new String
                            ("p" + l +
                             "_" + crypts[i] + modes[j] + paddings[k] + ".txt");

                        init(crypts[i], modes[j], paddings[k]);
                        padding = paddings[k];
                        runTest();
                    }
                }
            }
        }
    }

    public void init(String crypt, String mode, String padding)
        throws Exception {

        SunJCE jce = new SunJCE();
        Security.addProvider(jce);

        KeySpec desKeySpec = null;
        SecretKeyFactory factory = null;

        StringBuffer cipherName = new StringBuffer(crypt);
        if (mode.length() != 0)
            cipherName.append("/" + mode);
        if (padding.length() != 0)
            cipherName.append("/" + padding);

        cipher = Cipher.getInstance(cipherName.toString());
        if (crypt.endsWith("ede")) {
            desKeySpec = new DESedeKeySpec(key3);
            factory = SecretKeyFactory.getInstance("DESede", "SunJCE");
        } else {
            desKeySpec = new DESKeySpec(key);
            factory = SecretKeyFactory.getInstance("DES", "SunJCE");
        }

        // retrieve the cipher key
        cipherKey = factory.generateSecret(desKeySpec);

        // retrieve iv
        if (!mode.equals("ECB"))
            params = new IvParameterSpec(iv);
        else
            params = null;
    }

    public void runTest() throws Exception {

        int bufferLen = 512;
        byte[] input = new byte[bufferLen];
        int len;
        int totalInputLen = 0;

        BufferedInputStream pin = null;
        BufferedOutputStream cout = null;
        BufferedInputStream cin = null;
        BufferedOutputStream pout = null;

        try {
            pin = new BufferedInputStream(new FileInputStream(pinfile));
            cout = new BufferedOutputStream(new FileOutputStream(cfile));
            cipher.init(Cipher.ENCRYPT_MODE, cipherKey, params);

            while ((len = pin.read(input, 0, bufferLen)) > 0) {
                totalInputLen += len;
                byte[] output = cipher.update(input, 0, len);
                cout.write(output, 0, output.length);
                cout.flush();
            }

            len = cipher.getOutputSize(0);

            byte[] out = new byte[len];
            len = cipher.doFinal(out, 0);
            cout.write(out, 0, len);
            cout.flush();

            cin = new BufferedInputStream(new FileInputStream(cfile));
            pout = new BufferedOutputStream(new FileOutputStream(poutfile));
            cipher.init(Cipher.DECRYPT_MODE, cipherKey, params);

            byte[] output = null;
            while ((len = cin.read(input, 0, bufferLen)) > 0) {
                output = cipher.update(input, 0, len);
                pout.write(output, 0, output.length);
                pout.flush();
            }

            len = cipher.getOutputSize(0);
            out = new byte[len];
            len = cipher.doFinal(out, 0);
            pout.write(out, 0, len);
            pout.flush();

            Process child = Runtime.getRuntime().exec
                ("diff " + pinfile + " " + poutfile);
            InputStream in = child.getInputStream();
            byte[] data = new byte[64];

            while((len = in.read(data)) != -1)
                System.out.write(data, 0, len);
            in.close();
            child.waitFor();
            System.out.println("child exited with " + child.exitValue());
        }
        catch (IllegalBlockSizeException ex) {
            if ((totalInputLen % 8 != 0) && (padding.equals("NoPadding")))
                return;
            else {
                System.out.println("Test failed!");
                throw ex;
            }
        }
        finally {
            try {
                if (pin != null)
                    pin.close();
                if (pout != null)
                    pout.close();
                if (cin != null)
                    cin.close();
                if (cout != null)
                    cout.close();
            }
            catch (IOException e) {
                e.printStackTrace();
                return;
            }
        }
    }

}