PaddingTest.java revision 6073:cea72c2bf071 
1/*
2 * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test
26 * @bug 0000000 6296075 6330275
27 * @summary PaddingTest
28 * @author Jan Luehe
29 */
30import java.io.*;
31import java.nio.file.Files;
32import java.nio.file.Paths;
33import java.security.spec.*;
34import javax.crypto.*;
35import javax.crypto.spec.*;
36import java.util.Arrays;
37
38public class PaddingTest {
39
40    Cipher cipher;
41    IvParameterSpec params = null;
42    SecretKey cipherKey = null;
43    String pinfile = null;
44    String cfile = null;
45    String poutfile = null;
46
47    public static byte[] key = {
48        (byte)0x01,(byte)0x23,(byte)0x45,(byte)0x67,
49        (byte)0x89,(byte)0xab,(byte)0xcd,(byte)0xef
50    };
51
52    public static byte[] key3 = {
53        (byte)0x01,(byte)0x23,(byte)0x45,(byte)0x67,
54        (byte)0x89,(byte)0xab,(byte)0xcd,(byte)0xef,
55        (byte)0xf0,(byte)0xe1,(byte)0xd2,(byte)0xc3,
56        (byte)0xb4,(byte)0xa5,(byte)0x96,(byte)0x87,
57        (byte)0xfe,(byte)0xdc,(byte)0xba,(byte)0x98,
58        (byte)0x76,(byte)0x54,(byte)0x32,(byte)0x10};
59
60    public static byte[] iv  = {
61        (byte)0xfe,(byte)0xdc,(byte)0xba,(byte)0x98,
62        (byte)0x76,(byte)0x54,(byte)0x32,(byte)0x10};
63
64    static String[] crypts = {"DES", "DESede"};
65    static String[] modes = {"ECB", "CBC", "CFB", "OFB", "PCBC"};
66    static String[] paddings = {"PKCS5Padding", "NoPadding"};
67    static int numFiles = 11;
68    static final String currDir = System.getProperty("test.src", ".");
69    static String dataDir = currDir + "/inputData/";
70
71    private String padding = null;
72
73    public static void main(String argv[]) throws Exception {
74        PaddingTest pt = new PaddingTest();
75        pt.run();
76    }
77
78    public PaddingTest() {
79    }
80
81    public void run() throws Exception {
82
83        for (int l=0; l<numFiles; l++) {
84            pinfile = dataDir + "plain" + l + ".txt";
85            for (int i=0; i<crypts.length; i++) {
86                for (int j=0; j<modes.length; j++) {
87                    for (int k=0; k<paddings.length; k++) {
88                        System.out.println
89                            ("===============================");
90                        System.out.println
91                            (crypts[i]+" "+modes[j]+" " + paddings[k]+ " " +
92                             "plain" + l + " test");
93                        cfile = "c" + l + "_" +
94                             crypts[i] + "_" +
95                             modes[j] + "_" +
96                             paddings[k] + ".bin";
97                        poutfile = "p" + l +
98                             "_" + crypts[i] + modes[j] + paddings[k] + ".txt";
99
100                        init(crypts[i], modes[j], paddings[k]);
101                        padding = paddings[k];
102                        runTest();
103                    }
104                }
105            }
106        }
107    }
108
109    public void init(String crypt, String mode, String padding)
110            throws Exception {
111
112        KeySpec desKeySpec = null;
113        SecretKeyFactory factory = null;
114
115        StringBuffer cipherName = new StringBuffer(crypt);
116        if (mode.length() != 0)
117            cipherName.append("/" + mode);
118        if (padding.length() != 0)
119            cipherName.append("/" + padding);
120
121        cipher = Cipher.getInstance(cipherName.toString());
122        if (crypt.endsWith("ede")) {
123            desKeySpec = new DESedeKeySpec(key3);
124            factory = SecretKeyFactory.getInstance("DESede", "SunJCE");
125        } else {
126            desKeySpec = new DESKeySpec(key);
127            factory = SecretKeyFactory.getInstance("DES", "SunJCE");
128        }
129
130        // retrieve the cipher key
131        cipherKey = factory.generateSecret(desKeySpec);
132
133        // retrieve iv
134        if (!mode.equals("ECB"))
135            params = new IvParameterSpec(iv);
136        else
137            params = null;
138    }
139
140    public void runTest() throws Exception {
141
142        int bufferLen = 512;
143        byte[] input = new byte[bufferLen];
144        int len;
145        int totalInputLen = 0;
146
147        try {
148            try (FileInputStream fin = new FileInputStream(pinfile);
149                    BufferedInputStream pin = new BufferedInputStream(fin);
150                    FileOutputStream fout = new FileOutputStream(cfile);
151                    BufferedOutputStream cout = new BufferedOutputStream(fout)) {
152                cipher.init(Cipher.ENCRYPT_MODE, cipherKey, params);
153
154                while ((len = pin.read(input, 0, bufferLen)) > 0) {
155                    totalInputLen += len;
156                    byte[] output = cipher.update(input, 0, len);
157                    cout.write(output, 0, output.length);
158                }
159
160                len = cipher.getOutputSize(0);
161
162                byte[] out = new byte[len];
163                len = cipher.doFinal(out, 0);
164                cout.write(out, 0, len);
165            }
166
167            try (FileInputStream fin = new FileInputStream(cfile);
168                    BufferedInputStream cin = new BufferedInputStream(fin);
169                    FileOutputStream fout = new FileOutputStream(poutfile);
170                    BufferedOutputStream pout = new BufferedOutputStream(fout)) {
171                cipher.init(Cipher.DECRYPT_MODE, cipherKey, params);
172
173                byte[] output = null;
174                while ((len = cin.read(input, 0, bufferLen)) > 0) {
175                    output = cipher.update(input, 0, len);
176                    pout.write(output, 0, output.length);
177                }
178
179                len = cipher.getOutputSize(0);
180                byte[] out = new byte[len];
181                len = cipher.doFinal(out, 0);
182                pout.write(out, 0, len);
183            }
184
185            diff(pinfile, poutfile);
186        } catch (IllegalBlockSizeException ex) {
187            if ((totalInputLen % 8 != 0) && (padding.equals("NoPadding"))) {
188                return;
189            } else {
190                System.out.println("Test failed!");
191                throw ex;
192            }
193        }
194    }
195
196    private static void diff(String fname1, String fname2) throws Exception {
197        if (!Arrays.equals(Files.readAllBytes(Paths.get(fname1)),
198                Files.readAllBytes(Paths.get(fname2)))) {
199            throw new Exception(
200                    "files " + fname1 + " and " + fname2 + " differ");
201        }
202    }
203}
204