From 84003ba9db0cdc88c8e30a66c66c3b6ea3455893 Mon Sep 17 00:00:00 2001
From: lidgren <lidgren@users.noreply.github.com>
Date: Wed, 29 Aug 2012 11:47:46 +0000
Subject: [PATCH] Fix for WriteByte outputting too many bits, unit test added

---
 Lidgren.Network/NetBitWriter.cs | 2 +-
 UnitTests/Program.cs            | 2 ++
 UnitTests/UnitTests.csproj      | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/Lidgren.Network/NetBitWriter.cs b/Lidgren.Network/NetBitWriter.cs
index e07bfc7..2d3ecb7 100644
--- a/Lidgren.Network/NetBitWriter.cs
+++ b/Lidgren.Network/NetBitWriter.cs
@@ -105,7 +105,7 @@ namespace Lidgren.Network
 			NetException.Assert(((numberOfBits >= 1) && (numberOfBits <= 8)), "Must write between 1 and 8 bits!");
 
 			// mask out unwanted bits in the source
-			byte isrc = (byte)((uint)source & ((~(uint)0) >> (8 - numberOfBits)));
+			byte isrc = (byte)(source & (0x000000FF >> (8 - numberOfBits)));
 
 			int bytePtr = destBitOffset >> 3;
 
diff --git a/UnitTests/Program.cs b/UnitTests/Program.cs
index 47a1936..1f7bf74 100644
--- a/UnitTests/Program.cs
+++ b/UnitTests/Program.cs
@@ -17,6 +17,8 @@ namespace UnitTests
 
 			Console.WriteLine("Unique identifier is " + NetUtility.ToHexString(peer.UniqueIdentifier));
 
+			NetBitWriterTests.Run();
+
 			ReadWriteTests.Run(peer);
 
 			NetQueueTests.Run();
diff --git a/UnitTests/UnitTests.csproj b/UnitTests/UnitTests.csproj
index 4665117..b1bab11 100644
--- a/UnitTests/UnitTests.csproj
+++ b/UnitTests/UnitTests.csproj
@@ -41,6 +41,7 @@
     <Compile Include="BitVectorTests.cs" />
     <Compile Include="EncryptionTests.cs" />
     <Compile Include="MiscTests.cs" />
+    <Compile Include="NetBitWriterTests.cs" />
     <Compile Include="NetQueueTests.cs" />
     <Compile Include="Program.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />