/*

 * 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.

 *

 */

#include "precompiled.hpp"

#include "ci/ciArrayKlass.hpp"

#include "ci/ciEnv.hpp"

#include "ci/ciKlass.hpp"

#include "ci/ciMethod.hpp"

#include "code/dependencies.hpp"

#include "compiler/compileLog.hpp"

#include "oops/oop.inline.hpp"

#include "runtime/handles.inline.hpp"

#include "utilities/copy.hpp"

#ifdef ASSERT

static bool must_be_in_vm() {

  Thread* thread = Thread::current();

  if (thread->is_Java_thread())

    return ((JavaThread*)thread)->thread_state() == _thread_in_vm;

  else

    return true;  //something like this: thread->is_VM_thread();

}

#endif //ASSERT

void Dependencies::initialize(ciEnv* env) {

  Arena* arena = env->arena();

  _oop_recorder = env->oop_recorder();

  _log = env->log();

  _dep_seen = new(arena) GrowableArray<int>(arena, 500, 0, 0);

  DEBUG_ONLY(_deps[end_marker] = NULL);

  for (int i = (int)FIRST_TYPE; i < (int)TYPE_LIMIT; i++) {

    _deps[i] = new(arena) GrowableArray<ciObject*>(arena, 10, 0, 0);

  }

  _content_bytes = NULL;

  _size_in_bytes = (size_t)-1;

  assert(TYPE_LIMIT <= (1<<LG2_TYPE_LIMIT), "sanity");

}

void Dependencies::assert_evol_method(ciMethod* m) {

  assert_common_1(evol_method, m);

}

void Dependencies::assert_leaf_type(ciKlass* ctxk) {

  if (ctxk->is_array_klass()) {

    // As a special case, support this assertion on an array type,

    // which reduces to an assertion on its element type.

    // Note that this cannot be done with assertions that

    // relate to concreteness or abstractness.

    ciType* elemt = ctxk->as_array_klass()->base_element_type();

    if (!elemt->is_instance_klass())  return;   // Ex:  int[][]

    ctxk = elemt->as_instance_klass();

    //if (ctxk->is_final())  return;            // Ex:  String[][]

  }

  check_ctxk(ctxk);

  assert_common_1(leaf_type, ctxk);

}

void Dependencies::assert_abstract_with_unique_concrete_subtype(ciKlass* ctxk, ciKlass* conck) {

  check_ctxk_abstract(ctxk);

  assert_common_2(abstract_with_unique_concrete_subtype, ctxk, conck);

}

void Dependencies::assert_abstract_with_no_concrete_subtype(ciKlass* ctxk) {

  check_ctxk_abstract(ctxk);

  assert_common_1(abstract_with_no_concrete_subtype, ctxk);

}

void Dependencies::assert_concrete_with_no_concrete_subtype(ciKlass* ctxk) {

  check_ctxk_concrete(ctxk);

  assert_common_1(concrete_with_no_concrete_subtype, ctxk);

}

void Dependencies::assert_unique_concrete_method(ciKlass* ctxk, ciMethod* uniqm) {

  check_ctxk(ctxk);

  assert_common_2(unique_concrete_method, ctxk, uniqm);

}

void Dependencies::assert_abstract_with_exclusive_concrete_subtypes(ciKlass* ctxk, ciKlass* k1, ciKlass* k2) {

  check_ctxk(ctxk);

  assert_common_3(abstract_with_exclusive_concrete_subtypes_2, ctxk, k1, k2);

}

void Dependencies::assert_exclusive_concrete_methods(ciKlass* ctxk, ciMethod* m1, ciMethod* m2) {

  check_ctxk(ctxk);

  assert_common_3(exclusive_concrete_methods_2, ctxk, m1, m2);

}

void Dependencies::assert_has_no_finalizable_subclasses(ciKlass* ctxk) {

  check_ctxk(ctxk);

  assert_common_1(no_finalizable_subclasses, ctxk);

}

// Helper function.  If we are adding a new dep. under ctxk2,

// try to find an old dep. under a broader* ctxk1.  If there is

//

bool Dependencies::maybe_merge_ctxk(GrowableArray<ciObject*>* deps,

                                    int ctxk_i, ciKlass* ctxk2) {

  ciKlass* ctxk1 = deps->at(ctxk_i)->as_klass();

  if (ctxk2->is_subtype_of(ctxk1)) {

    return true;  // success, and no need to change

  } else if (ctxk1->is_subtype_of(ctxk2)) {

    // new context class fully subsumes previous one

    deps->at_put(ctxk_i, ctxk2);

    return true;

  } else {

    return false;

  }

}

void Dependencies::assert_common_1(Dependencies::DepType dept, ciObject* x) {

  assert(dep_args(dept) == 1, "sanity");

  log_dependency(dept, x);

  GrowableArray<ciObject*>* deps = _deps[dept];

  // see if the same (or a similar) dep is already recorded

  if (note_dep_seen(dept, x)) {

    assert(deps->find(x) >= 0, "sanity");

  } else {

    deps->append(x);

  }

}

void Dependencies::assert_common_2(Dependencies::DepType dept,

                                   ciKlass* ctxk, ciObject* x) {

  assert(dep_context_arg(dept) == 0, "sanity");

  assert(dep_args(dept) == 2, "sanity");

  log_dependency(dept, ctxk, x);

  GrowableArray<ciObject*>* deps = _deps[dept];

  // see if the same (or a similar) dep is already recorded

  if (note_dep_seen(dept, x)) {

    // look in this bucket for redundant assertions

    const int stride = 2;

    for (int i = deps->length(); (i -= stride) >= 0; ) {

      ciObject* x1 = deps->at(i+1);

      if (x == x1) {  // same subject; check the context

        if (maybe_merge_ctxk(deps, i+0, ctxk)) {

          return;

        }

      }

    }

  }

  // append the assertion in the correct bucket:

  deps->append(ctxk);

  deps->append(x);

}

void Dependencies::assert_common_3(Dependencies::DepType dept,

                                   ciKlass* ctxk, ciObject* x, ciObject* x2) {

  assert(dep_context_arg(dept) == 0, "sanity");

  assert(dep_args(dept) == 3, "sanity");

  log_dependency(dept, ctxk, x, x2);

  GrowableArray<ciObject*>* deps = _deps[dept];

  // try to normalize an unordered pair:

  bool swap = false;

  switch (dept) {

  case abstract_with_exclusive_concrete_subtypes_2:

    swap = (x->ident() > x2->ident() && x != ctxk);

    break;

  case exclusive_concrete_methods_2:

    swap = (x->ident() > x2->ident() && x->as_method()->holder() != ctxk);

    break;

  }

  if (swap) { ciObject* t = x; x = x2; x2 = t; }

  // see if the same (or a similar) dep is already recorded

  if (note_dep_seen(dept, x) && note_dep_seen(dept, x2)) {

    // look in this bucket for redundant assertions

    const int stride = 3;

    for (int i = deps->length(); (i -= stride) >= 0; ) {

      ciObject* y  = deps->at(i+1);

      ciObject* y2 = deps->at(i+2);

      if (x == y && x2 == y2) {  // same subjects; check the context

        if (maybe_merge_ctxk(deps, i+0, ctxk)) {

          return;

        }

      }

    }

  }

  // append the assertion in the correct bucket:

  deps->append(ctxk);

  deps->append(x);

  deps->append(x2);

}

/// Support for encoding dependencies into an nmethod:

void Dependencies::copy_to(nmethod* nm) {

  address beg = nm->dependencies_begin();

  address end = nm->dependencies_end();

  guarantee(end - beg >= (ptrdiff_t) size_in_bytes(), "bad sizing");

  Copy::disjoint_words((HeapWord*) content_bytes(),

                       (HeapWord*) beg,

                       size_in_bytes() / sizeof(HeapWord));

  assert(size_in_bytes() % sizeof(HeapWord) == 0, "copy by words");

}

static int sort_dep(ciObject** p1, ciObject** p2, int narg) {

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

    int diff = p1[i]->ident() - p2[i]->ident();

    if (diff != 0)  return diff;

  }

  return 0;

}

static int sort_dep_arg_1(ciObject** p1, ciObject** p2)

{ return sort_dep(p1, p2, 1); }

static int sort_dep_arg_2(ciObject** p1, ciObject** p2)

{ return sort_dep(p1, p2, 2); }

static int sort_dep_arg_3(ciObject** p1, ciObject** p2)

{ return sort_dep(p1, p2, 3); }

void Dependencies::sort_all_deps() {

  for (int deptv = (int)FIRST_TYPE; deptv < (int)TYPE_LIMIT; deptv++) {

    DepType dept = (DepType)deptv;

    GrowableArray<ciObject*>* deps = _deps[dept];

    if (deps->length() <= 1)  continue;

    switch (dep_args(dept)) {

    case 1: deps->sort(sort_dep_arg_1, 1); break;

    case 2: deps->sort(sort_dep_arg_2, 2); break;

    case 3: deps->sort(sort_dep_arg_3, 3); break;

    default: ShouldNotReachHere();

    }

  }

}

size_t Dependencies::estimate_size_in_bytes() {

  size_t est_size = 100;

  for (int deptv = (int)FIRST_TYPE; deptv < (int)TYPE_LIMIT; deptv++) {

    DepType dept = (DepType)deptv;

    GrowableArray<ciObject*>* deps = _deps[dept];

    est_size += deps->length()*2;  // tags and argument(s)

  }

  return est_size;

}

ciKlass* Dependencies::ctxk_encoded_as_null(DepType dept, ciObject* x) {

  switch (dept) {

  case abstract_with_exclusive_concrete_subtypes_2:

    return x->as_klass();

  case unique_concrete_method:

  case exclusive_concrete_methods_2:

    return x->as_method()->holder();

  }

  return NULL;  // let NULL be NULL

}

klassOop Dependencies::ctxk_encoded_as_null(DepType dept, oop x) {

  assert(must_be_in_vm(), "raw oops here");

  switch (dept) {

  case abstract_with_exclusive_concrete_subtypes_2:

    assert(x->is_klass(), "sanity");

    return (klassOop) x;

  case unique_concrete_method:

  case exclusive_concrete_methods_2:

    assert(x->is_method(), "sanity");

    return ((methodOop)x)->method_holder();

  }

  return NULL;  // let NULL be NULL

}

void Dependencies::encode_content_bytes() {

  sort_all_deps();

  // cast is safe, no deps can overflow INT_MAX

  CompressedWriteStream bytes((int)estimate_size_in_bytes());

  for (int deptv = (int)FIRST_TYPE; deptv < (int)TYPE_LIMIT; deptv++) {

    DepType dept = (DepType)deptv;

    GrowableArray<ciObject*>* deps = _deps[dept];

    if (deps->length() == 0)  continue;

    int stride = dep_args(dept);

    int ctxkj  = dep_context_arg(dept);  // -1 if no context arg

    assert(stride > 0, "sanity");

    for (int i = 0; i < deps->length(); i += stride) {

      jbyte code_byte = (jbyte)dept;

      int skipj = -1;

      if (ctxkj >= 0 && ctxkj+1 < stride) {

        ciKlass*  ctxk = deps->at(i+ctxkj+0)->as_klass();

        ciObject* x    = deps->at(i+ctxkj+1);  // following argument

        if (ctxk == ctxk_encoded_as_null(dept, x)) {

          skipj = ctxkj;  // we win:  maybe one less oop to keep track of

          code_byte |= default_context_type_bit;

        }

      }

      bytes.write_byte(code_byte);

      for (int j = 0; j < stride; j++) {

        if (j == skipj)  continue;

        bytes.write_int(_oop_recorder->find_index(deps->at(i+j)->constant_encoding()));

      }

    }

  }

  // write a sentinel byte to mark the end

  bytes.write_byte(end_marker);

  // round it out to a word boundary

  while (bytes.position() % sizeof(HeapWord) != 0) {

    bytes.write_byte(end_marker);

  }

  // check whether the dept byte encoding really works

  assert((jbyte)default_context_type_bit != 0, "byte overflow");

  _content_bytes = bytes.buffer();

  _size_in_bytes = bytes.position();

}

const char* Dependencies::_dep_name[TYPE_LIMIT] = {

  "end_marker",

  "evol_method",

  "leaf_type",

  "abstract_with_unique_concrete_subtype",

  "abstract_with_no_concrete_subtype",

  "concrete_with_no_concrete_subtype",

  "unique_concrete_method",

  "abstract_with_exclusive_concrete_subtypes_2",

  "exclusive_concrete_methods_2",

  "no_finalizable_subclasses"

};

int Dependencies::_dep_args[TYPE_LIMIT] = {

  -1,// end_marker

  1, // evol_method m

  1, // leaf_type ctxk

  2, // abstract_with_unique_concrete_subtype ctxk, k

  1, // abstract_with_no_concrete_subtype ctxk

  1, // concrete_with_no_concrete_subtype ctxk

  2, // unique_concrete_method ctxk, m

  3, // unique_concrete_subtypes_2 ctxk, k1, k2

  3, // unique_concrete_methods_2 ctxk, m1, m2

  1  // no_finalizable_subclasses ctxk

};

const char* Dependencies::dep_name(Dependencies::DepType dept) {

  if (!dept_in_mask(dept, all_types))  return "?bad-dep?";

  return _dep_name[dept];

}

int Dependencies::dep_args(Dependencies::DepType dept) {

  if (!dept_in_mask(dept, all_types))  return -1;

  return _dep_args[dept];

}

// for the sake of the compiler log, print out current dependencies:

void Dependencies::log_all_dependencies() {

  if (log() == NULL)  return;

  ciObject* args[max_arg_count];

  for (int deptv = (int)FIRST_TYPE; deptv < (int)TYPE_LIMIT; deptv++) {

    DepType dept = (DepType)deptv;

    GrowableArray<ciObject*>* deps = _deps[dept];

    if (deps->length() == 0)  continue;

    int stride = dep_args(dept);

    for (int i = 0; i < deps->length(); i += stride) {

      for (int j = 0; j < stride; j++) {

        // flush out the identities before printing

        args[j] = deps->at(i+j);

      }

      write_dependency_to(log(), dept, stride, args);

    }

  }

}

void Dependencies::write_dependency_to(CompileLog* log,

                                       DepType dept,

                                       int nargs, oop args[],

                                       klassOop witness) {

  if (log == NULL) {

    return;

  }

  ciEnv* env = ciEnv::current();

  ciObject* ciargs[max_arg_count];

  assert(nargs <= max_arg_count, "oob");

  for (int j = 0; j < nargs; j++) {

    ciargs[j] = env->get_object(args[j]);

  }

  Dependencies::write_dependency_to(log, dept, nargs, ciargs, witness);

}

void Dependencies::write_dependency_to(CompileLog* log,

                                       DepType dept,

                                       int nargs, ciObject* args[],

                                       klassOop witness) {

  if (log == NULL)  return;

  assert(nargs <= max_arg_count, "oob");

  int argids[max_arg_count];

  int ctxkj = dep_context_arg(dept);  // -1 if no context arg

  int j;

  for (j = 0; j < nargs; j++) {

    argids[j] = log->identify(args[j]);

  }

  if (witness != NULL) {

    log->begin_elem("dependency_failed");

  } else {

    log->begin_elem("dependency");

  }

  log->print(" type='%s'", dep_name(dept));

  if (ctxkj >= 0) {

    log->print(" ctxk='%d'", argids[ctxkj]);

  }

  // write remaining arguments, if any.

  for (j = 0; j < nargs; j++) {

    if (j == ctxkj)  continue;  // already logged

    if (j == 1) {

      log->print(  " x='%d'",    argids[j]);

    } else {

      log->print(" x%d='%d'", j, argids[j]);

    }

  }

  if (witness != NULL) {

    log->object("witness", witness);

    log->stamp();

  }

  log->end_elem();

}

void Dependencies::write_dependency_to(xmlStream* xtty,

                                       DepType dept,

                                       int nargs, oop args[],

                                       klassOop witness) {

  if (xtty == NULL)  return;

  ttyLocker ttyl;

  int ctxkj = dep_context_arg(dept);  // -1 if no context arg

  if (witness != NULL) {

    xtty->begin_elem("dependency_failed");

  } else {

    xtty->begin_elem("dependency");

  }

  xtty->print(" type='%s'", dep_name(dept));

  if (ctxkj >= 0) {

    xtty->object("ctxk", args[ctxkj]);

  }

  // write remaining arguments, if any.

  for (int j = 0; j < nargs; j++) {

    if (j == ctxkj)  continue;  // already logged

    if (j == 1) {

      xtty->object("x", args[j]);

    } else {

      char xn[10]; sprintf(xn, "x%d", j);

      xtty->object(xn, args[j]);

    }

  }

  if (witness != NULL) {

    xtty->object("witness", witness);

    xtty->stamp();

  }

  xtty->end_elem();

}

void Dependencies::print_dependency(DepType dept, int nargs, oop args[],

                                    klassOop witness) {

  ResourceMark rm;

  ttyLocker ttyl;   // keep the following output all in one block

  tty->print_cr("%s of type %s",

                (witness == NULL)? "Dependency": "Failed dependency",

                dep_name(dept));

  // print arguments

  int ctxkj = dep_context_arg(dept);  // -1 if no context arg

  for (int j = 0; j < nargs; j++) {

    oop arg = args[j];

    bool put_star = false;

    if (arg == NULL)  continue;

    const char* what;

    if (j == ctxkj) {

      what = "context";

      put_star = !Dependencies::is_concrete_klass((klassOop)arg);

    } else if (arg->is_method()) {

      what = "method ";

      put_star = !Dependencies::is_concrete_method((methodOop)arg);

    } else if (arg->is_klass()) {

      what = "class  ";

    } else {

      what = "object ";

    }

    tty->print("  %s = %s", what, (put_star? "*": ""));

    if (arg->is_klass())

      tty->print("%s", Klass::cast((klassOop)arg)->external_name());

    else

      arg->print_value();

    tty->cr();

  }

  if (witness != NULL) {

    bool put_star = !Dependencies::is_concrete_klass(witness);

    tty->print_cr("  witness = %s%s",

                  (put_star? "*": ""),

                  Klass::cast(witness)->external_name());

  }

}

void Dependencies::DepStream::log_dependency(klassOop witness) {

  if (_deps == NULL && xtty == NULL)  return;  // fast cutout for runtime

  int nargs = argument_count();

  oop args[max_arg_count];

  for (int j = 0; j < nargs; j++) {

    args[j] = argument(j);

  }

  if (_deps != NULL && _deps->log() != NULL) {

    Dependencies::write_dependency_to(_deps->log(),

                                      type(), nargs, args, witness);

  } else {

    Dependencies::write_dependency_to(xtty,

                                      type(), nargs, args, witness);