1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081 |
- /*
- This file is part of GNUnet.
- Copyright (C) 2010,2011 Christian Grothoff (and other contributing authors)
- 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 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.
- */
- /**
- * @file ats/ats_api_performance.c
- * @brief automatic transport selection and outbound bandwidth determination
- * @author Christian Grothoff
- * @author Matthias Wachs
- */
- #include "platform.h"
- #include "gnunet_ats_service.h"
- #include "ats.h"
- #define LOG(kind,...) GNUNET_log_from(kind, "ats-performance-api", __VA_ARGS__)
- /**
- * Message in linked list we should send to the ATS service. The
- * actual binary message follows this struct.
- */
- struct PendingMessage
- {
- /**
- * Kept in a DLL.
- */
- struct PendingMessage *next;
- /**
- * Kept in a DLL.
- */
- struct PendingMessage *prev;
- /**
- * Size of the message.
- */
- size_t size;
- /**
- * Is this the 'ATS_START' message?
- */
- int is_init;
- };
- /**
- * Linked list of pending reservations.
- */
- struct GNUNET_ATS_ReservationContext
- {
- /**
- * Kept in a DLL.
- */
- struct GNUNET_ATS_ReservationContext *next;
- /**
- * Kept in a DLL.
- */
- struct GNUNET_ATS_ReservationContext *prev;
- /**
- * Target peer.
- */
- struct GNUNET_PeerIdentity peer;
- /**
- * Desired reservation
- */
- int32_t size;
- /**
- * Function to call on result.
- */
- GNUNET_ATS_ReservationCallback rcb;
- /**
- * Closure for @e rcb
- */
- void *rcb_cls;
- /**
- * Do we need to undo this reservation if it succeeded? Set to
- * #GNUNET_YES if a reservation is cancelled. (at that point, 'info'
- * is also set to NULL; however, info will ALSO be NULL for the
- * reservation context that is created to undo the original request,
- * so 'info' being NULL cannot be used to check if undo is
- * required).
- */
- int undo;
- };
- /**
- * Linked list of pending reservations.
- */
- struct GNUNET_ATS_AddressListHandle
- {
- /**
- * Kept in a DLL.
- */
- struct GNUNET_ATS_AddressListHandle *next;
- /**
- * Kept in a DLL.
- */
- struct GNUNET_ATS_AddressListHandle *prev;
- /**
- * Performance handle
- */
- struct GNUNET_ATS_PerformanceHandle *ph;
- /**
- * Callback
- */
- GNUNET_ATS_AddressInformationCallback cb;
- /**
- * Callback closure for @e cb
- */
- void *cb_cls;
- /**
- * Target peer.
- */
- struct GNUNET_PeerIdentity peer;
- /**
- * Return all or specific peer only
- */
- int all_peers;
- /**
- * Return all or used address only
- */
- int all_addresses;
- /**
- * Request multiplexing
- */
- uint32_t id;
- };
- /**
- * ATS Handle to obtain and/or modify performance information.
- */
- struct GNUNET_ATS_PerformanceHandle
- {
- /**
- * Our configuration.
- */
- const struct GNUNET_CONFIGURATION_Handle *cfg;
- /**
- * Callback to invoke when an address has performance changes.
- */
- GNUNET_ATS_AddressInformationCallback addr_info_cb;
- /**
- * Closure for @e addr_info_cb.
- */
- void *addr_info_cb_cls;
- /**
- * Connection to ATS service.
- */
- struct GNUNET_CLIENT_Connection *client;
- /**
- * Head of list of messages for the ATS service.
- */
- struct PendingMessage *pending_head;
- /**
- * Tail of list of messages for the ATS service
- */
- struct PendingMessage *pending_tail;
- /**
- * Head of linked list of pending reservation requests.
- */
- struct GNUNET_ATS_ReservationContext *reservation_head;
- /**
- * Tail of linked list of pending reservation requests.
- */
- struct GNUNET_ATS_ReservationContext *reservation_tail;
- /**
- * Head of linked list of pending address list requests.
- */
- struct GNUNET_ATS_AddressListHandle *addresslist_head;
- /**
- * Tail of linked list of pending address list requests.
- */
- struct GNUNET_ATS_AddressListHandle *addresslist_tail;
- /**
- * Current request for transmission to ATS.
- */
- struct GNUNET_CLIENT_TransmitHandle *th;
- /**
- * Task to trigger reconnect.
- */
- struct GNUNET_SCHEDULER_Task *task;
- /**
- * Reconnect backoff delay.
- */
- struct GNUNET_TIME_Relative backoff;
- /**
- * Monitor request multiplexing
- */
- uint32_t monitor_id;
- /**
- * Request multiplexing
- */
- uint32_t id;
- };
- /**
- * Re-establish the connection to the ATS service.
- *
- * @param ph handle to use to re-connect.
- */
- static void
- reconnect (struct GNUNET_ATS_PerformanceHandle *ph);
- /**
- * Re-establish the connection to the ATS service.
- *
- * @param cls handle to use to re-connect.
- * @param tc scheduler context
- */
- static void
- reconnect_task (void *cls,
- const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- struct GNUNET_ATS_PerformanceHandle *ph = cls;
- ph->task = NULL;
- reconnect (ph);
- }
- /**
- * Transmit messages from the message queue to the service
- * (if there are any, and if we are not already trying).
- *
- * @param ph handle to use
- */
- static void
- do_transmit (struct GNUNET_ATS_PerformanceHandle *ph);
- /**
- * Type of a function to call when we receive a message
- * from the service.
- *
- * @param cls the `struct GNUNET_ATS_SchedulingHandle`
- * @param msg message received, NULL on timeout or fatal error
- */
- static void
- process_ats_message (void *cls,
- const struct GNUNET_MessageHeader *msg);
- /**
- * We can now transmit a message to ATS. Do it.
- *
- * @param cls the `struct GNUNET_ATS_PerformanceHandle`
- * @param size number of bytes we can transmit to ATS
- * @param buf where to copy the messages
- * @return number of bytes copied into @a buf
- */
- static size_t
- transmit_message_to_ats (void *cls,
- size_t size,
- void *buf)
- {
- struct GNUNET_ATS_PerformanceHandle *ph = cls;
- struct PendingMessage *p;
- size_t ret;
- char *cbuf;
- ph->th = NULL;
- ret = 0;
- cbuf = buf;
- while ((NULL != (p = ph->pending_head)) && (p->size <= size))
- {
- memcpy (&cbuf[ret], &p[1], p->size);
- ret += p->size;
- size -= p->size;
- GNUNET_CONTAINER_DLL_remove (ph->pending_head,
- ph->pending_tail,
- p);
- GNUNET_free(p);
- }
- do_transmit (ph);
- return ret;
- }
- /**
- * Transmit messages from the message queue to the service
- * (if there are any, and if we are not already trying).
- *
- * @param ph handle to use
- */
- static void
- do_transmit (struct GNUNET_ATS_PerformanceHandle *ph)
- {
- struct PendingMessage *p;
- if (NULL != ph->th)
- return;
- if (NULL == (p = ph->pending_head))
- return;
- if (NULL == ph->client)
- return; /* currently reconnecting */
- ph->th = GNUNET_CLIENT_notify_transmit_ready (ph->client,
- p->size,
- GNUNET_TIME_UNIT_FOREVER_REL,
- GNUNET_YES,
- &transmit_message_to_ats, ph);
- }
- /**
- * We received a peer information message. Validate and process it.
- *
- * @param ph our context with the callback
- * @param msg the message
- * @return #GNUNET_OK if the message was well-formed
- */
- static int
- process_pi_message (struct GNUNET_ATS_PerformanceHandle *ph,
- const struct GNUNET_MessageHeader *msg)
- {
- const struct PeerInformationMessage *pi;
- const char *plugin_address;
- const char *plugin_name;
- struct GNUNET_HELLO_Address address;
- uint16_t plugin_address_length;
- uint16_t plugin_name_length;
- int addr_active;
- struct GNUNET_ATS_Properties prop;
- if (ntohs (msg->size) < sizeof(struct PeerInformationMessage))
- {
- GNUNET_break(0);
- return GNUNET_SYSERR;
- }
- pi = (const struct PeerInformationMessage *) msg;
- plugin_address_length = ntohs (pi->address_length);
- plugin_name_length = ntohs (pi->plugin_name_length);
- addr_active = (int) ntohl (pi->address_active);
- plugin_address = (const char *) &pi[1];
- plugin_name = &plugin_address[plugin_address_length];
- if ((plugin_address_length + plugin_name_length
- + sizeof(struct PeerInformationMessage) != ntohs (msg->size))
- || (plugin_name[plugin_name_length - 1] != '\0'))
- {
- GNUNET_break(0);
- return GNUNET_SYSERR;
- }
- if (NULL != ph->addr_info_cb)
- {
- GNUNET_ATS_properties_ntoh (&prop,
- &pi->properties);
- address.peer = pi->peer;
- address.address = plugin_address;
- address.address_length = plugin_address_length;
- address.transport_name = plugin_name;
- ph->addr_info_cb (ph->addr_info_cb_cls,
- &address,
- addr_active,
- pi->bandwidth_out,
- pi->bandwidth_in,
- &prop);
- }
- return GNUNET_OK;
- }
- /**
- * We received a reservation result message. Validate and process it.
- *
- * @param ph our context with the callback
- * @param msg the message
- * @return #GNUNET_OK if the message was well-formed
- */
- static int
- process_rr_message (struct GNUNET_ATS_PerformanceHandle *ph,
- const struct GNUNET_MessageHeader *msg)
- {
- const struct ReservationResultMessage *rr;
- struct GNUNET_ATS_ReservationContext *rc;
- int32_t amount;
- if (ntohs (msg->size) < sizeof(struct ReservationResultMessage))
- {
- GNUNET_break(0);
- return GNUNET_SYSERR;
- }
- rr = (const struct ReservationResultMessage *) msg;
- amount = ntohl (rr->amount);
- rc = ph->reservation_head;
- if (0 != memcmp (&rr->peer, &rc->peer, sizeof(struct GNUNET_PeerIdentity)))
- {
- GNUNET_break(0);
- return GNUNET_SYSERR;
- }
- GNUNET_CONTAINER_DLL_remove (ph->reservation_head,
- ph->reservation_tail,
- rc);
- if ( (0 == amount) ||
- (NULL != rc->rcb) )
- {
- /* tell client if not cancelled */
- if (rc->rcb != NULL )
- rc->rcb (rc->rcb_cls,
- &rr->peer,
- amount,
- GNUNET_TIME_relative_ntoh (rr->res_delay));
- GNUNET_free(rc);
- return GNUNET_OK;
- }
- /* amount non-zero, but client cancelled, consider undo! */
- if (GNUNET_YES != rc->undo)
- {
- GNUNET_free(rc);
- return GNUNET_OK; /* do not try to undo failed undos or negative amounts */
- }
- GNUNET_free(rc);
- (void) GNUNET_ATS_reserve_bandwidth (ph,
- &rr->peer,
- -amount,
- NULL, NULL);
- return GNUNET_OK;
- }
- /**
- * We received a reservation result message. Validate and process it.
- *
- * @param ph our context with the callback
- * @param msg the message
- * @return #GNUNET_OK if the message was well-formed
- */
- static int
- process_ar_message (struct GNUNET_ATS_PerformanceHandle *ph,
- const struct GNUNET_MessageHeader *msg)
- {
- const struct PeerInformationMessage *pi;
- struct GNUNET_ATS_AddressListHandle *alh;
- struct GNUNET_ATS_AddressListHandle *next;
- const char *plugin_address;
- const char *plugin_name;
- struct GNUNET_HELLO_Address address;
- struct GNUNET_PeerIdentity allzeros;
- struct GNUNET_BANDWIDTH_Value32NBO bandwidth_zero;
- struct GNUNET_ATS_Properties prop;
- uint16_t plugin_address_length;
- uint16_t plugin_name_length;
- uint32_t active;
- uint32_t id;
- if (ntohs (msg->size) < sizeof(struct PeerInformationMessage))
- {
- GNUNET_break(0);
- return GNUNET_SYSERR;
- }
- pi = (const struct PeerInformationMessage *) msg;
- id = ntohl (pi->id);
- active = ntohl (pi->address_active);
- plugin_address_length = ntohs (pi->address_length);
- plugin_name_length = ntohs (pi->plugin_name_length);
- plugin_address = (const char *) &pi[1];
- plugin_name = &plugin_address[plugin_address_length];
- if ( (plugin_address_length + plugin_name_length
- + sizeof (struct PeerInformationMessage) != ntohs (msg->size)) ||
- (plugin_name[plugin_name_length - 1] != '\0') )
- {
- GNUNET_break(0);
- return GNUNET_SYSERR;
- }
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Received ATS_ADDRESSLIST_RESPONSE message for peer %s and plugin %s\n",
- GNUNET_i2s (&pi->peer),
- plugin_name);
- next = ph->addresslist_head;
- while (NULL != (alh = next))
- {
- next = alh->next;
- if (alh->id == id)
- break;
- }
- if (NULL == alh)
- {
- /* was canceled */
- return GNUNET_SYSERR;
- }
- memset (&allzeros, '\0', sizeof (allzeros));
- if ( (0 == memcmp (&allzeros, &pi->peer, sizeof(allzeros))) &&
- (0 == plugin_name_length) &&
- (0 == plugin_address_length) )
- {
- /* Done */
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Received last message for ATS_ADDRESSLIST_RESPONSE\n");
- bandwidth_zero.value__ = htonl (0);
- GNUNET_CONTAINER_DLL_remove (ph->addresslist_head,
- ph->addresslist_tail,
- alh);
- if (NULL != alh->cb)
- alh->cb (ph->addr_info_cb_cls,
- NULL,
- GNUNET_NO,
- bandwidth_zero,
- bandwidth_zero,
- NULL);
- GNUNET_free (alh);
- return GNUNET_OK;
- }
- address.peer = pi->peer;
- address.address = plugin_address;
- address.address_length = plugin_address_length;
- address.transport_name = plugin_name;
- if ( ( (GNUNET_YES == alh->all_addresses) ||
- (GNUNET_YES == active) ) &&
- (NULL != alh->cb) )
- {
- GNUNET_ATS_properties_ntoh (&prop,
- &pi->properties);
- alh->cb (ph->addr_info_cb_cls,
- &address,
- active,
- pi->bandwidth_out,
- pi->bandwidth_in,
- &prop);
- }
- return GNUNET_OK;
- }
- /**
- * Type of a function to call when we receive a message
- * from the service.
- *
- * @param cls the 'struct GNUNET_ATS_SchedulingHandle'
- * @param msg message received, NULL on timeout or fatal error
- */
- static void
- process_ats_message (void *cls,
- const struct GNUNET_MessageHeader *msg)
- {
- struct GNUNET_ATS_PerformanceHandle *ph = cls;
- if (NULL == msg)
- goto reconnect;
- switch (ntohs (msg->type))
- {
- case GNUNET_MESSAGE_TYPE_ATS_PEER_INFORMATION:
- if (GNUNET_OK != process_pi_message (ph, msg))
- goto reconnect;
- break;
- case GNUNET_MESSAGE_TYPE_ATS_RESERVATION_RESULT:
- if (GNUNET_OK != process_rr_message (ph, msg))
- goto reconnect;
- break;
- case GNUNET_MESSAGE_TYPE_ATS_ADDRESSLIST_RESPONSE:
- if (GNUNET_OK != process_ar_message (ph, msg))
- goto reconnect;
- break;
- default:
- GNUNET_break(0);
- goto reconnect;
- }
- ph->backoff = GNUNET_TIME_UNIT_ZERO;
- GNUNET_CLIENT_receive (ph->client,
- &process_ats_message,
- ph,
- GNUNET_TIME_UNIT_FOREVER_REL);
- return;
- reconnect:
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Reconnecting!\n");
- if (NULL != ph->th)
- {
- GNUNET_CLIENT_notify_transmit_ready_cancel (ph->th);
- ph->th = NULL;
- }
- GNUNET_CLIENT_disconnect (ph->client);
- ph->client = NULL;
- if (NULL != ph->addr_info_cb)
- {
- /* Indicate reconnect */
- ph->addr_info_cb (ph->addr_info_cb_cls,
- NULL,
- GNUNET_NO,
- GNUNET_BANDWIDTH_value_init (0),
- GNUNET_BANDWIDTH_value_init (0),
- NULL);
- }
- ph->backoff = GNUNET_TIME_STD_BACKOFF (ph->backoff);
- ph->task = GNUNET_SCHEDULER_add_delayed (ph->backoff,
- &reconnect_task,
- ph);
- }
- /**
- * Re-establish the connection to the ATS service.
- *
- * @param ph handle to use to re-connect.
- */
- static void
- reconnect (struct GNUNET_ATS_PerformanceHandle *ph)
- {
- struct PendingMessage *p;
- struct ClientStartMessage *init;
- GNUNET_assert (NULL == ph->client);
- ph->client = GNUNET_CLIENT_connect ("ats",
- ph->cfg);
- GNUNET_assert (NULL != ph->client);
- GNUNET_CLIENT_receive (ph->client,
- &process_ats_message,
- ph,
- GNUNET_TIME_UNIT_FOREVER_REL);
- if ((NULL == (p = ph->pending_head)) || (GNUNET_YES != p->is_init))
- {
- p = GNUNET_malloc (sizeof (struct PendingMessage) +
- sizeof (struct ClientStartMessage));
- p->size = sizeof(struct ClientStartMessage);
- p->is_init = GNUNET_YES;
- init = (struct ClientStartMessage *) &p[1];
- init->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_START);
- init->header.size = htons (sizeof(struct ClientStartMessage));
- init->start_flag = htonl ( (NULL == ph->addr_info_cb)
- ? START_FLAG_PERFORMANCE_NO_PIC
- : START_FLAG_PERFORMANCE_WITH_PIC);
- GNUNET_CONTAINER_DLL_insert (ph->pending_head,
- ph->pending_tail,
- p);
- }
- do_transmit (ph);
- }
- /**
- * Get handle to access performance API of the ATS subsystem.
- *
- * @param cfg configuration to use
- * @param addr_info_cb callback called when performance characteristics for
- * an address change
- * @param addr_info_cb_cls closure for @a addr_info_cb
- * @return ats performance context
- */
- struct GNUNET_ATS_PerformanceHandle *
- GNUNET_ATS_performance_init (const struct GNUNET_CONFIGURATION_Handle *cfg,
- GNUNET_ATS_AddressInformationCallback addr_info_cb,
- void *addr_info_cb_cls)
- {
- struct GNUNET_ATS_PerformanceHandle *ph;
- ph = GNUNET_new (struct GNUNET_ATS_PerformanceHandle);
- ph->cfg = cfg;
- ph->addr_info_cb = addr_info_cb;
- ph->addr_info_cb_cls = addr_info_cb_cls;
- ph->id = 0;
- reconnect (ph);
- return ph;
- }
- /**
- * Client is done using the ATS performance subsystem, release resources.
- *
- * @param ph handle
- */
- void
- GNUNET_ATS_performance_done (struct GNUNET_ATS_PerformanceHandle *ph)
- {
- struct PendingMessage *p;
- struct GNUNET_ATS_ReservationContext *rc;
- struct GNUNET_ATS_AddressListHandle *alh;
- while (NULL != (p = ph->pending_head))
- {
- GNUNET_CONTAINER_DLL_remove (ph->pending_head,
- ph->pending_tail,
- p);
- GNUNET_free (p);
- }
- while (NULL != (alh = ph->addresslist_head))
- {
- GNUNET_CONTAINER_DLL_remove (ph->addresslist_head,
- ph->addresslist_tail,
- alh);
- GNUNET_free (alh);
- }
- while (NULL != (rc = ph->reservation_head))
- {
- GNUNET_CONTAINER_DLL_remove (ph->reservation_head,
- ph->reservation_tail,
- rc);
- GNUNET_break (NULL == rc->rcb);
- GNUNET_free (rc);
- }
- if (NULL != ph->task)
- {
- GNUNET_SCHEDULER_cancel (ph->task);
- ph->task = NULL;
- }
- if (NULL != ph->client)
- {
- GNUNET_CLIENT_disconnect (ph->client);
- ph->client = NULL;
- }
- GNUNET_free (ph);
- }
- /**
- * Reserve inbound bandwidth from the given peer. ATS will look at
- * the current amount of traffic we receive from the peer and ensure
- * that the peer could add 'amount' of data to its stream.
- *
- * @param ph performance handle
- * @param peer identifies the peer
- * @param amount reserve N bytes for receiving, negative
- * amounts can be used to undo a (recent) reservation;
- * @param rcb function to call with the resulting reservation information
- * @param rcb_cls closure for @a rcb
- * @return NULL on error
- * @deprecated will be replaced soon
- */
- struct GNUNET_ATS_ReservationContext *
- GNUNET_ATS_reserve_bandwidth (struct GNUNET_ATS_PerformanceHandle *ph,
- const struct GNUNET_PeerIdentity *peer,
- int32_t amount,
- GNUNET_ATS_ReservationCallback rcb, void *rcb_cls)
- {
- struct GNUNET_ATS_ReservationContext *rc;
- struct PendingMessage *p;
- struct ReservationRequestMessage *m;
- rc = GNUNET_new (struct GNUNET_ATS_ReservationContext);
- rc->size = amount;
- rc->peer = *peer;
- rc->rcb = rcb;
- rc->rcb_cls = rcb_cls;
- if ( (NULL != rcb) &&
- (amount > 0) )
- rc->undo = GNUNET_YES;
- GNUNET_CONTAINER_DLL_insert_tail (ph->reservation_head,
- ph->reservation_tail,
- rc);
- p = GNUNET_malloc (sizeof (struct PendingMessage) +
- sizeof (struct ReservationRequestMessage));
- p->size = sizeof(struct ReservationRequestMessage);
- p->is_init = GNUNET_NO;
- m = (struct ReservationRequestMessage *) &p[1];
- m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_RESERVATION_REQUEST);
- m->header.size = htons (sizeof(struct ReservationRequestMessage));
- m->amount = htonl (amount);
- m->peer = *peer;
- GNUNET_CONTAINER_DLL_insert_tail (ph->pending_head,
- ph->pending_tail,
- p);
- do_transmit (ph);
- return rc;
- }
- /**
- * Cancel request for reserving bandwidth.
- *
- * @param rc context returned by the original GNUNET_ATS_reserve_bandwidth call
- */
- void
- GNUNET_ATS_reserve_bandwidth_cancel (struct GNUNET_ATS_ReservationContext *rc)
- {
- rc->rcb = NULL;
- }
- /**
- * Get information about addresses known to the ATS subsystem.
- *
- * @param handle the performance handle to use
- * @param peer peer idm can be NULL for all peers
- * @param all #GNUNET_YES to get information about all addresses or #GNUNET_NO to
- * get only address currently used
- * @param infocb callback to call with the addresses,
- * will callback with address == NULL when done
- * @param infocb_cls closure for @a infocb
- * @return ats performance context
- */
- struct GNUNET_ATS_AddressListHandle*
- GNUNET_ATS_performance_list_addresses (struct GNUNET_ATS_PerformanceHandle *handle,
- const struct GNUNET_PeerIdentity *peer,
- int all,
- GNUNET_ATS_AddressInformationCallback infocb,
- void *infocb_cls)
- {
- struct GNUNET_ATS_AddressListHandle *alh;
- struct PendingMessage *p;
- struct AddressListRequestMessage *m;
- if (NULL == infocb)
- return NULL;
- alh = GNUNET_new (struct GNUNET_ATS_AddressListHandle);
- alh->id = handle->id;
- handle->id++;
- alh->cb = infocb;
- alh->cb_cls = infocb_cls;
- alh->ph = handle;
- alh->all_addresses = all;
- if (NULL == peer)
- {
- alh->all_peers = GNUNET_YES;
- }
- else
- {
- alh->all_peers = GNUNET_NO;
- alh->peer = *peer;
- }
- GNUNET_CONTAINER_DLL_insert (handle->addresslist_head,
- handle->addresslist_tail,
- alh);
- p = GNUNET_malloc (sizeof (struct PendingMessage) +
- sizeof (struct AddressListRequestMessage));
- p->size = sizeof (struct AddressListRequestMessage);
- m = (struct AddressListRequestMessage *) &p[1];
- m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_ADDRESSLIST_REQUEST);
- m->header.size = htons (sizeof(struct AddressListRequestMessage));
- m->all = htonl (all);
- m->id = htonl (alh->id);
- if (NULL != peer)
- m->peer = *peer;
- GNUNET_CONTAINER_DLL_insert_tail (handle->pending_head,
- handle->pending_tail,
- p);
- do_transmit (handle);
- return alh;
- }
- /**
- * Cancel a pending address listing operation
- *
- * @param handle the handle of the request to cancel
- */
- void
- GNUNET_ATS_performance_list_addresses_cancel (struct GNUNET_ATS_AddressListHandle *handle)
- {
- GNUNET_CONTAINER_DLL_remove (handle->ph->addresslist_head,
- handle->ph->addresslist_tail,
- handle);
- GNUNET_free (handle);
- }
- /**
- * Convert a `enum GNUNET_ATS_PreferenceType` to a string
- *
- * @param type the preference type
- * @return a string or NULL if invalid
- */
- const char *
- GNUNET_ATS_print_preference_type (uint32_t type)
- {
- const char *prefs[] = GNUNET_ATS_PreferenceTypeString;
- if (type < GNUNET_ATS_PREFERENCE_END)
- return prefs[type];
- return NULL;
- }
- /**
- * Change preferences for the given peer. Preference changes are forgotten if peers
- * disconnect.
- *
- * @param ph performance handle
- * @param peer identifies the peer
- * @param ... #GNUNET_ATS_PREFERENCE_END-terminated specification of the desired changes
- */
- void
- GNUNET_ATS_performance_change_preference (struct GNUNET_ATS_PerformanceHandle *ph,
- const struct GNUNET_PeerIdentity *peer, ...)
- {
- struct PendingMessage *p;
- struct ChangePreferenceMessage *m;
- size_t msize;
- uint32_t count;
- struct PreferenceInformation *pi;
- va_list ap;
- enum GNUNET_ATS_PreferenceKind kind;
- count = 0;
- va_start(ap, peer);
- while (GNUNET_ATS_PREFERENCE_END !=
- (kind = va_arg (ap, enum GNUNET_ATS_PreferenceKind) ))
- {
- switch (kind)
- {
- case GNUNET_ATS_PREFERENCE_BANDWIDTH:
- count++;
- (void) va_arg (ap, double);
- break;
- case GNUNET_ATS_PREFERENCE_LATENCY:
- count++;
- (void) va_arg (ap, double);
- break;
- default:
- GNUNET_assert(0);
- }
- }
- va_end(ap);
- msize = count * sizeof(struct PreferenceInformation)
- + sizeof(struct ChangePreferenceMessage);
- p = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
- p->size = msize;
- p->is_init = GNUNET_NO;
- m = (struct ChangePreferenceMessage *) &p[1];
- m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_PREFERENCE_CHANGE);
- m->header.size = htons (msize);
- m->num_preferences = htonl (count);
- m->peer = *peer;
- pi = (struct PreferenceInformation *) &m[1];
- count = 0;
- va_start(ap, peer);
- while (GNUNET_ATS_PREFERENCE_END != (kind =
- va_arg (ap, enum GNUNET_ATS_PreferenceKind) ))
- {
- pi[count].preference_kind = htonl (kind);
- switch (kind)
- {
- case GNUNET_ATS_PREFERENCE_BANDWIDTH:
- pi[count].preference_value = (float) va_arg (ap, double);
- count++;
- break;
- case GNUNET_ATS_PREFERENCE_LATENCY:
- pi[count].preference_value = (float) va_arg (ap, double);
- count++;
- break;
- default:
- GNUNET_assert(0);
- }
- }
- va_end(ap);
- GNUNET_CONTAINER_DLL_insert_tail(ph->pending_head, ph->pending_tail, p);
- do_transmit (ph);
- }
- /**
- * Send feedback to ATS on how good a the requirements for a peer and a
- * preference is satisfied by ATS
- *
- * @param ph performance handle
- * @param scope the time interval this valid for: [now - scope .. now]
- * @param peer identifies the peer
- * @param ... #GNUNET_ATS_PREFERENCE_END-terminated specification of the desired changes
- */
- void
- GNUNET_ATS_performance_give_feedback (struct GNUNET_ATS_PerformanceHandle *ph,
- const struct GNUNET_PeerIdentity *peer,
- const struct GNUNET_TIME_Relative scope, ...)
- {
- struct PendingMessage *p;
- struct FeedbackPreferenceMessage *m;
- size_t msize;
- uint32_t count;
- struct PreferenceInformation *pi;
- va_list ap;
- enum GNUNET_ATS_PreferenceKind kind;
- count = 0;
- va_start(ap, scope);
- while (GNUNET_ATS_PREFERENCE_END !=
- (kind = va_arg (ap, enum GNUNET_ATS_PreferenceKind) ))
- {
- switch (kind)
- {
- case GNUNET_ATS_PREFERENCE_BANDWIDTH:
- count++;
- (void) va_arg (ap, double);
- break;
- case GNUNET_ATS_PREFERENCE_LATENCY:
- count++;
- (void) va_arg (ap, double);
- break;
- default:
- GNUNET_assert(0);
- }
- }
- va_end(ap);
- msize = count * sizeof(struct PreferenceInformation)
- + sizeof(struct FeedbackPreferenceMessage);
- p = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
- p->size = msize;
- p->is_init = GNUNET_NO;
- m = (struct FeedbackPreferenceMessage *) &p[1];
- m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_PREFERENCE_FEEDBACK);
- m->header.size = htons (msize);
- m->scope = GNUNET_TIME_relative_hton (scope);
- m->num_feedback = htonl (count);
- m->peer = *peer;
- pi = (struct PreferenceInformation *) &m[1];
- count = 0;
- va_start(ap, scope);
- while (GNUNET_ATS_PREFERENCE_END != (kind =
- va_arg (ap, enum GNUNET_ATS_PreferenceKind) ))
- {
- pi[count].preference_kind = htonl (kind);
- switch (kind)
- {
- case GNUNET_ATS_PREFERENCE_BANDWIDTH:
- pi[count].preference_value = (float) va_arg (ap, double);
- count++;
- break;
- case GNUNET_ATS_PREFERENCE_LATENCY:
- pi[count].preference_value = (float) va_arg (ap, double);
- count++;
- break;
- default:
- GNUNET_assert(0);
- }
- }
- va_end(ap);
- GNUNET_CONTAINER_DLL_insert_tail (ph->pending_head,
- ph->pending_tail,
- p);
- do_transmit (ph);
- }
- /* end of ats_api_performance.c */
|