/*
 * Copyright (c) 2017, 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.
 *
 */

#ifndef SHARE_VM_RUNTIME_THREADSMR_HPP
#define SHARE_VM_RUNTIME_THREADSMR_HPP

#include "memory/allocation.hpp"
#include "runtime/timer.hpp"

// Thread Safe Memory Reclamation (Thread-SMR) support.
//
// ThreadsListHandles are used to safely perform operations on one or more
// threads without the risk of the thread or threads exiting during the
// operation. It is no longer necessary to hold the Threads_lock to safely
// perform an operation on a target thread.
//
// There are several different ways to refer to java.lang.Thread objects
// so we have a few ways to get a protected JavaThread *:
//
// JNI jobject example:
//   jobject jthread = ...;
//   :
//   ThreadsListHandle tlh;
//   JavaThread* jt = NULL;
//   bool is_alive = tlh.cv_internal_thread_to_JavaThread(jthread, &jt, NULL);
//   if (is_alive) {
//     :  // do stuff with 'jt'...
//   }
//
// JVM/TI jthread example:
//   jthread thread = ...;
//   :
//   JavaThread* jt = NULL;
//   ThreadsListHandle tlh;
//   jvmtiError err = JvmtiExport::cv_external_thread_to_JavaThread(tlh.list(), thread, &jt, NULL);
//   if (err != JVMTI_ERROR_NONE) {
//     return err;
//   }
//   :  // do stuff with 'jt'...
//
// JVM/TI oop example (this one should be very rare):
//   oop thread_obj = ...;
//   :
//   JavaThread *jt = NULL;
//   ThreadsListHandle tlh;
//   jvmtiError err = JvmtiExport::cv_oop_to_JavaThread(tlh.list(), thread_obj, &jt);
//   if (err != JVMTI_ERROR_NONE) {
//     return err;
//   }
//   :  // do stuff with 'jt'...
//
// A JavaThread * that is included in the ThreadsList that is held by
// a ThreadsListHandle is protected as long as the ThreadsListHandle
// remains in scope. The target JavaThread * may have logically exited,
// but that target JavaThread * will not be deleted until it is no
// longer protected by a ThreadsListHandle.


// SMR Support for the Threads class.
//
class ThreadsSMRSupport : AllStatic {
  // The coordination between ThreadsSMRSupport::release_stable_list() and
  // ThreadsSMRSupport::smr_delete() uses the smr_delete_lock in order to
  // reduce the traffic on the Threads_lock.
  static Monitor*              _smr_delete_lock;
  // The '_cnt', '_max' and '_times" fields are enabled via
  // -XX:+EnableThreadSMRStatistics (see thread.cpp for a
  // description about each field):
  static uint                  _smr_delete_lock_wait_cnt;
  static uint                  _smr_delete_lock_wait_max;
  // The smr_delete_notify flag is used for proper double-check
  // locking in order to reduce the traffic on the smr_delete_lock.
  static volatile uint         _smr_delete_notify;
  static volatile uint         _smr_deleted_thread_cnt;
  static volatile uint         _smr_deleted_thread_time_max;
  static volatile uint         _smr_deleted_thread_times;
  static ThreadsList* volatile _smr_java_thread_list;
  static uint64_t              _smr_java_thread_list_alloc_cnt;
  static uint64_t              _smr_java_thread_list_free_cnt;
  static uint                  _smr_java_thread_list_max;
  static uint                  _smr_nested_thread_list_max;
  static volatile uint         _smr_tlh_cnt;
  static volatile uint         _smr_tlh_time_max;
  static volatile uint         _smr_tlh_times;
  static ThreadsList*          _smr_to_delete_list;
  static uint                  _smr_to_delete_list_cnt;
  static uint                  _smr_to_delete_list_max;

  static ThreadsList *acquire_stable_list_fast_path(Thread *self);
  static ThreadsList *acquire_stable_list_nested_path(Thread *self);
  static void add_smr_deleted_thread_times(uint add_value);
  static void add_smr_tlh_times(uint add_value);
  static void clear_smr_delete_notify();
  static void inc_smr_deleted_thread_cnt();
  static void inc_smr_java_thread_list_alloc_cnt();
  static void inc_smr_tlh_cnt();
  static bool is_a_protected_JavaThread(JavaThread *thread);
  static void release_stable_list_fast_path(Thread *self);
  static void release_stable_list_nested_path(Thread *self);
  static void release_stable_list_wake_up(char *log_str);
  static void set_smr_delete_notify();
  static Monitor* smr_delete_lock() { return _smr_delete_lock; }
  static bool smr_delete_notify();
  static void smr_free_list(ThreadsList* threads);
  static void update_smr_deleted_thread_time_max(uint new_value);
  static void update_smr_java_thread_list_max(uint new_value);
  static void update_smr_tlh_time_max(uint new_value);
  static ThreadsList* xchg_smr_java_thread_list(ThreadsList* new_list);

 public:
  static ThreadsList *acquire_stable_list(Thread *self, bool is_ThreadsListSetter);
  static void add_thread(JavaThread *thread);
  static ThreadsList* get_smr_java_thread_list();
  static bool is_a_protected_JavaThread_with_lock(JavaThread *thread);
  static void release_stable_list(Thread *self);
  static void remove_thread(JavaThread *thread);
  static void smr_delete(JavaThread *thread);
  static void update_smr_tlh_stats(uint millis);

  // Logging and printing support:
  static void log_smr_statistics();
  static void print_smr_info_elements_on(outputStream* st, ThreadsList* t_list);
  static void print_smr_info_on(outputStream* st);
};

// A fast list of JavaThreads.
//
class ThreadsList : public CHeapObj<mtThread> {
  friend class ThreadsSMRSupport;  // for next_list(), set_next_list() access

  const uint _length;
  ThreadsList* _next_list;
  JavaThread *const *const _threads;

  template <class T>
  void threads_do_dispatch(T *cl, JavaThread *const thread) const;

  ThreadsList *next_list() const        { return _next_list; }
  void set_next_list(ThreadsList *list) { _next_list = list; }

  static ThreadsList* add_thread(ThreadsList* list, JavaThread* java_thread);
  static ThreadsList* remove_thread(ThreadsList* list, JavaThread* java_thread);

public:
  ThreadsList(int entries);
  ~ThreadsList();

  template <class T>
  void threads_do(T *cl) const;

  uint length() const                       { return _length; }

  JavaThread *const thread_at(uint i) const { return _threads[i]; }

  JavaThread *const *threads() const        { return _threads; }

  // Returns -1 if target is not found.
  int find_index_of_JavaThread(JavaThread* target);
  JavaThread* find_JavaThread_from_java_tid(jlong java_tid) const;
  bool includes(const JavaThread * const p) const;
};

// Linked list of ThreadsLists to support nested ThreadsListHandles.
class NestedThreadsList : public CHeapObj<mtThread> {
  ThreadsList*const _t_list;
  NestedThreadsList* _next;

public:
  NestedThreadsList(ThreadsList* t_list) : _t_list(t_list) {
    assert(Threads_lock->owned_by_self(),
           "must own Threads_lock for saved t_list to be valid.");
  }

  ThreadsList* t_list() { return _t_list; }
  NestedThreadsList* next() { return _next; }
  void set_next(NestedThreadsList* value) { _next = value; }
};

// A helper to optionally set the hazard ptr in ourself. This helper can
// be used by ourself or by another thread. If the hazard ptr is set(),
// then the destructor will release it.
//
class ThreadsListSetter : public StackObj {
private:
  bool _target_needs_release;  // needs release only when set()
  Thread * _target;

public:
  ThreadsListSetter() : _target_needs_release(false), _target(Thread::current()) {
  }
  ~ThreadsListSetter();
  ThreadsList* list();
  void set();
  bool target_needs_release() { return _target_needs_release; }
};

// This stack allocated ThreadsListHandle keeps all JavaThreads in the
// ThreadsList from being deleted until it is safe.
//
class ThreadsListHandle : public StackObj {
  ThreadsList * _list;
  Thread *const _self;
  elapsedTimer _timer;  // Enabled via -XX:+EnableThreadSMRStatistics.

public:
  ThreadsListHandle(Thread *self = Thread::current());
  ~ThreadsListHandle();

  ThreadsList *list() const {
    return _list;
  }

  template <class T>
  void threads_do(T *cl) const {
    return _list->threads_do(cl);
  }

  bool cv_internal_thread_to_JavaThread(jobject jthread, JavaThread ** jt_pp, oop * thread_oop_p);

  bool includes(JavaThread* p) {
    return _list->includes(p);
  }

  uint length() const {
    return _list->length();
  }
};

// This stack allocated JavaThreadIterator is used to walk the
// specified ThreadsList using the following style:
//
//   JavaThreadIterator jti(t_list);
//   for (JavaThread *jt = jti.first(); jt != NULL; jt = jti.next()) {
//     ...
//   }
//
class JavaThreadIterator : public StackObj {
  ThreadsList * _list;
  uint _index;

public:
  JavaThreadIterator(ThreadsList *list) : _list(list), _index(0) {
    assert(list != NULL, "ThreadsList must not be NULL.");
  }

  JavaThread *first() {
    _index = 0;
    return _list->thread_at(_index);
  }

  uint length() const {
    return _list->length();
  }

  ThreadsList *list() const {
    return _list;
  }

  JavaThread *next() {
    if (++_index >= length()) {
      return NULL;
    }
    return _list->thread_at(_index);
  }
};

// This stack allocated ThreadsListHandle and JavaThreadIterator combo
// is used to walk the ThreadsList in the included ThreadsListHandle
// using the following style:
//
//   for (JavaThreadIteratorWithHandle jtiwh; JavaThread *jt = jtiwh.next(); ) {
//     ...
//   }
//
class JavaThreadIteratorWithHandle : public StackObj {
  ThreadsListHandle _tlh;
  uint _index;

public:
  JavaThreadIteratorWithHandle() : _index(0) {}

  uint length() const {
    return _tlh.length();
  }

  ThreadsList *list() const {
    return _tlh.list();
  }

  JavaThread *next() {
    if (_index >= length()) {
      return NULL;
    }
    return _tlh.list()->thread_at(_index++);
  }

  void rewind() {
    _index = 0;
  }
};

#endif // SHARE_VM_RUNTIME_THREADSMR_HPP