Tox: handle cookie-request message.

1 files changed, 12 insertions, 2 deletions

diff --git a/src/Network/Tox/DHT/Transport.hs b/src/Network/Tox/DHT/Transport.hs
index bebb8ae7..5ebe8b15 100644
--- a/src/Network/Tox/DHT/Transport.hs
+++ b/src/Network/Tox/DHT/Transport.hs
@@ -20,8 +20,9 @@ module Network.Tox.DHT.Transport
     , FriendRequest(..)
     , NoSpam(..)
     , verifyChecksum
-    , CookieRequest
-    , Cookie
+    , CookieRequest(..)
+    , Cookie(..)
+    , CookieData(..)
     , DHTRequest
     , mapMessage
     , encrypt
@@ -378,6 +379,15 @@ data CookieData = CookieData   --   16 (mac)
 instance Sized CookieData where
     size = ConstSize 72
 
+instance Serialize CookieData where
+    get = CookieData <$> get
+                     <*> (id2key <$> get)
+                     <*> (id2key <$> get)
+    put (CookieData tm userkey dhtkey) = do
+        put tm
+        put (key2id userkey)
+        put (key2id dhtkey)
+
 instance Sized CookieRequest where
     size = ConstSize 64 -- 32 byte key + 32 byte padding