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