1/* 2 * Copyright (c) 2001, 2016, 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 * @bug 8156002 26 * @summary Unit test for socket-channel adaptors 27 * @library .. 28 */ 29 30import java.io.*; 31import java.net.*; 32import java.nio.channels.*; 33import java.util.Arrays; 34 35 36public class AdaptSocket { 37 38 static java.io.PrintStream out = System.out; 39 40 static void test(TestServers.DayTimeServer dayTimeServer, 41 int timeout, 42 boolean shouldTimeout) 43 throws Exception 44 { 45 out.println(); 46 47 InetSocketAddress isa 48 = new InetSocketAddress(dayTimeServer.getAddress(), 49 dayTimeServer.getPort()); 50 SocketChannel sc = SocketChannel.open(); 51 Socket so = sc.socket(); 52 out.println("opened: " + so); 53 out.println(" " + sc); 54 55 //out.println("opts: " + sc.options()); 56 so.setTcpNoDelay(true); 57 //so.setTrafficClass(SocketOpts.IP.TOS_THROUGHPUT); 58 so.setKeepAlive(true); 59 so.setSoLinger(true, 42); 60 so.setOOBInline(true); 61 so.setReceiveBufferSize(512); 62 so.setSendBufferSize(512); 63 //out.println(" " + sc.options()); 64 65 if (timeout == 0) 66 so.connect(isa); 67 else { 68 try { 69 so.connect(isa, timeout); 70 } catch (SocketTimeoutException x) { 71 if (shouldTimeout) { 72 out.println("Connection timed out, as expected"); 73 return; 74 } else { 75 throw x; 76 } 77 } 78 } 79 out.println("connected: " + so); 80 out.println(" " + sc); 81 byte[] bb = new byte[100]; 82 int n = so.getInputStream().read(bb); 83 String s = new String(bb, 0, n - 2, "US-ASCII"); 84 out.println(isa + " says: \"" + s + "\""); 85 so.shutdownInput(); 86 out.println("ishut: " + sc); 87 so.shutdownOutput(); 88 out.println("oshut: " + sc); 89 so.close(); 90 out.println("closed: " + so); 91 out.println(" " + sc); 92 } 93 94 static String dataString = "foo\r\n"; 95 96 static void testRead(Socket so, boolean shouldTimeout) 97 throws Exception 98 { 99 String data = "foo\r\n"; 100 so.getOutputStream().write(dataString.getBytes("US-ASCII")); 101 InputStream is = so.getInputStream(); 102 try { 103 byte[] b = new byte[100]; 104 int n = is.read(b); 105 if (shouldTimeout) { 106 throw new Exception("Should time out, but not, data: " + Arrays.toString(b)); 107 } 108 if (n != 5) { 109 throw new Exception("Incorrect number of bytes read: " + n); 110 } 111 if (!dataString.equals(new String(b, 0, n, "US-ASCII"))) { 112 throw new Exception("Incorrect data read: " + n); 113 } 114 } catch (SocketTimeoutException x) { 115 if (shouldTimeout) { 116 out.println("Read timed out, as expected"); 117 return; 118 } 119 throw x; 120 } 121 } 122 123 static void testRead(TestServers.EchoServer echoServer, 124 int timeout, 125 boolean shouldTimeout) 126 throws Exception 127 { 128 out.println(); 129 130 InetSocketAddress isa 131 = new InetSocketAddress(echoServer.getAddress(), 132 echoServer.getPort()); 133 SocketChannel sc = SocketChannel.open(); 134 sc.connect(isa); 135 Socket so = sc.socket(); 136 out.println("connected: " + so); 137 out.println(" " + sc); 138 139 if (timeout > 0) 140 so.setSoTimeout(timeout); 141 out.println("timeout: " + so.getSoTimeout()); 142 143 testRead(so, shouldTimeout); 144 for (int i = 0; i < 4; i++) { 145 out.println("loop: " + i); 146 testRead(so, shouldTimeout); 147 } 148 149 sc.close(); 150 } 151 152 public static void main(String[] args) throws Exception { 153 154 try (TestServers.DayTimeServer dayTimeServer 155 = TestServers.DayTimeServer.startNewServer()) { 156 test(dayTimeServer, 0, false); 157 test(dayTimeServer, 1000, false); 158 } 159 160 try (TestServers.DayTimeServer lingerDayTimeServer 161 = TestServers.DayTimeServer.startNewServer(100)) { 162 // this test no longer really test the connection timeout 163 // since there is no way to prevent the server from eagerly 164 // accepting connection... 165 test(lingerDayTimeServer, 10, true); 166 } 167 168 try (TestServers.EchoServer echoServer 169 = TestServers.EchoServer.startNewServer()) { 170 testRead(echoServer, 0, false); 171 testRead(echoServer, 8000, false); 172 } 173 174 try (TestServers.NoResponseServer noResponseServer 175 = TestServers.NoResponseServer.startNewServer()) { 176 testRead(noResponseServer, 10, true); 177 } 178 } 179} 180