2 files changed, 13 insertions, 29 deletions

diff --git a/dht/src/Network/Tox/Onion/Transport.hs b/dht/src/Network/Tox/Onion/Transport.hs
index 913d339d..6319ed2f 100644
--- a/dht/src/Network/Tox/Onion/Transport.hs
+++ b/dht/src/Network/Tox/Onion/Transport.hs
@@ -58,55 +58,39 @@ import Network.Tox.DHT.Transport (SendNodes(..))
 import Network.Tox.NodeId
 import Network.QueryResponse
 
-{-
 encodeOnionAddr :: TransportCrypto
                    -> (NodeInfo -> RouteId -> IO (Maybe OnionRoute))
                    -> (OnionMessage Encrypted,OnionDestination RouteId)
                    -> IO (Maybe (ByteString, SockAddr))
--}
-encodeOnionAddr :: TransportCrypto
-                     -> (NodeInfo -> RouteId -> IO (Maybe OnionRoute))
-                     -> (OnionMessage Encrypted, OnionDestination RouteId)
-                     -> IO (Maybe
-                             (Either (TCP.RelayPacket, TCP.NodeInfo) (ByteString, SockAddr)))
 encodeOnionAddr crypto _ (msg,OnionToOwner ni p) =
-    return $ Just $ Right ( runPut $ putResponse (OnionResponse p msg)
-                          , nodeAddr ni )
+    return $ Just ( runPut $ putResponse (OnionResponse p msg)
+                  , nodeAddr ni )
 encodeOnionAddr crypto getRoute (msg,OnionDestination x ni Nothing) = do
     encodeOnionAddr crypto getRoute (msg,OnionDestination x ni (Just $ routeId $ nodeId ni) )
-    -- dput XMisc $ "ONION encode missing routeid"
-    -- return Nothing
 encodeOnionAddr crypto getRoute (msg,OnionDestination _ ni (Just rid)) = do
     let go route = do
-            mreq <- wrapForRoute crypto msg ni route
-            case mreq of
-                Right req -> return $ Right ( runPut $ putRequest req , nodeAddr $ routeNodeA route)
-                Left o | Just port <- routeRelayPort route
-                          -> return $ Left ( o, TCP.NodeInfo (routeNodeA route) port)
+            req <- wrapForRoute crypto msg ni route
+            return ( runPut $ putRequest req , nodeAddr $ routeNodeA route)
     m <- {-# SCC "encodeOnionAddr.getRoute" #-} getRoute ni rid
     x <- {-# SCC "encodeOnionAddr.wrapForRoute" #-} mapM go m
     return x
 
--- wrapForRoute :: TransportCrypto -> OnionMessage Encrypted -> NodeInfo -> OnionRoute -> IO (OnionRequest N0)
 wrapForRoute :: TransportCrypto
                       -> OnionMessage Encrypted
                       -> NodeInfo
                       -> OnionRoute
-                      -> IO (Either TCP.RelayPacket (OnionRequest N0))
-wrapForRoute crypto msg ni r@OnionRoute{routeRelayPort=Nothing} =
+                      -> IO (OnionRequest N0)
+wrapForRoute crypto msg ni r =
     wrapIndirectHops crypto msg ni r $ \nonce saddr msg' -> do
         fwd <- wrapOnion crypto (routeAliasA r)
                                 nonce
                                 (id2key . nodeId $ routeNodeA r)
                                 saddr
                                 msg'
-        return $ Right OnionRequest { onionNonce    = nonce
-                                    , onionForward  = fwd
-                                    , pathFromOwner = NoReturnPath
-                                    }
-wrapForRoute crypto msg ni r@OnionRoute { routeRelayPort = Just _ } =
-    wrapIndirectHops crypto msg ni r $ \nonce saddr fwd ->
-        return $ Left $ TCP.OnionPacket nonce $ Addressed (nodeAddr $ routeNodeB r) fwd
+        return $ OnionRequest { onionNonce    = nonce
+                              , onionForward  = fwd
+                              , pathFromOwner = NoReturnPath
+                              }
 
 wrapIndirectHops :: TransportCrypto
                       -> OnionMessage Encrypted
diff --git a/dht/src/Network/Tox/Transport.hs b/dht/src/Network/Tox/Transport.hs
index 12886245..ff99b747 100644
--- a/dht/src/Network/Tox/Transport.hs
+++ b/dht/src/Network/Tox/Transport.hs
@@ -38,14 +38,14 @@ toxTransport ::
           -> SockAddr -- ^ UDP bind-address
           -> UDPTransport
           -> Transport String ViaRelay B.ByteString
-          -> (TCP.NodeInfo -> RelayPacket -> IO ())   -- ^ TCP server-bound callback.
+          -> (TCP.NodeInfo -> RelayPacket -> IO ())   -- ^ (UNUSED) TCP server-bound callback.
           -> (Int -> OnionMessage Encrypted -> IO ()) -- ^ TCP client-bound callback.
           -> IO ( Transport String Multi.SessionAddress (CryptoPacket Encrypted)
                 , Transport String Multi.NodeInfo (DHTMessage Encrypted8)
                 , Transport String (OnionDestination RouteId) (OnionMessage Encrypted)
                 , Transport String AnnouncedRendezvous (PublicKey,OnionData)
                 , Transport String Multi.SessionAddress (Handshake Encrypted))
-toxTransport crypto orouter closeLookup addr udp relaynet tcp2server tcp2client = do
+toxTransport crypto orouter closeLookup addr udp relaynet _ tcp2client = do
     (netcrypto, udp0) <- partitionTransport parseCrypto encodeCrypto udp
     (dhtUDP,udp1) <- partitionTransportM (parseDHTAddr (pendingCookiesUDP crypto) nodeInfo)
                                       (fmap Just . encodeDHTAddr nodeAddr)
@@ -60,7 +60,7 @@ toxTransport crypto orouter closeLookup addr udp relaynet tcp2server tcp2client
         [ Multi.UDP :=> ByAddress dhtUDP
         , Multi.TCP :=> ByAddress dhtTCP
         ]
-    (onion1,udp2) <- partitionAndForkTransport tcp2server
+    (onion1,udp2) <- partitionTransportM
                              (parseOnionAddr $ lookupSender orouter)
                              (encodeOnionAddr crypto $ lookupRoute orouter)
                              udp1