undefect. CWE-407 — 63 sites patched across 27 ecosystems
Authors: russell@unturf.com · brackishbert@gmail.com · foxhop.net · TimeHexOn.com Patches, unit tests, benchmarks, whitepaper, and outreach briefs. Public domain — no copyright claimed. Use freely.
This commit is contained in:
commit
0a580b313d
70422 changed files with 17213626 additions and 0 deletions
187
test/jdk/javax/crypto/Cipher/TestCipherMode.java
Normal file
187
test/jdk/javax/crypto/Cipher/TestCipherMode.java
Normal file
|
|
@ -0,0 +1,187 @@
|
|||
/*
|
||||
* Copyright (c) 2004, 2024, 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 4953556 8210838 8248268
|
||||
* @summary ensure that IllegalStateException is thrown if the
|
||||
* Cipher object is initialized with a wrong mode, e.g. WRAP_MODE
|
||||
* for update()/doFinal() calls.
|
||||
* @author Valerie Peng
|
||||
*/
|
||||
|
||||
|
||||
import java.security.*;
|
||||
import java.security.spec.*;
|
||||
import java.util.Arrays;
|
||||
|
||||
import javax.crypto.*;
|
||||
import javax.crypto.spec.SecretKeySpec;
|
||||
|
||||
public class TestCipherMode {
|
||||
|
||||
private static final String[] TRANSFORMATIONS = {
|
||||
"DES/ECB/PKCS5Padding", // CipherCore
|
||||
"AES/GCM/NoPadding", // GaloisCounterMode
|
||||
"AES/KW/NoPadding", // KeyWrapCipher
|
||||
"AES/KW/PKCS5Padding", // KeyWrapCipher
|
||||
"AES/KWP/NoPadding", // KeyWrapCipher
|
||||
"RSA/ECB/NoPadding", // RSACipher
|
||||
"DESedeWrap/CBC/NoPadding", // DESedeWrapCipher
|
||||
"ChaCha20-Poly1305", // ChaCha20Cipher
|
||||
};
|
||||
|
||||
private static final byte[] BYTES32 =
|
||||
Arrays.copyOf(TRANSFORMATIONS[0].getBytes(), 32);
|
||||
private static final SecretKey DES_KEY =
|
||||
new SecretKeySpec(BYTES32, 0, 8, "DES");
|
||||
private static final SecretKey AES_KEY =
|
||||
new SecretKeySpec(BYTES32, 0, 16, "AES");
|
||||
|
||||
private static enum CipherMode {
|
||||
ENCRYPT(Cipher.ENCRYPT_MODE),
|
||||
DECRYPT(Cipher.DECRYPT_MODE),
|
||||
WRAP(Cipher.WRAP_MODE),
|
||||
UNWRAP(Cipher.UNWRAP_MODE),
|
||||
NONEXISTENT(100);
|
||||
|
||||
int value;
|
||||
|
||||
CipherMode(int value) {
|
||||
this.value = value;
|
||||
}
|
||||
}
|
||||
|
||||
private static Key getKey(String t, CipherMode m)
|
||||
throws NoSuchAlgorithmException, NoSuchProviderException {
|
||||
Key key;
|
||||
String algo = t.split("/")[0];
|
||||
switch (algo) {
|
||||
case "AES":
|
||||
key = AES_KEY;
|
||||
break;
|
||||
case "RSA":
|
||||
KeyPairGenerator kpg = KeyPairGenerator.getInstance(algo);
|
||||
KeyPair kp = kpg.generateKeyPair();
|
||||
key = ((m == CipherMode.ENCRYPT || m == CipherMode.UNWRAP)?
|
||||
kp.getPrivate() : kp.getPublic());
|
||||
break;
|
||||
case "ChaCha20-Poly1305":
|
||||
key = new SecretKeySpec(BYTES32, 0, 32, "ChaCha20");
|
||||
break;
|
||||
case "DES":
|
||||
key = new SecretKeySpec(BYTES32, 0, 8, algo);
|
||||
break;
|
||||
case "DESedeWrap":
|
||||
key = new SecretKeySpec(BYTES32, 0, 24, "DESede");
|
||||
break;
|
||||
default:
|
||||
throw new RuntimeException("Unknown transformation: " + t);
|
||||
}
|
||||
return key;
|
||||
}
|
||||
|
||||
public static void main(String[] argv) throws Exception {
|
||||
|
||||
TestCipherMode test = new TestCipherMode(
|
||||
System.getProperty("test.provider.name", "SunJCE"),
|
||||
TRANSFORMATIONS);
|
||||
System.out.println("All Tests Passed");
|
||||
}
|
||||
|
||||
private Cipher c = null;
|
||||
private SecretKey key = null;
|
||||
|
||||
private TestCipherMode(String provName, String... transformations)
|
||||
throws Exception {
|
||||
|
||||
System.out.println("Testing " + provName);
|
||||
|
||||
for (String t : transformations) {
|
||||
for (CipherMode m : CipherMode.values()) {
|
||||
checkMode(t, m, provName);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void checkMode(String t, CipherMode mode, String provName)
|
||||
throws Exception {
|
||||
Cipher c = Cipher.getInstance(t, provName);
|
||||
Key key = getKey(t, mode);
|
||||
|
||||
System.out.println(c.getAlgorithm() + " with " + mode.name());
|
||||
try {
|
||||
c.init(mode.value, key, c.getParameters());
|
||||
if (mode == CipherMode.NONEXISTENT) {
|
||||
throw new Exception("ERROR: should throw IPE for init()");
|
||||
}
|
||||
} catch (UnsupportedOperationException uoe) {
|
||||
// some may not support wrap/unwrap or enc/dec
|
||||
if (mode != CipherMode.NONEXISTENT) {
|
||||
System.out.println("Expected UOE thrown with init()");
|
||||
return;
|
||||
}
|
||||
throw uoe;
|
||||
} catch (InvalidParameterException ipe) {
|
||||
if (mode == CipherMode.NONEXISTENT) {
|
||||
System.out.println("=> expected IPE thrown for init()");
|
||||
return;
|
||||
}
|
||||
throw ipe;
|
||||
}
|
||||
|
||||
switch (mode) {
|
||||
case ENCRYPT:
|
||||
case DECRYPT:
|
||||
// call wrap()/unwrap() and see if ISE is thrown.
|
||||
try {
|
||||
c.wrap(key);
|
||||
throw new Exception("ERROR: should throw ISE for wrap()");
|
||||
} catch (IllegalStateException ise) {
|
||||
System.out.println("=> expected ISE thrown for wrap()");
|
||||
}
|
||||
try {
|
||||
c.unwrap(new byte[16], key.getAlgorithm(), Cipher.SECRET_KEY);
|
||||
throw new Exception("ERROR: should throw ISE for unwrap()");
|
||||
} catch (IllegalStateException ise) {
|
||||
System.out.println("=> expected ISE thrown for unwrap()");
|
||||
}
|
||||
break;
|
||||
case WRAP:
|
||||
case UNWRAP:
|
||||
try {
|
||||
c.update(new byte[16]);
|
||||
throw new Exception("ERROR: should throw ISE for update()");
|
||||
} catch (IllegalStateException ise) {
|
||||
System.out.println("=> expected ISE thrown for update()");
|
||||
}
|
||||
try {
|
||||
c.doFinal();
|
||||
throw new Exception("ERROR: should throw ISE for doFinal()");
|
||||
} catch (IllegalStateException ise) {
|
||||
System.out.println("=> expected ISE thrown for doFinal()");
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue