-dead

[oweals/gnunet.git] / src / util / gnunet-rsa.c

diff --git a/src/util/gnunet-rsa.c b/src/util/gnunet-rsa.c
index 61e1b66dffa71f52af562b5caf3386fef4dbd65f..cdf69384141a04e2ecca34268620dfbe5748b85e 100644 (file)
--- a/src/util/gnunet-rsa.c
+++ b/src/util/gnunet-rsa.c
@@ -43,6 +43,15 @@ static int print_peer_identity;
  */
 static int print_short_identity;
 
+/**
+ * Use weak random number generator for key generation.
+ */
+static int weak_random;
+
+/**
+ * Option set to create a bunch of keys at once.
+ */
+static unsigned int make_keys;
 
 /**
  * The private information of an RSA key pair.
@@ -54,14 +63,13 @@ struct GNUNET_CRYPTO_RsaPrivateKey
 };
 
 
-#if 0
 /**
  * Create a new private key. Caller must free return value.
  *
  * @return fresh private key
  */
-struct GNUNET_CRYPTO_RsaPrivateKey *
-GNUNET_CRYPTO_rsa_key_create ()
+static struct GNUNET_CRYPTO_RsaPrivateKey *
+rsa_key_create ()
 {
   struct GNUNET_CRYPTO_RsaPrivateKey *ret;
   gcry_sexp_t s_key;
@@ -70,7 +78,7 @@ GNUNET_CRYPTO_rsa_key_create ()
   GNUNET_assert (0 ==
                  gcry_sexp_build (&s_keyparam, NULL,
                                   "(genkey(rsa(nbits %d)(rsa-use-e 3:257)))",
-                                  HOSTKEY_LEN));
+                                  2048));
   GNUNET_assert (0 == gcry_pk_genkey (&s_key, s_keyparam));
   gcry_sexp_release (s_keyparam);
 #if EXTRA_CHECKS
@@ -80,7 +88,57 @@ GNUNET_CRYPTO_rsa_key_create ()
   ret->sexp = s_key;
   return ret;
 }
-#endif
+
+
+/**
+ * Create a flat file with a large number of key pairs for testing.
+ */
+static void
+create_keys (const char *fn)
+{
+  FILE *f;
+  struct GNUNET_CRYPTO_RsaPrivateKey *pk;
+  struct GNUNET_CRYPTO_RsaPrivateKeyBinaryEncoded *enc;
+
+  if (NULL == (f = fopen (fn, "w+")))
+    {
+      fprintf (stderr,
+              _("Failed to open `%s': %s\n"),
+              fn,
+              STRERROR (errno));
+      return;
+    }
+  fprintf (stderr,
+          _("Generating %u keys, please wait"),
+          make_keys);
+  while (0 < make_keys--)
+  {    
+    fprintf (stderr,
+            ".");
+    if (NULL == (pk = rsa_key_create ()))
+    {
+       GNUNET_break (0);
+       break;
+    }
+    enc = GNUNET_CRYPTO_rsa_encode_key (pk);
+    if (htons (enc->len) != fwrite (enc, 1, htons (enc->len), f))
+      {
+       fprintf (stderr,
+                _("\nFailed to write to `%s': %s\n"),
+                fn,
+                STRERROR (errno));
+       GNUNET_CRYPTO_rsa_key_free (pk);
+       GNUNET_free (enc);
+       break;
+      }
+    GNUNET_CRYPTO_rsa_key_free (pk);
+    GNUNET_free (enc);
+  }
+  if (0 == make_keys)
+    fprintf (stderr,
+            _("Finished!\n"));
+  fclose (f);
+}
 
 
 /**
@@ -104,6 +162,13 @@ run (void *cls, char *const *args, const char *cfgfile,
     fprintf (stderr, _("No hostkey file specified on command line\n"));
     return;
   }
+  if (0 != weak_random)    
+    GNUNET_CRYPTO_random_disable_entropy_gathering ();  
+  if (make_keys > 0)
+  {
+    create_keys (args[0]);
+    return;
+  }
   pk = GNUNET_CRYPTO_rsa_key_create_from_file (args[0]);
   if (NULL == pk)
     return;
@@ -150,6 +215,9 @@ int
 main (int argc, char *const *argv)
 {
   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
+    { 'g', "generate-keys", "COUNT",
+      gettext_noop ("create COUNT public-private key pairs (for testing)"),
+      1, &GNUNET_GETOPT_set_uint, &make_keys },
     { 'p', "print-public-key", NULL,
       gettext_noop ("print the public key in ASCII format"),
       0, &GNUNET_GETOPT_set_one, &print_public_key },
@@ -159,6 +227,9 @@ main (int argc, char *const *argv)
     { 's', "print-short-identity", NULL,
       gettext_noop ("print the short hash of the public key in ASCII format"),
       0, &GNUNET_GETOPT_set_one, &print_short_identity },
+    { 'w', "weak-random", NULL,
+      gettext_noop ("use insecure, weak random number generator for key generation (for testing only)"),
+      0, &GNUNET_GETOPT_set_one, &weak_random },
     GNUNET_GETOPT_OPTION_END
   };