Use the server port number in the tracker request

[functorrent.git] / src / main / Main.hs

diff --git a/src/main/Main.hs b/src/main/Main.hs
index 997801bf5aa9c6acfd7278075a204b65376e878e..0a514104703485beda12b9206226fe44acbb6bc5 100644 (file)
--- a/src/main/Main.hs
+++ b/src/main/Main.hs
@@ -5,6 +5,7 @@ import Prelude hiding (log, length, readFile, getContents)
 
 import Control.Concurrent (forkIO)
 import Data.ByteString.Char8 (ByteString, getContents, readFile, unpack)
+import Network (PortID (PortNumber))
 import System.Environment (getArgs)
 import System.Exit (exitSuccess)
 import System.Directory (doesFileExist)
@@ -60,11 +61,12 @@ main = do
        log $ "Downloading file : " ++ name (info m)
 
        log $ "starting server"
-       (serverSock, portnum) <- Server.start
+       (serverSock, (PortNumber portnum)) <- Server.start
+       log $ "server started on " ++ show portnum
        log "Trying to fetch peers"
        forkIO $ Server.run serverSock peerId m
        log $ "Trackers: " ++ head (announceList m)
-       trackerResp <- getTrackerResponse peerId m
+       trackerResp <- getTrackerResponse portnum peerId m
        case  trackerResp of
         Left e -> log $ "Error" ++ unpack e
         Right peerList -> do