1 /* 2 * Copyright (c) 2001, 2008, 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 * @build HandshakeTimeout 37 * @run main/othervm HandshakeTimeout 38 */ 39 40 import java.net.ServerSocket; 41 import java.rmi.ConnectException; 42 import java.rmi.ConnectIOException; 43 import java.rmi.MarshalException; 44 import java.rmi.registry.LocateRegistry; 45 import java.rmi.registry.Registry; 46 47 public class HandshakeTimeout { 48 49 private static final int PORT = 2020; 50 private static final int TIMEOUT = 10000; 51 52 public static void main(String[] args) throws Exception { 53 54 System.setProperty("sun.rmi.transport.tcp.handshakeTimeout", 55 String.valueOf(TIMEOUT / 2)); 56 57 /* 58 * Listen on port, but never process connections made to it. 59 */ 60 ServerSocket serverSocket = new ServerSocket(PORT); 61 62 /* 63 * Attempt RMI call to port in separate thread. 64 */ 65 Registry registry = LocateRegistry.getRegistry(PORT); 66 Connector connector = new Connector(registry); 67 Thread t = new Thread(connector); 68 t.setDaemon(true); 69 t.start(); 70 71 /* 72 * Wait for call attempt to finished, and analyze result. 73 */ 74 t.join(TIMEOUT); 75 synchronized (connector) { 76 if (connector.success) { 77 throw new RuntimeException( 78 "TEST FAILED: remote call succeeded??"); 79 } 80 if (connector.exception == null) { 81 throw new RuntimeException( 82 "TEST FAILED: remote call did not time out"); 83 } else { 84 System.err.println("remote call failed with exception:"); 85 connector.exception.printStackTrace(); 86 System.err.println(); 87 88 if (connector.exception instanceof MarshalException) { 89 System.err.println( 90 "TEST FAILED: MarshalException thrown, expecting " + 91 "java.rmi.ConnectException or ConnectIOException"); 92 } else if (connector.exception instanceof ConnectException || 93 connector.exception instanceof ConnectIOException) 94 { 95 System.err.println( 96 "TEST PASSED: java.rmi.ConnectException or " + 97 "ConnectIOException thrown"); 98 } else { 99 throw new RuntimeException( 100 "TEST FAILED: unexpected Exception thrown", 101 connector.exception); 102 } 103 } 104 } 105 } 106 107 private static class Connector implements Runnable { 108 109 private final Registry registry; 110 111 boolean success = false; 112 Exception exception = null; 113 114 Connector(Registry registry) { 115 this.registry = registry; 116 } 117 118 public void run() { 119 try { 120 registry.lookup("Dale Cooper"); 121 synchronized (this) { 122 success = true; 123 } 124 } catch (Exception e) { 125 synchronized (this) { 126 exception = e; 127 } 128 } 129 } 130 } 131 }