Update rng cli - can make multiple requests

1 files changed, 10 insertions, 5 deletions

diff --git a/src/cli/utils.cpp b/src/cli/utils.cpp
index 199a7894f..76e445126 100644
--- a/src/cli/utils.cpp
+++ b/src/cli/utils.cpp
@@ -149,24 +149,29 @@ BOTAN_REGISTER_COMMAND("hash", Hash);
 class RNG final : public Command
    {
    public:
-      RNG() : Command("rng bytes --system") {}
+      RNG() : Command("rng --system *bytes") {}
 
       void go() override
          {
-         const size_t bytes = get_arg_sz("bytes");
+         std::unique_ptr<Botan::RNG> rng;
 
          if(flag_set("system"))
             {
 #if defined(BOTAN_HAS_SYSTEM_RNG)
-            output() << Botan::hex_encode(Botan::system_rng().random_vec(bytes)) << "\n";
+            rng.reset(new Botan::System_RNG);
 #else
             error_output() << "system_rng disabled in build\n";
+            return;
 #endif
             }
          else
             {
-            Botan::AutoSeeded_RNG rng;
-            output() << Botan::hex_encode(rng.random_vec(bytes)) << "\n";
+            rng.reset(new Botan::AutoSeeded_RNG);
+            }
+
+         for(const std::string& req : get_arg_list("bytes"))
+            {
+            output() << Botan::hex_encode(rng->random_vec(Botan::to_u32bit(req))) << "\n";
             }
          }
    };