/*
 * Copyright 2021 Code Intelligence GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <jni.h>

#include <string>

namespace jazzer {

void DumpJvmStackTraces();

// JVM is a thin wrapper around JNI_CreateJavaVM and DestroyJavaVM. The JVM
// instance is created inside the constructor with some default JNI options
// + options which can be added to via command line flags.
class JVM {
 private:
  JavaVM *jvm_;
  JNIEnv *env_;

 public:
  // Creates a JVM instance with default options + options that were provided as
  // command line flags.
  explicit JVM(const std::string &executable_path);

  // Destroy the running JVM instance.
  ~JVM();

  // Get the JNI environment for interaction with the running JVM instance.
  JNIEnv &GetEnv() const;

  jclass FindClass(std::string class_name) const;
  jmethodID GetStaticMethodID(jclass class_id, const std::string &method_name,
                              const std::string &signature,
                              bool is_required = true) const;
  jmethodID GetMethodID(jclass class_id, const std::string &method_name,
                        const std::string &signature) const;
  jfieldID GetStaticFieldID(jclass jclass, const std::string &field_name,
                            const std::string &type) const;
};

// Adds convenience methods to convert a jvm exception to std::string
// using StringWriter and PrintWriter. The stack trace can be subjected to
// further processing, such as deduplication token computation and severity
// annotation.
class ExceptionPrinter {
 private:
  const JVM &jvm_;

  jclass string_writer_class_;
  jmethodID string_writer_constructor_;
  jmethodID string_writer_to_string_method_;

  jclass print_writer_class_;
  jmethodID print_writer_constructor_;
  jmethodID print_stack_trace_method_;

  jclass exception_utils_;
  jmethodID compute_dedup_token_method_;
  jmethodID preprocess_throwable_method_;

 protected:
  explicit ExceptionPrinter(JVM &jvm);

  // returns the current JVM exception stack trace as a string
  std::string getStackTrace(jthrowable exception) const;
  // augments the throwable with additional information such as severity markers
  jthrowable preprocessException(jthrowable exception) const;
  // returns a hash of the exception stack trace for deduplication purposes
  jlong computeDedupToken(jthrowable exception) const;
};

} /* namespace jazzer */