/* 
 * Copyright 2005 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/* @test
 * @bug 4924577
 * @summary When the current RMIFailureHandler (if any) is invoked
 * because of a server socket accept failure, if it returns false,
 * then (in addition to the accept loop terminating) the associated
 * server socket should be closed.  The server socket should also be
 * closed if the accept loop terminates because of an unexpected
 * exception for which it doesn't even consult the RMIFailureHandler.
 * @author Peter Jones
 *
 * @build CloseServerSocketOnTermination
 * @run main/othervm CloseServerSocketOnTermination
 */

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.rmi.Remote;
import java.rmi.server.RMIFailureHandler;
import java.rmi.server.RMIServerSocketFactory;
import java.rmi.server.RMISocketFactory;
import java.rmi.server.UnicastRemoteObject;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

public class CloseServerSocketOnTermination {

    private static long TIMEOUT = 5000;

    public static void main(String[] args) throws Exception {
	System.err.println("\nRegression test for bug 4924577\n");

	RMISocketFactory.setFailureHandler(new RMIFailureHandler() {
	    public boolean failure(Exception e) { return false; }
	});

	tryWith(new IOException());
	tryWith(new NullPointerException());
	tryWith(new OutOfMemoryError());
	tryWith(new NoClassDefFoundError());
	tryWith(new InternalError());
	tryWith(new Throwable());

	System.err.println("TEST PASSED");
    }

    private static void tryWith(Throwable t) throws Exception {
	Remote impl = new Remote() { };
	try {
	    CountDownLatch latch = new CountDownLatch(1);
	    UnicastRemoteObject.exportObject(impl, 0, null, new SSF(t, latch));
	    if (!latch.await(TIMEOUT, TimeUnit.MILLISECONDS)) {
		throw new Error("server socket not closed");
	    }
	} finally {
	    UnicastRemoteObject.unexportObject(impl, true);
	}
    }

    private static class SSF implements RMIServerSocketFactory {
	private final Throwable acceptFailure;
	private final CountDownLatch closedLatch;
	SSF(Throwable acceptFailure, CountDownLatch closedLatch) {
	    this.acceptFailure = acceptFailure;
	    this.closedLatch = closedLatch;
	}
	public ServerSocket createServerSocket(int port) throws IOException {
	    return new ServerSocket(port) {
		private int acceptInvocations = 0;
		public synchronized Socket accept() throws IOException {
		    if (acceptInvocations++ == 0) {
			throwException(acceptFailure);
		    }
		    return super.accept();
		}
		public void close() throws IOException {
		    closedLatch.countDown();
		    super.close();
		}
	    };
	}

	// hack to throw an arbitrary (possibly checked) Throwable
	private static void throwException(Throwable t) {
	    try {
		toThrow.set(t);
		Thrower.class.newInstance();
	    } catch (IllegalAccessException e) {
		throw new AssertionError();
	    } catch (InstantiationException e) {
		throw new AssertionError();
	    } finally {
		toThrow.remove();
	    }
	}
	private static ThreadLocal<Throwable> toThrow =
	    new ThreadLocal<Throwable>();
	private static class Thrower {
	    Thrower() throws Throwable { throw toThrow.get(); }
	}
    }
}