| /* |
| * Copyright (c) 1997, 2012, 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 0000000 6296075 6330275 |
| * @summary PaddingTest |
| * @author Jan Luehe |
| */ |
| import java.io.*; |
| import java.nio.file.Files; |
| import java.nio.file.Paths; |
| import java.security.spec.*; |
| import javax.crypto.*; |
| import javax.crypto.spec.*; |
| import java.util.Arrays; |
| |
| 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 = 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 = "c" + l + "_" + |
| crypts[i] + "_" + |
| modes[j] + "_" + |
| paddings[k] + ".bin"; |
| poutfile = "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 { |
| |
| 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; |
| |
| try { |
| try (FileInputStream fin = new FileInputStream(pinfile); |
| BufferedInputStream pin = new BufferedInputStream(fin); |
| FileOutputStream fout = new FileOutputStream(cfile); |
| BufferedOutputStream cout = new BufferedOutputStream(fout)) { |
| 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); |
| } |
| |
| len = cipher.getOutputSize(0); |
| |
| byte[] out = new byte[len]; |
| len = cipher.doFinal(out, 0); |
| cout.write(out, 0, len); |
| } |
| |
| try (FileInputStream fin = new FileInputStream(cfile); |
| BufferedInputStream cin = new BufferedInputStream(fin); |
| FileOutputStream fout = new FileOutputStream(poutfile); |
| BufferedOutputStream pout = new BufferedOutputStream(fout)) { |
| 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); |
| } |
| |
| len = cipher.getOutputSize(0); |
| byte[] out = new byte[len]; |
| len = cipher.doFinal(out, 0); |
| pout.write(out, 0, len); |
| } |
| |
| diff(pinfile, poutfile); |
| } catch (IllegalBlockSizeException ex) { |
| if ((totalInputLen % 8 != 0) && (padding.equals("NoPadding"))) { |
| return; |
| } else { |
| System.out.println("Test failed!"); |
| throw ex; |
| } |
| } |
| } |
| |
| private static void diff(String fname1, String fname2) throws Exception { |
| if (!Arrays.equals(Files.readAllBytes(Paths.get(fname1)), |
| Files.readAllBytes(Paths.get(fname2)))) { |
| throw new Exception( |
| "files " + fname1 + " and " + fname2 + " differ"); |
| } |
| } |
| } |