1/* 2 * Copyright (c) 2015, 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/* 25 * @test 26 * @bug 8067105 27 * @summary Socket returned by ServerSocket.accept() is inherited by child process on Windows 28 * @author Chris Hegarty 29 */ 30 31import java.io.*; 32import java.net.*; 33import java.nio.channels.ServerSocketChannel; 34import java.util.ArrayList; 35import java.util.List; 36import java.util.concurrent.TimeUnit; 37import java.util.function.Supplier; 38 39public class AcceptInheritHandle { 40 41 enum ServerSocketProducer { 42 JAVA_NET(() -> { 43 try { 44 return new ServerSocket(); } 45 catch(IOException x) { 46 throw new UncheckedIOException(x); 47 } 48 }), 49 NIO_CHANNELS(() -> { 50 try { 51 return ServerSocketChannel.open().socket(); 52 } catch (IOException x) { 53 throw new UncheckedIOException(x); 54 } 55 }); 56 57 final Supplier<ServerSocket> supplier; 58 ServerSocketProducer(Supplier<ServerSocket> supplier) { 59 this.supplier = supplier; 60 } 61 Supplier<ServerSocket> supplier () { return supplier; } 62 } 63 64 static final String JAVA = System.getProperty("java.home") 65 + File.separator + "bin" + File.separator + "java"; 66 67 static final String CLASSPATH = System.getProperty("java.class.path"); 68 69 public static void main(String[] args) throws Exception { 70 if (args.length == 1) 71 server(ServerSocketProducer.valueOf(args[0])); 72 else 73 mainEntry(); 74 } 75 76 static void mainEntry() throws Exception { 77 testJavaNetServerSocket(); 78 testNioServerSocketChannel(); 79 } 80 81 static void testJavaNetServerSocket() throws Exception { 82 test(ServerSocketProducer.JAVA_NET); 83 test(ServerSocketProducer.JAVA_NET, "-Djava.net.preferIPv4Stack=true"); 84 } 85 static void testNioServerSocketChannel() throws Exception { 86 test(ServerSocketProducer.NIO_CHANNELS); 87 } 88 89 static void test(ServerSocketProducer ssp, String... sysProps) throws Exception { 90 System.out.println("\nStarting test for " + ssp.name()); 91 92 List<String> commands = new ArrayList<>(); 93 commands.add(JAVA); 94 for (String prop : sysProps) 95 commands.add(prop); 96 commands.add("-cp"); 97 commands.add(CLASSPATH); 98 commands.add("AcceptInheritHandle"); 99 commands.add(ssp.name()); 100 101 System.out.println("Executing: "+ commands); 102 ProcessBuilder pb = new ProcessBuilder(commands); 103 pb.redirectError(ProcessBuilder.Redirect.INHERIT); 104 Process serverProcess = pb.start(); 105 DataInputStream dis = new DataInputStream(serverProcess.getInputStream()); 106 107 int port = dis.readInt(); 108 System.out.println("Server process listening on " + port + ", connecting..."); 109 110 Socket socket = new Socket("localhost", port); 111 String s = dis.readUTF(); 112 System.out.println("Server process said " + s); 113 114 serverProcess.destroy(); 115 serverProcess.waitFor(30, TimeUnit.SECONDS); 116 System.out.println("serverProcess exitCode:" + serverProcess.exitValue()); 117 118 try { 119 socket.setSoTimeout(10 * 1000); 120 socket.getInputStream().read(); 121 } catch (SocketTimeoutException x) { 122 // failed 123 throw new RuntimeException("Failed: should get reset, not " + x); 124 } catch (SocketException x) { 125 System.out.println("Expected:" + x); 126 } 127 } 128 129 static void server(ServerSocketProducer producer) throws Exception { 130 try (ServerSocket ss = producer.supplier().get()) { 131 ss.bind(new InetSocketAddress(0)); 132 int port = ss.getLocalPort(); 133 DataOutputStream dos = new DataOutputStream(System.out); 134 dos.writeInt(port); 135 dos.flush(); 136 137 ss.accept(); // do not close 138 139 Runtime.getRuntime().exec("sleep 20"); 140 Thread.sleep(3 * 1000); 141 142 dos.writeUTF("kill me!"); 143 dos.flush(); 144 Thread.sleep(30 * 1000); 145 } 146 } 147} 148