test/hotspot/jtreg/vmTestbase/nsk/share/jvmti/aod/jvmti_aod.c
changeset 50260 46c67f5e27c2
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/hotspot/jtreg/vmTestbase/nsk/share/jvmti/aod/jvmti_aod.c	Thu May 24 17:12:15 2018 -0700
@@ -0,0 +1,352 @@
+/*
+ * 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