Add Agent

build.gradle.kts

@@ -46,6 +46,7 @@ tasks {
  duplicatesStrategy = DuplicatesStrategy.EXCLUDE
  manifest.attributes["Specification-Version"] = project.version
  manifest.attributes["Main-Class"] = "${project.group}.${project.name.toLowerCase()}.Main"
+ manifest.attributes["Agent-Class"] = "${project.group}.${project.name.toLowerCase()}.Agent"
  from(configurations.compileClasspath.get().map { if (it.isDirectory) it else zipTree(it) })
  }
 

src/main/kotlin/dev/shota/decompiler/Agent.kt

@@ -0,0 +1,8 @@
+@file:JvmName("Agent")
+package dev.shota.decompiler
+
+import java.lang.instrument.Instrumentation
+
+fun agentmain(args: String?, inst: Instrumentation) {
+ println("Hello World!")
+}

src/main/kotlin/dev/shota/decompiler/window/popup/Processes.kt

@@ -7,24 +7,32 @@ import javafx.scene.control.Button
 import javafx.scene.control.TableColumn
 import javafx.scene.control.TableView
 import javafx.scene.layout.VBox
+import java.io.File
+import java.lang.management.ManagementFactory
 
 class Processes : Popup("file.openProcess") {
 
  init {
  val root = VBox()
+
  val table = TableView<VirtualMachineDescriptor>()
+ root.children.add(table)
+ for (vm in VirtualMachine.list())
+ if (ManagementFactory.getRuntimeMXBean().pid != vm.id().toLong())
+ table.items.add(vm)
+
  val id = TableColumn<VirtualMachineDescriptor, String>("ID")
  val entry = TableColumn<VirtualMachineDescriptor, String>("Entry")
  id.setCellValueFactory { SimpleStringProperty(it.value.id()) }
  entry.setCellValueFactory { SimpleStringProperty(it.value.displayName().split(" ").first()) }
  table.columns.addAll(id, entry)
- root.children.add(table)
-
- for (vm in VirtualMachine.list())
- table.items.add(vm)
 
  val button = Button("Open")
- button.setOnAction {}
+ button.setOnAction {
+ val vm = VirtualMachine.attach(table.selectionModel.selectedItem)
+ vm.loadAgent(File(javaClass.protectionDomain.codeSource.location.path).path)
+ vm.detach()
+ }
  root.children.add(button)
 
  run(root)