c16f94464e5fd9f54d363de1ef005b366cd0a81d
[oweals/gnunet.git] / src / fs / gnunet-service-fs_pr.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_pr.c
23  * @brief API to handle pending requests
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_load_lib.h"
28 #include "gnunet-service-fs.h"
29 #include "gnunet-service-fs_cp.h"
30 #include "gnunet-service-fs_indexing.h"
31 #include "gnunet-service-fs_pe.h"
32 #include "gnunet-service-fs_pr.h"
33
34 /**
35  * Maximum size of the datastore queue for P2P operations.  Needs to
36  * be large enough to queue MAX_QUEUE_PER_PEER operations for roughly
37  * the number of active (connected) peers.
38  */
39 #define MAX_DATASTORE_QUEUE (16 * MAX_QUEUE_PER_PEER)
40
41 /**
42  * Bandwidth value of a 0-priority content (must be fairly high
43  * compared to query since content is typically significantly larger
44  * -- and more valueable since it can take many queries to get one
45  * piece of content).
46  */
47 #define CONTENT_BANDWIDTH_VALUE 800
48
49 /**
50  * Hard limit on the number of results we may get from the datastore per query.
51  */
52 #define MAX_RESULTS (100 * 1024)
53
54 /**
55  * An active request.
56  */
57 struct GSF_PendingRequest
58 {
59   /**
60    * Public data for the request.
61    */
62   struct GSF_PendingRequestData public_data;
63
64   /**
65    * Function to call if we encounter a reply.
66    */
67   GSF_PendingRequestReplyHandler rh;
68
69   /**
70    * Closure for 'rh'
71    */
72   void *rh_cls;
73
74   /**
75    * Array of hash codes of replies we've already seen.
76    */
77   GNUNET_HashCode *replies_seen;
78
79   /**
80    * Bloomfilter masking replies we've already seen.
81    */
82   struct GNUNET_CONTAINER_BloomFilter *bf;
83
84   /**
85    * Entry for this pending request in the expiration heap, or NULL.
86    */
87   struct GNUNET_CONTAINER_HeapNode *hnode;
88
89   /**
90    * Datastore queue entry for this request (or NULL for none).
91    */
92   struct GNUNET_DATASTORE_QueueEntry *qe;
93
94   /**
95    * DHT request handle for this request (or NULL for none).
96    */
97   struct GNUNET_DHT_GetHandle *gh;
98
99   /**
100    * Function to call upon completion of the local get
101    * request, or NULL for none.
102    */
103   GSF_LocalLookupContinuation llc_cont;
104
105   /**
106    * Closure for llc_cont.
107    */
108   void *llc_cont_cls;
109
110   /**
111    * Last result from the local datastore lookup evaluation.
112    */
113   enum GNUNET_BLOCK_EvaluationResult local_result;
114
115   /**
116    * Identity of the peer that we should use for the 'sender'
117    * (recipient of the response) when forwarding (0 for none).
118    */
119   GNUNET_PEER_Id sender_pid;
120
121   /**
122    * Identity of the peer that we should never forward this query
123    * to since it originated this query (0 for none).
124    */
125   GNUNET_PEER_Id origin_pid;
126
127   /**
128    * Time we started the last datastore lookup.
129    */
130   struct GNUNET_TIME_Absolute qe_start;
131
132   /**
133    * Task that warns us if the local datastore lookup takes too long.
134    */
135   GNUNET_SCHEDULER_TaskIdentifier warn_task;
136
137   /**
138    * Current offset for querying our local datastore for results.
139    * Starts at a random value, incremented until we get the same
140    * UID again (detected using 'first_uid'), which is then used
141    * to termiante the iteration.
142    */
143   uint64_t local_result_offset;
144
145   /**
146    * Unique ID of the first result from the local datastore;
147    * used to detect wrap-around of the offset.
148    */
149   uint64_t first_uid;
150
151   /**
152    * Number of valid entries in the 'replies_seen' array.
153    */
154   unsigned int replies_seen_count;
155
156   /**
157    * Length of the 'replies_seen' array.
158    */
159   unsigned int replies_seen_size;
160
161   /**
162    * Mingle value we currently use for the bf.
163    */
164   uint32_t mingle;
165
166   /**
167    * Do we have a first UID yet?
168    */
169   unsigned int have_first_uid;
170
171 };
172
173
174 /**
175  * All pending requests, ordered by the query.  Entries
176  * are of type 'struct GSF_PendingRequest*'.
177  */
178 static struct GNUNET_CONTAINER_MultiHashMap *pr_map;
179
180
181 /**
182  * Datastore 'PUT' load tracking.
183  */
184 static struct GNUNET_LOAD_Value *datastore_put_load;
185
186
187 /**
188  * Are we allowed to migrate content to this peer.
189  */
190 static int active_to_migration;
191
192
193 /**
194  * Size of the datastore queue we assume for common requests.
195  * Determined based on the network quota.
196  */
197 static unsigned int datastore_queue_size;
198
199 /**
200  * Heap with the request that will expire next at the top.  Contains
201  * pointers of type "struct PendingRequest*"; these will *also* be
202  * aliased from the "requests_by_peer" data structures and the
203  * "requests_by_query" table.  Note that requests from our clients
204  * don't expire and are thus NOT in the "requests_by_expiration"
205  * (or the "requests_by_peer" tables).
206  */
207 static struct GNUNET_CONTAINER_Heap *requests_by_expiration_heap;
208
209
210 /**
211  * Maximum number of requests (from other peers, overall) that we're
212  * willing to have pending at any given point in time.  Can be changed
213  * via the configuration file (32k is just the default).
214  */
215 static unsigned long long max_pending_requests = (32 * 1024);
216
217
218
219 /**
220  * Recalculate our bloom filter for filtering replies.  This function
221  * will create a new bloom filter from scratch, so it should only be
222  * called if we have no bloomfilter at all (and hence can create a
223  * fresh one of minimal size without problems) OR if our peer is the
224  * initiator (in which case we may resize to larger than mimimum size).
225  *
226  * @param pr request for which the BF is to be recomputed
227  */
228 static void
229 refresh_bloomfilter (struct GSF_PendingRequest *pr)
230 {
231   if (pr->bf != NULL)
232     GNUNET_CONTAINER_bloomfilter_free (pr->bf);
233   pr->mingle =
234       GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, UINT32_MAX);
235   pr->bf =
236       GNUNET_BLOCK_construct_bloomfilter (pr->mingle, pr->replies_seen,
237                                           pr->replies_seen_count);
238 }
239
240
241 /**
242  * Create a new pending request.
243  *
244  * @param options request options
245  * @param type type of the block that is being requested
246  * @param query key for the lookup
247  * @param namespace namespace to lookup, NULL for no namespace
248  * @param target preferred target for the request, NULL for none
249  * @param bf_data raw data for bloom filter for known replies, can be NULL
250  * @param bf_size number of bytes in bf_data
251  * @param mingle mingle value for bf
252  * @param anonymity_level desired anonymity level
253  * @param priority maximum outgoing cummulative request priority to use
254  * @param ttl current time-to-live for the request
255  * @param sender_pid peer ID to use for the sender when forwarding, 0 for none
256  * @param origin_pid peer ID of origin of query (do not loop back)
257  * @param replies_seen hash codes of known local replies
258  * @param replies_seen_count size of the 'replies_seen' array
259  * @param rh handle to call when we get a reply
260  * @param rh_cls closure for rh
261  * @return handle for the new pending request
262  */
263 struct GSF_PendingRequest *
264 GSF_pending_request_create_ (enum GSF_PendingRequestOptions options,
265                              enum GNUNET_BLOCK_Type type,
266                              const GNUNET_HashCode * query,
267                              const GNUNET_HashCode * namespace,
268                              const struct GNUNET_PeerIdentity *target,
269                              const char *bf_data, size_t bf_size,
270                              uint32_t mingle, uint32_t anonymity_level,
271                              uint32_t priority, int32_t ttl,
272                              GNUNET_PEER_Id sender_pid,
273                              GNUNET_PEER_Id origin_pid,
274                              const GNUNET_HashCode * replies_seen,
275                              unsigned int replies_seen_count,
276                              GSF_PendingRequestReplyHandler rh, void *rh_cls)
277 {
278   struct GSF_PendingRequest *pr;
279   struct GSF_PendingRequest *dpr;
280
281 #if DEBUG_FS
282   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
283               "Creating request handle for `%s' of type %d\n",
284               GNUNET_h2s (query), type);
285 #endif
286   GNUNET_STATISTICS_update (GSF_stats,
287                             gettext_noop ("# Pending requests created"), 1,
288                             GNUNET_NO);
289   pr = GNUNET_malloc (sizeof (struct GSF_PendingRequest));
290   pr->local_result_offset =
291       GNUNET_CRYPTO_random_u64 (GNUNET_CRYPTO_QUALITY_WEAK, UINT64_MAX);
292   pr->public_data.query = *query;
293   if (GNUNET_BLOCK_TYPE_FS_SBLOCK == type)
294   {
295     GNUNET_assert (NULL != namespace);
296     pr->public_data.namespace = *namespace;
297   }
298   if (NULL != target)
299   {
300     pr->public_data.target = *target;
301     pr->public_data.has_target = GNUNET_YES;
302   }
303   pr->public_data.anonymity_level = anonymity_level;
304   pr->public_data.priority = priority;
305   pr->public_data.original_priority = priority;
306   pr->public_data.options = options;
307   pr->public_data.type = type;
308   pr->public_data.start_time = GNUNET_TIME_absolute_get ();
309   pr->sender_pid = sender_pid;
310   pr->origin_pid = origin_pid;
311   pr->rh = rh;
312   pr->rh_cls = rh_cls;
313   GNUNET_assert ((sender_pid != 0) || (0 == (options & GSF_PRO_FORWARD_ONLY)));
314   if (ttl >= 0)
315     pr->public_data.ttl =
316         GNUNET_TIME_relative_to_absolute (GNUNET_TIME_relative_multiply
317                                           (GNUNET_TIME_UNIT_SECONDS,
318                                            (uint32_t) ttl));
319   else
320     pr->public_data.ttl =
321         GNUNET_TIME_absolute_subtract (pr->public_data.start_time,
322                                        GNUNET_TIME_relative_multiply
323                                        (GNUNET_TIME_UNIT_SECONDS,
324                                         (uint32_t) (-ttl)));
325   if (replies_seen_count > 0)
326   {
327     pr->replies_seen_size = replies_seen_count;
328     pr->replies_seen =
329         GNUNET_malloc (sizeof (GNUNET_HashCode) * pr->replies_seen_size);
330     memcpy (pr->replies_seen, replies_seen,
331             replies_seen_count * sizeof (GNUNET_HashCode));
332     pr->replies_seen_count = replies_seen_count;
333   }
334   if (NULL != bf_data)
335   {
336     pr->bf =
337         GNUNET_CONTAINER_bloomfilter_init (bf_data, bf_size,
338                                            GNUNET_CONSTANTS_BLOOMFILTER_K);
339     pr->mingle = mingle;
340   }
341   else if ((replies_seen_count > 0) &&
342            (0 != (options & GSF_PRO_BLOOMFILTER_FULL_REFRESH)))
343   {
344     refresh_bloomfilter (pr);
345   }
346   GNUNET_CONTAINER_multihashmap_put (pr_map, query, pr,
347                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
348   if (0 == (options & GSF_PRO_REQUEST_NEVER_EXPIRES))
349   {
350     pr->hnode =
351         GNUNET_CONTAINER_heap_insert (requests_by_expiration_heap, pr,
352                                       pr->public_data.ttl.abs_value);
353     /* make sure we don't track too many requests */
354     while (GNUNET_CONTAINER_heap_get_size (requests_by_expiration_heap) >
355            max_pending_requests)
356     {
357       dpr = GNUNET_CONTAINER_heap_peek (requests_by_expiration_heap);
358       GNUNET_assert (dpr != NULL);
359       if (pr == dpr)
360         break;                  /* let the request live briefly... */
361       if (NULL != dpr->rh)
362         dpr->rh (dpr->rh_cls, GNUNET_BLOCK_EVALUATION_REQUEST_VALID, dpr,
363                  UINT32_MAX, GNUNET_TIME_UNIT_FOREVER_ABS, GNUNET_TIME_UNIT_FOREVER_ABS,
364                  GNUNET_BLOCK_TYPE_ANY, NULL, 0);
365       GSF_pending_request_cancel_ (dpr, GNUNET_YES);
366     }
367   }
368   GNUNET_STATISTICS_update (GSF_stats,
369                             gettext_noop ("# Pending requests active"), 1,
370                             GNUNET_NO);
371   return pr;
372 }
373
374 /**
375  * Obtain the public data associated with a pending request
376  *
377  * @param pr pending request
378  * @return associated public data
379  */
380 struct GSF_PendingRequestData *
381 GSF_pending_request_get_data_ (struct GSF_PendingRequest *pr)
382 {
383   return &pr->public_data;
384 }
385
386
387 /**
388  * Test if two pending requests are compatible (would generate
389  * the same query modulo filters and should thus be processed
390  * jointly).
391  *
392  * @param pra a pending request
393  * @param prb another pending request
394  * @return GNUNET_OK if the requests are compatible
395  */
396 int
397 GSF_pending_request_is_compatible_ (struct GSF_PendingRequest *pra,
398                                     struct GSF_PendingRequest *prb)
399 {
400   if ((pra->public_data.type != prb->public_data.type) ||
401       (0 !=
402        memcmp (&pra->public_data.query, &prb->public_data.query,
403                sizeof (GNUNET_HashCode))) ||
404       ((pra->public_data.type == GNUNET_BLOCK_TYPE_FS_SBLOCK) &&
405        (0 !=
406         memcmp (&pra->public_data.namespace, &prb->public_data.namespace,
407                 sizeof (GNUNET_HashCode)))))
408     return GNUNET_NO;
409   return GNUNET_OK;
410 }
411
412
413
414 /**
415  * Update a given pending request with additional replies
416  * that have been seen.
417  *
418  * @param pr request to update
419  * @param replies_seen hash codes of replies that we've seen
420  * @param replies_seen_count size of the replies_seen array
421  */
422 void
423 GSF_pending_request_update_ (struct GSF_PendingRequest *pr,
424                              const GNUNET_HashCode * replies_seen,
425                              unsigned int replies_seen_count)
426 {
427   unsigned int i;
428   GNUNET_HashCode mhash;
429
430   if (replies_seen_count + pr->replies_seen_count < pr->replies_seen_count)
431     return;                     /* integer overflow */
432   if (0 != (pr->public_data.options & GSF_PRO_BLOOMFILTER_FULL_REFRESH))
433   {
434     /* we're responsible for the BF, full refresh */
435     if (replies_seen_count + pr->replies_seen_count > pr->replies_seen_size)
436       GNUNET_array_grow (pr->replies_seen, pr->replies_seen_size,
437                          replies_seen_count + pr->replies_seen_count);
438     memcpy (&pr->replies_seen[pr->replies_seen_count], replies_seen,
439             sizeof (GNUNET_HashCode) * replies_seen_count);
440     pr->replies_seen_count += replies_seen_count;
441     refresh_bloomfilter (pr);
442   }
443   else
444   {
445     if (NULL == pr->bf)
446     {
447       /* we're not the initiator, but the initiator did not give us
448        * any bloom-filter, so we need to create one on-the-fly */
449       pr->mingle =
450           GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, UINT32_MAX);
451       pr->bf =
452           GNUNET_BLOCK_construct_bloomfilter (pr->mingle, replies_seen,
453                                               replies_seen_count);
454     }
455     else
456     {
457       for (i = 0; i < pr->replies_seen_count; i++)
458       {
459         GNUNET_BLOCK_mingle_hash (&replies_seen[i], pr->mingle, &mhash);
460         GNUNET_CONTAINER_bloomfilter_add (pr->bf, &mhash);
461       }
462     }
463   }
464 }
465
466
467 /**
468  * Generate the message corresponding to the given pending request for
469  * transmission to other peers (or at least determine its size).
470  *
471  * @param pr request to generate the message for
472  * @param buf_size number of bytes available in buf
473  * @param buf where to copy the message (can be NULL)
474  * @return number of bytes needed (if > buf_size) or used
475  */
476 size_t
477 GSF_pending_request_get_message_ (struct GSF_PendingRequest *pr,
478                                   size_t buf_size, void *buf)
479 {
480   char lbuf[GNUNET_SERVER_MAX_MESSAGE_SIZE];
481   struct GetMessage *gm;
482   GNUNET_HashCode *ext;
483   size_t msize;
484   unsigned int k;
485   uint32_t bm;
486   uint32_t prio;
487   size_t bf_size;
488   struct GNUNET_TIME_Absolute now;
489   int64_t ttl;
490   int do_route;
491
492 #if DEBUG_FS
493   if (buf_size > 0)
494     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
495                 "Building request message for `%s' of type %d\n",
496                 GNUNET_h2s (&pr->public_data.query), pr->public_data.type);
497 #endif
498   k = 0;
499   bm = 0;
500   do_route = (0 == (pr->public_data.options & GSF_PRO_FORWARD_ONLY));
501   if ((!do_route) && (pr->sender_pid == 0))
502   {
503     GNUNET_break (0);
504     do_route = GNUNET_YES;
505   }
506   if (!do_route)
507   {
508     bm |= GET_MESSAGE_BIT_RETURN_TO;
509     k++;
510   }
511   if (GNUNET_BLOCK_TYPE_FS_SBLOCK == pr->public_data.type)
512   {
513     bm |= GET_MESSAGE_BIT_SKS_NAMESPACE;
514     k++;
515   }
516   if (GNUNET_YES == pr->public_data.has_target)
517   {
518     bm |= GET_MESSAGE_BIT_TRANSMIT_TO;
519     k++;
520   }
521   bf_size = GNUNET_CONTAINER_bloomfilter_get_size (pr->bf);
522   msize = sizeof (struct GetMessage) + bf_size + k * sizeof (GNUNET_HashCode);
523   GNUNET_assert (msize < GNUNET_SERVER_MAX_MESSAGE_SIZE);
524   if (buf_size < msize)
525     return msize;
526   gm = (struct GetMessage *) lbuf;
527   gm->header.type = htons (GNUNET_MESSAGE_TYPE_FS_GET);
528   gm->header.size = htons (msize);
529   gm->type = htonl (pr->public_data.type);
530   if (do_route)
531     prio =
532         GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
533                                   pr->public_data.priority + 1);
534   else
535     prio = 0;
536   pr->public_data.priority -= prio;
537   gm->priority = htonl (prio);
538   now = GNUNET_TIME_absolute_get ();
539   ttl = (int64_t) (pr->public_data.ttl.abs_value - now.abs_value);
540   gm->ttl = htonl (ttl / 1000);
541   gm->filter_mutator = htonl (pr->mingle);
542   gm->hash_bitmap = htonl (bm);
543   gm->query = pr->public_data.query;
544   ext = (GNUNET_HashCode *) & gm[1];
545   k = 0;
546   if (!do_route)
547     GNUNET_PEER_resolve (pr->sender_pid,
548                          (struct GNUNET_PeerIdentity *) &ext[k++]);
549   if (GNUNET_BLOCK_TYPE_FS_SBLOCK == pr->public_data.type)
550     memcpy (&ext[k++], &pr->public_data.namespace, sizeof (GNUNET_HashCode));
551   if (GNUNET_YES == pr->public_data.has_target)
552     ext[k++] = pr->public_data.target.hashPubKey;
553   if (pr->bf != NULL)
554     GNUNET_assert (GNUNET_SYSERR !=
555                    GNUNET_CONTAINER_bloomfilter_get_raw_data (pr->bf,
556                                                               (char *) &ext[k],
557                                                               bf_size));
558   memcpy (buf, gm, msize);
559   return msize;
560 }
561
562
563 /**
564  * Iterator to free pending requests.
565  *
566  * @param cls closure, unused
567  * @param key current key code
568  * @param value value in the hash map (pending request)
569  * @return GNUNET_YES (we should continue to iterate)
570  */
571 static int
572 clean_request (void *cls, const GNUNET_HashCode * key, void *value)
573 {
574   struct GSF_PendingRequest *pr = value;
575   GSF_LocalLookupContinuation cont;
576
577 #if DEBUG_FS
578   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
579               "Cleaning up pending request for `%s'.\n", GNUNET_h2s (key));
580 #endif
581   if (NULL != (cont = pr->llc_cont))
582   {
583     pr->llc_cont = NULL;
584     cont (pr->llc_cont_cls, pr, pr->local_result);
585   }
586   GSF_plan_notify_request_done_ (pr);
587   GNUNET_free_non_null (pr->replies_seen);
588   if (NULL != pr->bf)
589   {
590     GNUNET_CONTAINER_bloomfilter_free (pr->bf);
591     pr->bf = NULL;
592   }
593   GNUNET_PEER_change_rc (pr->sender_pid, -1);
594   pr->sender_pid = 0;
595   GNUNET_PEER_change_rc (pr->origin_pid, -1);
596   pr->origin_pid = 0;
597   if (NULL != pr->hnode)
598   {
599     GNUNET_CONTAINER_heap_remove_node (pr->hnode);
600     pr->hnode = NULL;
601   }
602   if (NULL != pr->qe)
603   {
604     GNUNET_DATASTORE_cancel (pr->qe);
605     pr->qe = NULL;
606   }
607   if (NULL != pr->gh)
608   {
609     GNUNET_DHT_get_stop (pr->gh);
610     pr->gh = NULL;
611   }
612   if (GNUNET_SCHEDULER_NO_TASK != pr->warn_task)
613   {
614     GNUNET_SCHEDULER_cancel (pr->warn_task);
615     pr->warn_task = GNUNET_SCHEDULER_NO_TASK;
616   }
617   GNUNET_assert (GNUNET_OK ==
618                  GNUNET_CONTAINER_multihashmap_remove (pr_map,
619                                                        &pr->public_data.query,
620                                                        pr));
621   GNUNET_STATISTICS_update (GSF_stats,
622                             gettext_noop ("# Pending requests active"), -1,
623                             GNUNET_NO);
624   GNUNET_free (pr);
625   return GNUNET_YES;
626 }
627
628
629 /**
630  * Explicitly cancel a pending request.
631  *
632  * @param pr request to cancel
633  * @param full_cleanup fully purge the request
634  */
635 void
636 GSF_pending_request_cancel_ (struct GSF_PendingRequest *pr, int full_cleanup)
637 {
638   GSF_LocalLookupContinuation cont;
639
640   if (NULL == pr_map)
641     return;                     /* already cleaned up! */
642   if (GNUNET_YES != full_cleanup)
643   {
644     /* make request inactive (we're no longer interested in more results),
645      * but do NOT remove from our data-structures, we still need it there
646      * to prevent the request from looping */
647     pr->rh = NULL;
648     if (NULL != (cont = pr->llc_cont))
649     {
650       pr->llc_cont = NULL;
651       cont (pr->llc_cont_cls, pr, pr->local_result);
652     }
653     GSF_plan_notify_request_done_ (pr);
654     if (NULL != pr->qe)
655     {
656       GNUNET_DATASTORE_cancel (pr->qe);
657       pr->qe = NULL;
658     }
659     if (NULL != pr->gh)
660     {
661       GNUNET_DHT_get_stop (pr->gh);
662       pr->gh = NULL;
663     }
664     if (GNUNET_SCHEDULER_NO_TASK != pr->warn_task)
665     {
666       GNUNET_SCHEDULER_cancel (pr->warn_task);
667       pr->warn_task = GNUNET_SCHEDULER_NO_TASK;
668     }
669     return;
670   }
671   GNUNET_assert (GNUNET_YES ==
672                  clean_request (NULL, &pr->public_data.query, pr));
673 }
674
675
676 /**
677  * Iterate over all pending requests.
678  *
679  * @param it function to call for each request
680  * @param cls closure for it
681  */
682 void
683 GSF_iterate_pending_requests_ (GSF_PendingRequestIterator it, void *cls)
684 {
685   GNUNET_CONTAINER_multihashmap_iterate (pr_map,
686                                          (GNUNET_CONTAINER_HashMapIterator) it,
687                                          cls);
688 }
689
690
691
692
693 /**
694  * Closure for "process_reply" function.
695  */
696 struct ProcessReplyClosure
697 {
698   /**
699    * The data for the reply.
700    */
701   const void *data;
702
703   /**
704    * Who gave us this reply? NULL for local host (or DHT)
705    */
706   struct GSF_ConnectedPeer *sender;
707
708   /**
709    * When the reply expires.
710    */
711   struct GNUNET_TIME_Absolute expiration;
712
713   /**
714    * Size of data.
715    */
716   size_t size;
717
718   /**
719    * Type of the block.
720    */
721   enum GNUNET_BLOCK_Type type;
722
723   /**
724    * How much was this reply worth to us?
725    */
726   uint32_t priority;
727
728   /**
729    * Anonymity requirements for this reply.
730    */
731   uint32_t anonymity_level;
732
733   /**
734    * Evaluation result (returned).
735    */
736   enum GNUNET_BLOCK_EvaluationResult eval;
737
738   /**
739    * Did we find a matching request?
740    */
741   int request_found;
742 };
743
744
745 /**
746  * Update the performance data for the sender (if any) since
747  * the sender successfully answered one of our queries.
748  *
749  * @param prq information about the sender
750  * @param pr request that was satisfied
751  */
752 static void
753 update_request_performance_data (struct ProcessReplyClosure *prq,
754                                  struct GSF_PendingRequest *pr)
755 {
756   if (prq->sender == NULL)
757     return;
758   GSF_peer_update_performance_ (prq->sender, pr->public_data.start_time,
759                                 prq->priority);
760 }
761
762
763 /**
764  * We have received a reply; handle it!
765  *
766  * @param cls response (struct ProcessReplyClosure)
767  * @param key our query
768  * @param value value in the hash map (info about the query)
769  * @return GNUNET_YES (we should continue to iterate)
770  */
771 static int
772 process_reply (void *cls, const GNUNET_HashCode * key, void *value)
773 {
774   struct ProcessReplyClosure *prq = cls;
775   struct GSF_PendingRequest *pr = value;
776   GNUNET_HashCode chash;
777   struct GNUNET_TIME_Absolute last_transmission;
778
779   if (NULL == pr->rh)
780     return GNUNET_YES;
781 #if DEBUG_FS
782   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
783               "Matched result (type %u) for query `%s' with pending request\n",
784               (unsigned int) prq->type, GNUNET_h2s (key));
785 #endif
786   GNUNET_STATISTICS_update (GSF_stats,
787                             gettext_noop ("# replies received and matched"), 1,
788                             GNUNET_NO);
789   prq->eval =
790       GNUNET_BLOCK_evaluate (GSF_block_ctx, prq->type, key, &pr->bf, pr->mingle,
791                              &pr->public_data.namespace,
792                              (prq->type ==
793                               GNUNET_BLOCK_TYPE_FS_SBLOCK) ?
794                              sizeof (GNUNET_HashCode) : 0, prq->data,
795                              prq->size);
796   switch (prq->eval)
797   {
798   case GNUNET_BLOCK_EVALUATION_OK_MORE:
799     update_request_performance_data (prq, pr);
800     break;
801   case GNUNET_BLOCK_EVALUATION_OK_LAST:
802     /* short cut: stop processing early, no BF-update, etc. */
803     update_request_performance_data (prq, pr);
804     GNUNET_LOAD_update (GSF_rt_entry_lifetime,
805                         GNUNET_TIME_absolute_get_duration (pr->
806                                                            public_data.start_time).rel_value);
807     if (!GSF_request_plan_reference_get_last_transmission_ (pr->public_data.rpr_head, prq->sender, &last_transmission))
808       last_transmission.abs_value = GNUNET_TIME_UNIT_FOREVER_ABS.abs_value;
809     /* pass on to other peers / local clients */
810     pr->rh (pr->rh_cls, prq->eval, pr, prq->anonymity_level, prq->expiration,
811             last_transmission, prq->type, prq->data, prq->size);
812     return GNUNET_YES;
813   case GNUNET_BLOCK_EVALUATION_OK_DUPLICATE:
814     GNUNET_STATISTICS_update (GSF_stats,
815                               gettext_noop
816                               ("# duplicate replies discarded (bloomfilter)"),
817                               1, GNUNET_NO);
818 #if DEBUG_FS && 0
819     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
820                 "Duplicate response `%s', discarding.\n", GNUNET_h2s (&mhash));
821 #endif
822     return GNUNET_YES;          /* duplicate */
823   case GNUNET_BLOCK_EVALUATION_RESULT_INVALID:
824     return GNUNET_YES;          /* wrong namespace */
825   case GNUNET_BLOCK_EVALUATION_REQUEST_VALID:
826     GNUNET_break (0);
827     return GNUNET_YES;
828   case GNUNET_BLOCK_EVALUATION_REQUEST_INVALID:
829     GNUNET_break (0);
830     return GNUNET_YES;
831   case GNUNET_BLOCK_EVALUATION_TYPE_NOT_SUPPORTED:
832     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Unsupported block type %u\n"),
833                 prq->type);
834     return GNUNET_NO;
835   }
836   /* update bloomfilter */
837   GNUNET_CRYPTO_hash (prq->data, prq->size, &chash);
838   GSF_pending_request_update_ (pr, &chash, 1);
839   if (NULL == prq->sender)
840   {
841 #if DEBUG_FS
842     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
843                 "Found result for query `%s' in local datastore\n",
844                 GNUNET_h2s (key));
845 #endif
846     GNUNET_STATISTICS_update (GSF_stats,
847                               gettext_noop ("# results found locally"), 1,
848                               GNUNET_NO);
849   }
850   else
851   {
852     GSF_dht_lookup_ (pr);
853   }
854   prq->priority += pr->public_data.original_priority;
855   pr->public_data.priority = 0;
856   pr->public_data.original_priority = 0;
857   pr->public_data.results_found++;
858   prq->request_found = GNUNET_YES;
859   /* finally, pass on to other peer / local client */
860   if (!GSF_request_plan_reference_get_last_transmission_ (pr->public_data.rpr_head, prq->sender, &last_transmission))
861     last_transmission.abs_value = GNUNET_TIME_UNIT_FOREVER_ABS.abs_value;
862   pr->rh (pr->rh_cls, prq->eval, pr, prq->anonymity_level, prq->expiration,
863           last_transmission, prq->type, prq->data, prq->size);
864   return GNUNET_YES;
865 }
866
867
868 /**
869  * Context for the 'put_migration_continuation'.
870  */
871 struct PutMigrationContext
872 {
873
874   /**
875    * Start time for the operation.
876    */
877   struct GNUNET_TIME_Absolute start;
878
879   /**
880    * Request origin.
881    */
882   struct GNUNET_PeerIdentity origin;
883
884   /**
885    * GNUNET_YES if we had a matching request for this block,
886    * GNUNET_NO if not.
887    */
888   int requested;
889 };
890
891
892 /**
893  * Continuation called to notify client about result of the
894  * operation.
895  *
896  * @param cls closure
897  * @param success GNUNET_SYSERR on failure
898  * @param min_expiration minimum expiration time required for content to be stored
899  * @param msg NULL on success, otherwise an error message
900  */
901 static void
902 put_migration_continuation (void *cls, int success, 
903                             struct GNUNET_TIME_Absolute min_expiration,
904                             const char *msg)
905 {
906   struct PutMigrationContext *pmc = cls;
907   struct GSF_ConnectedPeer *cp;
908   struct GNUNET_TIME_Relative mig_pause;
909   struct GSF_PeerPerformanceData *ppd;
910
911   if (NULL != datastore_put_load)
912   {
913     if (GNUNET_SYSERR != success)
914     {
915       GNUNET_LOAD_update (datastore_put_load, 
916                           GNUNET_TIME_absolute_get_duration (pmc->start).rel_value);
917     }
918     else
919     {
920       /* on queue failure / timeout, increase the put load dramatically */
921       GNUNET_LOAD_update (datastore_put_load, 
922                           GNUNET_TIME_UNIT_MINUTES.rel_value);
923     }
924   }
925   cp = GSF_peer_get_ (&pmc->origin);
926   if (GNUNET_OK == success)
927   {
928     if (NULL != cp)
929     {
930       ppd = GSF_get_peer_performance_data_ (cp);
931       ppd->migration_delay.rel_value /= 2;
932     }
933     GNUNET_free (pmc);
934     return;
935   }
936   if ( (GNUNET_NO == success) && 
937        (GNUNET_NO == pmc->requested) && 
938        (NULL != cp) )
939   {
940     ppd = GSF_get_peer_performance_data_ (cp);
941     if (min_expiration.abs_value > 0)
942     {
943 #if DEBUG_FS
944       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
945                   "Asking to stop migration for %llu ms because datastore is full\n",
946                   (unsigned long long) GNUNET_TIME_absolute_get_remaining (min_expiration).rel_value);
947 #endif
948       GSF_block_peer_migration_ (cp, min_expiration);      
949     }
950     else
951     {
952       ppd->migration_delay = GNUNET_TIME_relative_max (GNUNET_TIME_UNIT_SECONDS,
953                                                        ppd->migration_delay);
954       ppd->migration_delay = GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_HOURS,
955                                                        ppd->migration_delay);
956       mig_pause.rel_value = GNUNET_CRYPTO_random_u64 (GNUNET_CRYPTO_QUALITY_WEAK,
957                                                       ppd->migration_delay.rel_value);
958       ppd->migration_delay = GNUNET_TIME_relative_multiply (ppd->migration_delay, 2);
959 #if DEBUG_FS
960       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
961                   "Replicated content already exists locally, asking to stop migration for %llu ms\n",
962                   (unsigned long long) mig_pause.rel_value);
963 #endif
964       GSF_block_peer_migration_ (cp, GNUNET_TIME_relative_to_absolute (mig_pause));
965     }
966   }
967   GNUNET_free (pmc);
968   GNUNET_STATISTICS_update (GSF_stats,
969                             gettext_noop ("# Datastore `PUT' failures"), 1,
970                             GNUNET_NO);
971 }
972
973
974 /**
975  * Test if the DATABASE (PUT) load on this peer is too high
976  * to even consider processing the query at
977  * all.
978  *
979  * @return GNUNET_YES if the load is too high to do anything (load high)
980  *         GNUNET_NO to process normally (load normal or low)
981  */
982 static int
983 test_put_load_too_high (uint32_t priority)
984 {
985   double ld;
986
987   if (NULL == datastore_put_load)
988     return GNUNET_NO;
989   if (GNUNET_LOAD_get_average (datastore_put_load) < 50)
990     return GNUNET_NO;           /* very fast */
991   ld = GNUNET_LOAD_get_load (datastore_put_load);
992   if (ld < 2.0 * (1 + priority))
993     return GNUNET_NO;
994   GNUNET_STATISTICS_update (GSF_stats,
995                             gettext_noop
996                             ("# storage requests dropped due to high load"), 1,
997                             GNUNET_NO);
998   return GNUNET_YES;
999 }
1000
1001
1002 /**
1003  * Iterator called on each result obtained for a DHT
1004  * operation that expects a reply
1005  *
1006  * @param cls closure
1007  * @param exp when will this value expire
1008  * @param key key of the result
1009  * @param get_path peers on reply path (or NULL if not recorded)
1010  * @param get_path_length number of entries in get_path
1011  * @param put_path peers on the PUT path (or NULL if not recorded)
1012  * @param put_path_length number of entries in get_path
1013  * @param type type of the result
1014  * @param size number of bytes in data
1015  * @param data pointer to the result data
1016  */
1017 static void
1018 handle_dht_reply (void *cls, struct GNUNET_TIME_Absolute exp,
1019                   const GNUNET_HashCode * key,
1020                   const struct GNUNET_PeerIdentity *get_path,
1021                   unsigned int get_path_length,
1022                   const struct GNUNET_PeerIdentity *put_path,
1023                   unsigned int put_path_length, enum GNUNET_BLOCK_Type type,
1024                   size_t size, const void *data)
1025 {
1026   struct GSF_PendingRequest *pr = cls;
1027   struct ProcessReplyClosure prq;
1028   struct PutMigrationContext *pmc;
1029
1030   GNUNET_STATISTICS_update (GSF_stats,
1031                             gettext_noop ("# Replies received from DHT"), 1,
1032                             GNUNET_NO);
1033   memset (&prq, 0, sizeof (prq));
1034   prq.data = data;
1035   prq.expiration = exp;
1036   /* do not allow migrated content to live longer than 1 year */
1037   prq.expiration = GNUNET_TIME_absolute_min (GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_YEARS),
1038                                              prq.expiration);
1039   prq.size = size;
1040   prq.type = type;
1041   process_reply (&prq, key, pr);
1042   if ((GNUNET_YES == active_to_migration) &&
1043       (GNUNET_NO == test_put_load_too_high (prq.priority)))
1044   {
1045 #if DEBUG_FS
1046     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1047                 "Replicating result for query `%s' with priority %u\n",
1048                 GNUNET_h2s (key), prq.priority);
1049 #endif
1050     pmc = GNUNET_malloc (sizeof (struct PutMigrationContext));
1051     pmc->start = GNUNET_TIME_absolute_get ();
1052     pmc->requested = GNUNET_YES;
1053     if (NULL ==
1054         GNUNET_DATASTORE_put (GSF_dsh, 0, key, size, data, type, prq.priority,
1055                               1 /* anonymity */ ,
1056                               0 /* replication */ ,
1057                               exp, 1 + prq.priority, MAX_DATASTORE_QUEUE,
1058                               GNUNET_CONSTANTS_SERVICE_TIMEOUT,
1059                               &put_migration_continuation, pmc))
1060     {
1061       put_migration_continuation (pmc, GNUNET_SYSERR, GNUNET_TIME_UNIT_ZERO_ABS, NULL);
1062     }
1063   }
1064 }
1065
1066
1067 /**
1068  * Consider looking up the data in the DHT (anonymity-level permitting).
1069  *
1070  * @param pr the pending request to process
1071  */
1072 void
1073 GSF_dht_lookup_ (struct GSF_PendingRequest *pr)
1074 {
1075   const void *xquery;
1076   size_t xquery_size;
1077   struct GNUNET_PeerIdentity pi;
1078   char buf[sizeof (GNUNET_HashCode) * 2];
1079
1080   if (0 != pr->public_data.anonymity_level)
1081     return;
1082   if (NULL != pr->gh)
1083   {
1084     GNUNET_DHT_get_stop (pr->gh);
1085     pr->gh = NULL;
1086   }
1087   xquery = NULL;
1088   xquery_size = 0;
1089   if (GNUNET_BLOCK_TYPE_FS_SBLOCK == pr->public_data.type)
1090   {
1091     xquery = buf;
1092     memcpy (buf, &pr->public_data.namespace, sizeof (GNUNET_HashCode));
1093     xquery_size = sizeof (GNUNET_HashCode);
1094   }
1095   if (0 != (pr->public_data.options & GSF_PRO_FORWARD_ONLY))
1096   {
1097     GNUNET_assert (0 != pr->sender_pid);
1098     GNUNET_PEER_resolve (pr->sender_pid, &pi);
1099     memcpy (&buf[xquery_size], &pi, sizeof (struct GNUNET_PeerIdentity));
1100     xquery_size += sizeof (struct GNUNET_PeerIdentity);
1101   }
1102   pr->gh =
1103       GNUNET_DHT_get_start (GSF_dht, GNUNET_TIME_UNIT_FOREVER_REL,
1104                             pr->public_data.type, &pr->public_data.query,
1105                             5 /* DEFAULT_GET_REPLICATION */ ,
1106                             GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
1107                             /* FIXME: can no longer pass pr->bf/pr->mingle... */
1108                             xquery, xquery_size, &handle_dht_reply, pr);
1109 }
1110
1111
1112 /**
1113  * Task that issues a warning if the datastore lookup takes too long.
1114  *
1115  * @param cls the 'struct GSF_PendingRequest'
1116  * @param tc task context
1117  */
1118 static void
1119 warn_delay_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1120 {
1121   struct GSF_PendingRequest *pr = cls;
1122
1123   GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1124               _("Datastore lookup already took %llu ms!\n"),
1125               (unsigned long long)
1126               GNUNET_TIME_absolute_get_duration (pr->qe_start).rel_value);
1127   pr->warn_task =
1128       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES, &warn_delay_task,
1129                                     pr);
1130 }
1131
1132
1133 /**
1134  * Task that issues a warning if the datastore lookup takes too long.
1135  *
1136  * @param cls the 'struct GSF_PendingRequest'
1137  * @param tc task context
1138  */
1139 static void
1140 odc_warn_delay_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1141 {
1142   struct GSF_PendingRequest *pr = cls;
1143
1144   GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1145               _("On-demand lookup already took %llu ms!\n"),
1146               (unsigned long long)
1147               GNUNET_TIME_absolute_get_duration (pr->qe_start).rel_value);
1148   pr->warn_task =
1149       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
1150                                     &odc_warn_delay_task, pr);
1151 }
1152
1153
1154 /**
1155  * We're processing (local) results for a search request
1156  * from another peer.  Pass applicable results to the
1157  * peer and if we are done either clean up (operation
1158  * complete) or forward to other peers (more results possible).
1159  *
1160  * @param cls our closure (struct PendingRequest)
1161  * @param key key for the content
1162  * @param size number of bytes in data
1163  * @param data content stored
1164  * @param type type of the content
1165  * @param priority priority of the content
1166  * @param anonymity anonymity-level for the content
1167  * @param expiration expiration time for the content
1168  * @param uid unique identifier for the datum;
1169  *        maybe 0 if no unique identifier is available
1170  */
1171 static void
1172 process_local_reply (void *cls, const GNUNET_HashCode * key, size_t size,
1173                      const void *data, enum GNUNET_BLOCK_Type type,
1174                      uint32_t priority, uint32_t anonymity,
1175                      struct GNUNET_TIME_Absolute expiration, uint64_t uid)
1176 {
1177   struct GSF_PendingRequest *pr = cls;
1178   GSF_LocalLookupContinuation cont;
1179   struct ProcessReplyClosure prq;
1180   GNUNET_HashCode query;
1181   unsigned int old_rf;
1182
1183   GNUNET_SCHEDULER_cancel (pr->warn_task);
1184   pr->warn_task = GNUNET_SCHEDULER_NO_TASK;
1185   if (NULL != pr->qe)
1186   {
1187     pr->qe = NULL;
1188     if (NULL == key)
1189     {
1190       GNUNET_STATISTICS_update (GSF_stats,
1191                                 gettext_noop
1192                                 ("# Datastore lookups concluded (no results)"),
1193                                 1, GNUNET_NO);
1194     }
1195     if (GNUNET_NO == pr->have_first_uid)
1196     {
1197       pr->first_uid = uid;
1198       pr->have_first_uid = 1;
1199     }
1200     else
1201     {
1202       if ((uid == pr->first_uid) && (key != NULL))
1203       {
1204         GNUNET_STATISTICS_update (GSF_stats,
1205                                   gettext_noop
1206                                   ("# Datastore lookups concluded (seen all)"),
1207                                   1, GNUNET_NO);
1208         key = NULL;             /* all replies seen! */
1209       }
1210       pr->have_first_uid++;
1211       if ((pr->have_first_uid > MAX_RESULTS) && (key != NULL))
1212       {
1213         GNUNET_STATISTICS_update (GSF_stats,
1214                                   gettext_noop
1215                                   ("# Datastore lookups aborted (more than MAX_RESULTS)"),
1216                                   1, GNUNET_NO);
1217         key = NULL;             /* all replies seen! */
1218       }
1219     }
1220   }
1221   if (NULL == key)
1222   {
1223 #if DEBUG_FS
1224     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
1225                 "No further local responses available.\n");
1226 #endif
1227     if ((pr->public_data.type == GNUNET_BLOCK_TYPE_FS_DBLOCK) ||
1228         (pr->public_data.type == GNUNET_BLOCK_TYPE_FS_IBLOCK))
1229       GNUNET_STATISTICS_update (GSF_stats,
1230                                 gettext_noop
1231                                 ("# requested DBLOCK or IBLOCK not found"), 1,
1232                                 GNUNET_NO);
1233     goto check_error_and_continue;
1234   }
1235 #if DEBUG_FS
1236   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1237               "Received reply for `%s' of type %d with UID %llu from datastore.\n",
1238               GNUNET_h2s (key), type, (unsigned long long) uid);
1239 #endif
1240   if (type == GNUNET_BLOCK_TYPE_FS_ONDEMAND)
1241   {
1242 #if DEBUG_FS
1243     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1244                 "Found ONDEMAND block, performing on-demand encoding\n");
1245 #endif
1246     GNUNET_STATISTICS_update (GSF_stats,
1247                               gettext_noop
1248                               ("# on-demand blocks matched requests"), 1,
1249                               GNUNET_NO);
1250     pr->qe_start = GNUNET_TIME_absolute_get ();
1251     pr->warn_task =
1252         GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
1253                                       &odc_warn_delay_task, pr);
1254     if (GNUNET_OK ==
1255         GNUNET_FS_handle_on_demand_block (key, size, data, type, priority,
1256                                           anonymity, expiration, uid,
1257                                           &process_local_reply, pr))
1258     {
1259       GNUNET_STATISTICS_update (GSF_stats,
1260                                 gettext_noop
1261                                 ("# on-demand lookups performed successfully"),
1262                                 1, GNUNET_NO);
1263       return;                   /* we're done */
1264     }
1265     GNUNET_STATISTICS_update (GSF_stats,
1266                               gettext_noop ("# on-demand lookups failed"), 1,
1267                               GNUNET_NO);
1268     GNUNET_SCHEDULER_cancel (pr->warn_task);
1269     pr->warn_task =
1270         GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
1271                                       &warn_delay_task, pr);
1272     pr->qe =
1273         GNUNET_DATASTORE_get_key (GSF_dsh, pr->local_result_offset - 1,
1274                                   &pr->public_data.query,
1275                                   pr->public_data.type ==
1276                                   GNUNET_BLOCK_TYPE_FS_DBLOCK ?
1277                                   GNUNET_BLOCK_TYPE_ANY : pr->public_data.type,
1278                                   (0 !=
1279                                    (GSF_PRO_PRIORITY_UNLIMITED &
1280                                     pr->public_data.options)) ? UINT_MAX : 1
1281                                   /* queue priority */ ,
1282                                   (0 !=
1283                                    (GSF_PRO_PRIORITY_UNLIMITED &
1284                                     pr->public_data.options)) ? UINT_MAX :
1285                                   datastore_queue_size
1286                                   /* max queue size */ ,
1287                                   GNUNET_TIME_UNIT_FOREVER_REL,
1288                                   &process_local_reply, pr);
1289     if (NULL != pr->qe)
1290       return;                   /* we're done */
1291     GNUNET_STATISTICS_update (GSF_stats,
1292                               gettext_noop
1293                               ("# Datastore lookups concluded (error queueing)"),
1294                               1, GNUNET_NO);
1295     goto check_error_and_continue;
1296   }
1297   old_rf = pr->public_data.results_found;
1298   memset (&prq, 0, sizeof (prq));
1299   prq.data = data;
1300   prq.expiration = expiration;
1301   prq.size = size;
1302   if (GNUNET_OK !=
1303       GNUNET_BLOCK_get_key (GSF_block_ctx, type, data, size, &query))
1304   {
1305     GNUNET_break (0);
1306     GNUNET_DATASTORE_remove (GSF_dsh, key, size, data, -1, -1,
1307                              GNUNET_TIME_UNIT_FOREVER_REL, NULL, NULL);
1308     pr->qe_start = GNUNET_TIME_absolute_get ();
1309     pr->warn_task =
1310         GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
1311                                       &warn_delay_task, pr);
1312     pr->qe =
1313         GNUNET_DATASTORE_get_key (GSF_dsh, pr->local_result_offset - 1,
1314                                   &pr->public_data.query,
1315                                   pr->public_data.type ==
1316                                   GNUNET_BLOCK_TYPE_FS_DBLOCK ?
1317                                   GNUNET_BLOCK_TYPE_ANY : pr->public_data.type,
1318                                   (0 !=
1319                                    (GSF_PRO_PRIORITY_UNLIMITED &
1320                                     pr->public_data.options)) ? UINT_MAX : 1
1321                                   /* queue priority */ ,
1322                                   (0 !=
1323                                    (GSF_PRO_PRIORITY_UNLIMITED &
1324                                     pr->public_data.options)) ? UINT_MAX :
1325                                   datastore_queue_size
1326                                   /* max queue size */ ,
1327                                   GNUNET_TIME_UNIT_FOREVER_REL,
1328                                   &process_local_reply, pr);
1329     if (pr->qe == NULL)
1330     {
1331       GNUNET_STATISTICS_update (GSF_stats,
1332                                 gettext_noop
1333                                 ("# Datastore lookups concluded (error queueing)"),
1334                                 1, GNUNET_NO);
1335       goto check_error_and_continue;
1336     }
1337     return;
1338   }
1339   prq.type = type;
1340   prq.priority = priority;
1341   prq.request_found = GNUNET_NO;
1342   prq.anonymity_level = anonymity;
1343   if ((old_rf == 0) && (pr->public_data.results_found == 0))
1344     GSF_update_datastore_delay_ (pr->public_data.start_time);
1345   process_reply (&prq, key, pr);
1346   pr->local_result = prq.eval;
1347   if (prq.eval == GNUNET_BLOCK_EVALUATION_OK_LAST)
1348   {
1349     GNUNET_STATISTICS_update (GSF_stats,
1350                               gettext_noop
1351                               ("# Datastore lookups concluded (found last result)"),
1352                               1, GNUNET_NO);
1353     goto check_error_and_continue;
1354   }
1355   if ((0 == (GSF_PRO_PRIORITY_UNLIMITED & pr->public_data.options)) &&
1356       ((GNUNET_YES == GSF_test_get_load_too_high_ (0)) ||
1357        (pr->public_data.results_found > 5 + 2 * pr->public_data.priority)))
1358   {
1359 #if DEBUG_FS > 2
1360     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Load too high, done with request\n");
1361 #endif
1362     GNUNET_STATISTICS_update (GSF_stats,
1363                               gettext_noop
1364                               ("# Datastore lookups concluded (load too high)"),
1365                               1, GNUNET_NO);
1366     goto check_error_and_continue;
1367   }
1368   pr->qe_start = GNUNET_TIME_absolute_get ();
1369   pr->warn_task =
1370       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES, &warn_delay_task,
1371                                     pr);
1372   pr->qe =
1373       GNUNET_DATASTORE_get_key (GSF_dsh, pr->local_result_offset++,
1374                                 &pr->public_data.query,
1375                                 pr->public_data.type ==
1376                                 GNUNET_BLOCK_TYPE_FS_DBLOCK ?
1377                                 GNUNET_BLOCK_TYPE_ANY : pr->public_data.type,
1378                                 (0 !=
1379                                  (GSF_PRO_PRIORITY_UNLIMITED & pr->
1380                                   public_data.options)) ? UINT_MAX : 1
1381                                 /* queue priority */ ,
1382                                 (0 !=
1383                                  (GSF_PRO_PRIORITY_UNLIMITED & pr->
1384                                   public_data.options)) ? UINT_MAX :
1385                                 datastore_queue_size
1386                                 /* max queue size */ ,
1387                                 GNUNET_TIME_UNIT_FOREVER_REL,
1388                                 &process_local_reply, pr);
1389   /* check if we successfully queued another datastore request;
1390    * if so, return, otherwise call our continuation (if we have
1391    * any) */
1392 check_error_and_continue:
1393   if (NULL != pr->qe)
1394     return;
1395   if (GNUNET_SCHEDULER_NO_TASK != pr->warn_task)
1396   {
1397     GNUNET_SCHEDULER_cancel (pr->warn_task);
1398     pr->warn_task = GNUNET_SCHEDULER_NO_TASK;
1399   }
1400   if (NULL == (cont = pr->llc_cont))
1401     return;                     /* no continuation */
1402   pr->llc_cont = NULL;
1403   cont (pr->llc_cont_cls, pr, pr->local_result);
1404 }
1405
1406
1407 /**
1408  * Is the given target a legitimate peer for forwarding the given request?
1409  *
1410  * @param pr request
1411  * @param target
1412  * @return GNUNET_YES if this request could be forwarded to the given peer
1413  */
1414 int
1415 GSF_pending_request_test_target_ (struct GSF_PendingRequest *pr,
1416                                   const struct GNUNET_PeerIdentity *target)
1417 {
1418   struct GNUNET_PeerIdentity pi;
1419
1420   if (0 == pr->origin_pid)
1421     return GNUNET_YES;
1422   GNUNET_PEER_resolve (pr->origin_pid, &pi);
1423   return (0 ==
1424           memcmp (&pi, target,
1425                   sizeof (struct GNUNET_PeerIdentity))) ? GNUNET_NO :
1426       GNUNET_YES;
1427 }
1428
1429
1430 /**
1431  * Look up the request in the local datastore.
1432  *
1433  * @param pr the pending request to process
1434  * @param cont function to call at the end
1435  * @param cont_cls closure for cont
1436  */
1437 void
1438 GSF_local_lookup_ (struct GSF_PendingRequest *pr,
1439                    GSF_LocalLookupContinuation cont, void *cont_cls)
1440 {
1441   GNUNET_assert (NULL == pr->gh);
1442   GNUNET_assert (NULL == pr->llc_cont);
1443   pr->llc_cont = cont;
1444   pr->llc_cont_cls = cont_cls;
1445   pr->qe_start = GNUNET_TIME_absolute_get ();
1446   pr->warn_task =
1447       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES, &warn_delay_task,
1448                                     pr);
1449   GNUNET_STATISTICS_update (GSF_stats,
1450                             gettext_noop ("# Datastore lookups initiated"), 1,
1451                             GNUNET_NO);
1452   pr->qe =
1453       GNUNET_DATASTORE_get_key (GSF_dsh, pr->local_result_offset++,
1454                                 &pr->public_data.query,
1455                                 pr->public_data.type ==
1456                                 GNUNET_BLOCK_TYPE_FS_DBLOCK ?
1457                                 GNUNET_BLOCK_TYPE_ANY : pr->public_data.type,
1458                                 (0 !=
1459                                  (GSF_PRO_PRIORITY_UNLIMITED & pr->
1460                                   public_data.options)) ? UINT_MAX : 1
1461                                 /* queue priority */ ,
1462                                 (0 !=
1463                                  (GSF_PRO_PRIORITY_UNLIMITED & pr->
1464                                   public_data.options)) ? UINT_MAX :
1465                                 datastore_queue_size
1466                                 /* max queue size */ ,
1467                                 GNUNET_TIME_UNIT_FOREVER_REL,
1468                                 &process_local_reply, pr);
1469   if (NULL != pr->qe)
1470     return;
1471   GNUNET_STATISTICS_update (GSF_stats,
1472                             gettext_noop
1473                             ("# Datastore lookups concluded (error queueing)"),
1474                             1, GNUNET_NO);
1475   GNUNET_SCHEDULER_cancel (pr->warn_task);
1476   pr->warn_task = GNUNET_SCHEDULER_NO_TASK;
1477   pr->llc_cont = NULL;
1478   if (NULL != cont)
1479     cont (cont_cls, pr, pr->local_result);
1480 }
1481
1482
1483
1484 /**
1485  * Handle P2P "CONTENT" message.  Checks that the message is
1486  * well-formed and then checks if there are any pending requests for
1487  * this content and possibly passes it on (to local clients or other
1488  * peers).  Does NOT perform migration (content caching at this peer).
1489  *
1490  * @param cp the other peer involved (sender or receiver, NULL
1491  *        for loopback messages where we are both sender and receiver)
1492  * @param message the actual message
1493  * @return GNUNET_OK if the message was well-formed,
1494  *         GNUNET_SYSERR if the message was malformed (close connection,
1495  *         do not cache under any circumstances)
1496  */
1497 int
1498 GSF_handle_p2p_content_ (struct GSF_ConnectedPeer *cp,
1499                          const struct GNUNET_MessageHeader *message)
1500 {
1501   const struct PutMessage *put;
1502   uint16_t msize;
1503   size_t dsize;
1504   enum GNUNET_BLOCK_Type type;
1505   struct GNUNET_TIME_Absolute expiration;
1506   GNUNET_HashCode query;
1507   struct ProcessReplyClosure prq;
1508   struct GNUNET_TIME_Relative block_time;
1509   double putl;
1510   struct PutMigrationContext *pmc;
1511
1512   msize = ntohs (message->size);
1513   if (msize < sizeof (struct PutMessage))
1514   {
1515     GNUNET_break_op (0);
1516     return GNUNET_SYSERR;
1517   }
1518   put = (const struct PutMessage *) message;
1519   dsize = msize - sizeof (struct PutMessage);
1520   type = ntohl (put->type);
1521   expiration = GNUNET_TIME_absolute_ntoh (put->expiration);
1522   /* do not allow migrated content to live longer than 1 year */
1523   expiration = GNUNET_TIME_absolute_min (GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_YEARS),
1524                                          expiration);
1525   if (type == GNUNET_BLOCK_TYPE_FS_ONDEMAND)
1526     return GNUNET_SYSERR;
1527   if (GNUNET_OK !=
1528       GNUNET_BLOCK_get_key (GSF_block_ctx, type, &put[1], dsize, &query))
1529   {
1530     GNUNET_break_op (0);
1531     return GNUNET_SYSERR;
1532   }
1533   GNUNET_STATISTICS_update (GSF_stats,
1534                             gettext_noop ("# GAP PUT messages received"), 1,
1535                             GNUNET_NO);
1536   /* now, lookup 'query' */
1537   prq.data = (const void *) &put[1];
1538   if (NULL != cp)
1539     prq.sender = cp;
1540   else
1541     prq.sender = NULL;
1542   prq.size = dsize;
1543   prq.type = type;
1544   prq.expiration = expiration;
1545   prq.priority = 0;
1546   prq.anonymity_level = UINT32_MAX;
1547   prq.request_found = GNUNET_NO;
1548   GNUNET_CONTAINER_multihashmap_get_multiple (pr_map, &query, &process_reply,
1549                                               &prq);
1550   if (NULL != cp)
1551   {
1552     GSF_connected_peer_change_preference_ (cp,
1553                                            CONTENT_BANDWIDTH_VALUE +
1554                                            1000 * prq.priority);
1555     GSF_get_peer_performance_data_ (cp)->trust += prq.priority;
1556   }
1557   if ((GNUNET_YES == active_to_migration) &&
1558       (GNUNET_NO == test_put_load_too_high (prq.priority)))
1559   {
1560 #if DEBUG_FS
1561     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1562                 "Replicating result for query `%s' with priority %u\n",
1563                 GNUNET_h2s (&query), prq.priority);
1564 #endif
1565     pmc = GNUNET_malloc (sizeof (struct PutMigrationContext));
1566     pmc->start = GNUNET_TIME_absolute_get ();
1567     pmc->requested = prq.request_found;
1568     GNUNET_assert (0 != GSF_get_peer_performance_data_ (cp)->pid);
1569     GNUNET_PEER_resolve (GSF_get_peer_performance_data_ (cp)->pid,
1570                          &pmc->origin);
1571     if (NULL ==
1572         GNUNET_DATASTORE_put (GSF_dsh, 0, &query, dsize, &put[1], type,
1573                               prq.priority, 1 /* anonymity */ ,
1574                               0 /* replication */ ,
1575                               expiration, 1 + prq.priority, MAX_DATASTORE_QUEUE,
1576                               GNUNET_CONSTANTS_SERVICE_TIMEOUT,
1577                               &put_migration_continuation, pmc))
1578     {
1579       put_migration_continuation (pmc, GNUNET_SYSERR, GNUNET_TIME_UNIT_ZERO_ABS, NULL);
1580     }
1581   }
1582   else
1583   {
1584 #if DEBUG_FS
1585     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1586                 "Choosing not to keep content `%s' (%d/%d)\n",
1587                 GNUNET_h2s (&query), active_to_migration,
1588                 test_put_load_too_high (prq.priority));
1589 #endif
1590   }
1591   putl = GNUNET_LOAD_get_load (datastore_put_load);
1592   if ((NULL != (cp = prq.sender)) && (GNUNET_NO == prq.request_found) &&
1593       ((GNUNET_YES != active_to_migration) ||
1594        (putl > 2.5 * (1 + prq.priority))))
1595   {
1596     if (GNUNET_YES != active_to_migration)
1597       putl = 1.0 + GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, 5);
1598     block_time =
1599         GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
1600                                        5000 +
1601                                        GNUNET_CRYPTO_random_u32
1602                                        (GNUNET_CRYPTO_QUALITY_WEAK,
1603                                         (unsigned int) (60000 * putl * putl)));
1604 #if DEBUG_FS
1605     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
1606                 "Asking to stop migration for %llu ms because of load %f and events %d/%d\n",
1607                 (unsigned long long) block_time.rel_value,
1608                 putl,
1609                 active_to_migration,
1610                 (GNUNET_NO == prq.request_found));
1611 #endif
1612     GSF_block_peer_migration_ (cp, GNUNET_TIME_relative_to_absolute (block_time));
1613   }
1614   return GNUNET_OK;
1615 }
1616
1617
1618 /**
1619  * Setup the subsystem.
1620  */
1621 void
1622 GSF_pending_request_init_ ()
1623 {
1624   unsigned long long bps;
1625
1626   if (GNUNET_OK !=
1627       GNUNET_CONFIGURATION_get_value_number (GSF_cfg, "fs",
1628                                              "MAX_PENDING_REQUESTS",
1629                                              &max_pending_requests))
1630   {
1631     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1632                 _
1633                 ("Configuration fails to specify `%s', assuming default value."),
1634                 "MAX_PENDING_REQUESTS");
1635   }
1636   if (GNUNET_OK !=
1637       GNUNET_CONFIGURATION_get_value_size (GSF_cfg, "ats", "WAN_QUOTA_OUT",
1638                                            &bps))
1639   {
1640     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1641                 _
1642                 ("Configuration fails to specify `%s', assuming default value."),
1643                 "WAN_QUOTA_OUT");
1644     bps = 65536;
1645   }
1646   /* queue size should be #queries we can have pending and satisfy within
1647    * a carry interval: */
1648   datastore_queue_size =
1649       bps * GNUNET_CONSTANTS_MAX_BANDWIDTH_CARRY_S / DBLOCK_SIZE;
1650
1651   active_to_migration =
1652       GNUNET_CONFIGURATION_get_value_yesno (GSF_cfg, "FS", "CONTENT_CACHING");
1653   datastore_put_load = GNUNET_LOAD_value_init (DATASTORE_LOAD_AUTODECLINE);
1654   pr_map = GNUNET_CONTAINER_multihashmap_create (32 * 1024);
1655   requests_by_expiration_heap =
1656       GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN);
1657 }
1658
1659
1660 /**
1661  * Shutdown the subsystem.
1662  */
1663 void
1664 GSF_pending_request_done_ ()
1665 {
1666   GNUNET_CONTAINER_multihashmap_iterate (pr_map, &clean_request, NULL);
1667   GNUNET_CONTAINER_multihashmap_destroy (pr_map);
1668   pr_map = NULL;
1669   GNUNET_CONTAINER_heap_destroy (requests_by_expiration_heap);
1670   requests_by_expiration_heap = NULL;
1671   GNUNET_LOAD_value_free (datastore_put_load);
1672   datastore_put_load = NULL;
1673 }
1674
1675
1676 /* end of gnunet-service-fs_pr.c */