| /* |
| * Copyright (c) 2003, 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. |
| */ |
| |
| /* |
| * |
| * |
| * Tests that the channel returned by System.inheritedChannel() |
| * is in blocking mode, bound, and in the case of a SocketChannel |
| * connected to a peer. |
| * |
| * The test works by launching a test service (called StateTestService) so |
| * that it inherits each type of channel. The test service checks the |
| * socket state and replies back to this class via an out-of-band |
| * channel. |
| */ |
| import java.io.*; |
| import java.net.*; |
| import java.nio.ByteBuffer; |
| import java.nio.channels.*; |
| |
| public class StateTest { |
| |
| private static int failures = 0; |
| |
| private static String TEST_SERVICE = "StateTestService"; |
| |
| /* |
| * Reads the test result from the "out-of-band" connection to the test service. |
| * |
| * The out-of-band connection is just a TCP connection from the service to |
| * this class. waitForTestResult just waits (with timeout) for the service |
| * to connect. Once connected it waits (with timeout) for the test result. |
| * The test result is examined. |
| */ |
| private static void waitForTestResult(ServerSocketChannel ssc, boolean expectFail) throws IOException { |
| Selector sel = ssc.provider().openSelector(); |
| SelectionKey sk; |
| SocketChannel sc; |
| |
| /* |
| * Wait for service to connect |
| */ |
| ssc.configureBlocking(false); |
| sk = ssc.register(sel, SelectionKey.OP_ACCEPT); |
| long to = 15*1000; |
| sc = null; |
| for (;;) { |
| long st = System.currentTimeMillis(); |
| sel.select(to); |
| if (sk.isAcceptable() && ((sc = ssc.accept()) != null)) { |
| // connection established |
| break; |
| } |
| sel.selectedKeys().remove(sk); |
| to -= System.currentTimeMillis() - st; |
| if (to <= 0) { |
| throw new IOException("Timed out waiting for service to report test result"); |
| } |
| } |
| sk.cancel(); |
| ssc.configureBlocking(false); |
| |
| /* |
| * Wait for service to report test result |
| */ |
| sc.configureBlocking(false); |
| sk = sc.register(sel, SelectionKey.OP_READ); |
| to = 5000; |
| ByteBuffer bb = ByteBuffer.allocateDirect(20); |
| for (;;) { |
| long st = System.currentTimeMillis(); |
| sel.select(to); |
| if (sk.isReadable()) { |
| int n = sc.read(bb); |
| if (n > 0) { |
| break; |
| } |
| if (n < 0) { |
| throw new IOException("Premature EOF - no test result from service"); |
| } |
| } |
| sel.selectedKeys().remove(sk); |
| to -= System.currentTimeMillis() - st; |
| if (to <= 0) { |
| throw new IOException("Timed out waiting for service to report test result"); |
| } |
| } |
| sk.cancel(); |
| sc.close(); |
| sel.close(); |
| |
| /* |
| * Examine the test result |
| */ |
| bb.flip(); |
| byte b = bb.get(); |
| |
| if (expectFail && b == 'P') { |
| System.err.println("Test passed - test is expected to fail!!!"); |
| failures++; |
| } |
| if (!expectFail && b != 'P') { |
| System.err.println("Test failed!"); |
| failures++; |
| } |
| } |
| |
| public static void main(String args[]) throws IOException { |
| boolean expectFail = false; |
| |
| /* |
| * [-expectFail] [options...] |
| */ |
| String options[] = args; |
| if (args.length > 0 && args[0].equals("-expectFail")) { |
| // shift out first arg to create options |
| expectFail = true; |
| options = new String[args.length-1]; |
| if (args.length > 1) { |
| System.arraycopy(args, 1, options, 0, args.length-1); |
| } |
| } |
| |
| /* |
| * Create the listener which will be used to read the test result |
| * from the service. |
| */ |
| ServerSocketChannel ssc = ServerSocketChannel.open(); |
| ssc.socket().bind(new InetSocketAddress(0)); |
| |
| /* |
| * The port is passed to the service as an argument. |
| */ |
| int port = ssc.socket().getLocalPort(); |
| String arg[] = new String[1]; |
| arg[0] = String.valueOf(port); |
| |
| /* |
| * Launch service with a SocketChannel (tcp nowait) |
| */ |
| SocketChannel sc = Launcher.launchWithSocketChannel(TEST_SERVICE, options, arg); |
| waitForTestResult(ssc, expectFail); |
| sc.close(); |
| |
| /* |
| * Launch service with a ServerSocketChannel (tcp wait) |
| * launchWithServerSocketChannel establishes a connection to the service |
| * and the returned SocketChannel is connected to the service. |
| */ |
| sc = Launcher.launchWithServerSocketChannel(TEST_SERVICE, options, arg); |
| waitForTestResult(ssc, expectFail); |
| sc.close(); |
| |
| /* |
| * Launch service with a DatagramChannel (udp wait) |
| */ |
| DatagramChannel dc = Launcher.launchWithDatagramChannel(TEST_SERVICE, options, arg); |
| waitForTestResult(ssc, expectFail); |
| dc.close(); |
| |
| if (failures > 0) { |
| throw new RuntimeException("Test failed - see log for details"); |
| } else { |
| System.out.println("All tests passed."); |
| } |
| } |
| } |