2 This file is part of GNUnet.
3 (C) 2008, 2009, 2010 Christian Grothoff (and other contributing authors)
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 2, or (at your
8 option) any later version.
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.
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.
22 * @file hostlist/hostlist-server.c
23 * @author Christian Grothoff
24 * @brief application to provide an integrated hostlist HTTP server
28 #include <microhttpd.h>
29 #include "hostlist-server.h"
30 #include "gnunet_hello_lib.h"
31 #include "gnunet_peerinfo_service.h"
33 #define DEBUG_HOSTLIST_SERVER GNUNET_NO
36 * How often should we recalculate our response to hostlist requests?
38 #define RESPONSE_UPDATE_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 5)
41 * Handle to the HTTP server as provided by libmicrohttpd for IPv6.
43 static struct MHD_Daemon *daemon_handle_v6;
46 * Handle to the HTTP server as provided by libmicrohttpd for IPv4.
48 static struct MHD_Daemon *daemon_handle_v4;
53 static const struct GNUNET_CONFIGURATION_Handle *cfg;
58 static struct GNUNET_SCHEDULER_Handle *sched;
61 * Our primary task for IPv4.
63 static GNUNET_SCHEDULER_TaskIdentifier hostlist_task_v4;
66 * Our primary task for IPv6.
68 static GNUNET_SCHEDULER_TaskIdentifier hostlist_task_v6;
71 * Task that updates our HTTP response.
73 static GNUNET_SCHEDULER_TaskIdentifier response_task;
76 * Our canonical response.
78 static struct MHD_Response *response;
81 * NULL if we are not currenlty iterating over peer information.
83 static struct GNUNET_PEERINFO_IteratorContext *pitr;
86 * Context for host processor.
97 * Task that will produce a new response object.
100 update_response (void *cls,
101 const struct GNUNET_SCHEDULER_TaskContext *tc);
105 * Function that assembles our response.
108 finish_response (struct HostSet *results)
110 struct GNUNET_TIME_Relative freq;
112 if (response != NULL)
113 MHD_destroy_response (response);
114 #if DEBUG_HOSTLIST_SERVER
115 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
116 "Creating hostlist response with %u bytes\n",
117 (unsigned int) results->size);
119 response = MHD_create_response_from_data (results->size,
120 results->data, MHD_YES, MHD_NO);
121 if ( (daemon_handle_v4 != NULL) ||
122 (daemon_handle_v6 != NULL) )
124 freq = RESPONSE_UPDATE_FREQUENCY;
125 if (results->size == 0)
126 freq = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, 50);
127 /* schedule next update of the response */
128 response_task = GNUNET_SCHEDULER_add_delayed (sched,
135 /* already past shutdown */
136 MHD_destroy_response (response);
139 GNUNET_free (results);
144 * Callback that processes each of the known HELLOs for the
145 * hostlist response construction.
148 host_processor (void *cls,
149 const struct GNUNET_PeerIdentity * peer,
150 const struct GNUNET_HELLO_Message *hello,
153 struct HostSet *results = cls;
160 finish_response (results);
164 s = GNUNET_HELLO_size(hello);
165 #if DEBUG_HOSTLIST_SERVER
166 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
167 "Received %u bytes of `%s' from peer `%s' for hostlist.\n",
172 if (old + s >= GNUNET_MAX_MALLOC_CHECKED)
173 return; /* too large, skip! */
174 GNUNET_array_grow (results->data,
177 memcpy (&results->data[old], hello, s);
182 * Task that will produce a new response object.
185 update_response (void *cls,
186 const struct GNUNET_SCHEDULER_TaskContext *tc)
188 struct HostSet *results;
190 response_task = GNUNET_SCHEDULER_NO_TASK;
191 results = GNUNET_malloc(sizeof(struct HostSet));
192 pitr = GNUNET_PEERINFO_iterate (cfg, sched,
195 GNUNET_TIME_UNIT_MINUTES,
202 * Hostlist access policy (very permissive, allows everything).
205 accept_policy_callback (void *cls,
206 const struct sockaddr *addr, socklen_t addrlen)
208 if (NULL == response)
210 #if DEBUG_HOSTLIST_SERVER
211 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
212 "Received request for hostlist, but I am not yet ready; rejecting!\n");
216 return MHD_YES; /* accept all */
221 * Main request handler.
224 access_handler_callback (void *cls,
225 struct MHD_Connection *connection,
229 const char *upload_data,
230 size_t*upload_data_size, void **con_cls)
234 if (0 != strcmp (method, MHD_HTTP_METHOD_GET))
236 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
237 _("Refusing `%s' request to hostlist server\n"),
241 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
242 _("Received request for our hostlist\n"));
243 if (NULL == *con_cls)
246 #if DEBUG_HOSTLIST_SERVER
247 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
248 _("Sending 100 CONTINUE reply\n"));
250 return MHD_YES; /* send 100 continue */
252 if (*upload_data_size != 0)
254 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
255 _("Refusing `%s' request with %llu bytes of upload data\n"),
257 (unsigned long long) *upload_data_size);
258 return MHD_NO; /* do not support upload data */
260 if (response == NULL)
262 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
263 _("Could not handle hostlist request since I do not have a response yet\n"));
264 return MHD_NO; /* internal error, no response yet */
266 return MHD_queue_response (connection, MHD_HTTP_OK, response);
271 * Function that queries MHD's select sets and
272 * starts the task waiting for them.
274 static GNUNET_SCHEDULER_TaskIdentifier
275 prepare_daemon (struct MHD_Daemon *daemon_handle);
278 * Call MHD to process pending requests and then go back
279 * and schedule the next run.
282 run_daemon (void *cls,
283 const struct GNUNET_SCHEDULER_TaskContext *tc)
285 struct MHD_Daemon *daemon_handle = cls;
287 if (daemon_handle == daemon_handle_v4)
288 hostlist_task_v4 = GNUNET_SCHEDULER_NO_TASK;
290 hostlist_task_v6 = GNUNET_SCHEDULER_NO_TASK;
292 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
294 GNUNET_assert (MHD_YES == MHD_run (daemon_handle));
295 if (daemon_handle == daemon_handle_v4)
296 hostlist_task_v4 = prepare_daemon (daemon_handle);
298 hostlist_task_v6 = prepare_daemon (daemon_handle);
303 * Function that queries MHD's select sets and
304 * starts the task waiting for them.
306 static GNUNET_SCHEDULER_TaskIdentifier
307 prepare_daemon (struct MHD_Daemon *daemon_handle)
309 GNUNET_SCHEDULER_TaskIdentifier ret;
313 struct GNUNET_NETWORK_FDSet *wrs;
314 struct GNUNET_NETWORK_FDSet *wws;
315 struct GNUNET_NETWORK_FDSet *wes;
317 unsigned long long timeout;
319 struct GNUNET_TIME_Relative tv;
324 wrs = GNUNET_NETWORK_fdset_create ();
325 wes = GNUNET_NETWORK_fdset_create ();
326 wws = GNUNET_NETWORK_fdset_create ();
328 GNUNET_assert (MHD_YES ==
329 MHD_get_fdset (daemon_handle,
334 haveto = MHD_get_timeout (daemon_handle, &timeout);
335 if (haveto == MHD_YES)
336 tv.value = (uint64_t) timeout;
338 tv = GNUNET_TIME_UNIT_FOREVER_REL;
339 GNUNET_NETWORK_fdset_copy_native (wrs, &rs, max);
340 GNUNET_NETWORK_fdset_copy_native (wws, &ws, max);
341 GNUNET_NETWORK_fdset_copy_native (wes, &es, max);
342 ret = GNUNET_SCHEDULER_add_select (sched,
343 GNUNET_SCHEDULER_PRIORITY_HIGH,
344 GNUNET_SCHEDULER_NO_TASK,
350 GNUNET_NETWORK_fdset_destroy (wrs);
351 GNUNET_NETWORK_fdset_destroy (wws);
352 GNUNET_NETWORK_fdset_destroy (wes);
359 * Start server offering our hostlist.
361 * @return GNUNET_OK on success
364 GNUNET_HOSTLIST_server_start (const struct GNUNET_CONFIGURATION_Handle *c,
365 struct GNUNET_SCHEDULER_Handle *s,
366 struct GNUNET_STATISTICS_Handle *st)
368 unsigned long long port;
372 if (-1 == GNUNET_CONFIGURATION_get_value_number (cfg,
376 return GNUNET_SYSERR;
377 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
378 _("Hostlist service starts on port %llu\n"),
380 daemon_handle_v6 = MHD_start_daemon (MHD_USE_IPv6
381 #if DEBUG_HOSTLIST_SERVER
385 (unsigned short) port,
386 &accept_policy_callback,
388 &access_handler_callback,
390 MHD_OPTION_CONNECTION_LIMIT, (unsigned int) 16,
391 MHD_OPTION_PER_IP_CONNECTION_LIMIT, (unsigned int) 1,
392 MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
393 MHD_OPTION_CONNECTION_MEMORY_LIMIT, (size_t) (16 * 1024),
395 daemon_handle_v4 = MHD_start_daemon (MHD_NO_FLAG
396 #if DEBUG_HOSTLIST_SERVER
400 (unsigned short) port,
401 &accept_policy_callback,
403 &access_handler_callback,
405 MHD_OPTION_CONNECTION_LIMIT, (unsigned int) 16,
406 MHD_OPTION_PER_IP_CONNECTION_LIMIT, (unsigned int) 1,
407 MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
408 MHD_OPTION_CONNECTION_MEMORY_LIMIT, (size_t) (16 * 1024),
411 if ( (daemon_handle_v6 == NULL) &&
412 (daemon_handle_v4 == NULL) )
414 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
415 _("Could not start hostlist HTTP server on port %u\n"),
416 (unsigned short) port);
417 return GNUNET_SYSERR;
419 if (daemon_handle_v4 != NULL)
420 hostlist_task_v4 = prepare_daemon (daemon_handle_v4);
421 if (daemon_handle_v6 != NULL)
422 hostlist_task_v6 = prepare_daemon (daemon_handle_v6);
423 response_task = GNUNET_SCHEDULER_add_now (sched,
430 * Stop server offering our hostlist.
433 GNUNET_HOSTLIST_server_stop ()
435 #if DEBUG_HOSTLIST_SERVER
436 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
437 "Hostlist server shutdown\n");
439 if (GNUNET_SCHEDULER_NO_TASK != hostlist_task_v6)
441 GNUNET_SCHEDULER_cancel (sched, hostlist_task_v6);
442 hostlist_task_v6 = GNUNET_SCHEDULER_NO_TASK;
444 if (GNUNET_SCHEDULER_NO_TASK != hostlist_task_v4)
446 GNUNET_SCHEDULER_cancel (sched, hostlist_task_v4);
447 hostlist_task_v4 = GNUNET_SCHEDULER_NO_TASK;
451 GNUNET_PEERINFO_iterate_cancel (pitr);
454 if (GNUNET_SCHEDULER_NO_TASK != response_task)
456 GNUNET_SCHEDULER_cancel (sched, response_task);
457 response_task = GNUNET_SCHEDULER_NO_TASK;
459 if (NULL != daemon_handle_v4)
461 MHD_stop_daemon (daemon_handle_v4);
462 daemon_handle_v4 = NULL;
464 if (NULL != daemon_handle_v6)
466 MHD_stop_daemon (daemon_handle_v6);
467 daemon_handle_v6 = NULL;
469 if (response != NULL)
471 MHD_destroy_response (response);
476 /* end of hostlist-server.c */