From 6fd08c3b9d7946bc3a45365e249b2cd3f169ebef Mon Sep 17 00:00:00 2001
From: Chaoscaot <chaoscaot444@gmail.com>
Date: Thu, 30 Dec 2021 21:12:00 +0100
Subject: [PATCH] Fix Invalid Schem Names

Signed-off-by: Chaoscaot <chaoscaot444@gmail.com>
---
 .../steamwar/inventory/SchematicSelector.java | 10 +++++---
 .../src/de/steamwar/sql/SchematicNode.java    | 24 +++++++++++++++++++
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/SpigotCore_Main/src/de/steamwar/inventory/SchematicSelector.java b/SpigotCore_Main/src/de/steamwar/inventory/SchematicSelector.java
index 90e96d4..6a98371 100644
--- a/SpigotCore_Main/src/de/steamwar/inventory/SchematicSelector.java
+++ b/SpigotCore_Main/src/de/steamwar/inventory/SchematicSelector.java
@@ -228,10 +228,14 @@ public class SchematicSelector {
         SWAnvilInv inv = new SWAnvilInv(player, Core.MESSAGE.parse("SCHEM_SELECTOR_CREATE_DIR_TITLE", player));
         inv.setItem(Material.CHEST);
         inv.setCallback(s -> {
-            if(injectable.onFolderCreate(this, s)) {
-                SchematicNode.createSchematicDirectory(user.getId(), s, parent==null?0:parent.getId());
-                openList(parent);
+            if(!SchematicNode.invalidSchemName(new String[] {s})) {
+                if(injectable.onFolderCreate(this, s)) {
+                    SchematicNode.createSchematicDirectory(user.getId(), s, parent==null?0:parent.getId());
+                    openList(parent);
+                }
+                return;
             }
+            player.closeInventory();
         });
         inv.open();
     }
diff --git a/SpigotCore_Main/src/de/steamwar/sql/SchematicNode.java b/SpigotCore_Main/src/de/steamwar/sql/SchematicNode.java
index d888e5e..08abbe2 100644
--- a/SpigotCore_Main/src/de/steamwar/sql/SchematicNode.java
+++ b/SpigotCore_Main/src/de/steamwar/sql/SchematicNode.java
@@ -604,4 +604,28 @@ public class SchematicNode {
         list.remove("//copy");
         return list;
     }
+
+    private static final List<String> FORBIDDEN_NAMES = Collections.unmodifiableList(Arrays.asList("public"));
+    public static boolean invalidSchemName(String[] layers) {
+        for (String layer : layers) {
+            if (layer.isEmpty()) {
+                return true;
+            }
+            if (layer.contains("/") ||
+                    layer.contains("\\") ||
+                    layer.contains("<") ||
+                    layer.contains(">") ||
+                    layer.contains("^") ||
+                    layer.contains("°") ||
+                    layer.contains("'") ||
+                    layer.contains("\"") ||
+                    layer.contains(" ")) {
+                return true;
+            }
+            if(FORBIDDEN_NAMES.contains(layer.toLowerCase())) {
+                return true;
+            }
+        }
+        return false;
+    }
 }