prevent crash on book copy

MeteorDevelopment · Nov 29, 2023 · a4d3cac · a4d3cac
1 parent 66544a9
commit a4d3cac
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/main/java/meteordevelopment/meteorclient/mixin/BookScreenMixin.java b/src/main/java/meteordevelopment/meteorclient/mixin/BookScreenMixin.java
@@ -8,6 +8,7 @@
 import it.unimi.dsi.fastutil.io.FastByteArrayOutputStream;
 import meteordevelopment.meteorclient.gui.GuiThemes;
 import meteordevelopment.meteorclient.gui.screens.EditBookTitleAndAuthorScreen;
+import meteordevelopment.meteorclient.utils.player.ChatUtils;
 import net.minecraft.client.gui.screen.Screen;
 import net.minecraft.client.gui.screen.ingame.BookScreen;
 import net.minecraft.client.gui.widget.ButtonWidget;
@@ -20,6 +21,8 @@
 import net.minecraft.text.Text;
 import net.minecraft.util.Hand;
 import org.lwjgl.glfw.GLFW;
+import org.lwjgl.system.MemoryStack;
+import org.lwjgl.system.MemoryUtil;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
 import org.spongepowered.asm.mixin.injection.At;
@@ -63,7 +66,17 @@ private void onInit(CallbackInfo info) {
                         e.printStackTrace();
                     }
 
-                    GLFW.glfwSetClipboardString(mc.getWindow().getHandle(), Base64.getEncoder().encodeToString(bytes.array));
+                    String encoded = Base64.getEncoder().encodeToString(bytes.array);
+
+                    @SuppressWarnings("resource")
+                    long available = MemoryStack.stackGet().getPointer();
+                    long size = MemoryUtil.memLengthUTF8(encoded, true);
+
+                    if (size > available) {
+                        ChatUtils.error("Could not copy to clipboard: Out of memory.");
+                    } else {
+                        GLFW.glfwSetClipboardString(mc.getWindow().getHandle(), encoded);
+                    }
                 })
                 .position(4, 4)
                 .size(120, 20)