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 * @bug 4322806 26 * @summary When an RMI (JRMP) connection is made to a TCP address that is 27 * listening, so the connection is accepted, but the server never responds 28 * to the initial JRMP handshake (nor does it terminate the connection), 29 * the client should not hang forever; instead, it should throw an exception 30 * after a reasonable timeout interval. The exception should be a 31 * java.rmi.ConnectException or ConnectIOException, not a MarshalException, 32 * because it should be clear that no partial call execution has occurred at 33 * this point (because no data for the invocation has yet been written). 34 * @author Peter Jones 35 * 36 * @library ../../testlibrary 37 * @modules java.rmi/sun.rmi.registry 38 * java.rmi/sun.rmi.server 39 * java.rmi/sun.rmi.transport 40 * java.rmi/sun.rmi.transport.tcp 41 * @build TestLibrary 42 * @run main/othervm HandshakeTimeout 43 */ 44 45 import java.net.ServerSocket; 46 import java.rmi.ConnectException; 47 import java.rmi.ConnectIOException; 48 import java.rmi.MarshalException; 49 import java.rmi.registry.LocateRegistry; 50 import java.rmi.registry.Registry; 51 52 public class HandshakeTimeout { 53 54 private static final int PORT = TestLibrary.getUnusedRandomPort(); 55 private static final int TIMEOUT = 10000; 56 57 public static void main(String[] args) throws Exception { 58 59 System.setProperty("sun.rmi.transport.tcp.handshakeTimeout", 60 String.valueOf(TIMEOUT / 2)); 61 62 /* 63 * Listen on port, but never process connections made to it. 64 */ 65 ServerSocket serverSocket = new ServerSocket(PORT); 66 67 /* 68 * Attempt RMI call to port in separate thread. 69 */ 70 Registry registry = LocateRegistry.getRegistry(PORT); 71 Connector connector = new Connector(registry); 72 Thread t = new Thread(connector); 73 t.setDaemon(true); 74 t.start(); 75 76 /* 77 * Wait for call attempt to finished, and analyze result. 78 */ 79 t.join(TIMEOUT); 80 synchronized (connector) { 81 if (connector.success) { 82 throw new RuntimeException( 83 "TEST FAILED: remote call succeeded??"); 84 } 85 if (connector.exception == null) { 86 throw new RuntimeException( 87 "TEST FAILED: remote call did not time out"); 88 } else { 89 System.err.println("remote call failed with exception:"); 90 connector.exception.printStackTrace(); 91 System.err.println(); 92 93 if (connector.exception instanceof MarshalException) { 94 System.err.println( 95 "TEST FAILED: MarshalException thrown, expecting " + 96 "java.rmi.ConnectException or ConnectIOException"); 97 } else if (connector.exception instanceof ConnectException || 98 connector.exception instanceof ConnectIOException) 99 { 100 System.err.println( 101 "TEST PASSED: java.rmi.ConnectException or " + 102 "ConnectIOException thrown"); 103 } else { 104 throw new RuntimeException( 105 "TEST FAILED: unexpected Exception thrown", 106 connector.exception); 107 } 108 } 109 } 110 } 111 112 private static class Connector implements Runnable { 113 114 private final Registry registry; 115 116 boolean success = false; 117 Exception exception = null; 118 119 Connector(Registry registry) { 120 this.registry = registry; 121 } 122 123 public void run() { 124 try { 125 registry.lookup("Dale Cooper"); 126 synchronized (this) { 127 success = true; 128 } 129 } catch (Exception e) { 130 synchronized (this) { 131 exception = e; 132 } 133 } 134 } 135 } 136 }