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 }