1 /*
   2  * Copyright (c) 2005, 2013, 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 4457683
  26  * @summary After all of the remote objects (including a registry, if
  27  * applicable) that had been exported with a given
  28  * RMIServerSocketFactory value (including null) have been unexported,
  29  * the server socket created for the exports should be closed (so that
  30  * the local port is released).
  31  * @author Peter Jones
  32  *
  33  * @library ../../testlibrary
  34  * @modules java.rmi/sun.rmi.registry
  35  *          java.rmi/sun.rmi.server
  36  *          java.rmi/sun.rmi.transport
  37  *          java.rmi/sun.rmi.transport.tcp
  38  * @build TestLibrary
  39  * @run main/othervm CloseServerSocket
  40  */
  41 
  42 import java.io.IOException;
  43 import java.net.BindException;
  44 import java.net.ServerSocket;
  45 import java.rmi.Remote;
  46 import java.rmi.registry.LocateRegistry;
  47 import java.rmi.registry.Registry;
  48 import java.rmi.server.RMIServerSocketFactory;
  49 import java.rmi.server.UnicastRemoteObject;
  50 
  51 public class CloseServerSocket implements Remote {
  52     private static final int PORT = TestLibrary.getUnusedRandomPort();
  53 
  54     private CloseServerSocket() { }
  55 
  56     public static void main(String[] args) throws Exception {
  57         System.err.println("\nRegression test for bug 4457683\n");
  58 
  59         verifyPortFree(PORT);
  60         Registry registry = LocateRegistry.createRegistry(PORT);
  61         System.err.println("- exported registry: " + registry);
  62         verifyPortInUse(PORT);
  63         UnicastRemoteObject.unexportObject(registry, true);
  64         System.err.println("- unexported registry");
  65         Thread.sleep(1000);        // work around BindException (bug?)
  66         verifyPortFree(PORT);
  67 
  68         /*
  69          * The follow portion of this test is disabled temporarily
  70          * because 4457683 was partially backed out because of
  71          * 6269166; for now, only server sockets originally opened for
  72          * exports on non-anonymous ports will be closed when all of
  73          * the corresponding remote objects have been exported.  A
  74          * separate bug will be filed to represent the remainder of
  75          * 4457683 for anonymous-port exports.
  76          */
  77 
  78 //      SSF ssf = new SSF();
  79 //      Remote impl = new CloseServerSocket();
  80 //      Remote stub = UnicastRemoteObject.exportObject(impl, 0, null, ssf);
  81 //      System.err.println("- exported object: " + stub);
  82 //      UnicastRemoteObject.unexportObject(impl, true);
  83 //      System.err.println("- unexported object");
  84 //      synchronized (ssf) {
  85 //          if (!ssf.serverSocketClosed) {
  86 //              throw new RuntimeException("TEST FAILED: " +
  87 //                                         "server socket not closed");
  88 //          }
  89 //      }
  90 
  91         System.err.println("TEST PASSED");
  92     }
  93 
  94     private static void verifyPortFree(int port) throws IOException {
  95         ServerSocket ss = new ServerSocket(port);
  96         ss.close();
  97         System.err.println("- port " + port + " is free");
  98     }
  99 
 100     private static void verifyPortInUse(int port) throws IOException {
 101         try {
 102             verifyPortFree(port);
 103         } catch (BindException e) {
 104             System.err.println("- port " + port + " is in use");
 105             return;
 106         }
 107     }
 108 
 109     private static class SSF implements RMIServerSocketFactory {
 110         boolean serverSocketClosed = false;
 111         SSF() { };
 112 
 113         public ServerSocket createServerSocket(int port) throws IOException {
 114             return new SS(port);
 115         }
 116 
 117         private class SS extends ServerSocket {
 118             SS(int port) throws IOException {
 119                 super(port);
 120                 System.err.println("- created server socket: " + this);
 121             };
 122 
 123             public void close() throws IOException {
 124                 synchronized (SSF.this) {
 125                     serverSocketClosed = true;
 126                     SSF.this.notifyAll();
 127                 }
 128                 System.err.println("- closing server socket: " + this);
 129                 super.close();
 130             }
 131         }
 132     }
 133 }