aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorLulu13022002 <[email protected]>2024-08-17 22:30:21 +0200
committerGitHub <[email protected]>2024-08-17 22:30:21 +0200
commit66a97cc929740475bc8dd192382c4ae59fdeace9 (patch)
treea217a60878c8a3c3088ecabf9a11a379b8bff00e
parent0e7361704a16c8386d829c26603f3e5c0bf72b39 (diff)
downloadPaper-66a97cc929740475bc8dd192382c4ae59fdeace9.tar.gz
Paper-66a97cc929740475bc8dd192382c4ae59fdeace9.zip
Update item data sanitization (#11227)
-rw-r--r--patches/server/0975-Prevent-sending-oversized-item-data-in-equipment-and.patch46
1 files changed, 27 insertions, 19 deletions
diff --git a/patches/server/0975-Prevent-sending-oversized-item-data-in-equipment-and.patch b/patches/server/0975-Prevent-sending-oversized-item-data-in-equipment-and.patch
index c9d47f41f8..ce1eb08d18 100644
--- a/patches/server/0975-Prevent-sending-oversized-item-data-in-equipment-and.patch
+++ b/patches/server/0975-Prevent-sending-oversized-item-data-in-equipment-and.patch
@@ -7,23 +7,25 @@ Co-authored-by: Jake Potrebic <[email protected]>
diff --git a/src/main/java/io/papermc/paper/util/DataSanitizationUtil.java b/src/main/java/io/papermc/paper/util/DataSanitizationUtil.java
new file mode 100644
-index 0000000000000000000000000000000000000000..e9436f8a73ee0a02096d66e14d73edaae28d5a41
+index 0000000000000000000000000000000000000000..72483dedd3b1864fca3463d3ba3f6fad22680b97
--- /dev/null
+++ b/src/main/java/io/papermc/paper/util/DataSanitizationUtil.java
-@@ -0,0 +1,100 @@
+@@ -0,0 +1,108 @@
+package io.papermc.paper.util;
+
-+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.function.UnaryOperator;
++import it.unimi.dsi.fastutil.objects.ObjectArrayList;
+import net.minecraft.network.RegistryFriendlyByteBuf;
+import net.minecraft.network.codec.StreamCodec;
++import net.minecraft.util.Mth;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.item.Items;
+import net.minecraft.world.item.component.BundleContents;
+import net.minecraft.world.item.component.ChargedProjectiles;
+import net.minecraft.world.item.component.ItemContainerContents;
++import org.apache.commons.lang3.math.Fraction;
+import org.checkerframework.checker.nullness.qual.NonNull;
+import org.checkerframework.framework.qual.DefaultQualifier;
+
@@ -48,26 +50,33 @@ index 0000000000000000000000000000000000000000..e9436f8a73ee0a02096d66e14d73edaa
+ if (projectiles.isEmpty()) {
+ return projectiles;
+ }
-+ final List<ItemStack> items = projectiles.getItems();
-+ final List<ItemStack> sanitized = new ArrayList<>();
-+ for (int i = 0; i < Math.min(items.size(), 3); i++) {
-+ // we want to preserve item type as vanilla client can change visuals based on type
-+ sanitized.add(new ItemStack(items.get(i).getItemHolder()));
-+ }
-+ return ChargedProjectiles.of(sanitized);
++
++ return ChargedProjectiles.of(List.of(
++ new ItemStack(projectiles.contains(Items.FIREWORK_ROCKET) ? Items.FIREWORK_ROCKET : Items.ARROW)
++ ));
+ }
+
+ private static BundleContents sanitizeBundleContents(final BundleContents contents) {
++ if (contents.isEmpty()) {
++ return contents;
++ }
++
+ // Bundles change their texture based on their fullness.
-+ int sizeUsed = 0;
-+ for (final ItemStack item : contents.items()) {
-+ final int scale = 64 / item.getMaxStackSize();
-+ sizeUsed += scale * item.getCount();
++ // A bundles content weight may be anywhere from 0 to, basically, infinity.
++ // A weight of 1 is the usual maximum case
++ int sizeUsed = Mth.mulAndTruncate(contents.weight(), 64);
++ // Early out, *most* bundles should not be overfilled above a weight of one.
++ if (sizeUsed <= 64) return new BundleContents(List.of(new ItemStack(Items.PAPER, Math.max(1, sizeUsed))));
++
++ final List<ItemStack> sanitizedRepresentation = new ObjectArrayList<>(sizeUsed / 64 + 1);
++ while (sizeUsed > 0) {
++ final int stackCount = Math.min(64, sizeUsed);
++ sanitizedRepresentation.add(new ItemStack(Items.PAPER, stackCount));
++ sizeUsed -= stackCount;
+ }
+ // Now we add a single fake item that uses the same amount of slots as all other items.
-+ final List<ItemStack> items = new ArrayList<>();
-+ items.add(new ItemStack(Items.PAPER, sizeUsed));
-+ return new BundleContents(items);
++ // Ensure that potentially overstacked bundles are not represented by empty (count=0) itemstacks.
++ return new BundleContents(sanitizedRepresentation);
+ }
+
+ private static <B, A> StreamCodec<B, A> codec(final StreamCodec<B, A> delegate, final UnaryOperator<A> sanitizer) {
@@ -109,7 +118,6 @@ index 0000000000000000000000000000000000000000..e9436f8a73ee0a02096d66e14d73edaa
+
+ private DataSanitizationUtil() {
+ }
-+
+}
diff --git a/src/main/java/net/minecraft/core/component/DataComponents.java b/src/main/java/net/minecraft/core/component/DataComponents.java
index c9aef759c1485da753e820f9b509117ca50a31e4..60757f8df706cba92350d73503b73913cff3bcfc 100644
@@ -201,7 +209,7 @@ index 0e7ace92522fbd4cef7b2c2b8a0f8b86c2cce192..1d849ce4e2c85f149af25318b8ffb6dc
((LivingEntity) this.entity).detectEquipmentUpdatesPublic(); // CraftBukkit - SPIGOT-3789: sync again immediately after sending
}
diff --git a/src/main/java/net/minecraft/server/network/ServerGamePacketListenerImpl.java b/src/main/java/net/minecraft/server/network/ServerGamePacketListenerImpl.java
-index 4f9a17e7cd5aca2e8bd55b4892a20dce181289cd..a04fdcecf00f128f0afba1a33fad1d9e7694ef63 100644
+index d5f33021754047971b69f90caff3664f596d6eec..9da231673888bb8ba2ed922204ddfd0220f091b1 100644
--- a/src/main/java/net/minecraft/server/network/ServerGamePacketListenerImpl.java
+++ b/src/main/java/net/minecraft/server/network/ServerGamePacketListenerImpl.java
@@ -2727,7 +2727,7 @@ public class ServerGamePacketListenerImpl extends ServerCommonPacketListenerImpl