/*

 * Copyright (c) 2005, 2019, 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 "classfile/classLoaderDataGraph.hpp"

#include "jfr/jfrEvents.hpp"

#include "jfr/support/jfrThreadId.hpp"

#include "logging/log.hpp"

#include "memory/resourceArea.hpp"

#include "oops/klass.inline.hpp"

#include "oops/markWord.hpp"

#include "oops/oop.inline.hpp"

#include "runtime/atomic.hpp"

#include "runtime/basicLock.hpp"

#include "runtime/biasedLocking.hpp"

#include "runtime/handles.inline.hpp"

#include "runtime/task.hpp"

#include "runtime/threadSMR.hpp"

#include "runtime/vframe.hpp"

#include "runtime/vmThread.hpp"

#include "runtime/vmOperations.hpp"

static bool _biased_locking_enabled = false;

BiasedLockingCounters BiasedLocking::_counters;

static GrowableArray<Handle>*   _preserved_oop_stack  = NULL;

static GrowableArray<markWord>* _preserved_mark_stack = NULL;

static void enable_biased_locking(InstanceKlass* k) {

  k->set_prototype_header(markWord::biased_locking_prototype());

}

static void enable_biased_locking() {

  _biased_locking_enabled = true;

  log_info(biasedlocking)("Biased locking enabled");

}

class VM_EnableBiasedLocking: public VM_Operation {

 public:

  VM_EnableBiasedLocking() {}

  VMOp_Type type() const          { return VMOp_EnableBiasedLocking; }

  void doit() {

    // Iterate the class loader data dictionaries enabling biased locking for all

    // currently loaded classes.

    ClassLoaderDataGraph::dictionary_classes_do(enable_biased_locking);

    // Indicate that future instances should enable it as well

    enable_biased_locking();

  }

  bool allow_nested_vm_operations() const        { return false; }

};

// One-shot PeriodicTask subclass for enabling biased locking

class EnableBiasedLockingTask : public PeriodicTask {

 public:

  EnableBiasedLockingTask(size_t interval_time) : PeriodicTask(interval_time) {}

  virtual void task() {

    // Reclaim our storage and disenroll ourself

    delete this;

  }

};

void BiasedLocking::init() {

  // If biased locking is enabled and BiasedLockingStartupDelay is set,

  // schedule a task to fire after the specified delay which turns on

  // biased locking for all currently loaded classes as well as future

  // ones. This could be a workaround for startup time regressions

  // due to large number of safepoints being taken during VM startup for

  // bias revocation.

  if (UseBiasedLocking) {

    if (BiasedLockingStartupDelay > 0) {

      EnableBiasedLockingTask* task = new EnableBiasedLockingTask(BiasedLockingStartupDelay);

      task->enroll();

    } else {

      enable_biased_locking();

    }

  }

}

bool BiasedLocking::enabled() {

  assert(UseBiasedLocking, "precondition");

  // We check "BiasedLockingStartupDelay == 0" here to cover the

  // possibility of calls to BiasedLocking::enabled() before

  // BiasedLocking::init().

  return _biased_locking_enabled || BiasedLockingStartupDelay == 0;

}

// Returns MonitorInfos for all objects locked on this thread in youngest to oldest order

static GrowableArray<MonitorInfo*>* get_or_compute_monitor_info(JavaThread* thread) {

  GrowableArray<MonitorInfo*>* info = thread->cached_monitor_info();

  if (info != NULL) {

    return info;

  }

  info = new GrowableArray<MonitorInfo*>();

  // It's possible for the thread to not have any Java frames on it,

  // i.e., if it's the main thread and it's already returned from main()

  if (thread->has_last_Java_frame()) {

    RegisterMap rm(thread);

    for (javaVFrame* vf = thread->last_java_vframe(&rm); vf != NULL; vf = vf->java_sender()) {

      GrowableArray<MonitorInfo*> *monitors = vf->monitors();

      if (monitors != NULL) {

        int len = monitors->length();

        // Walk monitors youngest to oldest

        for (int i = len - 1; i >= 0; i--) {

          MonitorInfo* mon_info = monitors->at(i);

          if (mon_info->eliminated()) continue;

          oop owner = mon_info->owner();

          if (owner != NULL) {

            info->append(mon_info);

          }

        }

      }

    }

  }

  thread->set_cached_monitor_info(info);

  return info;

}

// After the call, *biased_locker will be set to obj->mark()->biased_locker() if biased_locker != NULL,

// AND it is a living thread. Otherwise it will not be updated, (i.e. the caller is responsible for initialization).

void BiasedLocking::single_revoke_at_safepoint(oop obj, bool is_bulk, JavaThread* requesting_thread, JavaThread** biased_locker) {

  assert(SafepointSynchronize::is_at_safepoint(), "must be done at safepoint");

  assert(Thread::current()->is_VM_thread(), "must be VMThread");

  markWord mark = obj->mark();

  if (!mark.has_bias_pattern()) {

    if (log_is_enabled(Info, biasedlocking)) {

      ResourceMark rm;

      log_info(biasedlocking)("  (Skipping revocation of object " INTPTR_FORMAT

                              ", mark " INTPTR_FORMAT ", type %s"

                              ", requesting thread " INTPTR_FORMAT

                              " because it's no longer biased)",

                              p2i((void *)obj), mark.value(),

                              obj->klass()->external_name(),

                              (intptr_t) requesting_thread);

    }

    return;

  }

  uint age = mark.age();

  markWord unbiased_prototype = markWord::prototype().set_age(age);

  // Log at "info" level if not bulk, else "trace" level

  if (!is_bulk) {

    ResourceMark rm;

    log_info(biasedlocking)("Revoking bias of object " INTPTR_FORMAT ", mark "

                            INTPTR_FORMAT ", type %s, prototype header " INTPTR_FORMAT

                            ", requesting thread " INTPTR_FORMAT,

                            p2i((void *)obj),

                            mark.value(),

                            obj->klass()->external_name(),

                            obj->klass()->prototype_header().value(),

                            (intptr_t) requesting_thread);

  } else {

    ResourceMark rm;

    log_trace(biasedlocking)("Revoking bias of object " INTPTR_FORMAT " , mark "

                             INTPTR_FORMAT " , type %s , prototype header " INTPTR_FORMAT

                             " , requesting thread " INTPTR_FORMAT,

                             p2i((void *)obj),

                             mark.value(),

                             obj->klass()->external_name(),

                             obj->klass()->prototype_header().value(),

                             (intptr_t) requesting_thread);

  }

  JavaThread* biased_thread = mark.biased_locker();

  if (biased_thread == NULL) {

    // Object is anonymously biased. We can get here if, for

    // example, we revoke the bias due to an identity hash code

    // being computed for an object.

    obj->set_mark(unbiased_prototype);

    // Log at "info" level if not bulk, else "trace" level

    if (!is_bulk) {

      log_info(biasedlocking)("  Revoked bias of anonymously-biased object");

    } else {

      log_trace(biasedlocking)("  Revoked bias of anonymously-biased object");

    }

    return;

  }

  // Handle case where the thread toward which the object was biased has exited

  bool thread_is_alive = false;

  if (requesting_thread == biased_thread) {

    thread_is_alive = true;

  } else {

    ThreadsListHandle tlh;

    thread_is_alive = tlh.includes(biased_thread);

  }

  if (!thread_is_alive) {

    obj->set_mark(unbiased_prototype);

    // Log at "info" level if not bulk, else "trace" level

    if (!is_bulk) {

      log_info(biasedlocking)("  Revoked bias of object biased toward dead thread ("

                              PTR_FORMAT ")", p2i(biased_thread));

    } else {

      log_trace(biasedlocking)("  Revoked bias of object biased toward dead thread ("

                               PTR_FORMAT ")", p2i(biased_thread));

    }

    return;

  }

  // Log at "info" level if not bulk, else "trace" level

  if (!is_bulk) {

    log_info(biasedlocking)("  Revoked bias of object biased toward live thread ("

                            PTR_FORMAT ")", p2i(biased_thread));

  } else {

    log_trace(biasedlocking)("  Revoked bias of object biased toward live thread ("

                               PTR_FORMAT ")", p2i(biased_thread));

  }

  // Thread owning bias is alive.

  // Check to see whether it currently owns the lock and, if so,

  // write down the needed displaced headers to the thread's stack.

  // Otherwise, restore the object's header either to the unlocked

  // or unbiased state.

  GrowableArray<MonitorInfo*>* cached_monitor_info = get_or_compute_monitor_info(biased_thread);

  BasicLock* highest_lock = NULL;

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

    MonitorInfo* mon_info = cached_monitor_info->at(i);

    if (mon_info->owner() == obj) {

      log_trace(biasedlocking)("   mon_info->owner (" PTR_FORMAT ") == obj (" PTR_FORMAT ")",

                               p2i((void *) mon_info->owner()),

                               p2i((void *) obj));

      // Assume recursive case and fix up highest lock below

      markWord mark = markWord::encode((BasicLock*) NULL);

      highest_lock = mon_info->lock();

      highest_lock->set_displaced_header(mark);

    } else {

      log_trace(biasedlocking)("   mon_info->owner (" PTR_FORMAT ") != obj (" PTR_FORMAT ")",

                               p2i((void *) mon_info->owner()),

                               p2i((void *) obj));

    }

  }

  if (highest_lock != NULL) {

    // Fix up highest lock to contain displaced header and point

    // object at it

    highest_lock->set_displaced_header(unbiased_prototype);

    // Reset object header to point to displaced mark.

    // Must release store the lock address for platforms without TSO

    // ordering (e.g. ppc).

    obj->release_set_mark(markWord::encode(highest_lock));

    assert(!obj->mark().has_bias_pattern(), "illegal mark state: stack lock used bias bit");

    // Log at "info" level if not bulk, else "trace" level

    if (!is_bulk) {

      log_info(biasedlocking)("  Revoked bias of currently-locked object");

    } else {

      log_trace(biasedlocking)("  Revoked bias of currently-locked object");

    }

  } else {

    // Log at "info" level if not bulk, else "trace" level

    if (!is_bulk) {

      log_info(biasedlocking)("  Revoked bias of currently-unlocked object");

    } else {

      log_trace(biasedlocking)("  Revoked bias of currently-unlocked object");

    }

    // Store the unlocked value into the object's header.

    obj->set_mark(unbiased_prototype);

  }

  // If requested, return information on which thread held the bias

  if (biased_locker != NULL) {

    *biased_locker = biased_thread;

  }

}

enum HeuristicsResult {

  HR_NOT_BIASED    = 1,

  HR_SINGLE_REVOKE = 2,

  HR_BULK_REBIAS   = 3,

  HR_BULK_REVOKE   = 4

};

static HeuristicsResult update_heuristics(oop o) {

  markWord mark = o->mark();

  if (!mark.has_bias_pattern()) {

    return HR_NOT_BIASED;

  }

  // Heuristics to attempt to throttle the number of revocations.

  // Stages:

  // 1. Revoke the biases of all objects in the heap of this type,

  //    but allow rebiasing of those objects if unlocked.

  // 2. Revoke the biases of all objects in the heap of this type

  //    and don't allow rebiasing of these objects. Disable

  //    allocation of objects of that type with the bias bit set.

  Klass* k = o->klass();

  jlong cur_time = os::javaTimeMillis();

  jlong last_bulk_revocation_time = k->last_biased_lock_bulk_revocation_time();

  int revocation_count = k->biased_lock_revocation_count();

  if ((revocation_count >= BiasedLockingBulkRebiasThreshold) &&

      (revocation_count <  BiasedLockingBulkRevokeThreshold) &&

      (last_bulk_revocation_time != 0) &&

      (cur_time - last_bulk_revocation_time >= BiasedLockingDecayTime)) {

    // This is the first revocation we've seen in a while of an

    // object of this type since the last time we performed a bulk

    // rebiasing operation. The application is allocating objects in

    // bulk which are biased toward a thread and then handing them

    // off to another thread. We can cope with this allocation

    // pattern via the bulk rebiasing mechanism so we reset the

    // klass's revocation count rather than allow it to increase

    // monotonically. If we see the need to perform another bulk

    // rebias operation later, we will, and if subsequently we see

    // many more revocation operations in a short period of time we

    // will completely disable biasing for this type.

    k->set_biased_lock_revocation_count(0);

    revocation_count = 0;

  }

  // Make revocation count saturate just beyond BiasedLockingBulkRevokeThreshold

  if (revocation_count <= BiasedLockingBulkRevokeThreshold) {

    revocation_count = k->atomic_incr_biased_lock_revocation_count();

  }

  if (revocation_count == BiasedLockingBulkRevokeThreshold) {

    return HR_BULK_REVOKE;

  }

  if (revocation_count == BiasedLockingBulkRebiasThreshold) {

    return HR_BULK_REBIAS;

  }

  return HR_SINGLE_REVOKE;

}

void BiasedLocking::bulk_revoke_at_safepoint(oop o, bool bulk_rebias, JavaThread* requesting_thread) {

  assert(SafepointSynchronize::is_at_safepoint(), "must be done at safepoint");

  assert(Thread::current()->is_VM_thread(), "must be VMThread");

  log_info(biasedlocking)("* Beginning bulk revocation (kind == %s) because of object "

                          INTPTR_FORMAT " , mark " INTPTR_FORMAT " , type %s",

                          (bulk_rebias ? "rebias" : "revoke"),

                          p2i((void *) o),

                          o->mark().value(),

                          o->klass()->external_name());

  jlong cur_time = os::javaTimeMillis();

  o->klass()->set_last_biased_lock_bulk_revocation_time(cur_time);

  Klass* k_o = o->klass();

  Klass* klass = k_o;

  {

    JavaThreadIteratorWithHandle jtiwh;

    if (bulk_rebias) {

      // Use the epoch in the klass of the object to implicitly revoke

      // all biases of objects of this data type and force them to be

      // reacquired. However, we also need to walk the stacks of all

      // threads and update the headers of lightweight locked objects

      // with biases to have the current epoch.

      // If the prototype header doesn't have the bias pattern, don't

      // try to update the epoch -- assume another VM operation came in

      // and reset the header to the unbiased state, which will

      // implicitly cause all existing biases to be revoked

      if (klass->prototype_header().has_bias_pattern()) {

        int prev_epoch = klass->prototype_header().bias_epoch();

        klass->set_prototype_header(klass->prototype_header().incr_bias_epoch());

        int cur_epoch = klass->prototype_header().bias_epoch();

        // Now walk all threads' stacks and adjust epochs of any biased

        // and locked objects of this data type we encounter

        for (; JavaThread *thr = jtiwh.next(); ) {

          GrowableArray<MonitorInfo*>* cached_monitor_info = get_or_compute_monitor_info(thr);

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

            MonitorInfo* mon_info = cached_monitor_info->at(i);

            oop owner = mon_info->owner();

            markWord mark = owner->mark();

            if ((owner->klass() == k_o) && mark.has_bias_pattern()) {

              // We might have encountered this object already in the case of recursive locking

              assert(mark.bias_epoch() == prev_epoch || mark.bias_epoch() == cur_epoch, "error in bias epoch adjustment");

              owner->set_mark(mark.set_bias_epoch(cur_epoch));

            }

          }

        }

      }

      // At this point we're done. All we have to do is potentially

      // adjust the header of the given object to revoke its bias.

      single_revoke_at_safepoint(o, true, requesting_thread, NULL);

    } else {

      if (log_is_enabled(Info, biasedlocking)) {

        ResourceMark rm;

        log_info(biasedlocking)("* Disabling biased locking for type %s", klass->external_name());

      }

      // Disable biased locking for this data type. Not only will this

      // cause future instances to not be biased, but existing biased

      // instances will notice that this implicitly caused their biases

      // to be revoked.

      klass->set_prototype_header(markWord::prototype());

      // Now walk all threads' stacks and forcibly revoke the biases of

      // any locked and biased objects of this data type we encounter.

      for (; JavaThread *thr = jtiwh.next(); ) {

        GrowableArray<MonitorInfo*>* cached_monitor_info = get_or_compute_monitor_info(thr);

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

          MonitorInfo* mon_info = cached_monitor_info->at(i);

          oop owner = mon_info->owner();

          markWord mark = owner->mark();

          if ((owner->klass() == k_o) && mark.has_bias_pattern()) {

            single_revoke_at_safepoint(owner, true, requesting_thread, NULL);

          }

        }

      }

      // Must force the bias of the passed object to be forcibly revoked

      // as well to ensure guarantees to callers

      single_revoke_at_safepoint(o, true, requesting_thread, NULL);

    }

  } // ThreadsListHandle is destroyed here.

  log_info(biasedlocking)("* Ending bulk revocation");

  assert(!o->mark().has_bias_pattern(), "bug in bulk bias revocation");

}

static void clean_up_cached_monitor_info(JavaThread* thread = NULL) {

  if (thread != NULL) {

    thread->set_cached_monitor_info(NULL);

  } else {

    // Walk the thread list clearing out the cached monitors

    for (JavaThreadIteratorWithHandle jtiwh; JavaThread *thr = jtiwh.next(); ) {

      thr->set_cached_monitor_info(NULL);

    }

  }

}

class VM_BulkRevokeBias : public VM_Operation {

private:

  Handle* _obj;

  JavaThread* _requesting_thread;

  bool _bulk_rebias;

  uint64_t _safepoint_id;

public:

  VM_BulkRevokeBias(Handle* obj, JavaThread* requesting_thread,

                    bool bulk_rebias)

    : _obj(obj)

    , _requesting_thread(requesting_thread)

    , _bulk_rebias(bulk_rebias)

    , _safepoint_id(0) {}

  virtual VMOp_Type type() const { return VMOp_BulkRevokeBias; }

  virtual void doit() {

    BiasedLocking::bulk_revoke_at_safepoint((*_obj)(), _bulk_rebias, _requesting_thread);

    _safepoint_id = SafepointSynchronize::safepoint_id();

    clean_up_cached_monitor_info();

  }

  bool is_bulk_rebias() const {

    return _bulk_rebias;

  }

  uint64_t safepoint_id() const {

    return _safepoint_id;

  }

};