/*

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

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

 * questions.

 */

package jdk.vm.ci.hotspot;

import static jdk.vm.ci.hotspot.CompilerToVM.compilerToVM;

import static jdk.vm.ci.hotspot.HotSpotJVMCIRuntime.runtime;

import static jdk.vm.ci.hotspot.HotSpotResolvedJavaMethod.Options.UseProfilingInformation;

import static jdk.vm.ci.hotspot.HotSpotVMConfig.config;

import static jdk.vm.ci.hotspot.UnsafeAccess.UNSAFE;

import java.lang.annotation.Annotation;

import java.lang.reflect.Executable;

import java.lang.reflect.InvocationTargetException;

import java.lang.reflect.Method;

import java.lang.reflect.Modifier;

import java.lang.reflect.Type;

import java.util.HashMap;

import java.util.Map;

import jdk.vm.ci.common.JVMCIError;

import jdk.vm.ci.meta.Constant;

import jdk.vm.ci.meta.ConstantPool;

import jdk.vm.ci.meta.DefaultProfilingInfo;

import jdk.vm.ci.meta.ExceptionHandler;

import jdk.vm.ci.meta.JavaConstant;

import jdk.vm.ci.meta.JavaMethod;

import jdk.vm.ci.meta.JavaType;

import jdk.vm.ci.meta.LineNumberTable;

import jdk.vm.ci.meta.LineNumberTableImpl;

import jdk.vm.ci.meta.Local;

import jdk.vm.ci.meta.LocalImpl;

import jdk.vm.ci.meta.LocalVariableTable;

import jdk.vm.ci.meta.LocalVariableTableImpl;

import jdk.vm.ci.meta.ModifiersProvider;

import jdk.vm.ci.meta.ProfilingInfo;

import jdk.vm.ci.meta.ResolvedJavaMethod;

import jdk.vm.ci.meta.ResolvedJavaType;

import jdk.vm.ci.meta.Signature;

import jdk.vm.ci.meta.SpeculationLog;

import jdk.vm.ci.meta.TriState;

/**

 * Implementation of {@link JavaMethod} for resolved HotSpot methods.

 */

final class HotSpotResolvedJavaMethodImpl extends HotSpotMethod implements HotSpotResolvedJavaMethod, HotSpotProxified, MetaspaceWrapperObject {

    /**

     * Reference to metaspace Method object.

     */

    private final long metaspaceMethod;

    private final HotSpotResolvedObjectTypeImpl holder;

    private final HotSpotConstantPool constantPool;

    private final HotSpotSignature signature;

    private HotSpotMethodData methodData;

    private byte[] code;

    private Executable toJavaCache;

    /**

     * Gets the holder of a HotSpot metaspace method native object.

     *

     * @param metaspaceMethod a metaspace Method object

     * @return the {@link ResolvedJavaType} corresponding to the holder of the

     *         {@code metaspaceMethod}

     */

    private static HotSpotResolvedObjectTypeImpl getHolder(long metaspaceMethod) {

        HotSpotVMConfig config = config();

        final long metaspaceConstMethod = UNSAFE.getAddress(metaspaceMethod + config.methodConstMethodOffset);

        final long metaspaceConstantPool = UNSAFE.getAddress(metaspaceConstMethod + config.constMethodConstantsOffset);

        return compilerToVM().getResolvedJavaType(null, metaspaceConstantPool + config.constantPoolHolderOffset, false);

    }

    /**

     * Gets the JVMCI mirror from a HotSpot method. The VM is responsible for ensuring that the

     * Method* is kept alive for the duration of this call and the

     * {@link HotSpotJVMCIMetaAccessContext} keeps it alive after that.

     *

     * Called from the VM.

     *

     * @param metaspaceMethod a metaspace Method object

     * @return the {@link ResolvedJavaMethod} corresponding to {@code metaspaceMethod}

     */

    @SuppressWarnings("unused")

    private static HotSpotResolvedJavaMethod fromMetaspace(long metaspaceMethod) {

        HotSpotResolvedObjectTypeImpl holder = getHolder(metaspaceMethod);

        return holder.createMethod(metaspaceMethod);

    }

    HotSpotResolvedJavaMethodImpl(HotSpotResolvedObjectTypeImpl holder, long metaspaceMethod) {

        // It would be too much work to get the method name here so we fill it in later.

        super(null);

        this.metaspaceMethod = metaspaceMethod;

        this.holder = holder;

        HotSpotVMConfig config = config();

        final long constMethod = getConstMethod();

        /*

         * Get the constant pool from the metaspace method. Some methods (e.g. intrinsics for

         * signature-polymorphic method handle methods) have their own constant pool instead of the

         * one from their holder.

         */

        final long metaspaceConstantPool = UNSAFE.getAddress(constMethod + config.constMethodConstantsOffset);

        if (metaspaceConstantPool == holder.getConstantPool().getMetaspaceConstantPool()) {

            this.constantPool = holder.getConstantPool();

        } else {

            this.constantPool = compilerToVM().getConstantPool(null, constMethod + config.constMethodConstantsOffset);

        }

        final int nameIndex = UNSAFE.getChar(constMethod + config.constMethodNameIndexOffset);

        this.name = constantPool.lookupUtf8(nameIndex);

        final int signatureIndex = UNSAFE.getChar(constMethod + config.constMethodSignatureIndexOffset);

        this.signature = (HotSpotSignature) constantPool.lookupSignature(signatureIndex);

    }

    /**

     * Returns a pointer to this method's constant method data structure (

     * {@code Method::_constMethod}). This pointer isn't wrapped since it should be safe to use it

     * within the context of this HotSpotResolvedJavaMethodImpl since the Method* and ConstMethod*

     * are kept alive as a pair.

     *

     * @return pointer to this method's ConstMethod

     */

    private long getConstMethod() {

        assert metaspaceMethod != 0;

        return UNSAFE.getAddress(metaspaceMethod + config().methodConstMethodOffset);

    }

    @Override

    public boolean equals(Object obj) {

        if (this == obj) {

            return true;

        }

        if (obj instanceof HotSpotResolvedJavaMethodImpl) {

            HotSpotResolvedJavaMethodImpl that = (HotSpotResolvedJavaMethodImpl) obj;

            return that.metaspaceMethod == metaspaceMethod;

        }

        return false;

    }

    @Override

    public int hashCode() {

        return (int) metaspaceMethod;

    }

    /**

     * Returns this method's flags ({@code Method::_flags}).

     *

     * @return flags of this method

     */

    private int getFlags() {

    }

    /**

     * Returns this method's constant method flags ({@code ConstMethod::_flags}).

     *

     * @return flags of this method's ConstMethod

     */

    private int getConstMethodFlags() {

        return UNSAFE.getChar(getConstMethod() + config().constMethodFlagsOffset);

    }

    @Override

    public HotSpotResolvedObjectTypeImpl getDeclaringClass() {

        return holder;

    }

    /**

     * Gets the address of the C++ Method object for this method.

     */

    public Constant getMetaspaceMethodConstant() {

        return HotSpotMetaspaceConstantImpl.forMetaspaceObject(this, false);

    }

    public long getMetaspacePointer() {

        return metaspaceMethod;

    }

    @Override

    public Constant getEncoding() {

        return getMetaspaceMethodConstant();

    }

    /**

     * Gets the complete set of modifiers for this method which includes the JVM specification

     * modifiers as well as the HotSpot internal modifiers.

     */

    public int getAllModifiers() {

        return UNSAFE.getInt(metaspaceMethod + config().methodAccessFlagsOffset);

    }

    @Override

    public int getModifiers() {

        return getAllModifiers() & ModifiersProvider.jvmMethodModifiers();

    }

    @Override

    public boolean canBeStaticallyBound() {

        return (isFinal() || isPrivate() || isStatic() || holder.isLeaf()) && isConcrete();

    }

    @Override

    public byte[] getCode() {

        if (getCodeSize() == 0) {

            return null;

        }

        if (code == null && holder.isLinked()) {

            code = compilerToVM().getBytecode(this);

            assert code.length == getCodeSize() : "expected: " + getCodeSize() + ", actual: " + code.length;

        }

        return code;

    }

    @Override

    public int getCodeSize() {

        return UNSAFE.getChar(getConstMethod() + config().constMethodCodeSizeOffset);

    }

    @Override

    public ExceptionHandler[] getExceptionHandlers() {

        final boolean hasExceptionTable = (getConstMethodFlags() & config().constMethodHasExceptionTable) != 0;

        if (!hasExceptionTable) {

            return new ExceptionHandler[0];

        }

        HotSpotVMConfig config = config();

        final int exceptionTableLength = compilerToVM().getExceptionTableLength(this);

        ExceptionHandler[] handlers = new ExceptionHandler[exceptionTableLength];

        long exceptionTableElement = compilerToVM().getExceptionTableStart(this);

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

            final int startPc = UNSAFE.getChar(exceptionTableElement + config.exceptionTableElementStartPcOffset);

            final int endPc = UNSAFE.getChar(exceptionTableElement + config.exceptionTableElementEndPcOffset);

            final int handlerPc = UNSAFE.getChar(exceptionTableElement + config.exceptionTableElementHandlerPcOffset);

            int catchTypeIndex = UNSAFE.getChar(exceptionTableElement + config.exceptionTableElementCatchTypeIndexOffset);

            JavaType catchType;

            if (catchTypeIndex == 0) {

                catchType = null;

            } else {

                final int opcode = -1;  // opcode is not used

                catchType = constantPool.lookupType(catchTypeIndex, opcode);

                // Check for Throwable which catches everything.

                if (catchType instanceof HotSpotResolvedObjectTypeImpl) {

                    HotSpotResolvedObjectTypeImpl resolvedType = (HotSpotResolvedObjectTypeImpl) catchType;

                    if (resolvedType.mirror() == Throwable.class) {

                        catchTypeIndex = 0;

                        catchType = null;

                    }

                }

            }

            handlers[i] = new ExceptionHandler(startPc, endPc, handlerPc, catchTypeIndex, catchType);

            // Go to the next ExceptionTableElement

            exceptionTableElement += config.exceptionTableElementSize;

        }

        return handlers;

    }

    /**

     * Returns true if this method has a {@code CallerSensitive} annotation.

     *

     * @return true if CallerSensitive annotation present, false otherwise

     */

    public boolean isCallerSensitive() {

        return (getFlags() & config().methodFlagsCallerSensitive) != 0;

    }

    /**

     * Returns true if this method has a {@code ForceInline} annotation.

     *

     * @return true if ForceInline annotation present, false otherwise

     */

    public boolean isForceInline() {

        return (getFlags() & config().methodFlagsForceInline) != 0;

    }

    /**

     * Returns true if this method has a {@code DontInline} annotation.

     *

     * @return true if DontInline annotation present, false otherwise

     */

    public boolean isDontInline() {

        return (getFlags() & config().methodFlagsDontInline) != 0;

    }

    /**

     * Manually adds a DontInline annotation to this method.

     */

    public void setNotInlineable() {

        compilerToVM().doNotInlineOrCompile(this);

    }

    /**

     * Returns true if this method is one of the special methods that is ignored by security stack

     * walks.

     *

     * @return true if special method ignored by security stack walks, false otherwise

     */

    public boolean ignoredBySecurityStackWalk() {

        return compilerToVM().methodIsIgnoredBySecurityStackWalk(this);

    }

    @Override

    public boolean isClassInitializer() {

        return "<clinit>".equals(name) && isStatic();

    }

    @Override

    public boolean isConstructor() {

        return "<init>".equals(name) && !isStatic();

    }

    @Override

    public int getMaxLocals() {

        if (isAbstract() || isNative()) {

            return 0;

        }

        HotSpotVMConfig config = config();

        return UNSAFE.getChar(getConstMethod() + config.methodMaxLocalsOffset);

    }

    @Override

    public int getMaxStackSize() {

        if (isAbstract() || isNative()) {

            return 0;

        }

        HotSpotVMConfig config = config();

        return config.extraStackEntries + UNSAFE.getChar(getConstMethod() + config.constMethodMaxStackOffset);

    }

    @Override

    public StackTraceElement asStackTraceElement(int bci) {

        if (bci < 0 || bci >= getCodeSize()) {

            // HotSpot code can only construct stack trace elements for valid bcis

            StackTraceElement ste = compilerToVM().getStackTraceElement(this, 0);

            return new StackTraceElement(ste.getClassName(), ste.getMethodName(), ste.getFileName(), -1);

        }

        return compilerToVM().getStackTraceElement(this, bci);

    }

    public ResolvedJavaMethod uniqueConcreteMethod(HotSpotResolvedObjectType receiver) {

        if (receiver.isInterface()) {

            // Cannot trust interfaces. Because of:

            // interface I { void foo(); }

            // class A { public void foo() {} }

            // class B extends A implements I { }

            // class C extends B { public void foo() { } }

            // class D extends B { }

            // Would lead to identify C.foo() as the unique concrete method for I.foo() without

            // seeing A.foo().

            return null;

        }

        if (this.isDefault()) {

            // CHA for default methods doesn't work and may crash the VM

            return null;

        }

        return compilerToVM().findUniqueConcreteMethod(((HotSpotResolvedObjectTypeImpl) receiver), this);

    }

    @Override

    public HotSpotSignature getSignature() {

        return signature;

    }

    /**

     * Gets the value of {@code Method::_code}.

     *

     * @return the value of {@code Method::_code}

     */

    private long getCompiledCode() {

        HotSpotVMConfig config = config();

        return UNSAFE.getAddress(metaspaceMethod + config.methodCodeOffset);

    }

    /**

     * Returns whether this method has compiled code.

     *

     * @return true if this method has compiled code, false otherwise

     */

    public boolean hasCompiledCode() {

        return getCompiledCode() != 0L;

    }

    /**

     * @param level

     * @return true if the currently installed code was generated at {@code level}.

     */

    public boolean hasCompiledCodeAtLevel(int level) {

        long compiledCode = getCompiledCode();

        if (compiledCode != 0) {

            return UNSAFE.getInt(compiledCode + config().nmethodCompLevelOffset) == level;

        }

        return false;

    }

    private static final String TraceMethodDataFilter = System.getProperty("jvmci.traceMethodDataFilter");

    @Override

    public ProfilingInfo getProfilingInfo(boolean includeNormal, boolean includeOSR) {

        ProfilingInfo info;

        if (UseProfilingInformation.getValue() && methodData == null) {

            long metaspaceMethodData = UNSAFE.getAddress(metaspaceMethod + config().methodDataOffset);

            if (metaspaceMethodData != 0) {

                methodData = new HotSpotMethodData(metaspaceMethodData, this);

                if (TraceMethodDataFilter != null && this.format("%H.%n").contains(TraceMethodDataFilter)) {

                    System.out.println("Raw method data for " + this.format("%H.%n(%p)") + ":");

                    System.out.println(methodData.toString());

                }

            }

        }

        if (methodData == null || (!methodData.hasNormalData() && !methodData.hasExtraData())) {

            // Be optimistic and return false for exceptionSeen. A methodDataOop is allocated in

            // case of a deoptimization.

            info = DefaultProfilingInfo.get(TriState.FALSE);

        } else {

            info = new HotSpotProfilingInfo(methodData, this, includeNormal, includeOSR);

        }

        return info;

    }

    @Override

    public void reprofile() {

        compilerToVM().reprofile(this);

    }

    @Override

    public ConstantPool getConstantPool() {

        return constantPool;

    }

    @Override

    public Annotation[][] getParameterAnnotations() {

        Executable javaMethod = toJava();

        return javaMethod == null ? null : javaMethod.getParameterAnnotations();

    }

    @Override

    public Annotation[] getAnnotations() {

        Executable javaMethod = toJava();

        return javaMethod == null ? new Annotation[0] : javaMethod.getAnnotations();

    }

    @Override

    public <T extends Annotation> T getAnnotation(Class<T> annotationClass) {

        Executable javaMethod = toJava();

        return javaMethod == null ? null : javaMethod.getAnnotation(annotationClass);

    }

    public boolean isDefault() {

        if (isConstructor()) {

            return false;

        }

        // Copied from java.lang.Method.isDefault()

        int mask = Modifier.ABSTRACT | Modifier.PUBLIC | Modifier.STATIC;

        return ((getModifiers() & mask) == Modifier.PUBLIC) && getDeclaringClass().isInterface();

    }

    @Override

    public Type[] getGenericParameterTypes() {

        Executable javaMethod = toJava();

        return javaMethod == null ? null : javaMethod.getGenericParameterTypes();

    }

    public Class<?>[] signatureToTypes() {

        Signature sig = getSignature();

        int count = sig.getParameterCount(false);

        Class<?>[] result = new Class<?>[count];

        for (int i = 0; i < result.length; ++i) {

            JavaType parameterType = sig.getParameterType(i, holder);

            HotSpotResolvedJavaType resolvedParameterType = (HotSpotResolvedJavaType) parameterType.resolve(holder);

            result[i] = resolvedParameterType.mirror();

        }

        return result;

    }

    private Executable toJava() {

        if (toJavaCache != null) {

            return toJavaCache;

        }

        try {

            Class<?>[] parameterTypes = signatureToTypes();

            Executable result = isConstructor() ? holder.mirror().getDeclaredConstructor(parameterTypes) : holder.mirror().getDeclaredMethod(name, parameterTypes);

            toJavaCache = result;

            return result;

        } catch (NoSuchMethodException | NoClassDefFoundError e) {

            return null;

        }

    }

    @Override