src/dht/gnunet-service-dht.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2009, 2010, 2011, 2016 GNUnet e.V.
   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., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, 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_transport_hello_service.h"
  32 #include "gnunet_hello_lib.h"
  33 #include "gnunet_dht_service.h"
  34 #include "gnunet_statistics_service.h"
  35 #include "gnunet-service-dht.h"
  36 #include "gnunet-service-dht_clients.h"
  37 #include "gnunet-service-dht_datacache.h"
  38 #include "gnunet-service-dht_hello.h"
  39 #include "gnunet-service-dht_neighbours.h"
  40 #include "gnunet-service-dht_nse.h"
  41 #include "gnunet-service-dht_routing.h"
  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  * Handle to our server.
  61  */
  62 struct GNUNET_SERVER_Handle *GDS_server;
  63
  64 /**
  65  * Our HELLO
  66  */
  67 struct GNUNET_MessageHeader *GDS_my_hello;
  68
  69 /**
  70  * Handle to get our current HELLO.
  71  */
  72 static struct GNUNET_TRANSPORT_HelloGetHandle *ghh;
  73
  74 /**
  75  * Hello address expiration
  76  */
  77 struct GNUNET_TIME_Relative hello_expiration;
  78
  79
  80 /**
  81  * Receive the HELLO from transport service, free current and replace
  82  * if necessary.
  83  *
  84  * @param cls NULL
  85  * @param message HELLO message of peer
  86  */
  87 static void
  88 process_hello (void *cls,
  89                const struct GNUNET_MessageHeader *message)
  90 {
  91   GNUNET_free_non_null (GDS_my_hello);
  92   GDS_my_hello = GNUNET_malloc (ntohs (message->size));
  93   GNUNET_memcpy (GDS_my_hello, message, ntohs (message->size));
  94 }
  95
  96
  97 /**
  98  * Task run during shutdown.
  99  *
 100  * @param cls unused
 101  */
 102 static void
 103 shutdown_task (void *cls)
 104 {
 105   if (NULL != ghh)
 106   {
 107     GNUNET_TRANSPORT_hello_get_cancel (ghh);
 108     ghh = NULL;
 109   }
 110   GDS_NEIGHBOURS_done ();
 111   GDS_DATACACHE_done ();
 112   GDS_ROUTING_done ();
 113   GDS_HELLO_done ();
 114   GDS_NSE_done ();
 115   if (NULL != GDS_block_context)
 116   {
 117     GNUNET_BLOCK_context_destroy (GDS_block_context);
 118     GDS_block_context = NULL;
 119   }
 120   if (NULL != GDS_stats)
 121   {
 122     GNUNET_STATISTICS_destroy (GDS_stats,
 123                                GNUNET_YES);
 124     GDS_stats = NULL;
 125   }
 126   GNUNET_free_non_null (GDS_my_hello);
 127   GDS_my_hello = NULL;
 128 }
 129
 130
 131 /**
 132  * Process dht requests.
 133  *
 134  * @param cls closure
 135  * @param server the initialized server
 136  * @param c configuration to use
 137  */
 138 static void
 139 run (void *cls,
 140      struct GNUNET_SERVER_Handle *server,
 141      const struct GNUNET_CONFIGURATION_Handle *c)
 142 {
 143   GDS_cfg = c;
 144   GDS_server = server;
 145   GNUNET_SERVER_suspend (server);
 146   if (GNUNET_OK !=
 147       GNUNET_CONFIGURATION_get_value_time (c,
 148                                            "transport",
 149                                            "HELLO_EXPIRATION",
 150                                            &hello_expiration))
 151   {
 152     hello_expiration = GNUNET_CONSTANTS_HELLO_ADDRESS_EXPIRATION;
 153   }
 154   GDS_block_context = GNUNET_BLOCK_context_create (GDS_cfg);
 155   GDS_stats = GNUNET_STATISTICS_create ("dht", GDS_cfg);
 156   GDS_ROUTING_init ();
 157   GDS_NSE_init ();
 158   GDS_DATACACHE_init ();
 159   GDS_HELLO_init ();
 160   if (GNUNET_OK != GDS_NEIGHBOURS_init ())
 161   {
 162     shutdown_task (NULL);
 163     return;
 164   }
 165   GNUNET_SCHEDULER_add_shutdown (&shutdown_task,
 166                                  NULL);
 167   ghh = GNUNET_TRANSPORT_hello_get (GDS_cfg,
 168                                     GNUNET_TRANSPORT_AC_GLOBAL,
 169                                     &process_hello,
 170                                     NULL);
 171 }
 172
 173
 174 /**
 175  * The main function for the dht service.
 176  *
 177  * @param argc number of arguments from the command line
 178  * @param argv command line arguments
 179  * @return 0 ok, 1 on error
 180  */
 181 int
 182 main (int argc,
 183       char *const *argv)
 184 {
 185   int ret;
 186
 187   ret = (GNUNET_OK ==
 188          GNUNET_SERVICE_run (argc,
 189                              argv,
 190                              "dht",
 191                              GNUNET_SERVICE_OPTION_NONE,
 192                              &run,
 193                              NULL)) ? 0 : 1;
 194   GDS_CLIENTS_done ();
 195   return ret;
 196 }
 197
 198 /* end of gnunet-service-dht.c */