/*
 * Copyright (c) 2008, 2018, 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.
 */
#include <string.h>
#include <stdlib.h>
#include <jvmti_aod.h>

#ifdef __cplusplus
extern "C" {
#endif

void nsk_jvmti_aod_disableEventAndFinish(const char* agentName, jvmtiEvent event, int success, jvmtiEnv *jvmti, JNIEnv* jni) {
    if (!nsk_jvmti_aod_disableEvent(jvmti, event))
        success = 0;

    nsk_aod_agentFinished(jni, agentName, success);
}

void nsk_jvmti_aod_disableEventsAndFinish(const char* agentName, jvmtiEvent events[], int eventsNumber, int success, jvmtiEnv *jvmti, JNIEnv* jni) {
    if (!nsk_jvmti_aod_disableEvents(jvmti, events, eventsNumber))
        success = 0;

    nsk_aod_agentFinished(jni, agentName, success);
}

/*
 * Work with agent options
 */

struct {
    jvmtiEnv *jvmti;
    Options *options;
} multiagentsOptions[MAX_MULTIPLE_AGENTS];

static volatile int multiagentsCount = 0;

int nsk_jvmti_aod_addMultiagentsOptions(jvmtiEnv *jvmti, Options *options) {
    if (multiagentsCount >= MAX_MULTIPLE_AGENTS) {
        NSK_COMPLAIN1("To many agents, max agents count is %d\n", MAX_MULTIPLE_AGENTS);
        return NSK_FALSE;
    }

    multiagentsOptions[multiagentsCount].jvmti = jvmti;
    multiagentsOptions[multiagentsCount].options = options;
    multiagentsCount++;

    NSK_DISPLAY3("Options for agent %s were added (jvmtiEnv: %p, agentsCount: %d)\n",
            nsk_aod_getOptionValue(options, NSK_AOD_AGENT_NAME_OPTION),
            jvmti,
            multiagentsCount);

    return NSK_TRUE;
}

Options* nsk_jvmti_aod_getMultiagentsOptions(jvmtiEnv *jvmti) {
    int i;
    for (i = 0; i < multiagentsCount; i++) {
        if (multiagentsOptions[i].jvmti == jvmti) {
            return multiagentsOptions[i].options;
        }
    }

    NSK_COMPLAIN1("Options for jvmtiEnv %p weren't found\n", jvmti);

    return NULL;
}

/*
 * Auxiliary functions
 */

void nsk_jvmti_aod_deallocate(jvmtiEnv *jvmti, unsigned char* mem) {
    if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(Deallocate, jvmti, mem))) {
        NSK_COMPLAIN0("Deallocate failed\n");

        /*
         * if deallocate fails it isn't critical and test execution can continue without problems,
         * just call nsk_aod_internal_error to inform framework about this error
         */
        nsk_aod_internal_error();
    }
}

int nsk_jvmti_aod_getClassName(jvmtiEnv *jvmti, jclass klass, char classNameBuffer[]) {
    char* className;

    if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB4(GetClassSignature, jvmti, klass, &className, NULL))) {
        NSK_COMPLAIN0("Failed to get class name\n");
        classNameBuffer[0] = '\0';
        return NSK_FALSE;
    } else {
        strcpy(classNameBuffer, className);

        nsk_jvmti_aod_deallocate(jvmti, (unsigned char*)className);

        return NSK_TRUE;
    }
}

int nsk_jvmti_aod_getThreadName(jvmtiEnv * jvmti, jthread thread, char threadNameBuffer[]) {
    jvmtiThreadInfo info;
    if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB3(GetThreadInfo, jvmti, thread, &info))){
        NSK_COMPLAIN0("Failed to get thread info\n");
        threadNameBuffer[0] = '\0';
        return NSK_FALSE;
    } else {
        strcpy(threadNameBuffer, info.name);

        nsk_jvmti_aod_deallocate(jvmti, (unsigned char*)info.name);

        return NSK_TRUE;
    }
}

// events enabling/disabling

int nsk_jvmti_aod_disableEvents(jvmtiEnv* jvmti, jvmtiEvent events[], int eventsNumber) {
    int i;
    int status = NSK_TRUE;

    for (i = 0; i < eventsNumber; i++) {
        if (!nsk_jvmti_aod_disableEvent(jvmti, events[i])) {
            status = NSK_FALSE;
        }
    }

    return status;
}

int nsk_jvmti_aod_enableEvents(jvmtiEnv* jvmti, jvmtiEvent events[], int eventsNumber) {
    int i;
    for (i = 0; i < eventsNumber; i++) {
        if (!nsk_jvmti_aod_enableEvent(jvmti, events[i]))
            return NSK_FALSE;
    }

    return NSK_TRUE;
}

// java threads creation

jthread nsk_jvmti_aod_createThread(JNIEnv *jni) {
    jclass klass ;
    jmethodID threadConstructor;
    jthread thread;

    if (!NSK_JNI_VERIFY(jni,
            (klass = NSK_CPP_STUB2(FindClass, jni, "java/lang/Thread")) != NULL )) {
        NSK_COMPLAIN0("Failed to get the java.lang.Thread class\n");
        return NULL;
    }
    if (!NSK_JNI_VERIFY(jni,
            (threadConstructor = NSK_CPP_STUB4(GetMethodID, jni, klass, "<init>", "()V") )  != NULL )) {
        NSK_COMPLAIN0("Failed to get java.lang.Thread constructor\n");
        return NULL;
    }

    if (!NSK_JNI_VERIFY (jni,
            (thread = NSK_CPP_STUB4(NewObject, jni, klass, threadConstructor, NULL)) != NULL ) ) {
        NSK_COMPLAIN0("Failed to create Thread object\n");
        return NULL;
    }

    if (!NSK_JNI_VERIFY(jni, (thread =
        NSK_CPP_STUB2(NewGlobalRef, jni, thread)) != NULL)) {
        NSK_COMPLAIN0("Failed to create global reference\n");
        return NULL;
    }

    return thread;
}

jthread nsk_jvmti_aod_createThreadWithName(JNIEnv *jni, char* threadName) {
    jclass klass ;
    jmethodID threadConstructor;
    jthread thread;
    jstring threadNameString;

    if (!NSK_JNI_VERIFY(jni, (threadNameString =
        NSK_CPP_STUB2(NewStringUTF, jni, threadName)) != NULL))
        return NULL;

    if (!NSK_JNI_VERIFY(jni,
            (klass = NSK_CPP_STUB2(FindClass, jni, "java/lang/Thread")) != NULL )) {
        NSK_COMPLAIN0("Failed to get the java.lang.Thread class\n");
        return NULL;
    }
    if (!NSK_JNI_VERIFY(jni,
            (threadConstructor = NSK_CPP_STUB4(GetMethodID, jni, klass, "<init>", "(Ljava/lang/String;)V") )  != NULL )) {
        NSK_COMPLAIN0("Failed to get java.lang.Thread constructor\n");
        return NULL;
    }

    if (!NSK_JNI_VERIFY(jni,
            (thread = NSK_CPP_STUB4(NewObject, jni, klass, threadConstructor, threadNameString)) != NULL ) ) {
        NSK_COMPLAIN0("Failed to create Thread object\n");
        return NULL;
    }

    if (!NSK_JNI_VERIFY(jni, (thread =
        NSK_CPP_STUB2(NewGlobalRef, jni, thread)) != NULL)) {
        NSK_COMPLAIN0("Failed to create global reference\n");
        return NULL;
    }

    return thread;
}

// class redefinition

int nsk_jvmti_aod_redefineClass(
        Options * options,
        jvmtiEnv * jvmti,
        jclass classToRedefine,
        const char * fileName) {

    if (!nsk_aod_optionSpecified(options, PATH_TO_NEW_BYTE_CODE_OPTION)) {
        NSK_COMPLAIN1("Option '%s' isn't specified\n", PATH_TO_NEW_BYTE_CODE_OPTION);
        return NSK_FALSE;
    }
    if (fileName == NULL) {
        NSK_COMPLAIN0("File name is NULL\n");
        return NSK_FALSE;
    }
    {
        char file [1024];

        sprintf(file,"%s/%s.class",
                nsk_aod_getOptionValue(options, PATH_TO_NEW_BYTE_CODE_OPTION),
                fileName);
        NSK_DISPLAY1("File with new bytecode: '%s'\n", file);

        {
            FILE *bytecode;
            unsigned char * classBytes;
            jvmtiError error;
            jint size;

            bytecode = fopen(file, "rb");
            error= JVMTI_ERROR_NONE;
            if ( bytecode == NULL ) {
                NSK_COMPLAIN1("Error opening file '%s'\n", file);
                return NSK_FALSE;
            }

            NSK_DISPLAY1("Opening file '%s' \n", file);
            fseek(bytecode, 0, SEEK_END);
            size = ftell(bytecode);
            NSK_DISPLAY1("File size= %ld\n", ftell(bytecode));
            rewind(bytecode);
            error = (*jvmti)->Allocate(jvmti, size, &classBytes);
            if ( error != JVMTI_ERROR_NONE) {
                NSK_DISPLAY1("Failed to create memory %s\n", TranslateError(error));
                return NSK_FALSE;
            }

            if ( ((jint) fread( classBytes, 1, size, bytecode )) != size ) {
                NSK_COMPLAIN0("Failed to read all the bytes, could be less or more\n");
                fclose(bytecode);
                return NSK_FALSE;
            } else {
                NSK_DISPLAY0("File read completely \n");
            }
            fclose(bytecode);
            {
                jvmtiClassDefinition classDef;
                classDef.klass = classToRedefine;
                classDef.class_byte_count= size;
                classDef.class_bytes = classBytes;
                NSK_DISPLAY0("Redefining\n");
                error = (*jvmti)->RedefineClasses(jvmti, 1, &classDef);
                if ( error != JVMTI_ERROR_NONE ) {
                    NSK_DISPLAY1("# error occured while redefining %s ",
                            TranslateError(error) );
                    return NSK_FALSE;
                }
            }
        }
    }
    return NSK_TRUE;
}

// capabilities

void printCapabilities(jvmtiCapabilities caps) {
    #define printCap(name) NSK_DISPLAY1(#name ": %d\n", caps.name)

    printCap(can_tag_objects);
    printCap(can_generate_field_modification_events);
    printCap(can_generate_field_access_events);
    printCap(can_get_bytecodes);
    printCap(can_get_synthetic_attribute);
    printCap(can_get_owned_monitor_info);
    printCap(can_get_current_contended_monitor);
    printCap(can_get_monitor_info);
    printCap(can_pop_frame);
    printCap(can_redefine_classes);
    printCap(can_signal_thread);
    printCap(can_get_source_file_name);
    printCap(can_get_line_numbers);
    printCap(can_get_source_debug_extension);
    printCap(can_access_local_variables);
    printCap(can_maintain_original_method_order);
    printCap(can_generate_single_step_events);
    printCap(can_generate_exception_events);
    printCap(can_generate_frame_pop_events);
    printCap(can_generate_breakpoint_events);
    printCap(can_suspend);
    printCap(can_redefine_any_class);
    printCap(can_get_current_thread_cpu_time);
    printCap(can_get_thread_cpu_time);
    printCap(can_generate_method_entry_events);
    printCap(can_generate_method_exit_events);
    printCap(can_generate_all_class_hook_events);
    printCap(can_generate_compiled_method_load_events);
    printCap(can_generate_monitor_events);
    printCap(can_generate_vm_object_alloc_events);
    printCap(can_generate_native_method_bind_events);
    printCap(can_generate_garbage_collection_events);
    printCap(can_generate_object_free_events);
    printCap(can_force_early_return);
    printCap(can_get_owned_monitor_stack_depth_info);
    printCap(can_get_constant_pool);
    printCap(can_set_native_method_prefix);
    printCap(can_retransform_classes);
    printCap(can_retransform_any_class);

    #undef printCap
}

#ifdef __cplusplus
}
#endif