Various small fixes and cleanups, new is_prime util

1 files changed, 33 insertions, 0 deletions

diff --git a/src/cmd/is_prime.cpp b/src/cmd/is_prime.cpp
new file mode 100644
index 000000000..658401690
--- /dev/null
+++ b/src/cmd/is_prime.cpp
@@ -0,0 +1,33 @@
+#include "apps.h"
+#include <botan/numthry.h>
+
+int is_prime_main(int argc, char* argv[])
+   {
+   if(argc != 2 && argc != 3)
+      {
+      std::cerr << "Usage: " << argv[0] << " n <prob>\n";
+      return 2;
+      }
+
+   BigInt n(argv[1]);
+
+   size_t prob = 56;
+
+   if(argc == 3)
+      prob = to_u32bit(argv[2]);
+
+   AutoSeeded_RNG rng;
+
+   const bool prime = is_prime(n, rng, prob);
+
+   if(prime)
+      {
+      std::cout << n << " is prime\n";
+      return 0;
+      }
+   else
+      {
+      std::cout << n << " is not prime\n";
+      return 1;
+      }
+   }