230 lines
7.6 KiB
Java
230 lines
7.6 KiB
Java
|
/*
|
||
|
* 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;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|