--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/java/nio/channels/SocketChannel/FinishConnect.java Tue Sep 12 19:03:39 2017 +0200
@@ -0,0 +1,158 @@
+/*
+ * Copyright (c) 2001, 2012, 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
+ * @summary Test SocketChannel.finishConnect
+ * @library ..
+ */
+
+import java.net.*;
+import java.nio.*;
+import java.nio.channels.*;
+import java.nio.channels.spi.SelectorProvider;
+import java.nio.charset.*;
+import java.util.*;
+
+
+public class FinishConnect {
+
+ public static void main(String[] args) throws Exception {
+ 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 = SocketChannel.open();
+ 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 = SelectorProvider.provider().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 = SocketChannel.open();
+ 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();
+ }
+ }
+
+}