/*

 * Copyright (c) 2012, 2014, Oracle and/or its affiliates. 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 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 "memory/universe.hpp"

#include "oops/oop.inline.hpp"

#include "classfile/classLoaderData.hpp"

#include "prims/whitebox.hpp"

#include "prims/wbtestmethods/parserTests.hpp"

#include "runtime/arguments.hpp"

#include "runtime/interfaceSupport.hpp"

#include "runtime/os.hpp"

#include "utilities/debug.hpp"

#include "utilities/macros.hpp"

#include "utilities/exceptions.hpp"

#if INCLUDE_ALL_GCS

#include "gc_implementation/g1/concurrentMark.hpp"

#include "gc_implementation/g1/g1CollectedHeap.inline.hpp"

#include "gc_implementation/g1/heapRegionRemSet.hpp"

#endif // INCLUDE_ALL_GCS

#ifdef INCLUDE_NMT

#include "services/memTracker.hpp"

#endif // INCLUDE_NMT

#include "compiler/compileBroker.hpp"

#include "runtime/compilationPolicy.hpp"

#define SIZE_T_MAX_VALUE ((size_t) -1)

bool WhiteBox::_used = false;

WB_ENTRY(jlong, WB_GetObjectAddress(JNIEnv* env, jobject o, jobject obj))

  return (jlong)(void*)JNIHandles::resolve(obj);

WB_END

WB_ENTRY(jint, WB_GetHeapOopSize(JNIEnv* env, jobject o))

  return heapOopSize;

WB_END

class WBIsKlassAliveClosure : public KlassClosure {

    Symbol* _name;

    bool _found;

public:

    WBIsKlassAliveClosure(Symbol* name) : _name(name), _found(false) {}

    void do_klass(Klass* k) {

      if (_found) return;

      Symbol* ksym = k->name();

      if (ksym->fast_compare(_name) == 0) {

        _found = true;

      }

    }

    bool found() const {

        return _found;

    }

};

WB_ENTRY(jboolean, WB_IsClassAlive(JNIEnv* env, jobject target, jstring name))

  Handle h_name = JNIHandles::resolve(name);

  if (h_name.is_null()) return false;

  Symbol* sym = java_lang_String::as_symbol(h_name, CHECK_false);

  TempNewSymbol tsym(sym); // Make sure to decrement reference count on sym on return

  WBIsKlassAliveClosure closure(sym);

  ClassLoaderDataGraph::classes_do(&closure);

  return closure.found();

WB_END

WB_ENTRY(jlong, WB_GetCompressedOopsMaxHeapSize(JNIEnv* env, jobject o)) {

  return (jlong)Arguments::max_heap_for_compressed_oops();

}

WB_END

WB_ENTRY(void, WB_PrintHeapSizes(JNIEnv* env, jobject o)) {

  CollectorPolicy * p = Universe::heap()->collector_policy();

  gclog_or_tty->print_cr("Minimum heap "SIZE_FORMAT" Initial heap "

    SIZE_FORMAT" Maximum heap "SIZE_FORMAT" Space alignment "SIZE_FORMAT" Heap alignment "SIZE_FORMAT,

    p->min_heap_byte_size(), p->initial_heap_byte_size(), p->max_heap_byte_size(),

    p->space_alignment(), p->heap_alignment());

}

WB_END

#ifndef PRODUCT

// Forward declaration

void TestReservedSpace_test();

void TestReserveMemorySpecial_test();

void TestVirtualSpace_test();

void TestMetaspaceAux_test();

#endif

WB_ENTRY(void, WB_RunMemoryUnitTests(JNIEnv* env, jobject o))

#ifndef PRODUCT

  TestReservedSpace_test();

  TestReserveMemorySpecial_test();

  TestVirtualSpace_test();

  TestMetaspaceAux_test();

#endif

WB_END

WB_ENTRY(void, WB_ReadFromNoaccessArea(JNIEnv* env, jobject o))

  size_t granularity = os::vm_allocation_granularity();

  ReservedHeapSpace rhs(100 * granularity, granularity, false, NULL);

  VirtualSpace vs;

  vs.initialize(rhs, 50 * granularity);

  //Check if constraints are complied

  if (!( UseCompressedOops && rhs.base() != NULL &&

         Universe::narrow_oop_base() != NULL &&

         Universe::narrow_oop_use_implicit_null_checks() )) {

    tty->print_cr("WB_ReadFromNoaccessArea method is useless:\n "

                  "\tUseCompressedOops is %d\n"

                  "\trhs.base() is "PTR_FORMAT"\n"

                  "\tUniverse::narrow_oop_base() is "PTR_FORMAT"\n"

                  "\tUniverse::narrow_oop_use_implicit_null_checks() is %d",

                  UseCompressedOops,

                  rhs.base(),

                  Universe::narrow_oop_base(),

                  Universe::narrow_oop_use_implicit_null_checks());

    return;

  }

  tty->print_cr("Reading from no access area... ");

  tty->print_cr("*(vs.low_boundary() - rhs.noaccess_prefix() / 2 ) = %c",

                *(vs.low_boundary() - rhs.noaccess_prefix() / 2 ));

WB_END

static jint wb_stress_virtual_space_resize(size_t reserved_space_size,

                                           size_t magnitude, size_t iterations) {

  size_t granularity = os::vm_allocation_granularity();

  ReservedHeapSpace rhs(reserved_space_size * granularity, granularity, false, NULL);

  VirtualSpace vs;

  if (!vs.initialize(rhs, 0)) {

    tty->print_cr("Failed to initialize VirtualSpace. Can't proceed.");

    return 3;

  }

  long seed = os::random();

  tty->print_cr("Random seed is %ld", seed);

  os::init_random(seed);

  for (size_t i = 0; i < iterations; i++) {

    // Whether we will shrink or grow

    bool shrink = os::random() % 2L == 0;

    // Get random delta to resize virtual space

    size_t delta = (size_t)os::random() % magnitude;

    // If we are about to shrink virtual space below zero, then expand instead

    if (shrink && vs.committed_size() < delta) {

      shrink = false;

    }

    // Resizing by delta

    if (shrink) {

      vs.shrink_by(delta);

    } else {

      // If expanding fails expand_by will silently return false

      vs.expand_by(delta, true);

    }

  }

  return 0;

}

WB_ENTRY(jint, WB_StressVirtualSpaceResize(JNIEnv* env, jobject o,

        jlong reserved_space_size, jlong magnitude, jlong iterations))

  tty->print_cr("reservedSpaceSize="JLONG_FORMAT", magnitude="JLONG_FORMAT", "

                "iterations="JLONG_FORMAT"\n", reserved_space_size, magnitude,

                iterations);

  if (reserved_space_size < 0 || magnitude < 0 || iterations < 0) {

    tty->print_cr("One of variables printed above is negative. Can't proceed.\n");

    return 1;

  }

  // sizeof(size_t) depends on whether OS is 32bit or 64bit. sizeof(jlong) is

  // always 8 byte. That's why we should avoid overflow in case of 32bit platform.

  if (sizeof(size_t) < sizeof(jlong)) {

    jlong size_t_max_value = (jlong) SIZE_T_MAX_VALUE;

    if (reserved_space_size > size_t_max_value || magnitude > size_t_max_value

        || iterations > size_t_max_value) {

      tty->print_cr("One of variables printed above overflows size_t. Can't proceed.\n");

      return 2;

    }

  }

  return wb_stress_virtual_space_resize((size_t) reserved_space_size,

                                        (size_t) magnitude, (size_t) iterations);

WB_END

#if INCLUDE_ALL_GCS

WB_ENTRY(jboolean, WB_G1IsHumongous(JNIEnv* env, jobject o, jobject obj))

  G1CollectedHeap* g1 = G1CollectedHeap::heap();

  oop result = JNIHandles::resolve(obj);

  const HeapRegion* hr = g1->heap_region_containing(result);

  return hr->isHumongous();

WB_END

WB_ENTRY(jlong, WB_G1NumFreeRegions(JNIEnv* env, jobject o))

  G1CollectedHeap* g1 = G1CollectedHeap::heap();

  size_t nr = g1->free_regions();

  return (jlong)nr;

WB_END

WB_ENTRY(jboolean, WB_G1InConcurrentMark(JNIEnv* env, jobject o))

  G1CollectedHeap* g1 = G1CollectedHeap::heap();

  ConcurrentMark* cm = g1->concurrent_mark();

  return cm->concurrent_marking_in_progress();

WB_END

WB_ENTRY(jint, WB_G1RegionSize(JNIEnv* env, jobject o))

  return (jint)HeapRegion::GrainBytes;

WB_END

#endif // INCLUDE_ALL_GCS

#if INCLUDE_NMT

// Alloc memory using the test memory type so that we can use that to see if

// NMT picks it up correctly

WB_ENTRY(jlong, WB_NMTMalloc(JNIEnv* env, jobject o, jlong size))

  jlong addr = 0;

  if (MemTracker::is_on() && !MemTracker::shutdown_in_progress()) {

    addr = (jlong)(uintptr_t)os::malloc(size, mtTest);

  }

  return addr;

WB_END

// Free the memory allocated by NMTAllocTest

WB_ENTRY(void, WB_NMTFree(JNIEnv* env, jobject o, jlong mem))

  os::free((void*)(uintptr_t)mem, mtTest);

WB_END

WB_ENTRY(jlong, WB_NMTReserveMemory(JNIEnv* env, jobject o, jlong size))

  jlong addr = 0;

  if (MemTracker::is_on() && !MemTracker::shutdown_in_progress()) {

    addr = (jlong)(uintptr_t)os::reserve_memory(size);

    MemTracker::record_virtual_memory_type((address)addr, mtTest);

  }

  return addr;

WB_END

WB_ENTRY(void, WB_NMTCommitMemory(JNIEnv* env, jobject o, jlong addr, jlong size))

  os::commit_memory((char *)(uintptr_t)addr, size, !ExecMem);

  MemTracker::record_virtual_memory_type((address)(uintptr_t)addr, mtTest);

WB_END

WB_ENTRY(void, WB_NMTUncommitMemory(JNIEnv* env, jobject o, jlong addr, jlong size))

  os::uncommit_memory((char *)(uintptr_t)addr, size);

WB_END

WB_ENTRY(void, WB_NMTReleaseMemory(JNIEnv* env, jobject o, jlong addr, jlong size))

  os::release_memory((char *)(uintptr_t)addr, size);

WB_END

// Block until the current generation of NMT data to be merged, used to reliably test the NMT feature

WB_ENTRY(jboolean, WB_NMTWaitForDataMerge(JNIEnv* env))

  if (!MemTracker::is_on() || MemTracker::shutdown_in_progress()) {

    return false;

  }

  return MemTracker::wbtest_wait_for_data_merge();

WB_END

WB_ENTRY(jboolean, WB_NMTIsDetailSupported(JNIEnv* env))

  return MemTracker::tracking_level() == MemTracker::NMT_detail;

WB_END

#endif // INCLUDE_NMT

static jmethodID reflected_method_to_jmid(JavaThread* thread, JNIEnv* env, jobject method) {

  assert(method != NULL, "method should not be null");

  ThreadToNativeFromVM ttn(thread);

  return env->FromReflectedMethod(method);

}

WB_ENTRY(void, WB_DeoptimizeAll(JNIEnv* env, jobject o))

  MutexLockerEx mu(Compile_lock);

  CodeCache::mark_all_nmethods_for_deoptimization();

  VM_Deoptimize op;

  VMThread::execute(&op);

WB_END

WB_ENTRY(jint, WB_DeoptimizeMethod(JNIEnv* env, jobject o, jobject method, jboolean is_osr))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  int result = 0;

  CHECK_JNI_EXCEPTION_(env, result);

  MutexLockerEx mu(Compile_lock);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  nmethod* code;

  if (is_osr) {

    int bci = InvocationEntryBci;

    while ((code = mh->lookup_osr_nmethod_for(bci, CompLevel_none, false)) != NULL) {

      code->mark_for_deoptimization();

      ++result;

      bci = code->osr_entry_bci() + 1;

    }

  } else {

    code = mh->code();

  }

  if (code != NULL) {

    code->mark_for_deoptimization();

    ++result;

  }

  result += CodeCache::mark_for_deoptimization(mh());

  if (result > 0) {

    VM_Deoptimize op;

    VMThread::execute(&op);

  }

  return result;

WB_END

WB_ENTRY(jboolean, WB_IsMethodCompiled(JNIEnv* env, jobject o, jobject method, jboolean is_osr))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, JNI_FALSE);

  MutexLockerEx mu(Compile_lock);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  nmethod* code = is_osr ? mh->lookup_osr_nmethod_for(InvocationEntryBci, CompLevel_none, false) : mh->code();

  if (code == NULL) {

    return JNI_FALSE;

  }

  return (code->is_alive() && !code->is_marked_for_deoptimization());

WB_END

WB_ENTRY(jboolean, WB_IsMethodCompilable(JNIEnv* env, jobject o, jobject method, jint comp_level, jboolean is_osr))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, JNI_FALSE);

  MutexLockerEx mu(Compile_lock);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  if (is_osr) {

    return CompilationPolicy::can_be_osr_compiled(mh, comp_level);

  } else {

    return CompilationPolicy::can_be_compiled(mh, comp_level);

  }

WB_END

WB_ENTRY(jboolean, WB_IsMethodQueuedForCompilation(JNIEnv* env, jobject o, jobject method))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, JNI_FALSE);

  MutexLockerEx mu(Compile_lock);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  return mh->queued_for_compilation();

WB_END

WB_ENTRY(jint, WB_GetMethodCompilationLevel(JNIEnv* env, jobject o, jobject method, jboolean is_osr))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, CompLevel_none);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  nmethod* code = is_osr ? mh->lookup_osr_nmethod_for(InvocationEntryBci, CompLevel_none, false) : mh->code();

  return (code != NULL ? code->comp_level() : CompLevel_none);

WB_END

WB_ENTRY(void, WB_MakeMethodNotCompilable(JNIEnv* env, jobject o, jobject method, jint comp_level, jboolean is_osr))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION(env);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  if (is_osr) {

    mh->set_not_osr_compilable(comp_level, true /* report */, "WhiteBox");

  } else {

    mh->set_not_compilable(comp_level, true /* report */, "WhiteBox");

  }

WB_END

WB_ENTRY(jint, WB_GetMethodEntryBci(JNIEnv* env, jobject o, jobject method))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, InvocationEntryBci);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  nmethod* code = mh->lookup_osr_nmethod_for(InvocationEntryBci, CompLevel_none, false);

  return (code != NULL && code->is_osr_method() ? code->osr_entry_bci() : InvocationEntryBci);

WB_END

WB_ENTRY(jboolean, WB_TestSetDontInlineMethod(JNIEnv* env, jobject o, jobject method, jboolean value))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, JNI_FALSE);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  bool result = mh->dont_inline();

  mh->set_dont_inline(value == JNI_TRUE);

  return result;

WB_END

WB_ENTRY(jint, WB_GetCompileQueueSize(JNIEnv* env, jobject o, jint comp_level))

  if (comp_level == CompLevel_any) {

    return CompileBroker::queue_size(CompLevel_full_optimization) /* C2 */ +

        CompileBroker::queue_size(CompLevel_full_profile) /* C1 */;

  } else {

    return CompileBroker::queue_size(comp_level);

  }

WB_END

WB_ENTRY(jboolean, WB_TestSetForceInlineMethod(JNIEnv* env, jobject o, jobject method, jboolean value))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, JNI_FALSE);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  bool result = mh->force_inline();

  mh->set_force_inline(value == JNI_TRUE);

  return result;

WB_END

WB_ENTRY(jboolean, WB_EnqueueMethodForCompilation(JNIEnv* env, jobject o, jobject method, jint comp_level, jint bci))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION_(env, JNI_FALSE);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  nmethod* nm = CompileBroker::compile_method(mh, bci, comp_level, mh, mh->invocation_count(), "WhiteBox", THREAD);

  MutexLockerEx mu(Compile_lock);

  return (mh->queued_for_compilation() || nm != NULL);

WB_END

class VM_WhiteBoxOperation : public VM_Operation {

 public:

  VM_WhiteBoxOperation()                         { }

  VMOp_Type type()                  const        { return VMOp_WhiteBoxOperation; }

  bool allow_nested_vm_operations() const        { return true; }

};

class AlwaysFalseClosure : public BoolObjectClosure {

 public:

  bool do_object_b(oop p) { return false; }

};

static AlwaysFalseClosure always_false;

class VM_WhiteBoxCleanMethodData : public VM_WhiteBoxOperation {

 public:

  VM_WhiteBoxCleanMethodData(MethodData* mdo) : _mdo(mdo) { }

  void doit() {

    _mdo->clean_method_data(&always_false);

  }

 private:

  MethodData* _mdo;

};

WB_ENTRY(void, WB_ClearMethodState(JNIEnv* env, jobject o, jobject method))

  jmethodID jmid = reflected_method_to_jmid(thread, env, method);

  CHECK_JNI_EXCEPTION(env);

  methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));

  MutexLockerEx mu(Compile_lock);

  MethodData* mdo = mh->method_data();

  MethodCounters* mcs = mh->method_counters();

  if (mdo != NULL) {

    mdo->init();

    ResourceMark rm;

    int arg_count = mdo->method()->size_of_parameters();

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

      mdo->set_arg_modified(i, 0);

    }

    VM_WhiteBoxCleanMethodData op(mdo);

    VMThread::execute(&op);

  }

  mh->clear_not_c1_compilable();

  mh->clear_not_c2_compilable();

  mh->clear_not_c2_osr_compilable();

  NOT_PRODUCT(mh->set_compiled_invocation_count(0));

  if (mcs != NULL) {

    mcs->backedge_counter()->init();

    mcs->invocation_counter()->init();

    mcs->set_interpreter_invocation_count(0);

    mcs->set_interpreter_throwout_count(0);