prevent loopback routing, add some statistics, faster re-transmit
[oweals/gnunet.git] / src / fs / gnunet-service-fs.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 fs/gnunet-service-fs.c
23  * @brief gnunet anonymity protocol implementation
24  * @author Christian Grothoff
25  *
26  * To use:
27  * - consider re-issue GSF_dht_lookup_ after non-DHT reply received
28  * - implement 'SUPPORT_DELAYS'
29  *
30  */
31 #include "platform.h"
32 #include <float.h>
33 #include "gnunet_constants.h"
34 #include "gnunet_core_service.h"
35 #include "gnunet_dht_service.h"
36 #include "gnunet_datastore_service.h"
37 #include "gnunet_load_lib.h"
38 #include "gnunet_peer_lib.h"
39 #include "gnunet_protocols.h"
40 #include "gnunet_signatures.h"
41 #include "gnunet_statistics_service.h"
42 #include "gnunet_transport_service.h"
43 #include "gnunet_util_lib.h"
44 #include "gnunet-service-fs_cp.h"
45 #include "gnunet-service-fs_indexing.h"
46 #include "gnunet-service-fs_lc.h"
47 #include "gnunet-service-fs_pe.h"
48 #include "gnunet-service-fs_pr.h"
49 #include "gnunet-service-fs_push.h"
50 #include "gnunet-service-fs_put.h"
51 #include "fs.h"
52
53 /**
54  * Size for the hash map for DHT requests from the FS
55  * service.  Should be about the number of concurrent
56  * DHT requests we plan to make.
57  */
58 #define FS_DHT_HT_SIZE 1024
59
60
61 /**
62  * How quickly do we age cover traffic?  At the given
63  * time interval, remaining cover traffic counters are
64  * decremented by 1/16th.
65  */
66 #define COVER_AGE_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
67
68
69 /* ****************************** globals ****************************** */
70
71 /**
72  * Our connection to the datastore.
73  */
74 struct GNUNET_DATASTORE_Handle *GSF_dsh;
75
76 /**
77  * Our configuration.
78  */
79 const struct GNUNET_CONFIGURATION_Handle *GSF_cfg;
80
81 /**
82  * Handle for reporting statistics.
83  */
84 struct GNUNET_STATISTICS_Handle *GSF_stats;
85
86 /**
87  * Handle for DHT operations.
88  */
89 struct GNUNET_DHT_Handle *GSF_dht;
90
91 /**
92  * How long do requests typically stay in the routing table?
93  */
94 struct GNUNET_LOAD_Value *GSF_rt_entry_lifetime;
95
96 /**
97  * Typical priorities we're seeing from other peers right now.  Since
98  * most priorities will be zero, this value is the weighted average of
99  * non-zero priorities seen "recently".  In order to ensure that new
100  * values do not dramatically change the ratio, values are first
101  * "capped" to a reasonable range (+N of the current value) and then
102  * averaged into the existing value by a ratio of 1:N.  Hence
103  * receiving the largest possible priority can still only raise our
104  * "current_priorities" by at most 1.
105  */
106 double GSF_current_priorities;
107
108 /**
109  * How many query messages have we received 'recently' that
110  * have not yet been claimed as cover traffic?
111  */
112 unsigned int GSF_cover_query_count;
113
114 /**
115  * How many content messages have we received 'recently' that
116  * have not yet been claimed as cover traffic?
117  */
118 unsigned int GSF_cover_content_count;
119
120 /**
121  * Our block context.
122  */
123 struct GNUNET_BLOCK_Context *GSF_block_ctx;
124
125 /**
126  * Pointer to handle to the core service (points to NULL until we've
127  * connected to it).
128  */
129 struct GNUNET_CORE_Handle *GSF_core;
130
131 /**
132  * Are we introducing randomized delays for better anonymity?
133  */
134 int GSF_enable_randomized_delays;
135
136 /* ***************************** locals ******************************* */
137
138 /**
139  * Configuration for block library.
140  */
141 static struct GNUNET_CONFIGURATION_Handle *block_cfg;
142
143 /**
144  * ID of our task that we use to age the cover counters.
145  */
146 static GNUNET_SCHEDULER_TaskIdentifier cover_age_task;
147
148 /**
149  * Datastore 'GET' load tracking.
150  */
151 static struct GNUNET_LOAD_Value *datastore_get_load;
152
153 /**
154  * Identity of this peer.
155  */
156 static struct GNUNET_PeerIdentity my_id;
157
158 /**
159  * Task that periodically ages our cover traffic statistics.
160  *
161  * @param cls unused closure
162  * @param tc task context
163  */
164 static void
165 age_cover_counters (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
166 {
167   GSF_cover_content_count = (GSF_cover_content_count * 15) / 16;
168   GSF_cover_query_count = (GSF_cover_query_count * 15) / 16;
169   cover_age_task =
170       GNUNET_SCHEDULER_add_delayed (COVER_AGE_FREQUENCY, &age_cover_counters,
171                                     NULL);
172 }
173
174
175
176 /**
177  * We've just now completed a datastore request.  Update our
178  * datastore load calculations.
179  *
180  * @param start time when the datastore request was issued
181  */
182 void
183 GSF_update_datastore_delay_ (struct GNUNET_TIME_Absolute start)
184 {
185   struct GNUNET_TIME_Relative delay;
186
187   delay = GNUNET_TIME_absolute_get_duration (start);
188   GNUNET_LOAD_update (datastore_get_load, delay.rel_value);
189 }
190
191
192 /**
193  * Test if the DATABASE (GET) load on this peer is too high
194  * to even consider processing the query at
195  * all.
196  *
197  * @return GNUNET_YES if the load is too high to do anything (load high)
198  *         GNUNET_NO to process normally (load normal)
199  *         GNUNET_SYSERR to process for free (load low)
200  */
201 int
202 GSF_test_get_load_too_high_ (uint32_t priority)
203 {
204   double ld;
205
206   ld = GNUNET_LOAD_get_load (datastore_get_load);
207   if (ld < 1)
208     return GNUNET_SYSERR;
209   if (ld <= priority)
210     return GNUNET_NO;
211   return GNUNET_YES;
212 }
213
214
215 /**
216  * Handle P2P "PUT" message.
217  *
218  * @param cls closure, always NULL
219  * @param other the other peer involved (sender or receiver, NULL
220  *        for loopback messages where we are both sender and receiver)
221  * @param message the actual message
222  * @param atsi performance information
223  * @return GNUNET_OK to keep the connection open,
224  *         GNUNET_SYSERR to close it (signal serious error)
225  */
226 static int
227 handle_p2p_put (void *cls, const struct GNUNET_PeerIdentity *other,
228                 const struct GNUNET_MessageHeader *message,
229                 const struct GNUNET_ATS_Information *atsi)
230 {
231   struct GSF_ConnectedPeer *cp;
232
233   cp = GSF_peer_get_ (other);
234   if (NULL == cp)
235   {
236     GNUNET_break (0);
237     return GNUNET_OK;
238   }
239   GSF_cover_content_count++;
240   return GSF_handle_p2p_content_ (cp, message);
241 }
242
243
244 /**
245  * We have a new request, consider forwarding it to the given
246  * peer.
247  *
248  * @param cls the 'struct GSF_PendingRequest'
249  * @param peer identity of the peer
250  * @param cp handle to the connected peer record
251  * @param ppd peer performance data
252  */
253 static void
254 consider_request_for_forwarding (void *cls,
255                                  const struct GNUNET_PeerIdentity *peer,
256                                  struct GSF_ConnectedPeer *cp,
257                                  const struct GSF_PeerPerformanceData *ppd)
258 {
259   struct GSF_PendingRequest *pr = cls;
260
261   if (GNUNET_YES !=
262       GSF_pending_request_test_target_ (pr,
263                                         peer))
264   {
265     GNUNET_STATISTICS_update (GSF_stats,
266                               gettext_noop ("# Loopback routes suppressed"), 1,
267                               GNUNET_NO);
268     return;
269   }
270   GSF_plan_add_ (cp, pr);
271 }
272
273
274 /**
275  * Function to be called after we're done processing
276  * replies from the local lookup.  If the result status
277  * code indicates that there may be more replies, plan
278  * forwarding the request.
279  *
280  * @param cls closure (NULL)
281  * @param pr the pending request we were processing
282  * @param result final datastore lookup result
283  */
284 static void
285 consider_forwarding (void *cls, struct GSF_PendingRequest *pr,
286                      enum GNUNET_BLOCK_EvaluationResult result)
287 {
288   if (GNUNET_BLOCK_EVALUATION_OK_LAST == result)
289     return;                     /* we're done... */
290   GSF_iterate_connected_peers_ (&consider_request_for_forwarding, pr);
291 }
292
293
294 /**
295  * Handle P2P "GET" request.
296  *
297  * @param cls closure, always NULL
298  * @param other the other peer involved (sender or receiver, NULL
299  *        for loopback messages where we are both sender and receiver)
300  * @param message the actual message
301  * @param atsi performance information
302  * @return GNUNET_OK to keep the connection open,
303  *         GNUNET_SYSERR to close it (signal serious error)
304  */
305 static int
306 handle_p2p_get (void *cls, const struct GNUNET_PeerIdentity *other,
307                 const struct GNUNET_MessageHeader *message,
308                 const struct GNUNET_ATS_Information *atsi)
309 {
310   struct GSF_PendingRequest *pr;
311
312   pr = GSF_handle_p2p_query_ (other, message);
313   if (NULL == pr)
314     return GNUNET_SYSERR;
315   GSF_local_lookup_ (pr, &consider_forwarding, NULL);
316   return GNUNET_OK;
317 }
318
319
320 /**
321  * We're done with the local lookup, now consider
322  * P2P processing (depending on request options and
323  * result status).  Also signal that we can now
324  * receive more request information from the client.
325  *
326  * @param cls the client doing the request ('struct GNUNET_SERVER_Client')
327  * @param pr the pending request we were processing
328  * @param result final datastore lookup result
329  */
330 static void
331 start_p2p_processing (void *cls, struct GSF_PendingRequest *pr,
332                       enum GNUNET_BLOCK_EvaluationResult result)
333 {
334   struct GNUNET_SERVER_Client *client = cls;
335   struct GSF_PendingRequestData *prd;
336
337   prd = GSF_pending_request_get_data_ (pr);
338 #if DEBUG_FS_CLIENT
339   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
340               "Finished database lookup for local request `%s' with result %d\n",
341               GNUNET_h2s (&prd->query), result);
342 #endif
343   GNUNET_SERVER_receive_done (client, GNUNET_OK);
344   if (GNUNET_BLOCK_EVALUATION_OK_LAST == result)
345     return;                     /* we're done, 'pr' was already destroyed... */
346   if (0 != (GSF_PRO_LOCAL_ONLY & prd->options))
347   {
348     GSF_pending_request_cancel_ (pr, GNUNET_YES);
349     return;
350   }
351   GSF_dht_lookup_ (pr);
352   consider_forwarding (NULL, pr, result);
353 }
354
355
356 /**
357  * Handle START_SEARCH-message (search request from client).
358  *
359  * @param cls closure
360  * @param client identification of the client
361  * @param message the actual message
362  */
363 static void
364 handle_start_search (void *cls, struct GNUNET_SERVER_Client *client,
365                      const struct GNUNET_MessageHeader *message)
366 {
367   struct GSF_PendingRequest *pr;
368
369   pr = GSF_local_client_start_search_handler_ (client, message);
370   if (NULL == pr)
371   {
372     /* GNUNET_SERVER_receive_done was already called! */
373     return;
374   }
375   GSF_local_lookup_ (pr, &start_p2p_processing, client);
376 }
377
378
379 /**
380  * Task run during shutdown.
381  *
382  * @param cls unused
383  * @param tc unused
384  */
385 static void
386 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
387 {
388   if (NULL != GSF_core)
389   {
390     GNUNET_CORE_disconnect (GSF_core);
391     GSF_core = NULL;
392   }
393   GSF_put_done_ ();
394   GSF_push_done_ ();
395   GSF_pending_request_done_ ();
396   GSF_plan_done ();
397   GSF_connected_peer_done_ ();
398   GNUNET_DATASTORE_disconnect (GSF_dsh, GNUNET_NO);
399   GSF_dsh = NULL;
400   GNUNET_DHT_disconnect (GSF_dht);
401   GSF_dht = NULL;
402   GNUNET_BLOCK_context_destroy (GSF_block_ctx);
403   GSF_block_ctx = NULL;
404   GNUNET_CONFIGURATION_destroy (block_cfg);
405   block_cfg = NULL;
406   GNUNET_STATISTICS_destroy (GSF_stats, GNUNET_NO);
407   GSF_stats = NULL;
408   if (GNUNET_SCHEDULER_NO_TASK != cover_age_task)
409   {
410     GNUNET_SCHEDULER_cancel (cover_age_task);
411     cover_age_task = GNUNET_SCHEDULER_NO_TASK;
412   }
413   GNUNET_FS_indexing_done ();
414   GNUNET_LOAD_value_free (datastore_get_load);
415   datastore_get_load = NULL;
416   GNUNET_LOAD_value_free (GSF_rt_entry_lifetime);
417   GSF_rt_entry_lifetime = NULL;
418 }
419
420
421 /**
422  * Function called for each pending request whenever a new
423  * peer connects, giving us a chance to decide about submitting
424  * the existing request to the new peer.
425  *
426  * @param cls the 'struct GSF_ConnectedPeer' of the new peer
427  * @param key query for the request
428  * @param pr handle to the pending request
429  * @return GNUNET_YES to continue to iterate
430  */
431 static int
432 consider_peer_for_forwarding (void *cls, const GNUNET_HashCode * key,
433                               struct GSF_PendingRequest *pr)
434 {
435   struct GSF_ConnectedPeer *cp = cls;
436   struct GNUNET_PeerIdentity pid;
437
438   GSF_connected_peer_get_identity_ (cp, &pid);
439   if (GNUNET_YES !=
440       GSF_pending_request_test_target_ (pr,
441                                         &pid))
442   {
443     GNUNET_STATISTICS_update (GSF_stats,
444                               gettext_noop ("# Loopback routes suppressed"), 1,
445                               GNUNET_NO);
446     return GNUNET_YES;
447   }
448   GSF_plan_add_ (cp, pr);
449   return GNUNET_YES;
450 }
451
452
453 /**
454  * Method called whenever a given peer connects.
455  *
456  * @param cls closure, not used
457  * @param peer peer identity this notification is about
458  * @param atsi performance information
459  */
460 static void
461 peer_connect_handler (void *cls, const struct GNUNET_PeerIdentity *peer,
462                       const struct GNUNET_ATS_Information *atsi)
463 {
464   struct GSF_ConnectedPeer *cp;
465
466   if (0 == memcmp (&my_id, peer, sizeof (struct GNUNET_PeerIdentity)))
467     return;
468   cp = GSF_peer_connect_handler_ (peer, atsi);
469   if (NULL == cp)
470     return;
471   GSF_iterate_pending_requests_ (&consider_peer_for_forwarding, cp);
472 }
473
474
475 /**
476  * Function called after GNUNET_CORE_connect has succeeded
477  * (or failed for good).  Note that the private key of the
478  * peer is intentionally not exposed here; if you need it,
479  * your process should try to read the private key file
480  * directly (which should work if you are authorized...).
481  *
482  * @param cls closure
483  * @param server handle to the server, NULL if we failed
484  * @param my_identity ID of this peer, NULL if we failed
485  */
486 static void
487 peer_init_handler (void *cls, struct GNUNET_CORE_Handle *server,
488                    const struct GNUNET_PeerIdentity *my_identity)
489 {
490   my_id = *my_identity;
491 }
492
493
494 /**
495  * Process fs requests.
496  *
497  * @param server the initialized server
498  * @param c configuration to use
499  */
500 static int
501 main_init (struct GNUNET_SERVER_Handle *server,
502            const struct GNUNET_CONFIGURATION_Handle *c)
503 {
504   static const struct GNUNET_CORE_MessageHandler p2p_handlers[] = {
505     {&handle_p2p_get,
506      GNUNET_MESSAGE_TYPE_FS_GET, 0},
507     {&handle_p2p_put,
508      GNUNET_MESSAGE_TYPE_FS_PUT, 0},
509     {&GSF_handle_p2p_migration_stop_,
510      GNUNET_MESSAGE_TYPE_FS_MIGRATION_STOP,
511      sizeof (struct MigrationStopMessage)},
512     {NULL, 0, 0}
513   };
514   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
515     {&GNUNET_FS_handle_index_start, NULL,
516      GNUNET_MESSAGE_TYPE_FS_INDEX_START, 0},
517     {&GNUNET_FS_handle_index_list_get, NULL,
518      GNUNET_MESSAGE_TYPE_FS_INDEX_LIST_GET,
519      sizeof (struct GNUNET_MessageHeader)},
520     {&GNUNET_FS_handle_unindex, NULL, GNUNET_MESSAGE_TYPE_FS_UNINDEX,
521      sizeof (struct UnindexMessage)},
522     {&handle_start_search, NULL, GNUNET_MESSAGE_TYPE_FS_START_SEARCH,
523      0},
524     {NULL, NULL, 0, 0}
525   };
526
527   GSF_core = GNUNET_CORE_connect (GSF_cfg, 2,   /* larger? */
528                                   NULL, &peer_init_handler,
529                                   &peer_connect_handler,
530                                   &GSF_peer_disconnect_handler_,
531                                   NULL, GNUNET_NO,
532                                   NULL, GNUNET_NO, p2p_handlers);
533   if (NULL == GSF_core)
534   {
535     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
536                 _("Failed to connect to `%s' service.\n"), "core");
537     return GNUNET_SYSERR;
538   }
539   GNUNET_SERVER_disconnect_notify (server, &GSF_client_disconnect_handler_,
540                                    NULL);
541   GNUNET_SERVER_add_handlers (server, handlers);
542   cover_age_task =
543       GNUNET_SCHEDULER_add_delayed (COVER_AGE_FREQUENCY, &age_cover_counters,
544                                     NULL);
545   datastore_get_load = GNUNET_LOAD_value_init (DATASTORE_LOAD_AUTODECLINE);
546   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
547                                 NULL);
548   return GNUNET_OK;
549 }
550
551
552 /**
553  * Process fs requests.
554  *
555  * @param cls closure
556  * @param server the initialized server
557  * @param cfg configuration to use
558  */
559 static void
560 run (void *cls, struct GNUNET_SERVER_Handle *server,
561      const struct GNUNET_CONFIGURATION_Handle *cfg)
562 {
563   GSF_cfg = cfg;
564   GSF_enable_randomized_delays =
565       GNUNET_CONFIGURATION_get_value_yesno (cfg, "fs", "DELAY");
566   GSF_dsh = GNUNET_DATASTORE_connect (cfg);
567   if (NULL == GSF_dsh)
568   {
569     GNUNET_SCHEDULER_shutdown ();
570     return;
571   }
572   GSF_rt_entry_lifetime = GNUNET_LOAD_value_init (GNUNET_TIME_UNIT_FOREVER_REL);
573   GSF_stats = GNUNET_STATISTICS_create ("fs", cfg);
574   block_cfg = GNUNET_CONFIGURATION_create ();
575   GNUNET_CONFIGURATION_set_value_string (block_cfg, "block", "PLUGINS", "fs");
576   GSF_block_ctx = GNUNET_BLOCK_context_create (block_cfg);
577   GNUNET_assert (NULL != GSF_block_ctx);
578   GSF_dht = GNUNET_DHT_connect (cfg, FS_DHT_HT_SIZE);
579   GSF_plan_init ();
580   GSF_pending_request_init_ ();
581   GSF_connected_peer_init_ ();
582   GSF_push_init_ ();
583   GSF_put_init_ ();
584   if ((GNUNET_OK != GNUNET_FS_indexing_init (cfg, GSF_dsh)) ||
585       (GNUNET_OK != main_init (server, cfg)))
586   {
587     GNUNET_SCHEDULER_shutdown ();
588     shutdown_task (NULL, NULL);
589     return;
590   }
591 }
592
593
594 /**
595  * The main function for the fs service.
596  *
597  * @param argc number of arguments from the command line
598  * @param argv command line arguments
599  * @return 0 ok, 1 on error
600  */
601 int
602 main (int argc, char *const *argv)
603 {
604   return (GNUNET_OK ==
605           GNUNET_SERVICE_run (argc, argv, "fs", GNUNET_SERVICE_OPTION_NONE,
606                               &run, NULL)) ? 0 : 1;
607 }
608
609 /* end of gnunet-service-fs.c */