88 lines
2.9 KiB
C++
88 lines
2.9 KiB
C++
/*
|
|
* 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 */
|