~ Fix message length bug in handshake.

1 files changed, 4 insertions, 1 deletions

diff --git a/src/Network/BitTorrent/PeerWire/Message.hs b/src/Network/BitTorrent/PeerWire/Message.hs
index 39102eed..1c520221 100644
--- a/src/Network/BitTorrent/PeerWire/Message.hs
+++ b/src/Network/BitTorrent/PeerWire/Message.hs
@@ -99,7 +99,10 @@ instance Serialize Message where
           0x0D -> SuggestPiece  <$> getInt
           0x10 -> RejectRequest <$> get
           0x11 -> AllowedFast   <$> getInt
-          _    -> fail $ "unknown message ID: " ++ show mid
+          _    -> do
+            rm <- remaining >>= getBytes
+            fail $ "unknown message ID: " ++ show mid ++ "\n"
+                ++ "remaining available bytes: " ++ show rm
 
     where
       getBlock :: Int -> Get Block