-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         GNUNET_free (enc);
139         break;
140       }
141     GNUNET_CRYPTO_rsa_key_free (pk);
142     GNUNET_free (enc);
143   }
144   if (0 == make_keys)
145     fprintf (stderr,
146              _("Finished!\n"));
147   fclose (f);
148 }
149
150
151 /**
152  * Main function that will be run by the scheduler.
153  *
154  * @param cls closure
155  * @param args remaining command-line arguments
156  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
157  * @param cfg configuration
158  */
159 static void
160 run (void *cls, char *const *args, const char *cfgfile,
161      const struct GNUNET_CONFIGURATION_Handle *cfg)
162 {
163   struct GNUNET_CRYPTO_RsaPrivateKey *pk;
164   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pub;
165   struct GNUNET_PeerIdentity pid;
166
167   if (NULL == args[0])
168   {
169     fprintf (stderr, _("No hostkey file specified on command line\n"));
170     return;
171   }
172   if (0 != weak_random)    
173     GNUNET_CRYPTO_random_disable_entropy_gathering ();  
174   if (make_keys > 0)
175   {
176     create_keys (args[0]);
177     return;
178   }
179   pk = GNUNET_CRYPTO_rsa_key_create_from_file (args[0]);
180   if (NULL == pk)
181     return;
182   if (print_public_key)
183   {
184     char *s;
185
186     GNUNET_CRYPTO_rsa_key_get_public (pk, &pub);
187     s = GNUNET_CRYPTO_rsa_public_key_to_string (&pub);
188     fprintf (stdout, "%s\n", s);
189     GNUNET_free (s);
190   }
191   if (print_peer_identity)
192   {
193     struct GNUNET_CRYPTO_HashAsciiEncoded enc;
194
195     GNUNET_CRYPTO_rsa_key_get_public (pk, &pub);
196     GNUNET_CRYPTO_hash (&pub, sizeof (pub), &pid.hashPubKey);
197     GNUNET_CRYPTO_hash_to_enc (&pid.hashPubKey, &enc);
198     fprintf (stdout, "%s\n", enc.encoding);
199   }
200   if (print_short_identity)
201   {
202     struct GNUNET_CRYPTO_ShortHashAsciiEncoded enc;
203     struct GNUNET_CRYPTO_ShortHashCode sh;
204
205     GNUNET_CRYPTO_rsa_key_get_public (pk, &pub);
206     GNUNET_CRYPTO_short_hash (&pub, sizeof (pub), &sh);
207     GNUNET_CRYPTO_short_hash_to_enc (&sh, &enc);
208     fprintf (stdout, "%s\n", enc.short_encoding);
209   }
210   GNUNET_CRYPTO_rsa_key_free (pk);
211 }
212
213
214 /**
215  * Program to manipulate RSA key files.
216  *
217  * @param argc number of arguments from the command line
218  * @param argv command line arguments
219  * @return 0 ok, 1 on error
220  */
221 int
222 main (int argc, char *const *argv)
223 {
224   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
225     { 'g', "generate-keys", "COUNT",
226       gettext_noop ("create COUNT public-private key pairs (for testing)"),
227       1, &GNUNET_GETOPT_set_uint, &make_keys },
228     { 'p', "print-public-key", NULL,
229       gettext_noop ("print the public key in ASCII format"),
230       0, &GNUNET_GETOPT_set_one, &print_public_key },
231     { 'P', "print-peer-identity", NULL,
232       gettext_noop ("print the hash of the public key in ASCII format"),
233       0, &GNUNET_GETOPT_set_one, &print_peer_identity },
234     { 's', "print-short-identity", NULL,
235       gettext_noop ("print the short hash of the public key in ASCII format"),
236       0, &GNUNET_GETOPT_set_one, &print_short_identity },
237     { 'w', "weak-random", NULL,
238       gettext_noop ("use insecure, weak random number generator for key generation (for testing only)"),
239       0, &GNUNET_GETOPT_set_one, &weak_random },
240     GNUNET_GETOPT_OPTION_END
241   };
242
243   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
244     return 2;
245
246   return (GNUNET_OK ==
247           GNUNET_PROGRAM_run (argc, argv, "gnunet-rsa [OPTIONS] keyfile",
248                               gettext_noop ("Manipulate GNUnet private RSA key files"),
249                               options, &run, NULL)) ? 0 : 1;
250 }
251
252 /* end of gnunet-rsa.c */