summaryrefslogtreecommitdiff
path: root/src/Network/BitTorrent/Exchange.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/Network/BitTorrent/Exchange.hs')
-rw-r--r--src/Network/BitTorrent/Exchange.hs57
1 files changed, 50 insertions, 7 deletions
diff --git a/src/Network/BitTorrent/Exchange.hs b/src/Network/BitTorrent/Exchange.hs
index d75fdf96..425ed2a3 100644
--- a/src/Network/BitTorrent/Exchange.hs
+++ b/src/Network/BitTorrent/Exchange.hs
@@ -6,18 +6,61 @@
6-- Portability : portable 6-- Portability : portable
7-- 7--
8{-# LANGUAGE DoAndIfThenElse #-} 8{-# LANGUAGE DoAndIfThenElse #-}
9module Network.BitTorrent.Exchange (module PW) where 9module Network.BitTorrent.Exchange
10 (
11 -- * Session
12 PeerSession, newLeacher, newSeeder
13 ) where
14
15import Control.Applicative
16import Control.Concurrent
17import Control.Concurrent.STM
18import Data.IORef
19import Data.Function
20import Data.Ord
21import Data.Set as S
22
23import Data.Conduit
24import Data.Conduit.Cereal
25import Data.Conduit.Network
26import Data.Serialize
10 27
11import Network.BitTorrent.Exchange.Selection as PW 28import Network.BitTorrent.Exchange.Selection as PW
12import Network.BitTorrent.Exchange.Protocol as PW 29import Network.BitTorrent.Exchange.Protocol as PW
13 30
31import Network.BitTorrent.Internal
32import Network.BitTorrent.Extension
33import Network.BitTorrent.Peer
34import Data.Bitfield as BF
35import Data.Torrent
36
37{-----------------------------------------------------------------------
38 P2P monad
39-----------------------------------------------------------------------}
40
14{- 41{-
42type P2P = Reader PeerSession (ConduitM Message Message IO)
15 43
16newtype P2P a = P2P { 44conduit :: Socket -> P2P a -> IO a
17 getP2P :: ReaderT PSession State PState (Conduit Message IO Message) a 45conduit sock p2p =
18 } 46 sourceSocket sock $=
47 conduitGet get $=
48 messageLoop p2p $=
49 conduitPut put $$
50 sinkSocket sock
19 51
20runP2P :: PConnection -> P2P a -> IO a 52messageLoop :: P2P () -> P2P ()
21recvMessage :: P2P Message 53messageLoop = undefined
22sendMessage :: Message -> P2P () 54
55runP2P :: SSession -> PeerAddr -> P2P a -> IO a
56runP2P se addr p2p = withPeer se addr $ conduit messageLoop
57
58data Event = Available
59 | Want
60 | Block
61
62{-
63waitForEvent :: P2P Event
64signalEvent :: Event -> P2P ()
65-}
23-} \ No newline at end of file 66-} \ No newline at end of file