-reduce calls in get_latest()
[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., 51 Franklin Street, Fifth Floor,
18      Boston, MA 02110-1301, 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",
662                                 peer, "respect",
663                                 GNUNET_TIME_UNIT_FOREVER_REL,
664                                 &peer_respect_cb,
665                                 cp);
666 }
667
668
669 /**
670  * It may be time to re-start migrating content to this
671  * peer.  Check, and if so, restart migration.
672  *
673  * @param cls the `struct GSF_ConnectedPeer`
674  * @param tc scheduler context
675  */
676 static void
677 revive_migration (void *cls,
678                   const struct GNUNET_SCHEDULER_TaskContext *tc)
679 {
680   struct GSF_ConnectedPeer *cp = cls;
681   struct GNUNET_TIME_Relative bt;
682
683   cp->mig_revive_task = NULL;
684   bt = GNUNET_TIME_absolute_get_remaining (cp->ppd.migration_blocked_until);
685   if (0 != bt.rel_value_us)
686   {
687     /* still time left... */
688     cp->mig_revive_task =
689         GNUNET_SCHEDULER_add_delayed (bt, &revive_migration, cp);
690     return;
691   }
692   GSF_push_start_ (cp);
693 }
694
695
696 /**
697  * Get a handle for a connected peer.
698  *
699  * @param peer peer's identity
700  * @return NULL if the peer is not currently connected
701  */
702 struct GSF_ConnectedPeer *
703 GSF_peer_get_ (const struct GNUNET_PeerIdentity *peer)
704 {
705   if (NULL == cp_map)
706     return NULL;
707   return GNUNET_CONTAINER_multipeermap_get (cp_map, peer);
708 }
709
710
711 /**
712  * Handle P2P "MIGRATION_STOP" message.
713  *
714  * @param cls closure, always NULL
715  * @param other the other peer involved (sender or receiver, NULL
716  *        for loopback messages where we are both sender and receiver)
717  * @param message the actual message
718  * @return #GNUNET_OK to keep the connection open,
719  *         #GNUNET_SYSERR to close it (signal serious error)
720  */
721 int
722 GSF_handle_p2p_migration_stop_ (void *cls,
723                                 const struct GNUNET_PeerIdentity *other,
724                                 const struct GNUNET_MessageHeader *message)
725 {
726   struct GSF_ConnectedPeer *cp;
727   const struct MigrationStopMessage *msm;
728   struct GNUNET_TIME_Relative bt;
729
730   msm = (const struct MigrationStopMessage *) message;
731   cp = GSF_peer_get_ (other);
732   if (NULL == cp)
733   {
734     GNUNET_break (0);
735     return GNUNET_OK;
736   }
737   GNUNET_STATISTICS_update (GSF_stats,
738                             gettext_noop ("# migration stop messages received"),
739                             1, GNUNET_NO);
740   bt = GNUNET_TIME_relative_ntoh (msm->duration);
741   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
742               _("Migration of content to peer `%s' blocked for %s\n"),
743               GNUNET_i2s (other),
744               GNUNET_STRINGS_relative_time_to_string (bt, GNUNET_YES));
745   cp->ppd.migration_blocked_until = GNUNET_TIME_relative_to_absolute (bt);
746   if ( (NULL == cp->mig_revive_task) &&
747        (NULL == cp->respect_iterate_req) )
748   {
749     GSF_push_stop_ (cp);
750     cp->mig_revive_task =
751         GNUNET_SCHEDULER_add_delayed (bt,
752                                       &revive_migration, cp);
753   }
754   return GNUNET_OK;
755 }
756
757
758 /**
759  * Copy reply and free put message.
760  *
761  * @param cls the `struct PutMessage`
762  * @param buf_size number of bytes available in @a buf
763  * @param buf where to copy the message, NULL on error (peer disconnect)
764  * @return number of bytes copied to @a buf, can be 0 (without indicating an error)
765  */
766 static size_t
767 copy_reply (void *cls,
768             size_t buf_size,
769             void *buf)
770 {
771   struct PutMessage *pm = cls;
772   size_t size;
773
774   if (NULL != buf)
775   {
776     GNUNET_assert (buf_size >= ntohs (pm->header.size));
777     size = ntohs (pm->header.size);
778     memcpy (buf, pm, size);
779     GNUNET_STATISTICS_update (GSF_stats,
780                               gettext_noop ("# replies transmitted to other peers"),
781                               1,
782                               GNUNET_NO);
783   }
784   else
785   {
786     size = 0;
787     GNUNET_STATISTICS_update (GSF_stats,
788                               gettext_noop ("# replies dropped"),
789                               1,
790                               GNUNET_NO);
791   }
792   GNUNET_free (pm);
793   return size;
794 }
795
796
797 /**
798  * Free resources associated with the given peer request.
799  *
800  * @param peerreq request to free
801  * @param query associated key for the request
802  */
803 static void
804 free_pending_request (struct PeerRequest *peerreq,
805                       const struct GNUNET_HashCode *query)
806 {
807   struct GSF_ConnectedPeer *cp = peerreq->cp;
808
809   if (NULL != peerreq->kill_task)
810   {
811     GNUNET_SCHEDULER_cancel (peerreq->kill_task);
812     peerreq->kill_task = NULL;
813   }
814   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
815                             -1, GNUNET_NO);
816   GNUNET_break (GNUNET_YES ==
817                 GNUNET_CONTAINER_multihashmap_remove (cp->request_map,
818                                                       query,
819                                                       peerreq));
820   GNUNET_free (peerreq);
821 }
822
823
824 /**
825  * Cancel all requests associated with the peer.
826  *
827  * @param cls unused
828  * @param query hash code of the request
829  * @param value the `struct GSF_PendingRequest`
830  * @return #GNUNET_YES (continue to iterate)
831  */
832 static int
833 cancel_pending_request (void *cls,
834                         const struct GNUNET_HashCode *query,
835                         void *value)
836 {
837   struct PeerRequest *peerreq = value;
838   struct GSF_PendingRequest *pr = peerreq->pr;
839   struct GSF_PendingRequestData *prd;
840
841   prd = GSF_pending_request_get_data_ (pr);
842   GSF_pending_request_cancel_ (pr, GNUNET_NO);
843   free_pending_request (peerreq, &prd->query);
844   return GNUNET_OK;
845 }
846
847
848 /**
849  * Free the given request.
850  *
851  * @param cls the request to free
852  * @param tc task context
853  */
854 static void
855 peer_request_destroy (void *cls,
856                       const struct GNUNET_SCHEDULER_TaskContext *tc)
857 {
858   struct PeerRequest *peerreq = cls;
859   struct GSF_PendingRequest *pr = peerreq->pr;
860   struct GSF_PendingRequestData *prd;
861
862   peerreq->kill_task = NULL;
863   prd = GSF_pending_request_get_data_ (pr);
864   cancel_pending_request (NULL, &prd->query, peerreq);
865 }
866
867
868 /**
869  * The artificial delay is over, transmit the message now.
870  *
871  * @param cls the `struct GSF_DelayedHandle` with the message
872  * @param tc scheduler context
873  */
874 static void
875 transmit_delayed_now (void *cls,
876                       const struct GNUNET_SCHEDULER_TaskContext *tc)
877 {
878   struct GSF_DelayedHandle *dh = cls;
879   struct GSF_ConnectedPeer *cp = dh->cp;
880
881   GNUNET_CONTAINER_DLL_remove (cp->delayed_head,
882                                cp->delayed_tail,
883                                dh);
884   cp->delay_queue_size--;
885   if (0 != (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason))
886   {
887     GNUNET_free (dh->pm);
888     GNUNET_free (dh);
889     return;
890   }
891   (void) GSF_peer_transmit_ (cp,
892                              GNUNET_NO,
893                              UINT32_MAX,
894                              REPLY_TIMEOUT,
895                              dh->msize,
896                              &copy_reply,
897                              dh->pm);
898   GNUNET_free (dh);
899 }
900
901
902 /**
903  * Get the randomized delay a response should be subjected to.
904  *
905  * @return desired delay
906  */
907 static struct GNUNET_TIME_Relative
908 get_randomized_delay ()
909 {
910   struct GNUNET_TIME_Relative ret;
911
912   ret =
913       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
914                                      GNUNET_CRYPTO_random_u32
915                                      (GNUNET_CRYPTO_QUALITY_WEAK,
916                                       2 * GSF_avg_latency.rel_value_us + 1));
917 #if INSANE_STATISTICS
918   GNUNET_STATISTICS_update (GSF_stats,
919                             gettext_noop
920                             ("# artificial delays introduced (ms)"),
921                             ret.rel_value_us / 1000LL, GNUNET_NO);
922 #endif
923   return ret;
924 }
925
926
927 /**
928  * Handle a reply to a pending request.  Also called if a request
929  * expires (then with data == NULL).  The handler may be called
930  * many times (depending on the request type), but will not be
931  * called during or after a call to GSF_pending_request_cancel
932  * and will also not be called anymore after a call signalling
933  * expiration.
934  *
935  * @param cls `struct PeerRequest` this is an answer for
936  * @param eval evaluation of the result
937  * @param pr handle to the original pending request
938  * @param reply_anonymity_level anonymity level for the reply, UINT32_MAX for "unknown"
939  * @param expiration when does @a data expire?
940  * @param last_transmission when did we last transmit a request for this block
941  * @param type type of the block
942  * @param data response data, NULL on request expiration
943  * @param data_len number of bytes in @a data
944  */
945 static void
946 handle_p2p_reply (void *cls,
947                   enum GNUNET_BLOCK_EvaluationResult eval,
948                   struct GSF_PendingRequest *pr,
949                   uint32_t reply_anonymity_level,
950                   struct GNUNET_TIME_Absolute expiration,
951                   struct GNUNET_TIME_Absolute last_transmission,
952                   enum GNUNET_BLOCK_Type type,
953                   const void *data,
954                   size_t data_len)
955 {
956   struct PeerRequest *peerreq = cls;
957   struct GSF_ConnectedPeer *cp = peerreq->cp;
958   struct GSF_PendingRequestData *prd;
959   struct PutMessage *pm;
960   size_t msize;
961
962   GNUNET_assert (data_len + sizeof (struct PutMessage) <
963                  GNUNET_SERVER_MAX_MESSAGE_SIZE);
964   GNUNET_assert (peerreq->pr == pr);
965   prd = GSF_pending_request_get_data_ (pr);
966   if (NULL == data)
967   {
968     free_pending_request (peerreq, &prd->query);
969     return;
970   }
971   GNUNET_break (GNUNET_BLOCK_TYPE_ANY != type);
972   if ((prd->type != type) && (GNUNET_BLOCK_TYPE_ANY != prd->type))
973   {
974     GNUNET_STATISTICS_update (GSF_stats,
975                               gettext_noop
976                               ("# replies dropped due to type mismatch"),
977                                 1, GNUNET_NO);
978     return;
979   }
980   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
981               "Transmitting result for query `%s' to peer\n",
982               GNUNET_h2s (&prd->query));
983   GNUNET_STATISTICS_update (GSF_stats,
984                             gettext_noop ("# replies received for other peers"),
985                             1, GNUNET_NO);
986   msize = sizeof (struct PutMessage) + data_len;
987   if (msize >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
988   {
989     GNUNET_break (0);
990     return;
991   }
992   if ((UINT32_MAX != reply_anonymity_level) && (reply_anonymity_level > 1))
993   {
994     if (reply_anonymity_level - 1 > GSF_cover_content_count)
995     {
996       GNUNET_STATISTICS_update (GSF_stats,
997                                 gettext_noop
998                                 ("# replies dropped due to insufficient cover traffic"),
999                                 1, GNUNET_NO);
1000       return;
1001     }
1002     GSF_cover_content_count -= (reply_anonymity_level - 1);
1003   }
1004
1005   pm = GNUNET_malloc (msize);
1006   pm->header.type = htons (GNUNET_MESSAGE_TYPE_FS_PUT);
1007   pm->header.size = htons (msize);
1008   pm->type = htonl (type);
1009   pm->expiration = GNUNET_TIME_absolute_hton (expiration);
1010   memcpy (&pm[1], data, data_len);
1011   if ( (UINT32_MAX != reply_anonymity_level) &&
1012        (0 != reply_anonymity_level) &&
1013        (GNUNET_YES == GSF_enable_randomized_delays) )
1014   {
1015     struct GSF_DelayedHandle *dh;
1016
1017     dh = GNUNET_new (struct GSF_DelayedHandle);
1018     dh->cp = cp;
1019     dh->pm = pm;
1020     dh->msize = msize;
1021     GNUNET_CONTAINER_DLL_insert (cp->delayed_head,
1022                                  cp->delayed_tail,
1023                                  dh);
1024     cp->delay_queue_size++;
1025     dh->delay_task =
1026         GNUNET_SCHEDULER_add_delayed (get_randomized_delay (),
1027                                       &transmit_delayed_now,
1028                                       dh);
1029   }
1030   else
1031   {
1032     (void) GSF_peer_transmit_ (cp,
1033                                GNUNET_NO,
1034                                UINT32_MAX,
1035                                REPLY_TIMEOUT,
1036                                msize,
1037                                &copy_reply,
1038                                pm);
1039   }
1040   if (GNUNET_BLOCK_EVALUATION_OK_LAST != eval)
1041     return;
1042   if (NULL == peerreq->kill_task)
1043   {
1044     GNUNET_STATISTICS_update (GSF_stats,
1045                               gettext_noop
1046                               ("# P2P searches destroyed due to ultimate reply"),
1047                               1, GNUNET_NO);
1048     peerreq->kill_task =
1049         GNUNET_SCHEDULER_add_now (&peer_request_destroy, peerreq);
1050   }
1051 }
1052
1053
1054 /**
1055  * Increase the peer's respect by a value.
1056  *
1057  * @param cp which peer to change the respect value on
1058  * @param value is the int value by which the
1059  *  peer's credit is to be increased or decreased
1060  * @returns the actual change in respect (positive or negative)
1061  */
1062 static int
1063 change_peer_respect (struct GSF_ConnectedPeer *cp, int value)
1064 {
1065   if (0 == value)
1066     return 0;
1067   GNUNET_assert (NULL != cp);
1068   if (value > 0)
1069   {
1070     if (cp->ppd.respect + value < cp->ppd.respect)
1071     {
1072       value = UINT32_MAX - cp->ppd.respect;
1073       cp->ppd.respect = UINT32_MAX;
1074     }
1075     else
1076       cp->ppd.respect += value;
1077   }
1078   else
1079   {
1080     if (cp->ppd.respect < -value)
1081     {
1082       value = -cp->ppd.respect;
1083       cp->ppd.respect = 0;
1084     }
1085     else
1086       cp->ppd.respect += value;
1087   }
1088   return value;
1089 }
1090
1091
1092 /**
1093  * We've received a request with the specified priority.  Bound it
1094  * according to how much we respect the given peer.
1095  *
1096  * @param prio_in requested priority
1097  * @param cp the peer making the request
1098  * @return effective priority
1099  */
1100 static int32_t
1101 bound_priority (uint32_t prio_in, struct GSF_ConnectedPeer *cp)
1102 {
1103 #define N ((double)128.0)
1104   uint32_t ret;
1105   double rret;
1106   int ld;
1107
1108   ld = GSF_test_get_load_too_high_ (0);
1109   if (GNUNET_SYSERR == ld)
1110   {
1111 #if INSANE_STATISTICS
1112     GNUNET_STATISTICS_update (GSF_stats,
1113                               gettext_noop
1114                               ("# requests done for free (low load)"), 1,
1115                               GNUNET_NO);
1116 #endif
1117     return 0;                   /* excess resources */
1118   }
1119   if (prio_in > INT32_MAX)
1120     prio_in = INT32_MAX;
1121   ret = -change_peer_respect (cp, -(int) prio_in);
1122   if (ret > 0)
1123   {
1124     if (ret > GSF_current_priorities + N)
1125       rret = GSF_current_priorities + N;
1126     else
1127       rret = ret;
1128     GSF_current_priorities = (GSF_current_priorities * (N - 1) + rret) / N;
1129   }
1130   if ((GNUNET_YES == ld) && (ret > 0))
1131   {
1132     /* try with charging */
1133     ld = GSF_test_get_load_too_high_ (ret);
1134   }
1135   if (GNUNET_YES == ld)
1136   {
1137     GNUNET_STATISTICS_update (GSF_stats,
1138                               gettext_noop
1139                               ("# request dropped, priority insufficient"), 1,
1140                               GNUNET_NO);
1141     /* undo charge */
1142     change_peer_respect (cp, (int) ret);
1143     return -1;                  /* not enough resources */
1144   }
1145   else
1146   {
1147     GNUNET_STATISTICS_update (GSF_stats,
1148                               gettext_noop
1149                               ("# requests done for a price (normal load)"), 1,
1150                               GNUNET_NO);
1151   }
1152 #undef N
1153   return ret;
1154 }
1155
1156
1157 /**
1158  * The priority level imposes a bound on the maximum
1159  * value for the ttl that can be requested.
1160  *
1161  * @param ttl_in requested ttl
1162  * @param prio given priority
1163  * @return @a ttl_in if @a ttl_in is below the limit,
1164  *         otherwise the ttl-limit for the given @a prio
1165  */
1166 static int32_t
1167 bound_ttl (int32_t ttl_in,
1168            uint32_t prio)
1169 {
1170   unsigned long long allowed;
1171
1172   if (ttl_in <= 0)
1173     return ttl_in;
1174   allowed = ((unsigned long long) prio) * TTL_DECREMENT / 1000;
1175   if (ttl_in > allowed)
1176   {
1177     if (allowed >= (1 << 30))
1178       return 1 << 30;
1179     return allowed;
1180   }
1181   return ttl_in;
1182 }
1183
1184
1185 /**
1186  * Closure for #test_exist_cb().
1187  */
1188 struct TestExistClosure
1189 {
1190
1191   /**
1192    * Priority of the incoming request.
1193    */
1194   int32_t priority;
1195
1196   /**
1197    * Relative TTL of the incoming request.
1198    */
1199   int32_t ttl;
1200
1201   /**
1202    * Type of the incoming request.
1203    */
1204   enum GNUNET_BLOCK_Type type;
1205
1206   /**
1207    * Set to #GNUNET_YES if we are done handling the query.
1208    */
1209   int finished;
1210
1211 };
1212
1213
1214 /**
1215  * Test if the query already exists.  If so, merge it, otherwise
1216  * keep `finished` at #GNUNET_NO.
1217  *
1218  * @param cls our `struct TestExistClosure`
1219  * @param hc the key of the query
1220  * @param value the existing `struct PeerRequest`.
1221  * @return #GNUNET_YES to continue to iterate,
1222  *         #GNUNET_NO if we successfully merged
1223  */
1224 static int
1225 test_exist_cb (void *cls,
1226                const struct GNUNET_HashCode *hc,
1227                void *value)
1228 {
1229   struct TestExistClosure *tec = cls;
1230   struct PeerRequest *peerreq = value;
1231   struct GSF_PendingRequest *pr;
1232   struct GSF_PendingRequestData *prd;
1233
1234   pr = peerreq->pr;
1235   prd = GSF_pending_request_get_data_ (pr);
1236   if (prd->type != tec->type)
1237     return GNUNET_YES;
1238   if (prd->ttl.abs_value_us >=
1239       GNUNET_TIME_absolute_get ().abs_value_us + tec->ttl * 1000LL)
1240   {
1241     /* existing request has higher TTL, drop new one! */
1242     prd->priority += tec->priority;
1243     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1244                 "Have existing request with higher TTL, dropping new request.\n");
1245     GNUNET_STATISTICS_update (GSF_stats,
1246                               gettext_noop
1247                               ("# requests dropped due to higher-TTL request"),
1248                               1, GNUNET_NO);
1249     tec->finished = GNUNET_YES;
1250     return GNUNET_NO;
1251   }
1252   /* existing request has lower TTL, drop old one! */
1253   tec->priority += prd->priority;
1254   GSF_pending_request_cancel_ (pr, GNUNET_YES);
1255   free_pending_request (peerreq,
1256                         hc);
1257   return GNUNET_NO;
1258 }
1259
1260
1261 /**
1262  * Handle P2P "QUERY" message.  Creates the pending request entry
1263  * and sets up all of the data structures to that we will
1264  * process replies properly.  Does not initiate forwarding or
1265  * local database lookups.
1266  *
1267  * @param other the other peer involved (sender or receiver, NULL
1268  *        for loopback messages where we are both sender and receiver)
1269  * @param message the actual message
1270  * @return pending request handle, NULL on error
1271  */
1272 struct GSF_PendingRequest *
1273 GSF_handle_p2p_query_ (const struct GNUNET_PeerIdentity *other,
1274                        const struct GNUNET_MessageHeader *message)
1275 {
1276   struct PeerRequest *peerreq;
1277   struct GSF_PendingRequest *pr;
1278   struct GSF_ConnectedPeer *cp;
1279   struct GSF_ConnectedPeer *cps;
1280   const struct GNUNET_PeerIdentity *target;
1281   enum GSF_PendingRequestOptions options;
1282   uint16_t msize;
1283   const struct GetMessage *gm;
1284   unsigned int bits;
1285   const struct GNUNET_PeerIdentity *opt;
1286   uint32_t bm;
1287   size_t bfsize;
1288   uint32_t ttl_decrement;
1289   struct TestExistClosure tec;
1290   GNUNET_PEER_Id spid;
1291
1292   msize = ntohs (message->size);
1293   if (msize < sizeof (struct GetMessage))
1294   {
1295     GNUNET_break_op (0);
1296     return NULL;
1297   }
1298   GNUNET_STATISTICS_update (GSF_stats,
1299                             gettext_noop
1300                             ("# GET requests received (from other peers)"),
1301                             1,
1302                             GNUNET_NO);
1303   gm = (const struct GetMessage *) message;
1304   tec.type = ntohl (gm->type);
1305   bm = ntohl (gm->hash_bitmap);
1306   bits = 0;
1307   while (bm > 0)
1308   {
1309     if (1 == (bm & 1))
1310       bits++;
1311     bm >>= 1;
1312   }
1313   if (msize < sizeof (struct GetMessage) + bits * sizeof (struct GNUNET_PeerIdentity))
1314   {
1315     GNUNET_break_op (0);
1316     return NULL;
1317   }
1318   opt = (const struct GNUNET_PeerIdentity *) &gm[1];
1319   bfsize = msize - sizeof (struct GetMessage) - bits * sizeof (struct GNUNET_PeerIdentity);
1320   /* bfsize must be power of 2, check! */
1321   if (0 != ((bfsize - 1) & bfsize))
1322   {
1323     GNUNET_break_op (0);
1324     return NULL;
1325   }
1326   GSF_cover_query_count++;
1327   bm = ntohl (gm->hash_bitmap);
1328   bits = 0;
1329   cps = GSF_peer_get_ (other);
1330   if (NULL == cps)
1331   {
1332     /* peer must have just disconnected */
1333     GNUNET_STATISTICS_update (GSF_stats,
1334                               gettext_noop
1335                               ("# requests dropped due to initiator not being connected"),
1336                               1, GNUNET_NO);
1337     return NULL;
1338   }
1339   if (0 != (bm & GET_MESSAGE_BIT_RETURN_TO))
1340     cp = GSF_peer_get_ (&opt[bits++]);
1341   else
1342     cp = cps;
1343   if (NULL == cp)
1344   {
1345     if (0 != (bm & GET_MESSAGE_BIT_RETURN_TO))
1346       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1347                   "Failed to find RETURN-TO peer `%s' in connection set. Dropping query.\n",
1348                   GNUNET_i2s (&opt[bits - 1]));
1349
1350     else
1351       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1352                   "Failed to find peer `%s' in connection set. Dropping query.\n",
1353                   GNUNET_i2s (other));
1354     GNUNET_STATISTICS_update (GSF_stats,
1355                               gettext_noop
1356                               ("# requests dropped due to missing reverse route"),
1357                               1,
1358                               GNUNET_NO);
1359     return NULL;
1360   }
1361   if (cp->ppd.pending_replies + cp->delay_queue_size > MAX_QUEUE_PER_PEER)
1362   {
1363     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1364                 "Peer `%s' has too many replies queued already. Dropping query.\n",
1365                 GNUNET_i2s (other));
1366     GNUNET_STATISTICS_update (GSF_stats,
1367                               gettext_noop ("# requests dropped due to full reply queue"),
1368                               1,
1369                               GNUNET_NO);
1370     return NULL;
1371   }
1372   /* note that we can really only check load here since otherwise
1373    * peers could find out that we are overloaded by not being
1374    * disconnected after sending us a malformed query... */
1375   tec.priority = bound_priority (ntohl (gm->priority),
1376                                  cps);
1377   if (tec.priority < 0)
1378   {
1379     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1380                 "Dropping query from `%s', this peer is too busy.\n",
1381                 GNUNET_i2s (other));
1382     return NULL;
1383   }
1384   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1385               "Received request for `%s' of type %u from peer `%s' with flags %u\n",
1386               GNUNET_h2s (&gm->query),
1387               (unsigned int) tec.type,
1388               GNUNET_i2s (other),
1389               (unsigned int) bm);
1390   target =
1391       (0 !=
1392        (bm & GET_MESSAGE_BIT_TRANSMIT_TO)) ? (&opt[bits++]) : NULL;
1393   options = GSF_PRO_DEFAULTS;
1394   spid = 0;
1395   if ((GNUNET_LOAD_get_load (cp->ppd.transmission_delay) > 3 * (1 + tec.priority))
1396       || (GNUNET_LOAD_get_average (cp->ppd.transmission_delay) >
1397           GNUNET_CONSTANTS_MAX_CORK_DELAY.rel_value_us * 2 +
1398           GNUNET_LOAD_get_average (GSF_rt_entry_lifetime)))
1399   {
1400     /* don't have BW to send to peer, or would likely take longer than we have for it,
1401      * so at best indirect the query */
1402     tec.priority = 0;
1403     options |= GSF_PRO_FORWARD_ONLY;
1404     spid = GNUNET_PEER_intern (other);
1405     GNUNET_assert (0 != spid);
1406   }
1407   tec.ttl = bound_ttl (ntohl (gm->ttl),
1408                        tec.priority);
1409   /* decrement ttl (always) */
1410   ttl_decrement =
1411       2 * TTL_DECREMENT + GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
1412                                                     TTL_DECREMENT);
1413   if ((tec.ttl < 0) && (((int32_t) (tec.ttl - ttl_decrement)) > 0))
1414   {
1415     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1416                 "Dropping query from `%s' due to TTL underflow (%d - %u).\n",
1417                 GNUNET_i2s (other),
1418                 tec.ttl,
1419                 ttl_decrement);
1420     GNUNET_STATISTICS_update (GSF_stats,
1421                               gettext_noop
1422                               ("# requests dropped due TTL underflow"), 1,
1423                               GNUNET_NO);
1424     /* integer underflow => drop (should be very rare)! */
1425     return NULL;
1426   }
1427   tec.ttl -= ttl_decrement;
1428
1429   /* test if the request already exists */
1430   tec.finished = GNUNET_NO;
1431   GNUNET_CONTAINER_multihashmap_get_multiple (cp->request_map,
1432                                               &gm->query,
1433                                               &test_exist_cb,
1434                                               &tec);
1435   if (GNUNET_YES == tec.finished)
1436     return NULL; /* merged into existing request, we're done */
1437
1438   peerreq = GNUNET_new (struct PeerRequest);
1439   peerreq->cp = cp;
1440   pr = GSF_pending_request_create_ (options,
1441                                     tec.type,
1442                                     &gm->query,
1443                                     target,
1444                                     (bfsize > 0)
1445                                     ? (const char *) &opt[bits]
1446                                     : NULL,
1447                                     bfsize,
1448                                     ntohl (gm->filter_mutator),
1449                                     1 /* anonymity */,
1450                                     (uint32_t) tec.priority,
1451                                     tec.ttl,
1452                                     spid,
1453                                     GNUNET_PEER_intern (other),
1454                                     NULL, 0,        /* replies_seen */
1455                                     &handle_p2p_reply, peerreq);
1456   GNUNET_assert (NULL != pr);
1457   peerreq->pr = pr;
1458   GNUNET_break (GNUNET_OK ==
1459                 GNUNET_CONTAINER_multihashmap_put (cp->request_map,
1460                                                    &gm->query,
1461                                                    peerreq,
1462                                                    GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
1463   GNUNET_STATISTICS_update (GSF_stats,
1464                             gettext_noop
1465                             ("# P2P query messages received and processed"), 1,
1466                             GNUNET_NO);
1467   GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
1468                             1, GNUNET_NO);
1469   return pr;
1470 }
1471
1472
1473 /**
1474  * Function called if there has been a timeout trying to satisfy
1475  * a transmission request.
1476  *
1477  * @param cls the `struct GSF_PeerTransmitHandle` of the request
1478  * @param tc scheduler context
1479  */
1480 static void
1481 peer_transmit_timeout (void *cls,
1482                        const struct GNUNET_SCHEDULER_TaskContext *tc)
1483 {
1484   struct GSF_PeerTransmitHandle *pth = cls;
1485   struct GSF_ConnectedPeer *cp;
1486
1487   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1488               "Timeout trying to transmit to other peer\n");
1489   pth->timeout_task = NULL;
1490   cp = pth->cp;
1491   GNUNET_CONTAINER_DLL_remove (cp->pth_head,
1492                                cp->pth_tail,
1493                                pth);
1494   if (GNUNET_YES == pth->is_query)
1495     GNUNET_assert (0 < cp->ppd.pending_queries--);
1496   else if (GNUNET_NO == pth->is_query)
1497     GNUNET_assert (0 < cp->ppd.pending_replies--);
1498   GNUNET_LOAD_update (cp->ppd.transmission_delay,
1499                       UINT64_MAX);
1500   if (NULL != cp->cth)
1501   {
1502     GNUNET_CORE_notify_transmit_ready_cancel (cp->cth);
1503     cp->cth = NULL;
1504   }
1505   pth->gmc (pth->gmc_cls, 0, NULL);
1506   GNUNET_assert (0 == cp->cth_in_progress);
1507   GNUNET_free (pth);
1508 }
1509
1510
1511 /**
1512  * Transmit a message to the given peer as soon as possible.
1513  * If the peer disconnects before the transmission can happen,
1514  * the callback is invoked with a `NULL` @a buffer.
1515  *
1516  * @param cp target peer
1517  * @param is_query is this a query (#GNUNET_YES) or content (#GNUNET_NO) or neither (#GNUNET_SYSERR)
1518  * @param priority how important is this request?
1519  * @param timeout when does this request timeout (call gmc with error)
1520  * @param size number of bytes we would like to send to the peer
1521  * @param gmc function to call to get the message
1522  * @param gmc_cls closure for @a gmc
1523  * @return handle to cancel request
1524  */
1525 struct GSF_PeerTransmitHandle *
1526 GSF_peer_transmit_ (struct GSF_ConnectedPeer *cp,
1527                     int is_query,
1528                     uint32_t priority,
1529                     struct GNUNET_TIME_Relative timeout,
1530                     size_t size,
1531                     GSF_GetMessageCallback gmc, void *gmc_cls)
1532 {
1533   struct GSF_PeerTransmitHandle *pth;
1534   struct GSF_PeerTransmitHandle *pos;
1535   struct GSF_PeerTransmitHandle *prev;
1536
1537   pth = GNUNET_new (struct GSF_PeerTransmitHandle);
1538   pth->transmission_request_start_time = GNUNET_TIME_absolute_get ();
1539   pth->timeout = GNUNET_TIME_relative_to_absolute (timeout);
1540   pth->gmc = gmc;
1541   pth->gmc_cls = gmc_cls;
1542   pth->size = size;
1543   pth->is_query = is_query;
1544   pth->priority = priority;
1545   pth->cp = cp;
1546   /* insertion sort (by priority, descending) */
1547   prev = NULL;
1548   pos = cp->pth_head;
1549   while ((NULL != pos) && (pos->priority > priority))
1550   {
1551     prev = pos;
1552     pos = pos->next;
1553   }
1554   GNUNET_CONTAINER_DLL_insert_after (cp->pth_head,
1555                                      cp->pth_tail,
1556                                      prev,
1557                                      pth);
1558   if (GNUNET_YES == is_query)
1559     cp->ppd.pending_queries++;
1560   else if (GNUNET_NO == is_query)
1561     cp->ppd.pending_replies++;
1562   pth->timeout_task
1563     = GNUNET_SCHEDULER_add_delayed (timeout,
1564                                     &peer_transmit_timeout,
1565                                     pth);
1566   schedule_transmission (pth);
1567   return pth;
1568 }
1569
1570
1571 /**
1572  * Cancel an earlier request for transmission.
1573  *
1574  * @param pth request to cancel
1575  */
1576 void
1577 GSF_peer_transmit_cancel_ (struct GSF_PeerTransmitHandle *pth)
1578 {
1579   struct GSF_ConnectedPeer *cp;
1580
1581   if (NULL != pth->timeout_task)
1582   {
1583     GNUNET_SCHEDULER_cancel (pth->timeout_task);
1584     pth->timeout_task = NULL;
1585   }
1586   cp = pth->cp;
1587   GNUNET_CONTAINER_DLL_remove (cp->pth_head,
1588                                cp->pth_tail,
1589                                pth);
1590   if (GNUNET_YES == pth->is_query)
1591     GNUNET_assert (0 < cp->ppd.pending_queries--);
1592   else if (GNUNET_NO == pth->is_query)
1593     GNUNET_assert (0 < cp->ppd.pending_replies--);
1594   GNUNET_free (pth);
1595 }
1596
1597
1598 /**
1599  * Report on receiving a reply; update the performance record of the given peer.
1600  *
1601  * @param cp responding peer (will be updated)
1602  * @param request_time time at which the original query was transmitted
1603  * @param request_priority priority of the original request
1604  */
1605 void
1606 GSF_peer_update_performance_ (struct GSF_ConnectedPeer *cp,
1607                               struct GNUNET_TIME_Absolute request_time,
1608                               uint32_t request_priority)
1609 {
1610   struct GNUNET_TIME_Relative delay;
1611
1612   delay = GNUNET_TIME_absolute_get_duration (request_time);
1613   cp->ppd.avg_reply_delay.rel_value_us =
1614       (cp->ppd.avg_reply_delay.rel_value_us * (RUNAVG_DELAY_N - 1) +
1615        delay.rel_value_us) / RUNAVG_DELAY_N;
1616   cp->ppd.avg_priority =
1617       (cp->ppd.avg_priority * (RUNAVG_DELAY_N - 1) +
1618        request_priority) / RUNAVG_DELAY_N;
1619 }
1620
1621
1622 /**
1623  * Report on receiving a reply in response to an initiating client.
1624  * Remember that this peer is good for this client.
1625  *
1626  * @param cp responding peer (will be updated)
1627  * @param initiator_client local client on responsible for query
1628  */
1629 void
1630 GSF_peer_update_responder_client_ (struct GSF_ConnectedPeer *cp,
1631                                    struct GSF_LocalClient *initiator_client)
1632 {
1633   cp->ppd.last_client_replies[cp->last_client_replies_woff++ %
1634                               CS2P_SUCCESS_LIST_SIZE] = initiator_client;
1635 }
1636
1637
1638 /**
1639  * Report on receiving a reply in response to an initiating peer.
1640  * Remember that this peer is good for this initiating peer.
1641  *
1642  * @param cp responding peer (will be updated)
1643  * @param initiator_peer other peer responsible for query
1644  */
1645 void
1646 GSF_peer_update_responder_peer_ (struct GSF_ConnectedPeer *cp,
1647                                  const struct GSF_ConnectedPeer *initiator_peer)
1648 {
1649   unsigned int woff;
1650
1651   woff = cp->last_p2p_replies_woff % P2P_SUCCESS_LIST_SIZE;
1652   GNUNET_PEER_change_rc (cp->ppd.last_p2p_replies[woff], -1);
1653   cp->ppd.last_p2p_replies[woff] = initiator_peer->ppd.pid;
1654   GNUNET_PEER_change_rc (initiator_peer->ppd.pid, 1);
1655   cp->last_p2p_replies_woff = (woff + 1) % P2P_SUCCESS_LIST_SIZE;
1656 }
1657
1658
1659 /**
1660  * Write peer-respect information to a file - flush the buffer entry!
1661  *
1662  * @param cls unused
1663  * @param key peer identity
1664  * @param value the `struct GSF_ConnectedPeer` to flush
1665  * @return #GNUNET_OK to continue iteration
1666  */
1667 static int
1668 flush_respect (void *cls,
1669                const struct GNUNET_PeerIdentity *key,
1670                void *value)
1671 {
1672   struct GSF_ConnectedPeer *cp = value;
1673   struct GNUNET_PeerIdentity pid;
1674
1675   if (cp->ppd.respect == cp->disk_respect)
1676     return GNUNET_OK;           /* unchanged */
1677   GNUNET_assert (0 != cp->ppd.pid);
1678   GNUNET_PEER_resolve (cp->ppd.pid, &pid);
1679   GNUNET_PEERSTORE_store (peerstore, "fs", &pid, "respect", &cp->ppd.respect,
1680                           sizeof (cp->ppd.respect),
1681                           GNUNET_TIME_UNIT_FOREVER_ABS,
1682                           GNUNET_PEERSTORE_STOREOPTION_REPLACE, NULL, NULL);
1683   return GNUNET_OK;
1684 }
1685
1686
1687 /**
1688  * A peer disconnected from us.  Tear down the connected peer
1689  * record.
1690  *
1691  * @param cls unused
1692  * @param peer identity of peer that connected
1693  */
1694 void
1695 GSF_peer_disconnect_handler_ (void *cls,
1696                               const struct GNUNET_PeerIdentity *peer)
1697 {
1698   struct GSF_ConnectedPeer *cp;
1699   struct GSF_PeerTransmitHandle *pth;
1700   struct GSF_DelayedHandle *dh;
1701
1702   cp = GSF_peer_get_ (peer);
1703   if (NULL == cp)
1704     return;                     /* must have been disconnect from core with
1705                                  * 'peer' == my_id, ignore */
1706   flush_respect (NULL, peer, cp);
1707   GNUNET_assert (GNUNET_YES ==
1708                  GNUNET_CONTAINER_multipeermap_remove (cp_map,
1709                                                        peer,
1710                                                        cp));
1711   GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# peers connected"),
1712                          GNUNET_CONTAINER_multipeermap_size (cp_map),
1713                          GNUNET_NO);
1714   if (NULL != cp->respect_iterate_req)
1715   {
1716     GNUNET_PEERSTORE_iterate_cancel (cp->respect_iterate_req);
1717     cp->respect_iterate_req = NULL;
1718   }
1719   if (NULL != cp->migration_pth)
1720   {
1721     GSF_peer_transmit_cancel_ (cp->migration_pth);
1722     cp->migration_pth = NULL;
1723   }
1724   if (NULL != cp->rc)
1725   {
1726     GNUNET_ATS_reserve_bandwidth_cancel (cp->rc);
1727     cp->rc = NULL;
1728   }
1729   if (NULL != cp->rc_delay_task)
1730   {
1731     GNUNET_SCHEDULER_cancel (cp->rc_delay_task);
1732     cp->rc_delay_task = NULL;
1733   }
1734   GNUNET_CONTAINER_multihashmap_iterate (cp->request_map,
1735                                          &cancel_pending_request,
1736                                          cp);
1737   GNUNET_CONTAINER_multihashmap_destroy (cp->request_map);
1738   cp->request_map = NULL;
1739   GSF_plan_notify_peer_disconnect_ (cp);
1740   GNUNET_LOAD_value_free (cp->ppd.transmission_delay);
1741   GNUNET_PEER_decrement_rcs (cp->ppd.last_p2p_replies, P2P_SUCCESS_LIST_SIZE);
1742   memset (cp->ppd.last_p2p_replies, 0, sizeof (cp->ppd.last_p2p_replies));
1743   GSF_push_stop_ (cp);
1744   if (NULL != cp->cth)
1745   {
1746     GNUNET_CORE_notify_transmit_ready_cancel (cp->cth);
1747     cp->cth = NULL;
1748   }
1749   GNUNET_assert (0 == cp->cth_in_progress);
1750   while (NULL != (pth = cp->pth_head))
1751   {
1752     if (pth->timeout_task != NULL)
1753     {
1754       GNUNET_SCHEDULER_cancel (pth->timeout_task);
1755       pth->timeout_task = NULL;
1756     }
1757     GNUNET_CONTAINER_DLL_remove (cp->pth_head,
1758                                  cp->pth_tail,
1759                                  pth);
1760     if (GNUNET_YES == pth->is_query)
1761       GNUNET_assert (0 < cp->ppd.pending_queries--);
1762     else if (GNUNET_NO == pth->is_query)
1763       GNUNET_assert (0 < cp->ppd.pending_replies--);
1764     pth->gmc (pth->gmc_cls, 0, NULL);
1765     GNUNET_free (pth);
1766   }
1767   while (NULL != (dh = cp->delayed_head))
1768   {
1769     GNUNET_CONTAINER_DLL_remove (cp->delayed_head,
1770                                  cp->delayed_tail,
1771                                  dh);
1772     cp->delay_queue_size--;
1773     GNUNET_SCHEDULER_cancel (dh->delay_task);
1774     GNUNET_free (dh->pm);
1775     GNUNET_free (dh);
1776   }
1777   GNUNET_PEER_change_rc (cp->ppd.pid, -1);
1778   if (NULL != cp->mig_revive_task)
1779   {
1780     GNUNET_SCHEDULER_cancel (cp->mig_revive_task);
1781     cp->mig_revive_task = NULL;
1782   }
1783   GNUNET_break (0 == cp->ppd.pending_queries);
1784   GNUNET_break (0 == cp->ppd.pending_replies);
1785   GNUNET_free (cp);
1786 }
1787
1788
1789 /**
1790  * Closure for #call_iterator().
1791  */
1792 struct IterationContext
1793 {
1794   /**
1795    * Function to call on each entry.
1796    */
1797   GSF_ConnectedPeerIterator it;
1798
1799   /**
1800    * Closure for @e it.
1801    */
1802   void *it_cls;
1803 };
1804
1805
1806 /**
1807  * Function that calls the callback for each peer.
1808  *
1809  * @param cls the `struct IterationContext *`
1810  * @param key identity of the peer
1811  * @param value the `struct GSF_ConnectedPeer *`
1812  * @return #GNUNET_YES to continue iteration
1813  */
1814 static int
1815 call_iterator (void *cls,
1816                const struct GNUNET_PeerIdentity *key,
1817                void *value)
1818 {
1819   struct IterationContext *ic = cls;
1820   struct GSF_ConnectedPeer *cp = value;
1821
1822   ic->it (ic->it_cls,
1823           key, cp,
1824           &cp->ppd);
1825   return GNUNET_YES;
1826 }
1827
1828
1829 /**
1830  * Iterate over all connected peers.
1831  *
1832  * @param it function to call for each peer
1833  * @param it_cls closure for @a it
1834  */
1835 void
1836 GSF_iterate_connected_peers_ (GSF_ConnectedPeerIterator it,
1837                               void *it_cls)
1838 {
1839   struct IterationContext ic;
1840
1841   ic.it = it;
1842   ic.it_cls = it_cls;
1843   GNUNET_CONTAINER_multipeermap_iterate (cp_map,
1844                                          &call_iterator,
1845                                          &ic);
1846 }
1847
1848
1849 /**
1850  * Obtain the identity of a connected peer.
1851  *
1852  * @param cp peer to get identity of
1853  * @param id identity to set (written to)
1854  */
1855 void
1856 GSF_connected_peer_get_identity_ (const struct GSF_ConnectedPeer *cp,
1857                                   struct GNUNET_PeerIdentity *id)
1858 {
1859   GNUNET_assert (0 != cp->ppd.pid);
1860   GNUNET_PEER_resolve (cp->ppd.pid, id);
1861 }
1862
1863
1864 /**
1865  * Obtain the identity of a connected peer.
1866  *
1867  * @param cp peer to get identity of
1868  * @return reference to peer identity, valid until peer disconnects (!)
1869  */
1870 const struct GNUNET_PeerIdentity *
1871 GSF_connected_peer_get_identity2_ (const struct GSF_ConnectedPeer *cp)
1872 {
1873   GNUNET_assert (0 != cp->ppd.pid);
1874   return GNUNET_PEER_resolve2 (cp->ppd.pid);
1875 }
1876
1877
1878 /**
1879  * Assemble a migration stop message for transmission.
1880  *
1881  * @param cls the `struct GSF_ConnectedPeer` to use
1882  * @param size number of bytes we're allowed to write to @a buf
1883  * @param buf where to copy the message
1884  * @return number of bytes copied to @a buf
1885  */
1886 static size_t
1887 create_migration_stop_message (void *cls,
1888                                size_t size,
1889                                void *buf)
1890 {
1891   struct GSF_ConnectedPeer *cp = cls;
1892   struct MigrationStopMessage msm;
1893
1894   cp->migration_pth = NULL;
1895   if (NULL == buf)
1896     return 0;
1897   GNUNET_assert (size >= sizeof (struct MigrationStopMessage));
1898   msm.header.size = htons (sizeof (struct MigrationStopMessage));
1899   msm.header.type = htons (GNUNET_MESSAGE_TYPE_FS_MIGRATION_STOP);
1900   msm.reserved = htonl (0);
1901   msm.duration =
1902       GNUNET_TIME_relative_hton (GNUNET_TIME_absolute_get_remaining
1903                                  (cp->last_migration_block));
1904   memcpy (buf, &msm, sizeof (struct MigrationStopMessage));
1905   GNUNET_STATISTICS_update (GSF_stats,
1906                             gettext_noop ("# migration stop messages sent"),
1907                             1, GNUNET_NO);
1908   return sizeof (struct MigrationStopMessage);
1909 }
1910
1911
1912 /**
1913  * Ask a peer to stop migrating data to us until the given point
1914  * in time.
1915  *
1916  * @param cp peer to ask
1917  * @param block_time until when to block
1918  */
1919 void
1920 GSF_block_peer_migration_ (struct GSF_ConnectedPeer *cp,
1921                            struct GNUNET_TIME_Absolute block_time)
1922 {
1923   if (cp->last_migration_block.abs_value_us > block_time.abs_value_us)
1924   {
1925     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1926                 "Migration already blocked for another %s\n",
1927                 GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_remaining
1928                                                         (cp->last_migration_block), GNUNET_YES));
1929     return;                     /* already blocked */
1930   }
1931   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Asking to stop migration for %s\n",
1932               GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_remaining (block_time),
1933                                                       GNUNET_YES));
1934   cp->last_migration_block = block_time;
1935   if (NULL != cp->migration_pth)
1936     GSF_peer_transmit_cancel_ (cp->migration_pth);
1937   cp->migration_pth =
1938       GSF_peer_transmit_ (cp, GNUNET_SYSERR, UINT32_MAX,
1939                           GNUNET_TIME_UNIT_FOREVER_REL,
1940                           sizeof (struct MigrationStopMessage),
1941                           &create_migration_stop_message, cp);
1942 }
1943
1944
1945 /**
1946  * Notify core about a preference we have for the given peer
1947  * (to allocate more resources towards it).  The change will
1948  * be communicated the next time we reserve bandwidth with
1949  * core (not instantly).
1950  *
1951  * @param cp peer to reserve bandwidth from
1952  * @param pref preference change
1953  */
1954 void
1955 GSF_connected_peer_change_preference_ (struct GSF_ConnectedPeer *cp,
1956                                        uint64_t pref)
1957 {
1958   cp->inc_preference += pref;
1959 }
1960
1961
1962 /**
1963  * Call this method periodically to flush respect information to disk.
1964  *
1965  * @param cls closure, not used
1966  * @param tc task context, not used
1967  */
1968 static void
1969 cron_flush_respect (void *cls,
1970                     const struct GNUNET_SCHEDULER_TaskContext *tc)
1971 {
1972   if (NULL == cp_map)
1973     return;
1974   GNUNET_CONTAINER_multipeermap_iterate (cp_map,
1975                                          &flush_respect, NULL);
1976   if (NULL == tc)
1977     return;
1978   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1979     return;
1980   GNUNET_SCHEDULER_add_delayed_with_priority (RESPECT_FLUSH_FREQ,
1981                                               GNUNET_SCHEDULER_PRIORITY_HIGH,
1982                                               &cron_flush_respect, NULL);
1983 }
1984
1985
1986 /**
1987  * Initialize peer management subsystem.
1988  */
1989 void
1990 GSF_connected_peer_init_ ()
1991 {
1992   cp_map = GNUNET_CONTAINER_multipeermap_create (128, GNUNET_YES);
1993   peerstore = GNUNET_PEERSTORE_connect (GSF_cfg);
1994   GNUNET_SCHEDULER_add_with_priority (GNUNET_SCHEDULER_PRIORITY_HIGH,
1995                                       &cron_flush_respect, NULL);
1996 }
1997
1998
1999 /**
2000  * Iterator to free peer entries.
2001  *
2002  * @param cls closure, unused
2003  * @param key current key code
2004  * @param value value in the hash map (peer entry)
2005  * @return #GNUNET_YES (we should continue to iterate)
2006  */
2007 static int
2008 clean_peer (void *cls,
2009             const struct GNUNET_PeerIdentity *key,
2010             void *value)
2011 {
2012   GSF_peer_disconnect_handler_ (NULL, key);
2013   return GNUNET_YES;
2014 }
2015
2016
2017 /**
2018  * Shutdown peer management subsystem.
2019  */
2020 void
2021 GSF_connected_peer_done_ ()
2022 {
2023   cron_flush_respect (NULL, NULL);
2024   GNUNET_CONTAINER_multipeermap_iterate (cp_map,
2025                                          &clean_peer, NULL);
2026   GNUNET_CONTAINER_multipeermap_destroy (cp_map);
2027   cp_map = NULL;
2028   GNUNET_PEERSTORE_disconnect (peerstore, GNUNET_YES);
2029 }
2030
2031
2032 /**
2033  * Iterator to remove references to LC entry.
2034  *
2035  * @param cls the `struct GSF_LocalClient *` to look for
2036  * @param key current key code
2037  * @param value value in the hash map (peer entry)
2038  * @return #GNUNET_YES (we should continue to iterate)
2039  */
2040 static int
2041 clean_local_client (void *cls,
2042                     const struct GNUNET_PeerIdentity *key,
2043                     void *value)
2044 {
2045   const struct GSF_LocalClient *lc = cls;
2046   struct GSF_ConnectedPeer *cp = value;
2047   unsigned int i;
2048
2049   for (i = 0; i < CS2P_SUCCESS_LIST_SIZE; i++)
2050     if (cp->ppd.last_client_replies[i] == lc)
2051       cp->ppd.last_client_replies[i] = NULL;
2052   return GNUNET_YES;
2053 }
2054
2055
2056 /**
2057  * Notification that a local client disconnected.  Clean up all of our
2058  * references to the given handle.
2059  *
2060  * @param lc handle to the local client (henceforth invalid)
2061  */
2062 void
2063 GSF_handle_local_client_disconnect_ (const struct GSF_LocalClient *lc)
2064 {
2065   if (NULL == cp_map)
2066     return;                     /* already cleaned up */
2067   GNUNET_CONTAINER_multipeermap_iterate (cp_map, &clean_local_client,
2068                                          (void *) lc);
2069 }
2070
2071
2072 /* end of gnunet-service-fs_cp.c */