/* 

 * Copyright 1998-2000 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 4134233

 * @bug 4213186

 *

 * @summary synopsis: ActivationSystem.unregisterGroup should unregister objects in group

 * @author Ann Wollrath

 *

 * @library ../../../testlibrary

 * @build TestLibrary RMID JavaVM StreamPipe

 * @build ActivateMe CallbackInterface

 * @build UnregisterGroup

 * @build UnregisterGroup_Stub

 * @build Callback_Stub

 * @run main/othervm/policy=security.policy/timeout=480 UnregisterGroup

 */

import java.io.*;

import java.rmi.*;

import java.rmi.activation.*;

import java.rmi.server.*;

import java.rmi.registry.*;

import java.util.Properties;

class Callback extends UnicastRemoteObject implements CallbackInterface {

  public static int num_deactivated = 0;

  public Callback() throws RemoteException { super(); }

  public void inc() throws RemoteException {

    incNumDeactivated();

  }

  public synchronized int getNumDeactivated() throws RemoteException {

    return(num_deactivated);

  }

  public synchronized void incNumDeactivated() {

    num_deactivated++;

  }

}

public class UnregisterGroup

	extends Activatable

	implements ActivateMe, Runnable

{

    private static Exception exception = null;

    private static String error = null;

    private static boolean done = false;

    private static ActivateMe lastResortExitObj = null;

    private static final int NUM_OBJECTS = 10;

    private static int PORT = 2006;

    public UnregisterGroup(ActivationID id, MarshalledObject mobj)

	throws Exception

    {

	super(id, 0);

    }

    public void ping()

    {}

    public void unregister() throws Exception {

	super.unregister(super.getID());

    }

    /**

     * Spawns a thread to deactivate the object.

     */

    public void shutdown() throws Exception {

	(new Thread(this,"UnregisterGroup")).start();

    }

    /**

     * To support exiting of group VM as a last resort

     */

    public void justGoAway() {

	System.exit(0);

    }

    /**

     * Thread to deactivate object. First attempts to make object

     * inactive (via the inactive method).  If that fails (the

     * object may still have pending/executing calls), then

     * unexport the object forcibly.

     */

    public void run() {

	ActivationLibrary.deactivate(this, getID());

	System.err.println("\tActivationLibrary.deactivate returned");

	try {

	    CallbackInterface cobj =

		(CallbackInterface)Naming.lookup("//:" + PORT + "/Callback");

	    cobj.inc();

	} catch (Exception e) {

	    System.err.println("cobj.inc exception");

	    e.printStackTrace();

	}

    }

    public static void main(String[] args) {

	Registry registry;

 	System.err.println("\nRegression test for bug 4134233\n");

	TestLibrary.suggestSecurityManager("java.rmi.RMISecurityManager");		

	RMID rmid = null;

	try {

	    RMID.removeLog();

	    rmid = RMID.createRMID();

	    rmid.start();

	    /* Cause activation groups to have a security policy that will

	     * allow security managers to be downloaded and installed

	     */

	    final Properties p = new Properties();

	    // this test must always set policies/managers in its

	    // activation groups

	    p.put("java.security.policy", 

		  TestParams.defaultGroupPolicy);

	    p.put("java.security.manager", 

		  TestParams.defaultSecurityManager);

	    //final int NUM_OBJECTS = 10;

	    Thread t = new Thread() {

		public void run () {

		    try {

			System.err.println("Creating group descriptor");

			ActivationGroupDesc groupDesc =

			    new ActivationGroupDesc(p, null);

			ActivationSystem system = ActivationGroup.getSystem();

			ActivationGroupID groupID =

			    system.registerGroup(groupDesc);

			ActivateMe[] obj = new ActivateMe[NUM_OBJECTS];

			for (int i = 0; i < NUM_OBJECTS; i++) {

			    System.err.println("Creating descriptor: " + i);

			    ActivationDesc desc =

				new ActivationDesc(groupID, "UnregisterGroup",

						   null, null);

			    System.err.println("Registering descriptor: " + i);

			    obj[i] = (ActivateMe) Activatable.register(desc);

			    System.err.println("Activating object: " + i);

			    obj[i].ping();

			}

			lastResortExitObj = obj[0];

			System.err.println("Unregistering group");

			system.unregisterGroup(groupID);

			try {

			    System.err.println("Get the group descriptor");

			    system.getActivationGroupDesc(groupID);

			    error = "test failed: group still registered";

			} catch (UnknownGroupException e) {

			    System.err.println("Test passed: " +

					       "group unregistered");

			}

			/*

			 * Deactivate objects so group VM will exit.

			 */

			for (int i = 0; i < NUM_OBJECTS; i++) {

			    System.err.println("Deactivating object: " + i);

			    obj[i].shutdown();

			    obj[i] = null;

			}

			lastResortExitObj = null;

		    } catch (Exception e) {

			exception = e;

		    }

		    done = true;

		}

	    };

	    t.start();

	    t.join(120000);

	    if (exception != null) {

		TestLibrary.bomb("test failed", exception);

	    } else if (error != null) {

		TestLibrary.bomb(error, null);

	    } else if (!done) {

		TestLibrary.bomb("test failed: not completed before timeout", null);

	    } else {

		System.err.println("Test passed");

	    }

	} catch (Exception e) {

	    TestLibrary.bomb("test failed", e);

	} finally {

	    if (lastResortExitObj != null) {

		try {

		    lastResortExitObj.justGoAway();

		} catch (Exception munch) {

		}

	    }

	    // Wait for the object deactivation to take place first

	    try {

 		// create reg and export callback object

 		registry = LocateRegistry.createRegistry(PORT);

 		Callback robj = new Callback();

 		registry.bind("Callback", robj);

 		//get the callback object 

 		int maxwait=30;

 		int nd = robj.getNumDeactivated();

 		while ((nd < NUM_OBJECTS) && (maxwait> 0)) {

		    System.err.println("num_deactivated="+nd);

		    try {

			Thread.sleep(1000);

		    } catch (InterruptedException ie) {}

		    maxwait--;

		    nd = robj.getNumDeactivated();

 		}

	    } catch (Exception ce) {

 		System.err.println("E:"+ce);

 		ce.printStackTrace();

	    }

	    ActivationLibrary.rmidCleanup(rmid);

	}

    }

}