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