1 files changed, 10 insertions, 2 deletions

diff --git a/dht/examples/toxrelay.hs b/dht/examples/toxrelay.hs
index af08e8d7..d6b0da17 100644
--- a/dht/examples/toxrelay.hs
+++ b/dht/examples/toxrelay.hs
@@ -1,15 +1,23 @@
-import Network.Address (getBindAddress)
+import Network.Address (getBindAddress,sockAddrPort)
 import Network.SocketLike
 import Network.StreamServer
 import Network.Tox.Onion.Transport hiding (encrypt,decrypt)
 import Network.Tox.Relay
+import Crypto.Tox
+
+import DPut
+import DebugTag
+
+socketPort s = sockAddrPort <$> getSocketName s
 
 main :: IO ()
 main = do
     udp_addr <- getBindAddress "33445" True
     let sendOnion :: SockAddr -> OnionRequest N1 -> IO ()
         sendOnion _ _ = return ()
-    (h,sendTCP) <- tcpRelay udp_addr sendOnion
+    setVerbose XNetCrypto
+    crypto <- newCrypto
+    (h,sendTCP) <- tcpRelay crypto udp_addr sendOnion
     boundPort <- socketPort $ listenSocket h
     putStrLn $ "Listening on port: " ++ show boundPort