1 /* 2 * Copyright (c) 2001, 2012, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* @test 25 * @summary Test SocketChannel.finishConnect 26 * @library .. 27 */ 28 29 import java.net.*; 30 import java.nio.*; 31 import java.nio.channels.*; 32 import java.nio.channels.spi.SelectorProvider; 33 import java.nio.charset.*; 34 import java.util.*; 35 36 37 public class FinishConnect { 38 39 public static void main(String[] args) throws Exception { 40 try (TestClass.DayTimeServer dayTimeServer 41 = TestClass.DayTimeServer.startNewServer(100)) { 42 test1(dayTimeServer, true, true); 43 test1(dayTimeServer, true, false); 44 test1(dayTimeServer, false, true); 45 test1(dayTimeServer, false, false); 46 test2(dayTimeServer); 47 } 48 } 49 50 static void test1(TestClass.DayTimeServer daytimeServer, 51 boolean select, 52 boolean setBlocking) 53 throws Exception 54 { 55 InetSocketAddress isa 56 = new InetSocketAddress(daytimeServer.getAddress(), 57 daytimeServer.getPort()); 58 SocketChannel sc = SocketChannel.open(); 59 sc.configureBlocking(false); 60 boolean connected = sc.connect(isa); 61 int attempts = 0; 62 63 try { 64 sc.connect(isa); 65 throw new RuntimeException("Allowed another connect call"); 66 } catch (IllegalStateException ise) { 67 // Correct behavior 68 } 69 70 if (setBlocking) 71 sc.configureBlocking(true); 72 73 if (!connected && select && !setBlocking) { 74 Selector selector = SelectorProvider.provider().openSelector(); 75 sc.register(selector, SelectionKey.OP_CONNECT); 76 while (!connected) { 77 int keysAdded = selector.select(100); 78 if (keysAdded > 0) { 79 Set readyKeys = selector.selectedKeys(); 80 Iterator i = readyKeys.iterator(); 81 while (i.hasNext()) { 82 SelectionKey sk = (SelectionKey)i.next(); 83 SocketChannel nextReady = 84 (SocketChannel)sk.channel(); 85 connected = sc.finishConnect(); 86 } 87 } 88 } 89 selector.close(); 90 } 91 92 while (!connected) { 93 if (attempts++ > 30) 94 throw new RuntimeException("Failed to connect"); 95 Thread.sleep(100); 96 connected = sc.finishConnect(); 97 } 98 99 ByteBuffer bb = ByteBuffer.allocateDirect(100); 100 int bytesRead = 0; 101 int totalRead = 0; 102 while (totalRead < 20) { 103 bytesRead = sc.read(bb); 104 if (bytesRead > 0) 105 totalRead += bytesRead; 106 if (bytesRead < 0) 107 throw new RuntimeException("Message shorter than expected"); 108 } 109 bb.position(bb.position() - 2); // Drop CRLF 110 bb.flip(); 111 CharBuffer cb = Charset.forName("US-ASCII").newDecoder().decode(bb); 112 System.err.println(isa + " says: \"" + cb + "\""); 113 sc.close(); 114 } 115 116 static void test2(TestClass.DayTimeServer daytimeServer) throws Exception { 117 InetSocketAddress isa 118 = new InetSocketAddress(daytimeServer.getAddress(), 119 daytimeServer.getPort()); 120 boolean done = false; 121 int globalAttempts = 0; 122 int connectSuccess = 0; 123 while (!done) { 124 // When using a local daytime server it is not always possible 125 // to get a pending connection, as sc.connect(isa) may always 126 // return true. 127 // So we're going to throw the exception only if there was 128 // at least 1 case where we did not manage to connect. 129 if (globalAttempts++ > 50) { 130 if (globalAttempts == connectSuccess + 1) { 131 System.out.println("Can't fully test on " 132 + System.getProperty("os.name")); 133 break; 134 } 135 throw new RuntimeException("Failed to connect"); 136 } 137 SocketChannel sc = SocketChannel.open(); 138 sc.configureBlocking(false); 139 boolean connected = sc.connect(isa); 140 int localAttempts = 0; 141 while (!connected) { 142 if (localAttempts++ > 500) 143 throw new RuntimeException("Failed to connect"); 144 connected = sc.finishConnect(); 145 if (connected) { 146 done = true; 147 break; 148 } 149 Thread.sleep(10); 150 } 151 if (connected) { 152 connectSuccess++; 153 } 154 sc.close(); 155 } 156 } 157 158 }