+ Add some benchs.

2 files changed, 7 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/PeerWire/Message.hs b/src/Network/BitTorrent/PeerWire/Message.hs
index 188f0572..cc771966 100644
--- a/src/Network/BitTorrent/PeerWire/Message.hs
+++ b/src/Network/BitTorrent/PeerWire/Message.hs
@@ -12,6 +12,7 @@ import Data.Serialize
 import Network.BitTorrent.PeerWire.Block
 import Network.BitTorrent.PeerWire.Bitfield
 
+import Data.Array
 
 -- | Messages used in communication between peers.
 --
@@ -78,7 +79,7 @@ data Message = KeepAlive
 instance Serialize Message where
   get = do
     len <- getInt
-    _   <- lookAhead $ ensure len
+--    _   <- lookAhead $ ensure len
     if len == 0 then return KeepAlive
       else do
         mid <- getWord8
diff --git a/src/Network/BitTorrent/PeerWire/Selection.hs b/src/Network/BitTorrent/PeerWire/Selection.hs
index 04049812..2e412e06 100644
--- a/src/Network/BitTorrent/PeerWire/Selection.hs
+++ b/src/Network/BitTorrent/PeerWire/Selection.hs
@@ -23,7 +23,8 @@
 --
 module Network.BitTorrent.PeerWire.Selection
        ( Selector
-       , strictFirst, rarestFirst, randomFirst, endGame, autoSelector
+       , strictFirst, strictLast
+       , rarestFirst, randomFirst, endGame, autoSelector
        ) where
 
 import Network.BitTorrent.PeerWire.Block
@@ -41,6 +42,9 @@ type Selector =  Bitfield      -- ^ Indices of client "have" pieces.
 strictFirst :: Selector
 strictFirst h a _ = findMin (difference a h)
 
+-- | Select the last available piece.
+strictLast :: Selector
+strictLast h a _ = findMax (difference a h)
 
 -- |
 rarestFirst :: Selector