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