Allow loading agent on startup

parca-dev · Jan 4, 2022 · b169e83 · b169e83
1 parent d9843a0
commit b169e83
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/c/perf-map-agent.c b/src/c/perf-map-agent.c
@@ -337,6 +337,20 @@ jvmtiError set_callbacks(jvmtiEnv *jvmti) {
     return (*jvmti)->SetEventCallbacks(jvmti, &callbacks, (jint)sizeof(callbacks));
 }
 
+JNIEXPORT jint JNICALL 
+Agent_OnLoad(JavaVM *vm, char *options, void *reserved)
+{
+    open_map_file();
+
+    jvmtiEnv *jvmti;
+    (*vm)->GetEnv(vm, (void **)&jvmti, JVMTI_VERSION_1);
+    enable_capabilities(jvmti);
+    set_callbacks(jvmti);
+    set_notification_mode(jvmti, JVMTI_ENABLE);
+
+    return 0;
+}
+
 JNIEXPORT jint JNICALL
 Agent_OnAttach(JavaVM *vm, char *options, void *reserved) {
     open_map_file();