validate mannequin pose

Author: Lulu13022002

paper-server/patches/sources/net/minecraft/world/entity/decoration/Mannequin.java.patch

@@ -0,0 +1,11 @@
+--- a/net/minecraft/world/entity/decoration/Mannequin.java
++++ b/net/minecraft/world/entity/decoration/Mannequin.java
+@@ -33,7 +_,7 @@
+         Mannequin.class, EntityDataSerializers.OPTIONAL_COMPONENT
+     );
+     public static final byte ALL_LAYERS = (byte)Arrays.stream(PlayerModelPart.values()).mapToInt(PlayerModelPart::getMask).reduce(0, (i, i1) -> i | i1);
+-    private static final Set<Pose> VALID_POSES = Set.of(Pose.STANDING, Pose.CROUCHING, Pose.SWIMMING, Pose.FALL_FLYING, Pose.SLEEPING);
++    public static final Set<Pose> VALID_POSES = Set.of(Pose.STANDING, Pose.CROUCHING, Pose.SWIMMING, Pose.FALL_FLYING, Pose.SLEEPING); // Paper - public
+     public static final Codec<Pose> POSE_CODEC = Pose.CODEC
+         .validate(pose -> VALID_POSES.contains(pose) ? DataResult.success(pose) : DataResult.error(() -> "Invalid pose: " + pose.getSerializedName()));
+     private static final Codec<Byte> LAYERS_CODEC = PlayerModelPart.CODEC

paper-server/src/main/java/org/bukkit/craftbukkit/entity/CraftEntity.java

@@ -7,15 +7,16 @@
 import com.google.common.collect.Lists;
 import com.mojang.logging.LogUtils;
 import io.papermc.paper.datacomponent.DataComponentType;
+import io.papermc.paper.entity.LookAnchor;
 import io.papermc.paper.entity.TeleportFlag;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
 import java.util.UUID;
-import io.papermc.paper.entity.LookAnchor;
 import java.util.concurrent.CompletableFuture;
 import net.kyori.adventure.pointer.PointersSupplier;
 import net.kyori.adventure.util.TriState;
+import net.md_5.bungee.api.chat.BaseComponent;
 import net.minecraft.nbt.CompoundTag;
 import net.minecraft.network.chat.Component;
 import net.minecraft.server.MinecraftServer;
@@ -74,8 +75,6 @@
 import org.bukkit.util.BoundingBox;
 import org.bukkit.util.NumberConversions;
 import org.bukkit.util.Vector;
-
-import net.md_5.bungee.api.chat.BaseComponent; // Spigot
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.slf4j.Logger;
@@ -960,10 +959,14 @@ public boolean isSneaking() {
 
     @Override
     public void setPose(Pose pose, boolean fixed) {
-        Preconditions.checkNotNull(pose, "Pose cannot be null");
+        Preconditions.checkArgument(pose != null, "pose cannot be null");
+        this.setPose0(net.minecraft.world.entity.Pose.values()[pose.ordinal()], fixed);
+    }
+
+    public void setPose0(net.minecraft.world.entity.Pose pose, boolean fixed) {
         final Entity handle = this.getHandle();
         handle.fixedPose = false;
-        handle.setPose(net.minecraft.world.entity.Pose.values()[pose.ordinal()]);
+        handle.setPose(pose);
         handle.fixedPose = fixed;
     }
 

paper-server/src/main/java/org/bukkit/craftbukkit/entity/CraftMannequin.java

@@ -12,6 +12,7 @@
 import net.minecraft.world.entity.HumanoidArm;
 import org.bukkit.craftbukkit.CraftServer;
 import org.bukkit.entity.Mannequin;
+import org.bukkit.entity.Pose;
 import org.bukkit.inventory.MainHand;
 import org.jspecify.annotations.NullMarked;
 import org.jspecify.annotations.Nullable;
@@ -27,6 +28,20 @@ public net.minecraft.world.entity.decoration.Mannequin getHandle() {
         return (net.minecraft.world.entity.decoration.Mannequin) this.entity;
     }
 
+    @Override
+    public void setPose(Pose pose, boolean fixed) {
+        Preconditions.checkArgument(pose != null, "pose cannot be null");
+        net.minecraft.world.entity.Pose internalPose = net.minecraft.world.entity.Pose.values()[pose.ordinal()];
+        if (!net.minecraft.world.entity.decoration.Mannequin.VALID_POSES.contains(internalPose)) {
+            throw new IllegalArgumentException("Invalid pose '%s', expected one of: %s".formatted(
+                pose.name(),
+                net.minecraft.world.entity.decoration.Mannequin.VALID_POSES.stream().map(p -> Pose.values()[p.ordinal()]).toList() // name doesn't match
+            ));
+        }
+
+        this.setPose0(internalPose, fixed);
+    }
+
     @Override
     public io.papermc.paper.datacomponent.item.ResolvableProfile getProfile() {
         return new PaperResolvableProfile(this.getHandle().getProfile());