/*

 * 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 "jni.h"

#include "jvm.h"

#include "jfr/jfr.hpp"

#include "jfr/jfrEvents.hpp"

#include "jfr/periodic/sampling/jfrThreadSampler.hpp"

#include "jfr/recorder/jfrEventSetting.hpp"

#include "jfr/recorder/jfrRecorder.hpp"

#include "jfr/recorder/checkpoint/jfrMetadataEvent.hpp"

#include "jfr/recorder/checkpoint/types/traceid/jfrTraceId.inline.hpp"

#include "jfr/recorder/repository/jfrRepository.hpp"

#include "jfr/recorder/repository/jfrChunkRotation.hpp"

#include "jfr/recorder/repository/jfrChunkWriter.hpp"

#include "jfr/recorder/service/jfrOptionSet.hpp"

#include "jfr/recorder/stacktrace/jfrStackTraceRepository.hpp"

#include "jfr/recorder/stringpool/jfrStringPool.hpp"

#include "jfr/jni/jfrGetAllEventClasses.hpp"

#include "jfr/jni/jfrJavaSupport.hpp"

#include "jfr/jni/jfrJniMethodRegistration.hpp"

#include "jfr/instrumentation/jfrEventClassTransformer.hpp"

#include "jfr/instrumentation/jfrJvmtiAgent.hpp"

#include "jfr/leakprofiler/leakProfiler.hpp"

#include "jfr/utilities/jfrJavaLog.hpp"

#include "jfr/utilities/jfrTimeConverter.hpp"

#include "jfr/utilities/jfrTime.hpp"

#include "jfr/writers/jfrJavaEventWriter.hpp"

#include "jfrfiles/jfrPeriodic.hpp"

#include "logging/log.hpp"

#include "memory/resourceArea.hpp"

#include "runtime/interfaceSupport.inline.hpp"

#include "runtime/mutexLocker.hpp"

#include "runtime/os.hpp"

#include "runtime/thread.hpp"

#include "utilities/debug.hpp"

#define NO_TRANSITION(result_type, header) extern "C" { result_type JNICALL header {

#define NO_TRANSITION_END } }

/*

 * NO_TRANSITION entries

 *

 * Thread remains _thread_in_native

 */

NO_TRANSITION(void, jfr_register_natives(JNIEnv* env, jclass jvmclass))

  JfrJniMethodRegistration register_native_methods(env);

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_is_enabled())

  return Jfr::is_enabled() ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_is_disabled())

  return Jfr::is_disabled() ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_is_started())

  return JfrRecorder::is_created() ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jstring, jfr_get_pid(JNIEnv* env, jobject jvm))

  char pid_buf[32] = { 0 };

  jio_snprintf(pid_buf, sizeof(pid_buf), "%d", os::current_process_id());

  jstring pid_string = env->NewStringUTF(pid_buf);

  return pid_string; // exception pending if NULL

NO_TRANSITION_END

NO_TRANSITION(jlong, jfr_elapsed_frequency(JNIEnv* env, jobject jvm))

  return JfrTime::frequency();

NO_TRANSITION_END

NO_TRANSITION(jlong, jfr_elapsed_counter(JNIEnv* env, jobject jvm))

  return JfrTicks::now();

NO_TRANSITION_END

NO_TRANSITION(void, jfr_retransform_classes(JNIEnv* env, jobject jvm, jobjectArray classes))

  JfrJvmtiAgent::retransform_classes(env, classes, JavaThread::thread_from_jni_environment(env));

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_enabled(JNIEnv* env, jobject jvm, jlong event_type_id, jboolean enabled))

  JfrEventSetting::set_enabled(event_type_id, JNI_TRUE == enabled);

  if (EventOldObjectSample::eventId == event_type_id) {

    ThreadInVMfromNative transition(JavaThread::thread_from_jni_environment(env));

    if (JNI_TRUE == enabled) {

      LeakProfiler::start(JfrOptionSet::old_object_queue_size());

    } else {

      LeakProfiler::stop();

    }

  }

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_file_notification(JNIEnv* env, jobject jvm, jlong threshold))

  JfrChunkRotation::set_threshold(threshold);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_sample_threads(JNIEnv* env, jobject jvm, jboolean sampleThreads))

  JfrOptionSet::set_sample_threads(sampleThreads);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_stack_depth(JNIEnv* env, jobject jvm, jint depth))

  JfrOptionSet::set_stackdepth((jlong)depth);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_stacktrace_enabled(JNIEnv* env, jobject jvm, jlong event_type_id, jboolean enabled))

  JfrEventSetting::set_stacktrace(event_type_id, JNI_TRUE == enabled);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_global_buffer_count(JNIEnv* env, jobject jvm, jlong count))

  JfrOptionSet::set_num_global_buffers(count);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_global_buffer_size(JNIEnv* env, jobject jvm, jlong size))

JfrOptionSet::set_global_buffer_size(size);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_thread_buffer_size(JNIEnv* env, jobject jvm, jlong size))

  JfrOptionSet::set_thread_buffer_size(size);

NO_TRANSITION_END

NO_TRANSITION(void, jfr_set_memory_size(JNIEnv* env, jobject jvm, jlong size))

  JfrOptionSet::set_memory_size(size);

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_set_threshold(JNIEnv* env, jobject jvm, jlong event_type_id, jlong thresholdTicks))

  return JfrEventSetting::set_threshold(event_type_id, thresholdTicks) ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_allow_event_retransforms(JNIEnv* env, jobject jvm))

  return JfrOptionSet::allow_event_retransforms() ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_is_available(JNIEnv* env, jclass jvm))

  return !Jfr::is_disabled() ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jlong, jfr_get_epoch_address(JNIEnv* env, jobject jvm))

  return JfrTraceIdEpoch::epoch_address();

NO_TRANSITION_END

NO_TRANSITION(jlong, jfr_get_unloaded_event_classes_count(JNIEnv* env, jobject jvm))

  return JfrEventClasses::unloaded_event_classes_count();

NO_TRANSITION_END

NO_TRANSITION(jdouble, jfr_time_conv_factor(JNIEnv* env, jobject jvm))

  return (jdouble)JfrTimeConverter::nano_to_counter_multiplier();

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_set_cutoff(JNIEnv* env, jobject jvm, jlong event_type_id, jlong cutoff_ticks))

  return JfrEventSetting::set_cutoff(event_type_id, cutoff_ticks) ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

NO_TRANSITION(jboolean, jfr_should_rotate_disk(JNIEnv* env, jobject jvm))

  return JfrChunkRotation::should_rotate() ? JNI_TRUE : JNI_FALSE;

NO_TRANSITION_END

/*

 * JVM_ENTRY_NO_ENV entries

 *

 * Transitions:

 *   Entry: _thread_in_native -> _thread_in_vm

 *   Exit:  _thread_in_vm -> _thread_in_native

 *

 * Current JavaThread available as "thread" variable

 */

JVM_ENTRY_NO_ENV(jboolean, jfr_create_jfr(JNIEnv* env, jobject jvm, jboolean simulate_failure))

  if (JfrRecorder::is_created()) {

    return JNI_TRUE;

  }

  if (!JfrRecorder::create(simulate_failure == JNI_TRUE)) {

    JfrJavaSupport::throw_illegal_state_exception("Unable to start Jfr", thread);

    return JNI_FALSE;

  }

  return JNI_TRUE;

JVM_END

JVM_ENTRY_NO_ENV(jboolean, jfr_destroy_jfr(JNIEnv* env, jobject jvm))

  JfrRecorder::destroy();

  return JNI_TRUE;

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_begin_recording(JNIEnv* env, jobject jvm))

  if (JfrRecorder::is_recording()) {

    return;

  }

  JfrRecorder::start_recording();

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_end_recording(JNIEnv* env, jobject jvm))

  if (!JfrRecorder::is_recording()) {

    return;

  }

  JfrRecorder::stop_recording();

JVM_END

JVM_ENTRY_NO_ENV(jboolean, jfr_emit_event(JNIEnv* env, jobject jvm, jlong eventTypeId, jlong timeStamp, jlong when))

  JfrPeriodicEventSet::requestEvent((JfrEventId)eventTypeId);

  return thread->has_pending_exception() ? JNI_FALSE : JNI_TRUE;

JVM_END

JVM_ENTRY_NO_ENV(jobject, jfr_get_all_event_classes(JNIEnv* env, jobject jvm))

  return JfrEventClasses::get_all_event_classes(thread);

JVM_END

JVM_ENTRY_NO_ENV(jlong, jfr_class_id(JNIEnv* env, jclass jvm, jclass jc))

  return JfrTraceId::use(jc);

JVM_END

JVM_ENTRY_NO_ENV(jlong, jfr_stacktrace_id(JNIEnv* env, jobject jvm, jint skip))

  return JfrStackTraceRepository::record(thread, skip);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_log(JNIEnv* env, jobject jvm, jint tag_set, jint level, jstring message))

  JfrJavaLog::log(tag_set, level, message, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_subscribe_log_level(JNIEnv* env, jobject jvm, jobject log_tag, jint id))

  JfrJavaLog::subscribe_log_level(log_tag, id, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_set_output(JNIEnv* env, jobject jvm, jstring path))

  JfrRepository::set_chunk_path(path, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_set_method_sampling_interval(JNIEnv* env, jobject jvm, jlong type, jlong intervalMillis))

  if (intervalMillis < 0) {

    intervalMillis = 0;

  }

  JfrEventId typed_event_id = (JfrEventId)type;

  assert(EventExecutionSample::eventId == typed_event_id || EventNativeMethodSample::eventId == typed_event_id, "invariant");

  if (intervalMillis > 0) {

    JfrEventSetting::set_enabled(typed_event_id, true); // ensure sampling event is enabled

  }

  if (EventExecutionSample::eventId == type) {

    JfrThreadSampling::set_java_sample_interval(intervalMillis);

  } else {

    JfrThreadSampling::set_native_sample_interval(intervalMillis);

  }

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_store_metadata_descriptor(JNIEnv* env, jobject jvm, jbyteArray descriptor))

  JfrMetadataEvent::update(descriptor);

JVM_END

// trace thread id for a thread object

JVM_ENTRY_NO_ENV(jlong, jfr_id_for_thread(JNIEnv* env, jobject jvm, jobject t))

  return JfrJavaSupport::jfr_thread_id(t);

JVM_END

JVM_ENTRY_NO_ENV(jobject, jfr_get_event_writer(JNIEnv* env, jclass cls))

  return JfrJavaEventWriter::event_writer(thread);

JVM_END

JVM_ENTRY_NO_ENV(jobject, jfr_new_event_writer(JNIEnv* env, jclass cls))

  return JfrJavaEventWriter::new_event_writer(thread);

JVM_END

JVM_ENTRY_NO_ENV(jboolean, jfr_event_writer_flush(JNIEnv* env, jclass cls, jobject writer, jint used_size, jint requested_size))

  return JfrJavaEventWriter::flush(writer, used_size, requested_size, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_flush(JNIEnv* env, jobject jvm, jboolean include_metadata, jshort flush_counter))

  JfrRepository::flush(include_metadata == JNI_TRUE, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_set_repository_location(JNIEnv* env, jobject repo, jstring location))

  return JfrRepository::set_path(location, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_uncaught_exception(JNIEnv* env, jobject jvm, jobject t, jthrowable throwable))

  JfrJavaSupport::uncaught_exception(throwable, thread);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_abort(JNIEnv* env, jobject jvm, jstring errorMsg))

  JfrJavaSupport::abort(errorMsg, thread);

JVM_END

JVM_ENTRY_NO_ENV(jlong, jfr_type_id(JNIEnv* env, jobject jvm, jclass jc))

  return JfrTraceId::get(jc);

JVM_END

JVM_ENTRY_NO_ENV(jboolean, jfr_add_string_constant(JNIEnv* env, jclass jvm, jboolean epoch, jlong id, jstring string))

  return JfrStringPool::add(epoch == JNI_TRUE, id, string, thread) ? JNI_TRUE : JNI_FALSE;

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_set_force_instrumentation(JNIEnv* env, jobject jvm, jboolean force_instrumentation))

  JfrEventClassTransformer::set_force_instrumentation(force_instrumentation == JNI_TRUE);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_emit_old_object_samples(JNIEnv* env, jobject jvm, jlong cutoff_ticks, jboolean emit_all))

  LeakProfiler::emit_events(cutoff_ticks, emit_all == JNI_TRUE);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_exclude_thread(JNIEnv* env, jobject jvm, jobject t))

  JfrJavaSupport::exclude(t);

JVM_END

JVM_ENTRY_NO_ENV(void, jfr_include_thread(JNIEnv* env, jobject jvm, jobject t))

  JfrJavaSupport::include(t);

JVM_END

JVM_ENTRY_NO_ENV(jboolean, jfr_is_thread_excluded(JNIEnv* env, jobject jvm, jobject t))

  return JfrJavaSupport::is_excluded(t);

JVM_END

JVM_ENTRY_NO_ENV(jlong, jfr_chunk_start_nanos(JNIEnv* env, jobject jvm))

  return JfrRepository::current_chunk_start_nanos();

JVM_END