1 files changed, 9 insertions, 8 deletions

diff --git a/dht/src/Network/Tox/Session.hs b/dht/src/Network/Tox/Session.hs
index cbb516a7..457ed1a0 100644
--- a/dht/src/Network/Tox/Session.hs
+++ b/dht/src/Network/Tox/Session.hs
@@ -136,14 +136,15 @@ plainHandshakeH sp saddr0 skey handshake = do
                       (sessionKey hd)
                       <$> atomically (newTVar $ baseNonce hd)
                       <*> atomically (newTVar $ baseNonce hd_sent)
-    m <- newSession (spSessions sp) (\() p -> return p) (decryptPacket sk) saddr
+    let addr_lbl = Multi.showSessionAddr saddr
+    m <- newSession (spSessions sp) (\() p -> return p) (\_ -> decryptPacket sk addr_lbl) saddr
     dput XNetCrypto $ prelude ++ "plainHandshakeH: session " ++ maybe "Nothing" (const "Just") m
     forM_ m $ \(sid, t) -> do
         (t2,resend,getMissing)
             <- lossless (take 8 (showKey256 them) ++ "." ++ Multi.showSessionAddr saddr)
                         (\cp a -> return $ fmap (,a) $ checkLossless $ runIdentity $ pktData cp)
                         (\seqno p@(Pkt m :=> _) _ -> do
-                            y <- encryptPacket sk $ bookKeeping seqno p
+                            y <- encryptPacket sk addr_lbl $ bookKeeping seqno p
                             return OutgoingInfo
                                 { oIsLossy         = lossyness m == Lossy
                                 , oEncoded         = y
@@ -182,8 +183,8 @@ data SessionKeys = SessionKeys
     }
 
 -- | Decrypt an inbound session packet and update the nonce for the next one.
-decryptPacket :: SessionKeys -> addr -> CryptoPacket Encrypted -> IO (Maybe (CryptoPacket Identity, ()))
-decryptPacket sk saddr (CryptoPacket n16 ciphered) = do
+decryptPacket :: SessionKeys -> String -> CryptoPacket Encrypted -> IO (Maybe (CryptoPacket Identity, ()))
+decryptPacket sk lbl (CryptoPacket n16 ciphered) = do
     (n24,δ) <- atomically $ do
         n <- readTVar (skNonceIncoming sk)
         let δ = n16 - nonce24ToWord16 n
@@ -198,13 +199,13 @@ decryptPacket sk saddr (CryptoPacket n16 ciphered) = do
             do let them = key2id $ skThem sk
                    CryptoData ack seqno _ = x
                    cm = decodeRawCryptoMsg x
-               dput XNetCrypto $ unwords [take 8 (show them),"-->",show (msgID cm),show (n24,ack,seqno)]
+               dput XNetCrypto $ unwords [lbl,"-->",show (msgID cm),show (n24,ack,seqno)]
 
             return $ Just ( CryptoPacket n16 (pure x), () )
 
 -- | Encrypt an outbound session packet and update the nonce for the next one.
-encryptPacket :: SessionKeys -> CryptoData -> IO (CryptoPacket Encrypted)
-encryptPacket sk plain = do
+encryptPacket :: SessionKeys -> String -> CryptoData -> IO (CryptoPacket Encrypted)
+encryptPacket sk lbl plain = do
     n24 <- atomically $ do
         n24 <- readTVar (skNonceOutgoing sk)
         modifyTVar' (skNonceOutgoing sk) incrementNonce24
@@ -214,7 +215,7 @@ encryptPacket sk plain = do
 
     do let them = key2id $ skThem sk
            CryptoData ack seqno cm = plain
-       dput XNetCrypto $ unwords [take 8 (show them),"<--",show (msgID cm),show (n24,ack,seqno)]
+       dput XNetCrypto $ unwords [lbl,"<--",show (msgID cm),show (n24,ack,seqno)]
 
     return $ CryptoPacket (nonce24ToWord16 n24) ciphered