From da01fbb447f17117ac3782e76ca783190230757f Mon Sep 17 00:00:00 2001
From: Lixfel <agga-games@gmx.de>
Date: Wed, 24 Jan 2024 11:29:44 +0100
Subject: [PATCH 1/2] Reduce checkpoint exceptions by TCP retry and silent jar
 updates

---
 .../src/de/steamwar/core/CheckpointUtils.java       | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/SpigotCore_Main/src/de/steamwar/core/CheckpointUtils.java b/SpigotCore_Main/src/de/steamwar/core/CheckpointUtils.java
index 15d063a..6f0af7f 100644
--- a/SpigotCore_Main/src/de/steamwar/core/CheckpointUtils.java
+++ b/SpigotCore_Main/src/de/steamwar/core/CheckpointUtils.java
@@ -69,8 +69,17 @@ public class CheckpointUtils {
         try {
             freezeInternal(path);
         } catch (Exception e) {
+            String message = e.getMessage() != null ? e.getMessage() : "";
+            if(message.contains("Connected TCP socket")) {
+                Core.getInstance().getLogger().log(Level.INFO, "Connected TCP socket, waiting for checkpointing");
+                Bukkit.getScheduler().runTaskLater(Core.getInstance(), CheckpointUtils::freeze, 1);
+                return;
+            }
+
             Bukkit.shutdown();
-            throw new SecurityException(e);
+
+            if(message.contains("Can't dump ghost file")) // File/Jar has been updated
+                throw new SecurityException(e);
         } finally {
             // Delete checkpoint
             try (Stream<Path> stream = Files.walk(path)) {
@@ -109,8 +118,6 @@ public class CheckpointUtils {
         try {
             criu.checkpointJVM();
         } catch (JVMCRIUException e) {
-            Bukkit.shutdown();
-
             Path logfile = path.resolve("criu.log");
             if(logfile.toFile().exists())
                 throw new IllegalStateException("Could not create checkpoint, criu log:\n" + new String(Files.readAllBytes(logfile)), e);

From 65f83eb0aafbf0a2c7839540dd085d40a36754db Mon Sep 17 00:00:00 2001
From: Lixfel <agga-games@gmx.de>
Date: Wed, 24 Jan 2024 11:30:08 +0100
Subject: [PATCH 2/2] Utilize "new" Reflection utils in CommandRemover

---
 .../src/de/steamwar/core/CommandRemover.java  | 37 +++++--------------
 1 file changed, 9 insertions(+), 28 deletions(-)

diff --git a/SpigotCore_Main/src/de/steamwar/core/CommandRemover.java b/SpigotCore_Main/src/de/steamwar/core/CommandRemover.java
index 6543e64..bbce137 100644
--- a/SpigotCore_Main/src/de/steamwar/core/CommandRemover.java
+++ b/SpigotCore_Main/src/de/steamwar/core/CommandRemover.java
@@ -19,41 +19,22 @@
 
 package de.steamwar.core;
 
+import com.comphenix.tinyprotocol.Reflection;
+import lombok.experimental.UtilityClass;
 import org.bukkit.Bukkit;
 import org.bukkit.command.Command;
 import org.bukkit.command.SimpleCommandMap;
 
-import java.lang.reflect.Field;
 import java.util.Map;
-import java.util.logging.Level;
 
+@UtilityClass
 public class CommandRemover {
-    private CommandRemover(){}
-
-    private static String packageName = Bukkit.getServer().getClass().getPackage().getName();
-    private static String version = packageName.substring(packageName.lastIndexOf('.') + 1);
-
-    public static void removeAll(String... cmds){
-        for (String cmd : cmds) removeCommand(cmd);
-    }
-
-
-    private static void removeCommand(String command) {
-        try {
-            Class<?> serverClass = Class.forName("org.bukkit.craftbukkit." + version + ".CraftServer");
-
-
-            Field f1 = serverClass.getDeclaredField("commandMap");
-            f1.setAccessible(true);
-            SimpleCommandMap commandMap = (SimpleCommandMap) f1.get(Bukkit.getServer());
-
-            Field f2 = SimpleCommandMap.class.getDeclaredField("knownCommands");
-            f2.setAccessible(true);
-            Map<String, Command> knownCommands = (Map<String, Command>) f2.get(commandMap);
-
-            knownCommands.remove(command.toLowerCase());
-        } catch (Exception e) {
-            Bukkit.getLogger().log(Level.SEVERE, "Could not remove command", e);
+    private static final Reflection.FieldAccessor<SimpleCommandMap> commandMap = Reflection.getField("{obc}.CraftServer", "commandMap", SimpleCommandMap.class);
+    private static final Reflection.FieldAccessor<Map> knownCommands = Reflection.getField(SimpleCommandMap.class, "knownCommands", Map.class);
+    public static void removeAll(String... cmds) {
+        Map<String, Command> knownCmds = knownCommands.get(commandMap.get(Bukkit.getServer()));
+        for (String cmd : cmds) {
+            knownCmds.remove(cmd.toLowerCase());
         }
     }
 }