1 files changed, 39 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/PeerWire/Bitfield.hs b/src/Network/BitTorrent/PeerWire/Bitfield.hs
new file mode 100644
index 00000000..c9357a25
--- /dev/null
+++ b/src/Network/BitTorrent/PeerWire/Bitfield.hs
@@ -0,0 +1,39 @@
+-- |
+--   Copyright   :  (c) Sam T. 2013
+--   License     :  MIT
+--   Maintainer  :  pxqr.sta@gmail.com
+--   Stability   :  experimental
+--   Portability :  portable
+--
+--
+--   This module provides Bitfield datatype used to represent sets of
+--   piece indexes any peer have. All associated operations should be
+--   defined here as well.
+--
+module Network.BitTorrent.PeerWire.Bitfield
+       ( Bitfield(..)
+       , getBitfield, putBitfield, bitfieldByteCount
+       ) where
+
+import Control.Applicative
+import           Data.ByteString (ByteString)
+import qualified Data.ByteString as B
+
+import Data.Serialize
+
+
+newtype Bitfield = MkBitfield {
+    bfBits :: ByteString
+  } deriving (Show, Eq, Ord)
+
+bitfieldByteCount :: Bitfield -> Int
+bitfieldByteCount = B.length . bfBits
+{-# INLINE bitfieldByteCount #-}
+
+getBitfield :: Int -> Get Bitfield
+getBitfield n = MkBitfield <$> getBytes n
+{-# INLINE getBitfield #-}
+
+putBitfield :: Bitfield -> Put
+putBitfield = putByteString . bfBits
+{-# INLINE putBitfield #-}
\ No newline at end of file