/*

 * Copyright (c) 2008, 2018, 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 java.lang.invoke;

import jdk.internal.misc.Unsafe;

import jdk.internal.vm.annotation.ForceInline;

import jdk.internal.vm.annotation.Stable;

import sun.invoke.util.ValueConversions;

import sun.invoke.util.VerifyAccess;

import sun.invoke.util.VerifyType;

import sun.invoke.util.Wrapper;

import java.lang.ref.WeakReference;

import java.util.Arrays;

import java.util.Objects;

import static java.lang.invoke.LambdaForm.*;

import static java.lang.invoke.LambdaForm.Kind.*;

import static java.lang.invoke.MethodHandleNatives.Constants.*;

import static java.lang.invoke.MethodHandleStatics.UNSAFE;

import static java.lang.invoke.MethodHandleStatics.newInternalError;

import static java.lang.invoke.MethodTypeForm.*;

/**

 * The flavor of method handle which implements a constant reference

 * to a class member.

 * @author jrose

 */

class DirectMethodHandle extends MethodHandle {

    final MemberName member;

    // Constructors and factory methods in this class *must* be package scoped or private.

    private DirectMethodHandle(MethodType mtype, LambdaForm form, MemberName member) {

        super(mtype, form);

        if (!member.isResolved())  throw new InternalError();

        if (member.getDeclaringClass().isInterface() &&

            // Check for corner case: invokeinterface of Object method

            MemberName m = new MemberName(Object.class, member.getName(), member.getMethodType(), member.getReferenceKind());

            m = MemberName.getFactory().resolveOrNull(m.getReferenceKind(), m, null);

            if (m != null && m.isPublic()) {

                assert(member.getReferenceKind() == m.getReferenceKind());  // else this.form is wrong

                member = m;

            }

        }

        this.member = member;

    }

    // Factory methods:

    static DirectMethodHandle make(byte refKind, Class<?> refc, MemberName member, Class<?> callerClass) {

        MethodType mtype = member.getMethodOrFieldType();

        if (!member.isStatic()) {

            if (!member.getDeclaringClass().isAssignableFrom(refc) || member.isConstructor())

                throw new InternalError(member.toString());

            mtype = mtype.insertParameterTypes(0, refc);

        }

        if (!member.isField()) {

            switch (refKind) {

                case REF_invokeSpecial: {

                    member = member.asSpecial();

                    }

                }

                case REF_invokeInterface: {

                    return new Interface(mtype, lform, member, refc);

                }

                default: {

                    return new DirectMethodHandle(mtype, lform, member);

                }

            }

        } else {

            LambdaForm lform = preparedFieldLambdaForm(member);

            if (member.isStatic()) {

                long offset = MethodHandleNatives.staticFieldOffset(member);

                Object base = MethodHandleNatives.staticFieldBase(member);

                return new StaticAccessor(mtype, lform, member, base, offset);

            } else {

                long offset = MethodHandleNatives.objectFieldOffset(member);

                assert(offset == (int)offset);

                return new Accessor(mtype, lform, member, (int)offset);

            }

        }

    }

    static DirectMethodHandle make(Class<?> refc, MemberName member) {

        byte refKind = member.getReferenceKind();

        if (refKind == REF_invokeSpecial)

            refKind =  REF_invokeVirtual;

        return make(refKind, refc, member, null /* no callerClass context */);

    }

    static DirectMethodHandle make(MemberName member) {

        if (member.isConstructor())

            return makeAllocator(member);

        return make(member.getDeclaringClass(), member);

    }

    private static DirectMethodHandle makeAllocator(MemberName ctor) {

        assert(ctor.isConstructor() && ctor.getName().equals("<init>"));

        Class<?> instanceClass = ctor.getDeclaringClass();

        ctor = ctor.asConstructor();

        assert(ctor.isConstructor() && ctor.getReferenceKind() == REF_newInvokeSpecial) : ctor;

        MethodType mtype = ctor.getMethodType().changeReturnType(instanceClass);

        LambdaForm lform = preparedLambdaForm(ctor);

        MemberName init = ctor.asSpecial();

        assert(init.getMethodType().returnType() == void.class);

        return new Constructor(mtype, lform, ctor, init, instanceClass);

    }

    @Override

    BoundMethodHandle rebind() {

        return BoundMethodHandle.makeReinvoker(this);

    }

    @Override

    MethodHandle copyWith(MethodType mt, LambdaForm lf) {

        assert(this.getClass() == DirectMethodHandle.class);  // must override in subclasses

        return new DirectMethodHandle(mt, lf, member);

    }

    @Override

    String internalProperties() {

        return "\n& DMH.MN="+internalMemberName();

    }

    //// Implementation methods.

    @Override

    @ForceInline

    MemberName internalMemberName() {

        return member;

    }

    private static final MemberName.Factory IMPL_NAMES = MemberName.getFactory();

    /**

     * Create a LF which can invoke the given method.

     * Cache and share this structure among all methods with

     * the same basicType and refKind.

     */

        assert(m.isInvocable()) : m;  // call preparedFieldLambdaForm instead

        MethodType mtype = m.getInvocationType().basicType();

        assert(!m.isMethodHandleInvoke()) : m;

        int which;

        switch (m.getReferenceKind()) {

        case REF_invokeVirtual:    which = LF_INVVIRTUAL;    break;

        case REF_invokeStatic:     which = LF_INVSTATIC;     break;

        case REF_invokeSpecial:    which = LF_INVSPECIAL;    break;

        case REF_invokeInterface:  which = LF_INVINTERFACE;  break;

        case REF_newInvokeSpecial: which = LF_NEWINVSPECIAL; break;

        default:  throw new InternalError(m.toString());

        }

        if (which == LF_INVSTATIC && shouldBeInitialized(m)) {

            // precompute the barrier-free version:

            preparedLambdaForm(mtype, which);

            which = LF_INVSTATIC_INIT;

        }

            which = LF_INVSPECIAL_IFC;

        }

        LambdaForm lform = preparedLambdaForm(mtype, which);

        maybeCompile(lform, m);

        assert(lform.methodType().dropParameterTypes(0, 1)

                .equals(m.getInvocationType().basicType()))

                : Arrays.asList(m, m.getInvocationType().basicType(), lform, lform.methodType());

        return lform;

    }

    private static LambdaForm preparedLambdaForm(MemberName m) {

    }

    private static LambdaForm preparedLambdaForm(MethodType mtype, int which) {

        LambdaForm lform = mtype.form().cachedLambdaForm(which);

        if (lform != null)  return lform;

        lform = makePreparedLambdaForm(mtype, which);

        return mtype.form().setCachedLambdaForm(which, lform);

    }

    static LambdaForm makePreparedLambdaForm(MethodType mtype, int which) {

        boolean needsInit = (which == LF_INVSTATIC_INIT);

        boolean doesAlloc = (which == LF_NEWINVSPECIAL);

        boolean needsReceiverCheck = (which == LF_INVINTERFACE ||

                                      which == LF_INVSPECIAL_IFC);

        String linkerName;

        LambdaForm.Kind kind;

        switch (which) {

        case LF_INVVIRTUAL:    linkerName = "linkToVirtual";   kind = DIRECT_INVOKE_VIRTUAL;     break;

        case LF_INVSTATIC:     linkerName = "linkToStatic";    kind = DIRECT_INVOKE_STATIC;      break;

        case LF_INVSTATIC_INIT:linkerName = "linkToStatic";    kind = DIRECT_INVOKE_STATIC_INIT; break;

        case LF_INVSPECIAL_IFC:linkerName = "linkToSpecial";   kind = DIRECT_INVOKE_SPECIAL_IFC; break;

        case LF_INVSPECIAL:    linkerName = "linkToSpecial";   kind = DIRECT_INVOKE_SPECIAL;     break;

        case LF_INVINTERFACE:  linkerName = "linkToInterface"; kind = DIRECT_INVOKE_INTERFACE;   break;

        case LF_NEWINVSPECIAL: linkerName = "linkToSpecial";   kind = DIRECT_NEW_INVOKE_SPECIAL; break;

        default:  throw new InternalError("which="+which);

        }

        MethodType mtypeWithArg = mtype.appendParameterTypes(MemberName.class);

        if (doesAlloc)

            mtypeWithArg = mtypeWithArg

                    .insertParameterTypes(0, Object.class)  // insert newly allocated obj

                    .changeReturnType(void.class);          // <init> returns void

        MemberName linker = new MemberName(MethodHandle.class, linkerName, mtypeWithArg, REF_invokeStatic);

        try {

            linker = IMPL_NAMES.resolveOrFail(REF_invokeStatic, linker, null, NoSuchMethodException.class);

        } catch (ReflectiveOperationException ex) {

            throw newInternalError(ex);

        }

        final int DMH_THIS    = 0;

        final int ARG_BASE    = 1;

        final int ARG_LIMIT   = ARG_BASE + mtype.parameterCount();

        int nameCursor = ARG_LIMIT;

        final int NEW_OBJ     = (doesAlloc ? nameCursor++ : -1);

        final int GET_MEMBER  = nameCursor++;

        final int CHECK_RECEIVER = (needsReceiverCheck ? nameCursor++ : -1);

        final int LINKER_CALL = nameCursor++;

        Name[] names = arguments(nameCursor - ARG_LIMIT, mtype.invokerType());

        assert(names.length == nameCursor);

        if (doesAlloc) {

            // names = { argx,y,z,... new C, init method }

            names[NEW_OBJ] = new Name(getFunction(NF_allocateInstance), names[DMH_THIS]);

            names[GET_MEMBER] = new Name(getFunction(NF_constructorMethod), names[DMH_THIS]);

        } else if (needsInit) {

            names[GET_MEMBER] = new Name(getFunction(NF_internalMemberNameEnsureInit), names[DMH_THIS]);

        } else {

            names[GET_MEMBER] = new Name(getFunction(NF_internalMemberName), names[DMH_THIS]);

        }

        assert(findDirectMethodHandle(names[GET_MEMBER]) == names[DMH_THIS]);

        Object[] outArgs = Arrays.copyOfRange(names, ARG_BASE, GET_MEMBER+1, Object[].class);

        if (needsReceiverCheck) {

            names[CHECK_RECEIVER] = new Name(getFunction(NF_checkReceiver), names[DMH_THIS], names[ARG_BASE]);

            outArgs[0] = names[CHECK_RECEIVER];

        }

        assert(outArgs[outArgs.length-1] == names[GET_MEMBER]);  // look, shifted args!

        int result = LAST_RESULT;

        if (doesAlloc) {

            assert(outArgs[outArgs.length-2] == names[NEW_OBJ]);  // got to move this one

            System.arraycopy(outArgs, 0, outArgs, 1, outArgs.length-2);

            outArgs[0] = names[NEW_OBJ];

            result = NEW_OBJ;

        }

        names[LINKER_CALL] = new Name(linker, outArgs);

        LambdaForm lform = new LambdaForm(ARG_LIMIT, names, result, kind);

        // This is a tricky bit of code.  Don't send it through the LF interpreter.

        lform.compileToBytecode();

        return lform;

    }

    /* assert */ static Object findDirectMethodHandle(Name name) {

        if (name.function.equals(getFunction(NF_internalMemberName)) ||

            name.function.equals(getFunction(NF_internalMemberNameEnsureInit)) ||

            name.function.equals(getFunction(NF_constructorMethod))) {

            assert(name.arguments.length == 1);

            return name.arguments[0];

        }

        return null;

    }

    private static void maybeCompile(LambdaForm lform, MemberName m) {

        if (lform.vmentry == null && VerifyAccess.isSamePackage(m.getDeclaringClass(), MethodHandle.class))

            // Help along bootstrapping...

            lform.compileToBytecode();

    }

    /** Static wrapper for DirectMethodHandle.internalMemberName. */

    @ForceInline

    /*non-public*/ static Object internalMemberName(Object mh) {

        return ((DirectMethodHandle)mh).member;

    }

    /** Static wrapper for DirectMethodHandle.internalMemberName.

     * This one also forces initialization.

     */

    /*non-public*/ static Object internalMemberNameEnsureInit(Object mh) {

        DirectMethodHandle dmh = (DirectMethodHandle)mh;

        dmh.ensureInitialized();

        return dmh.member;

    }

    /*non-public*/ static

    boolean shouldBeInitialized(MemberName member) {

        switch (member.getReferenceKind()) {

        case REF_invokeStatic:

        case REF_getStatic:

        case REF_putStatic:

        case REF_newInvokeSpecial:

            break;

        default:

            // No need to initialize the class on this kind of member.

            return false;

        }

        Class<?> cls = member.getDeclaringClass();

        if (cls == ValueConversions.class ||

            cls == MethodHandleImpl.class ||

            cls == Invokers.class) {

            // These guys have lots of <clinit> DMH creation but we know

            // the MHs will not be used until the system is booted.

            return false;

        }

        if (VerifyAccess.isSamePackage(MethodHandle.class, cls) ||

            VerifyAccess.isSamePackage(ValueConversions.class, cls)) {

            // It is a system class.  It is probably in the process of

            // being initialized, but we will help it along just to be safe.

            if (UNSAFE.shouldBeInitialized(cls)) {

                UNSAFE.ensureClassInitialized(cls);

            }

            return false;

        }

        return UNSAFE.shouldBeInitialized(cls);

    }

    private static class EnsureInitialized extends ClassValue<WeakReference<Thread>> {

        @Override

        protected WeakReference<Thread> computeValue(Class<?> type) {

            UNSAFE.ensureClassInitialized(type);

            if (UNSAFE.shouldBeInitialized(type))

                // If the previous call didn't block, this can happen.

                // We are executing inside <clinit>.

                return new WeakReference<>(Thread.currentThread());

            return null;

        }

        static final EnsureInitialized INSTANCE = new EnsureInitialized();

    }

    private void ensureInitialized() {

        if (checkInitialized(member)) {

            // The coast is clear.  Delete the <clinit> barrier.

            if (member.isField())

                updateForm(preparedFieldLambdaForm(member));

            else

                updateForm(preparedLambdaForm(member));

        }

    }

    private static boolean checkInitialized(MemberName member) {

        Class<?> defc = member.getDeclaringClass();

        WeakReference<Thread> ref = EnsureInitialized.INSTANCE.get(defc);

        if (ref == null) {

            return true;  // the final state

        }

        Thread clinitThread = ref.get();

        // Somebody may still be running defc.<clinit>.

        if (clinitThread == Thread.currentThread()) {

            // If anybody is running defc.<clinit>, it is this thread.

            if (UNSAFE.shouldBeInitialized(defc))

                // Yes, we are running it; keep the barrier for now.

                return false;

        } else {

            // We are in a random thread.  Block.

            UNSAFE.ensureClassInitialized(defc);

        }

        assert(!UNSAFE.shouldBeInitialized(defc));

        // put it into the final state

        EnsureInitialized.INSTANCE.remove(defc);

        return true;

    }

    /*non-public*/ static void ensureInitialized(Object mh) {

        ((DirectMethodHandle)mh).ensureInitialized();

    }

    /** This subclass represents invokespecial instructions. */

    static class Special extends DirectMethodHandle {

        private final Class<?> caller;

        private Special(MethodType mtype, LambdaForm form, MemberName member, Class<?> caller) {

            super(mtype, form, member);

            this.caller = caller;

        }

        @Override

        boolean isInvokeSpecial() {

            return true;

        }

        @Override

        MethodHandle copyWith(MethodType mt, LambdaForm lf) {

            return new Special(mt, lf, member, caller);

        }

        Object checkReceiver(Object recv) {

            if (!caller.isInstance(recv)) {

                String msg = String.format("Receiver class %s is not a subclass of caller class %s",

                                           recv.getClass().getName(), caller.getName());

                throw new IncompatibleClassChangeError(msg);

            }

            return recv;

        }

    }

    /** This subclass represents invokeinterface instructions. */

    static class Interface extends DirectMethodHandle {

        private final Class<?> refc;

        private Interface(MethodType mtype, LambdaForm form, MemberName member, Class<?> refc) {

            super(mtype, form, member);

            assert refc.isInterface() : refc;

            this.refc = refc;

        }

        @Override

        MethodHandle copyWith(MethodType mt, LambdaForm lf) {

            return new Interface(mt, lf, member, refc);

        }

        @Override

        Object checkReceiver(Object recv) {

            if (!refc.isInstance(recv)) {

                String msg = String.format("Receiver class %s does not implement the requested interface %s",

                                           recv.getClass().getName(), refc.getName());

                throw new IncompatibleClassChangeError(msg);

            }

            return recv;

        }

    }

    /** Used for interface receiver type checks, by Interface and Special modes. */

    Object checkReceiver(Object recv) {

        throw new InternalError("Should only be invoked on a subclass");

    }

    /** This subclass handles constructor references. */

    static class Constructor extends DirectMethodHandle {

        final MemberName initMethod;

        final Class<?>   instanceClass;

        private Constructor(MethodType mtype, LambdaForm form, MemberName constructor,

                            MemberName initMethod, Class<?> instanceClass) {

            super(mtype, form, constructor);

            this.initMethod = initMethod;

            this.instanceClass = instanceClass;

            assert(initMethod.isResolved());

        }

        @Override

        MethodHandle copyWith(MethodType mt, LambdaForm lf) {

            return new Constructor(mt, lf, member, initMethod, instanceClass);

        }

    }

    /*non-public*/ static Object constructorMethod(Object mh) {

        Constructor dmh = (Constructor)mh;

        return dmh.initMethod;

    }

    /*non-public*/ static Object allocateInstance(Object mh) throws InstantiationException {

        Constructor dmh = (Constructor)mh;

        return UNSAFE.allocateInstance(dmh.instanceClass);

    }

    /** This subclass handles non-static field references. */

    static class Accessor extends DirectMethodHandle {

        final Class<?> fieldType;

        final int      fieldOffset;

        private Accessor(MethodType mtype, LambdaForm form, MemberName member,