Initial commit

Author: Artur114Projects

.gitignore

@@ -0,0 +1,43 @@
+.gradle
+build/
+!gradle/wrapper/gradle-wrapper.jar
+!**/src/main/**/build/
+!**/src/test/**/build/
+.kotlin
+
+### IntelliJ IDEA ###
+.idea/modules.xml
+.idea/jarRepositories.xml
+.idea/compiler.xml
+.idea/libraries/
+*.iws
+*.iml
+*.ipr
+out/
+!**/src/main/**/out/
+!**/src/test/**/out/
+
+### Eclipse ###
+.apt_generated
+.classpath
+.factorypath
+.project
+.settings
+.springBeans
+.sts4-cache
+bin/
+!**/src/main/**/bin/
+!**/src/test/**/bin/
+
+### NetBeans ###
+/nbproject/private/
+/nbbuild/
+/dist/
+/nbdist/
+/.nb-gradle/
+
+### VS Code ###
+.vscode/
+
+### Mac OS ###
+.DS_Store

build.gradle

@@ -0,0 +1,52 @@
+plugins {
+    id 'java'
+}
+
+group = 'com.artur114.bytecodegrab.main'
+archivesBaseName = "ByteCodeGrabber"
+version = '1.0-SNAPSHOT'
+
+repositories {
+    mavenCentral()
+}
+
+configurations {
+    include; implementation.extendsFrom(include)
+}
+
+dependencies {
+    testImplementation 'org.junit.jupiter:junit-jupiter-api:5.10.0'
+    testImplementation 'org.junit.jupiter:junit-jupiter-params:5.10.0'
+    testRuntimeOnly 'org.junit.jupiter:junit-jupiter-engine:5.10.0'
+    implementation files(getToolsJar())
+
+    include 'org.apache.logging.log4j:log4j-slf4j2-impl:2.20.0'
+    include 'org.apache.logging.log4j:log4j-api:2.0-beta9'
+    include 'com.google.code.gson:gson:2.8.9'
+    include 'com.formdev:flatlaf:3.7'
+}
+
+jar {
+    manifest { attributes 'Main-Class': 'com.artur114.bytecodegrab.main.Main' }
+    from { configurations.include.collect { it.isDirectory() ? it : zipTree(it) } }
+    duplicatesStrategy = DuplicatesStrategy.EXCLUDE
+}
+
+tasks.withType(JavaCompile).configureEach {
+    options.encoding = 'UTF-8'
+}
+
+test {
+    useJUnitPlatform()
+}
+
+def getToolsJar() {
+    def jar = files(
+            "${System.properties['java.home']}/../lib/tools.jar",
+            "${System.properties['java.home']}/lib/tools.jar"
+    ).find { it.exists() }
+    if (jar != null) {
+        return jar
+    }
+    throw new GradleException()
+}

settings.gradle

@@ -0,0 +1 @@
+rootProject.name = 'ByteCodeGrabber'

src/main/java/com/artur114/bytecodegrab/agent/AbortListener.java

@@ -0,0 +1,48 @@
+package com.artur114.bytecodegrab.agent;
+
+import java.io.DataInputStream;
+import java.io.EOFException;
+import java.io.IOException;
+
+public class AbortListener extends Thread {
+    private final DataInputStream connection;
+    private final IAgentLogger logger;
+    private boolean aborted = false;
+
+    public AbortListener(IAgentLogger logger, SocketA connection) {
+        try {
+            this.connection = new DataInputStream(connection.getInputStream());
+            this.logger = logger;
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public boolean isAborted() {
+        return this.aborted;
+    }
+
+    public void shutdown() {
+        this.interrupt();
+        try {
+            this.connection.close();
+        } catch (IOException ignored) {}
+    }
+
+    @Override
+    public void run() {
+        try {
+            String message = this.connection.readUTF();
+
+            if ("ABORT".equals(message)) {
+                this.aborted = true;
+            } else {
+                this.logger.warn("AbortListener listen not abort message {}!", message);
+            }
+        } catch (EOFException e) {
+            this.aborted = true;
+        } catch (IOException e) {
+            e.printStackTrace(System.err);
+        }
+    }
+}

src/main/java/com/artur114/bytecodegrab/agent/AgentLoggerAsync.java

@@ -0,0 +1,78 @@
+package com.artur114.bytecodegrab.agent;
+
+public class AgentLoggerAsync implements IAgentLogger {
+    private final AsyncTaskProcessor processor;
+    private final IAgentLogger output;
+
+    public AgentLoggerAsync(IAgentLogger output) {
+        this.output = output;
+
+        this.processor = new AsyncTaskProcessor();
+        this.processor.setName("BCG Logging Thread");
+        this.processor.start();
+    }
+
+    public void shutdown() {
+        this.processor.shutdown();
+    }
+
+    @Override
+    public void warn(String log) {
+        this.processor.addTask(() -> this.output.warn(log));
+    }
+
+    @Override
+    public void info(String log) {
+        this.processor.addTask(() -> this.output.info(log));
+    }
+
+    @Override
+    public void error(String log) {
+        this.processor.addTask(() -> this.output.error(log));
+    }
+
+    @Override
+    public void warn(String log, Object arg) {
+        this.processor.addTask(() -> this.output.warn(log, arg));
+    }
+
+    @Override
+    public void info(String log, Object arg) {
+        this.processor.addTask(() -> this.output.info(log, arg));
+    }
+
+    @Override
+    public void error(String log, Object arg) {
+        this.processor.addTask(() -> this.output.error(log, arg));
+    }
+
+    @Override
+    public void warn(String log, Object... args) {
+        this.processor.addTask(() -> this.output.warn(log, args));
+    }
+
+    @Override
+    public void info(String log, Object... args) {
+        this.processor.addTask(() -> this.output.info(log, args));
+    }
+
+    @Override
+    public void error(String log, Object... args) {
+        this.processor.addTask(() -> this.output.error(log, args));
+    }
+
+    @Override
+    public void warn(String log, Object arg, Object arg1) {
+        this.processor.addTask(() -> this.output.warn(log, arg, arg1));
+    }
+
+    @Override
+    public void info(String log, Object arg, Object arg1) {
+        this.processor.addTask(() -> this.output.info(log, arg, arg1));
+    }
+
+    @Override
+    public void error(String log, Object arg, Object arg1) {
+        this.processor.addTask(() -> this.output.error(log, arg, arg1));
+    }
+}

src/main/java/com/artur114/bytecodegrab/agent/AgentLoggerLog4j.java

@@ -0,0 +1,68 @@
+package com.artur114.bytecodegrab.agent;
+
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+public class AgentLoggerLog4j implements IAgentLogger {
+    private final Logger logger = LogManager.getLogger("BYTECODEGARB-AGENT");
+
+    @Override
+    public void warn(String log) {
+        this.logger.warn(log);
+    }
+
+    @Override
+    public void info(String log) {
+        this.logger.info(log);
+    }
+
+    @Override
+    public void error(String log) {
+        this.logger.error(log);
+    }
+
+    @Override
+    public void warn(String log, Object arg) {
+        this.logger.warn(log, arg);
+    }
+
+    @Override
+    public void info(String log, Object arg) {
+        this.logger.info(log, arg);
+    }
+
+    @Override
+    public void error(String log, Object arg) {
+        this.logger.error(log, arg);
+    }
+
+    @Override
+    public void warn(String log, Object... args) {
+        this.logger.warn(log, args);
+    }
+
+    @Override
+    public void info(String log, Object... args) {
+        this.logger.info(log, args);
+    }
+
+    @Override
+    public void error(String log, Object... args) {
+        this.logger.error(log, args);
+    }
+
+    @Override
+    public void warn(String log, Object arg, Object arg1) {
+        this.logger.warn(log, arg, arg1);
+    }
+
+    @Override
+    public void info(String log, Object arg, Object arg1) {
+        this.logger.info(log, arg, arg1);
+    }
+
+    @Override
+    public void error(String log, Object arg, Object arg1) {
+        this.logger.error(log, arg, arg1);
+    }
+}