Decryption layer for data-to-route messages.

1 files changed, 22 insertions, 11 deletions

diff --git a/src/Network/Tox/Onion/Transport.hs b/src/Network/Tox/Onion/Transport.hs
index bdaef651..4c3de3e6 100644
--- a/src/Network/Tox/Onion/Transport.hs
+++ b/src/Network/Tox/Onion/Transport.hs
@@ -738,16 +738,27 @@ instance Show Rendezvous where
 parseDataToRoute
     :: TransportCrypto
          -> (OnionMessage Encrypted,OnionDestination r)
-         -> Either (Asymm (Encrypted DataToRoute),Rendezvous) (OnionMessage Encrypted, OnionDestination r)
-parseDataToRoute crypto (OnionToRouteResponse dta, od)
-                            = Left ( dta
-                                   , Rendezvous (rendezvousAliasPublic crypto) $ onionNodeInfo od )
-parseDataToRoute _      msg = Right msg
+         -> IO (Either (DataToRoute,Rendezvous) (OnionMessage Encrypted, OnionDestination r))
+parseDataToRoute crypto (OnionToRouteResponse dta, od) =
+    return $ either (const $ Right (OnionToRouteResponse dta,od)) Left $ do
+        -- XXX: Do something with decryption failure?
+        decrypted <- uncomposed
+             $ decryptMessage (rendezvousSecret crypto,rendezvousPublic crypto)
+                              (asymmNonce dta)
+                              (Right dta)
+        return ( runIdentity decrypted
+               , Rendezvous (rendezvousPublic crypto) $ onionNodeInfo od )
+parseDataToRoute _ msg = return $ Right msg
 
 encodeDataToRoute :: TransportCrypto
-                     -> (Asymm (Encrypted DataToRoute),Rendezvous)
-                     -> Maybe (OnionMessage Encrypted,OnionDestination r)
-encodeDataToRoute crypto (dta, Rendezvous pub ni)
-    = Just ( OnionToRoute pub -- Public key of destination node
-                          dta
-           , OnionDestination SearchingAlias ni Nothing )
+                     -> (DataToRoute,Rendezvous)
+                     -> IO (Maybe (OnionMessage Encrypted,OnionDestination r))
+encodeDataToRoute crypto (plain, Rendezvous pub ni) = do
+    nonce <- atomically $ transportNewNonce crypto
+    let dta = encryptMessage (onionAliasSecret crypto) pub nonce plain
+    return $ Just ( OnionToRoute pub -- Public key of destination node
+                                 Asymm { senderKey  = onionAliasPublic crypto
+                                       , asymmNonce = nonce
+                                       , asymmData  = dta
+                                       }
+                  , OnionDestination SearchingAlias ni Nothing )