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