/*

 * Copyright (c) 2017, 2019, Oracle and/or its affiliates. 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.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package sun.rmi.transport;

import sun.rmi.transport.tcp.TCPConnection;

import java.io.IOException;

import java.rmi.dgc.Lease;

import java.rmi.dgc.VMID;

import java.rmi.server.UID;

import java.util.ArrayList;

import sun.misc.ObjectInputFilter;

/**

 * Stubs to invoke DGC remote methods.

 * Originally generated from RMIC but frozen to insert serialFilter.

 */

@SuppressWarnings({"deprecation", "serial"})

public final class DGCImpl_Stub

        extends java.rmi.server.RemoteStub

        implements java.rmi.dgc.DGC {

    private static final java.rmi.server.Operation[] operations = {

            new java.rmi.server.Operation("void clean(java.rmi.server.ObjID[], long, java.rmi.dgc.VMID, boolean)"),

            new java.rmi.server.Operation("java.rmi.dgc.Lease dirty(java.rmi.server.ObjID[], long, java.rmi.dgc.Lease)")

    };

    private static final long interfaceHash = -669196253586618813L;

    /** Registry max depth of remote invocations. **/

    private static int DGCCLIENT_MAX_DEPTH = 6;

    /** Registry maximum array size in remote invocations. **/

    private static int DGCCLIENT_MAX_ARRAY_SIZE = 10000;

    // constructors

    public DGCImpl_Stub() {

        super();

    }

    public DGCImpl_Stub(java.rmi.server.RemoteRef ref) {

        super(ref);

    }

    // methods from remote interfaces

    // implementation of clean(ObjID[], long, VMID, boolean)

    public void clean(java.rmi.server.ObjID[] $param_arrayOf_ObjID_1, long $param_long_2, java.rmi.dgc.VMID $param_VMID_3, boolean $param_boolean_4)

            throws java.rmi.RemoteException {

        try {

            StreamRemoteCall call = (StreamRemoteCall)ref.newCall((java.rmi.server.RemoteObject) this,

                    operations, 0, interfaceHash);

            call.setObjectInputFilter(DGCImpl_Stub::leaseFilter);

            try {

                java.io.ObjectOutput out = call.getOutputStream();

                out.writeObject($param_arrayOf_ObjID_1);

                out.writeLong($param_long_2);

                out.writeObject($param_VMID_3);

                out.writeBoolean($param_boolean_4);

            } catch (java.io.IOException e) {

                throw new java.rmi.MarshalException("error marshalling arguments", e);

            }

            ref.invoke(call);

            ref.done(call);

        } catch (java.lang.RuntimeException e) {

            throw e;

        } catch (java.rmi.RemoteException e) {

            throw e;

        } catch (java.lang.Exception e) {

            throw new java.rmi.UnexpectedException("undeclared checked exception", e);

        }

    }

    // implementation of dirty(ObjID[], long, Lease)

    public java.rmi.dgc.Lease dirty(java.rmi.server.ObjID[] $param_arrayOf_ObjID_1, long $param_long_2, java.rmi.dgc.Lease $param_Lease_3)

            throws java.rmi.RemoteException {

        try {

            StreamRemoteCall call =

                    (StreamRemoteCall)ref.newCall((java.rmi.server.RemoteObject) this,

                            operations, 1, interfaceHash);

            call.setObjectInputFilter(DGCImpl_Stub::leaseFilter);

            try {

                java.io.ObjectOutput out = call.getOutputStream();

                out.writeObject($param_arrayOf_ObjID_1);

                out.writeLong($param_long_2);

                out.writeObject($param_Lease_3);

            } catch (java.io.IOException e) {

                throw new java.rmi.MarshalException("error marshalling arguments", e);

            }

            ref.invoke(call);

            java.rmi.dgc.Lease $result;

            Connection connection = call.getConnection();

            try {

                java.io.ObjectInput in = call.getInputStream();

                $result = (java.rmi.dgc.Lease) in.readObject();

            } catch (ClassCastException | IOException | ClassNotFoundException e) {

                if (connection instanceof TCPConnection) {

                    // Modified to prevent re-use of the connection after an exception

                    ((TCPConnection) connection).getChannel().free(connection, false);

                }

                call.discardPendingRefs();

                throw new java.rmi.UnmarshalException("error unmarshalling return", e);

            } finally {

                ref.done(call);

            }

            return $result;

        } catch (java.lang.RuntimeException e) {

            throw e;

        } catch (java.rmi.RemoteException e) {

            throw e;

        } catch (java.lang.Exception e) {

            throw new java.rmi.UnexpectedException("undeclared checked exception", e);

        }

    }

    /**

     * ObjectInputFilter to filter DGCClient return value (a Lease).

     * The list of acceptable classes is very short and explicit.

     * The depth and array sizes are limited.

     * <p>

     * The filter must accept normal and exception returns.

     * A DGC server may throw exceptions that may have a cause

     * and suppressed exceptions.

     *

     * @param filterInfo access to class, arrayLength, etc.

     * @return  {@link ObjectInputFilter.Status#ALLOWED} if allowed,

     *          {@link ObjectInputFilter.Status#REJECTED} if rejected,

     *          otherwise {@link ObjectInputFilter.Status#UNDECIDED}

     */

    private static ObjectInputFilter.Status leaseFilter(ObjectInputFilter.FilterInfo filterInfo) {

        if (filterInfo.depth() > DGCCLIENT_MAX_DEPTH) {

            return ObjectInputFilter.Status.REJECTED;

        }

        Class<?> clazz = filterInfo.serialClass();

        if (clazz != null) {

            while (clazz.isArray()) {

                if (filterInfo.arrayLength() >= 0 && filterInfo.arrayLength() > DGCCLIENT_MAX_ARRAY_SIZE) {

                    return ObjectInputFilter.Status.REJECTED;

                }

                // Arrays are allowed depending on the component type

                clazz = clazz.getComponentType();

            }

            if (clazz.isPrimitive()) {

                // Arrays of primitives are allowed

                return ObjectInputFilter.Status.ALLOWED;

            }

            return (clazz == UID.class ||

                    clazz == VMID.class ||

                    clazz == Lease.class ||

                    (Throwable.class.isAssignableFrom(clazz) &&

                            clazz.getClassLoader() ==

                            Object.class.getClassLoader()) ||

                    clazz == StackTraceElement.class ||

                    clazz == ArrayList.class ||     // for suppressed exceptions, if any

                    clazz == Object.class ||

                    clazz.getName().equals("java.util.Collections$UnmodifiableList") ||

                    clazz.getName().equals("java.util.Collections$UnmodifiableCollection") ||

                    clazz.getName().equals("java.util.Collections$UnmodifiableRandomAccessList"))

                    ? ObjectInputFilter.Status.ALLOWED

                    : ObjectInputFilter.Status.REJECTED;

        }

        // Not a class, not size limited

        return ObjectInputFilter.Status.UNDECIDED;

    }

}