1 files changed, 20 insertions, 2 deletions

diff --git a/src/Network/Tox.hs b/src/Network/Tox.hs
index 3c3bce49..170a07e8 100644
--- a/src/Network/Tox.hs
+++ b/src/Network/Tox.hs
@@ -1,3 +1,4 @@
+{-# LANGUAGE ViewPatterns #-}
 {-# LANGUAGE CPP                        #-}
 {-# LANGUAGE DeriveDataTypeable         #-}
 {-# LANGUAGE DeriveFoldable             #-}
@@ -19,7 +20,11 @@ import Debug.Trace
 import Control.Exception hiding (Handler)
 import Control.Applicative
 import Control.Arrow
-import Control.Concurrent                     (MVar,killThread)
+#ifdef THREAD_DEBUG
+import Control.Concurrent.Lifted.Instrument
+#else
+import Control.Concurrent.Lifted
+#endif
 import Control.Concurrent.STM
 import Control.Monad
 import Control.Monad.Fix
@@ -113,7 +118,7 @@ import Data.HashMap.Strict (HashMap)
 import qualified Data.Map.Strict as Map
 import Control.Concurrent (threadDelay)
 import DPut
-
+import Network.Tox.Avahi
 
 newCrypto :: IO TransportCrypto
 newCrypto = do
@@ -476,6 +481,17 @@ newTox keydb addr mbSessionsState suppliedDHTKey = do
 onionTimeout :: Tox -> DHT.TransactionId -> Onion.OnionDestination RouteId -> STM (Onion.OnionDestination RouteId, Int)
 onionTimeout Tox { toxOnionRoutes = or } (DHT.TransactionId n8 _) od = lookupTimeout or n8 od
 
+routing4nodeInfo :: DHT.Routing -> IO NodeInfo
+routing4nodeInfo (DHT.routing4 -> tv) = R.thisNode <$> readTVarIO tv
+
+dnssdAnnounce :: Tox -> IO ()
+dnssdAnnounce (toxRouting -> r) = do
+    ni <- routing4nodeInfo r
+    announceToxService (nodePort ni) (nodeId ni)
+
+dnssdDiscover :: Tox -> NodeInfo -> IO ()
+dnssdDiscover (toxDHT -> client) ni = void $ DHT.ping client ni
+
 forkTox :: Tox -> IO (IO (), [NodeInfo] -> [NodeInfo] -> IO (), [NodeInfo] -> [NodeInfo] -> IO ())
 forkTox tox = do
     _ <- forkListener "toxHandshakes" (toxHandshakes tox)
@@ -485,6 +501,8 @@ forkTox tox = do
     quit <- forkListener "toxCrypto" (toxCrypto tox)
     forkPollForRefresh (DHT.refresher4 $ toxRouting tox)
     forkPollForRefresh (DHT.refresher6 $ toxRouting tox)
+    dnssdIn <- forkIO $ queryToxService (dnssdDiscover tox)
+    dnssdOut <- forkIO $ dnssdAnnounce tox
     keygc <- Onion.forkAnnouncedKeysGC (toxAnnouncedKeys tox)
     return ( killThread keygc >> quit
            , bootstrap (DHT.refresher4 $ toxRouting tox)