--- /dev/null 2018-06-30 13:49:24.634979840 -0700 +++ new/test/jdk/jdk/net/Sockets/rsocket/SocketChannel/BasicSocketChannelTest.java 2018-06-30 14:07:47.684165944 -0700 @@ -0,0 +1,168 @@ +/* + * 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 8195160 + * @summary Test for basic functionality for RdmaSocketChannel + * and RdmaServerSocketChannel + * @requires (os.family == "linux") + * @library .. /test/lib + * @build RsocketTest + * @run main/othervm -Djava.net.preferIPv4Stack=true BasicSocketChannelTest + */ + +import java.io.InputStream; +import java.io.OutputStream; +import java.net.InetAddress; +import java.net.InetSocketAddress; +import java.net.ServerSocket; +import java.net.Socket; +import java.nio.channels.ServerSocketChannel; +import java.nio.channels.SocketChannel; +import java.nio.ByteBuffer; +import java.nio.CharBuffer; +import jdk.net.Sockets; + +public class BasicSocketChannelTest implements Runnable { + static ServerSocketChannel ssc; + static SocketChannel sc1, sc2; + static InetAddress iaddr; + static int port = 0; + static String message = "This is a message!"; + static int length = -1; + + public static void main(String args[]) throws Exception { + if (!RsocketTest.isRsocketAvailable()) + return; + + iaddr = InetAddress.getLocalHost(); + length = message.length(); + String result; + + //test SocketChannel and ServerSocketChannel + try { + ssc = Sockets.openRdmaServerSocketChannel(); + sc1 = Sockets.openRdmaSocketChannel(); + + ssc.bind(new InetSocketAddress(iaddr, port)); + + Thread t = new Thread(new BasicSocketChannelTest(), "Channel"); + t.start(); + + sc2 = ssc.accept(); + ByteBuffer output = ByteBuffer.allocate(length); + int outputNum = 0; + while (outputNum < length) { + outputNum += sc2.read(output); + } + + result = new String(output.array()); + if(!result.equals(message)) + throw new RuntimeException("Test Failed!"); + sc2.shutdownInput(); + sc2.shutdownOutput(); + } catch (Exception e) { + e.printStackTrace(); + throw new RuntimeException("Test Failed!"); + } finally { + ssc.close(); + sc1.close(); + sc2.close(); + } + + //test SocketChannel.socket() and ServerSocketChannel.socket() + try { + ssc = Sockets.openRdmaServerSocketChannel(); + sc1 = Sockets.openRdmaSocketChannel(); + + ssc.socket().bind(new InetSocketAddress(iaddr, port)); + + Thread t = new Thread(new BasicSocketChannelTest(), "Socket"); + t.start(); + + sc2 = ssc.accept(); + Socket conn = sc2.socket(); + InputStream is = conn.getInputStream(); + + int num = 0; + byte[] buf = new byte[length]; + while (num < length) { + int l = is.read(buf); + num += l; + } + + result = new String(buf); + if(!result.equals(message)) + throw new RuntimeException("Test Failed!"); + conn.shutdownInput(); + conn.shutdownOutput(); + if (!conn.isInputShutdown() || !conn.isOutputShutdown()) + throw new RuntimeException("Test Failed!"); + } catch (Exception e) { + e.printStackTrace(); + throw new RuntimeException("Test Failed!"); + } finally { + ssc.close(); + sc1.close(); + sc2.close(); + } + } + + public void run() { + int port = ssc.socket().getLocalPort(); + byte[] arr = new byte[length]; + arr = message.getBytes(); + + try { + if (Thread.currentThread().getName().startsWith("Channel")) { + sc1.connect(new InetSocketAddress(iaddr, port)); + if (!sc1.isConnected()) + throw new RuntimeException("Test Failed!"); + ByteBuffer input = ByteBuffer.allocate(length); + input.put(arr); + input.flip(); + int inputNum = 0; + while (inputNum < length) { + inputNum += sc1.write(input); + } + sc1.shutdownInput(); + sc1.shutdownOutput(); + } else if (Thread.currentThread().getName().startsWith("Socket")) { + Socket client = sc1.socket(); + client.connect(new InetSocketAddress(iaddr, port)); + if (!client.isConnected()) + throw new RuntimeException("Test Failed!"); + OutputStream os = client.getOutputStream(); + os.write(arr); + client.shutdownInput(); + client.shutdownOutput(); + if (!client.isInputShutdown() || !client.isOutputShutdown()) + throw new RuntimeException("Test Failed!"); + } + } catch (Exception e) { + e.printStackTrace(); + throw new RuntimeException("Test Failed!"); + } + } +}