X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Fats%2Fgnunet-service-ats_reservations.c;h=a272170b706cfe20550a89571bb1189bbe643545;hb=e871f8368c8a1f0867f68f656875e4c37c26f298;hp=3356aefbd409b2e1c9e2ba02454c8a030d3aae84;hpb=d86cd012c0372c1746fed2a50b7cccca9de99be3;p=oweals%2Fgnunet.git diff --git a/src/ats/gnunet-service-ats_reservations.c b/src/ats/gnunet-service-ats_reservations.c index 3356aefbd..a272170b7 100644 --- a/src/ats/gnunet-service-ats_reservations.c +++ b/src/ats/gnunet-service-ats_reservations.c @@ -1,23 +1,20 @@ /* This file is part of GNUnet. - (C) 2011 Christian Grothoff (and other contributing authors) + Copyright (C) 2011 GNUnet e.V. - GNUnet is free software; you can redistribute it and/or modify - it under the terms of the GNU General Public License as published - by the Free Software Foundation; either version 3, or (at your - option) any later version. + GNUnet is free software: you can redistribute it and/or modify it + under the terms of the GNU Affero General Public License as published + by the Free Software Foundation, either version 3 of the License, + or (at your option) any later version. GNUnet is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - General Public License for more details. - - You should have received a copy of the GNU General Public License - along with GNUnet; see the file COPYING. If not, write to the - Free Software Foundation, Inc., 59 Temple Place - Suite 330, - Boston, MA 02111-1307, USA. + Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . */ - /** * @file ats/gnunet-service-ats_reservations.c * @brief ats service, inbound bandwidth reservation management @@ -25,15 +22,193 @@ */ #include "platform.h" #include "gnunet-service-ats_reservations.h" +#include "gnunet-service-ats.h" +#include "ats.h" + +/** + * Number of seconds that available bandwidth carries over + * (can accumulate). Note that the + * test_ats_reservation_api test depends on this value! + */ +#define MAX_BANDWIDTH_CARRY_S 5 + + +/** + * Map of peer identities to `struct GNUNET_BANDWIDTH_Tracker *`s + */ +static struct GNUNET_CONTAINER_MultiPeerMap *trackers; -struct GNUNET_TIME_Relative -GAS_reservations_reserve (const struct GNUNET_PeerIdentity *peer, - int32_t amount) +/** + * Reserve the given amount of incoming bandwidth (in bytes) from the + * given peer. If a reservation is not possible right now, return how + * long the client should wait before trying again. + * + * @param peer peer to reserve bandwidth from + * @param amount number of bytes to reserve + * @return 0 if the reservation was successful, FOREVER if the + * peer is not connected, otherwise the time to wait + * until the reservation might succeed + */ +static struct GNUNET_TIME_Relative +reservations_reserve (const struct GNUNET_PeerIdentity *peer, + int32_t amount) { - /* FIXME: implement... */ - /* permit all reservations instantly for now */ + struct GNUNET_BANDWIDTH_Tracker *tracker; + struct GNUNET_TIME_Relative ret; + + tracker = GNUNET_CONTAINER_multipeermap_get (trackers, + peer); + if (NULL == tracker) + { + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Not connected, allowing reservation of %d bytes\n", + (int) amount); + return GNUNET_TIME_UNIT_ZERO; /* not connected, satisfy now */ + } + if (amount >= 0) + { + ret = GNUNET_BANDWIDTH_tracker_get_delay (tracker, amount); + if (ret.rel_value_us > 0) + { + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Delay to satisfy reservation for %d bytes is %s\n", + (int) amount, + GNUNET_STRINGS_relative_time_to_string (ret, + GNUNET_YES)); + return ret; + } + } + (void) GNUNET_BANDWIDTH_tracker_consume (tracker, amount); + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Reserved %d bytes\n", + (int) amount); return GNUNET_TIME_UNIT_ZERO; } + +/** + * Set the amount of bandwidth the other peer could currently transmit + * to us (as far as we know) to the given value. + * + * @param peer identity of the peer + * @param bandwidth_in currently available bandwidth from that peer to + * this peer (estimate) + */ +void +GAS_reservations_set_bandwidth (const struct GNUNET_PeerIdentity *peer, + struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in) +{ + struct GNUNET_BANDWIDTH_Tracker *tracker; + + tracker = GNUNET_CONTAINER_multipeermap_get (trackers, peer); + if (0 == ntohl (bandwidth_in.value__)) + { + if (NULL == tracker) + return; + GNUNET_assert (GNUNET_YES == + GNUNET_CONTAINER_multipeermap_remove (trackers, + peer, + tracker)); + GNUNET_free (tracker); + return; + } + if (NULL == tracker) + { + tracker = GNUNET_new (struct GNUNET_BANDWIDTH_Tracker); + GNUNET_BANDWIDTH_tracker_init (tracker, + NULL, + NULL, + bandwidth_in, + MAX_BANDWIDTH_CARRY_S); + GNUNET_assert (GNUNET_OK == + GNUNET_CONTAINER_multipeermap_put (trackers, + peer, + tracker, + GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY)); + return; + } + GNUNET_BANDWIDTH_tracker_update_quota (tracker, + bandwidth_in); +} + + +/** + * Handle 'reservation request' messages from clients. + * + * @param client client that sent the request + * @param msg the request message + */ +void +GAS_handle_reservation_request (struct GNUNET_SERVICE_Client *client, + const struct ReservationRequestMessage *msg) +{ + struct GNUNET_MQ_Envelope *env; + struct ReservationResultMessage *result; + int32_t amount; + struct GNUNET_TIME_Relative res_delay; + + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Received RESERVATION_REQUEST message\n"); + amount = (int32_t) ntohl (msg->amount); + res_delay = reservations_reserve (&msg->peer, amount); + if (res_delay.rel_value_us > 0) + amount = 0; + env = GNUNET_MQ_msg (result, + GNUNET_MESSAGE_TYPE_ATS_RESERVATION_RESULT); + result->amount = htonl (amount); + result->peer = msg->peer; + result->res_delay = GNUNET_TIME_relative_hton (res_delay); + GNUNET_STATISTICS_update (GSA_stats, + "# reservation requests processed", + 1, + GNUNET_NO); + GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client), + env); +} + + +/** + * Initialize reservations subsystem. + */ +void +GAS_reservations_init () +{ + trackers = GNUNET_CONTAINER_multipeermap_create (128, + GNUNET_NO); +} + + +/** + * Free memory of bandwidth tracker. + * + * @param cls NULL + * @param key peer identity (unused) + * @param value the `struct GNUNET_BANDWIDTH_Tracker` to free + * @return #GNUNET_OK (continue to iterate) + */ +static int +free_tracker (void *cls, + const struct GNUNET_PeerIdentity *key, + void *value) +{ + struct GNUNET_BANDWIDTH_Tracker *tracker = value; + + GNUNET_free (tracker); + return GNUNET_OK; +} + + +/** + * Shutdown reservations subsystem. + */ +void +GAS_reservations_done () +{ + GNUNET_CONTAINER_multipeermap_iterate (trackers, + &free_tracker, + NULL); + GNUNET_CONTAINER_multipeermap_destroy (trackers); +} + /* end of gnunet-service-ats_reservations.c */