init

arthas-vmtool/src/main/native/head/windows/jawt_md.h

+/*
+ * Copyright (c) 1999, 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.  Oracle designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Oracle in the LICENSE file that accompanied this code.
+ *
+ * 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.
+ */
+
+#ifndef _JAVASOFT_JAWT_MD_H_
+#define _JAVASOFT_JAWT_MD_H_
+
+#include <windows.h>
+#include "jawt.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+/*
+ * Win32-specific declarations for AWT native interface.
+ * See notes in jawt.h for an example of use.
+ */
+typedef struct jawt_Win32DrawingSurfaceInfo {
+    /* Native window, DDB, or DIB handle */
+    union {
+        HWND hwnd;
+        HBITMAP hbitmap;
+        void* pbits;
+    };
+    /*
+     * This HDC should always be used instead of the HDC returned from
+     * BeginPaint() or any calls to GetDC().
+     */
+    HDC hdc;
+    HPALETTE hpalette;
+} JAWT_Win32DrawingSurfaceInfo;
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* !_JAVASOFT_JAWT_MD_H_ */

arthas-vmtool/src/main/native/head/windows/jni_md.h

+/*
+ * Copyright (c) 1996, 2020, 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.  Oracle designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Oracle in the LICENSE file that accompanied this code.
+ *
+ * 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.
+ */
+
+#ifndef _JAVASOFT_JNI_MD_H_
+#define _JAVASOFT_JNI_MD_H_
+
+#ifndef JNIEXPORT
+  #define JNIEXPORT __declspec(dllexport)
+#endif
+#define JNIIMPORT __declspec(dllimport)
+#define JNICALL __stdcall
+
+// 'long' is always 32 bit on windows so this matches what jdk expects
+typedef long jint;
+typedef __int64 jlong;
+typedef signed char jbyte;
+
+#endif /* !_JAVASOFT_JNI_MD_H_ */

arthas-vmtool/src/main/native/src/jni-library.cpp

+#include <iostream>
+#include <jni.h>
+#include <jni_md.h>
+#include <jvmti.h>
+#include "arthas_VmTool.h" // under target/native/javah/
+
+
+static jvmtiEnv *jvmti;
+static jlong tagCounter = 0;
+
+struct LimitCounter {
+    jint currentCounter;
+    jint limitValue;
+
+    void init(jint limit) {
+        currentCounter = 0;
+        limitValue = limit;
+    }
+
+    void countDown() {
+        currentCounter++;
+    }
+
+    bool allow() {
+        if (limitValue < 0) {
+            return true;
+        }
+        return limitValue > currentCounter;
+    }
+};
+
+// 每次 IterateOverInstancesOfClass 调用前需要先 init
+static LimitCounter limitCounter = {0, 0};
+
+extern "C"
+int init_agent(JavaVM *vm, void *reserved) {
+    jint rc;
+    /* Get JVMTI environment */
+    rc = vm->GetEnv((void **)&jvmti, JVMTI_VERSION_1_2);
+    if (rc != JNI_OK) {
+        fprintf(stderr, "ERROR: arthas vmtool Unable to create jvmtiEnv, GetEnv failed, error=%d\n", rc);
+        return -1;
+    }
+
+    jvmtiCapabilities capabilities = {0};
+    capabilities.can_tag_objects = 1;
+    jvmtiError error = jvmti->AddCapabilities(&capabilities);
+    if (error) {
+        fprintf(stderr, "ERROR: arthas vmtool JVMTI AddCapabilities failed!%u\n", error);
+        return JNI_FALSE;
+    }
+
+    return JNI_OK;
+}
+
+extern "C" JNIEXPORT jint JNICALL
+Agent_OnLoad(JavaVM *vm, char *options, void *reserved) {
+    return init_agent(vm, reserved);
+}
+
+extern "C" JNIEXPORT jint JNICALL
+Agent_OnAttach(JavaVM* vm, char* options, void* reserved) {
+    return init_agent(vm, reserved);
+}
+
+extern "C" JNIEXPORT jint JNICALL
+JNI_OnLoad(JavaVM* vm, void* reserved) {
+    init_agent(vm, reserved);
+    return JNI_VERSION_1_6;
+}
+
+extern "C"
+JNIEXPORT void JNICALL
+Java_arthas_VmTool_forceGc0(JNIEnv *env, jclass thisClass) {
+    jvmti->ForceGarbageCollection();
+}
+
+extern "C"
+jlong getTag() {
+    return ++tagCounter;
+}
+
+extern "C"
+jvmtiIterationControl JNICALL
+HeapObjectCallback(jlong class_tag, jlong size, jlong *tag_ptr, void *user_data) {
+    jlong *data = static_cast<jlong *>(user_data);
+    *tag_ptr = *data;
+
+    limitCounter.countDown();
+    if (limitCounter.allow()) {
+        return JVMTI_ITERATION_CONTINUE;
+    }else {
+        return JVMTI_ITERATION_ABORT;
+    }
+}
+
+extern "C"
+JNIEXPORT jobjectArray JNICALL
+Java_arthas_VmTool_getInstances0(JNIEnv *env, jclass thisClass, jclass klass, jint limit) {
+    jlong tag = getTag();
+    limitCounter.init(limit);
+    jvmtiError error = jvmti->IterateOverInstancesOfClass(klass, JVMTI_HEAP_OBJECT_EITHER,
+                                               HeapObjectCallback, &tag);
+    if (error) {
+        printf("ERROR: JVMTI IterateOverInstancesOfClass failed!%u\n", error);
+        return NULL;
+    }
+
+    jint count = 0;
+    jobject *instances;
+    error = jvmti->GetObjectsWithTags(1, &tag, &count, &instances, NULL);
+    if (error) {
+        printf("ERROR: JVMTI GetObjectsWithTags failed!%u\n", error);
+        return NULL;
+    }
+
+    jobjectArray array = env->NewObjectArray(count, klass, NULL);
+    //添加元素到数组
+    for (int i = 0; i < count; i++) {
+        env->SetObjectArrayElement(array, i, instances[i]);
+    }
+    jvmti->Deallocate(reinterpret_cast<unsigned char *>(instances));
+    return array;
+}
+
+extern "C"
+JNIEXPORT jlong JNICALL
+Java_arthas_VmTool_sumInstanceSize0(JNIEnv *env, jclass thisClass, jclass klass) {
+    jlong tag = getTag();
+    limitCounter.init(-1);
+    jvmtiError error = jvmti->IterateOverInstancesOfClass(klass, JVMTI_HEAP_OBJECT_EITHER,
+                                               HeapObjectCallback, &tag);
+    if (error) {
+        printf("ERROR: JVMTI IterateOverInstancesOfClass failed!%u\n", error);
+        return -1;
+    }
+
+    jint count = 0;
+    jobject *instances;
+    error = jvmti->GetObjectsWithTags(1, &tag, &count, &instances, NULL);
+    if (error) {
+        printf("ERROR: JVMTI GetObjectsWithTags failed!%u\n", error);
+        return -1;
+    }
+
+    jlong sum = 0;
+    for (int i = 0; i < count; i++) {
+        jlong size = 0;
+        jvmti->GetObjectSize(instances[i], &size);
+        sum = sum + size;
+    }
+    jvmti->Deallocate(reinterpret_cast<unsigned char *>(instances));
+    return sum;
+}
+
+extern "C"
+JNIEXPORT jlong JNICALL Java_arthas_VmTool_getInstanceSize0
+        (JNIEnv *env, jclass thisClass, jobject instance) {
+    jlong size = -1;
+    jvmtiError error = jvmti->GetObjectSize(instance, &size);
+    if (error) {
+        printf("ERROR: JVMTI GetObjectSize failed!%u\n", error);
+    }
+    return size;
+}
+
+extern "C"
+JNIEXPORT jlong JNICALL
+Java_arthas_VmTool_countInstances0(JNIEnv *env, jclass thisClass, jclass klass) {
+    jlong tag = getTag();
+    limitCounter.init(-1);
+    jvmtiError error = jvmti->IterateOverInstancesOfClass(klass, JVMTI_HEAP_OBJECT_EITHER,
+                                               HeapObjectCallback, &tag);
+    if (error) {
+        printf("ERROR: JVMTI IterateOverInstancesOfClass failed!%u\n", error);
+        return -1;
+    }
+
+    jint count = 0;
+    error = jvmti->GetObjectsWithTags(1, &tag, &count, NULL, NULL);
+    if (error) {
+        printf("ERROR: JVMTI GetObjectsWithTags failed!%u\n", error);
+        return -1;
+    }
+    return count;
+}
+
+extern "C"
+JNIEXPORT jobjectArray JNICALL Java_arthas_VmTool_getAllLoadedClasses0
+        (JNIEnv *env, jclass thisClass, jclass kclass) {
+    jclass *classes;
+    jint count = 0;
+
+    jvmtiError error = jvmti->GetLoadedClasses(&count, &classes);
+    if (error) {
+        printf("ERROR: JVMTI GetLoadedClasses failed!\n");
+        return NULL;
+    }
+
+    jobjectArray array = env->NewObjectArray(count, kclass, NULL);
+    //添加元素到数组
+    for (int i = 0; i < count; i++) {
+        env->SetObjectArrayElement(array, i, classes[i]);
+    }
+    jvmti->Deallocate(reinterpret_cast<unsigned char *>(classes));
+    return array;
+}
\ No newline at end of file

arthas-vmtool/target/classes/arthas-git.properties

+#Generated by Git-Commit-Id-Plugin
+git.build.version=3.6.4
+git.closest.tag.commit.count=
+git.closest.tag.name=
+git.commit.id=35f73a652cf4efc30cc57df74dbab7d5c1de757a
+git.commit.id.abbrev=35f73a6
+git.commit.id.describe=35f73a6
+git.commit.id.describe-short=35f73a6
+git.commit.message.full=ci-set-3
+git.commit.message.short=ci-set-3
+git.commit.user.email=shengnan.hu@ustchcs.com
+git.commit.user.name=shengnan.hu
+git.dirty=false