4 files changed, 17 insertions, 14 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index b3ffbe49..46d04999 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -127,10 +127,9 @@ executable example
                      , bittorrent
                      , mtl
 
-  ghc-options:         -O2 -rtsopts -threaded
--- -threaded -rtsopts
--- -eventlog
---  ghc-prof-options:    -prof -auto-all -caf-all
+  ghc-options:
+-- -threaded -eventlog
+  ghc-prof-options:   -rtsopts -auto-all -caf-all
 
   if !flag(testing)
      buildable:        False
diff --git a/examples/Main.hs b/examples/Main.hs
index 8d976aed..18cbefe3 100644
--- a/examples/Main.hs
+++ b/examples/Main.hs
@@ -13,7 +13,7 @@ main = do
 
   print (contentLayout "./" (tInfo torrent))
 
-  client  <- newClient 100 []
+  client  <- newClient 10 []
   swarm   <- newLeecher  client torrent
 
   storage <- swarm `bindTo`  "/tmp/"
@@ -22,5 +22,7 @@ main = do
 
   discover swarm $ do
     liftIO $ print "connected to peer"
-    forever $ exchange storage
-    liftIO $ print "disconnect to peer"
\ No newline at end of file
+    forever $ do
+      liftIO (getCurrentProgress client >>= print)
+      exchange storage
+    liftIO $ print "disconnected"
\ No newline at end of file
diff --git a/src/Network/BitTorrent.hs b/src/Network/BitTorrent.hs
index 30735023..b6e2eadf 100644
--- a/src/Network/BitTorrent.hs
+++ b/src/Network/BitTorrent.hs
@@ -154,4 +154,4 @@ exchange storage = awaitEvent >>= handler
         offer <- peerOffer
         if BF.null offer
           then return ()
-          else handler (Available offer)
\ No newline at end of file
+          else handler (Available offer)
diff --git a/src/Network/BitTorrent/Internal.hs b/src/Network/BitTorrent/Internal.hs
index bf47b87b..eaeb04e8 100644
--- a/src/Network/BitTorrent/Internal.hs
+++ b/src/Network/BitTorrent/Internal.hs
@@ -153,14 +153,14 @@ uploadedProgress (fromIntegral -> amount) = uploaded +~ amount
 {-# INLINE uploadedProgress #-}
 
 -- | Used when leecher join client session.
-enqueuedProgress :: Int -> Progress -> Progress
-enqueuedProgress (fromIntegral -> amount) = left +~ amount
+enqueuedProgress :: Integer -> Progress -> Progress
+enqueuedProgress amount = left +~ amount
 {-# INLINE enqueuedProgress #-}
 
 -- | Used when leecher leave client session.
 --   (e.g. user deletes not completed torrent)
-dequeuedProgress :: Int -> Progress -> Progress
-dequeuedProgress (fromIntegral -> amount) = left -~ amount
+dequeuedProgress :: Integer -> Progress -> Progress
+dequeuedProgress amount = left -~ amount
 {-# INLINE dequeuedProgress #-}
 
 {-----------------------------------------------------------------------
@@ -349,8 +349,10 @@ newSeeder cs t @ Torrent {..}
 
 -- | New swarm in which the client allowed both download and upload.
 newLeecher :: ClientSession -> Torrent -> IO SwarmSession
-newLeecher cs t @ Torrent {..}
-  = newSwarmSession defLeacherConns (haveNone (pieceCount tInfo)) cs t
+newLeecher cs t @ Torrent {..} = do
+  se <- newSwarmSession defLeacherConns (haveNone (pieceCount tInfo)) cs t
+  atomically $ modifyTVar' (currentProgress cs) (enqueuedProgress (contentLength tInfo))
+  return se
 
 --isLeacher :: SwarmSession -> IO Bool
 --isLeacher = undefined