Release 1.1. New config Options.

Author: gamerforEA

build.gradle

@@ -19,7 +19,7 @@ apply plugin: 'forge'
 
 [compileJava, compileTestJava]*.options*.encoding = 'UTF-8'
 
-version = "1.7.10-1.0"
+version = "1.7.10-1.1"
 group= "com.gamerforea"
 archivesBaseName = "PacketUnlimiter"
 

src/main/java/com/gamerforea/packetunlimiter/CoreMod.java

@@ -13,16 +13,22 @@ public final class CoreMod implements IFMLLoadingPlugin
 {
 	public static final String MODID = "PacketUnlimiter";
 	public static final String NAME = "PacketUnlimiter";
-	public static final String VERSION = "1.0";
+	public static final String VERSION = "1.1";
 
 	public static boolean isObfuscated = false;
-	public static boolean bigPacketWarning = false;
+	public static boolean bigPacketWarning = true;
+	public static boolean readNbtLimit = false;
+	public static boolean writeNbtLimit = false;
+	public static boolean unpackNbtLimit = false;
 
 	public CoreMod()
 	{
 		Configuration config = new Configuration(new File("config", NAME + ".cfg"));
 		config.load();
-		bigPacketWarning = config.getBoolean("bigPacketWarning", "general", true, "Включить оповещение при превышении стандартного лимита (2 MB)");
+		bigPacketWarning = config.getBoolean("bigPacketWarning", "general", bigPacketWarning, "Включить оповещение при превышении стандартного лимита (2 MB)");
+		readNbtLimit = config.getBoolean("readNbtLimit", "general", readNbtLimit, "Включить ограничение чтения NBT");
+		writeNbtLimit = config.getBoolean("writeNbtLimit", "general", writeNbtLimit, "Включить ограничение записи NBT");
+		unpackNbtLimit = config.getBoolean("unpackNbtLimit", "general", unpackNbtLimit, "Включить ограничение распаковки NBT");
 		config.save();
 	}
 

src/main/java/com/gamerforea/packetunlimiter/asm/ASMTransformer.java

@@ -10,6 +10,8 @@
 import org.objectweb.asm.tree.MethodNode;
 import org.objectweb.asm.tree.VarInsnNode;
 
+import com.gamerforea.packetunlimiter.CoreMod;
+
 import net.minecraft.launchwrapper.IClassTransformer;
 
 public final class ASMTransformer implements IClassTransformer, Opcodes
@@ -33,6 +35,9 @@ private static byte[] patchPacketBuffer(byte[] basicClass)
 		for (MethodNode mNode : cNode.methods)
 			if (mNode.name.equals(writeNbt) && mNode.desc.equals("(Lnet/minecraft/nbt/NBTTagCompound;)V"))
 			{
+				if (CoreMod.writeNbtLimit)
+					continue;
+
 				InsnList insn = new InsnList();
 				insn.add(new VarInsnNode(ALOAD, 0));
 				insn.add(new VarInsnNode(ALOAD, 1));
@@ -42,6 +47,9 @@ private static byte[] patchPacketBuffer(byte[] basicClass)
 			}
 			else if (mNode.name.equals(readNbt) && mNode.desc.equals("()Lnet/minecraft/nbt/NBTTagCompound;"))
 			{
+				if (CoreMod.readNbtLimit)
+					continue;
+
 				InsnList insn = new InsnList();
 				insn.add(new VarInsnNode(ALOAD, 0));
 				insn.add(new MethodInsnNode(INVOKESTATIC, "com/gamerforea/packetunlimiter/asm/PacketBufferPatch", "readNBTTagCompoundFromBuffer", "(Lnet/minecraft/network/PacketBuffer;)Lnet/minecraft/nbt/NBTTagCompound;", false));

src/main/java/com/gamerforea/packetunlimiter/asm/PacketBufferPatch.java

@@ -60,7 +60,7 @@ public static NBTTagCompound readNBTTagCompoundFromBuffer(PacketBuffer buf) thro
 				arr = ArrayUtils.addAll(arr, arr1);
 			}
 
-			return CompressedStreamTools.func_152457_a(arr, new NBTSizeTracker(Long.MAX_VALUE));
+			return CompressedStreamTools.func_152457_a(arr, new NBTSizeTracker(CoreMod.unpackNbtLimit ? 2097152 : Long.MAX_VALUE));
 		}
 	}
 }