From 4778126b1e7adab5dc1867fbc448df699bd9dae6 Mon Sep 17 00:00:00 2001
From: Sam Truzjan <pxqr.sta@gmail.com>
Date: Tue, 4 Feb 2014 01:27:55 +0400
Subject: Refactor UDP tracker host addr resolution

---
 src/Network/BitTorrent/Tracker/RPC/UDP.hs | 41 ++++++++++++++++++++-----------
 1 file changed, 26 insertions(+), 15 deletions(-)

(limited to 'src/Network/BitTorrent/Tracker/RPC')

diff --git a/src/Network/BitTorrent/Tracker/RPC/UDP.hs b/src/Network/BitTorrent/Tracker/RPC/UDP.hs
index a3927c2c..d7b359ed 100644
--- a/src/Network/BitTorrent/Tracker/RPC/UDP.hs
+++ b/src/Network/BitTorrent/Tracker/RPC/UDP.hs
@@ -94,6 +94,28 @@ closeManager Manager {..} = close sock
 withManager :: Options -> (Manager -> IO a) -> IO a
 withManager opts = bracket (newManager opts) closeManager
 
+{-----------------------------------------------------------------------
+--  Host Addr resolution
+-----------------------------------------------------------------------}
+
+setPort :: PortNumber -> SockAddr -> SockAddr
+setPort p (SockAddrInet  _ h)     = SockAddrInet  p h
+setPort p (SockAddrInet6 _ f h s) = SockAddrInet6 p f h s
+setPort _  addr = addr
+
+resolveURI :: URI -> IO SockAddr
+resolveURI URI { uriAuthority = Just (URIAuth {..}) } = do
+  infos <- getAddrInfo Nothing (Just uriRegName) Nothing
+  let port = fromMaybe 0 (readMaybe (L.drop 1 uriPort) :: Maybe Int)
+  case infos of
+    AddrInfo {..} : _ -> return $ setPort (fromIntegral port) addrAddress
+    _                 -> fail "getTrackerAddr: unable to lookup host addr"
+resolveURI _       = fail "getTrackerAddr: hostname unknown"
+
+-- TODO caching?
+getTrackerAddr :: Manager -> URI -> IO SockAddr
+getTrackerAddr _ = resolveURI
+
 {-----------------------------------------------------------------------
   Tokens
 -----------------------------------------------------------------------}
@@ -268,20 +290,6 @@ isExpired Connection {..} = do
 maxPacketSize :: Int
 maxPacketSize = 98 -- announce request packet
 
-setPort :: PortNumber -> SockAddr -> SockAddr
-setPort p (SockAddrInet  _ h)     = SockAddrInet  p h
-setPort p (SockAddrInet6 _ f h s) = SockAddrInet6 p f h s
-setPort _  addr = addr
-
-getTrackerAddr :: URI -> IO SockAddr
-getTrackerAddr URI { uriAuthority = Just (URIAuth {..}) } = do
-  infos <- getAddrInfo Nothing (Just uriRegName) Nothing
-  let port = fromMaybe 0 (readMaybe (L.drop 1 uriPort) :: Maybe Int)
-  case infos of
-    AddrInfo {..} : _ -> return $ setPort (fromIntegral port) addrAddress
-    _                 -> fail "getTrackerAddr: unable to lookup host addr"
-getTrackerAddr _       = fail "getTrackerAddr: hostname unknown"
-
 call :: Manager -> SockAddr -> ByteString -> IO ByteString
 call Manager {..} addr arg = do
   BS.sendAllTo sock arg addr
@@ -315,7 +323,7 @@ transaction m tracker @ UDPTracker {..} request = do
   tid <- genTransactionId
   let trans = TransactionQ cid tid request
 
-  addr <- getTrackerAddr trackerURI
+  addr <- getTrackerAddr m trackerURI
   res  <- call m addr (encode trans)
   case decode res of
     Right (TransactionR {..})
@@ -346,6 +354,9 @@ freshConnection m tracker @ UDPTracker {..} = do
     connId <- connectUDP m tracker
     updateConnection connId tracker
 
+getConnection :: Manager -> URI -> IO Connection
+getConnection _ = undefined
+
 announce :: Manager -> AnnounceQuery -> UDPTracker -> IO AnnounceInfo
 announce m ann tracker = do
   freshConnection m tracker
-- 
cgit v1.2.3