src/ats/ats_api_performance.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2010,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  * @file ats/ats_api_performance.c
  22  * @brief automatic transport selection and outbound bandwidth determination
  23  * @author Christian Grothoff
  24  * @author Matthias Wachs
  25   */
  26 #include "platform.h"
  27 #include "gnunet_ats_service.h"
  28 #include "ats.h"
  29
  30
  31 /**
  32  * Message in linked list we should send to the ATS service.  The
  33  * actual binary message follows this struct.
  34  */
  35 struct PendingMessage
  36 {
  37
  38   /**
  39    * Kept in a DLL.
  40    */
  41   struct PendingMessage *next;
  42
  43   /**
  44    * Kept in a DLL.
  45    */
  46   struct PendingMessage *prev;
  47
  48   /**
  49    * Size of the message.
  50    */
  51   size_t size;
  52
  53   /**
  54    * Is this the 'ATS_START' message?
  55    */
  56   int is_init;
  57 };
  58
  59
  60 /**
  61  * Linked list of pending reservations.
  62  */
  63 struct GNUNET_ATS_ReservationContext
  64 {
  65
  66   /**
  67    * Kept in a DLL.
  68    */
  69   struct GNUNET_ATS_ReservationContext *next;
  70
  71   /**
  72    * Kept in a DLL.
  73    */
  74   struct GNUNET_ATS_ReservationContext *prev;
  75
  76   /**
  77    * Target peer.
  78    */
  79   struct GNUNET_PeerIdentity peer;
  80
  81   /**
  82    * Desired reservation
  83    */
  84   int32_t size;
  85
  86   /**
  87    * Function to call on result.
  88    */
  89   GNUNET_ATS_ReservationCallback rcb;
  90
  91   /**
  92    * Closure for 'rcb'
  93    */
  94   void *rcb_cls;
  95
  96   /**
  97    * Do we need to undo this reservation if it succeeded?  Set to
  98    * GNUNET_YES if a reservation is cancelled.  (at that point, 'info'
  99    * is also set to NULL; however, info will ALSO be NULL for the
 100    * reservation context that is created to undo the original request,
 101    * so 'info' being NULL cannot be used to check if undo is
 102    * required).
 103    */
 104   int undo;
 105 };
 106
 107
 108 /**
 109  * Linked list of pending reservations.
 110  */
 111 struct GNUNET_ATS_AddressListHandle
 112 {
 113
 114   /**
 115    * Kept in a DLL.
 116    */
 117   struct GNUNET_ATS_AddressListHandle *next;
 118
 119   /**
 120    * Kept in a DLL.
 121    */
 122   struct GNUNET_ATS_AddressListHandle *prev;
 123
 124   /**
 125    * Performance handle
 126    */
 127   struct GNUNET_ATS_PerformanceHandle *ph;
 128
 129   /**
 130    * Callback
 131    */
 132   GNUNET_ATS_PeerInformationCallback cb;
 133
 134   /**
 135    * Callback closure
 136    */
 137   void *cb_cls;
 138
 139   /**
 140    * Target peer.
 141    */
 142   struct GNUNET_PeerIdentity peer;
 143
 144   /**
 145    * Return all or specific peer only
 146    */
 147   int all_peers;
 148
 149   /**
 150    * Return all or used address only
 151    */
 152   int all_addresses;
 153
 154   /**
 155    * Request multiplexing
 156    */
 157   uint32_t id;
 158 };
 159
 160 /**
 161  * ATS Handle to obtain and/or modify performance information.
 162  */
 163 struct GNUNET_ATS_PerformanceHandle
 164 {
 165
 166   /**
 167    * Our configuration.
 168    */
 169   const struct GNUNET_CONFIGURATION_Handle *cfg;
 170
 171   /**
 172    * Callback to invoke on performance changes.
 173    */
 174   GNUNET_ATS_PeerInformationCallback infocb;
 175
 176   /**
 177    * Closure for 'infocb'.
 178    */
 179   void *infocb_cls;
 180
 181   /**
 182    * Connection to ATS service.
 183    */
 184   struct GNUNET_CLIENT_Connection *client;
 185
 186   /**
 187    * Head of list of messages for the ATS service.
 188    */
 189   struct PendingMessage *pending_head;
 190
 191   /**
 192    * Tail of list of messages for the ATS service
 193    */
 194   struct PendingMessage *pending_tail;
 195
 196   /**
 197    * Head of linked list of pending reservation requests.
 198    */
 199   struct GNUNET_ATS_ReservationContext *reservation_head;
 200
 201   /**
 202    * Tail of linked list of pending reservation requests.
 203    */
 204   struct GNUNET_ATS_ReservationContext *reservation_tail;
 205
 206   /**
 207    * Head of linked list of pending address list requests.
 208    */
 209   struct GNUNET_ATS_AddressListHandle *addresslist_head;
 210
 211   /**
 212    * Tail of linked list of pending address list requests.
 213    */
 214   struct GNUNET_ATS_AddressListHandle *addresslist_tail;
 215
 216   /**
 217    * Current request for transmission to ATS.
 218    */
 219   struct GNUNET_CLIENT_TransmitHandle *th;
 220
 221   /**
 222    * Task to trigger reconnect.
 223    */
 224   GNUNET_SCHEDULER_TaskIdentifier task;
 225
 226   /**
 227    * Request multiplexing
 228    */
 229   uint32_t id;
 230 };
 231
 232
 233 /**
 234  * Re-establish the connection to the ATS service.
 235  *
 236  * @param ph handle to use to re-connect.
 237  */
 238 static void
 239 reconnect (struct GNUNET_ATS_PerformanceHandle *ph);
 240
 241
 242 /**
 243  * Re-establish the connection to the ATS service.
 244  *
 245  * @param cls handle to use to re-connect.
 246  * @param tc scheduler context
 247  */
 248 static void
 249 reconnect_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 250 {
 251   struct GNUNET_ATS_PerformanceHandle *ph = cls;
 252
 253   ph->task = GNUNET_SCHEDULER_NO_TASK;
 254   reconnect (ph);
 255 }
 256
 257
 258 /**
 259  * Transmit messages from the message queue to the service
 260  * (if there are any, and if we are not already trying).
 261  *
 262  * @param ph handle to use
 263  */
 264 static void
 265 do_transmit (struct GNUNET_ATS_PerformanceHandle *ph);
 266
 267
 268 /**
 269  * We can now transmit a message to ATS. Do it.
 270  *
 271  * @param cls the 'struct GNUNET_ATS_SchedulingHandle'
 272  * @param size number of bytes we can transmit to ATS
 273  * @param buf where to copy the messages
 274  * @return number of bytes copied into buf
 275  */
 276 static size_t
 277 transmit_message_to_ats (void *cls, size_t size, void *buf)
 278 {
 279   struct GNUNET_ATS_PerformanceHandle *ph = cls;
 280   struct PendingMessage *p;
 281   size_t ret;
 282   char *cbuf;
 283
 284   ph->th = NULL;
 285   ret = 0;
 286   cbuf = buf;
 287   while ((NULL != (p = ph->pending_head)) && (p->size <= size))
 288   {
 289     memcpy (&cbuf[ret], &p[1], p->size);
 290     ret += p->size;
 291     size -= p->size;
 292     GNUNET_CONTAINER_DLL_remove (ph->pending_head, ph->pending_tail, p);
 293     GNUNET_free (p);
 294   }
 295   do_transmit (ph);
 296   return ret;
 297 }
 298
 299
 300 /**
 301  * Transmit messages from the message queue to the service
 302  * (if there are any, and if we are not already trying).
 303  *
 304  * @param ph handle to use
 305  */
 306 static void
 307 do_transmit (struct GNUNET_ATS_PerformanceHandle *ph)
 308 {
 309   struct PendingMessage *p;
 310
 311   if (NULL != ph->th)
 312     return;
 313   if (NULL == (p = ph->pending_head))
 314     return;
 315   if (NULL == ph->client)
 316     return;                     /* currently reconnecting */
 317   ph->th =
 318       GNUNET_CLIENT_notify_transmit_ready (ph->client, p->size,
 319                                            GNUNET_TIME_UNIT_FOREVER_REL,
 320                                            GNUNET_YES, &transmit_message_to_ats,
 321                                            ph);
 322 }
 323
 324
 325 /**
 326  * We received a peer information message.  Validate and process it.
 327  *
 328  * @param ph our context with the callback
 329  * @param msg the message
 330  * @return GNUNET_OK if the message was well-formed
 331  */
 332 static int
 333 process_pi_message (struct GNUNET_ATS_PerformanceHandle *ph,
 334                     const struct GNUNET_MessageHeader *msg)
 335 {
 336   const struct PeerInformationMessage *pi;
 337   const struct GNUNET_ATS_Information *atsi;
 338   const char *plugin_address;
 339   const char *plugin_name;
 340   struct GNUNET_HELLO_Address address;
 341   uint16_t plugin_address_length;
 342   uint16_t plugin_name_length;
 343   uint32_t ats_count;
 344
 345   if (ntohs (msg->size) < sizeof (struct PeerInformationMessage))
 346   {
 347     GNUNET_break (0);
 348     return GNUNET_SYSERR;
 349   }
 350
 351   pi = (const struct PeerInformationMessage *) msg;
 352   ats_count = ntohl (pi->ats_count);
 353   plugin_address_length = ntohs (pi->address_length);
 354   plugin_name_length = ntohs (pi->plugin_name_length);
 355   atsi = (const struct GNUNET_ATS_Information *) &pi[1];
 356   plugin_address = (const char *) &atsi[ats_count];
 357   plugin_name = &plugin_address[plugin_address_length];
 358   if ((plugin_address_length + plugin_name_length +
 359        ats_count * sizeof (struct GNUNET_ATS_Information) +
 360        sizeof (struct PeerInformationMessage) != ntohs (msg->size)) ||
 361       (ats_count >
 362        GNUNET_SERVER_MAX_MESSAGE_SIZE / sizeof (struct GNUNET_ATS_Information))
 363       || (plugin_name[plugin_name_length - 1] != '\0'))
 364   {
 365     GNUNET_break (0);
 366     return GNUNET_SYSERR;
 367   }
 368   if (ph->infocb == NULL)
 369   {
 370     return GNUNET_OK;
 371   }
 372
 373   address.peer = pi->peer;
 374   address.address = plugin_address;
 375   address.address_length = plugin_address_length;
 376   address.transport_name = plugin_name;
 377   ph->infocb (ph->infocb_cls, &address, pi->bandwidth_out, pi->bandwidth_in,
 378               atsi, ats_count);
 379   return GNUNET_OK;
 380 }
 381
 382
 383 /**
 384  * We received a reservation result message.  Validate and process it.
 385  *
 386  * @param ph our context with the callback
 387  * @param msg the message
 388  * @return GNUNET_OK if the message was well-formed
 389  */
 390 static int
 391 process_rr_message (struct GNUNET_ATS_PerformanceHandle *ph,
 392                     const struct GNUNET_MessageHeader *msg)
 393 {
 394   const struct ReservationResultMessage *rr;
 395   struct GNUNET_ATS_ReservationContext *rc;
 396   int32_t amount;
 397
 398   if (ntohs (msg->size) < sizeof (struct ReservationResultMessage))
 399   {
 400     GNUNET_break (0);
 401     return GNUNET_SYSERR;
 402   }
 403   rr = (const struct ReservationResultMessage *) msg;
 404   amount = ntohl (rr->amount);
 405   rc = ph->reservation_head;
 406   if (0 != memcmp (&rr->peer, &rc->peer, sizeof (struct GNUNET_PeerIdentity)))
 407   {
 408     GNUNET_break (0);
 409     return GNUNET_SYSERR;
 410   }
 411   GNUNET_CONTAINER_DLL_remove (ph->reservation_head, ph->reservation_tail, rc);
 412   if ((amount == 0) || (rc->rcb != NULL))
 413   {
 414     /* tell client if not cancelled */
 415     if (rc->rcb != NULL)
 416       rc->rcb (rc->rcb_cls, &rr->peer, amount,
 417                GNUNET_TIME_relative_ntoh (rr->res_delay));
 418     GNUNET_free (rc);
 419     return GNUNET_OK;
 420   }
 421   /* amount non-zero, but client cancelled, consider undo! */
 422   if (GNUNET_YES != rc->undo)
 423   {
 424     GNUNET_free (rc);
 425     return GNUNET_OK;           /* do not try to undo failed undos or negative amounts */
 426   }
 427   GNUNET_free (rc);
 428   (void) GNUNET_ATS_reserve_bandwidth (ph, &rr->peer, -amount, NULL, NULL);
 429   return GNUNET_OK;
 430 }
 431
 432
 433 /**
 434  * We received a reservation result message.  Validate and process it.
 435  *
 436  * @param ph our context with the callback
 437  * @param msg the message
 438  * @return GNUNET_OK if the message was well-formed
 439  */
 440 static int
 441 process_ar_message (struct GNUNET_ATS_PerformanceHandle *ph,
 442                     const struct GNUNET_MessageHeader *msg)
 443 {
 444   const struct PeerInformationMessage *pi;
 445   struct GNUNET_ATS_AddressListHandle *alh;
 446   struct GNUNET_ATS_AddressListHandle *next;
 447   const struct GNUNET_ATS_Information *atsi;
 448   const char *plugin_address;
 449   const char *plugin_name;
 450   struct GNUNET_HELLO_Address address;
 451   struct GNUNET_PeerIdentity allzeros;
 452   struct GNUNET_BANDWIDTH_Value32NBO bandwidth_zero;
 453   uint16_t plugin_address_length;
 454   uint16_t plugin_name_length;
 455   uint32_t ats_count;
 456   uint32_t active;
 457   uint32_t id;
 458
 459   if (ntohs (msg->size) < sizeof (struct PeerInformationMessage))
 460   {
 461     GNUNET_break (0);
 462     return GNUNET_SYSERR;
 463   }
 464   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 465       _("Received %s message\n"), "ATS_ADDRESSLIST_RESPONSE");
 466
 467   pi = (const struct PeerInformationMessage *) msg;
 468   id = ntohl (pi->id);
 469   ats_count = ntohl (pi->ats_count);
 470   active = ntohl (pi->address_active);
 471   plugin_address_length = ntohs (pi->address_length);
 472   plugin_name_length = ntohs (pi->plugin_name_length);
 473   atsi = (const struct GNUNET_ATS_Information *) &pi[1];
 474   plugin_address = (const char *) &atsi[ats_count];
 475   plugin_name = &plugin_address[plugin_address_length];
 476   if ((plugin_address_length + plugin_name_length +
 477        ats_count * sizeof (struct GNUNET_ATS_Information) +
 478        sizeof (struct PeerInformationMessage) != ntohs (msg->size)) ||
 479       (ats_count >
 480        GNUNET_SERVER_MAX_MESSAGE_SIZE / sizeof (struct GNUNET_ATS_Information))
 481       || (plugin_name[plugin_name_length - 1] != '\0'))
 482   {
 483     GNUNET_break (0);
 484     return GNUNET_SYSERR;
 485   }
 486
 487   next = ph->addresslist_head;
 488   while (NULL != (alh = next))
 489   {
 490       next = alh->next;
 491       if (alh->id == id)
 492         break;
 493   }
 494   if (NULL == alh)
 495   {
 496       GNUNET_break (0);
 497       return GNUNET_SYSERR;
 498   }
 499
 500   memset (&allzeros, '\0', sizeof (allzeros));
 501   if ((0 == memcmp (&allzeros, &pi->peer, sizeof (allzeros))) &&
 502       (0 == plugin_name_length) &&
 503       (0 == plugin_address_length) &&
 504       (0 == ats_count))
 505   {
 506       /* Done */
 507       bandwidth_zero.value__ = htonl (0);
 508       alh->cb (ph->infocb_cls,
 509               NULL,
 510               bandwidth_zero, bandwidth_zero,
 511               NULL, 0);
 512       return GNUNET_OK;
 513   }
 514
 515   address.peer = pi->peer;
 516   address.address = plugin_address;
 517   address.address_length = plugin_address_length;
 518   address.transport_name = plugin_name;
 519
 520   if ((GNUNET_YES == alh->all_peers) || (GNUNET_YES == active))
 521   {
 522     alh->cb (ph->infocb_cls,
 523             &address,
 524             pi->bandwidth_out, pi->bandwidth_in,
 525             atsi, ats_count);
 526   }
 527   return GNUNET_OK;
 528 }
 529
 530
 531 /**
 532  * Type of a function to call when we receive a message
 533  * from the service.
 534  *
 535  * @param cls the 'struct GNUNET_ATS_SchedulingHandle'
 536  * @param msg message received, NULL on timeout or fatal error
 537  */
 538 static void
 539 process_ats_message (void *cls, const struct GNUNET_MessageHeader *msg)
 540 {
 541   struct GNUNET_ATS_PerformanceHandle *ph = cls;
 542
 543   if (NULL == msg)
 544     goto reconnect;
 545   switch (ntohs (msg->type))
 546   {
 547   case GNUNET_MESSAGE_TYPE_ATS_PEER_INFORMATION:
 548     if (GNUNET_OK != process_pi_message (ph, msg))
 549       goto reconnect;
 550     break;
 551   case GNUNET_MESSAGE_TYPE_ATS_RESERVATION_RESULT:
 552     if (GNUNET_OK != process_rr_message (ph, msg))
 553       goto reconnect;
 554     break;
 555   case GNUNET_MESSAGE_TYPE_ATS_ADDRESSLIST_RESPONSE:
 556     if (GNUNET_OK != process_ar_message (ph, msg))
 557       goto reconnect;
 558     break;
 559   default:
 560     GNUNET_break (0);
 561     goto reconnect;
 562   }
 563   GNUNET_CLIENT_receive (ph->client, &process_ats_message, ph,
 564                          GNUNET_TIME_UNIT_FOREVER_REL);
 565   return;
 566 reconnect:
 567   if (NULL != ph->th)
 568   {
 569     GNUNET_CLIENT_notify_transmit_ready_cancel (ph->th);
 570     ph->th = NULL;
 571   }
 572   GNUNET_CLIENT_disconnect (ph->client);
 573   ph->client = NULL;
 574   ph->task =
 575       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, &reconnect_task,
 576                                     ph);
 577 }
 578
 579
 580 /**
 581  * Re-establish the connection to the ATS service.
 582  *
 583  * @param ph handle to use to re-connect.
 584  */
 585 static void
 586 reconnect (struct GNUNET_ATS_PerformanceHandle *ph)
 587 {
 588   struct PendingMessage *p;
 589   struct ClientStartMessage *init;
 590
 591   GNUNET_assert (NULL == ph->client);
 592   ph->client = GNUNET_CLIENT_connect ("ats", ph->cfg);
 593   GNUNET_assert (NULL != ph->client);
 594   GNUNET_CLIENT_receive (ph->client, &process_ats_message, ph,
 595                          GNUNET_TIME_UNIT_FOREVER_REL);
 596   if ((NULL == (p = ph->pending_head)) || (GNUNET_YES != p->is_init))
 597   {
 598     p = GNUNET_malloc (sizeof (struct PendingMessage) +
 599                        sizeof (struct ClientStartMessage));
 600     p->size = sizeof (struct ClientStartMessage);
 601     p->is_init = GNUNET_YES;
 602     init = (struct ClientStartMessage *) &p[1];
 603     init->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_START);
 604     init->header.size = htons (sizeof (struct ClientStartMessage));
 605     init->start_flag =
 606         htonl ((ph->infocb ==
 607                 NULL) ? START_FLAG_PERFORMANCE_NO_PIC :
 608                START_FLAG_PERFORMANCE_WITH_PIC);
 609     GNUNET_CONTAINER_DLL_insert (ph->pending_head, ph->pending_tail, p);
 610   }
 611   do_transmit (ph);
 612 }
 613
 614
 615
 616 /**
 617  * Get handle to access performance API of the ATS subsystem.
 618  *
 619  * @param cfg configuration to use
 620  * @param infocb function to call on allocation changes, can be NULL
 621  * @param infocb_cls closure for infocb
 622  * @return ats performance context
 623  */
 624 struct GNUNET_ATS_PerformanceHandle *
 625 GNUNET_ATS_performance_init (const struct GNUNET_CONFIGURATION_Handle *cfg,
 626                              GNUNET_ATS_PeerInformationCallback infocb,
 627                              void *infocb_cls)
 628 {
 629   struct GNUNET_ATS_PerformanceHandle *ph;
 630
 631   ph = GNUNET_malloc (sizeof (struct GNUNET_ATS_PerformanceHandle));
 632   ph->cfg = cfg;
 633   ph->infocb = infocb;
 634   ph->infocb_cls = infocb_cls;
 635   ph->id  = 0;
 636   reconnect (ph);
 637   return ph;
 638 }
 639
 640
 641 /**
 642  * Client is done using the ATS performance subsystem, release resources.
 643  *
 644  * @param ph handle
 645  */
 646 void
 647 GNUNET_ATS_performance_done (struct GNUNET_ATS_PerformanceHandle *ph)
 648 {
 649   struct PendingMessage *p;
 650   struct GNUNET_ATS_ReservationContext *rc;
 651   struct GNUNET_ATS_AddressListHandle *alh;
 652
 653   while (NULL != (p = ph->pending_head))
 654   {
 655     GNUNET_CONTAINER_DLL_remove (ph->pending_head, ph->pending_tail, p);
 656     GNUNET_free (p);
 657   }
 658   while (NULL != (alh = ph->addresslist_head))
 659   {
 660     GNUNET_CONTAINER_DLL_remove (ph->addresslist_head, ph->addresslist_tail,
 661                                  alh);
 662     GNUNET_free (alh);
 663   }
 664   while (NULL != (rc = ph->reservation_head))
 665   {
 666     GNUNET_CONTAINER_DLL_remove (ph->reservation_head, ph->reservation_tail,
 667                                  rc);
 668     GNUNET_break (NULL == rc->rcb);
 669     GNUNET_free (rc);
 670   }
 671   if (GNUNET_SCHEDULER_NO_TASK != ph->task)
 672   {
 673     GNUNET_SCHEDULER_cancel (ph->task);
 674     ph->task = GNUNET_SCHEDULER_NO_TASK;
 675   }
 676   if (NULL != ph->client)
 677   {
 678     GNUNET_CLIENT_disconnect (ph->client);
 679     ph->client = NULL;
 680   }
 681   GNUNET_free (ph);
 682 }
 683
 684
 685 /**
 686  * Reserve inbound bandwidth from the given peer.  ATS will look at
 687  * the current amount of traffic we receive from the peer and ensure
 688  * that the peer could add 'amount' of data to its stream.
 689  *
 690  * @param ph performance handle
 691  * @param peer identifies the peer
 692  * @param amount reserve N bytes for receiving, negative
 693  *                amounts can be used to undo a (recent) reservation;
 694  * @param rcb function to call with the resulting reservation information
 695  * @param rcb_cls closure for info
 696  * @return NULL on error
 697  * @deprecated will be replaced soon
 698  */
 699 struct GNUNET_ATS_ReservationContext *
 700 GNUNET_ATS_reserve_bandwidth (struct GNUNET_ATS_PerformanceHandle *ph,
 701                               const struct GNUNET_PeerIdentity *peer,
 702                               int32_t amount,
 703                               GNUNET_ATS_ReservationCallback rcb, void *rcb_cls)
 704 {
 705   struct GNUNET_ATS_ReservationContext *rc;
 706   struct PendingMessage *p;
 707   struct ReservationRequestMessage *m;
 708
 709   rc = GNUNET_malloc (sizeof (struct GNUNET_ATS_ReservationContext));
 710   rc->size = amount;
 711   rc->peer = *peer;
 712   rc->rcb = rcb;
 713   rc->rcb_cls = rcb_cls;
 714   if ((rcb != NULL) && (amount > 0))
 715     rc->undo = GNUNET_YES;
 716   GNUNET_CONTAINER_DLL_insert_tail (ph->reservation_head, ph->reservation_tail,
 717                                     rc);
 718
 719   p = GNUNET_malloc (sizeof (struct PendingMessage) +
 720                      sizeof (struct ReservationRequestMessage));
 721   p->size = sizeof (struct ReservationRequestMessage);
 722   p->is_init = GNUNET_NO;
 723   m = (struct ReservationRequestMessage *) &p[1];
 724   m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_RESERVATION_REQUEST);
 725   m->header.size = htons (sizeof (struct ReservationRequestMessage));
 726   m->amount = htonl (amount);
 727   m->peer = *peer;
 728   GNUNET_CONTAINER_DLL_insert_tail (ph->pending_head, ph->pending_tail, p);
 729   do_transmit (ph);
 730   return rc;
 731 }
 732
 733
 734 /**
 735  * Cancel request for reserving bandwidth.
 736  *
 737  * @param rc context returned by the original GNUNET_ATS_reserve_bandwidth call
 738  */
 739 void
 740 GNUNET_ATS_reserve_bandwidth_cancel (struct GNUNET_ATS_ReservationContext *rc)
 741 {
 742   rc->rcb = NULL;
 743 }
 744
 745 /**
 746  * Get information about addresses known to the ATS subsystem.
 747  *
 748  * @param cfg configuration to use
 749  * @param peer peer idm can be NULL for all peers
 750  * @param all GNUNET_YES to get information about all addresses or GNUNET_NO to
 751  *        get only address currently used
 752  * @param infocb callback to call with the addresses,
 753  *        will callback with address == NULL when done
 754  * @param infocb_cls closure for infocb
 755  * @return ats performance context
 756  */
 757 struct GNUNET_ATS_AddressListHandle*
 758 GNUNET_ATS_performance_list_addresses (struct GNUNET_ATS_PerformanceHandle *handle,
 759                                        const struct GNUNET_PeerIdentity *peer,
 760                                        int all,
 761                                        GNUNET_ATS_PeerInformationCallback infocb,
 762                                        void *infocb_cls)
 763 {
 764   struct GNUNET_ATS_AddressListHandle *alh;
 765   struct PendingMessage *p;
 766   struct AddressListRequestMessage *m;
 767
 768   GNUNET_assert (NULL != handle);
 769
 770   alh = GNUNET_malloc (sizeof (struct GNUNET_ATS_AddressListHandle));
 771   alh->id = handle->id;
 772   handle->id ++;
 773   alh->cb = infocb;
 774   alh->cb_cls = infocb_cls;
 775   alh->ph = handle;
 776   alh->all_addresses = all;
 777   if (NULL == peer)
 778     alh->all_peers = GNUNET_YES;
 779   else
 780   {
 781       alh->all_peers = GNUNET_NO;
 782       alh->peer = (*peer);
 783   }
 784
 785   GNUNET_CONTAINER_DLL_insert (handle->addresslist_head, handle->addresslist_tail, alh);
 786
 787   p = GNUNET_malloc (sizeof (struct PendingMessage) +
 788                      sizeof (struct AddressListRequestMessage));
 789   p->size = sizeof (struct AddressListRequestMessage);
 790   m = (struct AddressListRequestMessage *) &p[1];
 791   m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_ADDRESSLIST_REQUEST);
 792   m->header.size = htons (sizeof (struct AddressListRequestMessage));
 793   m->all = htonl (all);
 794   m->id = htonl (alh->id);
 795   if (NULL != peer)
 796     m->peer = *peer;
 797   else
 798   {
 799       memset (&m->peer, '\0', sizeof (struct GNUNET_PeerIdentity));
 800   }
 801   GNUNET_CONTAINER_DLL_insert_tail (handle->pending_head, handle->pending_tail, p);
 802   do_transmit (handle);
 803
 804   return alh;
 805 }
 806
 807
 808 /**
 809  * Cancel a pending address listing operation
 810  *
 811  * @param handle the GNUNET_ATS_AddressListHandle handle to cancel
 812  */
 813 void
 814 GNUNET_ATS_performance_list_addresses_cancel (struct GNUNET_ATS_AddressListHandle *handle)
 815 {
 816   GNUNET_assert (NULL != handle);
 817
 818   GNUNET_CONTAINER_DLL_remove (handle->ph->addresslist_head, handle->ph->addresslist_tail, handle);
 819   GNUNET_free (handle);
 820 }
 821
 822
 823 /**
 824  * Change preferences for the given peer. Preference changes are forgotten if peers
 825  * disconnect.
 826  *
 827  * @param ph performance handle
 828  * @param peer identifies the peer
 829  * @param ... 0-terminated specification of the desired changes
 830  */
 831 void
 832 GNUNET_ATS_change_preference (struct GNUNET_ATS_PerformanceHandle *ph,
 833                               const struct GNUNET_PeerIdentity *peer, ...)
 834 {
 835   struct PendingMessage *p;
 836   struct ChangePreferenceMessage *m;
 837   size_t msize;
 838   uint32_t count;
 839   struct PreferenceInformation *pi;
 840   va_list ap;
 841   enum GNUNET_ATS_PreferenceKind kind;
 842
 843   count = 0;
 844   va_start (ap, peer);
 845   while (GNUNET_ATS_PREFERENCE_END !=
 846          (kind = va_arg (ap, enum GNUNET_ATS_PreferenceKind)))
 847   {
 848     switch (kind)
 849     {
 850     case GNUNET_ATS_PREFERENCE_BANDWIDTH:
 851       count++;
 852       (void) va_arg (ap, double);
 853
 854       break;
 855     case GNUNET_ATS_PREFERENCE_LATENCY:
 856       count++;
 857       (void) va_arg (ap, double);
 858
 859       break;
 860     default:
 861       GNUNET_assert (0);
 862     }
 863   }
 864   va_end (ap);
 865   msize =
 866       count * sizeof (struct PreferenceInformation) +
 867       sizeof (struct ChangePreferenceMessage);
 868   p = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
 869   p->size = msize;
 870   p->is_init = GNUNET_NO;
 871   m = (struct ChangePreferenceMessage *) &p[1];
 872   m->header.type = htons (GNUNET_MESSAGE_TYPE_ATS_PREFERENCE_CHANGE);
 873   m->header.size = htons (msize);
 874   m->num_preferences = htonl (count);
 875   m->peer = *peer;
 876   pi = (struct PreferenceInformation *) &m[1];
 877   count = 0;
 878   va_start (ap, peer);
 879   while (GNUNET_ATS_PREFERENCE_END !=
 880          (kind = va_arg (ap, enum GNUNET_ATS_PreferenceKind)))
 881   {
 882     pi[count].preference_kind = htonl (kind);
 883     switch (kind)
 884     {
 885     case GNUNET_ATS_PREFERENCE_BANDWIDTH:
 886       pi[count].preference_value = (float) va_arg (ap, double);
 887
 888       count++;
 889       break;
 890     case GNUNET_ATS_PREFERENCE_LATENCY:
 891       pi[count].preference_value = (float) va_arg (ap, double);
 892
 893       count++;
 894       break;
 895     default:
 896       GNUNET_assert (0);
 897     }
 898   }
 899   va_end (ap);
 900   GNUNET_CONTAINER_DLL_insert_tail (ph->pending_head, ph->pending_tail, p);
 901   do_transmit (ph);
 902 }
 903
 904 /* end of ats_api_performance.c */