Fork friendly byte buf correctly

Author: kennytv

paper-server/patches/sources/net/minecraft/nbt/NbtAccounter.java.patch

@@ -0,0 +1,19 @@
+--- a/net/minecraft/nbt/NbtAccounter.java
++++ b/net/minecraft/nbt/NbtAccounter.java
+@@ -48,6 +_,16 @@
+         }
+     }
+ 
++    // Paper start - Track codec depth
++    public void pushDepth(final int depth) {
++        if (this.depth + depth >= this.maxDepth) {
++            throw new NbtAccounterException("Tried to read NBT tag with too high complexity, depth > " + this.maxDepth);
++        } else {
++            this.depth += depth;
++        }
++    }
++    // Paper end - Track codec depth
++
+     public void pushDepth() {
+         if (this.depth >= this.maxDepth) {
+             throw new NbtAccounterException("Tried to read NBT tag with too high complexity, depth > " + this.maxDepth);

paper-server/patches/sources/net/minecraft/network/RegistryFriendlyByteBuf.java.patch

@@ -0,0 +1,18 @@
+--- a/net/minecraft/network/RegistryFriendlyByteBuf.java
++++ b/net/minecraft/network/RegistryFriendlyByteBuf.java
+@@ -12,6 +_,15 @@
+         this.registryAccess = registryAccess;
+     }
+ 
++    // Paper start - Track codec depth
++    public RegistryFriendlyByteBuf(final ByteBuf source, final RegistryFriendlyByteBuf parent) {
++        super(source);
++        this.registryAccess = parent.registryAccess;
++        this.trackCodecDepth = parent.trackCodecDepth;
++        this.codecDepth = parent.codecDepth;
++    }
++    // Paper end - Track codec depth
++
+     public RegistryAccess registryAccess() {
+         return this.registryAccess;
+     }

paper-server/patches/sources/net/minecraft/network/codec/ByteBufCodecs.java.patch

@@ -1,5 +1,20 @@
 --- a/net/minecraft/network/codec/ByteBufCodecs.java
 +++ b/net/minecraft/network/codec/ByteBufCodecs.java
+@@ -334,7 +_,13 @@
+         return new StreamCodec<ByteBuf, Tag>() {
+             @Override
+             public Tag decode(final ByteBuf input) {
+-                Tag result = FriendlyByteBuf.readNbt(input, accounter.get());
++                // Paper start - Track codec depth
++                final NbtAccounter acc = accounter.get();
++                if (input instanceof FriendlyByteBuf friendlyByteBuf && friendlyByteBuf.trackCodecDepth) {
++                    acc.pushDepth(friendlyByteBuf.codecDepth * 3);
++                }
++                Tag result = FriendlyByteBuf.readNbt(input, acc);
++                // Paper end - Track codec depth
+                 if (result == null) {
+                     throw new DecoderException("Expected non-null compound tag");
+                 } else {
 @@ -418,6 +_,48 @@
          };
      }
@@ -49,3 +64,12 @@
      static <B extends ByteBuf, V> StreamCodec<B, Optional<V>> optional(final StreamCodec<? super B, V> original) {
          return new StreamCodec<B, Optional<V>>() {
              @Override
+@@ -594,7 +_,7 @@
+     }
+ 
+     static <V> StreamCodec.CodecOperation<RegistryFriendlyByteBuf, V, V> registryFriendlyLengthPrefixed(final int maxSize) {
+-        return lengthPrefixed(maxSize, (parent, child) -> new RegistryFriendlyByteBuf(child, parent.registryAccess()));
++        return lengthPrefixed(maxSize, (parent, child) -> new RegistryFriendlyByteBuf(child, parent)); // Paper - Track codec depth
+     }
+ 
+     static <T> StreamCodec<ByteBuf, T> idMapper(final IntFunction<T> byId, final ToIntFunction<T> toId) {