From 6b146ebec2e000db6826ca2db272908b9c4f08f8 Mon Sep 17 00:00:00 2001
From: Zach Brown <1254957+zachbr@users.noreply.github.com>
Date: Sun, 13 Nov 2016 19:36:23 -0600
Subject: [PATCH] Expand NBT filtering from last commit

cc GH-478
---
 ...a-from-ArmorStand-and-SpawnEgg-items.patch | 43 +++++++++++++++++--
 1 file changed, 39 insertions(+), 4 deletions(-)

diff --git a/Spigot-Server-Patches/Filter-bad-data-from-ArmorStand-and-SpawnEgg-items.patch b/Spigot-Server-Patches/Filter-bad-data-from-ArmorStand-and-SpawnEgg-items.patch
index 70b3f7b619..7dc39dd3a5 100644
--- a/Spigot-Server-Patches/Filter-bad-data-from-ArmorStand-and-SpawnEgg-items.patch
+++ b/Spigot-Server-Patches/Filter-bad-data-from-ArmorStand-and-SpawnEgg-items.patch
@@ -4,20 +4,55 @@ Date: Sat, 12 Nov 2016 23:25:22 -0600
 Subject: [PATCH] Filter bad data from ArmorStand and SpawnEgg items
 
 
+diff --git a/src/main/java/com/destroystokyo/paper/PaperWorldConfig.java b/src/main/java/com/destroystokyo/paper/PaperWorldConfig.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/com/destroystokyo/paper/PaperWorldConfig.java
++++ b/src/main/java/com/destroystokyo/paper/PaperWorldConfig.java
+@@ -0,0 +0,0 @@ public class PaperWorldConfig {
+     private void removeCorruptTEs() {
+         removeCorruptTEs = getBoolean("remove-corrupt-tile-entities", false);
+     }
++
++    public boolean filterNBTFromSpawnEgg = true;
++    private void fitlerNBTFromSpawnEgg() {
++        filterNBTFromSpawnEgg = getBoolean("filter-nbt-data-from-spawn-eggs-and-related", true);
++        if (!filterNBTFromSpawnEgg) {
++            Bukkit.getLogger().warning("Spawn Egg and Armor Stand NBT filtering disabled, this is a potential security risk");
++        }
++    }
+ }
+diff --git a/src/main/java/net/minecraft/server/EntityFallingBlock.java b/src/main/java/net/minecraft/server/EntityFallingBlock.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/net/minecraft/server/EntityFallingBlock.java
++++ b/src/main/java/net/minecraft/server/EntityFallingBlock.java
+@@ -0,0 +0,0 @@ public class EntityFallingBlock extends Entity {
+             this.block = Block.getById(nbttagcompound.getByte("Tile") & 255).fromLegacyData(i);
+         }
+ 
++        // Paper start - Block FallingBlocks with Command Blocks
++        if (this.world.paperConfig.filterNBTFromSpawnEgg && this.block.getBlock() == Blocks.COMMAND_BLOCK) {
++            this.block = Blocks.STONE.getBlockData();
++        }
++        // Paper end
++
+         this.ticksLived = nbttagcompound.getInt("Time");
+         Block block = this.block.getBlock();
+ 
 diff --git a/src/main/java/net/minecraft/server/ItemMonsterEgg.java b/src/main/java/net/minecraft/server/ItemMonsterEgg.java
 index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
 --- a/src/main/java/net/minecraft/server/ItemMonsterEgg.java
 +++ b/src/main/java/net/minecraft/server/ItemMonsterEgg.java
 @@ -0,0 +0,0 @@ public class ItemMonsterEgg extends Item {
- 
                  NBTTagCompound nbttagcompound1 = entity.e(new NBTTagCompound());
                  UUID uuid = entity.getUniqueID();
--
+ 
 -                nbttagcompound1.a(nbttagcompound.getCompound("EntityTag"));
 +                // Paper start - Filter out position and motion information
 +                final NBTTagCompound entityTag = nbttagcompound.getCompound("EntityTag");
-+                entityTag.remove("Pos");
-+                entityTag.remove("Motion");
++                if (world.paperConfig.filterNBTFromSpawnEgg) {
++                    entityTag.remove("Pos");
++                    entityTag.remove("Motion");
++                }
 +                nbttagcompound1.a(entityTag);
 +                // Paper end
                  entity.a(uuid);