/* vim: set ts=4 sw=4 tw=99 et:

  *

  * Copyright (C) 2012 David Anderson

  *

  * This file is part of JITCraft.

  *

  * JITCraft is free software: you can redistribute it and/or modify it under

  * the terms of the GNU General Public License as published by the Free

  * Software Foundation, either version 3 of the License, or (at your option)

  * any later version.

  * 

  * Foobar 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 for more details.

  *

  * You should have received a copy of the GNU General Public License along with

  * JITCraft. If not, see http://www.gnu.org/licenses/.

  */

 #include <new>

 #include <stdlib.h>

 #include "Stack.h"

 #include "Zone.h"

 #include "Strings.h"

 #include "Modules.h"

 #include "Heap-inl.h"

 using namespace ke;

 Stack::Stack(Zone *zone)

   : zone_(zone),

     buffer_(NULL),

     end_(NULL)

 {

 }

 Stack::~Stack()

 {

     free(buffer_);

 }

 bool

 Stack::initialize()

 {

     buffer_ = (char *)malloc(STACK_SIZE);

     if (!buffer_)

         return false;

     top_.sp = reinterpret_cast<LocalSlot *>(buffer_);

     top_.pc = NULL;

     top_.fp = NULL;

     end_ = buffer_ + STACK_SIZE;

     return true;

 }

 InvokeArgs::~InvokeArgs()

 {

     stack_->popInvokeFrame(*this);

 }

 NativeArgs::~NativeArgs()

 {

     stack_->popNativeFrame(*this);

 }

 static inline size_t

 SizeForCodeFrame(Code *code)

 {

     return sizeof(Frame) + sizeof(LocalSlot) * (code->nlocals() + code->maxStackDepth());

 }

 bool

 Stack::pushInvokeFrame(Handle<Function> fun, InvokeArgs &args)

 {

     size_t nbytes = SizeForCodeFrame(fun->code()) +

                     sizeof(LocalSlot) * fun->nargs();

     if (reinterpret_cast<char *>(top_.sp) + nbytes >= end_) {

         assert(false);

         return false;

     }

     memset(top_.sp, 0, sizeof(LocalSlot) * fun->nargs());

     args.args_ = top_.sp;

     args.nargs_ = fun->nargs();

     args.stack_ = this;

     top_.sp += fun->nargs();

     Frame *frame = reinterpret_cast<Frame *>(top_.sp);

     new (frame) Frame(top_.fp, top_.pc, fun, fun->code(), zone_->heap().lifoPosition());

     top_.pc = frame->code()->start();

     top_.sp = frame->stackBase();

     top_.fp = frame;

     return true;

 }

 bool

 Stack::pushModuleFrame(Handle<Module> module, InvokeArgs &args)

 {

     size_t nbytes = SizeForCodeFrame(module->code());

     if (reinterpret_cast<char *>(top_.sp) + nbytes >= end_) {

         assert(false);

         return false;

     }

     args.args_ = top_.sp;

     args.nargs_ = 0;

     args.stack_ = this;

     Frame *frame = reinterpret_cast<Frame *>(top_.sp);

     new (frame) Frame(top_.fp, top_.pc, NULL, module->code(), zone_->heap().lifoPosition());

     top_.pc = frame->code()->start();

     top_.sp = frame->stackBase();

     top_.fp = frame;

     return true;

 }

 bool

 Stack::pushNativeFrame(Handle<Native> native, NativeArgs &args, unsigned argc)

 {

     size_t nbytes = sizeof(intptr_t) * argc + sizeof(Frame);

     if (reinterpret_cast<char *>(top_.sp) + nbytes >= end_) {

         assert(false);

         return false;

     }

     memset(top_.sp, 0, sizeof(intptr_t) * argc);

     args.argsBase_ = top_.sp;

     args.stack_ = this;

     top_.sp = reinterpret_cast<LocalSlot *>(reinterpret_cast<intptr_t *>(top_.sp) + argc);

     Frame *frame = reinterpret_cast<Frame *>(top_.sp);

     new (frame) Frame(top_.fp, top_.pc, native, zone_->heap().lifoPosition());

     top_.pc = NULL;

     top_.sp = reinterpret_cast<LocalSlot *>(frame + 1);

     top_.fp = frame;

     return true;

 }

 Frame *

 Stack::pushInlineFrame(Handle<Function> fun)

 {

     Code *code = fun->code();

     size_t nbytes = SizeForCodeFrame(code);

     if (reinterpret_cast<char *>(top_.sp) + nbytes >= end_) {

         zone_->reportError(Message_TooMuchRecursion);

         return NULL;

     }

     Frame *frame = reinterpret_cast<Frame *>(top_.sp);

     new (frame) Frame(top_.fp, top_.pc, fun, code, zone_->heap().lifoPosition());

     memset(frame->locals(), 0, sizeof(LocalSlot) * code->nlocals());

     top_.pc = frame->code()->start();

     top_.sp = frame->stackBase();

     top_.fp = frame;

     return frame;

 }

 Frame *

 Stack::popInlineFrame()

 {

     zone_->heap().unwindLifo(top_.fp->lifoPosition());

     Frame *prev = top_.fp->prev();

     Function *fun = top_.fp->fun();

     top_.pc = top_.fp->prevpc();

     top_.sp = reinterpret_cast<LocalSlot *>(top_.fp) - fun->nargs();

     top_.fp = prev;

     return prev;

 }

 void

 Stack::popInvokeFrame(InvokeArgs &args)

 {

     zone_->heap().unwindLifo(top_.fp->lifoPosition());

     top_.pc = top_.fp->prevpc();

     top_.sp = args.args_;

     top_.fp = top_.fp->prev();

 }

 void

 Stack::popNativeFrame(NativeArgs &args)

 {

     zone_->heap().unwindLifo(top_.fp->lifoPosition());

     top_.pc = top_.fp->prevpc();

     top_.sp = args.argsBase_;

     top_.fp = top_.fp->prev();

 }

 void

 Frame::accept(VirtualObjectVisitor *visitor)

 {

     visitor->visit(reinterpret_cast<Object **>(&callee_));

     visitor->visit(reinterpret_cast<Object **>(&code_));

 }

 static void

 TraceFunctionFrame(VirtualObjectVisitor *visitor, Frame *fp, LocalSlot *sp, bytecode *pc)

 {

     Code *code = fp->code();

     LocalSlot *slots = fp->locals();

     // Trace local variables.

     for (unsigned i = 0; i < code->nlocals(); i++) {

         Type *type = code->localType(i);

         if (type->isTraceable())

             visitor->visit(&slots[i].obj);

     }

     // Trace arguments, if a function call. Note that this means we usually

     // mark arguments twice, once on the caller side and once on the callee.

     // This makes it easier to deal with invoke frames which have no caller.

     if (Function *fun = fp->maybeFun()) {

         for (unsigned i = 0; i < fun->nargs(); i++) {

             Type *type = fun->argType(i);

             if (type->isTraceable())

                 visitor->visit(&fp->arg(i).obj);

         }

     }

     // Finally, trace stack slots. Not every gc-point in the bytecode will

     // have a safepoint; since they are actually uncommon, we don't emit

     // them unless at least one stack slot would be marked.

     if (uint8 *safepoint = code->safepointAt(code->pcToOffset(pc))) {

         LocalSlot *base = fp->stackBase();

         unsigned stackDepth = unsigned(sp - fp->stackBase());

         assert(stackDepth <= code->maxStackDepth());

         for (unsigned i = 0; i < stackDepth; i++) {

             if (safepoint[i / 8] & (1 << (i % 8)))

                 visitor->visit(&base[i].obj);

         }

     }

 }

 void

 Stack::accept(VirtualObjectVisitor *visitor)

 {

     LocalSlot *sp = top_.sp;

     bytecode *pc = top_.pc;

     for (Frame *frame = top_.fp; frame; frame = frame->prev()) {

         if (!frame->isNative())

             TraceFunctionFrame(visitor, frame, sp, pc);

         frame->accept(visitor);

         sp = reinterpret_cast<LocalSlot *>(frame);

         pc = frame->prevpc();

     }

 }

 bool

 Stack::isInCallStack(Handle<Module> module)

 {

     for (Frame *frame = top_.fp; frame; frame = frame->prev()) {

         if (frame->code()->module() == module)

             return true;

     }

     return false;

 }

 StackIter::StackIter(Zone *zone)

   : pc_(zone->stack().top()->pc),

     fp_(zone->stack().top()->fp)

 {

 }

 bool

 StackIter::done() const

 {

     return !fp_;

 }

 const char *

 StackIter::function() const

 {

     if (fp_->isNative())

         return fp_->native()->name()->chars();

     if (Function *fun = fp_->maybeFun())

         return fun->name()->chars();

     return "<toplevel>";

 }

 unsigned

 StackIter::lineNumber() const

 {

     if (fp_->isNative())

         return 0;

     if (fp_->maybeFun()) {

         unsigned offs = fp_->code()->pcToOffset(pc_);

         return fp_->code()->pcToSource(offs).line;

     }

     return fp_->code()->lineNumber();

 }

 const char *

 StackIter::source() const

 {

     if (fp_->isNative())

         return "<native>";

     return fp_->code()->module()->file()

            ? fp_->code()->module()->file()->chars()

            : "(anonymous)";

 }

 void

 StackIter::prev()

 {

     pc_ = fp_->prevpc();

     fp_ = fp_->prev();

 }