-whitespace
[oweals/gnunet.git] / src / fs / gnunet-service-fs_pe.c
1 /*
2      This file is part of GNUnet.
3      (C) 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_pe.c
23  * @brief API to manage query plan
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet-service-fs.h"
28 #include "gnunet-service-fs_cp.h"
29 #include "gnunet-service-fs_pe.h"
30 #include "gnunet-service-fs_pr.h"
31
32 /**
33  * Collect an instane number of statistics?  May cause excessive IPC.
34  */
35 #define INSANE_STATISTICS GNUNET_NO
36
37 /**
38  * List of GSF_PendingRequests this request plan
39  * participates with.
40  */
41 struct PendingRequestList;
42
43 /**
44  * Transmission plan for a peer.
45  */
46 struct PeerPlan;
47
48
49 /**
50  * M:N binding of plans to pending requests.
51  * Each pending request can be in a number of plans,
52  * and each plan can have a number of pending requests.
53  * Objects of this type indicate a mapping of a plan to
54  * a particular pending request.
55  *
56  * The corresponding head and tail of the "PE" MDLL
57  * are stored in a 'struct GSF_RequestPlan'. (We need
58  * to be able to lookup all pending requests corresponding
59  * to a given plan entry.)
60  *
61  * Similarly head and tail of the "PR" MDLL are stored
62  * with the 'struct GSF_PendingRequest'.  (We need
63  * to be able to lookup all plan entries corresponding
64  * to a given pending request.)
65  */
66 struct GSF_PendingRequestPlanBijection
67 {
68
69   /**
70    * This is a doubly-linked list.
71    */
72   struct GSF_PendingRequestPlanBijection *next_PR;
73
74   /**
75    * This is a doubly-linked list.
76    */
77   struct GSF_PendingRequestPlanBijection *prev_PR;
78
79   /**
80    * This is a doubly-linked list.
81    */
82   struct GSF_PendingRequestPlanBijection *next_PE;
83
84   /**
85    * This is a doubly-linked list.
86    */
87   struct GSF_PendingRequestPlanBijection *prev_PE;
88
89   /**
90    * Associated request plan.
91    */
92   struct GSF_RequestPlan *rp;
93
94   /**
95    * Associated pending request.
96    */
97   struct GSF_PendingRequest *pr;
98
99 };
100
101
102 /**
103  * Information we keep per request per peer.  This is a doubly-linked
104  * list (with head and tail in the 'struct GSF_PendingRequestData')
105  * with one entry in each heap of each 'struct PeerPlan'.  Each
106  * entry tracks information relevant for this request and this peer.
107  */
108 struct GSF_RequestPlan
109 {
110
111   /**
112    * This is a doubly-linked list.
113    */
114   struct GSF_RequestPlan *next;
115
116   /**
117    * This is a doubly-linked list.
118    */
119   struct GSF_RequestPlan *prev;
120
121   /**
122    * Heap node associated with this request and this peer.
123    */
124   struct GNUNET_CONTAINER_HeapNode *hn;
125
126   /**
127    * The transmission plan for a peer that this request is associated with.
128    */
129   struct PeerPlan *pp;
130
131   /**
132    * Head of list of associated pending requests.
133    */
134   struct GSF_PendingRequestPlanBijection *pe_head;
135
136   /**
137    * Tail of list of associated pending requests.
138    */
139   struct GSF_PendingRequestPlanBijection *pe_tail;
140
141   /**
142    * Earliest time we'd be happy to (re)transmit this request.
143    */
144   struct GNUNET_TIME_Absolute earliest_transmission;
145
146   /**
147    * When was the last time we transmitted this request to this peer? 0 for never.
148    */
149   struct GNUNET_TIME_Absolute last_transmission;
150
151   /**
152    * Current priority for this request for this target.
153    */
154   uint64_t priority;
155
156   /**
157    * How often did we transmit this request to this peer?
158    */
159   unsigned int transmission_counter;
160
161 };
162
163
164 /**
165  * Transmission plan for a peer.
166  */
167 struct PeerPlan
168 {
169   /**
170    * Heap with pending queries (struct GSF_RequestPlan), higher weights mean higher priority.
171    */
172   struct GNUNET_CONTAINER_Heap *priority_heap;
173
174   /**
175    * Heap with pending queries (struct GSF_RequestPlan), by transmission time, lowest first.
176    */
177   struct GNUNET_CONTAINER_Heap *delay_heap;
178
179   /**
180    * Map of queries to plan entries.  All entries in the priority_heap or delay_heap
181    * should be in the plan map.  Note that it IS possible for the plan map to have
182    * multiple entries for the same query.
183    */
184   struct GNUNET_CONTAINER_MultiHashMap *plan_map;
185
186   /**
187    * Current transmission request handle.
188    */
189   struct GSF_PeerTransmitHandle *pth;
190
191   /**
192    * Peer for which this is the plan.
193    */
194   struct GSF_ConnectedPeer *cp;
195
196   /**
197    * Current task for executing the plan.
198    */
199   GNUNET_SCHEDULER_TaskIdentifier task;
200 };
201
202
203 /**
204  * Hash map from peer identities to PeerPlans.
205  */
206 static struct GNUNET_CONTAINER_MultiPeerMap *plans;
207
208 /**
209  * Sum of all transmission counters (equals total delay for all plan entries).
210  */
211 static unsigned long long total_delay;
212
213 /**
214  * Number of plan entries.
215  */
216 static unsigned long long plan_count;
217
218
219 /**
220  * Return the query (key in the plan_map) for the given request plan.
221  * Note that this key may change as there can be multiple pending
222  * requests for the same key and we just return _one_ of them; this
223  * particular one might complete while another one might still be
224  * active, hence the lifetime of the returned hash code is NOT
225  * necessarily identical to that of the 'struct GSF_RequestPlan'
226  * given.
227  *
228  * @param rp a request plan
229  * @return the associated query
230  */
231 static const struct GNUNET_HashCode *
232 get_rp_key (struct GSF_RequestPlan *rp)
233 {
234   return &GSF_pending_request_get_data_ (rp->pe_head->pr)->query;
235 }
236
237
238 /**
239  * Figure out when and how to transmit to the given peer.
240  *
241  * @param cls the 'struct GSF_ConnectedPeer' for transmission
242  * @param tc scheduler context
243  */
244 static void
245 schedule_peer_transmission (void *cls,
246                             const struct GNUNET_SCHEDULER_TaskContext *tc);
247
248
249 /**
250  * Insert the given request plan into the heap with the appropriate weight.
251  *
252  * @param pp associated peer's plan
253  * @param rp request to plan
254  */
255 static void
256 plan (struct PeerPlan *pp, struct GSF_RequestPlan *rp)
257 {
258 #define N ((double)128.0)
259   /**
260    * Running average delay we currently impose.
261    */
262   static double avg_delay;
263
264   struct GSF_PendingRequestData *prd;
265   struct GNUNET_TIME_Relative delay;
266
267   GNUNET_assert (rp->pp == pp);
268   GNUNET_STATISTICS_set (GSF_stats,
269                          gettext_noop ("# average retransmission delay (ms)"),
270                          total_delay * 1000LL / plan_count, GNUNET_NO);
271   prd = GSF_pending_request_get_data_ (rp->pe_head->pr);
272
273   if (rp->transmission_counter < 8)
274     delay =
275         GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
276                                        rp->transmission_counter);
277   else if (rp->transmission_counter < 32)
278     delay =
279         GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
280                                        8 +
281                                        (1LL << (rp->transmission_counter - 8)));
282   else
283     delay =
284         GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
285                                        8 + (1LL << 24));
286   delay.rel_value_us =
287     GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
288                               delay.rel_value_us + 1);
289   /* Add 0.01 to avg_delay to avoid division-by-zero later */
290   avg_delay = (((avg_delay * (N - 1.0)) + delay.rel_value_us) / N) + 0.01;
291
292   /*
293    * For the priority, we need to consider a few basic rules:
294    * 1) if we just started requesting (delay is small), we should
295    * virtually always have a priority of zero.
296    * 2) for requests with average latency, our priority should match
297    * the average priority observed on the network
298    * 3) even the longest-running requests should not be WAY out of
299    * the observed average (thus we bound by a factor of 2)
300    * 4) we add +1 to the observed average priority to avoid everyone
301    * staying put at zero (2 * 0 = 0...).
302    *
303    * Using the specific calculation below, we get:
304    *
305    * delay = 0 => priority = 0;
306    * delay = avg delay => priority = running-average-observed-priority;
307    * delay >> avg_delay => priority = 2 * running-average-observed-priority;
308    *
309    * which satisfies all of the rules above.
310    *
311    * Note: M_PI_4 = PI/4 = arctan(1)
312    */
313   rp->priority =
314       round ((GSF_current_priorities +
315               1.0) * atan (delay.rel_value_us / avg_delay)) / M_PI_4;
316   /* Note: usage of 'round' and 'atan' requires -lm */
317
318   if (rp->transmission_counter != 0)
319     delay.rel_value_us += TTL_DECREMENT * 1000;
320   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
321               "Considering (re)transmission number %u in %s\n",
322               (unsigned int) rp->transmission_counter,
323               GNUNET_STRINGS_relative_time_to_string (delay,
324                                                       GNUNET_YES));
325   rp->earliest_transmission = GNUNET_TIME_relative_to_absolute (delay);
326   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
327               "Earliest (re)transmission for `%s' in %us\n",
328               GNUNET_h2s (&prd->query), rp->transmission_counter);
329   GNUNET_assert (rp->hn == NULL);
330   if (0 == GNUNET_TIME_absolute_get_remaining (rp->earliest_transmission).rel_value_us)
331     rp->hn = GNUNET_CONTAINER_heap_insert (pp->priority_heap, rp, rp->priority);
332   else
333     rp->hn =
334         GNUNET_CONTAINER_heap_insert (pp->delay_heap, rp,
335                                       rp->earliest_transmission.abs_value_us);
336   GNUNET_assert (GNUNET_YES ==
337                  GNUNET_CONTAINER_multihashmap_contains_value (pp->plan_map,
338                                                                get_rp_key (rp),
339                                                                rp));
340   if (GNUNET_SCHEDULER_NO_TASK != pp->task)
341     GNUNET_SCHEDULER_cancel (pp->task);
342   pp->task = GNUNET_SCHEDULER_add_now (&schedule_peer_transmission, pp);
343 #undef N
344 }
345
346
347 /**
348  * Get the pending request with the highest TTL from the given plan.
349  *
350  * @param rp plan to investigate
351  * @return pending request with highest TTL
352  */
353 struct GSF_PendingRequest *
354 get_latest (const struct GSF_RequestPlan *rp)
355 {
356   struct GSF_PendingRequest *ret;
357   struct GSF_PendingRequestPlanBijection *bi;
358
359   bi = rp->pe_head;
360   if (NULL == bi)
361     return NULL; /* should never happen */
362   ret = bi->pr;
363   bi = bi->next_PE;
364   while (NULL != bi)
365   {
366     if (GSF_pending_request_get_data_ (bi->pr)->ttl.abs_value_us >
367         GSF_pending_request_get_data_ (ret)->ttl.abs_value_us)
368       ret = bi->pr;
369     bi = bi->next_PE;
370   }
371   return ret;
372 }
373
374
375 /**
376  * Function called to get a message for transmission.
377  *
378  * @param cls closure
379  * @param buf_size number of bytes available in buf
380  * @param buf where to copy the message, NULL on error (peer disconnect)
381  * @return number of bytes copied to 'buf', can be 0 (without indicating an error)
382  */
383 static size_t
384 transmit_message_callback (void *cls, size_t buf_size, void *buf)
385 {
386   struct PeerPlan *pp = cls;
387   struct GSF_RequestPlan *rp;
388   size_t msize;
389
390   pp->pth = NULL;
391   if (NULL == buf)
392   {
393     /* failed, try again... */
394     if (GNUNET_SCHEDULER_NO_TASK != pp->task)
395       GNUNET_SCHEDULER_cancel (pp->task);
396
397     pp->task = GNUNET_SCHEDULER_add_now (&schedule_peer_transmission, pp);
398     GNUNET_STATISTICS_update (GSF_stats,
399                               gettext_noop
400                               ("# transmission failed (core has no bandwidth)"),
401                               1, GNUNET_NO);
402     return 0;
403   }
404   rp = GNUNET_CONTAINER_heap_peek (pp->priority_heap);
405   if (NULL == rp)
406   {
407     if (GNUNET_SCHEDULER_NO_TASK != pp->task)
408       GNUNET_SCHEDULER_cancel (pp->task);
409     pp->task = GNUNET_SCHEDULER_add_now (&schedule_peer_transmission, pp);
410     return 0;
411   }
412   msize = GSF_pending_request_get_message_ (get_latest (rp), buf_size, buf);
413   if (msize > buf_size)
414   {
415     if (GNUNET_SCHEDULER_NO_TASK != pp->task)
416       GNUNET_SCHEDULER_cancel (pp->task);
417     /* buffer to small (message changed), try again */
418     pp->task = GNUNET_SCHEDULER_add_now (&schedule_peer_transmission, pp);
419     return 0;
420   }
421   /* remove from root, add again elsewhere... */
422   GNUNET_assert (rp == GNUNET_CONTAINER_heap_remove_root (pp->priority_heap));
423   rp->hn = NULL;
424   rp->last_transmission = GNUNET_TIME_absolute_get ();
425   rp->transmission_counter++;
426   total_delay++;
427   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
428               "Executing plan %p executed %u times, planning retransmission\n",
429               rp, rp->transmission_counter);
430   plan (pp, rp);
431   GNUNET_STATISTICS_update (GSF_stats,
432                             gettext_noop
433                             ("# query messages sent to other peers"), 1,
434                             GNUNET_NO);
435   return msize;
436 }
437
438
439 /**
440  * Figure out when and how to transmit to the given peer.
441  *
442  * @param cls the 'struct PeerPlan'
443  * @param tc scheduler context
444  */
445 static void
446 schedule_peer_transmission (void *cls,
447                             const struct GNUNET_SCHEDULER_TaskContext *tc)
448 {
449   struct PeerPlan *pp = cls;
450   struct GSF_RequestPlan *rp;
451   size_t msize;
452   struct GNUNET_TIME_Relative delay;
453
454   pp->task = GNUNET_SCHEDULER_NO_TASK;
455   if (NULL != pp->pth)
456   {
457     GSF_peer_transmit_cancel_ (pp->pth);
458     pp->pth = NULL;
459   }
460   /* move ready requests to priority queue */
461   while ((NULL != (rp = GNUNET_CONTAINER_heap_peek (pp->delay_heap))) &&
462          (0 == GNUNET_TIME_absolute_get_remaining
463           (rp->earliest_transmission).rel_value_us))
464   {
465     GNUNET_assert (rp == GNUNET_CONTAINER_heap_remove_root (pp->delay_heap));
466     rp->hn = GNUNET_CONTAINER_heap_insert (pp->priority_heap, rp, rp->priority);
467   }
468   if (0 == GNUNET_CONTAINER_heap_get_size (pp->priority_heap))
469   {
470     /* priority heap (still) empty, check for delay... */
471     rp = GNUNET_CONTAINER_heap_peek (pp->delay_heap);
472     if (NULL == rp)
473     {
474       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "No active requests for plan %p.\n",
475                   pp);
476       return;                   /* both queues empty */
477     }
478     delay = GNUNET_TIME_absolute_get_remaining (rp->earliest_transmission);
479     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
480                 "Sleeping for %s before retrying requests on plan %p.\n",
481                 GNUNET_STRINGS_relative_time_to_string (delay,
482                                                         GNUNET_YES),
483                 pp);
484     GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# delay heap timeout (ms)"),
485                            delay.rel_value_us / 1000LL, GNUNET_NO);
486
487     pp->task =
488         GNUNET_SCHEDULER_add_delayed (delay, &schedule_peer_transmission, pp);
489     return;
490   }
491 #if INSANE_STATISTICS
492   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# query plans executed"),
493                             1, GNUNET_NO);
494 #endif
495   /* process from priority heap */
496   rp = GNUNET_CONTAINER_heap_peek (pp->priority_heap);
497   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Executing query plan %p\n", rp);
498   GNUNET_assert (NULL != rp);
499   msize = GSF_pending_request_get_message_ (get_latest (rp), 0, NULL);
500   pp->pth =
501       GSF_peer_transmit_ (pp->cp, GNUNET_YES, rp->priority,
502                           GNUNET_TIME_UNIT_FOREVER_REL, msize,
503                           &transmit_message_callback, pp);
504   GNUNET_assert (NULL != pp->pth);
505 }
506
507
508 /**
509  * Closure for 'merge_pr'.
510  */
511 struct MergeContext
512 {
513
514   struct GSF_PendingRequest *pr;
515
516   int merged;
517
518 };
519
520
521 /**
522  * Iterator that checks if an equivalent request is already
523  * present for this peer.
524  *
525  * @param cls closure
526  * @param query the query
527  * @param element request plan stored at the node
528  * @return GNUNET_YES if we should continue to iterate,
529  *         GNUNET_NO if not (merge success)
530  */
531 static int
532 merge_pr (void *cls, const struct GNUNET_HashCode * query, void *element)
533 {
534   struct MergeContext *mpr = cls;
535   struct GSF_RequestPlan *rp = element;
536   struct GSF_PendingRequestData *prd;
537   struct GSF_PendingRequestPlanBijection *bi;
538   struct GSF_PendingRequest *latest;
539
540   if (GNUNET_OK !=
541       GSF_pending_request_is_compatible_ (mpr->pr, rp->pe_head->pr))
542     return GNUNET_YES;
543   /* merge new request with existing request plan */
544   bi = GNUNET_new (struct GSF_PendingRequestPlanBijection);
545   bi->rp = rp;
546   bi->pr = mpr->pr;
547   prd = GSF_pending_request_get_data_ (mpr->pr);
548   GNUNET_CONTAINER_MDLL_insert (PR, prd->pr_head, prd->pr_tail, bi);
549   GNUNET_CONTAINER_MDLL_insert (PE, rp->pe_head, rp->pe_tail, bi);
550   mpr->merged = GNUNET_YES;
551 #if INSANE_STATISTICS
552   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# requests merged"), 1,
553                             GNUNET_NO);
554 #endif
555   latest = get_latest (rp);
556   if (GSF_pending_request_get_data_ (latest)->ttl.abs_value_us <
557       prd->ttl.abs_value_us)
558   {
559 #if INSANE_STATISTICS
560     GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# requests refreshed"),
561                               1, GNUNET_NO);
562 #endif
563     rp->transmission_counter = 0;       /* reset */
564   }
565   return GNUNET_NO;
566 }
567
568
569 /**
570  * Create a new query plan entry.
571  *
572  * @param cp peer with the entry
573  * @param pr request with the entry
574  */
575 void
576 GSF_plan_add_ (struct GSF_ConnectedPeer *cp, struct GSF_PendingRequest *pr)
577 {
578   const struct GNUNET_PeerIdentity *id;
579   struct PeerPlan *pp;
580   struct GSF_PendingRequestData *prd;
581   struct GSF_RequestPlan *rp;
582   struct GSF_PendingRequestPlanBijection *bi;
583   struct MergeContext mpc;
584
585   GNUNET_assert (NULL != cp);
586   id = GSF_connected_peer_get_identity2_ (cp);
587   pp = GNUNET_CONTAINER_multipeermap_get (plans, id);
588   if (NULL == pp)
589   {
590     pp = GNUNET_new (struct PeerPlan);
591     pp->plan_map = GNUNET_CONTAINER_multihashmap_create (128, GNUNET_NO);
592     pp->priority_heap =
593         GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MAX);
594     pp->delay_heap =
595         GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN);
596     pp->cp = cp;
597     GNUNET_CONTAINER_multipeermap_put (plans,
598                                        id, pp,
599                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
600   }
601   mpc.merged = GNUNET_NO;
602   mpc.pr = pr;
603   GNUNET_CONTAINER_multihashmap_get_multiple (pp->plan_map,
604                                               &GSF_pending_request_get_data_
605                                               (pr)->query, &merge_pr, &mpc); // 8 MB in 'merge_pr'
606   if (GNUNET_NO != mpc.merged)
607     return;
608   GNUNET_CONTAINER_multihashmap_get_multiple (pp->plan_map,
609                                               &GSF_pending_request_get_data_
610                                               (pr)->query, &merge_pr, &mpc);
611   if (GNUNET_NO != mpc.merged)
612     return;
613   plan_count++;
614   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# query plan entries"), 1,
615                             GNUNET_NO);
616   prd = GSF_pending_request_get_data_ (pr);
617   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
618               "Planning transmission of query `%s' to peer `%s'\n",
619               GNUNET_h2s (&prd->query), GNUNET_i2s (id));
620   rp = GNUNET_new (struct GSF_RequestPlan); // 8 MB
621   bi = GNUNET_new (struct GSF_PendingRequestPlanBijection);
622   bi->rp = rp;
623   bi->pr = pr;
624   GNUNET_CONTAINER_MDLL_insert (PR, prd->pr_head, prd->pr_tail, bi);
625   GNUNET_CONTAINER_MDLL_insert (PE, rp->pe_head, rp->pe_tail, bi);
626   rp->pp = pp;
627   GNUNET_assert (GNUNET_YES ==
628                  GNUNET_CONTAINER_multihashmap_put (pp->plan_map,
629                                                     get_rp_key (rp), rp,
630                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE)); // 8 MB
631   plan (pp, rp); // +5 MB (plan/heap-insert)
632 }
633
634
635 /**
636  * Notify the plan about a peer being no longer available;
637  * destroy all entries associated with this peer.
638  *
639  * @param cp connected peer
640  */
641 void
642 GSF_plan_notify_peer_disconnect_ (const struct GSF_ConnectedPeer *cp)
643 {
644   const struct GNUNET_PeerIdentity *id;
645   struct PeerPlan *pp;
646   struct GSF_RequestPlan *rp;
647   struct GSF_PendingRequestData *prd;
648   struct GSF_PendingRequestPlanBijection *bi;
649
650   id = GSF_connected_peer_get_identity2_ (cp);
651   pp = GNUNET_CONTAINER_multipeermap_get (plans, id);
652   if (NULL == pp)
653     return;                     /* nothing was ever planned for this peer */
654   GNUNET_assert (GNUNET_YES ==
655                  GNUNET_CONTAINER_multipeermap_remove (plans, id,
656                                                        pp));
657   if (NULL != pp->pth)
658   {
659     GSF_peer_transmit_cancel_ (pp->pth);
660     pp->pth = NULL;
661   }
662   if (GNUNET_SCHEDULER_NO_TASK != pp->task)
663   {
664     GNUNET_SCHEDULER_cancel (pp->task);
665     pp->task = GNUNET_SCHEDULER_NO_TASK;
666   }
667   while (NULL != (rp = GNUNET_CONTAINER_heap_remove_root (pp->priority_heap)))
668   {
669     GNUNET_break (GNUNET_YES ==
670                   GNUNET_CONTAINER_multihashmap_remove (pp->plan_map,
671                                                         get_rp_key (rp), rp));
672     while (NULL != (bi = rp->pe_head))
673     {
674       GNUNET_CONTAINER_MDLL_remove (PE, rp->pe_head, rp->pe_tail, bi);
675       prd = GSF_pending_request_get_data_ (bi->pr);
676       GNUNET_CONTAINER_MDLL_remove (PR, prd->pr_head, prd->pr_tail, bi);
677       GNUNET_free (bi);
678     }
679     plan_count--;
680     GNUNET_free (rp);
681   }
682   GNUNET_CONTAINER_heap_destroy (pp->priority_heap);
683   while (NULL != (rp = GNUNET_CONTAINER_heap_remove_root (pp->delay_heap)))
684   {
685     GNUNET_break (GNUNET_YES ==
686                   GNUNET_CONTAINER_multihashmap_remove (pp->plan_map,
687                                                         get_rp_key (rp), rp));
688     while (NULL != (bi = rp->pe_head))
689     {
690       prd = GSF_pending_request_get_data_ (bi->pr);
691       GNUNET_CONTAINER_MDLL_remove (PE, rp->pe_head, rp->pe_tail, bi);
692       GNUNET_CONTAINER_MDLL_remove (PR, prd->pr_head, prd->pr_tail, bi);
693       GNUNET_free (bi);
694     }
695     plan_count--;
696     GNUNET_free (rp);
697   }
698   GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# query plan entries"),
699                          plan_count, GNUNET_NO);
700   GNUNET_CONTAINER_heap_destroy (pp->delay_heap);
701   GNUNET_CONTAINER_multihashmap_destroy (pp->plan_map);
702   GNUNET_free (pp);
703 }
704
705
706 /**
707  * Get the last transmission attempt time for the request plan list
708  * referenced by 'pr_head', that was sent to 'sender'
709  *
710  * @param pr_head request plan reference list to check.
711  * @param sender the peer that we've sent the request to.
712  * @param result the timestamp to fill, set to "FOREVER" if never transmitted
713  * @return GNUNET_YES if 'result' was changed, GNUNET_NO otherwise.
714  */
715 int
716 GSF_request_plan_reference_get_last_transmission_ (
717     struct GSF_PendingRequestPlanBijection *pr_head, struct GSF_ConnectedPeer *sender,
718     struct GNUNET_TIME_Absolute *result)
719 {
720   struct GSF_PendingRequestPlanBijection *bi;
721
722   for (bi = pr_head; NULL != bi; bi = bi->next_PR)
723   {
724     if (bi->rp->pp->cp == sender)
725     {
726       if (0 == bi->rp->last_transmission.abs_value_us)
727         *result = GNUNET_TIME_UNIT_FOREVER_ABS;
728       else
729         *result = bi->rp->last_transmission;
730       return GNUNET_YES;
731     }
732   }
733   return GNUNET_NO;
734 }
735
736
737 /**
738  * Notify the plan about a request being done; destroy all entries
739  * associated with this request.
740  *
741  * @param pr request that is done
742  */
743 void
744 GSF_plan_notify_request_done_ (struct GSF_PendingRequest *pr)
745 {
746   struct GSF_RequestPlan *rp;
747   struct GSF_PendingRequestData *prd;
748   struct GSF_PendingRequestPlanBijection *bi;
749
750   prd = GSF_pending_request_get_data_ (pr);
751   while (NULL != (bi = prd->pr_head))
752   {
753     rp = bi->rp;
754     GNUNET_CONTAINER_MDLL_remove (PR, prd->pr_head, prd->pr_tail, bi);
755     GNUNET_CONTAINER_MDLL_remove (PE, rp->pe_head, rp->pe_tail, bi);
756     if (NULL == rp->pe_head)
757     {
758       GNUNET_CONTAINER_heap_remove_node (rp->hn);
759       plan_count--;
760       GNUNET_break (GNUNET_YES ==
761                     GNUNET_CONTAINER_multihashmap_remove (rp->pp->plan_map,
762                                                           &GSF_pending_request_get_data_
763                                                           (bi->pr)->query,
764                                                           rp));
765       GNUNET_free (rp);
766     }
767     GNUNET_free (bi);
768   }
769   GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# query plan entries"),
770                          plan_count, GNUNET_NO);
771 }
772
773
774 /**
775  * Initialize plan subsystem.
776  */
777 void
778 GSF_plan_init ()
779 {
780   plans = GNUNET_CONTAINER_multipeermap_create (256, GNUNET_YES);
781 }
782
783
784 /**
785  * Shutdown plan subsystem.
786  */
787 void
788 GSF_plan_done ()
789 {
790   GNUNET_assert (0 == GNUNET_CONTAINER_multipeermap_size (plans));
791   GNUNET_CONTAINER_multipeermap_destroy (plans);
792 }
793
794
795
796 /* end of gnunet-service-fs_pe.h */