/*

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

 *

 */

// TODO: we need to define a naming convention for perf counters

// to distinguish counters for:

//   - standard JSR174 use

//   - Hotspot extension (public and committed)

//   - Hotspot extension (private/internal and uncommitted)

// Default is disabled.

bool ThreadService::_thread_monitoring_contention_enabled = false;

bool ThreadService::_thread_cpu_time_enabled = false;

PerfCounter*  ThreadService::_total_threads_count = NULL;

PerfVariable* ThreadService::_live_threads_count = NULL;

PerfVariable* ThreadService::_peak_threads_count = NULL;

PerfVariable* ThreadService::_daemon_threads_count = NULL;

volatile int ThreadService::_exiting_threads_count = 0;

volatile int ThreadService::_exiting_daemon_threads_count = 0;

ThreadDumpResult* ThreadService::_threaddump_list = NULL;

static const int INITIAL_ARRAY_SIZE = 10;

void ThreadService::init() {

  EXCEPTION_MARK;

  // These counters are for java.lang.management API support.

  // They are created even if -XX:-UsePerfData is set and in

  // that case, they will be allocated on C heap.

  _total_threads_count =

                PerfDataManager::create_counter(JAVA_THREADS, "started",

                                                PerfData::U_Events, CHECK);

  _live_threads_count =

                PerfDataManager::create_variable(JAVA_THREADS, "live",

                                                 PerfData::U_None, CHECK);

  _peak_threads_count =

                PerfDataManager::create_variable(JAVA_THREADS, "livePeak",

                                                 PerfData::U_None, CHECK);

  _daemon_threads_count =

                PerfDataManager::create_variable(JAVA_THREADS, "daemon",

                                                 PerfData::U_None, CHECK);

  if (os::is_thread_cpu_time_supported()) {

    _thread_cpu_time_enabled = true;

  }

}

void ThreadService::reset_peak_thread_count() {

  // Acquire the lock to update the peak thread count

  // to synchronize with thread addition and removal.

  MutexLockerEx mu(Threads_lock);

  _peak_threads_count->set_value(get_live_thread_count());

}

void ThreadService::add_thread(JavaThread* thread, bool daemon) {

  // Do not count VM internal or JVMTI agent threads

  if (thread->is_hidden_from_external_view() ||

      thread->is_jvmti_agent_thread()) {

    return;

  }

  _total_threads_count->inc();

  _live_threads_count->inc();

  if (_live_threads_count->get_value() > _peak_threads_count->get_value()) {

    _peak_threads_count->set_value(_live_threads_count->get_value());

  }

  if (daemon) {

    _daemon_threads_count->inc();

  }

}

void ThreadService::remove_thread(JavaThread* thread, bool daemon) {

  Atomic::dec((jint*) &_exiting_threads_count);

  if (thread->is_hidden_from_external_view() ||

      thread->is_jvmti_agent_thread()) {

    return;

  }

  _live_threads_count->set_value(_live_threads_count->get_value() - 1);

  if (daemon) {

    _daemon_threads_count->set_value(_daemon_threads_count->get_value() - 1);

    Atomic::dec((jint*) &_exiting_daemon_threads_count);

  }

}

void ThreadService::current_thread_exiting(JavaThread* jt) {

  assert(jt == JavaThread::current(), "Called by current thread");

  Atomic::inc((jint*) &_exiting_threads_count);

  oop threadObj = jt->threadObj();

  if (threadObj != NULL && java_lang_Thread::is_daemon(threadObj)) {

    Atomic::inc((jint*) &_exiting_daemon_threads_count);

  }

}

// FIXME: JVMTI should call this function

Handle ThreadService::get_current_contended_monitor(JavaThread* thread) {

  assert(thread != NULL, "should be non-NULL");

  assert(Threads_lock->owned_by_self(), "must grab Threads_lock or be at safepoint");

  ObjectMonitor *wait_obj = thread->current_waiting_monitor();

  oop obj = NULL;

  if (wait_obj != NULL) {

    // thread is doing an Object.wait() call

    obj = (oop) wait_obj->object();

    assert(obj != NULL, "Object.wait() should have an object");

  } else {

    ObjectMonitor *enter_obj = thread->current_pending_monitor();

    if (enter_obj != NULL) {

      // thread is trying to enter() or raw_enter() an ObjectMonitor.

      obj = (oop) enter_obj->object();

    }

    // If obj == NULL, then ObjectMonitor is raw which doesn't count.

  }

  Handle h(obj);

  return h;

}

bool ThreadService::set_thread_monitoring_contention(bool flag) {

  MutexLocker m(Management_lock);

  bool prev = _thread_monitoring_contention_enabled;

  _thread_monitoring_contention_enabled = flag;

  return prev;

}

bool ThreadService::set_thread_cpu_time_enabled(bool flag) {

  MutexLocker m(Management_lock);

  bool prev = _thread_cpu_time_enabled;

  _thread_cpu_time_enabled = flag;

  return prev;

}

// GC support

void ThreadService::oops_do(OopClosure* f) {

  for (ThreadDumpResult* dump = _threaddump_list; dump != NULL; dump = dump->next()) {

    dump->oops_do(f);

  }

}

void ThreadService::add_thread_dump(ThreadDumpResult* dump) {

  MutexLocker ml(Management_lock);

  if (_threaddump_list == NULL) {

    _threaddump_list = dump;

  } else {

    dump->set_next(_threaddump_list);

    _threaddump_list = dump;

  }

}

void ThreadService::remove_thread_dump(ThreadDumpResult* dump) {

  MutexLocker ml(Management_lock);

  ThreadDumpResult* prev = NULL;

  bool found = false;

  for (ThreadDumpResult* d = _threaddump_list; d != NULL; prev = d, d = d->next()) {

    if (d == dump) {

      if (prev == NULL) {

        _threaddump_list = dump->next();

      } else {

        prev->set_next(dump->next());

      }

      found = true;

      break;

    }

  }

  assert(found, "The threaddump result to be removed must exist.");

}

// Dump stack trace of threads specified in the given threads array.

// Returns StackTraceElement[][] each element is the stack trace of a thread in

// the corresponding entry in the given threads array

Handle ThreadService::dump_stack_traces(GrowableArray<instanceHandle>* threads,

                                        int num_threads,

                                        TRAPS) {

  assert(num_threads > 0, "just checking");

  ThreadDumpResult dump_result;

  VM_ThreadDump op(&dump_result,

                   threads,

                   num_threads,

                   -1,    /* entire stack */

                   false, /* with locked monitors */

                   false  /* with locked synchronizers */);

  VMThread::execute(&op);

  // Allocate the resulting StackTraceElement[][] object

  ResourceMark rm(THREAD);

  klassOop k = SystemDictionary::resolve_or_fail(vmSymbolHandles::java_lang_StackTraceElement_array(), true, CHECK_NH);

  objArrayKlassHandle ik (THREAD, k);

  objArrayOop r = oopFactory::new_objArray(ik(), num_threads, CHECK_NH);

  objArrayHandle result_obj(THREAD, r);

  int num_snapshots = dump_result.num_snapshots();

  assert(num_snapshots == num_threads, "Must have num_threads thread snapshots");

  int i = 0;

  for (ThreadSnapshot* ts = dump_result.snapshots(); ts != NULL; i++, ts = ts->next()) {

    ThreadStackTrace* stacktrace = ts->get_stack_trace();

    if (stacktrace == NULL) {

      // No stack trace

      result_obj->obj_at_put(i, NULL);

    } else {

      // Construct an array of java/lang/StackTraceElement object

      Handle backtrace_h = stacktrace->allocate_fill_stack_trace_element_array(CHECK_NH);

      result_obj->obj_at_put(i, backtrace_h());

    }

  }

  return result_obj;

}

void ThreadService::reset_contention_count_stat(JavaThread* thread) {

  ThreadStatistics* stat = thread->get_thread_stat();

  if (stat != NULL) {

    stat->reset_count_stat();

  }

}

void ThreadService::reset_contention_time_stat(JavaThread* thread) {

  ThreadStatistics* stat = thread->get_thread_stat();

  if (stat != NULL) {

    stat->reset_time_stat();

  }

}

// Find deadlocks involving object monitors and concurrent locks if concurrent_locks is true

DeadlockCycle* ThreadService::find_deadlocks_at_safepoint(bool concurrent_locks) {

  // This code was modified from the original Threads::find_deadlocks code.

  int globalDfn = 0, thisDfn;

  ObjectMonitor* waitingToLockMonitor = NULL;

  oop waitingToLockBlocker = NULL;

  bool blocked_on_monitor = false;

  JavaThread *currentThread, *previousThread;

  int num_deadlocks = 0;

  for (JavaThread* p = Threads::first(); p != NULL; p = p->next()) {

    // Initialize the depth-first-number

    p->set_depth_first_number(-1);

  }

  DeadlockCycle* deadlocks = NULL;

  DeadlockCycle* last = NULL;

  DeadlockCycle* cycle = new DeadlockCycle();

  for (JavaThread* jt = Threads::first(); jt != NULL; jt = jt->next()) {

    if (jt->depth_first_number() >= 0) {

      // this thread was already visited

      continue;

    }

    thisDfn = globalDfn;

    jt->set_depth_first_number(globalDfn++);

    previousThread = jt;

    currentThread = jt;

    cycle->reset();

    // When there is a deadlock, all the monitors involved in the dependency

    // cycle must be contended and heavyweight. So we only care about the

    // heavyweight monitor a thread is waiting to lock.

    waitingToLockMonitor = (ObjectMonitor*)jt->current_pending_monitor();

    if (concurrent_locks) {

      waitingToLockBlocker = jt->current_park_blocker();

    }

    while (waitingToLockMonitor != NULL || waitingToLockBlocker != NULL) {

      cycle->add_thread(currentThread);

      if (waitingToLockMonitor != NULL) {

        currentThread = Threads::owning_thread_from_monitor_owner((address)waitingToLockMonitor->owner(),

                                                                  false /* no locking needed */);

      } else {

        if (concurrent_locks) {

          if (waitingToLockBlocker->is_a(SystemDictionary::abstract_ownable_synchronizer_klass())) {

            oop threadObj = java_util_concurrent_locks_AbstractOwnableSynchronizer::get_owner_threadObj(waitingToLockBlocker);

            currentThread = threadObj != NULL ? java_lang_Thread::thread(threadObj) : NULL;

          } else {

            currentThread = NULL;

          }

        }

      }

      if (currentThread == NULL) {

        // No dependency on another thread

        break;

      }

      if (currentThread->depth_first_number() < 0) {

        // First visit to this thread

        currentThread->set_depth_first_number(globalDfn++);

      } else if (currentThread->depth_first_number() < thisDfn) {

        // Thread already visited, and not on a (new) cycle

        break;

      } else if (currentThread == previousThread) {

        // Self-loop, ignore

        break;

      } else {

        // We have a (new) cycle

        num_deadlocks++;

        cycle->set_deadlock(true);

        // add this cycle to the deadlocks list

        if (deadlocks == NULL) {

          deadlocks = cycle;

        } else {

          last->set_next(cycle);

        }

        last = cycle;

        cycle = new DeadlockCycle();

        break;

      }

      previousThread = currentThread;

      waitingToLockMonitor = (ObjectMonitor*)currentThread->current_pending_monitor();

      if (concurrent_locks) {

        waitingToLockBlocker = currentThread->current_park_blocker();

      }

    }

  }

  return deadlocks;

}

ThreadDumpResult::ThreadDumpResult() : _num_threads(0), _num_snapshots(0), _snapshots(NULL), _next(NULL), _last(NULL) {

  // Create a new ThreadDumpResult object and append to the list.

  // If GC happens before this function returns, methodOop

  // in the stack trace will be visited.

  ThreadService::add_thread_dump(this);

}

ThreadDumpResult::ThreadDumpResult(int num_threads) : _num_threads(num_threads), _num_snapshots(0), _snapshots(NULL), _next(NULL), _last(NULL) {

  // Create a new ThreadDumpResult object and append to the list.

  // If GC happens before this function returns, oops

  // will be visited.

  ThreadService::add_thread_dump(this);

}

ThreadDumpResult::~ThreadDumpResult() {

  ThreadService::remove_thread_dump(this);

  // free all the ThreadSnapshot objects created during

  // the VM_ThreadDump operation

  ThreadSnapshot* ts = _snapshots;

  while (ts != NULL) {

    ThreadSnapshot* p = ts;

    ts = ts->next();

    delete p;

  }

}

void ThreadDumpResult::add_thread_snapshot(ThreadSnapshot* ts) {

  assert(_num_threads == 0 || _num_snapshots < _num_threads,

         "_num_snapshots must be less than _num_threads");

  _num_snapshots++;

  if (_snapshots == NULL) {

    _snapshots = ts;

  } else {

    _last->set_next(ts);

  }

  _last = ts;

}

void ThreadDumpResult::oops_do(OopClosure* f) {

  for (ThreadSnapshot* ts = _snapshots; ts != NULL; ts = ts->next()) {

    ts->oops_do(f);

  }

}

StackFrameInfo::StackFrameInfo(javaVFrame* jvf, bool with_lock_info) {

  _method = jvf->method();

  _bci = jvf->bci();

  _locked_monitors = NULL;

  if (with_lock_info) {

    ResourceMark rm;

    GrowableArray<MonitorInfo*>* list = jvf->locked_monitors();

    int length = list->length();

    if (length > 0) {

      _locked_monitors = new (ResourceObj::C_HEAP) GrowableArray<oop>(length, true);

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

        MonitorInfo* monitor = list->at(i);

        assert(monitor->owner(), "This monitor must have an owning object");

        _locked_monitors->append(monitor->owner());

      }

    }

  }

}

void StackFrameInfo::oops_do(OopClosure* f) {

  f->do_oop((oop*) &_method);

  if (_locked_monitors != NULL) {

    int length = _locked_monitors->length();

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

      f->do_oop((oop*) _locked_monitors->adr_at(i));

    }

  }

}

void StackFrameInfo::print_on(outputStream* st) const {

  ResourceMark rm;

  java_lang_Throwable::print_stack_element(st, method(), bci());

  int len = (_locked_monitors != NULL ? _locked_monitors->length() : 0);

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

    oop o = _locked_monitors->at(i);

    instanceKlass* ik = instanceKlass::cast(o->klass());

    st->print_cr("\t- locked <" INTPTR_FORMAT "> (a %s)", (address)o, ik->external_name());

  }

}

// Iterate through monitor cache to find JNI locked monitors

class InflatedMonitorsClosure: public MonitorClosure {

private:

  ThreadStackTrace* _stack_trace;

  Thread* _thread;

public:

  InflatedMonitorsClosure(Thread* t, ThreadStackTrace* st) {

    _thread = t;

    _stack_trace = st;

  }

  void do_monitor(ObjectMonitor* mid) {

    if (mid->owner() == _thread) {

      oop object = (oop) mid->object();

      if (!_stack_trace->is_owned_monitor_on_stack(object)) {

        _stack_trace->add_jni_locked_monitor(object);

      }

    }

  }

};

ThreadStackTrace::ThreadStackTrace(JavaThread* t, bool with_locked_monitors) {

  _thread = t;

  _frames = new (ResourceObj::C_HEAP) GrowableArray<StackFrameInfo*>(INITIAL_ARRAY_SIZE, true);

  _depth = 0;

  _with_locked_monitors = with_locked_monitors;

  if (_with_locked_monitors) {

    _jni_locked_monitors = new (ResourceObj::C_HEAP) GrowableArray<oop>(INITIAL_ARRAY_SIZE, true);

  } else {

    _jni_locked_monitors = NULL;

  }

}

ThreadStackTrace::~ThreadStackTrace() {

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

    delete _frames->at(i);

  }

  delete _frames;

  if (_jni_locked_monitors != NULL) {

    delete _jni_locked_monitors;

  }

}

void ThreadStackTrace::dump_stack_at_safepoint(int maxDepth) {

  assert(SafepointSynchronize::is_at_safepoint(), "all threads are stopped");

  if (_thread->has_last_Java_frame()) {

    RegisterMap reg_map(_thread);

    vframe* start_vf = _thread->last_java_vframe(&reg_map);

    int count = 0;

    for (vframe* f = start_vf; f; f = f->sender() ) {

      if (f->is_java_frame()) {

        javaVFrame* jvf = javaVFrame::cast(f);

        add_stack_frame(jvf);

        count++;

      } else {

        // Ignore non-Java frames

      }

      if (maxDepth > 0 && count == maxDepth) {

        // Skip frames if more than maxDepth

        break;

      }

    }

  }

  if (_with_locked_monitors) {

    // Iterate inflated monitors and find monitors locked by this thread

    // not found in the stack

    InflatedMonitorsClosure imc(_thread, this);