-fix leak
[oweals/gnunet.git] / src / util / gnunet-rsa.c
1 /*
2      This file is part of GNUnet.
3      (C) 2012 Christian Grothoff (and other contributing authors)
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20
21 /**
22  * @file util/gnunet-rsa.c
23  * @brief tool to manipulate RSA key files
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28 #include <gcrypt.h>
29
30
31 /**
32  * Flag for printing public key.
33  */
34 static int print_public_key;
35
36 /**
37  * Flag for printing hash of public key.
38  */
39 static int print_peer_identity;
40
41 /**
42  * Flag for printing short hash of public key.
43  */
44 static int print_short_identity;
45
46 /**
47  * Use weak random number generator for key generation.
48  */
49 static int weak_random;
50
51 /**
52  * Option set to create a bunch of keys at once.
53  */
54 static unsigned int make_keys;
55
56 /**
57  * The private information of an RSA key pair.
58  * NOTE: this must match the definition in crypto_ksk.c and crypto_rsa.c!
59  */
60 struct GNUNET_CRYPTO_RsaPrivateKey
61 {
62   gcry_sexp_t sexp;
63 };
64
65
66 #if 0
67 /**
68  * Create a new private key. Caller must free return value.
69  *
70  * @return fresh private key
71  */
72 struct GNUNET_CRYPTO_RsaPrivateKey *
73 GNUNET_CRYPTO_rsa_key_create ()
74 {
75   struct GNUNET_CRYPTO_RsaPrivateKey *ret;
76   gcry_sexp_t s_key;
77   gcry_sexp_t s_keyparam;
78
79   GNUNET_assert (0 ==
80                  gcry_sexp_build (&s_keyparam, NULL,
81                                   "(genkey(rsa(nbits %d)(rsa-use-e 3:257)))",
82                                   HOSTKEY_LEN));
83   GNUNET_assert (0 == gcry_pk_genkey (&s_key, s_keyparam));
84   gcry_sexp_release (s_keyparam);
85 #if EXTRA_CHECKS
86   GNUNET_assert (0 == gcry_pk_testkey (s_key));
87 #endif
88   ret = GNUNET_malloc (sizeof (struct GNUNET_CRYPTO_RsaPrivateKey));
89   ret->sexp = s_key;
90   return ret;
91 }
92 #endif
93
94
95 /**
96  * Create a flat file with a large number of key pairs for testing.
97  */
98 static void
99 create_keys (const char *fn)
100 {
101   time_t start;
102   struct GNUNET_HashCode hc;
103   struct GNUNET_HashCode h2;
104   struct GNUNET_HashCode h3;
105   FILE *f;
106   struct GNUNET_CRYPTO_RsaPrivateKey *pk;
107   struct GNUNET_CRYPTO_RsaPrivateKeyBinaryEncoded *enc;
108
109   start = time (NULL);
110   GNUNET_CRYPTO_hash (&start, sizeof (start), &hc);
111   if (NULL == (f = fopen (fn, "w+")))
112     {
113       fprintf (stderr,
114                _("Failed to open `%s': %s\n"),
115                fn,
116                STRERROR (errno));
117       return;
118     }
119   fprintf (stderr,
120            _("Generating %u keys, please wait"),
121            make_keys);
122   while (0 < make_keys--)
123   {    
124     fprintf (stderr,
125              ".");
126     GNUNET_CRYPTO_hash (&make_keys, sizeof (make_keys), &h2);
127     GNUNET_CRYPTO_hash (&hc, sizeof (hc), &h3);
128     GNUNET_CRYPTO_hash_xor (&h2, &h3, &hc);
129     pk = GNUNET_CRYPTO_rsa_key_create_from_hash (&hc);
130     enc = GNUNET_CRYPTO_rsa_encode_key (pk);
131     if (htons (enc->len) != fwrite (enc, 1, htons (enc->len), f))
132       {
133         fprintf (stderr,
134                  _("\nFailed to write to `%s': %s\n"),
135                  fn,
136                  STRERROR (errno));
137         GNUNET_CRYPTO_rsa_key_free (pk);
138         break;
139       }
140     GNUNET_CRYPTO_rsa_key_free (pk);
141   }
142   if (0 == make_keys)
143     fprintf (stderr,
144              _("Finished!\n"));
145   fclose (f);
146 }
147
148
149 /**
150  * Main function that will be run by the scheduler.
151  *
152  * @param cls closure
153  * @param args remaining command-line arguments
154  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
155  * @param cfg configuration
156  */
157 static void
158 run (void *cls, char *const *args, const char *cfgfile,
159      const struct GNUNET_CONFIGURATION_Handle *cfg)
160 {
161   struct GNUNET_CRYPTO_RsaPrivateKey *pk;
162   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pub;
163   struct GNUNET_PeerIdentity pid;
164
165   if (NULL == args[0])
166   {
167     fprintf (stderr, _("No hostkey file specified on command line\n"));
168     return;
169   }
170   if (0 != weak_random)    
171     GNUNET_CRYPTO_random_disable_entropy_gathering ();  
172   if (make_keys > 0)
173   {
174     create_keys (args[0]);
175     return;
176   }
177   pk = GNUNET_CRYPTO_rsa_key_create_from_file (args[0]);
178   if (NULL == pk)
179     return;
180   if (print_public_key)
181   {
182     char *s;
183
184     GNUNET_CRYPTO_rsa_key_get_public (pk, &pub);
185     s = GNUNET_CRYPTO_rsa_public_key_to_string (&pub);
186     fprintf (stdout, "%s\n", s);
187     GNUNET_free (s);
188   }
189   if (print_peer_identity)
190   {
191     struct GNUNET_CRYPTO_HashAsciiEncoded enc;
192
193     GNUNET_CRYPTO_rsa_key_get_public (pk, &pub);
194     GNUNET_CRYPTO_hash (&pub, sizeof (pub), &pid.hashPubKey);
195     GNUNET_CRYPTO_hash_to_enc (&pid.hashPubKey, &enc);
196     fprintf (stdout, "%s\n", enc.encoding);
197   }
198   if (print_short_identity)
199   {
200     struct GNUNET_CRYPTO_ShortHashAsciiEncoded enc;
201     struct GNUNET_CRYPTO_ShortHashCode sh;
202
203     GNUNET_CRYPTO_rsa_key_get_public (pk, &pub);
204     GNUNET_CRYPTO_short_hash (&pub, sizeof (pub), &sh);
205     GNUNET_CRYPTO_short_hash_to_enc (&sh, &enc);
206     fprintf (stdout, "%s\n", enc.short_encoding);
207   }
208   GNUNET_CRYPTO_rsa_key_free (pk);
209 }
210
211
212 /**
213  * Program to manipulate RSA key files.
214  *
215  * @param argc number of arguments from the command line
216  * @param argv command line arguments
217  * @return 0 ok, 1 on error
218  */
219 int
220 main (int argc, char *const *argv)
221 {
222   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
223     { 'g', "generate-keys", "COUNT",
224       gettext_noop ("create COUNT public-private key pairs (for testing)"),
225       1, &GNUNET_GETOPT_set_uint, &make_keys },
226     { 'p', "print-public-key", NULL,
227       gettext_noop ("print the public key in ASCII format"),
228       0, &GNUNET_GETOPT_set_one, &print_public_key },
229     { 'P', "print-peer-identity", NULL,
230       gettext_noop ("print the hash of the public key in ASCII format"),
231       0, &GNUNET_GETOPT_set_one, &print_peer_identity },
232     { 's', "print-short-identity", NULL,
233       gettext_noop ("print the short hash of the public key in ASCII format"),
234       0, &GNUNET_GETOPT_set_one, &print_short_identity },
235     { 'w', "weak-random", NULL,
236       gettext_noop ("use insecure, weak random number generator for key generation (for testing only)"),
237       0, &GNUNET_GETOPT_set_one, &weak_random },
238     GNUNET_GETOPT_OPTION_END
239   };
240
241   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
242     return 2;
243
244   return (GNUNET_OK ==
245           GNUNET_PROGRAM_run (argc, argv, "gnunet-rsa [OPTIONS] keyfile",
246                               gettext_noop ("Manipulate GNUnet private RSA key files"),
247                               options, &run, NULL)) ? 0 : 1;
248 }
249
250 /* end of gnunet-rsa.c */