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 4485966
  26  * @summary Whan an RMI (JRMP) connection is made to a TCP address that is
  27  * listening, so the connection is accepted, but the server responds with
  28  * invalid JRMP protocol (such as because a non-JRMP server is currently
  29  * listening at that address), the client application should receive a
  30  * java.rmi.ConnectException or ConnectIOException, not a MarshalException.
  31  * @author Peter Jones
  32  *
  33  * @library ../../testlibrary
  34  * @build HandshakeFailure TestLibrary
  35  * @run main/othervm HandshakeFailure
  36  */
  37 
  38 import java.net.ServerSocket;
  39 import java.net.Socket;
  40 import java.rmi.ConnectException;
  41 import java.rmi.ConnectIOException;
  42 import java.rmi.MarshalException;
  43 import java.rmi.registry.LocateRegistry;
  44 import java.rmi.registry.Registry;
  45 
  46 public class HandshakeFailure {
  47 
  48     private static final int PORT = TestLibrary.getUnusedRandomPort();
  49     private static final int TIMEOUT = 10000;
  50 
  51     public static void main(String[] args) throws Exception {
  52 
  53         /*
  54          * Listen on port...
  55          */
  56         ServerSocket serverSocket = new ServerSocket(PORT);
  57 
  58         /*
  59          * (Attempt RMI call to port in separate thread.)
  60          */
  61         Registry registry = LocateRegistry.getRegistry(PORT);
  62         Connector connector = new Connector(registry);
  63         Thread t = new Thread(connector);
  64         t.setDaemon(true);
  65         t.start();
  66 
  67         /*
  68          * ...accept one connection from port and send non-JRMP data.
  69          */
  70         Socket socket = serverSocket.accept();
  71         socket.getOutputStream().write("Wrong way".getBytes());
  72         socket.close();
  73 
  74         /*
  75          * Wait for call attempt to finish, and analyze result.
  76          */
  77         t.join(TIMEOUT);
  78         synchronized (connector) {
  79             if (connector.success) {
  80                 throw new RuntimeException(
  81                     "TEST FAILED: remote call succeeded??");
  82             }
  83             if (connector.exception == null) {
  84                 throw new RuntimeException(
  85                     "TEST FAILED: remote call did not time out");
  86             } else {
  87                 System.err.println("remote call failed with exception:");
  88                 connector.exception.printStackTrace();
  89                 System.err.println();
  90 
  91                 if (connector.exception instanceof MarshalException) {
  92                     System.err.println(
  93                         "TEST FAILED: MarshalException thrown, expecting " +
  94                         "java.rmi.ConnectException or ConnectIOException");
  95                 } else if (connector.exception instanceof ConnectException ||
  96                            connector.exception instanceof ConnectIOException)
  97                 {
  98                     System.err.println(
  99                         "TEST PASSED: java.rmi.ConnectException or " +
 100                         "ConnectIOException thrown");
 101                 } else {
 102                     throw new RuntimeException(
 103                         "TEST FAILED: unexpected Exception thrown",
 104                         connector.exception);
 105                 }
 106             }
 107         }
 108     }
 109 
 110     private static class Connector implements Runnable {
 111 
 112         private final Registry registry;
 113 
 114         boolean success = false;
 115         Exception exception = null;
 116 
 117         Connector(Registry registry) {
 118             this.registry = registry;
 119         }
 120 
 121         public void run() {
 122             try {
 123                 registry.lookup("Dale Cooper");
 124                 synchronized (this) {
 125                     success = true;
 126                 }
 127             } catch (Exception e) {
 128                 synchronized (this) {
 129                     exception = e;
 130                 }
 131             }
 132         }
 133     }
 134 }