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 }