refactoring how we handle peer addresses in peerinfo/ats/transport/hello subsystems...
[oweals/gnunet.git] / src / peerinfo-tool / gnunet-peerinfo.c
1 /*
2      This file is part of GNUnet.
3      (C) 2001, 2002, 2003, 2004, 2006, 2009, 2010 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 peerinfo-tool/gnunet-peerinfo.c
23  * @brief Print information about other known peers.
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_crypto_lib.h"
28 #include "gnunet_configuration_lib.h"
29 #include "gnunet_getopt_lib.h"
30 #include "gnunet_peerinfo_service.h"
31 #include "gnunet_transport_service.h"
32 #include "gnunet_program_lib.h"
33
34 static int no_resolve;
35
36 static int be_quiet;
37
38 static int get_self;
39
40 static struct GNUNET_PEERINFO_Handle *peerinfo;
41
42 static const struct GNUNET_CONFIGURATION_Handle *cfg;
43
44 struct PrintContext
45 {
46   struct GNUNET_PeerIdentity peer;
47   char **address_list;
48   unsigned int num_addresses;
49   uint32_t off;
50 };
51
52
53 static void
54 dump_pc (struct PrintContext *pc)
55 {
56   struct GNUNET_CRYPTO_HashAsciiEncoded enc;
57   unsigned int i;
58
59   GNUNET_CRYPTO_hash_to_enc (&pc->peer.hashPubKey, &enc);
60   printf (_("Peer `%s'\n"), (const char *) &enc);
61   for (i = 0; i < pc->num_addresses; i++)
62   {
63     printf ("\t%s\n", pc->address_list[i]);
64     GNUNET_free (pc->address_list[i]);
65   }
66   printf ("\n");
67   GNUNET_array_grow (pc->address_list, pc->num_addresses, 0);
68   GNUNET_free (pc);
69 }
70
71
72 /**
73  * Function to call with a human-readable format of an address
74  *
75  * @param cls closure
76  * @param address NULL on error, otherwise 0-terminated printable UTF-8 string
77  */
78 static void
79 process_resolved_address (void *cls, const char *address)
80 {
81   struct PrintContext *pc = cls;
82
83   if (address == NULL)
84   {
85     pc->off--;
86     if (pc->off == 0)
87       dump_pc (pc);
88     return;
89   }
90   GNUNET_array_append (pc->address_list, pc->num_addresses,
91                        GNUNET_strdup (address));
92 }
93
94
95 /**
96  * Iterator callback to go over all addresses.
97  *
98  * @param cls closure
99  * @param address the address
100  * @param expiration expiration time
101  * @return GNUNET_OK to keep the address and continue
102  */
103 static int
104 count_address (void *cls, 
105                const struct GNUNET_HELLO_Address *address,
106                struct GNUNET_TIME_Absolute expiration)
107 {
108   struct PrintContext *pc = cls;
109
110   pc->off++;
111   return GNUNET_OK;
112 }
113
114
115 /**
116  * Iterator callback to go over all addresses.
117  *
118  * @param cls closure
119  * @param address the address
120  * @param expiration expiration time
121  * @return GNUNET_OK to keep the address and continue
122  */
123 static int
124 print_address (void *cls,
125                const struct GNUNET_HELLO_Address *address,
126                struct GNUNET_TIME_Absolute expiration)
127 {
128   struct PrintContext *pc = cls;
129
130   GNUNET_TRANSPORT_address_lookup (cfg, 
131                                    address->address, 
132                                    address->address_length, no_resolve, 
133                                    address->transport_name,
134                                    GNUNET_TIME_relative_multiply
135                                    (GNUNET_TIME_UNIT_SECONDS, 10),
136                                    &process_resolved_address, pc);
137   return GNUNET_OK;
138 }
139
140
141 /**
142  * Print information about the peer.
143  * Currently prints the GNUNET_PeerIdentity and the IP.
144  * Could of course do more (e.g. resolve via DNS).
145  */
146 static void
147 print_peer_info (void *cls, const struct GNUNET_PeerIdentity *peer,
148                  const struct GNUNET_HELLO_Message *hello, const char *err_msg)
149 {
150   struct GNUNET_CRYPTO_HashAsciiEncoded enc;
151   struct PrintContext *pc;
152
153   if (peer == NULL)
154   {
155     if (err_msg != NULL)
156       fprintf (stderr, _("Error in communication with PEERINFO service\n"));
157     GNUNET_PEERINFO_disconnect (peerinfo);
158     return;
159   }
160   if ((be_quiet) || (NULL == hello))
161   {
162     GNUNET_CRYPTO_hash_to_enc (&peer->hashPubKey, &enc);
163     printf ("%s\n", (const char *) &enc);
164     return;
165   }
166   pc = GNUNET_malloc (sizeof (struct PrintContext));
167   pc->peer = *peer;
168   GNUNET_HELLO_iterate_addresses (hello, GNUNET_NO, &count_address, pc);
169   if (0 == pc->off)
170   {
171     dump_pc (pc);
172     return;
173   }
174   GNUNET_HELLO_iterate_addresses (hello, GNUNET_NO, &print_address, pc);
175 }
176
177
178 /**
179  * Main function that will be run by the scheduler.
180  *
181  * @param cls closure
182  * @param args remaining command-line arguments
183  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
184  * @param c configuration
185  */
186 static void
187 run (void *cls, char *const *args, const char *cfgfile,
188      const struct GNUNET_CONFIGURATION_Handle *c)
189 {
190   struct GNUNET_CRYPTO_RsaPrivateKey *priv;
191   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pub;
192   struct GNUNET_PeerIdentity pid;
193   struct GNUNET_CRYPTO_HashAsciiEncoded enc;
194   char *fn;
195
196   cfg = c;
197   if (args[0] != NULL)
198   {
199     fprintf (stderr, _("Invalid command line argument `%s'\n"), args[0]);
200     return;
201   }
202   if (get_self != GNUNET_YES)
203   {
204     peerinfo = GNUNET_PEERINFO_connect (cfg);
205     if (peerinfo == NULL)
206     {
207       fprintf (stderr, _("Could not access PEERINFO service.  Exiting.\n"));
208       return;
209     }
210     GNUNET_PEERINFO_iterate (peerinfo, NULL,
211                              GNUNET_TIME_relative_multiply
212                              (GNUNET_TIME_UNIT_SECONDS, 5), &print_peer_info,
213                              NULL);
214   }
215   else
216   {
217     if (GNUNET_OK !=
218         GNUNET_CONFIGURATION_get_value_filename (cfg, "GNUNETD", "HOSTKEY",
219                                                  &fn))
220     {
221       fprintf (stderr, _("Could not find option `%s:%s' in configuration.\n"),
222                "GNUNETD", "HOSTKEYFILE");
223       return;
224     }
225     priv = GNUNET_CRYPTO_rsa_key_create_from_file (fn);
226     if (priv == NULL)
227     {
228       fprintf (stderr, _("Loading hostkey from `%s' failed.\n"), fn);
229       GNUNET_free (fn);
230       return;
231     }
232     GNUNET_free (fn);
233     GNUNET_CRYPTO_rsa_key_get_public (priv, &pub);
234     GNUNET_CRYPTO_rsa_key_free (priv);
235     GNUNET_CRYPTO_hash (&pub, sizeof (pub), &pid.hashPubKey);
236     GNUNET_CRYPTO_hash_to_enc (&pid.hashPubKey, &enc);
237     if (be_quiet)
238       printf ("%s\n", (char *) &enc);
239     else
240       printf (_("I am peer `%s'.\n"), (const char *) &enc);
241   }
242 }
243
244
245 /**
246  * The main function to obtain peer information.
247  *
248  * @param argc number of arguments from the command line
249  * @param argv command line arguments
250  * @return 0 ok, 1 on error
251  */
252 int
253 main (int argc, char *const *argv)
254 {
255   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
256     {'n', "numeric", NULL,
257      gettext_noop ("don't resolve host names"),
258      0, &GNUNET_GETOPT_set_one, &no_resolve},
259     {'q', "quiet", NULL,
260      gettext_noop ("output only the identity strings"),
261      0, &GNUNET_GETOPT_set_one, &be_quiet},
262     {'s', "self", NULL,
263      gettext_noop ("output our own identity only"),
264      0, &GNUNET_GETOPT_set_one, &get_self},
265     GNUNET_GETOPT_OPTION_END
266   };
267   return (GNUNET_OK ==
268           GNUNET_PROGRAM_run (argc, argv, "gnunet-peerinfo",
269                               gettext_noop ("Print information about peers."),
270                               options, &run, NULL)) ? 0 : 1;
271 }
272
273 /* end of gnunet-peerinfo.c */