-adding a few extra checks for #3615
[oweals/gnunet.git] / src / fs / gnunet-service-fs_cp.c
1 /*
2      This file is part of GNUnet.
3      Copyright (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  * @file fs/gnunet-service-fs_cp.c
22  * @brief API to handle 'connected peers'
23  * @author Christian Grothoff
24  */
25 #include "platform.h"
26 #include "gnunet_util_lib.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_pe.h"
31 #include "gnunet-service-fs_pr.h"
32 #include "gnunet-service-fs_push.h"
33 #include "gnunet_peerstore_service.h"
34
35
36 /**
37  * Ratio for moving average delay calculation.  The previous
38  * average goes in with a factor of (n-1) into the calculation.
39  * Must be > 0.
40  */
41 #define RUNAVG_DELAY_N 16
42
43 /**
44  * How often do we flush respect values to disk?
45  */
46 #define RESPECT_FLUSH_FREQ GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 5)
47
48 /**
49  * After how long do we discard a reply?
50  */
51 #define REPLY_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 2)
52
53 /**
54  * Collect an instane number of statistics?  May cause excessive IPC.
55  */
56 #define INSANE_STATISTICS GNUNET_NO
57
58
59 /**
60  * Handle to cancel a transmission request.
61  */
62 struct GSF_PeerTransmitHandle
63 {
64
65   /**
66    * Kept in a doubly-linked list.
67    */
68   struct GSF_PeerTransmitHandle *next;
69
70   /**
71    * Kept in a doubly-linked list.
72    */
73   struct GSF_PeerTransmitHandle *prev;
74
75   /**
76    * Time when this transmission request was issued.
77    */
78   struct GNUNET_TIME_Absolute transmission_request_start_time;
79
80   /**
81    * Timeout for this request.
82    */
83   struct GNUNET_TIME_Absolute timeout;
84
85   /**
86    * Task called on timeout, or 0 for none.
87    */
88   struct GNUNET_SCHEDULER_Task *timeout_task;
89
90   /**
91    * Function to call to get the actual message.
92    */
93   GSF_GetMessageCallback gmc;
94
95   /**
96    * Peer this request targets.
97    */
98   struct GSF_ConnectedPeer *cp;
99
100   /**
101    * Closure for @e gmc.
102    */
103   void *gmc_cls;
104
105   /**
106    * Size of the message to be transmitted.
107    */
108   size_t size;
109
110   /**
111    * #GNUNET_YES if this is a query, #GNUNET_NO for content.
112    */
113   int is_query;
114
115   /**
116    * Did we get a reservation already?
117    */
118   int was_reserved;
119
120   /**
121    * Priority of this request.
122    */
123   uint32_t priority;
124
125 };
126
127
128 /**
129  * Handle for an entry in our delay list.
130  */
131 struct GSF_DelayedHandle
132 {
133
134   /**
135    * Kept in a doubly-linked list.
136    */
137   struct GSF_DelayedHandle *next;
138
139   /**
140    * Kept in a doubly-linked list.
141    */
142   struct GSF_DelayedHandle *prev;
143
144   /**
145    * Peer this transmission belongs to.
146    */
147   struct GSF_ConnectedPeer *cp;
148
149   /**
150    * The PUT that was delayed.
151    */
152   struct PutMessage *pm;
153
154   /**
155    * Task for the delay.
156    */
157   struct GNUNET_SCHEDULER_Task *delay_task;
158
159   /**
160    * Size of the message.
161    */
162   size_t msize;
163
164 };
165
166
167 /**
168  * Information per peer and request.
169  */
170 struct PeerRequest
171 {
172
173   /**
174    * Handle to generic request.
175    */
176   struct GSF_PendingRequest *pr;
177
178   /**
179    * Handle to specific peer.
180    */
181   struct GSF_ConnectedPeer *cp;
182
183   /**
184    * Task for asynchronous stopping of this request.
185    */
186   struct GNUNET_SCHEDULER_Task *kill_task;
187
188 };
189
190
191 /**
192  * A connected peer.
193  */
194 struct GSF_ConnectedPeer
195 {
196
197   /**
198    * Performance data for this peer.
199    */
200   struct GSF_PeerPerformanceData ppd;
201
202   /**
203    * Time until when we blocked this peer from migrating
204    * data to us.
205    */
206   struct GNUNET_TIME_Absolute last_migration_block;
207
208   /**
209    * Task scheduled to revive migration to this peer.
210    */
211   struct GNUNET_SCHEDULER_Task *mig_revive_task;
212
213   /**
214    * Messages (replies, queries, content migration) we would like to
215    * send to this peer in the near future.  Sorted by priority, head.
216    */
217   struct GSF_PeerTransmitHandle *pth_head;
218
219   /**
220    * Messages (replies, queries, content migration) we would like to
221    * send to this peer in the near future.  Sorted by priority, tail.
222    */
223   struct GSF_PeerTransmitHandle *pth_tail;
224
225   /**
226    * Messages (replies, queries, content migration) we would like to
227    * send to this peer in the near future.  Sorted by priority, head.
228    */
229   struct GSF_DelayedHandle *delayed_head;
230
231   /**
232    * Messages (replies, queries, content migration) we would like to
233    * send to this peer in the near future.  Sorted by priority, tail.
234    */
235   struct GSF_DelayedHandle *delayed_tail;
236
237   /**
238    * Migration stop message in our queue, or NULL if we have none pending.
239    */
240   struct GSF_PeerTransmitHandle *migration_pth;
241
242   /**
243    * Context of our GNUNET_ATS_reserve_bandwidth call (or NULL).
244    */
245   struct GNUNET_ATS_ReservationContext *rc;
246
247   /**
248    * Task scheduled if we need to retry bandwidth reservation later.
249    */
250   struct GNUNET_SCHEDULER_Task *rc_delay_task;
251
252   /**
253    * Active requests from this neighbour, map of query to 'struct PeerRequest'.
254    */
255   struct GNUNET_CONTAINER_MultiHashMap *request_map;
256
257   /**
258    * Handle for an active request for transmission to this
259    * peer, or NULL (if core queue was full).
260    */
261   struct GNUNET_CORE_TransmitHandle *cth;
262
263   /**
264    * Increase in traffic preference still to be submitted
265    * to the core service for this peer.
266    */
267   uint64_t inc_preference;
268
269   /**
270    * Set to 1 if we're currently in the process of calling
271    * #GNUNET_CORE_notify_transmit_ready() (so while @e cth is
272    * NULL, we should not call notify_transmit_ready for this
273    * handle right now).
274    */
275   unsigned int cth_in_progress;
276
277   /**
278    * Number of entries in @e delayed_head DLL.
279    */
280   unsigned int delay_queue_size;
281
282   /**
283    * Respect rating for this peer on disk.
284    */
285   uint32_t disk_respect;
286
287   /**
288    * Which offset in @e last_p2p_replies will be updated next?
289    * (we go round-robin).
290    */
291   unsigned int last_p2p_replies_woff;
292
293   /**
294    * Which offset in @e last_client_replies will be updated next?
295    * (we go round-robin).
296    */
297   unsigned int last_client_replies_woff;
298
299   /**
300    * Current offset into @e last_request_times ring buffer.
301    */
302   unsigned int last_request_times_off;
303
304   /**
305    * #GNUNET_YES if we did successfully reserve 32k bandwidth,
306    * #GNUNET_NO if not.
307    */
308   int did_reserve;
309
310   /**
311    * Function called when the creation of this record is complete.
312    */
313   GSF_ConnectedPeerCreationCallback creation_cb;
314
315   /**
316    * Closure for @e creation_cb
317    */
318   void *creation_cb_cls;
319
320   /**
321    * Handle to the PEERSTORE iterate request for peer respect value
322    */
323   struct GNUNET_PEERSTORE_IterateContext *respect_iterate_req;
324
325 };
326
327
328 /**
329  * Map from peer identities to `struct GSF_ConnectPeer` entries.
330  */
331 static struct GNUNET_CONTAINER_MultiPeerMap *cp_map;
332
333 /**
334  * Handle to peerstore service.
335  */
336 static struct GNUNET_PEERSTORE_Handle *peerstore;
337
338
339 /**
340  * Update the latency information kept for the given peer.
341  *
342  * @param id peer record to update
343  * @param latency current latency value
344  */
345 void
346 GSF_update_peer_latency_ (const struct GNUNET_PeerIdentity *id,
347                           struct GNUNET_TIME_Relative latency)
348 {
349   struct GSF_ConnectedPeer *cp;
350
351   cp = GSF_peer_get_ (id);
352   if (NULL == cp)
353     return; /* we're not yet connected at the core level, ignore */
354   GNUNET_LOAD_value_set_decline (cp->ppd.transmission_delay,
355                                  latency);
356 }
357
358
359 /**
360  * Return the performance data record for the given peer
361  *
362  * @param cp peer to query
363  * @return performance data record for the peer
364  */
365 struct GSF_PeerPerformanceData *
366 GSF_get_peer_performance_data_ (struct GSF_ConnectedPeer *cp)
367 {
368   return &cp->ppd;
369 }
370
371
372 /**
373  * Core is ready to transmit to a peer, get the message.
374  *
375  * @param cls the `struct GSF_PeerTransmitHandle` of the message
376  * @param size number of bytes core is willing to take
377  * @param buf where to copy the message
378  * @return number of bytes copied to @a buf
379  */
380 static size_t
381 peer_transmit_ready_cb (void *cls,
382                         size_t size,
383                         void *buf);
384
385
386 /**
387  * Function called by core upon success or failure of our bandwidth reservation request.
388  *
389  * @param cls the `struct GSF_ConnectedPeer` of the peer for which we made the request
390  * @param peer identifies the peer
391  * @param amount set to the amount that was actually reserved or unreserved;
392  *               either the full requested amount or zero (no partial reservations)
393  * @param res_delay if the reservation could not be satisfied (amount was 0), how
394  *        long should the client wait until re-trying?
395  */
396 static void
397 ats_reserve_callback (void *cls,
398                       const struct GNUNET_PeerIdentity *peer,
399                       int32_t amount,
400                       struct GNUNET_TIME_Relative res_delay);
401
402
403 /**
404  * If ready (bandwidth reserved), try to schedule transmission via
405  * core for the given handle.
406  *
407  * @param pth transmission handle to schedule
408  */
409 static void
410 schedule_transmission (struct GSF_PeerTransmitHandle *pth)
411 {
412   struct GSF_ConnectedPeer *cp;
413   struct GNUNET_PeerIdentity target;
414
415   cp = pth->cp;
416   if ((NULL != cp->cth) || (0 != cp->cth_in_progress))
417     return;                     /* already done */
418   GNUNET_assert (0 != cp->ppd.pid);
419   GNUNET_PEER_resolve (cp->ppd.pid, &target);
420
421   if (0 != cp->inc_preference)
422   {
423     GNUNET_ATS_performance_change_preference (GSF_ats,
424                                               &target,
425                                               GNUNET_ATS_PREFERENCE_BANDWIDTH,
426                                               (double) cp->inc_preference,
427                                               GNUNET_ATS_PREFERENCE_END);
428     cp->inc_preference = 0;
429   }
430
431   if ( (GNUNET_YES == pth->is_query) &&
432        (GNUNET_YES != pth->was_reserved) )
433   {
434     /* query, need reservation */
435     if (GNUNET_YES != cp->did_reserve)
436       return;                   /* not ready */
437     cp->did_reserve = GNUNET_NO;
438     /* reservation already done! */
439     pth->was_reserved = GNUNET_YES;
440     cp->rc = GNUNET_ATS_reserve_bandwidth (GSF_ats,
441                                            &target,
442                                            DBLOCK_SIZE,
443                                            &ats_reserve_callback,
444                                            cp);
445     return;
446   }
447   GNUNET_assert (NULL == cp->cth);
448   cp->cth_in_progress++;
449   cp->cth =
450     GNUNET_CORE_notify_transmit_ready (GSF_core,
451                                        GNUNET_YES,
452                                        GNUNET_CORE_PRIO_BACKGROUND,
453                                        GNUNET_TIME_absolute_get_remaining (pth->timeout),
454                                        &target,
455                                        pth->size,
456                                        &peer_transmit_ready_cb, cp);
457   GNUNET_assert (NULL != cp->cth);
458   GNUNET_assert (0 < cp->cth_in_progress--);
459 }
460
461
462 /**
463  * Core is ready to transmit to a peer, get the message.
464  *
465  * @param cls the `struct GSF_PeerTransmitHandle` of the message
466  * @param size number of bytes core is willing to take
467  * @param buf where to copy the message
468  * @return number of bytes copied to @a buf
469  */
470 static size_t
471 peer_transmit_ready_cb (void *cls,
472                         size_t size,
473                         void *buf)
474 {
475   struct GSF_ConnectedPeer *cp = cls;
476   struct GSF_PeerTransmitHandle *pth = cp->pth_head;
477   struct GSF_PeerTransmitHandle *pos;
478   size_t ret;
479
480   cp->cth = NULL;
481   if (NULL == pth)
482     return 0;
483   if (pth->size > size)
484   {
485     schedule_transmission (pth);
486     return 0;
487   }
488   if (NULL != pth->timeout_task)
489   {
490     GNUNET_SCHEDULER_cancel (pth->timeout_task);
491     pth->timeout_task = NULL;
492   }
493   GNUNET_CONTAINER_DLL_remove (cp->pth_head,
494                                cp->pth_tail,
495                                pth);
496   if (GNUNET_YES == pth->is_query)
497   {
498     cp->ppd.last_request_times[(cp->last_request_times_off++) %
499                                MAX_QUEUE_PER_PEER] =
500       GNUNET_TIME_absolute_get ();
501     GNUNET_assert (0 < cp->ppd.pending_queries--);
502   }
503   else if (GNUNET_NO == pth->is_query)
504   {
505     GNUNET_assert (0 < cp->ppd.pending_replies--);
506   }
507   GNUNET_LOAD_update (cp->ppd.transmission_delay,
508                       GNUNET_TIME_absolute_get_duration
509                       (pth->transmission_request_start_time).rel_value_us);
510   ret = pth->gmc (pth->gmc_cls, size, buf);
511   if (NULL != (pos = cp->pth_head))
512   {
513     GNUNET_assert (pos != pth);
514     schedule_transmission (pos);
515   }
516   GNUNET_free (pth);
517   return ret;
518 }
519
520
521 /**
522  * (re)try to reserve bandwidth from the given peer.
523  *
524  * @param cls the `struct GSF_ConnectedPeer` to reserve from
525  * @param tc scheduler context
526  */
527 static void
528 retry_reservation (void *cls,
529                    const struct GNUNET_SCHEDULER_TaskContext *tc)
530 {
531   struct GSF_ConnectedPeer *cp = cls;
532   struct GNUNET_PeerIdentity target;
533
534   GNUNET_PEER_resolve (cp->ppd.pid, &target);
535   cp->rc_delay_task = NULL;
536   cp->rc =
537     GNUNET_ATS_reserve_bandwidth (GSF_ats,
538                                   &target,
539                                   DBLOCK_SIZE,
540                                   &ats_reserve_callback, cp);
541 }
542
543
544 /**
545  * Function called by core upon success or failure of our bandwidth reservation request.
546  *
547  * @param cls the `struct GSF_ConnectedPeer` of the peer for which we made the request
548  * @param peer identifies the peer
549  * @param amount set to the amount that was actually reserved or unreserved;
550  *               either the full requested amount or zero (no partial reservations)
551  * @param res_delay if the reservation could not be satisfied (amount was 0), how
552  *        long should the client wait until re-trying?
553  */
554 static void
555 ats_reserve_callback (void *cls,
556                       const struct GNUNET_PeerIdentity *peer,
557                       int32_t amount,
558                       struct GNUNET_TIME_Relative res_delay)
559 {
560   struct GSF_ConnectedPeer *cp = cls;
561   struct GSF_PeerTransmitHandle *pth;
562
563   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
564               "Reserved %d bytes / need to wait %s for reservation\n",
565               (int) amount,
566               GNUNET_STRINGS_relative_time_to_string (res_delay, GNUNET_YES));
567   cp->rc = NULL;
568   if (0 == amount)
569   {
570     cp->rc_delay_task =
571         GNUNET_SCHEDULER_add_delayed (res_delay,
572                                       &retry_reservation,
573                                       cp);
574     return;
575   }
576   cp->did_reserve = GNUNET_YES;
577   pth = cp->pth_head;
578   if ( (NULL != pth) &&
579        (NULL == cp->cth) &&
580        (0 == cp->cth_in_progress) )
581   {
582     /* reservation success, try transmission now! */
583     cp->cth_in_progress++;
584     cp->cth =
585         GNUNET_CORE_notify_transmit_ready (GSF_core,
586                                            GNUNET_YES,
587                                            GNUNET_CORE_PRIO_BACKGROUND,
588                                            GNUNET_TIME_absolute_get_remaining (pth->timeout),
589                                            peer,
590                                            pth->size,
591                                            &peer_transmit_ready_cb,
592                                            cp);
593     GNUNET_assert (NULL != cp->cth);
594     GNUNET_assert (0 < cp->cth_in_progress--);
595   }
596 }
597
598
599 /**
600  * Function called by PEERSTORE with peer respect record
601  *
602  * @param cls handle to connected peer entry
603  * @param record peerstore record information
604  * @param emsg error message, or NULL if no errors
605  * @return #GNUNET_NO to stop iterating since we only expect 0 or 1 records
606  */
607 static int
608 peer_respect_cb (void *cls,
609                  const struct GNUNET_PEERSTORE_Record *record,
610                  const char *emsg)
611 {
612   struct GSF_ConnectedPeer *cp = cls;
613
614   GNUNET_assert (NULL != cp->respect_iterate_req);
615   cp->respect_iterate_req = NULL;
616   if ((NULL != record) && (sizeof (cp->disk_respect) == record->value_size))
617     cp->disk_respect = cp->ppd.respect = *((uint32_t *)record->value);
618   GSF_push_start_ (cp);
619   if (NULL != cp->creation_cb)
620     cp->creation_cb (cp->creation_cb_cls, cp);
621   return GNUNET_NO;
622 }
623
624
625 /**
626  * A peer connected to us.  Setup the connected peer
627  * records.
628  *
629  * @param peer identity of peer that connected
630  * @param creation_cb callback function when the record is created.
631  * @param creation_cb_cls closure for @creation_cb
632  */
633 void
634 GSF_peer_connect_handler_ (const struct GNUNET_PeerIdentity *peer,
635                            GSF_ConnectedPeerCreationCallback creation_cb,
636                            void *creation_cb_cls)
637 {
638   struct GSF_ConnectedPeer *cp;
639
640   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
641               "Connected to peer %s\n",
642               GNUNET_i2s (peer));
643   cp = GNUNET_new (struct GSF_ConnectedPeer);
644   cp->ppd.pid = GNUNET_PEER_intern (peer);
645   cp->ppd.transmission_delay = GNUNET_LOAD_value_init (GNUNET_TIME_UNIT_ZERO);
646   cp->rc =
647       GNUNET_ATS_reserve_bandwidth (GSF_ats, peer, DBLOCK_SIZE,
648                                     &ats_reserve_callback, cp);
649   cp->request_map = GNUNET_CONTAINER_multihashmap_create (128, GNUNET_NO);
650   GNUNET_break (GNUNET_OK ==
651                 GNUNET_CONTAINER_multipeermap_put (cp_map,
652                GSF_connected_peer_get_identity2_ (cp),
653                                                    cp,
654                                                    GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
655   GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# peers connected"),
656                          GNUNET_CONTAINER_multipeermap_size (cp_map),
657                          GNUNET_NO);
658   cp->creation_cb = creation_cb;
659   cp->creation_cb_cls = creation_cb_cls;
660   cp->respect_iterate_req =
661       GNUNET_PEERSTORE_iterate (peerstore, "fs", peer, "respect",
662                                 GNUNET_TIME_UNIT_FOREVER_REL, &peer_respect_cb,
663                                 cp);
664 }
665
666
667 /**
668  * It may be time to re-start migrating content to this
669  * peer.  Check, and if so, restart migration.
670  *
671  * @param cls the `struct GSF_ConnectedPeer`
672  * @param tc scheduler context
673  */
674 static void
675 revive_migration (void *cls,
676                   const struct GNUNET_SCHEDULER_TaskContext *tc)
677 {
678   struct GSF_ConnectedPeer *cp = cls;
679   struct GNUNET_TIME_Relative bt;
680
681   cp->mig_revive_task = NULL;
682   bt = GNUNET_TIME_absolute_get_remaining (cp->ppd.migration_blocked_until);
683   if (0 != bt.rel_value_us)
684   {
685     /* still time left... */
686     cp->mig_revive_task =
687         GNUNET_SCHEDULER_add_delayed (bt, &revive_migration, cp);
688     return;
689   }
690   GSF_push_start_ (cp);
691 }
692
693
694 /**
695  * Get a handle for a connected peer.
696  *
697  * @param peer peer's identity
698  * @return NULL if the peer is not currently connected
699  */
700 struct GSF_ConnectedPeer *
701 GSF_peer_get_ (const struct GNUNET_PeerIdentity *peer)
702 {
703   if (NULL == cp_map)
704     return NULL;
705   return GNUNET_CONTAINER_multipeermap_get (cp_map, peer);
706 }
707
708
709 /**
710  * Handle P2P "MIGRATION_STOP" message.
711  *
712  * @param cls closure, always NULL
713  * @param other the other peer involved (sender or receiver, NULL
714  *        for loopback messages where we are both sender and receiver)
715  * @param message the actual message
716  * @return #GNUNET_OK to keep the connection open,
717  *         #GNUNET_SYSERR to close it (signal serious error)
718  */
719 int
720 GSF_handle_p2p_migration_stop_ (void *cls,
721                                 const struct GNUNET_PeerIdentity *other,
722                                 const struct GNUNET_MessageHeader *message)
723 {
724   struct GSF_ConnectedPeer *cp;
725   const struct MigrationStopMessage *msm;
726   struct GNUNET_TIME_Relative bt;
727
728   msm = (const struct MigrationStopMessage *) message;
729   cp = GSF_peer_get_ (other);
730   if (NULL == cp)
731   {
732     GNUNET_break (0);
733     return GNUNET_OK;
734   }
735   GNUNET_STATISTICS_update (GSF_stats,
736                             gettext_noop ("# migration stop messages received"),
737                             1, GNUNET_NO);
738   bt = GNUNET_TIME_relative_ntoh (msm->duration);
739   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
740               _("Migration of content to peer `%s' blocked for %s\n"),
741               GNUNET_i2s (other),
742               GNUNET_STRINGS_relative_time_to_string (bt, GNUNET_YES));
743   cp->ppd.migration_blocked_until = GNUNET_TIME_relative_to_absolute (bt);
744   if (NULL == cp->mig_revive_task)
745   {
746     GSF_push_stop_ (cp);
747     cp->mig_revive_task =
748         GNUNET_SCHEDULER_add_delayed (bt,
749                                       &revive_migration, cp);
750   }
751   return GNUNET_OK;
752 }
753
754
755 /**
756  * Copy reply and free put message.
757  *
758  * @param cls the `struct PutMessage`
759  * @param buf_size number of bytes available in @a buf
760  * @param buf where to copy the message, NULL on error (peer disconnect)
761  * @return number of bytes copied to @a buf, can be 0 (without indicating an error)
762  */
763 static size_t
764 copy_reply (void *cls,
765             size_t buf_size,
766             void *buf)
767 {
768   struct PutMessage *pm = cls;
769   size_t size;
770
771   if (NULL != buf)
772   {
773     GNUNET_assert (buf_size >= ntohs (pm->header.size));
774     size = ntohs (pm->header.size);
775     memcpy (buf, pm, size);
776     GNUNET_STATISTICS_update (GSF_stats,
777                               gettext_noop ("# replies transmitted to other peers"),
778                               1,
779                               GNUNET_NO);
780   }
781   else
782   {
783     size = 0;
784     GNUNET_STATISTICS_update (GSF_stats,
785                               gettext_noop ("# replies dropped"),
786                               1,
787                               GNUNET_NO);
788   }
789   GNUNET_free (pm);
790   return size;
791 }
792
793
794 /**
795  * Free resources associated with the given peer request.
796  *
797  * @param peerreq request to free
798  * @param query associated key for the request
799  */
800 static void
801 free_pending_request (struct PeerRequest *peerreq,
802                       const struct GNUNET_HashCode *query)
803 {
804   struct GSF_ConnectedPeer *cp = peerreq->cp;
805
806   if (NULL != peerreq->kill_task)
807   {
808     GNUNET_SCHEDULER_cancel (peerreq->kill_task);
809     peerreq->kill_task = NULL;
810   }
811   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
812                             -1, GNUNET_NO);
813   GNUNET_break (GNUNET_YES ==
814                 GNUNET_CONTAINER_multihashmap_remove (cp->request_map,
815                                                       query, peerreq));
816   GNUNET_free (peerreq);
817 }
818
819
820 /**
821  * Cancel all requests associated with the peer.
822  *
823  * @param cls unused
824  * @param query hash code of the request
825  * @param value the `struct GSF_PendingRequest`
826  * @return #GNUNET_YES (continue to iterate)
827  */
828 static int
829 cancel_pending_request (void *cls,
830                         const struct GNUNET_HashCode *query,
831                         void *value)
832 {
833   struct PeerRequest *peerreq = value;
834   struct GSF_PendingRequest *pr = peerreq->pr;
835   struct GSF_PendingRequestData *prd;
836
837   prd = GSF_pending_request_get_data_ (pr);
838   GSF_pending_request_cancel_ (pr, GNUNET_NO);
839   free_pending_request (peerreq, &prd->query);
840   return GNUNET_OK;
841 }
842
843
844 /**
845  * Free the given request.
846  *
847  * @param cls the request to free
848  * @param tc task context
849  */
850 static void
851 peer_request_destroy (void *cls,
852                       const struct GNUNET_SCHEDULER_TaskContext *tc)
853 {
854   struct PeerRequest *peerreq = cls;
855   struct GSF_PendingRequest *pr = peerreq->pr;
856   struct GSF_PendingRequestData *prd;
857
858   peerreq->kill_task = NULL;
859   prd = GSF_pending_request_get_data_ (pr);
860   cancel_pending_request (NULL, &prd->query, peerreq);
861 }
862
863
864 /**
865  * The artificial delay is over, transmit the message now.
866  *
867  * @param cls the `struct GSF_DelayedHandle` with the message
868  * @param tc scheduler context
869  */
870 static void
871 transmit_delayed_now (void *cls,
872                       const struct GNUNET_SCHEDULER_TaskContext *tc)
873 {
874   struct GSF_DelayedHandle *dh = cls;
875   struct GSF_ConnectedPeer *cp = dh->cp;
876
877   GNUNET_CONTAINER_DLL_remove (cp->delayed_head,
878                                cp->delayed_tail,
879                                dh);
880   cp->delay_queue_size--;
881   if (0 != (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason))
882   {
883     GNUNET_free (dh->pm);
884     GNUNET_free (dh);
885     return;
886   }
887   (void) GSF_peer_transmit_ (cp,
888                              GNUNET_NO,
889                              UINT32_MAX,
890                              REPLY_TIMEOUT,
891                              dh->msize,
892                              &copy_reply,
893                              dh->pm);
894   GNUNET_free (dh);
895 }
896
897
898 /**
899  * Get the randomized delay a response should be subjected to.
900  *
901  * @return desired delay
902  */
903 static struct GNUNET_TIME_Relative
904 get_randomized_delay ()
905 {
906   struct GNUNET_TIME_Relative ret;
907
908   ret =
909       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
910                                      GNUNET_CRYPTO_random_u32
911                                      (GNUNET_CRYPTO_QUALITY_WEAK,
912                                       2 * GSF_avg_latency.rel_value_us + 1));
913 #if INSANE_STATISTICS
914   GNUNET_STATISTICS_update (GSF_stats,
915                             gettext_noop
916                             ("# artificial delays introduced (ms)"),
917                             ret.rel_value_us / 1000LL, GNUNET_NO);
918 #endif
919   return ret;
920 }
921
922
923 /**
924  * Handle a reply to a pending request.  Also called if a request
925  * expires (then with data == NULL).  The handler may be called
926  * many times (depending on the request type), but will not be
927  * called during or after a call to GSF_pending_request_cancel
928  * and will also not be called anymore after a call signalling
929  * expiration.
930  *
931  * @param cls `struct PeerRequest` this is an answer for
932  * @param eval evaluation of the result
933  * @param pr handle to the original pending request
934  * @param reply_anonymity_level anonymity level for the reply, UINT32_MAX for "unknown"
935  * @param expiration when does @a data expire?
936  * @param last_transmission when did we last transmit a request for this block
937  * @param type type of the block
938  * @param data response data, NULL on request expiration
939  * @param data_len number of bytes in @a data
940  */
941 static void
942 handle_p2p_reply (void *cls,
943                   enum GNUNET_BLOCK_EvaluationResult eval,
944                   struct GSF_PendingRequest *pr,
945                   uint32_t reply_anonymity_level,
946                   struct GNUNET_TIME_Absolute expiration,
947                   struct GNUNET_TIME_Absolute last_transmission,
948                   enum GNUNET_BLOCK_Type type,
949                   const void *data,
950                   size_t data_len)
951 {
952   struct PeerRequest *peerreq = cls;
953   struct GSF_ConnectedPeer *cp = peerreq->cp;
954   struct GSF_PendingRequestData *prd;
955   struct PutMessage *pm;
956   size_t msize;
957
958   GNUNET_assert (data_len + sizeof (struct PutMessage) <
959                  GNUNET_SERVER_MAX_MESSAGE_SIZE);
960   GNUNET_assert (peerreq->pr == pr);
961   prd = GSF_pending_request_get_data_ (pr);
962   if (NULL == data)
963   {
964     free_pending_request (peerreq, &prd->query);
965     return;
966   }
967   GNUNET_break (GNUNET_BLOCK_TYPE_ANY != type);
968   if ((prd->type != type) && (GNUNET_BLOCK_TYPE_ANY != prd->type))
969   {
970     GNUNET_STATISTICS_update (GSF_stats,
971                               gettext_noop
972                               ("# replies dropped due to type mismatch"),
973                                 1, GNUNET_NO);
974     return;
975   }
976   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
977               "Transmitting result for query `%s' to peer\n",
978               GNUNET_h2s (&prd->query));
979   GNUNET_STATISTICS_update (GSF_stats,
980                             gettext_noop ("# replies received for other peers"),
981                             1, GNUNET_NO);
982   msize = sizeof (struct PutMessage) + data_len;
983   if (msize >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
984   {
985     GNUNET_break (0);
986     return;
987   }
988   if ((UINT32_MAX != reply_anonymity_level) && (reply_anonymity_level > 1))
989   {
990     if (reply_anonymity_level - 1 > GSF_cover_content_count)
991     {
992       GNUNET_STATISTICS_update (GSF_stats,
993                                 gettext_noop
994                                 ("# replies dropped due to insufficient cover traffic"),
995                                 1, GNUNET_NO);
996       return;
997     }
998     GSF_cover_content_count -= (reply_anonymity_level - 1);
999   }
1000
1001   pm = GNUNET_malloc (msize);
1002   pm->header.type = htons (GNUNET_MESSAGE_TYPE_FS_PUT);
1003   pm->header.size = htons (msize);
1004   pm->type = htonl (type);
1005   pm->expiration = GNUNET_TIME_absolute_hton (expiration);
1006   memcpy (&pm[1], data, data_len);
1007   if ( (UINT32_MAX != reply_anonymity_level) &&
1008        (0 != reply_anonymity_level) &&
1009        (GNUNET_YES == GSF_enable_randomized_delays) )
1010   {
1011     struct GSF_DelayedHandle *dh;
1012
1013     dh = GNUNET_new (struct GSF_DelayedHandle);
1014     dh->cp = cp;
1015     dh->pm = pm;
1016     dh->msize = msize;
1017     GNUNET_CONTAINER_DLL_insert (cp->delayed_head,
1018                                  cp->delayed_tail,
1019                                  dh);
1020     cp->delay_queue_size++;
1021     dh->delay_task =
1022         GNUNET_SCHEDULER_add_delayed (get_randomized_delay (),
1023                                       &transmit_delayed_now,
1024                                       dh);
1025   }
1026   else
1027   {
1028     (void) GSF_peer_transmit_ (cp,
1029                                GNUNET_NO,
1030                                UINT32_MAX,
1031                                REPLY_TIMEOUT,
1032                                msize,
1033                                &copy_reply,
1034                                pm);
1035   }
1036   if (GNUNET_BLOCK_EVALUATION_OK_LAST != eval)
1037     return;
1038   if (NULL == peerreq->kill_task)
1039   {
1040     GNUNET_STATISTICS_update (GSF_stats,
1041                               gettext_noop
1042                               ("# P2P searches destroyed due to ultimate reply"),
1043                               1, GNUNET_NO);
1044     peerreq->kill_task =
1045         GNUNET_SCHEDULER_add_now (&peer_request_destroy, peerreq);
1046   }
1047 }
1048
1049
1050 /**
1051  * Increase the peer's respect by a value.
1052  *
1053  * @param cp which peer to change the respect value on
1054  * @param value is the int value by which the
1055  *  peer's credit is to be increased or decreased
1056  * @returns the actual change in respect (positive or negative)
1057  */
1058 static int
1059 change_peer_respect (struct GSF_ConnectedPeer *cp, int value)
1060 {
1061   if (0 == value)
1062     return 0;
1063   GNUNET_assert (NULL != cp);
1064   if (value > 0)
1065   {
1066     if (cp->ppd.respect + value < cp->ppd.respect)
1067     {
1068       value = UINT32_MAX - cp->ppd.respect;
1069       cp->ppd.respect = UINT32_MAX;
1070     }
1071     else
1072       cp->ppd.respect += value;
1073   }
1074   else
1075   {
1076     if (cp->ppd.respect < -value)
1077     {
1078       value = -cp->ppd.respect;
1079       cp->ppd.respect = 0;
1080     }
1081     else
1082       cp->ppd.respect += value;
1083   }
1084   return value;
1085 }
1086
1087
1088 /**
1089  * We've received a request with the specified priority.  Bound it
1090  * according to how much we respect the given peer.
1091  *
1092  * @param prio_in requested priority
1093  * @param cp the peer making the request
1094  * @return effective priority
1095  */
1096 static int32_t
1097 bound_priority (uint32_t prio_in, struct GSF_ConnectedPeer *cp)
1098 {
1099 #define N ((double)128.0)
1100   uint32_t ret;
1101   double rret;
1102   int ld;
1103
1104   ld = GSF_test_get_load_too_high_ (0);
1105   if (GNUNET_SYSERR == ld)
1106   {
1107 #if INSANE_STATISTICS
1108     GNUNET_STATISTICS_update (GSF_stats,
1109                               gettext_noop
1110                               ("# requests done for free (low load)"), 1,
1111                               GNUNET_NO);
1112 #endif
1113     return 0;                   /* excess resources */
1114   }
1115   if (prio_in > INT32_MAX)
1116     prio_in = INT32_MAX;
1117   ret = -change_peer_respect (cp, -(int) prio_in);
1118   if (ret > 0)
1119   {
1120     if (ret > GSF_current_priorities + N)
1121       rret = GSF_current_priorities + N;
1122     else
1123       rret = ret;
1124     GSF_current_priorities = (GSF_current_priorities * (N - 1) + rret) / N;
1125   }
1126   if ((GNUNET_YES == ld) && (ret > 0))
1127   {
1128     /* try with charging */
1129     ld = GSF_test_get_load_too_high_ (ret);
1130   }
1131   if (GNUNET_YES == ld)
1132   {
1133     GNUNET_STATISTICS_update (GSF_stats,
1134                               gettext_noop
1135                               ("# request dropped, priority insufficient"), 1,
1136                               GNUNET_NO);
1137     /* undo charge */
1138     change_peer_respect (cp, (int) ret);
1139     return -1;                  /* not enough resources */
1140   }
1141   else
1142   {
1143     GNUNET_STATISTICS_update (GSF_stats,
1144                               gettext_noop
1145                               ("# requests done for a price (normal load)"), 1,
1146                               GNUNET_NO);
1147   }
1148 #undef N
1149   return ret;
1150 }
1151
1152
1153 /**
1154  * The priority level imposes a bound on the maximum
1155  * value for the ttl that can be requested.
1156  *
1157  * @param ttl_in requested ttl
1158  * @param prio given priority
1159  * @return ttl_in if ttl_in is below the limit,
1160  *         otherwise the ttl-limit for the given priority
1161  */
1162 static int32_t
1163 bound_ttl (int32_t ttl_in, uint32_t prio)
1164 {
1165   unsigned long long allowed;
1166
1167   if (ttl_in <= 0)
1168     return ttl_in;
1169   allowed = ((unsigned long long) prio) * TTL_DECREMENT / 1000;
1170   if (ttl_in > allowed)
1171   {
1172     if (allowed >= (1 << 30))
1173       return 1 << 30;
1174     return allowed;
1175   }
1176   return ttl_in;
1177 }
1178
1179
1180 /**
1181  * Handle P2P "QUERY" message.  Creates the pending request entry
1182  * and sets up all of the data structures to that we will
1183  * process replies properly.  Does not initiate forwarding or
1184  * local database lookups.
1185  *
1186  * @param other the other peer involved (sender or receiver, NULL
1187  *        for loopback messages where we are both sender and receiver)
1188  * @param message the actual message
1189  * @return pending request handle, NULL on error
1190  */
1191 struct GSF_PendingRequest *
1192 GSF_handle_p2p_query_ (const struct GNUNET_PeerIdentity *other,
1193                        const struct GNUNET_MessageHeader *message)
1194 {
1195   struct PeerRequest *peerreq;
1196   struct GSF_PendingRequest *pr;
1197   struct GSF_PendingRequestData *prd;
1198   struct GSF_ConnectedPeer *cp;
1199   struct GSF_ConnectedPeer *cps;
1200   const struct GNUNET_PeerIdentity *target;
1201   enum GSF_PendingRequestOptions options;
1202   uint16_t msize;
1203   const struct GetMessage *gm;
1204   unsigned int bits;
1205   const struct GNUNET_PeerIdentity *opt;
1206   uint32_t bm;
1207   size_t bfsize;
1208   uint32_t ttl_decrement;
1209   int32_t priority;
1210   int32_t ttl;
1211   enum GNUNET_BLOCK_Type type;
1212   GNUNET_PEER_Id spid;
1213
1214   msize = ntohs (message->size);
1215   if (msize < sizeof (struct GetMessage))
1216   {
1217     GNUNET_break_op (0);
1218     return NULL;
1219   }
1220   GNUNET_STATISTICS_update (GSF_stats,
1221                             gettext_noop
1222                             ("# GET requests received (from other peers)"),
1223                             1,
1224                             GNUNET_NO);
1225   gm = (const struct GetMessage *) message;
1226   type = ntohl (gm->type);
1227   bm = ntohl (gm->hash_bitmap);
1228   bits = 0;
1229   while (bm > 0)
1230   {
1231     if (1 == (bm & 1))
1232       bits++;
1233     bm >>= 1;
1234   }
1235   if (msize < sizeof (struct GetMessage) + bits * sizeof (struct GNUNET_PeerIdentity))
1236   {
1237     GNUNET_break_op (0);
1238     return NULL;
1239   }
1240   opt = (const struct GNUNET_PeerIdentity *) &gm[1];
1241   bfsize = msize - sizeof (struct GetMessage) - bits * sizeof (struct GNUNET_PeerIdentity);
1242   /* bfsize must be power of 2, check! */
1243   if (0 != ((bfsize - 1) & bfsize))
1244   {
1245     GNUNET_break_op (0);
1246     return NULL;
1247   }
1248   GSF_cover_query_count++;
1249   bm = ntohl (gm->hash_bitmap);
1250   bits = 0;
1251   cps = GSF_peer_get_ (other);
1252   if (NULL == cps)
1253   {
1254     /* peer must have just disconnected */
1255     GNUNET_STATISTICS_update (GSF_stats,
1256                               gettext_noop
1257                               ("# requests dropped due to initiator not being connected"),
1258                               1, GNUNET_NO);
1259     return NULL;
1260   }
1261   if (0 != (bm & GET_MESSAGE_BIT_RETURN_TO))
1262     cp = GSF_peer_get_ (&opt[bits++]);
1263   else
1264     cp = cps;
1265   if (NULL == cp)
1266   {
1267     if (0 != (bm & GET_MESSAGE_BIT_RETURN_TO))
1268       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1269                   "Failed to find RETURN-TO peer `%s' in connection set. Dropping query.\n",
1270                   GNUNET_i2s (&opt[bits - 1]));
1271
1272     else
1273       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1274                   "Failed to find peer `%s' in connection set. Dropping query.\n",
1275                   GNUNET_i2s (other));
1276     GNUNET_STATISTICS_update (GSF_stats,
1277                               gettext_noop
1278                               ("# requests dropped due to missing reverse route"),
1279                               1,
1280                               GNUNET_NO);
1281     return NULL;
1282   }
1283   if (cp->ppd.pending_replies + cp->delay_queue_size > MAX_QUEUE_PER_PEER)
1284   {
1285     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1286                 "Peer `%s' has too many replies queued already. Dropping query.\n",
1287                 GNUNET_i2s (other));
1288     GNUNET_STATISTICS_update (GSF_stats,
1289                               gettext_noop ("# requests dropped due to full reply queue"),
1290                               1,
1291                               GNUNET_NO);
1292     return NULL;
1293   }
1294   /* note that we can really only check load here since otherwise
1295    * peers could find out that we are overloaded by not being
1296    * disconnected after sending us a malformed query... */
1297   priority = bound_priority (ntohl (gm->priority),
1298                              cps);
1299   if (priority < 0)
1300   {
1301     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1302                 "Dropping query from `%s', this peer is too busy.\n",
1303                 GNUNET_i2s (other));
1304     return NULL;
1305   }
1306   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1307               "Received request for `%s' of type %u from peer `%s' with flags %u\n",
1308               GNUNET_h2s (&gm->query),
1309               (unsigned int) type,
1310               GNUNET_i2s (other),
1311               (unsigned int) bm);
1312   target =
1313       (0 !=
1314        (bm & GET_MESSAGE_BIT_TRANSMIT_TO)) ? (&opt[bits++]) : NULL;
1315   options = GSF_PRO_DEFAULTS;
1316   spid = 0;
1317   if ((GNUNET_LOAD_get_load (cp->ppd.transmission_delay) > 3 * (1 + priority))
1318       || (GNUNET_LOAD_get_average (cp->ppd.transmission_delay) >
1319           GNUNET_CONSTANTS_MAX_CORK_DELAY.rel_value_us * 2 +
1320           GNUNET_LOAD_get_average (GSF_rt_entry_lifetime)))
1321   {
1322     /* don't have BW to send to peer, or would likely take longer than we have for it,
1323      * so at best indirect the query */
1324     priority = 0;
1325     options |= GSF_PRO_FORWARD_ONLY;
1326     spid = GNUNET_PEER_intern (other);
1327     GNUNET_assert (0 != spid);
1328   }
1329   ttl = bound_ttl (ntohl (gm->ttl), priority);
1330   /* decrement ttl (always) */
1331   ttl_decrement =
1332       2 * TTL_DECREMENT + GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
1333                                                     TTL_DECREMENT);
1334   if ((ttl < 0) && (((int32_t) (ttl - ttl_decrement)) > 0))
1335   {
1336     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1337                 "Dropping query from `%s' due to TTL underflow (%d - %u).\n",
1338                 GNUNET_i2s (other), ttl, ttl_decrement);
1339     GNUNET_STATISTICS_update (GSF_stats,
1340                               gettext_noop
1341                               ("# requests dropped due TTL underflow"), 1,
1342                               GNUNET_NO);
1343     /* integer underflow => drop (should be very rare)! */
1344     return NULL;
1345   }
1346   ttl -= ttl_decrement;
1347
1348   /* test if the request already exists */
1349   peerreq = GNUNET_CONTAINER_multihashmap_get (cp->request_map,
1350                                                &gm->query);
1351   if (NULL != peerreq)
1352   {
1353     pr = peerreq->pr;
1354     prd = GSF_pending_request_get_data_ (pr);
1355     if (prd->type == type)
1356     {
1357       if (prd->ttl.abs_value_us >= GNUNET_TIME_absolute_get ().abs_value_us + ttl * 1000LL)
1358       {
1359         /* existing request has higher TTL, drop new one! */
1360         prd->priority += priority;
1361         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1362                     "Have existing request with higher TTL, dropping new request.\n",
1363                     GNUNET_i2s (other));
1364         GNUNET_STATISTICS_update (GSF_stats,
1365                                   gettext_noop
1366                                   ("# requests dropped due to higher-TTL request"),
1367                                   1, GNUNET_NO);
1368         return NULL;
1369       }
1370       /* existing request has lower TTL, drop old one! */
1371       priority += prd->priority;
1372       GSF_pending_request_cancel_ (pr, GNUNET_YES);
1373       free_pending_request (peerreq, &gm->query);
1374     }
1375   }
1376
1377   peerreq = GNUNET_new (struct PeerRequest);
1378   peerreq->cp = cp;
1379   pr = GSF_pending_request_create_ (options,
1380                                     type,
1381                                     &gm->query,
1382                                     target,
1383                                     (bfsize > 0)
1384                                     ? (const char *) &opt[bits]
1385                                     : NULL,
1386                                     bfsize,
1387                                     ntohl (gm->filter_mutator),
1388                                     1 /* anonymity */,
1389                                     (uint32_t) priority,
1390                                     ttl,
1391                                     spid,
1392                                     GNUNET_PEER_intern (other),
1393                                     NULL, 0,        /* replies_seen */
1394                                     &handle_p2p_reply, peerreq);
1395   GNUNET_assert (NULL != pr);
1396   peerreq->pr = pr;
1397   GNUNET_break (GNUNET_OK ==
1398                 GNUNET_CONTAINER_multihashmap_put (cp->request_map, &gm->query,
1399                                                    peerreq,
1400                                                    GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
1401   GNUNET_STATISTICS_update (GSF_stats,
1402                             gettext_noop
1403                             ("# P2P query messages received and processed"), 1,
1404                             GNUNET_NO);
1405   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
1406                             1, GNUNET_NO);
1407   return pr;
1408 }
1409
1410
1411 /**
1412  * Function called if there has been a timeout trying to satisfy
1413  * a transmission request.
1414  *
1415  * @param cls the `struct GSF_PeerTransmitHandle` of the request
1416  * @param tc scheduler context
1417  */
1418 static void
1419 peer_transmit_timeout (void *cls,
1420                        const struct GNUNET_SCHEDULER_TaskContext *tc)
1421 {
1422   struct GSF_PeerTransmitHandle *pth = cls;
1423   struct GSF_ConnectedPeer *cp;
1424
1425   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1426               "Timeout trying to transmit to other peer\n");
1427   pth->timeout_task = NULL;
1428   cp = pth->cp;
1429   GNUNET_CONTAINER_DLL_remove (cp->pth_head,
1430                                cp->pth_tail,
1431                                pth);
1432   if (GNUNET_YES == pth->is_query)
1433     GNUNET_assert (0 < cp->ppd.pending_queries--);
1434   else if (GNUNET_NO == pth->is_query)
1435     GNUNET_assert (0 < cp->ppd.pending_replies--);
1436   GNUNET_LOAD_update (cp->ppd.transmission_delay, UINT64_MAX);
1437   if (NULL != cp->cth)
1438   {
1439     GNUNET_CORE_notify_transmit_ready_cancel (cp->cth);
1440     cp->cth = NULL;
1441   }
1442   pth->gmc (pth->gmc_cls, 0, NULL);
1443   GNUNET_assert (0 == cp->cth_in_progress);
1444   GNUNET_free (pth);
1445 }
1446
1447
1448 /**
1449  * Transmit a message to the given peer as soon as possible.
1450  * If the peer disconnects before the transmission can happen,
1451  * the callback is invoked with a `NULL` @a buffer.
1452  *
1453  * @param cp target peer
1454  * @param is_query is this a query (#GNUNET_YES) or content (#GNUNET_NO) or neither (#GNUNET_SYSERR)
1455  * @param priority how important is this request?
1456  * @param timeout when does this request timeout (call gmc with error)
1457  * @param size number of bytes we would like to send to the peer
1458  * @param gmc function to call to get the message
1459  * @param gmc_cls closure for @a gmc
1460  * @return handle to cancel request
1461  */
1462 struct GSF_PeerTransmitHandle *
1463 GSF_peer_transmit_ (struct GSF_ConnectedPeer *cp,
1464                     int is_query,
1465                     uint32_t priority,
1466                     struct GNUNET_TIME_Relative timeout,
1467                     size_t size,
1468                     GSF_GetMessageCallback gmc, void *gmc_cls)
1469 {
1470   struct GSF_PeerTransmitHandle *pth;
1471   struct GSF_PeerTransmitHandle *pos;
1472   struct GSF_PeerTransmitHandle *prev;
1473
1474   pth = GNUNET_new (struct GSF_PeerTransmitHandle);
1475   pth->transmission_request_start_time = GNUNET_TIME_absolute_get ();
1476   pth->timeout = GNUNET_TIME_relative_to_absolute (timeout);
1477   pth->gmc = gmc;
1478   pth->gmc_cls = gmc_cls;
1479   pth->size = size;
1480   pth->is_query = is_query;
1481   pth->priority = priority;
1482   pth->cp = cp;
1483   /* insertion sort (by priority, descending) */
1484   prev = NULL;
1485   pos = cp->pth_head;
1486   while ((NULL != pos) && (pos->priority > priority))
1487   {
1488     prev = pos;
1489     pos = pos->next;
1490   }
1491   GNUNET_CONTAINER_DLL_insert_after (cp->pth_head,
1492                                      cp->pth_tail,
1493                                      prev,
1494                                      pth);
1495   if (GNUNET_YES == is_query)
1496     cp->ppd.pending_queries++;
1497   else if (GNUNET_NO == is_query)
1498     cp->ppd.pending_replies++;
1499   pth->timeout_task
1500     = GNUNET_SCHEDULER_add_delayed (timeout,
1501                                     &peer_transmit_timeout,
1502                                     pth);
1503   schedule_transmission (pth);
1504   return pth;
1505 }
1506
1507
1508 /**
1509  * Cancel an earlier request for transmission.
1510  *
1511  * @param pth request to cancel
1512  */
1513 void
1514 GSF_peer_transmit_cancel_ (struct GSF_PeerTransmitHandle *pth)
1515 {
1516   struct GSF_ConnectedPeer *cp;
1517
1518   if (NULL != pth->timeout_task)
1519   {
1520     GNUNET_SCHEDULER_cancel (pth->timeout_task);
1521     pth->timeout_task = NULL;
1522   }
1523   cp = pth->cp;
1524   GNUNET_CONTAINER_DLL_remove (cp->pth_head,
1525                                cp->pth_tail,
1526                                pth);
1527   if (GNUNET_YES == pth->is_query)
1528     GNUNET_assert (0 < cp->ppd.pending_queries--);
1529   else if (GNUNET_NO == pth->is_query)
1530     GNUNET_assert (0 < cp->ppd.pending_replies--);
1531   GNUNET_free (pth);
1532 }
1533
1534
1535 /**
1536  * Report on receiving a reply; update the performance record of the given peer.
1537  *
1538  * @param cp responding peer (will be updated)
1539  * @param request_time time at which the original query was transmitted
1540  * @param request_priority priority of the original request
1541  */
1542 void
1543 GSF_peer_update_performance_ (struct GSF_ConnectedPeer *cp,
1544                               struct GNUNET_TIME_Absolute request_time,
1545                               uint32_t request_priority)
1546 {
1547   struct GNUNET_TIME_Relative delay;
1548
1549   delay = GNUNET_TIME_absolute_get_duration (request_time);
1550   cp->ppd.avg_reply_delay.rel_value_us =
1551       (cp->ppd.avg_reply_delay.rel_value_us * (RUNAVG_DELAY_N - 1) +
1552        delay.rel_value_us) / RUNAVG_DELAY_N;
1553   cp->ppd.avg_priority =
1554       (cp->ppd.avg_priority * (RUNAVG_DELAY_N - 1) +
1555        request_priority) / RUNAVG_DELAY_N;
1556 }
1557
1558
1559 /**
1560  * Report on receiving a reply in response to an initiating client.
1561  * Remember that this peer is good for this client.
1562  *
1563  * @param cp responding peer (will be updated)
1564  * @param initiator_client local client on responsible for query
1565  */
1566 void
1567 GSF_peer_update_responder_client_ (struct GSF_ConnectedPeer *cp,
1568                                    struct GSF_LocalClient *initiator_client)
1569 {
1570   cp->ppd.last_client_replies[cp->last_client_replies_woff++ %
1571                               CS2P_SUCCESS_LIST_SIZE] = initiator_client;
1572 }
1573
1574
1575 /**
1576  * Report on receiving a reply in response to an initiating peer.
1577  * Remember that this peer is good for this initiating peer.
1578  *
1579  * @param cp responding peer (will be updated)
1580  * @param initiator_peer other peer responsible for query
1581  */
1582 void
1583 GSF_peer_update_responder_peer_ (struct GSF_ConnectedPeer *cp,
1584                                  const struct GSF_ConnectedPeer *initiator_peer)
1585 {
1586   unsigned int woff;
1587
1588   woff = cp->last_p2p_replies_woff % P2P_SUCCESS_LIST_SIZE;
1589   GNUNET_PEER_change_rc (cp->ppd.last_p2p_replies[woff], -1);
1590   cp->ppd.last_p2p_replies[woff] = initiator_peer->ppd.pid;
1591   GNUNET_PEER_change_rc (initiator_peer->ppd.pid, 1);
1592   cp->last_p2p_replies_woff = (woff + 1) % P2P_SUCCESS_LIST_SIZE;
1593 }
1594
1595
1596 /**
1597  * Write peer-respect information to a file - flush the buffer entry!
1598  *
1599  * @param cls unused
1600  * @param key peer identity
1601  * @param value the `struct GSF_ConnectedPeer` to flush
1602  * @return #GNUNET_OK to continue iteration
1603  */
1604 static int
1605 flush_respect (void *cls,
1606                const struct GNUNET_PeerIdentity *key,
1607                void *value)
1608 {
1609   struct GSF_ConnectedPeer *cp = value;
1610   struct GNUNET_PeerIdentity pid;
1611
1612   if (cp->ppd.respect == cp->disk_respect)
1613     return GNUNET_OK;           /* unchanged */
1614   GNUNET_assert (0 != cp->ppd.pid);
1615   GNUNET_PEER_resolve (cp->ppd.pid, &pid);
1616   GNUNET_PEERSTORE_store (peerstore, "fs", &pid, "respect", &cp->ppd.respect,
1617                           sizeof (cp->ppd.respect),
1618                           GNUNET_TIME_UNIT_FOREVER_ABS,
1619                           GNUNET_PEERSTORE_STOREOPTION_REPLACE, NULL, NULL);
1620   return GNUNET_OK;
1621 }
1622
1623
1624 /**
1625  * A peer disconnected from us.  Tear down the connected peer
1626  * record.
1627  *
1628  * @param cls unused
1629  * @param peer identity of peer that connected
1630  */
1631 void
1632 GSF_peer_disconnect_handler_ (void *cls,
1633                               const struct GNUNET_PeerIdentity *peer)
1634 {
1635   struct GSF_ConnectedPeer *cp;
1636   struct GSF_PeerTransmitHandle *pth;
1637   struct GSF_DelayedHandle *dh;
1638
1639   cp = GSF_peer_get_ (peer);
1640   if (NULL == cp)
1641     return;                     /* must have been disconnect from core with
1642                                  * 'peer' == my_id, ignore */
1643   flush_respect (NULL, peer, cp);
1644   GNUNET_assert (GNUNET_YES ==
1645                  GNUNET_CONTAINER_multipeermap_remove (cp_map,
1646                                                        peer,
1647                                                        cp));
1648   GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# peers connected"),
1649                          GNUNET_CONTAINER_multipeermap_size (cp_map),
1650                          GNUNET_NO);
1651   if (NULL != cp->respect_iterate_req)
1652   {
1653     GNUNET_PEERSTORE_iterate_cancel (cp->respect_iterate_req);
1654     cp->respect_iterate_req = NULL;
1655   }
1656   if (NULL != cp->migration_pth)
1657   {
1658     GSF_peer_transmit_cancel_ (cp->migration_pth);
1659     cp->migration_pth = NULL;
1660   }
1661   if (NULL != cp->rc)
1662   {
1663     GNUNET_ATS_reserve_bandwidth_cancel (cp->rc);
1664     cp->rc = NULL;
1665   }
1666   if (NULL != cp->rc_delay_task)
1667   {
1668     GNUNET_SCHEDULER_cancel (cp->rc_delay_task);
1669     cp->rc_delay_task = NULL;
1670   }
1671   GNUNET_CONTAINER_multihashmap_iterate (cp->request_map,
1672                                          &cancel_pending_request, cp);
1673   GNUNET_CONTAINER_multihashmap_destroy (cp->request_map);
1674   cp->request_map = NULL;
1675   GSF_plan_notify_peer_disconnect_ (cp);
1676   GNUNET_LOAD_value_free (cp->ppd.transmission_delay);
1677   GNUNET_PEER_decrement_rcs (cp->ppd.last_p2p_replies, P2P_SUCCESS_LIST_SIZE);
1678   memset (cp->ppd.last_p2p_replies, 0, sizeof (cp->ppd.last_p2p_replies));
1679   GSF_push_stop_ (cp);
1680   if (NULL != cp->cth)
1681   {
1682     GNUNET_CORE_notify_transmit_ready_cancel (cp->cth);
1683     cp->cth = NULL;
1684   }
1685   GNUNET_assert (0 == cp->cth_in_progress);
1686   while (NULL != (pth = cp->pth_head))
1687   {
1688     if (pth->timeout_task != NULL)
1689     {
1690       GNUNET_SCHEDULER_cancel (pth->timeout_task);
1691       pth->timeout_task = NULL;
1692     }
1693     GNUNET_CONTAINER_DLL_remove (cp->pth_head,
1694                                  cp->pth_tail,
1695                                  pth);
1696     if (GNUNET_YES == pth->is_query)
1697       GNUNET_assert (0 < cp->ppd.pending_queries--);
1698     else if (GNUNET_NO == pth->is_query)
1699       GNUNET_assert (0 < cp->ppd.pending_replies--);
1700     pth->gmc (pth->gmc_cls, 0, NULL);
1701     GNUNET_free (pth);
1702   }
1703   while (NULL != (dh = cp->delayed_head))
1704   {
1705     GNUNET_CONTAINER_DLL_remove (cp->delayed_head,
1706                                  cp->delayed_tail,
1707                                  dh);
1708     cp->delay_queue_size--;
1709     GNUNET_SCHEDULER_cancel (dh->delay_task);
1710     GNUNET_free (dh->pm);
1711     GNUNET_free (dh);
1712   }
1713   GNUNET_PEER_change_rc (cp->ppd.pid, -1);
1714   if (NULL != cp->mig_revive_task)
1715   {
1716     GNUNET_SCHEDULER_cancel (cp->mig_revive_task);
1717     cp->mig_revive_task = NULL;
1718   }
1719   GNUNET_break (0 == cp->ppd.pending_queries);
1720   GNUNET_break (0 == cp->ppd.pending_replies);
1721   GNUNET_free (cp);
1722 }
1723
1724
1725 /**
1726  * Closure for #call_iterator().
1727  */
1728 struct IterationContext
1729 {
1730   /**
1731    * Function to call on each entry.
1732    */
1733   GSF_ConnectedPeerIterator it;
1734
1735   /**
1736    * Closure for @e it.
1737    */
1738   void *it_cls;
1739 };
1740
1741
1742 /**
1743  * Function that calls the callback for each peer.
1744  *
1745  * @param cls the `struct IterationContext *`
1746  * @param key identity of the peer
1747  * @param value the `struct GSF_ConnectedPeer *`
1748  * @return #GNUNET_YES to continue iteration
1749  */
1750 static int
1751 call_iterator (void *cls,
1752                const struct GNUNET_PeerIdentity *key,
1753                void *value)
1754 {
1755   struct IterationContext *ic = cls;
1756   struct GSF_ConnectedPeer *cp = value;
1757
1758   ic->it (ic->it_cls,
1759           key, cp,
1760           &cp->ppd);
1761   return GNUNET_YES;
1762 }
1763
1764
1765 /**
1766  * Iterate over all connected peers.
1767  *
1768  * @param it function to call for each peer
1769  * @param it_cls closure for @a it
1770  */
1771 void
1772 GSF_iterate_connected_peers_ (GSF_ConnectedPeerIterator it,
1773                               void *it_cls)
1774 {
1775   struct IterationContext ic;
1776
1777   ic.it = it;
1778   ic.it_cls = it_cls;
1779   GNUNET_CONTAINER_multipeermap_iterate (cp_map,
1780                                          &call_iterator,
1781                                          &ic);
1782 }
1783
1784
1785 /**
1786  * Obtain the identity of a connected peer.
1787  *
1788  * @param cp peer to get identity of
1789  * @param id identity to set (written to)
1790  */
1791 void
1792 GSF_connected_peer_get_identity_ (const struct GSF_ConnectedPeer *cp,
1793                                   struct GNUNET_PeerIdentity *id)
1794 {
1795   GNUNET_assert (0 != cp->ppd.pid);
1796   GNUNET_PEER_resolve (cp->ppd.pid, id);
1797 }
1798
1799
1800 /**
1801  * Obtain the identity of a connected peer.
1802  *
1803  * @param cp peer to get identity of
1804  * @return reference to peer identity, valid until peer disconnects (!)
1805  */
1806 const struct GNUNET_PeerIdentity *
1807 GSF_connected_peer_get_identity2_ (const struct GSF_ConnectedPeer *cp)
1808 {
1809   GNUNET_assert (0 != cp->ppd.pid);
1810   return GNUNET_PEER_resolve2 (cp->ppd.pid);
1811 }
1812
1813
1814 /**
1815  * Assemble a migration stop message for transmission.
1816  *
1817  * @param cls the `struct GSF_ConnectedPeer` to use
1818  * @param size number of bytes we're allowed to write to @a buf
1819  * @param buf where to copy the message
1820  * @return number of bytes copied to @a buf
1821  */
1822 static size_t
1823 create_migration_stop_message (void *cls,
1824                                size_t size,
1825                                void *buf)
1826 {
1827   struct GSF_ConnectedPeer *cp = cls;
1828   struct MigrationStopMessage msm;
1829
1830   cp->migration_pth = NULL;
1831   if (NULL == buf)
1832     return 0;
1833   GNUNET_assert (size >= sizeof (struct MigrationStopMessage));
1834   msm.header.size = htons (sizeof (struct MigrationStopMessage));
1835   msm.header.type = htons (GNUNET_MESSAGE_TYPE_FS_MIGRATION_STOP);
1836   msm.reserved = htonl (0);
1837   msm.duration =
1838       GNUNET_TIME_relative_hton (GNUNET_TIME_absolute_get_remaining
1839                                  (cp->last_migration_block));
1840   memcpy (buf, &msm, sizeof (struct MigrationStopMessage));
1841   GNUNET_STATISTICS_update (GSF_stats,
1842                             gettext_noop ("# migration stop messages sent"),
1843                             1, GNUNET_NO);
1844   return sizeof (struct MigrationStopMessage);
1845 }
1846
1847
1848 /**
1849  * Ask a peer to stop migrating data to us until the given point
1850  * in time.
1851  *
1852  * @param cp peer to ask
1853  * @param block_time until when to block
1854  */
1855 void
1856 GSF_block_peer_migration_ (struct GSF_ConnectedPeer *cp,
1857                            struct GNUNET_TIME_Absolute block_time)
1858 {
1859   if (cp->last_migration_block.abs_value_us > block_time.abs_value_us)
1860   {
1861     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1862                 "Migration already blocked for another %s\n",
1863                 GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_remaining
1864                                                         (cp->last_migration_block), GNUNET_YES));
1865     return;                     /* already blocked */
1866   }
1867   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Asking to stop migration for %s\n",
1868               GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_remaining (block_time),
1869                                                       GNUNET_YES));
1870   cp->last_migration_block = block_time;
1871   if (NULL != cp->migration_pth)
1872     GSF_peer_transmit_cancel_ (cp->migration_pth);
1873   cp->migration_pth =
1874       GSF_peer_transmit_ (cp, GNUNET_SYSERR, UINT32_MAX,
1875                           GNUNET_TIME_UNIT_FOREVER_REL,
1876                           sizeof (struct MigrationStopMessage),
1877                           &create_migration_stop_message, cp);
1878 }
1879
1880
1881 /**
1882  * Notify core about a preference we have for the given peer
1883  * (to allocate more resources towards it).  The change will
1884  * be communicated the next time we reserve bandwidth with
1885  * core (not instantly).
1886  *
1887  * @param cp peer to reserve bandwidth from
1888  * @param pref preference change
1889  */
1890 void
1891 GSF_connected_peer_change_preference_ (struct GSF_ConnectedPeer *cp,
1892                                        uint64_t pref)
1893 {
1894   cp->inc_preference += pref;
1895 }
1896
1897
1898 /**
1899  * Call this method periodically to flush respect information to disk.
1900  *
1901  * @param cls closure, not used
1902  * @param tc task context, not used
1903  */
1904 static void
1905 cron_flush_respect (void *cls,
1906                     const struct GNUNET_SCHEDULER_TaskContext *tc)
1907 {
1908   if (NULL == cp_map)
1909     return;
1910   GNUNET_CONTAINER_multipeermap_iterate (cp_map,
1911                                          &flush_respect, NULL);
1912   if (NULL == tc)
1913     return;
1914   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1915     return;
1916   GNUNET_SCHEDULER_add_delayed_with_priority (RESPECT_FLUSH_FREQ,
1917                                               GNUNET_SCHEDULER_PRIORITY_HIGH,
1918                                               &cron_flush_respect, NULL);
1919 }
1920
1921
1922 /**
1923  * Initialize peer management subsystem.
1924  */
1925 void
1926 GSF_connected_peer_init_ ()
1927 {
1928   cp_map = GNUNET_CONTAINER_multipeermap_create (128, GNUNET_YES);
1929   peerstore = GNUNET_PEERSTORE_connect (GSF_cfg);
1930   GNUNET_SCHEDULER_add_with_priority (GNUNET_SCHEDULER_PRIORITY_HIGH,
1931                                       &cron_flush_respect, NULL);
1932 }
1933
1934
1935 /**
1936  * Iterator to free peer entries.
1937  *
1938  * @param cls closure, unused
1939  * @param key current key code
1940  * @param value value in the hash map (peer entry)
1941  * @return #GNUNET_YES (we should continue to iterate)
1942  */
1943 static int
1944 clean_peer (void *cls,
1945             const struct GNUNET_PeerIdentity *key,
1946             void *value)
1947 {
1948   GSF_peer_disconnect_handler_ (NULL, key);
1949   return GNUNET_YES;
1950 }
1951
1952
1953 /**
1954  * Shutdown peer management subsystem.
1955  */
1956 void
1957 GSF_connected_peer_done_ ()
1958 {
1959   cron_flush_respect (NULL, NULL);
1960   GNUNET_CONTAINER_multipeermap_iterate (cp_map,
1961                                          &clean_peer, NULL);
1962   GNUNET_CONTAINER_multipeermap_destroy (cp_map);
1963   cp_map = NULL;
1964   GNUNET_PEERSTORE_disconnect (peerstore, GNUNET_YES);
1965 }
1966
1967
1968 /**
1969  * Iterator to remove references to LC entry.
1970  *
1971  * @param cls the `struct GSF_LocalClient *` to look for
1972  * @param key current key code
1973  * @param value value in the hash map (peer entry)
1974  * @return #GNUNET_YES (we should continue to iterate)
1975  */
1976 static int
1977 clean_local_client (void *cls,
1978                     const struct GNUNET_PeerIdentity *key,
1979                     void *value)
1980 {
1981   const struct GSF_LocalClient *lc = cls;
1982   struct GSF_ConnectedPeer *cp = value;
1983   unsigned int i;
1984
1985   for (i = 0; i < CS2P_SUCCESS_LIST_SIZE; i++)
1986     if (cp->ppd.last_client_replies[i] == lc)
1987       cp->ppd.last_client_replies[i] = NULL;
1988   return GNUNET_YES;
1989 }
1990
1991
1992 /**
1993  * Notification that a local client disconnected.  Clean up all of our
1994  * references to the given handle.
1995  *
1996  * @param lc handle to the local client (henceforth invalid)
1997  */
1998 void
1999 GSF_handle_local_client_disconnect_ (const struct GSF_LocalClient *lc)
2000 {
2001   if (NULL == cp_map)
2002     return;                     /* already cleaned up */
2003   GNUNET_CONTAINER_multipeermap_iterate (cp_map, &clean_local_client,
2004                                          (void *) lc);
2005 }
2006
2007
2008 /* end of gnunet-service-fs_cp.c */