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