--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/sun/security/ssl/SSLSocketImpl/SSLSocketKeyLimit.java Fri May 11 15:53:12 2018 -0700
@@ -0,0 +1,298 @@
+/*
+ * Copyright (c) 2018, 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 8164879
+ * @library /lib/testlibrary ../../
+ * @modules java.base/sun.security.util
+ * @summary Verify AES/GCM's limits set in the jdk.tls.keyLimits property
+ * @run main SSLSocketKeyLimit 0 server AES/GCM/NoPadding keyupdate 1000000
+ * @run main SSLSocketKeyLimit 0 client AES/GCM/NoPadding keyupdate 1000000
+ * @run main SSLSocketKeyLimit 1 client AES/GCM/NoPadding keyupdate 2^22
+ */
+
+ /**
+ * Verify AES/GCM's limits set in the jdk.tls.keyLimits property
+ * start a new handshake sequence to renegotiate the symmetric key with an
+ * SSLSocket connection. This test verifies the handshake method was called
+ * via debugging info. It does not verify the renegotiation was successful
+ * as that is very hard.
+ */
+
+import javax.net.ssl.KeyManagerFactory;
+import javax.net.ssl.SSLContext;
+import javax.net.ssl.SSLServerSocket;
+import javax.net.ssl.SSLServerSocketFactory;
+import javax.net.ssl.SSLSocket;
+import javax.net.ssl.SSLSocketFactory;
+import javax.net.ssl.TrustManagerFactory;
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.PrintWriter;
+import java.security.KeyStore;
+import java.security.SecureRandom;
+import java.util.Arrays;
+
+import jdk.testlibrary.ProcessTools;
+import jdk.testlibrary.Utils;
+import jdk.testlibrary.OutputAnalyzer;
+import sun.security.util.HexDumpEncoder;
+
+public class SSLSocketKeyLimit {
+
+ SSLSocket svr, c;
+ SSLServerSocketFactory ssf;
+ SSLServerSocket ss;
+ SSLSocketFactory sf;
+ InputStream in;
+ OutputStream out;
+
+ static boolean serverReady = false;
+ static int serverPort = 12345;
+
+ static String pathToStores = "../../../../javax/net/ssl/etc/";
+ static String keyStoreFile = "keystore";
+ static String passwd = "passphrase";
+ static int dataLen = 10240;
+ static byte[] data = new byte[dataLen];
+ static boolean serverwrite = true;
+ int totalDataLen = 0;
+ static boolean done = false;
+
+ SSLSocketKeyLimit() {
+ in = new ByteArrayInputStream(new byte[dataLen]);
+ out = new ByteArrayOutputStream();
+ }
+
+ SSLContext initContext() throws Exception {
+ SSLContext sc = SSLContext.getInstance("TLSv1.3");
+ KeyStore ks = KeyStore.getInstance(
+ new File(System.getProperty("javax.net.ssl.keyStore")),
+ passwd.toCharArray());
+ KeyManagerFactory kmf =
+ KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
+ kmf.init(ks, passwd.toCharArray());
+ TrustManagerFactory tmf = TrustManagerFactory.getInstance(TrustManagerFactory.getDefaultAlgorithm());
+ tmf.init(ks);
+ sc.init(kmf.getKeyManagers(), tmf.getTrustManagers(), new SecureRandom());
+ return sc;
+ }
+
+ /**
+ * args should have two values: server|client, <limit size>
+ * Prepending 'p' is for internal use only.
+ */
+ public static void main(String args[]) throws Exception {
+ if (args[0].compareTo("p") != 0) {
+
+ boolean expectedFail = (Integer.parseInt(args[0]) == 1);
+ if (expectedFail) {
+ System.out.println("Test expected to not find updated msg");
+ }
+ //Write security property file to overwrite default
+ File f = new File("keyusage."+ System.nanoTime());
+ PrintWriter p = new PrintWriter(f);
+ p.write("jdk.tls.keyLimits=");
+ for (int i = 2; i < args.length; i++) {
+ p.write(" "+ args[i]);
+ }
+ p.close();
+ System.out.println("Keyusage path = " + f.getAbsolutePath());
+ System.setProperty("test.java.opts",
+ "-Dtest.src=" + System.getProperty("test.src") +
+ " -Dtest.jdk=" + System.getProperty("test.jdk") +
+ " -Djavax.net.debug=ssl " +
+ " -Djava.security.properties=" + f.getName());
+
+ System.out.println("test.java.opts: " +
+ System.getProperty("test.java.opts"));
+
+ ProcessBuilder pb = ProcessTools.createJavaProcessBuilder(true,
+ Utils.addTestJavaOpts("SSLSocketKeyLimit", "p", args[1]));
+
+ OutputAnalyzer output = ProcessTools.executeProcess(pb);
+ try {
+ if (expectedFail) {
+ output.shouldNotContain("KeyUpdate: write key updated");
+ output.shouldNotContain("KeyUpdate: read key updated");
+ } else {
+ output.shouldContain("KeyUpdate: write key updated");
+ output.shouldContain("KeyUpdate: read key updated");
+ }
+ } catch (Exception e) {
+ throw e;
+ } finally {
+ System.out.println("-- BEGIN Stdout:");
+ System.out.println(output.getStdout());
+ System.out.println("-- END Stdout");
+ System.out.println("-- BEGIN Stderr:");
+ System.out.println(output.getStderr());
+ System.out.println("-- END Stderr");
+ }
+ return;
+ }
+
+ if (args.length > 0 && args[0].compareToIgnoreCase("client") == 0) {
+ serverwrite = false;
+ }
+
+ String keyFilename =
+ System.getProperty("test.src", "./") + "/" + pathToStores +
+ "/" + keyStoreFile;
+
+ System.setProperty("javax.net.ssl.keyStore", keyFilename);
+ System.setProperty("javax.net.ssl.keyStorePassword", passwd);
+
+ Arrays.fill(data, (byte)0x0A);
+ Thread ts = new Thread(new Server());
+
+ ts.start();
+ while (!serverReady) {
+ Thread.sleep(100);
+ }
+ new Client().run();
+ ts.interrupt();
+ ts.join(10000); // 10sec
+ System.exit(0);
+ }
+
+ void write(SSLSocket s) throws Exception {
+ int i = 0;
+ in = s.getInputStream();
+ out = s.getOutputStream();
+ System.out.print("Write-side writing... ");
+ while (i++ < 150) {
+ out.write(data, 0, dataLen);
+ }
+ out.flush();
+ out.write(0x0D);
+ out.flush();
+
+ // Let read side all the data
+ while (!done) {
+ Thread.sleep(100);
+ }
+ }
+
+
+ void read(SSLSocket s) throws Exception {
+ byte[] buf = new byte[dataLen];
+ int len;
+ int i = 0;
+ try {
+ System.out.println("connected " + s.getSession().getCipherSuite());
+ in = s.getInputStream();
+ out = s.getOutputStream();
+ while (true) {
+ len = in.read(buf, 0, dataLen);
+ System.out.print(".");
+ for (byte b: buf) {
+ if (b == 0x0A || b == 0x0D) {
+ continue;
+ }
+ System.out.println("\nData invalid: " + new HexDumpEncoder().encode(buf));
+ break;
+ }
+
+ if (len > 0 && buf[len-1] == 0x0D) {
+ System.out.print("got end byte");
+ break;
+ }
+ out.write(i++);
+ totalDataLen += len;
+ }
+ } catch (Exception e) {
+ System.out.println("\n" + e.getMessage());
+ e.printStackTrace();
+ } finally {
+ // Tell write side that we are done reading
+ done = true;
+ }
+ System.out.println("\nTotalDataLen = " + totalDataLen);
+ }
+
+ static class Server extends SSLSocketKeyLimit implements Runnable {
+ Server() {
+ super();
+ try {
+ ssf = initContext().getServerSocketFactory();
+ ss = (SSLServerSocket) ssf.createServerSocket(serverPort);
+ serverPort = ss.getLocalPort();
+ } catch (Exception e) {
+ System.out.println("server: " + e.getMessage());
+ e.printStackTrace();
+ }
+ }
+
+ public void run() {
+
+ try {
+ serverReady = true;
+ System.out.println("Server waiting... ");
+ svr = (SSLSocket) ss.accept();
+ if (serverwrite) {
+ write(svr);
+ } else {
+ read(svr);
+ }
+
+ svr.close();
+ } catch (Exception e) {
+ System.out.println("server: " + e.getMessage());
+ e.printStackTrace();
+ }
+ System.out.println("Server closed");
+ }
+ }
+
+
+ static class Client extends SSLSocketKeyLimit implements Runnable {
+ Client() {
+ super();
+ }
+
+ public void run() {
+ try {
+ sf = initContext().getSocketFactory();
+ System.out.print("Client connecting... ");
+ c = (SSLSocket)sf.createSocket("localhost", serverPort);
+ System.out.println("connected. " + c.getSession().getCipherSuite());
+
+ // Opposite of what the server does
+ if (!serverwrite) {
+ write(c);
+ } else {
+ read(c);
+ }
+
+ } catch (Exception e) {
+ System.err.println("client: " + e.getMessage());
+ e.printStackTrace();
+ }
+ }
+ }
+}