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