test/sun/security/ssl/javax/net/ssl/NewAPIs/SSLEngine/SSLEngineService.java - toolchain/jdk/jdk9_jdk - Git at Google

 /*
  * Copyright (c) 2006, 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.
  */

 /*
  *
  *
  * @bug 6388456
  * @summary Need adjustable TLS max record size for interoperability
  *      with non-compliant stacks
  *
  * Helper class of SSL/TLS client/server communication.
  *
  * @author Xuelei Fan
  */

 import javax.net.ssl.*;

 import java.io.*;
 import java.security.*;
 import java.nio.*;
 import java.nio.channels.*;

 public class SSLEngineService {

     private static String pathToStores = "../../../../../etc";
     private static String keyStoreFile = "keystore";
     private static String trustStoreFile = "truststore";
     private static char[] passphrase = "passphrase".toCharArray();

     private static String keyFilename =
             System.getProperty("test.src", "./") + "/" + pathToStores +
                 "/" + keyStoreFile;
     private static String trustFilename =
             System.getProperty("test.src", "./") + "/" + pathToStores +
                 "/" + trustStoreFile;

     // deliver local application data.
     protected static void deliver(SSLEngine ssle, SocketChannel sc)
         throws Exception {

         // create buufer.
         int appBufferMax = ssle.getSession().getApplicationBufferSize();
         int netBufferMax = ssle.getSession().getPacketBufferSize();
         int length = appBufferMax * (Integer.SIZE / 8);

         // allocate more in order to check large packet
         ByteBuffer localAppData = ByteBuffer.allocate(length);

         // allocate less in order to check BUFFER_OVERFLOW/BUFFER_UNDERFLOW
         ByteBuffer localNetData = ByteBuffer.allocate(netBufferMax/2);

         // prepare local application data
         localAppData.putInt(length);
         for (int i = 1; i < appBufferMax; i++) {
             localAppData.putInt(i);
         }
         localAppData.flip();


         while (localAppData.hasRemaining()) {
             // empty the local network packet buffer.
             localNetData.clear();

             // generated local network packet.
             SSLEngineResult res = ssle.wrap(localAppData, localNetData);

             // checking status
             switch (res.getStatus()) {

             case OK :
                 localNetData.flip();

                 // send the network packet
                 while (localNetData.hasRemaining()) {
                     if (sc.write(localNetData) < 0) {
                         throw new IOException("Unable write to socket channel");
                     }
                 }

                 if (res.getHandshakeStatus() ==
                         SSLEngineResult.HandshakeStatus.NEED_TASK) {
                     Runnable runnable;
                     while ((runnable = ssle.getDelegatedTask()) != null) {
                         runnable.run();
                     }
                 }

                 // detect large buffer
                 if (res.bytesProduced() >= Short.MAX_VALUE) {
                     System.out.println("Generate a " +
                         res.bytesProduced() + " bytes large packet ");
                 }
                 break;

             case BUFFER_OVERFLOW :
                 // maybe need to enlarge the local network packet buffer.
                 int size = ssle.getSession().getPacketBufferSize();
                 if (size > localNetData.capacity()) {
                     System.out.println("resize destination buffer upto " +
                                 size + " bytes for BUFFER_OVERFLOW");
                     localNetData = enlargeBuffer(localNetData, size);
                 }
                 break;

             default : // BUFFER_UNDERFLOW or CLOSED :
                 throw new IOException("Received invalid" + res.getStatus() +
                         "during transfer application data");
             }
         }
     }


     // receive peer application data.
     protected static void receive(SSLEngine ssle, SocketChannel sc)
         throws Exception {

         // create buufers.
         int appBufferMax = ssle.getSession().getApplicationBufferSize();
         int netBufferMax = ssle.getSession().getPacketBufferSize();

         // allocate less in order to check BUFFER_OVERFLOW/BUFFER_UNDERFLOW
         ByteBuffer peerAppData = ByteBuffer.allocate(appBufferMax/2);
         ByteBuffer peerNetData = ByteBuffer.allocate(netBufferMax/2);
         int received = -1;

         while (received != 0) {
             if (ssle.isInboundDone() || sc.read(peerNetData) < 0) {
                 break;
             }

             peerNetData.flip();
             SSLEngineResult res = ssle.unwrap(peerNetData, peerAppData);
             peerNetData.compact();

             // checking status
             switch (res.getStatus()) {

             case OK :
                 if (res.getHandshakeStatus() ==
                         SSLEngineResult.HandshakeStatus.NEED_TASK) {
                     Runnable runnable;
                     while ((runnable = ssle.getDelegatedTask()) != null) {
                         runnable.run();
                     }
                 }

                 if (received < 0 && res.bytesProduced() < 4 ) {
                     break;
                 }

                 if (received < 0) {
                     received = peerAppData.getInt(0);
                 }

                 System.out.println("received " + peerAppData.position() +
                         " bytes client application data");
                 System.out.println("\tcomsumed " + res.bytesConsumed() +
                         " byes network data");
                 peerAppData.clear();

                 received -= res.bytesProduced();

                 // detect large buffer
                 if (res.bytesConsumed() >= Short.MAX_VALUE) {
                     System.out.println("Consumes a " + res.bytesConsumed() +
                         " bytes large packet ");
                 }

                 break;

             case BUFFER_OVERFLOW :
                 // maybe need to enlarge the peer application data buffer.
                 int size = ssle.getSession().getApplicationBufferSize();
                 if (size > peerAppData.capacity()) {
                     System.out.println("resize destination buffer upto " +
                         size + " bytes for BUFFER_OVERFLOW");
                     peerAppData = enlargeBuffer(peerAppData, size);
                 }
                 break;

             case BUFFER_UNDERFLOW :
                 // maybe need to enlarge the peer network packet data buffer.
                 size = ssle.getSession().getPacketBufferSize();
                 if (size > peerNetData.capacity()) {
                     System.out.println("resize source buffer upto " + size +
                         " bytes for BUFFER_UNDERFLOW");
                     peerNetData = enlargeBuffer(peerNetData, size);
                 }
                 break;

             default : // CLOSED :
                 throw new IOException("Received invalid" + res.getStatus() +
                         "during transfer application data");
             }
         }
     }

     protected static void handshaking(SSLEngine ssle, SocketChannel sc)
         throws Exception {

         int appBufferMax = ssle.getSession().getApplicationBufferSize();
         int netBufferMax = ssle.getSession().getPacketBufferSize();

         // allocate less in order to check BUFFER_OVERFLOW/BUFFER_UNDERFLOW
         ByteBuffer localAppData = ByteBuffer.allocate(appBufferMax/10);
         ByteBuffer peerAppData = ByteBuffer.allocate(appBufferMax/10);
         ByteBuffer localNetData = ByteBuffer.allocate(netBufferMax/10);
         ByteBuffer peerNetData = ByteBuffer.allocate(netBufferMax/10);

         // begin handshake
         ssle.beginHandshake();
         SSLEngineResult.HandshakeStatus hs = ssle.getHandshakeStatus();

         // start handshaking from unwrap
         do {
             switch (hs) {

             case NEED_UNWRAP :
                 if (peerNetData.position() == 0) {
                     if (sc.read(peerNetData) < 0) {
                         ssle.closeInbound();
                         return;
                     }
                 }

                 peerNetData.flip();
                 SSLEngineResult res = ssle.unwrap(peerNetData, peerAppData);
                 peerNetData.compact();
                 hs = res.getHandshakeStatus();

                 switch (res.getStatus()) {
                 case OK :
                     break;
                 case BUFFER_UNDERFLOW :
                     // maybe need to enlarge the peer network packet buffer.
                     int size = ssle.getSession().getPacketBufferSize();
                     if (size > peerNetData.capacity()) {
                         System.out.println("resize source buffer upto " +
                                 size + " bytes for BUFFER_UNDERFLOW");
                         peerNetData = enlargeBuffer(peerNetData, size);
                     }
                     break;
                 case BUFFER_OVERFLOW :
                     // maybe need to enlarge the peer application data buffer.
                     size = ssle.getSession().getApplicationBufferSize();
                     if (size > peerAppData.capacity()) {
                         System.out.println("resize destination buffer upto " +
                                 size + " bytes for BUFFER_OVERFLOW");
                         peerAppData = enlargeBuffer(peerAppData, size);
                     }
                     break;
                 default : //CLOSED
                     throw new IOException("Received invalid" + res.getStatus() +
                         "during initial handshaking");
                 }
                 break;

             case NEED_WRAP :
                 // empty the local network packet buffer.
                 localNetData.clear();

                 // generated local network packet.
                 res = ssle.wrap(localAppData, localNetData);
                 hs = res.getHandshakeStatus();

                 // checking status
                 switch (res.getStatus()) {
                 case OK :
                     localNetData.flip();

                     // send the network packet
                     while (localNetData.hasRemaining()) {
                         if (sc.write(localNetData) < 0) {
                             throw new IOException(
                                 "Unable write to socket channel");
                         }
                     }
                     break;

                 case BUFFER_OVERFLOW :
                     // maybe need to enlarge the local network packet buffer.
                     int size = ssle.getSession().getPacketBufferSize();
                     if (size > localNetData.capacity()) {
                         System.out.println("resize destination buffer upto " +
                                 size + " bytes for BUFFER_OVERFLOW");
                         localNetData = enlargeBuffer(localNetData, size);
                     }
                     break;

                 default : // BUFFER_UNDERFLOW or CLOSED :
                     throw new IOException("Received invalid" + res.getStatus() +
                         "during initial handshaking");
                 }
                 break;

             case NEED_TASK :
                 Runnable runnable;
                 while ((runnable = ssle.getDelegatedTask()) != null) {
                     runnable.run();
                 }
                 hs = ssle.getHandshakeStatus();
                 break;

             default : // FINISHED or NOT_HANDSHAKING
                 // do nothing
             }
         } while (hs != SSLEngineResult.HandshakeStatus.FINISHED &&
                 hs != SSLEngineResult.HandshakeStatus.NOT_HANDSHAKING);
     }

     private static ByteBuffer enlargeBuffer(ByteBuffer buffer, int size) {
         ByteBuffer bb = ByteBuffer.allocate(size);
         buffer.flip();
         bb.put(buffer);

         return bb;
     }

     /*
      * Create an initialized SSLContext to use for this test.
      */
     protected static SSLEngine createSSLEngine(boolean mode) throws Exception {

         SSLEngine ssle;

         KeyStore ks = KeyStore.getInstance("JKS");
         KeyStore ts = KeyStore.getInstance("JKS");

         ks.load(new FileInputStream(keyFilename), passphrase);
         ts.load(new FileInputStream(trustFilename), 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();
         ssle.setUseClientMode(mode);

         return ssle;
     }
 }
	/*
	* Copyright (c) 2006, 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.
	*/

	/*
	*
	*
	* @bug 6388456
	* @summary Need adjustable TLS max record size for interoperability
	* with non-compliant stacks
	*
	* Helper class of SSL/TLS client/server communication.
	*
	* @author Xuelei Fan
	*/

	import javax.net.ssl.*;

	import java.io.*;
	import java.security.*;
	import java.nio.*;
	import java.nio.channels.*;

	public class SSLEngineService {

	private static String pathToStores = "../../../../../etc";
	private static String keyStoreFile = "keystore";
	private static String trustStoreFile = "truststore";
	private static char[] passphrase = "passphrase".toCharArray();

	private static String keyFilename =
	System.getProperty("test.src", "./") + "/" + pathToStores +
	"/" + keyStoreFile;
	private static String trustFilename =
	System.getProperty("test.src", "./") + "/" + pathToStores +
	"/" + trustStoreFile;

	// deliver local application data.
	protected static void deliver(SSLEngine ssle, SocketChannel sc)
	throws Exception {

	// create buufer.
	int appBufferMax = ssle.getSession().getApplicationBufferSize();
	int netBufferMax = ssle.getSession().getPacketBufferSize();
	int length = appBufferMax * (Integer.SIZE / 8);

	// allocate more in order to check large packet
	ByteBuffer localAppData = ByteBuffer.allocate(length);

	// allocate less in order to check BUFFER_OVERFLOW/BUFFER_UNDERFLOW
	ByteBuffer localNetData = ByteBuffer.allocate(netBufferMax/2);

	// prepare local application data
	localAppData.putInt(length);
	for (int i = 1; i < appBufferMax; i++) {
	localAppData.putInt(i);
	}
	localAppData.flip();


	while (localAppData.hasRemaining()) {
	// empty the local network packet buffer.
	localNetData.clear();

	// generated local network packet.
	SSLEngineResult res = ssle.wrap(localAppData, localNetData);

	// checking status
	switch (res.getStatus()) {

	case OK :
	localNetData.flip();

	// send the network packet
	while (localNetData.hasRemaining()) {
	if (sc.write(localNetData) < 0) {
	throw new IOException("Unable write to socket channel");
	}
	}

	if (res.getHandshakeStatus() ==
	SSLEngineResult.HandshakeStatus.NEED_TASK) {
	Runnable runnable;
	while ((runnable = ssle.getDelegatedTask()) != null) {
	runnable.run();
	}
	}

	// detect large buffer
	if (res.bytesProduced() >= Short.MAX_VALUE) {
	System.out.println("Generate a " +
	res.bytesProduced() + " bytes large packet ");
	}
	break;

	case BUFFER_OVERFLOW :
	// maybe need to enlarge the local network packet buffer.
	int size = ssle.getSession().getPacketBufferSize();
	if (size > localNetData.capacity()) {
	System.out.println("resize destination buffer upto " +
	size + " bytes for BUFFER_OVERFLOW");
	localNetData = enlargeBuffer(localNetData, size);
	}
	break;

	default : // BUFFER_UNDERFLOW or CLOSED :
	throw new IOException("Received invalid" + res.getStatus() +
	"during transfer application data");
	}
	}
	}


	// receive peer application data.
	protected static void receive(SSLEngine ssle, SocketChannel sc)
	throws Exception {

	// create buufers.
	int appBufferMax = ssle.getSession().getApplicationBufferSize();
	int netBufferMax = ssle.getSession().getPacketBufferSize();

	// allocate less in order to check BUFFER_OVERFLOW/BUFFER_UNDERFLOW
	ByteBuffer peerAppData = ByteBuffer.allocate(appBufferMax/2);
	ByteBuffer peerNetData = ByteBuffer.allocate(netBufferMax/2);
	int received = -1;

	while (received != 0) {
	if (ssle.isInboundDone() \|\| sc.read(peerNetData) < 0) {
	break;
	}

	peerNetData.flip();
	SSLEngineResult res = ssle.unwrap(peerNetData, peerAppData);
	peerNetData.compact();

	// checking status
	switch (res.getStatus()) {

	case OK :
	if (res.getHandshakeStatus() ==
	SSLEngineResult.HandshakeStatus.NEED_TASK) {
	Runnable runnable;
	while ((runnable = ssle.getDelegatedTask()) != null) {
	runnable.run();
	}
	}

	if (received < 0 && res.bytesProduced() < 4 ) {
	break;
	}

	if (received < 0) {
	received = peerAppData.getInt(0);
	}

	System.out.println("received " + peerAppData.position() +
	" bytes client application data");
	System.out.println("\tcomsumed " + res.bytesConsumed() +
	" byes network data");
	peerAppData.clear();

	received -= res.bytesProduced();

	// detect large buffer
	if (res.bytesConsumed() >= Short.MAX_VALUE) {
	System.out.println("Consumes a " + res.bytesConsumed() +
	" bytes large packet ");
	}

	break;

	case BUFFER_OVERFLOW :
	// maybe need to enlarge the peer application data buffer.
	int size = ssle.getSession().getApplicationBufferSize();
	if (size > peerAppData.capacity()) {
	System.out.println("resize destination buffer upto " +
	size + " bytes for BUFFER_OVERFLOW");
	peerAppData = enlargeBuffer(peerAppData, size);
	}
	break;

	case BUFFER_UNDERFLOW :
	// maybe need to enlarge the peer network packet data buffer.
	size = ssle.getSession().getPacketBufferSize();
	if (size > peerNetData.capacity()) {
	System.out.println("resize source buffer upto " + size +
	" bytes for BUFFER_UNDERFLOW");
	peerNetData = enlargeBuffer(peerNetData, size);
	}
	break;

	default : // CLOSED :
	throw new IOException("Received invalid" + res.getStatus() +
	"during transfer application data");
	}
	}
	}

	protected static void handshaking(SSLEngine ssle, SocketChannel sc)
	throws Exception {

	int appBufferMax = ssle.getSession().getApplicationBufferSize();
	int netBufferMax = ssle.getSession().getPacketBufferSize();

	// allocate less in order to check BUFFER_OVERFLOW/BUFFER_UNDERFLOW
	ByteBuffer localAppData = ByteBuffer.allocate(appBufferMax/10);
	ByteBuffer peerAppData = ByteBuffer.allocate(appBufferMax/10);
	ByteBuffer localNetData = ByteBuffer.allocate(netBufferMax/10);
	ByteBuffer peerNetData = ByteBuffer.allocate(netBufferMax/10);

	// begin handshake
	ssle.beginHandshake();
	SSLEngineResult.HandshakeStatus hs = ssle.getHandshakeStatus();

	// start handshaking from unwrap
	do {
	switch (hs) {

	case NEED_UNWRAP :
	if (peerNetData.position() == 0) {
	if (sc.read(peerNetData) < 0) {
	ssle.closeInbound();
	return;
	}
	}

	peerNetData.flip();
	SSLEngineResult res = ssle.unwrap(peerNetData, peerAppData);
	peerNetData.compact();
	hs = res.getHandshakeStatus();

	switch (res.getStatus()) {
	case OK :
	break;
	case BUFFER_UNDERFLOW :
	// maybe need to enlarge the peer network packet buffer.
	int size = ssle.getSession().getPacketBufferSize();
	if (size > peerNetData.capacity()) {
	System.out.println("resize source buffer upto " +
	size + " bytes for BUFFER_UNDERFLOW");
	peerNetData = enlargeBuffer(peerNetData, size);
	}
	break;
	case BUFFER_OVERFLOW :
	// maybe need to enlarge the peer application data buffer.
	size = ssle.getSession().getApplicationBufferSize();
	if (size > peerAppData.capacity()) {
	System.out.println("resize destination buffer upto " +
	size + " bytes for BUFFER_OVERFLOW");
	peerAppData = enlargeBuffer(peerAppData, size);
	}
	break;
	default : //CLOSED
	throw new IOException("Received invalid" + res.getStatus() +
	"during initial handshaking");
	}
	break;

	case NEED_WRAP :
	// empty the local network packet buffer.
	localNetData.clear();

	// generated local network packet.
	res = ssle.wrap(localAppData, localNetData);
	hs = res.getHandshakeStatus();

	// checking status
	switch (res.getStatus()) {
	case OK :
	localNetData.flip();

	// send the network packet
	while (localNetData.hasRemaining()) {
	if (sc.write(localNetData) < 0) {
	throw new IOException(
	"Unable write to socket channel");
	}
	}
	break;

	case BUFFER_OVERFLOW :
	// maybe need to enlarge the local network packet buffer.
	int size = ssle.getSession().getPacketBufferSize();
	if (size > localNetData.capacity()) {
	System.out.println("resize destination buffer upto " +
	size + " bytes for BUFFER_OVERFLOW");
	localNetData = enlargeBuffer(localNetData, size);
	}
	break;

	default : // BUFFER_UNDERFLOW or CLOSED :
	throw new IOException("Received invalid" + res.getStatus() +
	"during initial handshaking");
	}
	break;

	case NEED_TASK :
	Runnable runnable;
	while ((runnable = ssle.getDelegatedTask()) != null) {
	runnable.run();
	}
	hs = ssle.getHandshakeStatus();
	break;

	default : // FINISHED or NOT_HANDSHAKING
	// do nothing
	}
	} while (hs != SSLEngineResult.HandshakeStatus.FINISHED &&
	hs != SSLEngineResult.HandshakeStatus.NOT_HANDSHAKING);
	}

	private static ByteBuffer enlargeBuffer(ByteBuffer buffer, int size) {
	ByteBuffer bb = ByteBuffer.allocate(size);
	buffer.flip();
	bb.put(buffer);

	return bb;
	}

	/*
	* Create an initialized SSLContext to use for this test.
	*/
	protected static SSLEngine createSSLEngine(boolean mode) throws Exception {

	SSLEngine ssle;

	KeyStore ks = KeyStore.getInstance("JKS");
	KeyStore ts = KeyStore.getInstance("JKS");

	ks.load(new FileInputStream(keyFilename), passphrase);
	ts.load(new FileInputStream(trustFilename), 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();
	ssle.setUseClientMode(mode);

	return ssle;
	}
	}