/*

 * Copyright 1997-2009 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.

 *

 */

#include "incls/_precompiled.incl"

#include "incls/_methodHandles_x86.cpp.incl"

#define __ _masm->

address MethodHandleEntry::start_compiled_entry(MacroAssembler* _masm,

                                                address interpreted_entry) {

  // Just before the actual machine code entry point, allocate space

  // for a MethodHandleEntry::Data record, so that we can manage everything

  // from one base pointer.

  __ align(wordSize);

  address target = __ pc() + sizeof(Data);

  while (__ pc() < target) {

    __ nop();

    __ align(wordSize);

  }

  MethodHandleEntry* me = (MethodHandleEntry*) __ pc();

  me->set_end_address(__ pc());         // set a temporary end_address

  me->set_from_interpreted_entry(interpreted_entry);

  me->set_type_checking_entry(NULL);

  return (address) me;

}

MethodHandleEntry* MethodHandleEntry::finish_compiled_entry(MacroAssembler* _masm,

                                                address start_addr) {

  MethodHandleEntry* me = (MethodHandleEntry*) start_addr;

  assert(me->end_address() == start_addr, "valid ME");

  // Fill in the real end_address:

  __ align(wordSize);

  me->set_end_address(__ pc());

  return me;

}

#ifdef ASSERT

static void verify_argslot(MacroAssembler* _masm, Register rax_argslot,

                           const char* error_message) {

  // Verify that argslot lies within (rsp, rbp].

  Label L_ok, L_bad;

  __ cmpptr(rax_argslot, rbp);

  __ jccb(Assembler::above, L_bad);

  __ cmpptr(rsp, rax_argslot);

  __ jccb(Assembler::below, L_ok);

  __ bind(L_bad);

  __ stop(error_message);

  __ bind(L_ok);

}

#endif

// Code generation

address MethodHandles::generate_method_handle_interpreter_entry(MacroAssembler* _masm) {

  // rbx: methodOop

  // rcx: receiver method handle (must load from sp[MethodTypeForm.vmslots])

  // rsi/r13: sender SP (must preserve; see prepare_to_jump_from_interpreted)

  // rdx: garbage temp, blown away

  Register rbx_method = rbx;

  Register rcx_recv   = rcx;

  Register rax_mtype  = rax;

  Register rdx_temp   = rdx;

  // emit WrongMethodType path first, to enable jccb back-branch from main path

  Label wrong_method_type;

  __ bind(wrong_method_type);

  __ push(rax_mtype);       // required mtype

  __ push(rcx_recv);        // bad mh (1st stacked argument)

  __ jump(ExternalAddress(Interpreter::throw_WrongMethodType_entry()));

  // here's where control starts out:

  __ align(CodeEntryAlignment);

  address entry_point = __ pc();

  // fetch the MethodType from the method handle into rax (the 'check' register)

  {

    Register tem = rbx_method;

    for (jint* pchase = methodOopDesc::method_type_offsets_chain(); (*pchase) != -1; pchase++) {

      __ movptr(rax_mtype, Address(tem, *pchase));

      tem = rax_mtype;          // in case there is another indirection

    }

  }

  Register rbx_temp = rbx_method; // done with incoming methodOop

  // given the MethodType, find out where the MH argument is buried

  __ movptr(rdx_temp, Address(rax_mtype,

                              __ delayed_value(java_dyn_MethodType::form_offset_in_bytes, rbx_temp)));

  __ movl(rdx_temp, Address(rdx_temp,

                            __ delayed_value(java_dyn_MethodTypeForm::vmslots_offset_in_bytes, rbx_temp)));

  __ movptr(rcx_recv, __ argument_address(rdx_temp));

  __ check_method_handle_type(rax_mtype, rcx_recv, rdx_temp, wrong_method_type);

  __ jump_to_method_handle_entry(rcx_recv, rdx_temp);

  return entry_point;

}

// Helper to insert argument slots into the stack.

// arg_slots must be a multiple of stack_move_unit() and <= 0

void MethodHandles::insert_arg_slots(MacroAssembler* _masm,

                                     RegisterOrConstant arg_slots,

                                     int arg_mask,

                                     Register rax_argslot,

                                     Register rbx_temp, Register rdx_temp) {

  assert_different_registers(rax_argslot, rbx_temp, rdx_temp,

                             (!arg_slots.is_register() ? rsp : arg_slots.as_register()));

#ifdef ASSERT

  verify_argslot(_masm, rax_argslot, "insertion point must fall within current frame");

  if (arg_slots.is_register()) {

    Label L_ok, L_bad;

    __ cmpptr(arg_slots.as_register(), (int32_t) NULL_WORD);

    __ jccb(Assembler::greater, L_bad);

    __ testl(arg_slots.as_register(), -stack_move_unit() - 1);

    __ jccb(Assembler::zero, L_ok);

    __ bind(L_bad);

    __ stop("assert arg_slots <= 0 and clear low bits");

    __ bind(L_ok);

  } else {

    assert(arg_slots.as_constant() <= 0, "");

    assert(arg_slots.as_constant() % -stack_move_unit() == 0, "");

  }

#endif //ASSERT

#ifdef _LP64

  if (arg_slots.is_register()) {

    // clean high bits of stack motion register (was loaded as an int)

    __ movslq(arg_slots.as_register(), arg_slots.as_register());

  }

#endif

  // Make space on the stack for the inserted argument(s).

  // Then pull down everything shallower than rax_argslot.

  // The stacked return address gets pulled down with everything else.

  // That is, copy [rsp, argslot) downward by -size words.  In pseudo-code:

  //   rsp -= size;

  //   for (rdx = rsp + size; rdx < argslot; rdx++)

  //     rdx[-size] = rdx[0]

  //   argslot -= size;

  __ mov(rdx_temp, rsp);                        // source pointer for copy

  __ lea(rsp, Address(rsp, arg_slots, Address::times_ptr));

  {

    Label loop;

    __ bind(loop);

    // pull one word down each time through the loop

    __ movptr(rbx_temp, Address(rdx_temp, 0));

    __ movptr(Address(rdx_temp, arg_slots, Address::times_ptr), rbx_temp);

    __ addptr(rdx_temp, wordSize);

    __ cmpptr(rdx_temp, rax_argslot);

    __ jccb(Assembler::less, loop);

  }

  // Now move the argslot down, to point to the opened-up space.

  __ lea(rax_argslot, Address(rax_argslot, arg_slots, Address::times_ptr));

  if (TaggedStackInterpreter && arg_mask != _INSERT_NO_MASK) {

    // The caller has specified a bitmask of tags to put into the opened space.

    // This only works when the arg_slots value is an assembly-time constant.

    int constant_arg_slots = arg_slots.as_constant() / stack_move_unit();

    int tag_offset = Interpreter::tag_offset_in_bytes() - Interpreter::value_offset_in_bytes();

    for (int slot = 0; slot < constant_arg_slots; slot++) {

      BasicType slot_type   = ((arg_mask & (1 << slot)) == 0 ? T_OBJECT : T_INT);

      int       slot_offset = Interpreter::stackElementSize() * slot;

      Address   tag_addr(rax_argslot, slot_offset + tag_offset);

      __ movptr(tag_addr, frame::tag_for_basic_type(slot_type));

    }

    // Note that the new argument slots are tagged properly but contain

    // garbage at this point.  The value portions must be initialized

    // by the caller.  (Especially references!)

  }

}

// Helper to remove argument slots from the stack.

// arg_slots must be a multiple of stack_move_unit() and >= 0

void MethodHandles::remove_arg_slots(MacroAssembler* _masm,

                                    RegisterOrConstant arg_slots,

                                    Register rax_argslot,

                                    Register rbx_temp, Register rdx_temp) {

  assert_different_registers(rax_argslot, rbx_temp, rdx_temp,

                             (!arg_slots.is_register() ? rsp : arg_slots.as_register()));

#ifdef ASSERT

  {

    // Verify that [argslot..argslot+size) lies within (rsp, rbp).

    Label L_ok, L_bad;

    __ lea(rbx_temp, Address(rax_argslot, arg_slots, Address::times_ptr));

    __ cmpptr(rbx_temp, rbp);

    __ jccb(Assembler::above, L_bad);

    __ cmpptr(rsp, rax_argslot);

    __ jccb(Assembler::below, L_ok);

    __ bind(L_bad);

    __ stop("deleted argument(s) must fall within current frame");

    __ bind(L_ok);

  }

  if (arg_slots.is_register()) {

    Label L_ok, L_bad;

    __ cmpptr(arg_slots.as_register(), (int32_t) NULL_WORD);

    __ jccb(Assembler::less, L_bad);

    __ testl(arg_slots.as_register(), -stack_move_unit() - 1);

    __ jccb(Assembler::zero, L_ok);

    __ bind(L_bad);

    __ stop("assert arg_slots >= 0 and clear low bits");

    __ bind(L_ok);

  } else {

    assert(arg_slots.as_constant() >= 0, "");

    assert(arg_slots.as_constant() % -stack_move_unit() == 0, "");

  }

#endif //ASSERT

#ifdef _LP64

  if (false) {                  // not needed, since register is positive

    // clean high bits of stack motion register (was loaded as an int)

    if (arg_slots.is_register())

      __ movslq(arg_slots.as_register(), arg_slots.as_register());

  }

#endif

  // Pull up everything shallower than rax_argslot.

  // Then remove the excess space on the stack.

  // The stacked return address gets pulled up with everything else.

  // That is, copy [rsp, argslot) upward by size words.  In pseudo-code:

  //   for (rdx = argslot-1; rdx >= rsp; --rdx)

  //     rdx[size] = rdx[0]

  //   argslot += size;

  //   rsp += size;

  __ lea(rdx_temp, Address(rax_argslot, -wordSize)); // source pointer for copy

  {

    Label loop;

    __ bind(loop);

    // pull one word up each time through the loop

    __ movptr(rbx_temp, Address(rdx_temp, 0));

    __ movptr(Address(rdx_temp, arg_slots, Address::times_ptr), rbx_temp);

    __ addptr(rdx_temp, -wordSize);

    __ cmpptr(rdx_temp, rsp);

    __ jccb(Assembler::greaterEqual, loop);

  }

  // Now move the argslot up, to point to the just-copied block.

  __ lea(rsp, Address(rsp, arg_slots, Address::times_ptr));

  // And adjust the argslot address to point at the deletion point.

  __ lea(rax_argslot, Address(rax_argslot, arg_slots, Address::times_ptr));

}

#ifndef PRODUCT

extern "C" void print_method_handle(oop mh);

void trace_method_handle_stub(const char* adaptername,

                              oop mh,

                              intptr_t* entry_sp,

                              intptr_t* saved_sp,

                              intptr_t* saved_bp) {

  // called as a leaf from native code: do not block the JVM!

  intptr_t* last_sp = (intptr_t*) saved_bp[frame::interpreter_frame_last_sp_offset];

  intptr_t* base_sp = (intptr_t*) saved_bp[frame::interpreter_frame_monitor_block_top_offset];

  printf("MH %s mh="INTPTR_FORMAT" sp=("INTPTR_FORMAT"+"INTX_FORMAT") stack_size="INTX_FORMAT" bp="INTPTR_FORMAT"\n",

         adaptername, (intptr_t)mh, (intptr_t)entry_sp, (intptr_t)(saved_sp - entry_sp), (intptr_t)(base_sp - last_sp), (intptr_t)saved_bp);

  if (last_sp != saved_sp)

    printf("*** last_sp="INTPTR_FORMAT"\n", (intptr_t)last_sp);

  if (Verbose)  print_method_handle(mh);

}

#endif //PRODUCT

// Generate an "entry" field for a method handle.

// This determines how the method handle will respond to calls.

void MethodHandles::generate_method_handle_stub(MacroAssembler* _masm, MethodHandles::EntryKind ek) {

  // Here is the register state during an interpreted call,

  // as set up by generate_method_handle_interpreter_entry():

  // - rbx: garbage temp (was MethodHandle.invoke methodOop, unused)

  // - rcx: receiver method handle

  // - rax: method handle type (only used by the check_mtype entry point)

  // - rsi/r13: sender SP (must preserve; see prepare_to_jump_from_interpreted)

  // - rdx: garbage temp, can blow away

  Register rcx_recv    = rcx;

  Register rax_argslot = rax;

  Register rbx_temp    = rbx;

  Register rdx_temp    = rdx;

  // This guy is set up by prepare_to_jump_from_interpreted (from interpreted calls)

  // and gen_c2i_adapter (from compiled calls):

  Register saved_last_sp = LP64_ONLY(r13) NOT_LP64(rsi);

  guarantee(java_dyn_MethodHandle::vmentry_offset_in_bytes() != 0, "must have offsets");

  // some handy addresses

  Address rbx_method_fie(     rbx,      methodOopDesc::from_interpreted_offset() );

  Address rcx_mh_vmtarget(    rcx_recv, java_dyn_MethodHandle::vmtarget_offset_in_bytes() );

  Address rcx_dmh_vmindex(    rcx_recv, sun_dyn_DirectMethodHandle::vmindex_offset_in_bytes() );

  Address rcx_bmh_vmargslot(  rcx_recv, sun_dyn_BoundMethodHandle::vmargslot_offset_in_bytes() );

  Address rcx_bmh_argument(   rcx_recv, sun_dyn_BoundMethodHandle::argument_offset_in_bytes() );

  Address rcx_amh_vmargslot(  rcx_recv, sun_dyn_AdapterMethodHandle::vmargslot_offset_in_bytes() );

  Address rcx_amh_argument(   rcx_recv, sun_dyn_AdapterMethodHandle::argument_offset_in_bytes() );

  Address rcx_amh_conversion( rcx_recv, sun_dyn_AdapterMethodHandle::conversion_offset_in_bytes() );

  Address vmarg;                // __ argument_address(vmargslot)

  int tag_offset = -1;

  if (TaggedStackInterpreter) {

    tag_offset = Interpreter::tag_offset_in_bytes() - Interpreter::value_offset_in_bytes();

    assert(tag_offset = wordSize, "stack grows as expected");

  }

  const int java_mirror_offset = klassOopDesc::klass_part_offset_in_bytes() + Klass::java_mirror_offset_in_bytes();

  if (have_entry(ek)) {

    __ nop();                   // empty stubs make SG sick

    return;

  }

  address interp_entry = __ pc();

  if (UseCompressedOops)  __ unimplemented("UseCompressedOops");

#ifndef PRODUCT

  if (TraceMethodHandles) {

    __ push(rax); __ push(rbx); __ push(rcx); __ push(rdx); __ push(rsi); __ push(rdi);

    __ lea(rax, Address(rsp, wordSize*6)); // entry_sp

    // arguments:

    __ push(rbp);               // interpreter frame pointer

    __ push(rsi);               // saved_sp

    __ push(rax);               // entry_sp

    __ push(rcx);               // mh

    __ push(rcx);

    __ movptr(Address(rsp, 0), (intptr_t)entry_name(ek));

    __ call_VM_leaf(CAST_FROM_FN_PTR(address, trace_method_handle_stub), 5);

    __ pop(rdi); __ pop(rsi); __ pop(rdx); __ pop(rcx); __ pop(rbx); __ pop(rax);

  }

#endif //PRODUCT

  switch ((int) ek) {

  case _raise_exception:

    {

      // Not a real MH entry, but rather shared code for raising an exception.

      // Extra local arguments are pushed on stack, as required type at TOS+8,

      // failing object (or NULL) at TOS+4, failing bytecode type at TOS.

      // Beyond those local arguments are the PC, of course.

      Register rdx_code = rdx_temp;

      Register rcx_fail = rcx_recv;

      Register rax_want = rax_argslot;

      Register rdi_pc   = rdi;

      __ pop(rdx_code);  // TOS+0

      __ pop(rcx_fail);  // TOS+4

      __ pop(rax_want);  // TOS+8

      __ pop(rdi_pc);    // caller PC

      __ mov(rsp, rsi);   // cut the stack back to where the caller started

      // Repush the arguments as if coming from the interpreter.

      if (TaggedStackInterpreter)  __ push(frame::tag_for_basic_type(T_INT));

      __ push(rdx_code);

      if (TaggedStackInterpreter)  __ push(frame::tag_for_basic_type(T_OBJECT));

      __ push(rcx_fail);

      if (TaggedStackInterpreter)  __ push(frame::tag_for_basic_type(T_OBJECT));

      __ push(rax_want);

      Register rbx_method = rbx_temp;

      Label no_method;

      // FIXME: fill in _raise_exception_method with a suitable sun.dyn method

      __ movptr(rbx_method, ExternalAddress((address) &_raise_exception_method));

      __ testptr(rbx_method, rbx_method);

      __ jccb(Assembler::zero, no_method);

      int jobject_oop_offset = 0;

      __ movptr(rbx_method, Address(rbx_method, jobject_oop_offset));  // dereference the jobject

      __ testptr(rbx_method, rbx_method);

      __ jccb(Assembler::zero, no_method);

      __ verify_oop(rbx_method);

      __ push(rdi_pc);          // and restore caller PC

      __ jmp(rbx_method_fie);

      // If we get here, the Java runtime did not do its job of creating the exception.

      // Do something that is at least causes a valid throw from the interpreter.

      __ bind(no_method);

      __ pop(rax_want);

      if (TaggedStackInterpreter)  __ pop(rcx_fail);

      __ pop(rcx_fail);

      __ push(rax_want);

      __ push(rcx_fail);

      __ jump(ExternalAddress(Interpreter::throw_WrongMethodType_entry()));

    }

    break;

  case _invokestatic_mh:

  case _invokespecial_mh:

    {

      Register rbx_method = rbx_temp;

      __ movptr(rbx_method, rcx_mh_vmtarget); // target is a methodOop

      __ verify_oop(rbx_method);

      // same as TemplateTable::invokestatic or invokespecial,

      // minus the CP setup and profiling:

      if (ek == _invokespecial_mh) {

        // Must load & check the first argument before entering the target method.

        __ load_method_handle_vmslots(rax_argslot, rcx_recv, rdx_temp);

        __ movptr(rcx_recv, __ argument_address(rax_argslot, -1));

        __ null_check(rcx_recv);

        __ verify_oop(rcx_recv);

      }

      __ jmp(rbx_method_fie);

    }

    break;

  case _invokevirtual_mh:

    {

      // same as TemplateTable::invokevirtual,

      // minus the CP setup and profiling:

      // pick out the vtable index and receiver offset from the MH,

      // and then we can discard it:

      __ load_method_handle_vmslots(rax_argslot, rcx_recv, rdx_temp);

      Register rbx_index = rbx_temp;

      __ movl(rbx_index, rcx_dmh_vmindex);

      // Note:  The verifier allows us to ignore rcx_mh_vmtarget.

      __ movptr(rcx_recv, __ argument_address(rax_argslot, -1));

      __ null_check(rcx_recv, oopDesc::klass_offset_in_bytes());

      // get receiver klass

      Register rax_klass = rax_argslot;

      __ load_klass(rax_klass, rcx_recv);

      __ verify_oop(rax_klass);

      // get target methodOop & entry point

      const int base = instanceKlass::vtable_start_offset() * wordSize;

      assert(vtableEntry::size() * wordSize == wordSize, "adjust the scaling in the code below");

      Address vtable_entry_addr(rax_klass,

                                rbx_index, Address::times_ptr,

                                base + vtableEntry::method_offset_in_bytes());

      Register rbx_method = rbx_temp;

      __ movptr(rbx_method, vtable_entry_addr);

      __ verify_oop(rbx_method);

      __ jmp(rbx_method_fie);

    }

    break;

  case _invokeinterface_mh:

    {

      // same as TemplateTable::invokeinterface,

      // minus the CP setup and profiling:

      // pick out the interface and itable index from the MH.

      __ load_method_handle_vmslots(rax_argslot, rcx_recv, rdx_temp);

      Register rdx_intf  = rdx_temp;

      Register rbx_index = rbx_temp;

      __ movptr(rdx_intf,  rcx_mh_vmtarget);

      __ movl(rbx_index,   rcx_dmh_vmindex);

      __ movptr(rcx_recv, __ argument_address(rax_argslot, -1));

      __ null_check(rcx_recv, oopDesc::klass_offset_in_bytes());

      // get receiver klass

      Register rax_klass = rax_argslot;

      __ load_klass(rax_klass, rcx_recv);

      __ verify_oop(rax_klass);

      Register rdi_temp   = rdi;

      Register rbx_method = rbx_index;

      // get interface klass

      Label no_such_interface;

      __ verify_oop(rdx_intf);

      __ lookup_interface_method(rax_klass, rdx_intf,

                                 // note: next two args must be the same:

                                 rbx_index, rbx_method,

                                 rdi_temp,

                                 no_such_interface);

      __ verify_oop(rbx_method);

      __ jmp(rbx_method_fie);

      __ hlt();

      __ bind(no_such_interface);

      // Throw an exception.

      // For historical reasons, it will be IncompatibleClassChangeError.

      __ pushptr(Address(rdx_intf, java_mirror_offset));  // required interface

      __ push(rcx_recv);        // bad receiver

      __ push((int)Bytecodes::_invokeinterface);  // who is complaining?

      __ jump(ExternalAddress(from_interpreted_entry(_raise_exception)));

    }

    break;

  case _bound_ref_mh:

  case _bound_int_mh:

  case _bound_long_mh:

  case _bound_ref_direct_mh:

  case _bound_int_direct_mh:

  case _bound_long_direct_mh:

    {

      bool direct_to_method = (ek >= _bound_ref_direct_mh);

      BasicType arg_type = T_ILLEGAL;

      if (ek == _bound_long_mh || ek == _bound_long_direct_mh) {

        arg_type = T_LONG;

      } else if (ek == _bound_int_mh || ek == _bound_int_direct_mh) {

        arg_type = T_INT;

      } else {

        assert(ek == _bound_ref_mh || ek == _bound_ref_direct_mh, "must be ref");