1 files changed, 11 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Exchange.hs b/src/Network/BitTorrent/Exchange.hs
index 52b5f690..b2d639f6 100644
--- a/src/Network/BitTorrent/Exchange.hs
+++ b/src/Network/BitTorrent/Exchange.hs
@@ -60,6 +60,7 @@ module Network.BitTorrent.Exchange
        , yieldEvent
        , handleEvent
        , exchange
+       , p2p
 
          -- * Exceptions
        , disconnect
@@ -483,3 +484,13 @@ exchange storage = {-# SCC exchange #-} awaitEvent >>= handler
         if BF.null offer
           then return ()
           else handler (Available offer)
+
+yieldInit :: P2P ()
+yieldInit = yieldMessage . Bitfield =<< getClientBF
+
+p2p :: P2P ()
+p2p = do
+  yieldInit
+  storage <- asks (storage . swarmSession)
+  forever $ do
+    exchange storage
\ No newline at end of file