src/core/gnunet-core-list-connections.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2011 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 core/gnunet-core-list-connections.c
  23  * @brief Print information about other known _connected_ peers.
  24  * @author Nathan Evans
  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_core_service.h"
  33 #include "gnunet_program_lib.h"
  34
  35 #define VERBOSE 0
  36 static int no_resolve;
  37
  38 #if VERBOSE
  39 static unsigned int peer_count;
  40 #endif
  41
  42 static const struct GNUNET_CONFIGURATION_Handle *cfg;
  43
  44 struct AddressStringList
  45 {
  46   /**
  47    * Pointer to previous element.
  48    */
  49   struct AddressStringList *prev;
  50
  51   /**
  52    * Pointer to next element.
  53    */
  54   struct AddressStringList *next;
  55
  56   /**
  57    * Address as string.
  58    */
  59   char *address_string;
  60 };
  61
  62 struct PrintContext
  63 {
  64   struct GNUNET_PeerIdentity peer;
  65   struct AddressStringList *address_list_head;
  66   struct AddressStringList *address_list_tail;
  67 };
  68
  69
  70 static void
  71 dump_pc (struct PrintContext *pc)
  72 {
  73   struct GNUNET_CRYPTO_HashAsciiEncoded enc;
  74   struct AddressStringList *address;
  75
  76   GNUNET_CRYPTO_hash_to_enc (&pc->peer.hashPubKey, &enc);
  77   printf (_("Peer `%s'\n"), (const char *) &enc);
  78   while (NULL != (address = pc->address_list_head))
  79   {
  80     printf ("\t%s\n", address->address_string);
  81     GNUNET_free (address->address_string);
  82     GNUNET_CONTAINER_DLL_remove (pc->address_list_head, pc->address_list_tail,
  83                                  address);
  84     GNUNET_free (address);
  85   }
  86
  87   printf ("\n");
  88
  89   GNUNET_free (pc);
  90 }
  91
  92
  93 /**
  94  * Function to call with a human-readable format of an address
  95  *
  96  * @param cls closure
  97  * @param address NULL on error, otherwise 0-terminated printable UTF-8 string
  98  */
  99 static void
 100 process_resolved_address (void *cls, const char *address)
 101 {
 102   struct PrintContext *pc = cls;
 103   struct AddressStringList *new_address;
 104
 105   if (address == NULL)
 106   {
 107     dump_pc (pc);
 108     return;
 109   }
 110
 111   new_address = GNUNET_malloc (sizeof (struct AddressStringList));
 112 #if VERBOSE
 113   fprintf (stderr, "Received address %s\n", address);
 114 #endif
 115   new_address->address_string = GNUNET_strdup (address);
 116   GNUNET_CONTAINER_DLL_insert (pc->address_list_head, pc->address_list_tail,
 117                                new_address);
 118 }
 119
 120
 121 /**
 122  * Callback for retrieving a list of connected peers.
 123  */
 124 static void
 125 connected_peer_callback (void *cls, const struct GNUNET_PeerIdentity *peer,
 126                          const struct GNUNET_TRANSPORT_ATS_Information *atsi)
 127 {
 128   struct PrintContext *pc;
 129
 130   if (peer != NULL)             /* Not yet finished */
 131   {
 132 #if VERBOSE
 133     fprintf (stderr, "Learned about peer %s\n", GNUNET_i2s (peer));
 134     peer_count++;
 135 #endif
 136     pc = GNUNET_malloc (sizeof (struct PrintContext));
 137     pc->peer = *peer;
 138     GNUNET_TRANSPORT_peer_address_lookup (cfg, peer,
 139                                           GNUNET_TIME_UNIT_MINUTES,
 140                                           &process_resolved_address, pc);
 141   }
 142 #if VERBOSE
 143   else
 144   {
 145     fprintf (stderr, "Counted %u total connected peers.\n", peer_count);
 146   }
 147 #endif
 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 c configuration
 158  */
 159 static void
 160 run (void *cls,
 161      char *const *args,
 162      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *c)
 163 {
 164
 165   cfg = c;
 166   if (args[0] != NULL)
 167   {
 168     fprintf (stderr, _("Invalid command line argument `%s'\n"), args[0]);
 169     return;
 170   }
 171
 172   GNUNET_CORE_iterate_peers (cfg, &connected_peer_callback, NULL);
 173
 174 }
 175
 176
 177 /**
 178  * The main function to obtain peer information.
 179  *
 180  * @param argc number of arguments from the command line
 181  * @param argv command line arguments
 182  * @return 0 ok, 1 on error
 183  */
 184 int
 185 main (int argc, char *const *argv)
 186 {
 187   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 188     {'n', "numeric", NULL,
 189      gettext_noop ("don't resolve host names"),
 190      0, &GNUNET_GETOPT_set_one, &no_resolve},
 191     GNUNET_GETOPT_OPTION_END
 192   };
 193   return (GNUNET_OK ==
 194           GNUNET_PROGRAM_run (argc,
 195                               argv,
 196                               "gnunet-list-connections",
 197                               gettext_noop
 198                               ("Print information about connected peers."),
 199                               options, &run, NULL)) ? 0 : 1;
 200 }
 201
 202 /* end of gnunet-core-list-connections.c */