1 files changed, 11 insertions, 4 deletions

diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index bb9be560..dbedf801 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -60,6 +60,7 @@ import Control.Exception
 import Data.ByteString as BS
 import Data.Conduit as C
 import Data.Conduit.List as C
+import Data.Serialize
 import Network.Socket
 
 import Data.Torrent
@@ -182,15 +183,21 @@ isBootstrapped = T.full <$> getTable
 --
 --   This is blocking operation, use
 --   'Control.Concurrent.Async.Lifted.async' if needed.
-restore :: ByteString -> IO (Node ip)
-restore = error "DHT.restore: not implemented"
+restore :: Address ip => ByteString -> DHT ip ()
+restore bs = do
+    tblvar <- asks routingTable
+    case decode bs of
+        Right tbl -> restoreTable tbl
+        Left _ -> return ()
 
 -- | Serialize current DHT session to byte string.
 --
 --   This is blocking operation, use
 -- 'Control.Concurrent.Async.Lifted.async' if needed.
-snapshot :: DHT ip ByteString
-snapshot = error "DHT.snapshot: not implemented"
+snapshot :: Address ip => DHT ip ByteString
+snapshot = do
+    tbl <- getTable
+    return $ encode tbl
 
 {-----------------------------------------------------------------------
 --  Operations