/*

 * Copyright 2003-2008 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/_stubGenerator_x86_64.cpp.incl"

// Declaration and definition of StubGenerator (no .hpp file).

// For a more detailed description of the stub routine structure

// see the comment in stubRoutines.hpp

#define __ _masm->

#define TIMES_OOP (UseCompressedOops ? Address::times_4 : Address::times_8)

#ifdef PRODUCT

#define BLOCK_COMMENT(str) /* nothing */

#else

#define BLOCK_COMMENT(str) __ block_comment(str)

#endif

#define BIND(label) bind(label); BLOCK_COMMENT(#label ":")

const int MXCSR_MASK = 0xFFC0;  // Mask out any pending exceptions

// Stub Code definitions

static address handle_unsafe_access() {

  JavaThread* thread = JavaThread::current();

  address pc = thread->saved_exception_pc();

  // pc is the instruction which we must emulate

  // doing a no-op is fine:  return garbage from the load

  // therefore, compute npc

  address npc = Assembler::locate_next_instruction(pc);

  // request an async exception

  thread->set_pending_unsafe_access_error();

  // return address of next instruction to execute

  return npc;

}

class StubGenerator: public StubCodeGenerator {

 private:

#ifdef PRODUCT

#define inc_counter_np(counter) (0)

#else

  void inc_counter_np_(int& counter) {

    __ incrementl(ExternalAddress((address)&counter));

  }

#define inc_counter_np(counter) \

  BLOCK_COMMENT("inc_counter " #counter); \

  inc_counter_np_(counter);

#endif

  // Call stubs are used to call Java from C

  //

  // Linux Arguments:

  //    c_rarg0:   call wrapper address                   address

  //    c_rarg1:   result                                 address

  //    c_rarg2:   result type                            BasicType

  //    c_rarg3:   method                                 methodOop

  //    c_rarg4:   (interpreter) entry point              address

  //    c_rarg5:   parameters                             intptr_t*

  //    16(rbp): parameter size (in words)              int

  //    24(rbp): thread                                 Thread*

  //

  //     [ return_from_Java     ] <--- rsp

  //     [ argument word n      ]

  //      ...

  // -12 [ argument word 1      ]

  // -11 [ saved r15            ] <--- rsp_after_call

  // -10 [ saved r14            ]

  //  -9 [ saved r13            ]

  //  -8 [ saved r12            ]

  //  -7 [ saved rbx            ]

  //  -6 [ call wrapper         ]

  //  -5 [ result               ]

  //  -4 [ result type          ]

  //  -3 [ method               ]

  //  -2 [ entry point          ]

  //  -1 [ parameters           ]

  //   0 [ saved rbp            ] <--- rbp

  //   1 [ return address       ]

  //   2 [ parameter size       ]

  //   3 [ thread               ]

  //

  // Windows Arguments:

  //    c_rarg0:   call wrapper address                   address

  //    c_rarg1:   result                                 address

  //    c_rarg2:   result type                            BasicType

  //    c_rarg3:   method                                 methodOop

  //    48(rbp): (interpreter) entry point              address

  //    56(rbp): parameters                             intptr_t*

  //    64(rbp): parameter size (in words)              int

  //    72(rbp): thread                                 Thread*

  //

  //     [ return_from_Java     ] <--- rsp

  //     [ argument word n      ]

  //      ...

  //  -8 [ argument word 1      ]

  //  -7 [ saved r15            ] <--- rsp_after_call

  //  -6 [ saved r14            ]

  //  -5 [ saved r13            ]

  //  -4 [ saved r12            ]

  //  -3 [ saved rdi            ]

  //  -2 [ saved rsi            ]

  //  -1 [ saved rbx            ]

  //   0 [ saved rbp            ] <--- rbp

  //   1 [ return address       ]

  //   2 [ call wrapper         ]

  //   3 [ result               ]

  //   4 [ result type          ]

  //   5 [ method               ]

  //   6 [ entry point          ]

  //   7 [ parameters           ]

  //   8 [ parameter size       ]

  //   9 [ thread               ]

  //

  //    Windows reserves the callers stack space for arguments 1-4.

  //    We spill c_rarg0-c_rarg3 to this space.

  // Call stub stack layout word offsets from rbp

  enum call_stub_layout {

#ifdef _WIN64

    rsp_after_call_off = -7,

    r15_off            = rsp_after_call_off,

    r14_off            = -6,

    r13_off            = -5,

    r12_off            = -4,

    rdi_off            = -3,

    rsi_off            = -2,

    rbx_off            = -1,

    rbp_off            =  0,

    retaddr_off        =  1,

    call_wrapper_off   =  2,

    result_off         =  3,

    result_type_off    =  4,

    method_off         =  5,

    entry_point_off    =  6,

    parameters_off     =  7,

    parameter_size_off =  8,

    thread_off         =  9

#else

    rsp_after_call_off = -12,

    mxcsr_off          = rsp_after_call_off,

    r15_off            = -11,

    r14_off            = -10,

    r13_off            = -9,

    r12_off            = -8,

    rbx_off            = -7,

    call_wrapper_off   = -6,

    result_off         = -5,

    result_type_off    = -4,

    method_off         = -3,

    entry_point_off    = -2,

    parameters_off     = -1,

    rbp_off            =  0,

    retaddr_off        =  1,

    parameter_size_off =  2,

    thread_off         =  3

#endif

  };

  address generate_call_stub(address& return_address) {

    assert((int)frame::entry_frame_after_call_words == -(int)rsp_after_call_off + 1 &&

           (int)frame::entry_frame_call_wrapper_offset == (int)call_wrapper_off,

           "adjust this code");

    StubCodeMark mark(this, "StubRoutines", "call_stub");

    address start = __ pc();

    // same as in generate_catch_exception()!

    const Address rsp_after_call(rbp, rsp_after_call_off * wordSize);

    const Address call_wrapper  (rbp, call_wrapper_off   * wordSize);

    const Address result        (rbp, result_off         * wordSize);

    const Address result_type   (rbp, result_type_off    * wordSize);

    const Address method        (rbp, method_off         * wordSize);

    const Address entry_point   (rbp, entry_point_off    * wordSize);

    const Address parameters    (rbp, parameters_off     * wordSize);

    const Address parameter_size(rbp, parameter_size_off * wordSize);

    // same as in generate_catch_exception()!

    const Address thread        (rbp, thread_off         * wordSize);

    const Address r15_save(rbp, r15_off * wordSize);

    const Address r14_save(rbp, r14_off * wordSize);

    const Address r13_save(rbp, r13_off * wordSize);

    const Address r12_save(rbp, r12_off * wordSize);

    const Address rbx_save(rbp, rbx_off * wordSize);

    // stub code

    __ enter();

    __ subq(rsp, -rsp_after_call_off * wordSize);

    // save register parameters

#ifndef _WIN64

    __ movq(parameters,   c_rarg5); // parameters

    __ movq(entry_point,  c_rarg4); // entry_point

#endif

    __ movq(method,       c_rarg3); // method

    __ movl(result_type,  c_rarg2); // result type

    __ movq(result,       c_rarg1); // result

    __ movq(call_wrapper, c_rarg0); // call wrapper

    // save regs belonging to calling function

    __ movq(rbx_save, rbx);

    __ movq(r12_save, r12);

    __ movq(r13_save, r13);

    __ movq(r14_save, r14);

    __ movq(r15_save, r15);

#ifdef _WIN64

    const Address rdi_save(rbp, rdi_off * wordSize);

    const Address rsi_save(rbp, rsi_off * wordSize);

    __ movq(rsi_save, rsi);

    __ movq(rdi_save, rdi);

#else

    const Address mxcsr_save(rbp, mxcsr_off * wordSize);

    {

      Label skip_ldmx;

      __ stmxcsr(mxcsr_save);

      __ movl(rax, mxcsr_save);

      __ andl(rax, MXCSR_MASK);    // Only check control and mask bits

      ExternalAddress mxcsr_std(StubRoutines::amd64::mxcsr_std());

      __ cmp32(rax, mxcsr_std);

      __ jcc(Assembler::equal, skip_ldmx);

      __ ldmxcsr(mxcsr_std);

      __ bind(skip_ldmx);

    }

#endif

    // Load up thread register

    __ movq(r15_thread, thread);

    __ reinit_heapbase();

#ifdef ASSERT

    // make sure we have no pending exceptions

    {

      Label L;

      __ cmpq(Address(r15_thread, Thread::pending_exception_offset()), (int)NULL_WORD);

      __ jcc(Assembler::equal, L);

      __ stop("StubRoutines::call_stub: entered with pending exception");

      __ bind(L);

    }

#endif

    // pass parameters if any

    BLOCK_COMMENT("pass parameters if any");

    Label parameters_done;

    __ movl(c_rarg3, parameter_size);

    __ testl(c_rarg3, c_rarg3);

    __ jcc(Assembler::zero, parameters_done);

    Label loop;

    __ movq(c_rarg2, parameters);     // parameter pointer

    __ movl(c_rarg1, c_rarg3);        // parameter counter is in c_rarg1

    __ BIND(loop);

    if (TaggedStackInterpreter) {

      __ movq(rax, Address(c_rarg2, 0)); // get tag

      __ addq(c_rarg2, wordSize);     // advance to next tag

      __ pushq(rax);                  // pass tag

    }

    __ movq(rax, Address(c_rarg2, 0));  // get parameter

    __ addq(c_rarg2, wordSize);       // advance to next parameter

    __ decrementl(c_rarg1);           // decrement counter

    __ pushq(rax);                    // pass parameter

    __ jcc(Assembler::notZero, loop);

    // call Java function

    __ BIND(parameters_done);

    __ movq(rbx, method);             // get methodOop

    __ movq(c_rarg1, entry_point);    // get entry_point

    __ movq(r13, rsp);                // set sender sp

    BLOCK_COMMENT("call Java function");

    __ call(c_rarg1);

    BLOCK_COMMENT("call_stub_return_address:");

    return_address = __ pc();

    // store result depending on type (everything that is not

    // T_OBJECT, T_LONG, T_FLOAT or T_DOUBLE is treated as T_INT)

    __ movq(c_rarg0, result);

    Label is_long, is_float, is_double, exit;

    __ movl(c_rarg1, result_type);

    __ cmpl(c_rarg1, T_OBJECT);

    __ jcc(Assembler::equal, is_long);

    __ cmpl(c_rarg1, T_LONG);

    __ jcc(Assembler::equal, is_long);

    __ cmpl(c_rarg1, T_FLOAT);

    __ jcc(Assembler::equal, is_float);

    __ cmpl(c_rarg1, T_DOUBLE);

    __ jcc(Assembler::equal, is_double);

    // handle T_INT case

    __ movl(Address(c_rarg0, 0), rax);

    __ BIND(exit);

    // pop parameters

    __ leaq(rsp, rsp_after_call);

#ifdef ASSERT

    // verify that threads correspond

    {

      Label L, S;

      __ cmpq(r15_thread, thread);

      __ jcc(Assembler::notEqual, S);

      __ get_thread(rbx);

      __ cmpq(r15_thread, rbx);

      __ jcc(Assembler::equal, L);

      __ bind(S);

      __ jcc(Assembler::equal, L);

      __ stop("StubRoutines::call_stub: threads must correspond");

      __ bind(L);

    }

#endif

    // restore regs belonging to calling function

    __ movq(r15, r15_save);

    __ movq(r14, r14_save);

    __ movq(r13, r13_save);

    __ movq(r12, r12_save);

    __ movq(rbx, rbx_save);

#ifdef _WIN64

    __ movq(rdi, rdi_save);

    __ movq(rsi, rsi_save);

#else

    __ ldmxcsr(mxcsr_save);

#endif

    // restore rsp

    __ addq(rsp, -rsp_after_call_off * wordSize);

    // return

    __ popq(rbp);

    __ ret(0);

    // handle return types different from T_INT

    __ BIND(is_long);

    __ movq(Address(c_rarg0, 0), rax);

    __ jmp(exit);

    __ BIND(is_float);

    __ movflt(Address(c_rarg0, 0), xmm0);

    __ jmp(exit);

    __ BIND(is_double);

    __ movdbl(Address(c_rarg0, 0), xmm0);

    __ jmp(exit);

    return start;

  }

  // Return point for a Java call if there's an exception thrown in

  // Java code.  The exception is caught and transformed into a

  // pending exception stored in JavaThread that can be tested from

  // within the VM.

  //

  // Note: Usually the parameters are removed by the callee. In case

  // of an exception crossing an activation frame boundary, that is

  // not the case if the callee is compiled code => need to setup the

  // rsp.

  //

  // rax: exception oop

  address generate_catch_exception() {

    StubCodeMark mark(this, "StubRoutines", "catch_exception");

    address start = __ pc();

    // same as in generate_call_stub():

    const Address rsp_after_call(rbp, rsp_after_call_off * wordSize);

    const Address thread        (rbp, thread_off         * wordSize);

#ifdef ASSERT

    // verify that threads correspond

    {

      Label L, S;

      __ cmpq(r15_thread, thread);

      __ jcc(Assembler::notEqual, S);

      __ get_thread(rbx);

      __ cmpq(r15_thread, rbx);

      __ jcc(Assembler::equal, L);

      __ bind(S);

      __ stop("StubRoutines::catch_exception: threads must correspond");

      __ bind(L);

    }

#endif

    // set pending exception

    __ verify_oop(rax);

    __ movq(Address(r15_thread, Thread::pending_exception_offset()), rax);

    __ lea(rscratch1, ExternalAddress((address)__FILE__));

    __ movq(Address(r15_thread, Thread::exception_file_offset()), rscratch1);

    __ movl(Address(r15_thread, Thread::exception_line_offset()), (int)  __LINE__);

    // complete return to VM

    assert(StubRoutines::_call_stub_return_address != NULL,

           "_call_stub_return_address must have been generated before");

    __ jump(RuntimeAddress(StubRoutines::_call_stub_return_address));

    return start;

  }

  // Continuation point for runtime calls returning with a pending

  // exception.  The pending exception check happened in the runtime

  // or native call stub.  The pending exception in Thread is

  // converted into a Java-level exception.

  //

  // Contract with Java-level exception handlers:

  // rax: exception

  // rdx: throwing pc

  //

  // NOTE: At entry of this stub, exception-pc must be on stack !!

  address generate_forward_exception() {

    StubCodeMark mark(this, "StubRoutines", "forward exception");

    address start = __ pc();

    // Upon entry, the sp points to the return address returning into

    // Java (interpreted or compiled) code; i.e., the return address

    // becomes the throwing pc.

    //

    // Arguments pushed before the runtime call are still on the stack

    // but the exception handler will reset the stack pointer ->

    // ignore them.  A potential result in registers can be ignored as

    // well.

#ifdef ASSERT

    // make sure this code is only executed if there is a pending exception

    {

      Label L;

      __ cmpq(Address(r15_thread, Thread::pending_exception_offset()), (int) NULL);

      __ jcc(Assembler::notEqual, L);

      __ stop("StubRoutines::forward exception: no pending exception (1)");

      __ bind(L);

    }

#endif

    // compute exception handler into rbx

    __ movq(c_rarg0, Address(rsp, 0));

    BLOCK_COMMENT("call exception_handler_for_return_address");

    __ call_VM_leaf(CAST_FROM_FN_PTR(address,

                         SharedRuntime::exception_handler_for_return_address),

                    c_rarg0);

    __ movq(rbx, rax);

    // setup rax & rdx, remove return address & clear pending exception

    __ popq(rdx);

    __ movq(rax, Address(r15_thread, Thread::pending_exception_offset()));

    __ movptr(Address(r15_thread, Thread::pending_exception_offset()), (int)NULL_WORD);

#ifdef ASSERT

    // make sure exception is set

    {

      Label L;

      __ testq(rax, rax);

      __ jcc(Assembler::notEqual, L);

      __ stop("StubRoutines::forward exception: no pending exception (2)");

      __ bind(L);

    }

#endif

    // continue at exception handler (return address removed)

    // rax: exception

    // rbx: exception handler

    // rdx: throwing pc

    __ verify_oop(rax);

    __ jmp(rbx);

    return start;

  }

  // Support for jint atomic::xchg(jint exchange_value, volatile jint* dest)

  //

  // Arguments :

  //    c_rarg0: exchange_value

  //    c_rarg0: dest

  //

  // Result:

  //    *dest <- ex, return (orig *dest)

  address generate_atomic_xchg() {

    StubCodeMark mark(this, "StubRoutines", "atomic_xchg");

    address start = __ pc();

    __ movl(rax, c_rarg0); // Copy to eax we need a return value anyhow

    __ xchgl(rax, Address(c_rarg1, 0)); // automatic LOCK

    __ ret(0);

    return start;

  }

  // Support for intptr_t atomic::xchg_ptr(intptr_t exchange_value, volatile intptr_t* dest)

  //

  // Arguments :

  //    c_rarg0: exchange_value

  //    c_rarg1: dest

  //

  // Result:

  //    *dest <- ex, return (orig *dest)

  address generate_atomic_xchg_ptr() {

    StubCodeMark mark(this, "StubRoutines", "atomic_xchg_ptr");

    address start = __ pc();

    __ movq(rax, c_rarg0); // Copy to eax we need a return value anyhow