--- /dev/null 2018-11-27 03:18:47.532777276 -0500 +++ new/test/jdk/jdk/net/RdmaSockets/rsocket/SocketChannel/FinishConnect.java 2018-11-30 08:33:25.393538323 -0500 @@ -0,0 +1,175 @@ +/* + * 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 RDMA SocketChannel.finishConnect + * @requires (os.family == "linux") + * @library .. /test/lib + * @build jdk.test.lib.Utils TestServers + * @build RsocketTest + * @run main/othervm FinishConnect + */ + +import java.net.InetSocketAddress; +import java.net.StandardProtocolFamily; +import java.nio.ByteBuffer; +import java.nio.CharBuffer; +import java.nio.channels.Selector; +import java.nio.channels.SelectionKey; +import java.nio.channels.SocketChannel; +import java.nio.channels.spi.SelectorProvider; +import java.nio.charset.Charset; +import java.util.Iterator; +import java.util.Set; +import jdk.net.RdmaSockets; + +import jtreg.SkippedException; + +public class FinishConnect { + + public static void main(String[] args) throws Exception { + if (!RsocketTest.isRsocketAvailable()) + throw new SkippedException("rsocket is not available"); + + try (TestServers.DayTimeServer dayTimeServer + = TestServers.DayTimeServer.startNewServer(100)) { + test1(dayTimeServer, true, true); + test1(dayTimeServer, true, false); + test1(dayTimeServer, false, true); + test1(dayTimeServer, false, false); + test2(dayTimeServer); + } + } + + static void test1(TestServers.DayTimeServer daytimeServer, + boolean select, + boolean setBlocking) + throws Exception + { + InetSocketAddress isa + = new InetSocketAddress(daytimeServer.getAddress(), + daytimeServer.getPort()); + SocketChannel sc = RdmaSockets.openSocketChannel( + StandardProtocolFamily.INET); + sc.configureBlocking(false); + boolean connected = sc.connect(isa); + int attempts = 0; + + try { + sc.connect(isa); + throw new RuntimeException("Allowed another connect call"); + } catch (IllegalStateException ise) { + // Correct behavior + } + + if (setBlocking) + sc.configureBlocking(true); + + if (!connected && select && !setBlocking) { + Selector selector = RdmaSockets.openSelector(); + sc.register(selector, SelectionKey.OP_CONNECT); + while (!connected) { + int keysAdded = selector.select(100); + if (keysAdded > 0) { + Set readyKeys = selector.selectedKeys(); + Iterator i = readyKeys.iterator(); + while (i.hasNext()) { + SelectionKey sk = (SelectionKey)i.next(); + SocketChannel nextReady = + (SocketChannel)sk.channel(); + connected = sc.finishConnect(); + } + } + } + selector.close(); + } + + while (!connected) { + if (attempts++ > 30) + throw new RuntimeException("Failed to connect"); + Thread.sleep(100); + connected = sc.finishConnect(); + } + + ByteBuffer bb = ByteBuffer.allocateDirect(100); + int bytesRead = 0; + int totalRead = 0; + while (totalRead < 20) { + bytesRead = sc.read(bb); + if (bytesRead > 0) + totalRead += bytesRead; + if (bytesRead < 0) + throw new RuntimeException("Message shorter than expected"); + } + bb.position(bb.position() - 2); // Drop CRLF + bb.flip(); + CharBuffer cb = Charset.forName("US-ASCII").newDecoder().decode(bb); + System.err.println(isa + " says: \"" + cb + "\""); + sc.close(); + } + + static void test2(TestServers.DayTimeServer daytimeServer) throws Exception { + InetSocketAddress isa + = new InetSocketAddress(daytimeServer.getAddress(), + daytimeServer.getPort()); + boolean done = false; + int globalAttempts = 0; + int connectSuccess = 0; + while (!done) { + // When using a local daytime server it is not always possible + // to get a pending connection, as sc.connect(isa) may always + // return true. + // So we're going to throw the exception only if there was + // at least 1 case where we did not manage to connect. + if (globalAttempts++ > 50) { + if (globalAttempts == connectSuccess + 1) { + System.out.println("Can't fully test on " + + System.getProperty("os.name")); + break; + } + throw new RuntimeException("Failed to connect"); + } + SocketChannel sc = RdmaSockets.openSocketChannel( + StandardProtocolFamily.INET); + sc.configureBlocking(false); + boolean connected = sc.connect(isa); + int localAttempts = 0; + while (!connected) { + if (localAttempts++ > 500) + throw new RuntimeException("Failed to connect"); + connected = sc.finishConnect(); + if (connected) { + done = true; + break; + } + Thread.sleep(10); + } + if (connected) { + connectSuccess++; + } + sc.close(); + } + } + +}