fix
[oweals/gnunet.git] / src / dht / gnunet-service-dht-new.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 /**
72  * Handle to get our current HELLO.
73  */
74 static struct GNUNET_TRANSPORT_GetHelloHandle *ghh;
75
76
77 /**
78  * Receive the HELLO from transport service, free current and replace
79  * if necessary.
80  *
81  * @param cls NULL
82  * @param message HELLO message of peer
83  */
84 static void
85 process_hello (void *cls, 
86                const struct GNUNET_MessageHeader *message)
87 {
88   GNUNET_assert (message != NULL);
89   GNUNET_free_non_null (GDS_my_hello);
90   GDS_my_hello = GNUNET_malloc (ntohs (message->size));
91   memcpy (GDS_my_hello, message, ntohs (message->size));
92 }
93
94
95 /**
96  * Task run during shutdown.
97  *
98  * @param cls unused
99  * @param tc unused
100  */
101 static void
102 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
103 {
104   if (NULL != ghh)
105   {
106     GNUNET_TRANSPORT_get_hello_cancel (ghh);
107     ghh = NULL;
108   }
109   if (GDS_transport_handle != NULL)
110   {
111     GNUNET_TRANSPORT_disconnect (GDS_transport_handle);
112     GDS_transport_handle = NULL;
113   }
114   GDS_NEIGHBOURS_done ();
115   GDS_DATACACHE_done ();
116   GDS_ROUTING_done ();
117   GDS_HELLO_done ();
118   GDS_NSE_done ();
119   if (GDS_block_context != NULL)
120   {
121     GNUNET_BLOCK_context_destroy (GDS_block_context);
122     GDS_block_context = NULL;
123   }
124   if (GDS_stats != NULL)
125   {
126     GNUNET_STATISTICS_destroy (GDS_stats, GNUNET_YES);
127     GDS_stats = NULL;
128   }
129   GNUNET_free_non_null (GDS_my_hello);
130   GDS_my_hello = NULL;
131 }
132
133
134 /**
135  * Process dht requests.
136  *
137  * @param cls closure
138  * @param server the initialized server
139  * @param c configuration to use
140  */
141 static void
142 run (void *cls, struct GNUNET_SERVER_Handle *server,
143      const struct GNUNET_CONFIGURATION_Handle *c)
144 {
145   GDS_cfg = c;
146   GDS_block_context = GNUNET_BLOCK_context_create (GDS_cfg);
147   GDS_stats = GNUNET_STATISTICS_create ("dht", GDS_cfg);
148   GDS_ROUTING_init ();
149   GDS_NSE_init ();
150   GDS_DATACACHE_init ();
151   GDS_HELLO_init ();
152   GDS_CLIENTS_init (server);
153   if (GNUNET_OK !=
154       GDS_NEIGHBOURS_init ())
155     {
156       shutdown_task (NULL, NULL);
157       return;
158     }
159   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
160                                 &shutdown_task, NULL);
161   GDS_transport_handle =
162     GNUNET_TRANSPORT_connect (GDS_cfg, NULL, NULL, NULL, NULL, NULL);
163   if (GDS_transport_handle == NULL)
164     {
165       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
166                   _("Failed to connect to transport service!\n"));
167       return;
168     }
169   ghh = GNUNET_TRANSPORT_get_hello (GDS_transport_handle, 
170                                     &process_hello, 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, char *const *argv)
183 {
184   int ret;
185   
186   ret = (GNUNET_OK ==
187          GNUNET_SERVICE_run (argc, argv, 
188                              "dht", 
189                              GNUNET_SERVICE_OPTION_NONE,
190                              &run, NULL)) ? 0 : 1;
191   GDS_CLIENTS_done ();
192   return ret;
193 }
194
195 /* end of gnunet-service-dht.c */