import qualified Data.ByteString.Char8 as BC
import Data.ByteString.Lazy (fromStrict, toStrict)
import Data.Word (Word16, Word32, Word64)
-import Network.Socket (Socket, Family( AF_INET ), SocketType( Datagram ), defaultProtocol, SockAddr(..), socket, inet_addr, close, getAddrInfo, addrAddress, SockAddr(..))
+import Network.Socket (Socket, Family( AF_INET ), SocketType( Datagram ), defaultProtocol, SockAddr(..), socket, close, getAddrInfo, addrAddress, SockAddr(..))
import Network.Socket.ByteString (sendTo, recvFrom)
import System.Random (randomIO)
-import FuncTorrent.Tracker.Types (TrackerEventState(..))
-import FuncTorrent.Utils (IP, Port, toIP, toPort)
+import FuncTorrent.Tracker.Types (TrackerEventState(..), TState(..))
+import FuncTorrent.Utils (IP, Port, toIP, toPort, getHostname, getPort)
import qualified FuncTorrent.FileSystem as FS (MsgChannel, Stats(..), getStats)
-import FuncTorrent.Tracker.Types(TState(..))
-- UDP tracker: http://bittorrent.org/beps/bep_0015.html
data Action = Connect
putWord64be (fromIntegral up)
putWord32be $ fromIntegral (eventToInteger None)
putWord32be 0
- -- key is optional, we will not send it for now
+ putWord32be 0
putWord32be $ fromIntegral (-1)
putWord16be $ fromIntegral port
put (ScrapeReq _ _ _) = undefined
sendRequest :: UDPTrackerHandle -> ByteString -> IO ()
sendRequest h req = do
n <- sendTo (sock h) req (addr h)
+ print $ BC.length req
-- sanity check with n?
return ()
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) ->
return 0
_ -> return 0
-announceRequest :: Word64 -> ByteString -> Word64 -> Word64 -> Word64 -> Word16 -> ReaderT UDPTrackerHandle IO Word32
-announceRequest cid infohash up down left port = do
+announceRequest :: Word64 -> ByteString -> String -> Word64 -> Word64 -> Word64 -> Word16 -> ReaderT UDPTrackerHandle IO Word32
+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 "foo" down left up None port
+ let pkt = encode $ AnnounceReq cid tidi infohash peerId down left up None port
liftIO $ sendRequest h (toStrict pkt)
return tidi
s <- socket AF_INET Datagram defaultProtocol
addrinfos <- getAddrInfo Nothing (Just host) (Just (show port))
let (SockAddrInet p ip) = addrAddress $ head addrinfos
- hostAddr <- inet_addr (show ip)
putStrLn "connected to tracker"
- return $ UDPTrackerHandle { sock = s
- , addr = (SockAddrInet (fromIntegral port) hostAddr) }
+ return UDPTrackerHandle { sock = s
+ , addr = (SockAddrInet (fromIntegral port) ip) }
closeSession :: UDPTrackerHandle -> IO ()
closeSession (UDPTrackerHandle s _ _) = close s
trackerLoop :: String -> Port -> String -> ByteString -> FS.MsgChannel -> TState -> IO ()
-trackerLoop host port peerId infohash fschan tstate = do
+trackerLoop url sport peerId infohash fschan tstate = do
st' <- FS.getStats fschan
st <- readMVar st'
let up = FS.bytesRead st
down = FS.bytesWritten st
- handle <- startSession host 2710
+ port = getPort url
+ host = getHostname url
+ putStrLn $ "host = " ++ (show host) ++ " port= " ++ (show port)
+ handle <- startSession host port
flip runReaderT handle $ do
t1 <- connectRequest
cid <- connectResponse t1
- t2 <- announceRequest cid infohash (fromIntegral up) (fromIntegral down) (fromIntegral (left tstate)) (fromIntegral port)
+ liftIO $ print "connect response"
+ liftIO $ print cid
+ t2 <- announceRequest cid infohash peerId (fromIntegral up) (fromIntegral down) (fromIntegral (left tstate)) (fromIntegral sport)
+ liftIO $ print "announce request"
+ liftIO $ print t2
stats <- announceResponse t2
liftIO $ print stats