--- /dev/null 2018-09-16 08:14:05.604314476 -0700 +++ new/test/jdk/jdk/net/Sockets/rsocket/SocketChannel/VectorIO.java 2018-10-05 14:15:11.802757368 -0700 @@ -0,0 +1,228 @@ +/* + * 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 socketchannel vector IO (use -Dseed=X to set PRNG seed) + * @requires (os.family == "linux") + * @library .. /test/lib /test/jdk/java/nio/channels + * @build jdk.test.lib.RandomFactory + * @build RsocketTest + * @run main/othervm -Djava.net.preferIPv4Stack=true VectorIO + * @key randomness + */ + +import java.io.IOException; +import java.net.InetAddress; +import java.net.InetSocketAddress; +import java.nio.ByteBuffer; +import java.nio.channels.ServerSocketChannel; +import java.nio.channels.SocketChannel; +import java.util.Random; +import jdk.test.lib.RandomFactory; +import jdk.net.Sockets; + +public class VectorIO { + + private static Random generator = RandomFactory.getRandom(); + + static int testSize; + + // whether to use the write/read variant with a length parameter + static boolean setLength; + + public static void main(String[] args) throws Exception { + if (!RsocketTest.isRsocketAvailable()) + return; + + testSize = 1; + setLength = false; + runTest(); + for(int i = 15; i < 18; i++) { + testSize = i; + setLength = !setLength; + runTest(); + } + } + + static void runTest() throws Exception { + System.err.println("Length " + testSize); + Server sv = new Server(testSize); + sv.start(); + bufferTest(sv.port()); + if (sv.finish(8000) == 0) + throw new Exception("Failed: Length = " + testSize); + } + + static void bufferTest(int port) throws Exception { + ByteBuffer[] bufs = new ByteBuffer[testSize]; + long total = 0L; + for(int i = 0; i < testSize; i++) { + String source = "buffer" + i; + if (generator.nextBoolean()) + bufs[i] = ByteBuffer.allocateDirect(source.length()); + else + bufs[i] = ByteBuffer.allocate(source.length()); + + bufs[i].put(source.getBytes("8859_1")); + bufs[i].flip(); + total += bufs[i].remaining(); + } + + ByteBuffer[] bufsPlus1 = new ByteBuffer[bufs.length + 1]; + System.arraycopy(bufs, 0, bufsPlus1, 0, bufs.length); + + // Get a connection to the server + InetAddress lh = InetAddress.getLocalHost(); + InetSocketAddress isa = new InetSocketAddress(lh, port); + SocketChannel sc = Sockets.openRdmaSocketChannel(); + sc.connect(isa); + sc.configureBlocking(generator.nextBoolean()); + + // Write the data out + long rem = total; + while (rem > 0L) { + long bytesWritten; + if (setLength) { + bytesWritten = sc.write(bufsPlus1, 0, bufs.length); + } else { + bytesWritten = sc.write(bufs); + } + if (bytesWritten == 0) { + if (sc.isBlocking()) { + throw new RuntimeException("write did not block"); + } else { + System.err.println("Non-blocking write() wrote zero bytes"); + } + Thread.sleep(50); + } else { + rem -= bytesWritten; + } + } + + // Clean up + sc.close(); + } + + static class Server + extends TestThread + { + final int testSize; + final ServerSocketChannel ssc; + + Server(int testSize) throws IOException { + super("Server " + testSize); + this.testSize = testSize; + this.ssc = Sockets.openRdmaServerSocketChannel().bind(new InetSocketAddress(0)); + } + + int port() { + return ssc.socket().getLocalPort(); + } + + void go() throws Exception { + bufferTest(); + } + + void bufferTest() throws Exception { + long total = 0L; + ByteBuffer[] bufs = new ByteBuffer[testSize]; + for(int i=0; i 0) { + long bytesRead; + if (setLength) { + bytesRead = sc.read(bufsPlus1, 0, bufs.length); + } else { + bytesRead = sc.read(bufs); + } + if (bytesRead < 0) + break; + if (bytesRead == 0) { + if (sc.isBlocking()) { + throw new RuntimeException("read did not block"); + } else { + System.err.println + ("Non-blocking read() read zero bytes"); + } + Thread.sleep(50); + } + avail -= bytesRead; + } + + // Check results + for(int i=0; i