import Control.Concurrent.MVar (readMVar)
import Control.Monad.Reader (ReaderT, runReaderT, ask, liftIO)
import Data.Binary (Binary(..), encode, decode)
-import Data.Binary.Get (Get, isEmpty, getWord32be, getByteString)
+import Data.Binary.Get (Get, isEmpty, getWord32be, getWord64be, getByteString)
import Data.Binary.Put (putWord16be, putWord64be, putWord32be, putByteString)
import Data.ByteString (ByteString)
import qualified Data.ByteString.Char8 as BC
putWord64be (fromIntegral down)
putWord64be (fromIntegral left)
putWord64be (fromIntegral up)
- putWord32be $ fromIntegral (eventToInteger None)
+ putWord32be $ fromIntegral (eventToInteger event)
putWord32be 0
- -- key is optional, we will not send it for now
- putWord32be $ fromIntegral (-1)
+ putWord32be 0
+ putWord32be 10
putWord16be $ fromIntegral port
put (ScrapeReq _ _ _) = undefined
get = undefined
get = do
a <- getWord32be -- action
case a of
- 0 -> liftA2 ConnectResp (fromIntegral <$> getWord32be) (fromIntegral <$> getWord32be)
+ 0 -> liftA2 ConnectResp (fromIntegral <$> getWord32be) (fromIntegral <$> getWord64be)
1 -> do
tid <- fromIntegral <$> getWord32be
interval' <- fromIntegral <$> getWord32be
connectResponse tid = do
h <- ask
resp <- liftIO $ recvResponse h
- liftIO $ print resp
-- check if nbytes is at least 16 bytes long
case resp of
(ConnectResp tidr cid) ->
announceRequest cid infohash peerId up down left port = do
h <- ask
tidi <- liftIO randomIO
- -- connId transId infohash peerId down left up event port)
let pkt = encode $ AnnounceReq cid tidi infohash peerId down left up None port
liftIO $ sendRequest h (toStrict pkt)
return tidi
if empty
then return []
else do
- ip <- toIP <$> getByteString 6
+ ip <- toIP <$> getByteString 4
port <- toPort <$> getByteString 2
ipportpairs <- getIPPortPairs
return $ (ip, port) : ipportpairs
addrinfos <- getAddrInfo Nothing (Just host) (Just (show port))
let (SockAddrInet p ip) = addrAddress $ head addrinfos
putStrLn "connected to tracker"
- return $ UDPTrackerHandle { sock = s
- , addr = (SockAddrInet (fromIntegral port) ip) }
+ return UDPTrackerHandle { sock = s
+ , addr = (SockAddrInet (fromIntegral port) ip) }
closeSession :: UDPTrackerHandle -> IO ()
closeSession (UDPTrackerHandle s _ _) = close s
flip runReaderT handle $ do
t1 <- connectRequest
cid <- connectResponse t1
- liftIO $ print cid
t2 <- announceRequest cid infohash peerId (fromIntegral up) (fromIntegral down) (fromIntegral (left tstate)) (fromIntegral sport)
stats <- announceResponse t2
liftIO $ print stats
+-- _ <- threadDelay $