Command 'r' to generate random node ids.

1 files changed, 14 insertions, 0 deletions

diff --git a/dht/examples/dhtd.hs b/dht/examples/dhtd.hs
index 9c03a4f9..d028a308 100644
--- a/dht/examples/dhtd.hs
+++ b/dht/examples/dhtd.hs
@@ -28,6 +28,7 @@ import Control.Concurrent.STM.TMChan
 import Control.Exception
 import Control.Monad
 import Control.Monad.IO.Class (liftIO)
+import Crypto.Random (getRandomBytes)
 import Data.Array.MArray (getAssocs)
 import Data.Bool
 import Data.Bits (xor)
@@ -400,6 +401,7 @@ clientSession s@Session{..} sock cnum h = do
             , ["nid"]
             , ["lan"]
             , ["ls"]
+            , ["r"]
             , ["k"]
             , ["roster"]
             , ["sessions"]
@@ -553,6 +555,18 @@ clientSession s@Session{..} sock cnum h = do
                                          , ("node-id", show $ thisNode bkts)
                                          , ("network", netname) ]
 
+        ("r", s) | Just DHT{dhtQuery,dhtBuckets} <- Map.lookup netname dhts
+                 , Just DHTQuery{qsearch} <- Map.lookup "node" dhtQuery
+                   -> cmd0 $ do
+                        ni <- atomically $ thisNode <$> readTVar dhtBuckets
+                        let kad = searchSpace qsearch
+                            nid = kademliaLocation kad ni
+                            b = case readMaybe $ strp s of
+                                    Nothing -> bucketRange 0 True
+                                    Just n  -> bucketRange n False
+                        rnid <- kademliaSample kad getRandomBytes nid b
+                        hPutClient h $ show rnid
+
         --  TODO: online documentation.
         --
         --  k - manage key-pairs