-fix leak
[oweals/gnunet.git] / src / gns / gnunet-gns.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  * @file gnunet-gns.c
22  * @brief command line tool to access distributed GNS
23  * @author Christian Grothoff
24  *
25  * TODO:
26  * - everything
27  */
28 #include "platform.h"
29 #include <gnunet_util_lib.h>
30 #include <gnunet_dnsparser_lib.h>
31 #include <gnunet_namestore_service.h>
32 #include <gnunet_gns_service.h>
33
34 /**
35  * Handle to GNS service.
36  */
37 static struct GNUNET_GNS_Handle *gns;
38
39 /**
40  * GNS name to shorten. (-s option)
41  */
42 static char *shorten_name;
43
44 /**
45  * GNS name to lookup. (-u option)
46  */
47 static char *lookup_name;
48
49
50 /**
51  * record type to look up (-t option)
52  */
53 static char *lookup_type;
54
55 /**
56  * name to look up authority for (-a option)
57  */
58 static char *auth_name;
59
60 /**
61  * raw output
62  */
63 static int raw = 0;
64
65 static enum GNUNET_GNS_RecordType rtype;
66
67 /**
68  * Task run on shutdown.  Cleans up everything.
69  *
70  * @param cls unused
71  * @param tc scheduler context
72  */
73 static void
74 do_shutdown (void *cls,
75              const struct GNUNET_SCHEDULER_TaskContext *tc)
76 {
77   if (NULL != gns)
78   {
79     GNUNET_GNS_disconnect (gns);
80     gns = NULL;
81   }
82 }
83
84
85 static void
86 process_shorten_result(void* cls, const char* nshort)
87 {
88   printf("%s shortened to %s\n", (char*) cls, nshort);
89   GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
90 }
91
92 static void
93 process_lookup_result(void* cls, uint32_t rd_count,
94                       const struct GNUNET_NAMESTORE_RecordData *rd)
95 {
96   int i;
97   char* name = (char*) cls;
98   const char* typename;
99   char* string_val;
100   
101   if (!raw) {
102     if (rd_count == 0)
103       printf("No results.\n");
104     else
105       printf("%s:\n", name);
106   }
107
108
109
110   for (i=0; i<rd_count; i++)
111   {
112     typename = GNUNET_NAMESTORE_number_to_typename (rd[i].record_type);
113     string_val = GNUNET_NAMESTORE_value_to_string(rd[i].record_type,
114                                                   rd[i].data,
115                                                   rd[i].data_size);
116     if (raw)
117       printf("%s\n", string_val);
118     else
119       printf("Got %s record: %s\n", typename, string_val);
120
121   }
122
123   GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
124 }
125
126 static void
127 process_auth_result(void* cls, const char* auth)
128 {
129   printf ("%s\n", auth);
130   GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
131 }
132
133 /**
134  * Main function that will be run.
135  *
136  * @param cls closure
137  * @param args remaining command-line arguments
138  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
139  * @param cfg configuration
140  */
141 static void
142 run (void *cls, char *const *args, const char *cfgfile,
143      const struct GNUNET_CONFIGURATION_Handle *cfg)
144 {
145   char* keyfile;
146   struct GNUNET_CRYPTO_RsaPrivateKey *key = NULL;
147   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pkey;
148   struct GNUNET_CRYPTO_ShortHashCode *zone = NULL;
149   struct GNUNET_CRYPTO_ShortHashCode user_zone;
150   struct GNUNET_CRYPTO_ShortHashAsciiEncoded zonename;
151
152   if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns",
153                                                            "ZONEKEY", &keyfile))
154   {
155     if (!raw)
156       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
157                   "No private key for root zone found, using default!\n");
158     zone = NULL;
159   }
160   else
161   {
162     if (GNUNET_YES == GNUNET_DISK_file_test (keyfile))
163     {
164       key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
165       GNUNET_CRYPTO_rsa_key_get_public (key, &pkey);
166       GNUNET_CRYPTO_short_hash(&pkey,
167                          sizeof(struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
168                          &user_zone);
169       zone = &user_zone;
170       GNUNET_CRYPTO_short_hash_to_enc (zone, &zonename);
171       printf("asdsad\n");
172       if (!raw)
173         GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
174                     "Using zone: %s!\n", &zonename);
175       GNUNET_CRYPTO_rsa_key_free(key);
176     }
177     GNUNET_free(keyfile);
178   }
179
180   gns = GNUNET_GNS_connect (cfg);
181   if (lookup_type != NULL)
182     rtype = GNUNET_NAMESTORE_typename_to_number(lookup_type);
183   else
184     rtype = GNUNET_GNS_RECORD_TYPE_A;
185
186   if (NULL == gns)
187   {
188     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
189                 _("Failed to connect to GNS\n"));
190     return;
191   }
192   
193   if (shorten_name != NULL)
194   {
195     /** shorten name */
196     GNUNET_GNS_shorten_zone (gns, shorten_name, zone, &process_shorten_result,
197                        shorten_name);
198   }
199
200   if (lookup_name != NULL)
201   {
202     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
203                 "Lookup\n");
204     GNUNET_GNS_lookup_zone (gns, lookup_name, zone, rtype,
205                       &process_lookup_result, lookup_name);
206   }
207
208   if (auth_name != NULL)
209   {
210     GNUNET_GNS_get_authority(gns, auth_name, &process_auth_result, auth_name);
211   }
212   
213   // FIXME: do work here...
214   //GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
215 }
216
217
218 /**
219  * The main function for gnunet-gns.
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     {'s', "shorten", NULL,
230      gettext_noop ("try to shorten a given GNS name"), 1,
231      &GNUNET_GETOPT_set_string, &shorten_name},
232     {'u', "lookup", NULL,
233       gettext_noop ("Lookup a record using GNS (NOT IMPLEMENTED)"), 1,
234       &GNUNET_GETOPT_set_string, &lookup_name},
235     {'a', "authority", NULL,
236       gettext_noop ("Get the authority of a particular name"), 1,
237       &GNUNET_GETOPT_set_string, &auth_name},
238     {'t', "type", NULL,
239       gettext_noop ("Specify the type of the record lookup"), 1,
240       &GNUNET_GETOPT_set_string, &lookup_type},
241     {'r', "raw", NULL,
242       gettext_noop ("No unneeded output"), 0,
243       &GNUNET_GETOPT_set_one, &raw},
244     GNUNET_GETOPT_OPTION_END
245   };
246
247   int ret;
248
249   GNUNET_log_setup ("gnunet-gns", "WARNING", NULL);
250   ret =
251       (GNUNET_OK ==
252        GNUNET_PROGRAM_run (argc, argv, "gnunet-gns",
253                            _("GNUnet GNS access tool"), 
254                            options,
255                            &run, NULL)) ? 0 : 1;
256
257   return ret;
258 }
259
260 /* end of gnunet-gns.c */