2 This file is part of GNUnet.
3 (C) 2011 Christian Grothoff (and other contributing authors)
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.
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.
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.
22 * @file fs/gnunet-service-fs_cp.c
23 * @brief API to handle 'connected peers'
24 * @author Christian Grothoff
27 #include "gnunet_load_lib.h"
28 #include "gnunet_ats_service.h"
29 #include "gnunet-service-fs.h"
30 #include "gnunet-service-fs_cp.h"
31 #include "gnunet-service-fs_pe.h"
32 #include "gnunet-service-fs_pr.h"
33 #include "gnunet-service-fs_push.h"
36 * How often do we flush trust values to disk?
38 #define TRUST_FLUSH_FREQ GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 5)
41 * After how long do we discard a reply?
43 #define REPLY_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 2)
47 * Handle to cancel a transmission request.
49 struct GSF_PeerTransmitHandle
53 * Kept in a doubly-linked list.
55 struct GSF_PeerTransmitHandle *next;
58 * Kept in a doubly-linked list.
60 struct GSF_PeerTransmitHandle *prev;
63 * Handle for an active request for transmission to this
64 * peer, or NULL (if core queue was full).
66 struct GNUNET_CORE_TransmitHandle *cth;
69 * Time when this transmission request was issued.
71 struct GNUNET_TIME_Absolute transmission_request_start_time;
74 * Timeout for this request.
76 struct GNUNET_TIME_Absolute timeout;
79 * Task called on timeout, or 0 for none.
81 GNUNET_SCHEDULER_TaskIdentifier timeout_task;
84 * Function to call to get the actual message.
86 GSF_GetMessageCallback gmc;
89 * Peer this request targets.
91 struct GSF_ConnectedPeer *cp;
99 * Size of the message to be transmitted.
104 * Set to 1 if we're currently in the process of calling
105 * 'GNUNET_CORE_notify_transmit_ready' (so while cth is
106 * NULL, we should not call notify_transmit_ready for this
109 unsigned int cth_in_progress;
112 * GNUNET_YES if this is a query, GNUNET_NO for content.
117 * Did we get a reservation already?
122 * Priority of this request.
130 * Handle for an entry in our delay list.
132 struct GSF_DelayedHandle
136 * Kept in a doubly-linked list.
138 struct GSF_DelayedHandle *next;
141 * Kept in a doubly-linked list.
143 struct GSF_DelayedHandle *prev;
146 * Peer this transmission belongs to.
148 struct GSF_ConnectedPeer *cp;
151 * The PUT that was delayed.
153 struct PutMessage *pm;
156 * Task for the delay.
158 GNUNET_SCHEDULER_TaskIdentifier delay_task;
161 * Size of the message.
169 * Information per peer and request.
175 * Handle to generic request.
177 struct GSF_PendingRequest *pr;
180 * Handle to specific peer.
182 struct GSF_ConnectedPeer *cp;
185 * Task for asynchronous stopping of this request.
187 GNUNET_SCHEDULER_TaskIdentifier kill_task;
195 struct GSF_ConnectedPeer
199 * Performance data for this peer.
201 struct GSF_PeerPerformanceData ppd;
204 * Time until when we blocked this peer from migrating
207 struct GNUNET_TIME_Absolute last_migration_block;
210 * Task scheduled to revive migration to this peer.
212 GNUNET_SCHEDULER_TaskIdentifier mig_revive_task;
215 * Messages (replies, queries, content migration) we would like to
216 * send to this peer in the near future. Sorted by priority, head.
218 struct GSF_PeerTransmitHandle *pth_head;
221 * Messages (replies, queries, content migration) we would like to
222 * send to this peer in the near future. Sorted by priority, tail.
224 struct GSF_PeerTransmitHandle *pth_tail;
227 * Messages (replies, queries, content migration) we would like to
228 * send to this peer in the near future. Sorted by priority, head.
230 struct GSF_DelayedHandle *delayed_head;
233 * Messages (replies, queries, content migration) we would like to
234 * send to this peer in the near future. Sorted by priority, tail.
236 struct GSF_DelayedHandle *delayed_tail;
239 * Migration stop message in our queue, or NULL if we have none pending.
241 struct GSF_PeerTransmitHandle *migration_pth;
244 * Context of our GNUNET_ATS_reserve_bandwidth call (or NULL).
246 struct GNUNET_ATS_ReservationContext *rc;
249 * Task scheduled if we need to retry bandwidth reservation later.
251 GNUNET_SCHEDULER_TaskIdentifier rc_delay_task;
254 * Active requests from this neighbour, map of query to 'struct PeerRequest'.
256 struct GNUNET_CONTAINER_MultiHashMap *request_map;
259 * Increase in traffic preference still to be submitted
260 * to the core service for this peer.
262 uint64_t inc_preference;
265 * Trust rating for this peer on disk.
270 * Which offset in "last_p2p_replies" will be updated next?
271 * (we go round-robin).
273 unsigned int last_p2p_replies_woff;
276 * Which offset in "last_client_replies" will be updated next?
277 * (we go round-robin).
279 unsigned int last_client_replies_woff;
282 * Current offset into 'last_request_times' ring buffer.
284 unsigned int last_request_times_off;
287 * GNUNET_YES if we did successfully reserve 32k bandwidth,
296 * Map from peer identities to 'struct GSF_ConnectPeer' entries.
298 static struct GNUNET_CONTAINER_MultiHashMap *cp_map;
301 * Where do we store trust information?
303 static char *trustDirectory;
306 * Handle to ATS service.
308 static struct GNUNET_ATS_PerformanceHandle *ats;
311 * Get the filename under which we would store the GNUNET_HELLO_Message
312 * for the given host and protocol.
313 * @return filename of the form DIRECTORY/HOSTID
316 get_trust_filename (const struct GNUNET_PeerIdentity *id)
318 struct GNUNET_CRYPTO_HashAsciiEncoded fil;
321 GNUNET_CRYPTO_hash_to_enc (&id->hashPubKey, &fil);
322 GNUNET_asprintf (&fn, "%s%s%s", trustDirectory, DIR_SEPARATOR_STR, &fil);
328 * Find latency information in 'atsi'.
330 * @param atsi performance data
331 * @param atsi_count number of records in 'atsi'
332 * @return connection latency
334 static struct GNUNET_TIME_Relative
335 get_latency (const struct GNUNET_ATS_Information *atsi, unsigned int atsi_count)
339 for (i = 0; i < atsi_count; i++)
340 if (ntohl (atsi->type) == GNUNET_ATS_QUALITY_NET_DELAY)
341 return GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
342 ntohl (atsi->value));
343 return GNUNET_TIME_UNIT_SECONDS;
348 * Update the performance information kept for the given peer.
350 * @param cp peer record to update
351 * @param atsi transport performance data
352 * @param atsi_count number of records in 'atsi'
355 update_atsi (struct GSF_ConnectedPeer *cp,
356 const struct GNUNET_ATS_Information *atsi, unsigned int atsi_count)
358 struct GNUNET_TIME_Relative latency;
360 latency = get_latency (atsi, atsi_count);
361 GNUNET_LOAD_value_set_decline (cp->ppd.transmission_delay, latency);
362 /* LATER: merge atsi into cp's performance data (if we ever care...) */
367 * Return the performance data record for the given peer
369 * @param cp peer to query
370 * @return performance data record for the peer
372 struct GSF_PeerPerformanceData *
373 GSF_get_peer_performance_data_ (struct GSF_ConnectedPeer *cp)
380 * Core is ready to transmit to a peer, get the message.
382 * @param cls the 'struct GSF_PeerTransmitHandle' of the message
383 * @param size number of bytes core is willing to take
384 * @param buf where to copy the message
385 * @return number of bytes copied to buf
388 peer_transmit_ready_cb (void *cls, size_t size, void *buf);
392 * Function called by core upon success or failure of our bandwidth reservation request.
394 * @param cls the 'struct GSF_ConnectedPeer' of the peer for which we made the request
395 * @param peer identifies the peer
396 * @param amount set to the amount that was actually reserved or unreserved;
397 * either the full requested amount or zero (no partial reservations)
398 * @param res_delay if the reservation could not be satisfied (amount was 0), how
399 * long should the client wait until re-trying?
402 ats_reserve_callback (void *cls, const struct GNUNET_PeerIdentity *peer,
403 int32_t amount, struct GNUNET_TIME_Relative res_delay);
407 * If ready (bandwidth reserved), try to schedule transmission via
408 * core for the given handle.
410 * @param pth transmission handle to schedule
413 schedule_transmission (struct GSF_PeerTransmitHandle *pth)
415 struct GSF_ConnectedPeer *cp;
416 struct GNUNET_PeerIdentity target;
418 if ((NULL != pth->cth) || (0 != pth->cth_in_progress))
419 return; /* already done */
421 GNUNET_assert (0 != cp->ppd.pid);
422 GNUNET_PEER_resolve (cp->ppd.pid, &target);
424 if (0 != cp->inc_preference)
426 GNUNET_ATS_change_preference (ats, &target, GNUNET_ATS_PREFERENCE_BANDWIDTH,
427 (double) cp->inc_preference,
428 GNUNET_ATS_PREFERENCE_END);
429 cp->inc_preference = 0;
432 if ((GNUNET_YES == pth->is_query) && (GNUNET_YES != pth->was_reserved))
434 /* query, need reservation */
435 if (GNUNET_YES != cp->did_reserve)
436 return; /* not ready */
437 cp->did_reserve = GNUNET_NO;
438 /* reservation already done! */
439 pth->was_reserved = GNUNET_YES;
441 GNUNET_ATS_reserve_bandwidth (ats, &target, DBLOCK_SIZE,
442 &ats_reserve_callback, cp);
444 GNUNET_assert (pth->cth == NULL);
445 pth->cth_in_progress++;
447 GNUNET_CORE_notify_transmit_ready (GSF_core, GNUNET_YES, pth->priority,
448 GNUNET_TIME_absolute_get_remaining
449 (pth->timeout), &target, pth->size,
450 &peer_transmit_ready_cb, pth);
451 GNUNET_assert (0 < pth->cth_in_progress--);
456 * Core is ready to transmit to a peer, get the message.
458 * @param cls the 'struct GSF_PeerTransmitHandle' of the message
459 * @param size number of bytes core is willing to take
460 * @param buf where to copy the message
461 * @return number of bytes copied to buf
464 peer_transmit_ready_cb (void *cls, size_t size, void *buf)
466 struct GSF_PeerTransmitHandle *pth = cls;
467 struct GSF_PeerTransmitHandle *pos;
468 struct GSF_ConnectedPeer *cp;
471 GNUNET_assert ((NULL == buf) || (pth->size <= size));
473 if (pth->timeout_task != GNUNET_SCHEDULER_NO_TASK)
475 GNUNET_SCHEDULER_cancel (pth->timeout_task);
476 pth->timeout_task = GNUNET_SCHEDULER_NO_TASK;
479 GNUNET_CONTAINER_DLL_remove (cp->pth_head, cp->pth_tail, pth);
480 if (GNUNET_YES == pth->is_query)
482 cp->ppd.last_request_times[(cp->last_request_times_off++) %
483 MAX_QUEUE_PER_PEER] =
484 GNUNET_TIME_absolute_get ();
485 GNUNET_assert (0 < cp->ppd.pending_queries--);
487 else if (GNUNET_NO == pth->is_query)
489 GNUNET_assert (0 < cp->ppd.pending_replies--);
491 GNUNET_LOAD_update (cp->ppd.transmission_delay,
492 GNUNET_TIME_absolute_get_duration
493 (pth->transmission_request_start_time).rel_value);
494 ret = pth->gmc (pth->gmc_cls, size, buf);
495 GNUNET_assert (NULL == pth->cth);
496 for (pos = cp->pth_head; pos != NULL; pos = pos->next)
498 GNUNET_assert (pos != pth);
499 schedule_transmission (pos);
501 GNUNET_assert (pth->cth == NULL);
502 GNUNET_assert (pth->cth_in_progress == 0);
509 * (re)try to reserve bandwidth from the given peer.
511 * @param cls the 'struct GSF_ConnectedPeer' to reserve from
512 * @param tc scheduler context
515 retry_reservation (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
517 struct GSF_ConnectedPeer *cp = cls;
518 struct GNUNET_PeerIdentity target;
520 GNUNET_PEER_resolve (cp->ppd.pid, &target);
521 cp->rc_delay_task = GNUNET_SCHEDULER_NO_TASK;
523 GNUNET_ATS_reserve_bandwidth (ats, &target, DBLOCK_SIZE,
524 &ats_reserve_callback, cp);
529 * Function called by core upon success or failure of our bandwidth reservation request.
531 * @param cls the 'struct GSF_ConnectedPeer' of the peer for which we made the request
532 * @param peer identifies the peer
533 * @param amount set to the amount that was actually reserved or unreserved;
534 * either the full requested amount or zero (no partial reservations)
535 * @param res_delay if the reservation could not be satisfied (amount was 0), how
536 * long should the client wait until re-trying?
539 ats_reserve_callback (void *cls, const struct GNUNET_PeerIdentity *peer,
540 int32_t amount, struct GNUNET_TIME_Relative res_delay)
542 struct GSF_ConnectedPeer *cp = cls;
543 struct GSF_PeerTransmitHandle *pth;
545 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
546 "Reserved %d bytes / need to wait %llu ms for reservation\n",
547 (int) amount, (unsigned long long) res_delay.rel_value);
552 GNUNET_SCHEDULER_add_delayed (res_delay, &retry_reservation, cp);
555 cp->did_reserve = GNUNET_YES;
557 if ((NULL != pth) && (NULL == pth->cth))
559 /* reservation success, try transmission now! */
560 pth->cth_in_progress++;
562 GNUNET_CORE_notify_transmit_ready (GSF_core, GNUNET_YES, pth->priority,
563 GNUNET_TIME_absolute_get_remaining
564 (pth->timeout), peer, pth->size,
565 &peer_transmit_ready_cb, pth);
566 GNUNET_assert (0 < pth->cth_in_progress--);
572 * A peer connected to us. Setup the connected peer
575 * @param peer identity of peer that connected
576 * @param atsi performance data for the connection
577 * @param atsi_count number of records in 'atsi'
578 * @return handle to connected peer entry
580 struct GSF_ConnectedPeer *
581 GSF_peer_connect_handler_ (const struct GNUNET_PeerIdentity *peer,
582 const struct GNUNET_ATS_Information *atsi,
583 unsigned int atsi_count)
585 struct GSF_ConnectedPeer *cp;
589 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connected to peer %s\n",
591 cp = GNUNET_malloc (sizeof (struct GSF_ConnectedPeer));
592 cp->ppd.pid = GNUNET_PEER_intern (peer);
593 cp->ppd.transmission_delay = GNUNET_LOAD_value_init (GNUNET_TIME_UNIT_ZERO);
595 GNUNET_ATS_reserve_bandwidth (ats, peer, DBLOCK_SIZE,
596 &ats_reserve_callback, cp);
597 fn = get_trust_filename (peer);
598 if ((GNUNET_DISK_file_test (fn) == GNUNET_YES) &&
599 (sizeof (trust) == GNUNET_DISK_fn_read (fn, &trust, sizeof (trust))))
600 cp->disk_trust = cp->ppd.trust = ntohl (trust);
602 cp->request_map = GNUNET_CONTAINER_multihashmap_create (128);
603 GNUNET_break (GNUNET_OK ==
604 GNUNET_CONTAINER_multihashmap_put (cp_map, &peer->hashPubKey,
606 GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
607 GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# peers connected"),
608 GNUNET_CONTAINER_multihashmap_size (cp_map),
610 update_atsi (cp, atsi, atsi_count);
611 GSF_push_start_ (cp);
617 * It may be time to re-start migrating content to this
618 * peer. Check, and if so, restart migration.
620 * @param cls the 'struct GSF_ConnectedPeer'
621 * @param tc scheduler context
624 revive_migration (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
626 struct GSF_ConnectedPeer *cp = cls;
627 struct GNUNET_TIME_Relative bt;
629 cp->mig_revive_task = GNUNET_SCHEDULER_NO_TASK;
630 bt = GNUNET_TIME_absolute_get_remaining (cp->ppd.migration_blocked_until);
631 if (0 != bt.rel_value)
633 /* still time left... */
634 cp->mig_revive_task =
635 GNUNET_SCHEDULER_add_delayed (bt, &revive_migration, cp);
638 GSF_push_start_ (cp);
643 * Get a handle for a connected peer.
645 * @param peer peer's identity
646 * @return NULL if the peer is not currently connected
648 struct GSF_ConnectedPeer *
649 GSF_peer_get_ (const struct GNUNET_PeerIdentity *peer)
653 return GNUNET_CONTAINER_multihashmap_get (cp_map, &peer->hashPubKey);
658 * Handle P2P "MIGRATION_STOP" message.
660 * @param cls closure, always NULL
661 * @param other the other peer involved (sender or receiver, NULL
662 * for loopback messages where we are both sender and receiver)
663 * @param message the actual message
664 * @param atsi performance information
665 * @param atsi_count number of records in 'atsi'
666 * @return GNUNET_OK to keep the connection open,
667 * GNUNET_SYSERR to close it (signal serious error)
670 GSF_handle_p2p_migration_stop_ (void *cls,
671 const struct GNUNET_PeerIdentity *other,
672 const struct GNUNET_MessageHeader *message,
673 const struct GNUNET_ATS_Information *atsi,
674 unsigned int atsi_count)
676 struct GSF_ConnectedPeer *cp;
677 const struct MigrationStopMessage *msm;
678 struct GNUNET_TIME_Relative bt;
680 msm = (const struct MigrationStopMessage *) message;
681 cp = GSF_peer_get_ (other);
687 GNUNET_STATISTICS_update (GSF_stats,
688 gettext_noop ("# migration stop messages received"),
690 bt = GNUNET_TIME_relative_ntoh (msm->duration);
691 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
692 _("Migration of content to peer `%s' blocked for %llu ms\n"),
693 GNUNET_i2s (other), (unsigned long long) bt.rel_value);
694 cp->ppd.migration_blocked_until = GNUNET_TIME_relative_to_absolute (bt);
695 if (cp->mig_revive_task == GNUNET_SCHEDULER_NO_TASK)
698 cp->mig_revive_task =
699 GNUNET_SCHEDULER_add_delayed (bt, &revive_migration, cp);
701 update_atsi (cp, atsi, atsi_count);
707 * Copy reply and free put message.
709 * @param cls the 'struct PutMessage'
710 * @param buf_size number of bytes available in buf
711 * @param buf where to copy the message, NULL on error (peer disconnect)
712 * @return number of bytes copied to 'buf', can be 0 (without indicating an error)
715 copy_reply (void *cls, size_t buf_size, void *buf)
717 struct PutMessage *pm = cls;
722 GNUNET_assert (buf_size >= ntohs (pm->header.size));
723 size = ntohs (pm->header.size);
724 memcpy (buf, pm, size);
725 GNUNET_STATISTICS_update (GSF_stats,
727 ("# replies transmitted to other peers"), 1,
733 GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# replies dropped"), 1,
742 * Cancel all requests associated with the peer.
745 * @param query hash code of the request
746 * @param value the 'struct GSF_PendingRequest'
747 * @return GNUNET_YES (continue to iterate)
750 cancel_pending_request (void *cls, const GNUNET_HashCode * query, void *value)
752 struct PeerRequest *peerreq = value;
753 struct GSF_PendingRequest *pr = peerreq->pr;
754 struct GSF_ConnectedPeer *cp = peerreq->cp;
755 struct GSF_PendingRequestData *prd;
757 if (peerreq->kill_task != GNUNET_SCHEDULER_NO_TASK)
759 GNUNET_SCHEDULER_cancel (peerreq->kill_task);
760 peerreq->kill_task = GNUNET_SCHEDULER_NO_TASK;
762 GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
764 prd = GSF_pending_request_get_data_ (pr);
765 GNUNET_break (GNUNET_YES ==
766 GNUNET_CONTAINER_multihashmap_remove (cp->request_map,
767 &prd->query, peerreq));
768 GSF_pending_request_cancel_ (pr, GNUNET_NO);
769 GNUNET_free (peerreq);
775 * Free the given request.
777 * @param cls the request to free
778 * @param tc task context
781 peer_request_destroy (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
783 struct PeerRequest *peerreq = cls;
784 struct GSF_PendingRequest *pr = peerreq->pr;
785 struct GSF_PendingRequestData *prd;
787 peerreq->kill_task = GNUNET_SCHEDULER_NO_TASK;
788 prd = GSF_pending_request_get_data_ (pr);
789 cancel_pending_request (NULL, &prd->query, peerreq);
794 * The artificial delay is over, transmit the message now.
796 * @param cls the 'struct GSF_DelayedHandle' with the message
797 * @param tc scheduler context
800 transmit_delayed_now (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
802 struct GSF_DelayedHandle *dh = cls;
803 struct GSF_ConnectedPeer *cp = dh->cp;
805 GNUNET_CONTAINER_DLL_remove (cp->delayed_head, cp->delayed_tail, dh);
806 if (0 != (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason))
808 GNUNET_free (dh->pm);
812 (void) GSF_peer_transmit_ (cp, GNUNET_NO, UINT32_MAX, REPLY_TIMEOUT,
813 dh->msize, ©_reply, dh->pm);
819 * Get the randomized delay a response should be subjected to.
821 * @return desired delay
823 static struct GNUNET_TIME_Relative
824 get_randomized_delay ()
826 struct GNUNET_TIME_Relative ret;
829 GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
830 GNUNET_CRYPTO_random_u32
831 (GNUNET_CRYPTO_QUALITY_WEAK,
832 2 * GSF_avg_latency.rel_value + 1));
833 GNUNET_STATISTICS_update (GSF_stats,
835 ("# artificial delays introduced (ms)"),
836 ret.rel_value, GNUNET_NO);
843 * Handle a reply to a pending request. Also called if a request
844 * expires (then with data == NULL). The handler may be called
845 * many times (depending on the request type), but will not be
846 * called during or after a call to GSF_pending_request_cancel
847 * and will also not be called anymore after a call signalling
850 * @param cls 'struct PeerRequest' this is an answer for
851 * @param eval evaluation of the result
852 * @param pr handle to the original pending request
853 * @param reply_anonymity_level anonymity level for the reply, UINT32_MAX for "unknown"
854 * @param expiration when does 'data' expire?
855 * @param type type of the block
856 * @param data response data, NULL on request expiration
857 * @param data_len number of bytes in data
860 handle_p2p_reply (void *cls, enum GNUNET_BLOCK_EvaluationResult eval,
861 struct GSF_PendingRequest *pr, uint32_t reply_anonymity_level,
862 struct GNUNET_TIME_Absolute expiration,
863 enum GNUNET_BLOCK_Type type, const void *data,
866 struct PeerRequest *peerreq = cls;
867 struct GSF_ConnectedPeer *cp = peerreq->cp;
868 struct GSF_PendingRequestData *prd;
869 struct PutMessage *pm;
872 GNUNET_assert (data_len + sizeof (struct PutMessage) <
873 GNUNET_SERVER_MAX_MESSAGE_SIZE);
874 GNUNET_assert (peerreq->pr == pr);
875 prd = GSF_pending_request_get_data_ (pr);
878 GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
880 GNUNET_break (GNUNET_YES ==
881 GNUNET_CONTAINER_multihashmap_remove (cp->request_map,
882 &prd->query, peerreq));
883 GNUNET_free (peerreq);
886 GNUNET_break (type != GNUNET_BLOCK_TYPE_ANY);
887 if ((prd->type != type) && (prd->type != GNUNET_BLOCK_TYPE_ANY))
893 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
894 "Transmitting result for query `%s' to peer\n",
895 GNUNET_h2s (&prd->query));
897 GNUNET_STATISTICS_update (GSF_stats,
898 gettext_noop ("# replies received for other peers"),
900 msize = sizeof (struct PutMessage) + data_len;
901 if (msize >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
906 if ((reply_anonymity_level != UINT32_MAX) && (reply_anonymity_level > 1))
908 if (reply_anonymity_level - 1 > GSF_cover_content_count)
910 GNUNET_STATISTICS_update (GSF_stats,
912 ("# replies dropped due to insufficient cover traffic"),
916 GSF_cover_content_count -= (reply_anonymity_level - 1);
919 pm = GNUNET_malloc (msize);
920 pm->header.type = htons (GNUNET_MESSAGE_TYPE_FS_PUT);
921 pm->header.size = htons (msize);
922 pm->type = htonl (type);
923 pm->expiration = GNUNET_TIME_absolute_hton (expiration);
924 memcpy (&pm[1], data, data_len);
925 if ((reply_anonymity_level != UINT32_MAX) && (reply_anonymity_level != 0) &&
926 (GSF_enable_randomized_delays == GNUNET_YES))
928 struct GSF_DelayedHandle *dh;
930 dh = GNUNET_malloc (sizeof (struct GSF_DelayedHandle));
934 GNUNET_CONTAINER_DLL_insert (cp->delayed_head, cp->delayed_tail, dh);
936 GNUNET_SCHEDULER_add_delayed (get_randomized_delay (),
937 &transmit_delayed_now, dh);
941 (void) GSF_peer_transmit_ (cp, GNUNET_NO, UINT32_MAX, REPLY_TIMEOUT, msize,
944 if (eval != GNUNET_BLOCK_EVALUATION_OK_LAST)
946 if (GNUNET_SCHEDULER_NO_TASK == peerreq->kill_task)
948 GNUNET_STATISTICS_update (GSF_stats,
950 ("# P2P searches destroyed due to ultimate reply"),
953 GNUNET_SCHEDULER_add_now (&peer_request_destroy, peerreq);
959 * Increase the host credit by a value.
961 * @param cp which peer to change the trust value on
962 * @param value is the int value by which the
963 * host credit is to be increased or decreased
964 * @returns the actual change in trust (positive or negative)
967 change_host_trust (struct GSF_ConnectedPeer *cp, int value)
971 GNUNET_assert (cp != NULL);
974 if (cp->ppd.trust + value < cp->ppd.trust)
976 value = UINT32_MAX - cp->ppd.trust;
977 cp->ppd.trust = UINT32_MAX;
980 cp->ppd.trust += value;
984 if (cp->ppd.trust < -value)
986 value = -cp->ppd.trust;
990 cp->ppd.trust += value;
997 * We've received a request with the specified priority. Bound it
998 * according to how much we trust the given peer.
1000 * @param prio_in requested priority
1001 * @param cp the peer making the request
1002 * @return effective priority
1005 bound_priority (uint32_t prio_in, struct GSF_ConnectedPeer *cp)
1007 #define N ((double)128.0)
1012 ld = GSF_test_get_load_too_high_ (0);
1013 if (ld == GNUNET_SYSERR)
1015 GNUNET_STATISTICS_update (GSF_stats,
1017 ("# requests done for free (low load)"), 1,
1019 return 0; /* excess resources */
1021 if (prio_in > INT32_MAX)
1022 prio_in = INT32_MAX;
1023 ret = -change_host_trust (cp, -(int) prio_in);
1026 if (ret > GSF_current_priorities + N)
1027 rret = GSF_current_priorities + N;
1030 GSF_current_priorities = (GSF_current_priorities * (N - 1) + rret) / N;
1032 if ((ld == GNUNET_YES) && (ret > 0))
1034 /* try with charging */
1035 ld = GSF_test_get_load_too_high_ (ret);
1037 if (ld == GNUNET_YES)
1039 GNUNET_STATISTICS_update (GSF_stats,
1041 ("# request dropped, priority insufficient"), 1,
1044 change_host_trust (cp, (int) ret);
1045 return -1; /* not enough resources */
1049 GNUNET_STATISTICS_update (GSF_stats,
1051 ("# requests done for a price (normal load)"), 1,
1060 * The priority level imposes a bound on the maximum
1061 * value for the ttl that can be requested.
1063 * @param ttl_in requested ttl
1064 * @param prio given priority
1065 * @return ttl_in if ttl_in is below the limit,
1066 * otherwise the ttl-limit for the given priority
1069 bound_ttl (int32_t ttl_in, uint32_t prio)
1071 unsigned long long allowed;
1075 allowed = ((unsigned long long) prio) * TTL_DECREMENT / 1000;
1076 if (ttl_in > allowed)
1078 if (allowed >= (1 << 30))
1087 * Handle P2P "QUERY" message. Creates the pending request entry
1088 * and sets up all of the data structures to that we will
1089 * process replies properly. Does not initiate forwarding or
1090 * local database lookups.
1092 * @param other the other peer involved (sender or receiver, NULL
1093 * for loopback messages where we are both sender and receiver)
1094 * @param message the actual message
1095 * @return pending request handle, NULL on error
1097 struct GSF_PendingRequest *
1098 GSF_handle_p2p_query_ (const struct GNUNET_PeerIdentity *other,
1099 const struct GNUNET_MessageHeader *message)
1101 struct PeerRequest *peerreq;
1102 struct GSF_PendingRequest *pr;
1103 struct GSF_PendingRequestData *prd;
1104 struct GSF_ConnectedPeer *cp;
1105 struct GSF_ConnectedPeer *cps;
1106 const GNUNET_HashCode *namespace;
1107 const struct GNUNET_PeerIdentity *target;
1108 enum GSF_PendingRequestOptions options;
1110 const struct GetMessage *gm;
1112 const GNUNET_HashCode *opt;
1115 uint32_t ttl_decrement;
1118 enum GNUNET_BLOCK_Type type;
1119 GNUNET_PEER_Id spid;
1121 GNUNET_assert (other != NULL);
1122 msize = ntohs (message->size);
1123 if (msize < sizeof (struct GetMessage))
1125 GNUNET_break_op (0);
1128 GNUNET_STATISTICS_update (GSF_stats,
1130 ("# GET requests received (from other peers)"), 1,
1132 gm = (const struct GetMessage *) message;
1133 type = ntohl (gm->type);
1134 bm = ntohl (gm->hash_bitmap);
1142 if (msize < sizeof (struct GetMessage) + bits * sizeof (GNUNET_HashCode))
1144 GNUNET_break_op (0);
1147 opt = (const GNUNET_HashCode *) &gm[1];
1148 bfsize = msize - sizeof (struct GetMessage) - bits * sizeof (GNUNET_HashCode);
1149 /* bfsize must be power of 2, check! */
1150 if (0 != ((bfsize - 1) & bfsize))
1152 GNUNET_break_op (0);
1155 GSF_cover_query_count++;
1156 bm = ntohl (gm->hash_bitmap);
1158 cps = GSF_peer_get_ (other);
1161 /* peer must have just disconnected */
1162 GNUNET_STATISTICS_update (GSF_stats,
1164 ("# requests dropped due to initiator not being connected"),
1168 if (0 != (bm & GET_MESSAGE_BIT_RETURN_TO))
1169 cp = GSF_peer_get_ ((const struct GNUNET_PeerIdentity *) &opt[bits++]);
1175 if (0 != (bm & GET_MESSAGE_BIT_RETURN_TO))
1176 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1177 "Failed to find RETURN-TO peer `%4s' in connection set. Dropping query.\n",
1178 GNUNET_i2s ((const struct GNUNET_PeerIdentity *)
1182 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1183 "Failed to find peer `%4s' in connection set. Dropping query.\n",
1184 GNUNET_i2s (other));
1186 GNUNET_STATISTICS_update (GSF_stats,
1188 ("# requests dropped due to missing reverse route"),
1192 /* note that we can really only check load here since otherwise
1193 * peers could find out that we are overloaded by not being
1194 * disconnected after sending us a malformed query... */
1195 priority = bound_priority (ntohl (gm->priority), cps);
1199 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1200 "Dropping query from `%s', this peer is too busy.\n",
1201 GNUNET_i2s (other));
1206 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1207 "Received request for `%s' of type %u from peer `%4s' with flags %u\n",
1208 GNUNET_h2s (&gm->query), (unsigned int) type, GNUNET_i2s (other),
1211 namespace = (0 != (bm & GET_MESSAGE_BIT_SKS_NAMESPACE)) ? &opt[bits++] : NULL;
1212 if ((type == GNUNET_BLOCK_TYPE_FS_SBLOCK) && (namespace == NULL))
1214 GNUNET_break_op (0);
1217 if ((type != GNUNET_BLOCK_TYPE_FS_SBLOCK) && (namespace != NULL))
1219 GNUNET_break_op (0);
1224 (bm & GET_MESSAGE_BIT_TRANSMIT_TO)) ? ((const struct GNUNET_PeerIdentity
1225 *) &opt[bits++]) : NULL;
1228 if ((GNUNET_LOAD_get_load (cp->ppd.transmission_delay) > 3 * (1 + priority))
1229 || (GNUNET_LOAD_get_average (cp->ppd.transmission_delay) >
1230 GNUNET_CONSTANTS_MAX_CORK_DELAY.rel_value * 2 +
1231 GNUNET_LOAD_get_average (GSF_rt_entry_lifetime)))
1233 /* don't have BW to send to peer, or would likely take longer than we have for it,
1234 * so at best indirect the query */
1236 options |= GSF_PRO_FORWARD_ONLY;
1237 spid = GNUNET_PEER_intern (other);
1238 GNUNET_assert (0 != spid);
1240 ttl = bound_ttl (ntohl (gm->ttl), priority);
1241 /* decrement ttl (always) */
1243 2 * TTL_DECREMENT + GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
1245 if ((ttl < 0) && (((int32_t) (ttl - ttl_decrement)) > 0))
1248 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1249 "Dropping query from `%s' due to TTL underflow (%d - %u).\n",
1250 GNUNET_i2s (other), ttl, ttl_decrement);
1252 GNUNET_STATISTICS_update (GSF_stats,
1254 ("# requests dropped due TTL underflow"), 1,
1256 /* integer underflow => drop (should be very rare)! */
1259 ttl -= ttl_decrement;
1261 /* test if the request already exists */
1262 peerreq = GNUNET_CONTAINER_multihashmap_get (cp->request_map, &gm->query);
1263 if (peerreq != NULL)
1266 prd = GSF_pending_request_get_data_ (pr);
1267 if ((prd->type == type) &&
1268 ((type != GNUNET_BLOCK_TYPE_FS_SBLOCK) ||
1269 (0 == memcmp (&prd->namespace, namespace, sizeof (GNUNET_HashCode)))))
1271 if (prd->ttl.abs_value >= GNUNET_TIME_absolute_get ().abs_value + ttl)
1273 /* existing request has higher TTL, drop new one! */
1274 prd->priority += priority;
1276 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1277 "Have existing request with higher TTL, dropping new request.\n",
1278 GNUNET_i2s (other));
1280 GNUNET_STATISTICS_update (GSF_stats,
1282 ("# requests dropped due to higher-TTL request"),
1286 /* existing request has lower TTL, drop old one! */
1287 GNUNET_STATISTICS_update (GSF_stats,
1288 gettext_noop ("# P2P searches active"), -1,
1290 priority += prd->priority;
1291 GSF_pending_request_cancel_ (pr, GNUNET_YES);
1292 GNUNET_assert (GNUNET_YES ==
1293 GNUNET_CONTAINER_multihashmap_remove (cp->request_map,
1296 if (peerreq->kill_task != GNUNET_SCHEDULER_NO_TASK)
1298 GNUNET_SCHEDULER_cancel (peerreq->kill_task);
1299 peerreq->kill_task = GNUNET_SCHEDULER_NO_TASK;
1301 GNUNET_free (peerreq);
1305 peerreq = GNUNET_malloc (sizeof (struct PeerRequest));
1307 pr = GSF_pending_request_create_ (options, type, &gm->query, namespace,
1310 0) ? (const char *) &opt[bits] : NULL,
1311 bfsize, ntohl (gm->filter_mutator),
1313 (uint32_t) priority, ttl, spid, GNUNET_PEER_intern (other), NULL, 0, /* replies_seen */
1314 &handle_p2p_reply, peerreq);
1315 GNUNET_assert (NULL != pr);
1317 GNUNET_break (GNUNET_OK ==
1318 GNUNET_CONTAINER_multihashmap_put (cp->request_map, &gm->query,
1320 GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
1321 GNUNET_STATISTICS_update (GSF_stats,
1323 ("# P2P query messages received and processed"), 1,
1325 GNUNET_STATISTICS_update (GSF_stats, gettext_noop ("# P2P searches active"),
1332 * Function called if there has been a timeout trying to satisfy
1333 * a transmission request.
1335 * @param cls the 'struct GSF_PeerTransmitHandle' of the request
1336 * @param tc scheduler context
1339 peer_transmit_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1341 struct GSF_PeerTransmitHandle *pth = cls;
1342 struct GSF_ConnectedPeer *cp;
1345 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1346 "Timeout trying to transmit to other peer\n");
1348 pth->timeout_task = GNUNET_SCHEDULER_NO_TASK;
1350 GNUNET_CONTAINER_DLL_remove (cp->pth_head, cp->pth_tail, pth);
1351 if (GNUNET_YES == pth->is_query)
1352 GNUNET_assert (0 < cp->ppd.pending_queries--);
1353 else if (GNUNET_NO == pth->is_query)
1354 GNUNET_assert (0 < cp->ppd.pending_replies--);
1355 GNUNET_LOAD_update (cp->ppd.transmission_delay, UINT64_MAX);
1356 if (NULL != pth->cth)
1358 GNUNET_CORE_notify_transmit_ready_cancel (pth->cth);
1361 pth->gmc (pth->gmc_cls, 0, NULL);
1362 GNUNET_assert (0 == pth->cth_in_progress);
1368 * Transmit a message to the given peer as soon as possible.
1369 * If the peer disconnects before the transmission can happen,
1370 * the callback is invoked with a 'NULL' buffer.
1372 * @param cp target peer
1373 * @param is_query is this a query (GNUNET_YES) or content (GNUNET_NO) or neither (GNUNET_SYSERR)
1374 * @param priority how important is this request?
1375 * @param timeout when does this request timeout (call gmc with error)
1376 * @param size number of bytes we would like to send to the peer
1377 * @param gmc function to call to get the message
1378 * @param gmc_cls closure for gmc
1379 * @return handle to cancel request
1381 struct GSF_PeerTransmitHandle *
1382 GSF_peer_transmit_ (struct GSF_ConnectedPeer *cp, int is_query,
1383 uint32_t priority, struct GNUNET_TIME_Relative timeout,
1384 size_t size, GSF_GetMessageCallback gmc, void *gmc_cls)
1386 struct GSF_PeerTransmitHandle *pth;
1387 struct GSF_PeerTransmitHandle *pos;
1388 struct GSF_PeerTransmitHandle *prev;
1390 pth = GNUNET_malloc (sizeof (struct GSF_PeerTransmitHandle));
1391 pth->transmission_request_start_time = GNUNET_TIME_absolute_get ();
1392 pth->timeout = GNUNET_TIME_relative_to_absolute (timeout);
1394 pth->gmc_cls = gmc_cls;
1396 pth->is_query = is_query;
1397 pth->priority = priority;
1399 /* insertion sort (by priority, descending) */
1402 while ((pos != NULL) && (pos->priority > priority))
1408 GNUNET_CONTAINER_DLL_insert (cp->pth_head, cp->pth_tail, pth);
1410 GNUNET_CONTAINER_DLL_insert_after (cp->pth_head, cp->pth_tail, prev, pth);
1411 if (GNUNET_YES == is_query)
1412 cp->ppd.pending_queries++;
1413 else if (GNUNET_NO == is_query)
1414 cp->ppd.pending_replies++;
1416 GNUNET_SCHEDULER_add_delayed (timeout, &peer_transmit_timeout, pth);
1417 schedule_transmission (pth);
1423 * Cancel an earlier request for transmission.
1425 * @param pth request to cancel
1428 GSF_peer_transmit_cancel_ (struct GSF_PeerTransmitHandle *pth)
1430 struct GSF_ConnectedPeer *cp;
1432 if (pth->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1434 GNUNET_SCHEDULER_cancel (pth->timeout_task);
1435 pth->timeout_task = GNUNET_SCHEDULER_NO_TASK;
1437 if (NULL != pth->cth)
1439 GNUNET_CORE_notify_transmit_ready_cancel (pth->cth);
1443 GNUNET_CONTAINER_DLL_remove (cp->pth_head, cp->pth_tail, pth);
1444 if (GNUNET_YES == pth->is_query)
1445 GNUNET_assert (0 < cp->ppd.pending_queries--);
1446 else if (GNUNET_NO == pth->is_query)
1447 GNUNET_assert (0 < cp->ppd.pending_replies--);
1448 GNUNET_assert (0 == pth->cth_in_progress);
1454 * Report on receiving a reply; update the performance record of the given peer.
1456 * @param cp responding peer (will be updated)
1457 * @param request_time time at which the original query was transmitted
1458 * @param request_priority priority of the original request
1461 GSF_peer_update_performance_ (struct GSF_ConnectedPeer *cp,
1462 struct GNUNET_TIME_Absolute request_time,
1463 uint32_t request_priority)
1465 struct GNUNET_TIME_Relative delay;
1467 delay = GNUNET_TIME_absolute_get_duration (request_time);
1468 cp->ppd.avg_reply_delay.rel_value =
1469 (cp->ppd.avg_reply_delay.rel_value * (RUNAVG_DELAY_N - 1) +
1470 delay.rel_value) / RUNAVG_DELAY_N;
1471 cp->ppd.avg_priority =
1472 (cp->ppd.avg_priority * (RUNAVG_DELAY_N - 1) +
1473 request_priority) / RUNAVG_DELAY_N;
1478 * Report on receiving a reply in response to an initiating client.
1479 * Remember that this peer is good for this client.
1481 * @param cp responding peer (will be updated)
1482 * @param initiator_client local client on responsible for query
1485 GSF_peer_update_responder_client_ (struct GSF_ConnectedPeer *cp,
1486 struct GSF_LocalClient *initiator_client)
1488 cp->ppd.last_client_replies[cp->last_client_replies_woff++ %
1489 CS2P_SUCCESS_LIST_SIZE] = initiator_client;
1494 * Report on receiving a reply in response to an initiating peer.
1495 * Remember that this peer is good for this initiating peer.
1497 * @param cp responding peer (will be updated)
1498 * @param initiator_peer other peer responsible for query
1501 GSF_peer_update_responder_peer_ (struct GSF_ConnectedPeer *cp,
1502 const struct GSF_ConnectedPeer *initiator_peer)
1506 woff = cp->last_p2p_replies_woff % P2P_SUCCESS_LIST_SIZE;
1507 GNUNET_PEER_change_rc (cp->ppd.last_p2p_replies[woff], -1);
1508 cp->ppd.last_p2p_replies[woff] = initiator_peer->ppd.pid;
1509 GNUNET_PEER_change_rc (initiator_peer->ppd.pid, 1);
1510 cp->last_p2p_replies_woff = (woff + 1) % P2P_SUCCESS_LIST_SIZE;
1515 * A peer disconnected from us. Tear down the connected peer
1519 * @param peer identity of peer that connected
1522 GSF_peer_disconnect_handler_ (void *cls, const struct GNUNET_PeerIdentity *peer)
1524 struct GSF_ConnectedPeer *cp;
1525 struct GSF_PeerTransmitHandle *pth;
1526 struct GSF_DelayedHandle *dh;
1528 cp = GSF_peer_get_ (peer);
1530 return; /* must have been disconnect from core with
1531 * 'peer' == my_id, ignore */
1532 GNUNET_assert (GNUNET_YES ==
1533 GNUNET_CONTAINER_multihashmap_remove (cp_map,
1534 &peer->hashPubKey, cp));
1535 GNUNET_STATISTICS_set (GSF_stats, gettext_noop ("# peers connected"),
1536 GNUNET_CONTAINER_multihashmap_size (cp_map),
1538 if (NULL != cp->migration_pth)
1540 GSF_peer_transmit_cancel_ (cp->migration_pth);
1541 cp->migration_pth = NULL;
1545 GNUNET_ATS_reserve_bandwidth_cancel (cp->rc);
1548 if (GNUNET_SCHEDULER_NO_TASK != cp->rc_delay_task)
1550 GNUNET_SCHEDULER_cancel (cp->rc_delay_task);
1551 cp->rc_delay_task = GNUNET_SCHEDULER_NO_TASK;
1553 GNUNET_CONTAINER_multihashmap_iterate (cp->request_map,
1554 &cancel_pending_request, cp);
1555 GNUNET_CONTAINER_multihashmap_destroy (cp->request_map);
1556 cp->request_map = NULL;
1557 GSF_plan_notify_peer_disconnect_ (cp);
1558 GNUNET_LOAD_value_free (cp->ppd.transmission_delay);
1559 GNUNET_PEER_decrement_rcs (cp->ppd.last_p2p_replies, P2P_SUCCESS_LIST_SIZE);
1560 memset (cp->ppd.last_p2p_replies, 0, sizeof (cp->ppd.last_p2p_replies));
1561 GSF_push_stop_ (cp);
1562 while (NULL != (pth = cp->pth_head))
1564 if (NULL != pth->cth)
1566 GNUNET_CORE_notify_transmit_ready_cancel (pth->cth);
1569 if (pth->timeout_task != GNUNET_SCHEDULER_NO_TASK)
1571 GNUNET_SCHEDULER_cancel (pth->timeout_task);
1572 pth->timeout_task = GNUNET_SCHEDULER_NO_TASK;
1574 GNUNET_CONTAINER_DLL_remove (cp->pth_head, cp->pth_tail, pth);
1575 GNUNET_assert (0 == pth->cth_in_progress);
1576 pth->gmc (pth->gmc_cls, 0, NULL);
1579 while (NULL != (dh = cp->delayed_head))
1581 GNUNET_CONTAINER_DLL_remove (cp->delayed_head, cp->delayed_tail, dh);
1582 GNUNET_SCHEDULER_cancel (dh->delay_task);
1583 GNUNET_free (dh->pm);
1586 GNUNET_PEER_change_rc (cp->ppd.pid, -1);
1587 if (GNUNET_SCHEDULER_NO_TASK != cp->mig_revive_task)
1589 GNUNET_SCHEDULER_cancel (cp->mig_revive_task);
1590 cp->mig_revive_task = GNUNET_SCHEDULER_NO_TASK;
1597 * Closure for 'call_iterator'.
1599 struct IterationContext
1602 * Function to call on each entry.
1604 GSF_ConnectedPeerIterator it;
1614 * Function that calls the callback for each peer.
1616 * @param cls the 'struct IterationContext*'
1617 * @param key identity of the peer
1618 * @param value the 'struct GSF_ConnectedPeer*'
1619 * @return GNUNET_YES to continue iteration
1622 call_iterator (void *cls, const GNUNET_HashCode * key, void *value)
1624 struct IterationContext *ic = cls;
1625 struct GSF_ConnectedPeer *cp = value;
1627 ic->it (ic->it_cls, (const struct GNUNET_PeerIdentity *) key, cp, &cp->ppd);
1633 * Iterate over all connected peers.
1635 * @param it function to call for each peer
1636 * @param it_cls closure for it
1639 GSF_iterate_connected_peers_ (GSF_ConnectedPeerIterator it, void *it_cls)
1641 struct IterationContext ic;
1645 GNUNET_CONTAINER_multihashmap_iterate (cp_map, &call_iterator, &ic);
1650 * Obtain the identity of a connected peer.
1652 * @param cp peer to reserve bandwidth from
1653 * @param id identity to set (written to)
1656 GSF_connected_peer_get_identity_ (const struct GSF_ConnectedPeer *cp,
1657 struct GNUNET_PeerIdentity *id)
1659 GNUNET_assert (0 != cp->ppd.pid);
1660 GNUNET_PEER_resolve (cp->ppd.pid, id);
1665 * Assemble a migration stop message for transmission.
1667 * @param cls the 'struct GSF_ConnectedPeer' to use
1668 * @param size number of bytes we're allowed to write to buf
1669 * @param buf where to copy the message
1670 * @return number of bytes copied to buf
1673 create_migration_stop_message (void *cls, size_t size, void *buf)
1675 struct GSF_ConnectedPeer *cp = cls;
1676 struct MigrationStopMessage msm;
1678 cp->migration_pth = NULL;
1681 GNUNET_assert (size >= sizeof (struct MigrationStopMessage));
1682 msm.header.size = htons (sizeof (struct MigrationStopMessage));
1683 msm.header.type = htons (GNUNET_MESSAGE_TYPE_FS_MIGRATION_STOP);
1684 msm.reserved = htonl (0);
1686 GNUNET_TIME_relative_hton (GNUNET_TIME_absolute_get_remaining
1687 (cp->last_migration_block));
1688 memcpy (buf, &msm, sizeof (struct MigrationStopMessage));
1689 return sizeof (struct MigrationStopMessage);
1694 * Ask a peer to stop migrating data to us until the given point
1697 * @param cp peer to ask
1698 * @param block_time until when to block
1701 GSF_block_peer_migration_ (struct GSF_ConnectedPeer *cp,
1702 struct GNUNET_TIME_Relative block_time)
1704 if (GNUNET_TIME_absolute_get_remaining (cp->last_migration_block).rel_value >
1705 block_time.rel_value)
1708 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1709 "Migration already blocked for another %llu ms\n",
1710 (unsigned long long)
1711 GNUNET_TIME_absolute_get_remaining
1712 (cp->last_migration_block).rel_value);
1714 return; /* already blocked */
1717 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Asking to stop migration for %llu ms\n",
1718 (unsigned long long) block_time.rel_value);
1720 cp->last_migration_block = GNUNET_TIME_relative_to_absolute (block_time);
1721 if (cp->migration_pth != NULL)
1722 GSF_peer_transmit_cancel_ (cp->migration_pth);
1724 GSF_peer_transmit_ (cp, GNUNET_SYSERR, UINT32_MAX,
1725 GNUNET_TIME_UNIT_FOREVER_REL,
1726 sizeof (struct MigrationStopMessage),
1727 &create_migration_stop_message, cp);
1732 * Write host-trust information to a file - flush the buffer entry!
1734 * @param cls closure, not used
1735 * @param key host identity
1736 * @param value the 'struct GSF_ConnectedPeer' to flush
1737 * @return GNUNET_OK to continue iteration
1740 flush_trust (void *cls, const GNUNET_HashCode * key, void *value)
1742 struct GSF_ConnectedPeer *cp = value;
1745 struct GNUNET_PeerIdentity pid;
1747 if (cp->ppd.trust == cp->disk_trust)
1748 return GNUNET_OK; /* unchanged */
1749 GNUNET_assert (0 != cp->ppd.pid);
1750 GNUNET_PEER_resolve (cp->ppd.pid, &pid);
1751 fn = get_trust_filename (&pid);
1752 if (cp->ppd.trust == 0)
1754 if ((0 != UNLINK (fn)) && (errno != ENOENT))
1755 GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING |
1756 GNUNET_ERROR_TYPE_BULK, "unlink", fn);
1760 trust = htonl (cp->ppd.trust);
1761 if (sizeof (uint32_t) ==
1762 GNUNET_DISK_fn_write (fn, &trust, sizeof (uint32_t),
1763 GNUNET_DISK_PERM_USER_READ |
1764 GNUNET_DISK_PERM_USER_WRITE |
1765 GNUNET_DISK_PERM_GROUP_READ |
1766 GNUNET_DISK_PERM_OTHER_READ))
1767 cp->disk_trust = cp->ppd.trust;
1775 * Notify core about a preference we have for the given peer
1776 * (to allocate more resources towards it). The change will
1777 * be communicated the next time we reserve bandwidth with
1778 * core (not instantly).
1780 * @param cp peer to reserve bandwidth from
1781 * @param pref preference change
1784 GSF_connected_peer_change_preference_ (struct GSF_ConnectedPeer *cp,
1787 cp->inc_preference += pref;
1792 * Call this method periodically to flush trust information to disk.
1794 * @param cls closure, not used
1795 * @param tc task context, not used
1798 cron_flush_trust (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1803 GNUNET_CONTAINER_multihashmap_iterate (cp_map, &flush_trust, NULL);
1806 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1808 GNUNET_SCHEDULER_add_delayed (TRUST_FLUSH_FREQ, &cron_flush_trust, NULL);
1813 * Initialize peer management subsystem.
1816 GSF_connected_peer_init_ ()
1818 cp_map = GNUNET_CONTAINER_multihashmap_create (128);
1819 ats = GNUNET_ATS_performance_init (GSF_cfg, NULL, NULL);
1820 GNUNET_assert (GNUNET_OK ==
1821 GNUNET_CONFIGURATION_get_value_filename (GSF_cfg, "fs",
1824 GNUNET_DISK_directory_create (trustDirectory);
1825 GNUNET_SCHEDULER_add_with_priority (GNUNET_SCHEDULER_PRIORITY_HIGH,
1826 &cron_flush_trust, NULL);
1831 * Iterator to free peer entries.
1833 * @param cls closure, unused
1834 * @param key current key code
1835 * @param value value in the hash map (peer entry)
1836 * @return GNUNET_YES (we should continue to iterate)
1839 clean_peer (void *cls, const GNUNET_HashCode * key, void *value)
1841 GSF_peer_disconnect_handler_ (NULL, (const struct GNUNET_PeerIdentity *) key);
1847 * Shutdown peer management subsystem.
1850 GSF_connected_peer_done_ ()
1852 cron_flush_trust (NULL, NULL);
1853 GNUNET_CONTAINER_multihashmap_iterate (cp_map, &clean_peer, NULL);
1854 GNUNET_CONTAINER_multihashmap_destroy (cp_map);
1856 GNUNET_free (trustDirectory);
1857 trustDirectory = NULL;
1858 GNUNET_ATS_performance_done (ats);
1864 * Iterator to remove references to LC entry.
1866 * @param cls the 'struct GSF_LocalClient*' to look for
1867 * @param key current key code
1868 * @param value value in the hash map (peer entry)
1869 * @return GNUNET_YES (we should continue to iterate)
1872 clean_local_client (void *cls, const GNUNET_HashCode * key, void *value)
1874 const struct GSF_LocalClient *lc = cls;
1875 struct GSF_ConnectedPeer *cp = value;
1878 for (i = 0; i < CS2P_SUCCESS_LIST_SIZE; i++)
1879 if (cp->ppd.last_client_replies[i] == lc)
1880 cp->ppd.last_client_replies[i] = NULL;
1886 * Notification that a local client disconnected. Clean up all of our
1887 * references to the given handle.
1889 * @param lc handle to the local client (henceforth invalid)
1892 GSF_handle_local_client_disconnect_ (const struct GSF_LocalClient *lc)
1895 return; /* already cleaned up */
1896 GNUNET_CONTAINER_multihashmap_iterate (cp_map, &clean_local_client,
1901 /* end of gnunet-service-fs_cp.c */