/*

 * Copyright 1998-2007 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 4155894

 *

 * @summary remote references can't be downloaded

 * @author Ann Wollrath

 *

 * test currently needs RMISecurityManager because of

 * 4180392

 *

 * @library ../../testlibrary

 * @build UseCustomRef

 * @build Ping

 * @build UseCustomRef_Stub

 * @build UseCustomRef_Skel

 * @run main/othervm/policy=security.policy/secure=java.rmi.RMISecurityManager/timeout=120 UseCustomRef

 *

 * This test was failing to run because the synthetic access

 * control context used by the application class loader to find and define

 * CustomServerRef does not have accessClassInPackage.sun.rmi.server runtime

 * permission necessary to load its superclass sun.rmi.server.UnicastServerRef,

 * even though this test's code is granted that permission in its policy file.

 * That bug number is 4256530

 */

import java.io.*;

import java.rmi.*;

import java.rmi.server.*;

import java.rmi.registry.*;

import sun.rmi.transport.LiveRef;

public class UseCustomRef

        extends RemoteServer

        implements Ping

{

    public UseCustomRef() throws RemoteException {

        exportObject();

    }

    public void exportObject() throws RemoteException {

        ref = new CustomServerRef(new LiveRef(0));

        ((ServerRef) ref).exportObject(this, null);

    }

    public RemoteRef getRef() { return ref; }

    public void ping() {}

    public void receiveAndPing(Ping p) throws RemoteException {

        p.ping();

    }

    public static void main(String[] args) {

        Ping obj = null;

        Registry registry = null;

        try {

            /*

             * create registry

             */

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

            System.err.println("creating Registry...");

            registry = LocateRegistry.createRegistry(TestLibrary.REGISTRY_PORT);

            /*

             * create object with custom ref and bind in registry

             */

            System.err.println("creating UseCustomRef...");

            UseCustomRef cr = new UseCustomRef();

            RemoteRef ref = cr.getRef();

            if (!(ref instanceof CustomServerRef)) {

                TestLibrary.bomb("test failed: reference not " +

                                "instanceof CustomServerRef");

            }

            String name = "//:" + TestLibrary.REGISTRY_PORT + "/UseCustomRef";

            //      String name = "UseCustomRef";

            System.err.println("binding object in registry...");

            Naming.rebind(name, cr);

            /*

             * look up object and invoke its ping method

             */

            System.err.println("ping object...");

            obj = (Ping) Naming.lookup(name);

            obj.ping();

            /*

             * pass object with custom ref in remote call

             */

            System.err.println("pass object in remote call...");

            obj.receiveAndPing(cr);

            /*

             * write remote object with custom ref to output stream

             */

            System.err.println("writing remote object to stream...");

            ByteArrayOutputStream bout = new ByteArrayOutputStream();

            ObjectOutputStream out = new ObjectOutputStream(bout);

            out.writeObject(cr);

            out.flush();

            out.close();

            /*

             * read back remote object from output stream

             */

            System.err.println("reading remote object from stream...");

            ObjectInputStream in = new ObjectInputStream(

                new ByteArrayInputStream(bout.toByteArray()));

            cr = (UseCustomRef) in.readObject();

            /*

             * re-export object and ping

             */

            System.err.println("re-export object read...");

            cr.exportObject();

            System.err.println("look up object again...");

            Naming.rebind(name, cr);

            System.err.println("ping object read...");

            obj = (Ping) Naming.lookup(name);

            obj.ping();

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

            Naming.unbind(name);

            cr = null;

        } catch (Exception e) {

            TestLibrary.bomb("test failed with exception: ", e);

        } finally {

            TestLibrary.unexport(obj);

            TestLibrary.unexport(registry);

            registry = null;

            obj = null;

        }

    }

    public static class CustomServerRef

        extends sun.rmi.server.UnicastServerRef

    {

        public CustomServerRef() {}

        public CustomServerRef(LiveRef ref) {

            super(ref);

        }

        /*****

        public CustomServerRef(int port,

                               RMIClientSocketFactory csf,

                               RMIServerSocketFactory ssf)

        {

            super (new LiveRef(port, csf, ssf));

        }

        *****/

        public String getRefClass(ObjectOutput out) {

            return "";

        }

        protected void unmarshalCustomCallData(ObjectInput in)

            throws IOException, ClassNotFoundException

        {

            System.err.println("unmarshalling call data...");

            String s = (String) (in.readObject());

            System.err.println(s);

        }

        protected RemoteRef getClientRef() {

            return new CustomRef(ref);

        }

    }

    public static class CustomRef extends sun.rmi.server.UnicastRef {

        public CustomRef() {

        }

        public CustomRef(sun.rmi.transport.LiveRef ref) {

            super(ref);

        }

        protected void marshalCustomCallData(ObjectOutput out)

            throws IOException

        {

            // this custom data ensures that a custom server

            // ref has written the relevant information.

            System.err.println("marshalling call data...");

            out.writeObject("hello there.");

        }

        public String getRefClass(ObjectOutput out) {

            return "";

        }

    }

}