282 lines
9.4 KiB
Java
282 lines
9.4 KiB
Java
/*
|
|
* Copyright (c) 2004, 2021, 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 5019096
|
|
* @summary Add scatter/gather APIs for SSLEngine
|
|
*
|
|
* Check to see if the args are being parsed properly.
|
|
*
|
|
*/
|
|
|
|
import javax.net.ssl.*;
|
|
import javax.net.ssl.SSLEngineResult.*;
|
|
import java.io.*;
|
|
import java.security.*;
|
|
import java.nio.*;
|
|
|
|
public class ArgCheck {
|
|
|
|
private static boolean debug = false;
|
|
|
|
private static String pathToStores = "../etc";
|
|
private static String keyStoreFile = "keystore";
|
|
private static String trustStoreFile = "truststore";
|
|
private static String passwd = "passphrase";
|
|
|
|
private static String keyFilename =
|
|
System.getProperty("test.src", "./") + "/" + pathToStores +
|
|
"/" + keyStoreFile;
|
|
private static String trustFilename =
|
|
System.getProperty("test.src", "./") + "/" + pathToStores +
|
|
"/" + trustStoreFile;
|
|
|
|
private static void tryNull(SSLEngine ssle, ByteBuffer appData,
|
|
ByteBuffer netData) throws Exception {
|
|
try {
|
|
ssle.wrap(appData, netData);
|
|
throw new Exception();
|
|
} catch (IllegalArgumentException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
|
|
try {
|
|
ssle.unwrap(netData, appData);
|
|
throw new Exception();
|
|
} catch (IllegalArgumentException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
}
|
|
|
|
private static void tryNullArray(SSLEngine ssle, ByteBuffer [] appData,
|
|
ByteBuffer netData) throws Exception {
|
|
try {
|
|
ssle.wrap(appData, netData);
|
|
throw new Exception();
|
|
} catch (IllegalArgumentException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
|
|
try {
|
|
ssle.unwrap(netData, appData);
|
|
throw new Exception();
|
|
} catch (IllegalArgumentException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
}
|
|
|
|
private static void tryNullArrayLen(SSLEngine ssle, ByteBuffer [] appData,
|
|
int offset, int len, ByteBuffer netData) throws Exception {
|
|
try {
|
|
ssle.wrap(appData, offset, len, netData);
|
|
throw new Exception();
|
|
} catch (IllegalArgumentException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
|
|
try {
|
|
ssle.unwrap(netData, appData, offset, len);
|
|
throw new Exception();
|
|
} catch (IllegalArgumentException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
}
|
|
|
|
private static void tryReadOnly(SSLEngine ssle, ByteBuffer [] appData,
|
|
int offset, int len, ByteBuffer netData) throws Exception {
|
|
try {
|
|
if (netData.isReadOnly()) {
|
|
ssle.wrap(appData, offset, len, netData);
|
|
throw new Exception();
|
|
}
|
|
} catch (ReadOnlyBufferException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
|
|
try {
|
|
if (!netData.isReadOnly()) {
|
|
ssle.unwrap(netData, appData, offset, len);
|
|
throw new Exception();
|
|
}
|
|
} catch (ReadOnlyBufferException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
}
|
|
|
|
private static void tryOutOfBounds(SSLEngine ssle, ByteBuffer [] appData,
|
|
int offset, int len, ByteBuffer netData) throws Exception {
|
|
try {
|
|
ssle.wrap(appData, offset, len, netData);
|
|
throw new Exception();
|
|
} catch (IndexOutOfBoundsException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
|
|
try {
|
|
ssle.unwrap(netData, appData, offset, len);
|
|
throw new Exception();
|
|
} catch (IndexOutOfBoundsException e) {
|
|
System.out.println("Caught right exception");
|
|
}
|
|
}
|
|
|
|
private static void trySmallBufs(SSLEngine ssle,
|
|
ByteBuffer appBB, ByteBuffer smallNetBB,
|
|
ByteBuffer smallAppBB, ByteBuffer netBB) throws Exception {
|
|
|
|
SSLEngineResult res = ssle.wrap(appBB, smallNetBB);
|
|
if (res.getStatus() != Status.BUFFER_OVERFLOW) {
|
|
throw new Exception();
|
|
}
|
|
|
|
// For unwrap(), the BUFFER_OVERFLOW will not be generated
|
|
// until received SSL/TLS application data.
|
|
// Test test/jdk/javax/net/ssl/SSLEngine/LargePacket.java will check
|
|
// BUFFER_OVERFLOW/UNDERFLOW for both wrap() and unwrap().
|
|
//
|
|
//res = ssle.unwrap(netBB, smallAppBB);
|
|
//if (res.getStatus() != Status.BUFFER_OVERFLOW) {
|
|
// throw new Exception();
|
|
//}
|
|
}
|
|
|
|
private static void trySmallBufsArray(SSLEngine ssle,
|
|
ByteBuffer [] appBB, ByteBuffer smallNetBB,
|
|
ByteBuffer [] smallAppBB, ByteBuffer netBB) throws Exception {
|
|
|
|
SSLEngineResult res = ssle.wrap(appBB, 0, appBB.length, smallNetBB);
|
|
if (res.getStatus() != Status.BUFFER_OVERFLOW) {
|
|
throw new Exception();
|
|
}
|
|
|
|
// For unwrap(), the BUFFER_OVERFLOW will not be generated
|
|
// until received SSL/TLS application data.
|
|
// Test test/jdk/javax/net/ssl/SSLEngine/LargePacket.java will check
|
|
// BUFFER_OVERFLOW/UNDERFLOW for both wrap() and unwrap().
|
|
//
|
|
//res = ssle.unwrap(netBB, smallAppBB, 0, appBB.length);
|
|
//if (res.getStatus() != Status.BUFFER_OVERFLOW) {
|
|
// throw new Exception();
|
|
//}
|
|
}
|
|
|
|
private static void runTest(SSLEngine ssle) throws Exception {
|
|
ByteBuffer [] bufs;
|
|
|
|
ByteBuffer roBB = ByteBuffer.allocate(40).asReadOnlyBuffer();
|
|
|
|
ByteBuffer bb1K = ByteBuffer.allocate(1024);
|
|
ByteBuffer bb2K = ByteBuffer.allocate(2048);
|
|
ByteBuffer bb4K = ByteBuffer.allocate(5096);
|
|
ByteBuffer bb8K = ByteBuffer.allocate(10192);
|
|
|
|
SSLSession ssls = ssle.getSession();
|
|
|
|
ByteBuffer netBBMinus1 = ByteBuffer.allocate(
|
|
ssls.getPacketBufferSize() - 1);
|
|
ByteBuffer appBBMinus1 = ByteBuffer.allocate(
|
|
ssls.getApplicationBufferSize() - 1);
|
|
|
|
ByteBuffer bbNet = ByteBuffer.allocate(
|
|
ssls.getPacketBufferSize());
|
|
ByteBuffer bbApp = ByteBuffer.allocate(
|
|
ssls.getApplicationBufferSize());
|
|
|
|
bufs = new ByteBuffer [] { bb1K, bb2K, bb4K, bb8K};
|
|
|
|
tryNull(ssle, null, null);
|
|
tryNull(ssle, bb1K, null);
|
|
tryNull(ssle, null, bb1K);
|
|
|
|
tryNullArray(ssle, null, null);
|
|
tryNullArray(ssle, bufs, null);
|
|
tryNullArray(ssle, null, bb1K);
|
|
|
|
tryNullArrayLen(ssle, null, 0, 4, null);
|
|
tryNullArrayLen(ssle, bufs, 0, 4, null);
|
|
tryNullArrayLen(ssle, null, 0, 4, bb1K);
|
|
|
|
bufs[2] = null;
|
|
tryNullArray(ssle, bufs, bb1K);
|
|
|
|
bufs[2] = bb4K;
|
|
tryReadOnly(ssle, bufs, 0, 4, roBB);
|
|
|
|
bufs[2] = roBB;
|
|
tryReadOnly(ssle, bufs, 0, 4, bb4K);
|
|
|
|
bufs[2] = bb4K;
|
|
tryOutOfBounds(ssle, bufs, -1, 0, bb1K);
|
|
tryOutOfBounds(ssle, bufs, 0, -1, bb1K);
|
|
tryOutOfBounds(ssle, bufs, 0, bufs.length + 1, bb1K);
|
|
tryOutOfBounds(ssle, bufs, bufs.length, 1, bb1K);
|
|
|
|
bufs[3].position(bufs[3].limit());
|
|
trySmallBufs(ssle, bb1K, netBBMinus1, appBBMinus1, bb1K);
|
|
trySmallBufsArray(ssle, bufs, netBBMinus1, bufs, bb1K);
|
|
bufs[3].rewind();
|
|
|
|
}
|
|
|
|
public static void main(String args[]) throws Exception {
|
|
|
|
SSLEngine ssle = createSSLEngine(keyFilename, trustFilename);
|
|
runTest(ssle);
|
|
|
|
System.out.println("Test Passed.");
|
|
}
|
|
|
|
/*
|
|
* Create an initialized SSLContext to use for this test.
|
|
*/
|
|
static private SSLEngine createSSLEngine(String keyFile, String trustFile)
|
|
throws Exception {
|
|
|
|
SSLEngine ssle;
|
|
|
|
KeyStore ks = KeyStore.getInstance("JKS");
|
|
KeyStore ts = KeyStore.getInstance("JKS");
|
|
|
|
char[] passphrase = "passphrase".toCharArray();
|
|
|
|
ks.load(new FileInputStream(keyFile), passphrase);
|
|
ts.load(new FileInputStream(trustFile), passphrase);
|
|
|
|
KeyManagerFactory kmf = KeyManagerFactory.getInstance("SunX509");
|
|
kmf.init(ks, passphrase);
|
|
|
|
TrustManagerFactory tmf = TrustManagerFactory.getInstance("SunX509");
|
|
tmf.init(ts);
|
|
|
|
SSLContext sslCtx = SSLContext.getInstance("TLS");
|
|
|
|
sslCtx.init(kmf.getKeyManagers(), tmf.getTrustManagers(), null);
|
|
|
|
ssle = sslCtx.createSSLEngine("client", 1001);
|
|
ssle.setUseClientMode(true);
|
|
|
|
return ssle;
|
|
}
|
|
}
|