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 }