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