Fix reading and writing of signed compressed integers

jbevain · Jun 17, 2011 · 481ca4f · 481ca4f
1 parent 2e6f7c2
commit 481ca4f
Showing 1 changed file with 23 additions and 6 deletions.
diff --git a/Mono.Cecil.PE/ByteBuffer.cs b/Mono.Cecil.PE/ByteBuffer.cs
@@ -140,11 +140,16 @@ public uint ReadCompressedUInt32 ()
 
 		public int ReadCompressedInt32 ()
 		{
-			var value = (int) ReadCompressedUInt32 ();
-
-			return (value & 1) != 0
-				? -(value >> 1)
-				: value >> 1;
+			var value = (int) (ReadCompressedUInt32 () >> 1);
+			if ((value & 1) == 0)
+				return value;
+			if (value < 0x40)
+				return value - 0x40;
+			if (value < 0x2000)
+				return value - 0x2000;
+			if (value < 0x10000000)
+				return value - 0x10000000;
+			return value - 0x20000000;
 		}
 
 		public float ReadSingle ()
@@ -267,7 +272,19 @@ public void WriteCompressedUInt32 (uint value)
 
 		public void WriteCompressedInt32 (int value)
 		{
-			WriteCompressedUInt32 ((uint) ((value < 0) ? ((-value) << 1) | 1 : value << 1));
+			if (value >= 0) {
+				WriteCompressedUInt32 ((uint) (value << 1));
+				return;
+			}
+
+			if (value > -0x40)
+				value = 0x40 + value;
+			else if (value >= -0x2000)
+				value = 0x2000 + value;
+			else if (value >= -0x20000000)
+				value = 0x20000000 + value;
+
+			WriteCompressedUInt32 ((uint) ((value << 1) | 1));
 		}
 
 		public void WriteBytes (byte [] bytes)