From b1cddec0e4f43f949fd018d9e38718405da2c965 Mon Sep 17 00:00:00 2001
From: Gravit <gravit.min@ya.ru>
Date: Mon, 24 Sep 2018 18:36:20 +0700
Subject: [PATCH] ServerAgent

---
 .../gravit/launcher/server/ServerAgent.java   | 56 +++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 Launcher/src/main/java/ru/gravit/launcher/server/ServerAgent.java

diff --git a/Launcher/src/main/java/ru/gravit/launcher/server/ServerAgent.java b/Launcher/src/main/java/ru/gravit/launcher/server/ServerAgent.java
new file mode 100644
index 00000000..4b699ec4
--- /dev/null
+++ b/Launcher/src/main/java/ru/gravit/launcher/server/ServerAgent.java
@@ -0,0 +1,56 @@
+package ru.gravit.launcher.server;
+
+import ru.gravit.utils.helper.IOHelper;
+import ru.gravit.utils.helper.LogHelper;
+
+import java.io.IOException;
+import java.lang.instrument.Instrumentation;
+import java.nio.file.*;
+import java.nio.file.attribute.BasicFileAttributes;
+import java.util.Collections;
+import java.util.jar.JarFile;
+
+public class ServerAgent {
+    private static boolean isAgentStarted=false;
+    public static Instrumentation inst;
+    public static final class StarterVisitor extends SimpleFileVisitor<Path> {
+        private Instrumentation inst;
+
+        public StarterVisitor(Instrumentation inst) {
+            this.inst = inst;
+        }
+
+        @Override
+        public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
+            if (file.toFile().getName().endsWith(".jar")) addJVMClassPath(new JarFile(file.toFile()));
+            return super.visitFile(file, attrs);
+        }
+    }
+    public static void addJVMClassPath(String path) throws IOException {
+        LogHelper.debug("Load %s",path);
+        inst.appendToSystemClassLoaderSearch(new JarFile(path));
+    }
+    public static void addJVMClassPath(JarFile file) throws IOException {
+        LogHelper.debug("Load %s",file.getName());
+        inst.appendToSystemClassLoaderSearch(file);
+    }
+    public boolean isAgentStarted()
+    {
+        return isAgentStarted;
+    }
+    public static long getObjSize(Object obj) {
+        return inst.getObjectSize(obj);
+    }
+
+    public static void premain(String agentArgument, Instrumentation instrumentation) {
+        LogHelper.debug("Server Agent");
+        inst = instrumentation;
+        isAgentStarted = true;
+
+        try {
+            Files.walkFileTree(Paths.get("libraries"), Collections.singleton(FileVisitOption.FOLLOW_LINKS), Integer.MAX_VALUE, new StarterVisitor(inst));
+        } catch (IOException e) {
+            e.printStackTrace(System.err);
+        }
+    }
+}