src/dht/gnunet-service-xdht.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009, 2010, 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 dht/gnunet-service-xdht.c
  23  * @brief GNUnet DHT service
  24  * @author Christian Grothoff
  25  * @author Nathan Evans
  26  */
  27
  28 #include "platform.h"
  29 #include "gnunet_block_lib.h"
  30 #include "gnunet_util_lib.h"
  31 #include "gnunet_transport_service.h"
  32 #include "gnunet_hello_lib.h"
  33 #include "gnunet_dht_service.h"
  34 #include "gnunet_statistics_service.h"
  35 #include "gnunet-service-xdht.h"
  36 #include "gnunet-service-xdht_clients.h"
  37 #include "gnunet-service-xdht_datacache.h"
  38 #include "gnunet-service-xdht_hello.h"
  39 #include "gnunet-service-xdht_neighbours.h"
  40 #include "gnunet-service-xdht_nse.h"
  41 #include "gnunet-service-xdht_routing.h"
  42
  43
  44
  45 /**
  46  * Handle for the statistics service.
  47  */
  48 struct GNUNET_STATISTICS_Handle *GDS_stats;
  49
  50 /**
  51  * Our handle to the BLOCK library.
  52  */
  53 struct GNUNET_BLOCK_Context *GDS_block_context;
  54
  55 /**
  56  * The configuration the DHT service is running with
  57  */
  58 const struct GNUNET_CONFIGURATION_Handle *GDS_cfg;
  59
  60 /**
  61  * Our HELLO
  62  */
  63 struct GNUNET_MessageHeader *GDS_my_hello;
  64
  65 /**
  66  * Handle to the transport service, for getting our hello
  67  */
  68 struct GNUNET_TRANSPORT_Handle *GDS_transport_handle;
  69
  70 /**
  71  * Handle to get our current HELLO.
  72  */
  73 static struct GNUNET_TRANSPORT_GetHelloHandle *ghh;
  74
  75 /**
  76  * Hello address expiration
  77  */
  78 struct GNUNET_TIME_Relative hello_expiration;
  79
  80 /**
  81  * Should we store our topology predecessor and successor IDs into statistics?
  82  */
  83 extern unsigned int track_topology;
  84
  85 #if ENABLE_MALICIOUS
  86 /**
  87  * Should this peer act malicious?
  88  */
  89 unsigned int malicious;
  90 #endif
  91
  92
  93 /**
  94  * Receive the HELLO from transport service, free current and replace
  95  * if necessary.
  96  *
  97  * @param cls NULL
  98  * @param message HELLO message of peer
  99  */
 100 static void
 101 process_hello (void *cls, const struct GNUNET_MessageHeader *message)
 102 {
 103   GNUNET_assert (message != NULL);
 104   GNUNET_free_non_null (GDS_my_hello);
 105   GDS_my_hello = GNUNET_malloc (ntohs (message->size));
 106   memcpy (GDS_my_hello, message, ntohs (message->size));
 107 }
 108
 109
 110 /**
 111  * Task run during shutdown.
 112  *
 113  * @param cls unused
 114  * @param tc unused
 115  */
 116 static void
 117 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 118 {
 119   if (NULL != ghh)
 120   {
 121     GNUNET_TRANSPORT_get_hello_cancel (ghh);
 122     ghh = NULL;
 123   }
 124   if (GDS_transport_handle != NULL)
 125   {
 126     GNUNET_TRANSPORT_disconnect (GDS_transport_handle);
 127     GDS_transport_handle = NULL;
 128   }
 129
 130   GDS_NEIGHBOURS_done ();
 131   GDS_DATACACHE_done ();
 132   GDS_ROUTING_done ();
 133   GDS_HELLO_done ();
 134   GDS_NSE_done ();
 135   if (GDS_block_context != NULL)
 136   {
 137     GNUNET_BLOCK_context_destroy (GDS_block_context);
 138     GDS_block_context = NULL;
 139   }
 140   if (GDS_stats != NULL)
 141   {
 142     GNUNET_STATISTICS_destroy (GDS_stats, GNUNET_YES);
 143     GDS_stats = NULL;
 144   }
 145   GNUNET_free_non_null (GDS_my_hello);
 146   GDS_my_hello = NULL;
 147 }
 148
 149
 150 /**
 151  * Process dht requests.
 152  *
 153  * @param cls closure
 154  * @param server the initialized server
 155  * @param c configuration to use
 156  */
 157 static void
 158 run (void *cls, struct GNUNET_SERVER_Handle *server,
 159      const struct GNUNET_CONFIGURATION_Handle *c)
 160 {
 161   unsigned long long _track_topology;
 162   GDS_cfg = c;
 163   if (GNUNET_OK !=
 164       GNUNET_CONFIGURATION_get_value_time (c, "transport", "HELLO_EXPIRATION", &hello_expiration))
 165   {
 166     hello_expiration = GNUNET_CONSTANTS_HELLO_ADDRESS_EXPIRATION;
 167   }
 168   GDS_block_context = GNUNET_BLOCK_context_create (GDS_cfg);
 169   GDS_stats = GNUNET_STATISTICS_create ("dht", GDS_cfg);
 170   GDS_ROUTING_init ();
 171   GDS_NSE_init ();
 172   GDS_DATACACHE_init ();
 173   GDS_HELLO_init ();
 174   GDS_CLIENTS_init (server);
 175   if (GNUNET_OK ==
 176       GNUNET_CONFIGURATION_get_value_number (c, "xdht", "track_toplogy",
 177                                              &_track_topology))
 178   {
 179     track_topology = (unsigned int) _track_topology;
 180   }
 181   if (GNUNET_OK != GDS_NEIGHBOURS_init ())
 182   {
 183     shutdown_task (NULL, NULL);
 184     return;
 185   }
 186   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
 187                                 NULL);
 188   GDS_transport_handle =
 189       GNUNET_TRANSPORT_connect (GDS_cfg, NULL, NULL, NULL, NULL, NULL);
 190   if (GDS_transport_handle == NULL)
 191   {
 192     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 193                 _("Failed to connect to transport service!\n"));
 194     return;
 195   }
 196   ghh = GNUNET_TRANSPORT_get_hello (GDS_transport_handle, &process_hello, NULL);
 197 }
 198
 199
 200 /**
 201  * The main function for the dht service.
 202  *
 203  * @param argc number of arguments from the command line
 204  * @param argv command line arguments
 205  * @return 0 ok, 1 on error
 206  */
 207 int
 208 main (int argc, char *const *argv)
 209 {
 210   int ret;
 211
 212   ret =
 213       (GNUNET_OK ==
 214        GNUNET_SERVICE_run (argc, argv, "dht", GNUNET_SERVICE_OPTION_NONE, &run,
 215                            NULL)) ? 0 : 1;
 216   GDS_CLIENTS_done ();
 217   return ret;
 218 }
 219
 220 /* end of gnunet-service-dht.c */