src/ats/gnunet-service-ats_addresses.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2011 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 /**
  22  * @file ats/gnunet-service-ats_addresses.c
  23  * @brief ats service address management
  24  * @author Matthias Wachs
  25  * @author Christian Grothoff
  26  */
  27 #include "platform.h"
  28 #include "gnunet_ats_service.h"
  29 #include "gnunet-service-ats.h"
  30 #include "gnunet-service-ats_addresses.h"
  31 #include "gnunet-service-ats_performance.h"
  32 #include "gnunet-service-ats_scheduling.h"
  33 #include "gnunet-service-ats_reservations.h"
  34 #if HAVE_LIBGLPK
  35 #include "gnunet-service-ats_addresses_mlp.h"
  36 #endif
  37 #include "gnunet-service-ats_addresses_simplistic.h"
  38
  39 /**
  40  * NOTE: Do not change this documentation. This documentation is based on
  41  * gnunet.org:/vcs/fsnsg/ats-paper.git/tech-doku/ats-tech-guide.tex
  42  * use build_txt.sh to generate plaintext output
  43  *
  44  *   1 ATS addresses : ATS address management
  45  *
  46  *    This ATS addresses ("addresses") component manages the addresses known to
  47  *    ATS service and suggests addresses to transport service when it is
  48  *    interested in address suggestion for a peer. ATS addresses also
  49  *    instantiates the bandwidth assignment mechanism (solver), notifies it
  50  *    about changes to addresses and forwards changes to bandwidth assignments
  51  *    to transport, depending if transport is interested in this change.
  52  *
  53  *     1.1 Input data
  54  *
  55  *       1.1.1 Addresses
  56  *
  57  *    Addresses are added by specifying peer ID, plugin, address, address length
  58  *    and session, if available. ATS information can be specified if available.
  59  *
  60  *       1.1.2 Networks
  61  *
  62  *    ATS specifies a fix set of networks an address can belong to. For each
  63  *    network an inbound and outbound quota will be specified. The available
  64  *    networks and addtional helper varaibles are defined in
  65  *    gnunet_ats_service.h. At the moment 5 networks are defined:
  66  *      * GNUNET_ATS_NET_UNSPECIFIED
  67  *      * GNUNET_ATS_NET_LOOPBACK
  68  *      * GNUNET_ATS_NET_LAN
  69  *      * GNUNET_ATS_NET_WAN
  70  *      * GNUNET_ATS_NET_WLAN
  71  *
  72  *    The total number of networks defined is stored in
  73  *    GNUNET_ATS_NetworkTypeCount GNUNET_ATS_NetworkType can be used array
  74  *    initializer for an int array, while GNUNET_ATS_NetworkType is an
  75  *    initializer for a char array containing a string description of all
  76  *    networks
  77  *
  78  *       1.1.3 Quotas
  79  *
  80  *    An inbound and outbound quota for each of the networks mentioned in 1.1.2
  81  *    is loaded from ats configuration during initialization. This quota defines
  82  *    to total amount of inbound and outbound traffic allowed for a specific
  83  *    network. The configuration values used are in section ats:
  84  *      * "NETWORK"_QUOTA_IN = <value>
  85  *      * "NETWORK"_QUOTA_IN = <value>
  86  *
  87  *    You can specify quotas by setting the <value> to a:
  88  *      * unrestricted: unlimited
  89  *      * number of bytes: e.g. 10240
  90  *      * fancy value: e.g. 64 Kib
  91  *
  92  *    unlimited is defined as GNUNET_ATS_MaxBandwidthString and equivalent to
  93  *    the value GNUNET_ATS_MaxBandwidth Important predefined values for quotas
  94  *    are:
  95  *      * GNUNET_ATS_DefaultBandwidth: 65536
  96  *      * GNUNET_ATS_MaxBandwidth: UINT32_MAX
  97  *      * GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT: 1024
  98  *
  99  *    Details of loading quotas and default values will be described on
 100  *
 101  *       1.1.4 Preference values
 102  *
 103  *     1.2 Data structures used
 104  *
 105  *    Addresse uses struct ATS_Address for each address. The structs are stored
 106  *    in a linked list and provides a pointer void *solver_information for the
 107  *    solver to store address specific information. It provides the int values
 108  *    active which is set to GNUNET_YES if the address is select for transport
 109  *    use and used, representing that transport service is actively using this
 110  *    address. Address information are stored in peer, addr, addr_len, plugin.
 111  *
 112  *     1.3 Initialization
 113  *
 114  *    During initialization a hashmap to store addresses is created. The quotas
 115  *    for all networks defined for ATS are loaded from configuration. For each
 116  *    network first the logic will check if the string
 117  *    GNUNET_ATS_MaxBandwidthString is configured, if not it will try to convert
 118  *    the configured value as a fancy size and if this fails it will try to use
 119  *    it as a value_number. If no configuration value is found it will assign
 120  *    GNUNET_ATS_DefaultBandwidth. The most important step is to load the
 121  *    configured solver using configuration "[ats]:MODE". Current solvers are
 122  *    MODE_SIMPLISTIC, MODE_MLP. Interaction is done using a solver API
 123  *
 124  *     1.4 Solver API
 125  *
 126  *    Solver functions:
 127  *      * s_init: init the solver with required information
 128  *      * s_add: add a new address
 129  *      * s_update: update ATS values or session for an address
 130  *      * s_get: get prefered address for a peer
 131  *      * s_del: delete an address
 132  *      * s_pref: change preference value for a peer
 133  *      * s_done: shutdown solver
 134  *
 135  *    Callbacks: addresses provides a bandwidth_changed_cb callback to the
 136  *    solver which is called when bandwidth assigned to peer has changed
 137  *
 138  *     1.5 Shutdown
 139  *
 140  *    During shutdown all addresses are freed and the solver told to shutdown
 141  *
 142  *     1.6 Addresses and sessions
 143  *
 144  *    Addresses consist of the address itself and a numerical session. When a
 145  *    new address without a session is added it has no session, so it gets
 146  *    session 0 assigned. When an address with a session is added and an address
 147  *    object with session 0 is found, this object is updated with the session
 148  *    otherwise a new address object with this session assigned is created.
 149  *
 150  *       1.6.1 Terminology
 151  *
 152  *    Addresses a1,a2 with session s1, s2 are "exact" if:
 153  *    (a1 == a2)&&(s1 == s2)
 154  *    Addresses a1,a2 with session s1, s2 are "equivalent" if:
 155  *    (a1 == a2)&&((s1 == s2)||(s1 == 0)||(s2 == 0)
 156  *
 157  *     1.7 Address management
 158  *
 159  *    Transport service notifies ATS about changes to the addresses known to
 160  *    him.
 161  *
 162  *       1.7.1 Adding an address
 163  *
 164  *    When transport learns a new address it tells ATS and ATS is telling
 165  *    addresses about it using GAS_address_add. If not known to addresses it
 166  *    creates a new address object and calls solver's s_add. ATS information are
 167  *    deserialized and solver is notified about the session and ATS information
 168  *    using s_update.
 169  *
 170  *       1.7.2 Updating an address
 171  *
 172  *    Addresses does an lookup up for the existing address with the given
 173  *    session. If disassembles included ATS information and notifies the solver
 174  *    using s_update about the update.
 175  *
 176  *       1.7.3 Deleting an address
 177  *
 178  *    Addresses does an lookup for the exact address and session and if removes
 179  *    this address. If session != 0 the session is set to 0 and the address is
 180  *    kept. If session == 0, the addresses is removed.
 181  *
 182  *       1.7.4 Requesting an address suggestion
 183  *
 184  *    The address client issues a request address message to be notified about
 185  *    address suggestions for a specific peer. Addresses asks the solver with
 186  *    s_get. If no address is available, it will not send a response, otherwise
 187  *    it will respond with the choosen address.
 188  *
 189  *       1.7.5 Address suggestions
 190  *
 191  *    Addresses will notify the client automatically on any bandwidth_changed_cb
 192  *    by the solver if a address suggestion request is pending. If no address is
 193  *    available it will not respond at all If the client is not interested
 194  *    anymore, it has to cancel the address suggestion request.
 195  *
 196  *       1.7.6 Suggestions blocks and reset
 197  *
 198  *    After suggesting an address it is blocked for ATS_BLOCKING_DELTA sec. to
 199  *    prevent the client from being thrashed. If the client requires immediately
 200  *    it can reset this block using GAS_addresses_handle_backoff_reset.
 201  *
 202  *       1.7.7 Marking address in use
 203  *
 204  *    The client can notify addresses that it successfully uses an address and
 205  *    wants this address to be kept by calling GSA_address_in_use. Adresses will
 206  *    mark the address as used an notify the solver about the use.
 207  *
 208  *       1.7.8 Address lifecycle
 209  *
 210  *      * (add address)
 211  *      * (updated address) || (address in use)
 212  *      * (delete address)
 213  *
 214  *     1.8 Bandwidth assignment
 215  *
 216  *    The addresses are used to perform resource allocation operations. ATS
 217  *    addresses takes care of instantiating the solver configured and notifies
 218  *    the respective solver about address changes and receives changes to the
 219  *    bandwidth assignment from the solver. The current bandwidth assignment is
 220  *    sent to transport. The specific solvers will be described in the specific
 221  *    section.
 222  *
 223  *     1.9 Changing peer preferences
 224  *
 225  *    The bandwidth assigned to a peer can be influenced by setting a preference
 226  *    for a peer. The prefernce will be given to to the solver with s_pref which
 227  *    has to take care of the preference value
 228
 229  */
 230
 231
 232 /**
 233  * Available ressource assignment modes
 234  */
 235 enum ATS_Mode
 236 {
 237   /*
 238    * Simplistic mode:
 239    *
 240    * Assign each peer an equal amount of bandwidth (bw)
 241    *
 242    * bw_per_peer = bw_total / #active addresses
 243    */
 244   MODE_SIMPLISTIC,
 245
 246   /*
 247    * MLP mode:
 248    *
 249    * Solve ressource assignment as an optimization problem
 250    * Uses an mixed integer programming solver
 251    */
 252   MODE_MLP
 253 };
 254
 255
 256 /**
 257  * Pending Address suggestion requests
 258  */
 259 struct GAS_Addresses_Suggestion_Requests
 260 {
 261   /**
 262    * Next in DLL
 263    */
 264   struct GAS_Addresses_Suggestion_Requests *next;
 265
 266   /**
 267    * Previous in DLL
 268    */
 269   struct GAS_Addresses_Suggestion_Requests *prev;
 270
 271   /**
 272    * Peer ID
 273    */
 274   struct GNUNET_PeerIdentity id;
 275 };
 276
 277
 278 /**
 279  * Handle for ATS address component
 280  */
 281 struct GAS_Addresses_Handle
 282 {
 283   /**
 284    *
 285    */
 286   struct GNUNET_STATISTICS_Handle *stat;
 287
 288   /**
 289    * A multihashmap to store all addresses
 290    */
 291   struct GNUNET_CONTAINER_MultiHashMap *addresses;
 292
 293   /**
 294    * Configure WAN quota in
 295    */
 296   unsigned long long wan_quota_in;
 297
 298   /**
 299    * Configure WAN quota out
 300    */
 301   unsigned long long wan_quota_out;
 302
 303   /**
 304    * Is ATS addresses running
 305    */
 306   int running;
 307
 308   /**
 309    * Configured ATS solver
 310    */
 311   int ats_mode;
 312
 313   /**
 314    *  Solver handle
 315    */
 316   void *solver;
 317
 318   /**
 319    * Address suggestion requests DLL head
 320    */
 321   struct GAS_Addresses_Suggestion_Requests *r_head;
 322
 323   /**
 324    * Address suggestion requests DLL tail
 325    */
 326   struct GAS_Addresses_Suggestion_Requests *r_tail;
 327
 328   /* Solver functions */
 329
 330   /**
 331    * Initialize solver
 332    */
 333   GAS_solver_init s_init;
 334
 335   /**
 336    * Add an address to the solver
 337    */
 338   GAS_solver_address_add s_add;
 339
 340   /**
 341    * Update address in solver
 342    */
 343   GAS_solver_address_update s_update;
 344
 345   /**
 346    * Get address from solver
 347    */
 348   GAS_solver_get_preferred_address s_get;
 349
 350   /**
 351    * Get address from solver
 352    */
 353   GAS_solver_stop_get_preferred_address s_get_stop;
 354
 355   /**
 356    * Delete address in solver
 357    */
 358   GAS_solver_address_delete s_del;
 359
 360   /**
 361    * Change preference for quality in solver
 362    */
 363   GAS_solver_address_change_preference s_pref;
 364
 365   /**
 366    * Shutdown solver
 367    */
 368   GAS_solver_done s_done;
 369 };
 370
 371
 372 /**
 373  * Disassemble ATS information and update performance information in address
 374  *
 375  * Updates existing information and adds new information
 376  *
 377  * @param src source ATS information
 378  * @param ats_count number of ATS information
 379  * @param delta ats performance information which were updated
 380  *                              including previous value
 381  * @param delta_count number of ATS information in the delta
 382  * @param dest destination address
 383  * @return GNUNET_YES if address was address updated, GNUNET_NO otherwise
 384  */
 385 static unsigned int
 386 disassemble_ats_information (struct ATS_Address *dest,
 387                                                                                                                  const struct GNUNET_ATS_Information *update,
 388                              uint32_t update_count,
 389                              struct GNUNET_ATS_Information **delta_dest,
 390                              uint32_t *delta_count)
 391 {
 392
 393   int c1;
 394   int c2;
 395   int found;
 396   int change;
 397
 398   struct GNUNET_ATS_Information add_atsi[update_count];
 399   struct GNUNET_ATS_Information delta_atsi[update_count];
 400   struct GNUNET_ATS_Information *tmp_atsi;
 401   uint32_t add_atsi_count;
 402   uint32_t delta_atsi_count;
 403
 404   change = GNUNET_NO;
 405   add_atsi_count = 0;
 406   delta_atsi_count = 0;
 407
 408   if (0 == update_count)
 409         return GNUNET_NO;
 410
 411   if (NULL == dest->atsi)
 412   {
 413                 /* Create performance information */
 414                 dest->atsi = GNUNET_malloc (update_count * sizeof (struct GNUNET_ATS_Information));
 415                 dest->atsi_count = update_count;
 416                 memcpy (dest->atsi, update, update_count * sizeof (struct GNUNET_ATS_Information));
 417                 delta_atsi_count = update_count;
 418                 (*delta_dest) = GNUNET_malloc (update_count * sizeof (struct GNUNET_ATS_Information));
 419                 for (c1 = 0; c1 < update_count; c1 ++)
 420                 {
 421                         (*delta_dest)[c1].type = update[c1].type;
 422                         (*delta_dest)[c1].value = htonl(GNUNET_ATS_VALUE_UNDEFINED);
 423                 }
 424                 (*delta_count) = update_count;
 425                 return GNUNET_YES;
 426   }
 427
 428   for (c1 = 0; c1 < update_count; c1++)
 429   {
 430         /* Update existing performance information */
 431         found = GNUNET_NO;
 432         for (c2 = 0; c2 < dest->atsi_count; c2++)
 433         {
 434                         if (update[c1].type == dest->atsi[c2].type)
 435                         {
 436                                 if (update[c1].value != dest->atsi[c2].value)
 437                                 {
 438                                                 /* Save previous value in delta */
 439                                                 delta_atsi[delta_atsi_count] = dest->atsi[c2];
 440                                                 delta_atsi_count ++;
 441                                                 /* Set new value */
 442                                                 dest->atsi[c2].value = update[c1].value;
 443                                                 change = GNUNET_YES;
 444                                 }
 445                                 found = GNUNET_YES;
 446                                 break;
 447                         }
 448         }
 449                 if (GNUNET_NO == found)
 450                 {
 451                                 add_atsi[add_atsi_count] = update[c1];
 452                                 add_atsi_count ++;
 453                                 delta_atsi[delta_atsi_count].type = update[c1].type;
 454                                 delta_atsi[delta_atsi_count].value = htonl (GNUNET_ATS_VALUE_UNDEFINED);
 455                                 delta_atsi_count ++;
 456                 }
 457   }
 458
 459   if (add_atsi_count > 0)
 460   {
 461                 /* Extend ats performance information */
 462
 463                 tmp_atsi = GNUNET_malloc ((dest->atsi_count + add_atsi_count) *
 464                                                                                                                         (sizeof (struct GNUNET_ATS_Information)));
 465                 memcpy (tmp_atsi, dest->atsi, dest->atsi_count * sizeof (struct GNUNET_ATS_Information));
 466                 memcpy (&tmp_atsi[dest->atsi_count], add_atsi, add_atsi_count * sizeof (struct GNUNET_ATS_Information));
 467                 GNUNET_free (dest->atsi);
 468                 dest->atsi = tmp_atsi;
 469                 dest->atsi_count = dest->atsi_count + add_atsi_count;
 470                         change = GNUNET_YES;
 471   }
 472
 473   if (delta_atsi_count > 0)
 474   {
 475                 /* Copy delta */
 476                 (*delta_dest) = GNUNET_malloc (delta_atsi_count * sizeof (struct GNUNET_ATS_Information));
 477                 memcpy ((*delta_dest), delta_atsi, delta_atsi_count * sizeof (struct GNUNET_ATS_Information));
 478                 (*delta_count) = delta_atsi_count;
 479   }
 480
 481   return change;
 482 }
 483
 484 /**
 485  * Free the given address
 486  *
 487  * @param addr address to destroy
 488  */
 489 static void
 490 free_address (struct ATS_Address *addr)
 491 {
 492   GNUNET_free (addr->plugin);
 493   GNUNET_free_non_null (addr->atsi);
 494   GNUNET_free (addr);
 495 }
 496
 497 /**
 498  * Create a ATS_address with the given information
 499  *
 500  * @param peer peer
 501  * @param plugin_name plugin
 502  * @param plugin_addr address
 503  * @param plugin_addr_len address length
 504  * @param session_id session
 505  * @return the ATS_Address
 506  */
 507 static struct ATS_Address *
 508 create_address (const struct GNUNET_PeerIdentity *peer,
 509                 const char *plugin_name,
 510                 const void *plugin_addr, size_t plugin_addr_len,
 511                 uint32_t session_id)
 512 {
 513   struct ATS_Address *aa = NULL;
 514
 515   aa = GNUNET_malloc (sizeof (struct ATS_Address) + plugin_addr_len);
 516   aa->peer = *peer;
 517   aa->addr_len = plugin_addr_len;
 518   aa->addr = &aa[1];
 519   memcpy (&aa[1], plugin_addr, plugin_addr_len);
 520   aa->plugin = GNUNET_strdup (plugin_name);
 521   aa->session_id = session_id;
 522   aa->active = GNUNET_NO;
 523   aa->used = GNUNET_NO;
 524   aa->solver_information = NULL;
 525   aa->atsi = NULL;
 526   aa->atsi_count = 0;
 527   aa->assigned_bw_in = GNUNET_BANDWIDTH_value_init(0);
 528   aa->assigned_bw_out = GNUNET_BANDWIDTH_value_init(0);
 529   return aa;
 530 }
 531
 532
 533 /**
 534  * Destroy the given address.
 535  *
 536  * @param handle the address handle
 537  * @param addr address to destroy
 538  * @return GNUNET_YES if bandwidth allocations should be recalcualted
 539  */
 540 static int
 541 destroy_address (struct GAS_Addresses_Handle *handle, struct ATS_Address *addr)
 542 {
 543   int ret;
 544
 545   ret = GNUNET_NO;
 546   GNUNET_assert (GNUNET_YES ==
 547                  GNUNET_CONTAINER_multihashmap_remove (handle->addresses,
 548                                                        &addr->peer.hashPubKey,
 549                                                        addr));
 550   free_address (addr);
 551   return ret;
 552 }
 553
 554
 555 struct CompareAddressContext
 556 {
 557   const struct ATS_Address *search;
 558
 559   /* exact_address != NULL if address and session is equal */
 560   struct ATS_Address *exact_address;
 561   /* exact_address != NULL if address and session is 0 */
 562   struct ATS_Address *base_address;
 563 };
 564
 565 /**
 566  * Comapre addresses
 567  *
 568  * @param cls a CompareAddressContext containin the source address
 569  * @param key peer id
 570  * @param value the address to compare with
 571  * @return GNUNET_YES to continue, GNUNET_NO if address is founce
 572  */
 573
 574 static int
 575 compare_address_it (void *cls, const struct GNUNET_HashCode * key, void *value)
 576 {
 577   struct CompareAddressContext *cac = cls;
 578   struct ATS_Address *aa = value;
 579
 580   /* Find an matching exact address:
 581    *
 582    * Compare by:
 583    * aa->addr_len == cac->search->addr_len
 584    * aa->plugin == cac->search->plugin
 585    * aa->addr == cac->search->addr
 586    * aa->session == cac->search->session
 587    *
 588    * return as exact address
 589    */
 590   if ((aa->addr_len == cac->search->addr_len) && (0 == strcmp (aa->plugin, cac->search->plugin)))
 591   {
 592       if ((0 == memcmp (aa->addr, cac->search->addr, aa->addr_len)) && (aa->session_id == cac->search->session_id))
 593         cac->exact_address = aa;
 594   }
 595
 596   /* Find an matching base address:
 597    *
 598    * Properties:
 599    *
 600    * aa->session_id == 0
 601    *
 602    * Compare by:
 603    * aa->addr_len == cac->search->addr_len
 604    * aa->plugin == cac->search->plugin
 605    * aa->addr == cac->search->addr
 606    *
 607    * return as base address
 608    */
 609   if ((aa->addr_len == cac->search->addr_len) && (0 == strcmp (aa->plugin, cac->search->plugin)))
 610   {
 611       if ((0 == memcmp (aa->addr, cac->search->addr, aa->addr_len)) && (aa->session_id == 0))
 612         cac->base_address = aa;
 613   }
 614
 615   /* Find an matching exact address based on session:
 616    *
 617    * Properties:
 618    *
 619    * cac->search->addr_len == 0
 620    *
 621    * Compare by:
 622    * aa->plugin == cac->search->plugin
 623    * aa->session_id == cac->search->session_id
 624    *
 625    * return as exact address
 626    */
 627   if (0 == cac->search->addr_len)
 628   {
 629       if ((0 == strcmp (aa->plugin, cac->search->plugin)) && (aa->session_id == cac->search->session_id))
 630         cac->exact_address = aa;
 631   }
 632
 633   if (cac->exact_address == NULL)
 634     return GNUNET_YES; /* Continue iteration to find exact address */
 635   else
 636     return GNUNET_NO; /* Stop iteration since we have an exact address */
 637 }
 638
 639
 640 /**
 641  * Find an existing equivalent address record.
 642  * Compares by peer identity and network address OR by session ID
 643  * (one of the two must match).
 644  *
 645  * @param handle the address handle
 646  * @param peer peer to lookup addresses for
 647  * @param addr existing address record
 648  * @return existing address record, NULL for none
 649  */
 650 struct ATS_Address *
 651 find_equivalent_address (struct GAS_Addresses_Handle *handle,
 652                          const struct GNUNET_PeerIdentity *peer,
 653                          const struct ATS_Address *addr)
 654 {
 655   struct CompareAddressContext cac;
 656
 657   cac.exact_address = NULL;
 658   cac.base_address = NULL;
 659   cac.search = addr;
 660   GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses, &peer->hashPubKey,
 661                                               &compare_address_it, &cac);
 662
 663   if (cac.exact_address == NULL)
 664     return cac.base_address;
 665   return cac.exact_address;
 666 }
 667
 668
 669 /**
 670  * Lookup an ATS address by the address properties and session or return an
 671  * equivalent address with a session == 0
 672  *
 673  * @param handle the address handle to use
 674  * @param peer peer
 675  * @param plugin_name transport plugin name
 676  * @param plugin_addr plugin address
 677  * @param plugin_addr_len length of the plugin address
 678  * @param session_id session id, can be 0
 679  * @param atsi performance information for this address
 680  * @param atsi_count number of performance information contained
 681  * @return an ATS_address or NULL
 682  */
 683
 684 static struct ATS_Address *
 685 lookup_address (struct GAS_Addresses_Handle *handle,
 686                 const struct GNUNET_PeerIdentity *peer,
 687                 const char *plugin_name,
 688                 const void *plugin_addr,
 689                 size_t plugin_addr_len,
 690                 uint32_t session_id,
 691                 const struct GNUNET_ATS_Information *atsi,
 692                 uint32_t atsi_count)
 693 {
 694   struct ATS_Address *aa;
 695   struct ATS_Address *ea;
 696
 697   aa = create_address (peer,
 698                        plugin_name,
 699                        plugin_addr, plugin_addr_len,
 700                        session_id);
 701
 702   /* Get existing address or address with session == 0 */
 703   ea = find_equivalent_address (handle, peer, aa);
 704   free_address (aa);
 705   if (ea == NULL)
 706   {
 707     return NULL;
 708   }
 709   else if (ea->session_id != session_id)
 710   {
 711     return NULL;
 712   }
 713   return ea;
 714 }
 715
 716 /**
 717  * Extract an ATS performance info from an address
 718  *
 719  * @param address the address
 720  * @param type the type to extract in HBO
 721  * @return the value in HBO or GNUNET_ATS_VALUE_UNDEFINED in HBO if value does not exist
 722  */
 723 static int
 724 get_performance_info (struct ATS_Address *address, uint32_t type)
 725 {
 726         int c1;
 727         GNUNET_assert (NULL != address);
 728
 729         if ((NULL == address->atsi) || (0 == address->atsi_count))
 730                         return GNUNET_ATS_VALUE_UNDEFINED;
 731
 732         for (c1 = 0; c1 < address->atsi_count; c1++)
 733         {
 734                         if (ntohl(address->atsi[c1].type) == type)
 735                                 return ntohl(address->atsi[c1].value);
 736         }
 737         return GNUNET_ATS_VALUE_UNDEFINED;
 738 }
 739
 740
 741 /**
 742  * Add a new address for a peer.
 743  *
 744  * @param handle the address handle to use
 745  * @param peer peer
 746  * @param plugin_name transport plugin name
 747  * @param plugin_addr plugin address
 748  * @param plugin_addr_len length of the plugin address
 749  * @param session_id session id, can be 0
 750  * @param atsi performance information for this address
 751  * @param atsi_count number of performance information contained
 752  */
 753 void
 754 GAS_addresses_add (struct GAS_Addresses_Handle *handle,
 755                    const struct GNUNET_PeerIdentity *peer,
 756                    const char *plugin_name, const void *plugin_addr,
 757                    size_t plugin_addr_len, uint32_t session_id,
 758                    const struct GNUNET_ATS_Information *atsi,
 759                    uint32_t atsi_count)
 760 {
 761   struct ATS_Address *aa;
 762   struct ATS_Address *ea;
 763   struct GNUNET_ATS_Information *atsi_delta;
 764   uint32_t atsi_delta_count;
 765   uint32_t addr_net;
 766
 767
 768   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 769               "Received `%s' for peer `%s'\n",
 770               "ADDRESS ADD",
 771               GNUNET_i2s (peer));
 772
 773   if (GNUNET_NO == handle->running)
 774     return;
 775
 776   GNUNET_assert (NULL != handle->addresses);
 777
 778   aa = create_address (peer, plugin_name, plugin_addr, plugin_addr_len,
 779                        session_id);
 780   atsi_delta = NULL;
 781   disassemble_ats_information (aa, atsi, atsi_count, &atsi_delta, &atsi_delta_count);
 782   GNUNET_free_non_null (atsi_delta);
 783   addr_net = get_performance_info (aa, GNUNET_ATS_NETWORK_TYPE);
 784   if (GNUNET_ATS_VALUE_UNDEFINED == addr_net)
 785                 addr_net = GNUNET_ATS_NET_UNSPECIFIED;
 786
 787   /* Get existing address or address with session == 0 */
 788   ea = find_equivalent_address (handle, peer, aa);
 789   if (ea == NULL)
 790   {
 791     /* We have a new address */
 792     GNUNET_assert (GNUNET_OK ==
 793                    GNUNET_CONTAINER_multihashmap_put (handle->addresses,
 794                                                       &peer->hashPubKey, aa,
 795                                                       GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
 796     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Added new address for peer `%s' session id %u, %p\n",
 797                 GNUNET_i2s (peer), session_id, aa);
 798     /* Tell solver about new address */
 799     handle->s_add (handle->solver, handle->addresses, aa, addr_net);
 800     /* Notify performance clients about new address */
 801     GAS_performance_notify_all_clients (&aa->peer,
 802         aa->plugin,
 803         aa->addr, aa->addr_len,
 804         aa->session_id,
 805         aa->atsi, aa->atsi_count,
 806         aa->assigned_bw_out,
 807         aa->assigned_bw_in);
 808     return;
 809   }
 810
 811   GNUNET_free (aa->plugin);
 812   GNUNET_free_non_null (aa->atsi);
 813   GNUNET_free (aa);
 814   aa = NULL;
 815
 816   if (ea->session_id != 0)
 817   {
 818       /* This address with the same session is already existing
 819        * Should not happen */
 820       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 821                 "Added already existing address for peer `%s' `%s' %p with new session %u\n",
 822                 GNUNET_i2s (peer), plugin_name, session_id);
 823       GNUNET_break (0);
 824       return;
 825   }
 826
 827   /* We have an address without an session, update this address */
 828   atsi_delta = NULL;
 829   atsi_delta_count = 0;
 830   if (GNUNET_YES == disassemble_ats_information (ea, atsi, atsi_count, &atsi_delta, &atsi_delta_count))
 831   {
 832                 GAS_performance_notify_all_clients (&ea->peer,
 833                                 ea->plugin,
 834                                 ea->addr, ea->addr_len,
 835                                 ea->session_id,
 836                                 ea->atsi, ea->atsi_count,
 837                                 ea->assigned_bw_out,
 838                                 ea->assigned_bw_in);
 839   }
 840
 841   /* Notify solver about update with atsi information and session */
 842   handle->s_update (handle->solver, handle->addresses, ea, session_id, ea->used, atsi_delta, atsi_delta_count);
 843   GNUNET_free_non_null (atsi_delta);
 844
 845   /* Do the update */
 846   ea->session_id = session_id;
 847   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 848            "Updated existing address for peer `%s' %p with new session %u\n",
 849            GNUNET_i2s (peer), ea, session_id);
 850 }
 851
 852
 853 /**
 854  * Update an address with a session or performance information for a peer.
 855  *
 856  * If an address was added without a session it will be updated with the
 857  * session
 858  *
 859  * @param handle the address handle to use
 860  * @param peer peer
 861  * @param plugin_name transport plugin name
 862  * @param plugin_addr plugin address
 863  * @param plugin_addr_len length of the plugin address
 864  * @param session_id session id, can be 0
 865  * @param atsi performance information for this address
 866  * @param atsi_count number of performance information contained
 867  */
 868 void
 869 GAS_addresses_update (struct GAS_Addresses_Handle *handle,
 870                       const struct GNUNET_PeerIdentity *peer,
 871                       const char *plugin_name, const void *plugin_addr,
 872                       size_t plugin_addr_len, uint32_t session_id,
 873                       const struct GNUNET_ATS_Information *atsi,
 874                       uint32_t atsi_count)
 875 {
 876   struct ATS_Address *aa;
 877   struct GNUNET_ATS_Information *atsi_delta;
 878   uint32_t atsi_delta_count;
 879
 880   if (GNUNET_NO == handle->running)
 881     return;
 882
 883   GNUNET_assert (NULL != handle->addresses);
 884
 885   /* Get existing address */
 886   aa = lookup_address (handle, peer, plugin_name, plugin_addr, plugin_addr_len,
 887                        session_id, atsi, atsi_count);
 888   if (aa == NULL)
 889   {
 890     /* GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Tried to update unknown address for peer `%s' `%s' session id %u\n", */
 891     /*             GNUNET_i2s (peer), plugin_name, session_id); */
 892     /* GNUNET_break (0); */
 893     return;
 894   }
 895
 896   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 897                 "Received `%s' for peer `%s' address \n",
 898                 "ADDRESS UPDATE",
 899                 GNUNET_i2s (peer), aa);
 900
 901   /* Update address */
 902   atsi_delta = NULL;
 903   atsi_delta_count = 0;
 904   if (GNUNET_YES == disassemble_ats_information (aa, atsi, atsi_count, &atsi_delta, &atsi_delta_count))
 905   {
 906                 /* Notify performance clients about updated address */
 907                 GAS_performance_notify_all_clients (&aa->peer,
 908                                 aa->plugin,
 909                                 aa->addr, aa->addr_len,
 910                                 aa->session_id,
 911                                 aa->atsi, aa->atsi_count,
 912                                 aa->assigned_bw_out,
 913                                 aa->assigned_bw_in);
 914   }
 915   /* Tell solver about update */
 916   handle->s_update (handle->solver, handle->addresses, aa, session_id, aa->used, atsi_delta, atsi_delta_count);
 917   GNUNET_free_non_null (atsi_delta);
 918 }
 919
 920
 921 struct DestroyContext
 922 {
 923   struct ATS_Address *aa;
 924
 925   struct GAS_Addresses_Handle *handle;
 926
 927   /**
 928    * GNUNET_NO  : full address
 929    * GNUNET_YES : just session
 930    */
 931   int result;
 932 };
 933
 934
 935 /**
 936  * Delete an address
 937  *
 938  * If session != 0, just the session is deleted, the address itself still exists
 939  * If session == 0, remove full address
 940  * If session == 0 and addrlen == 0, destroy inbound address
 941  *
 942  * @param cls unused
 943  * @param key unused
 944  * @param value the 'struct ATS_Address'
 945  * @return GNUNET_OK (continue to iterate)
 946  */
 947 static int
 948 destroy_by_session_id (void *cls, const struct GNUNET_HashCode * key, void *value)
 949 {
 950   struct DestroyContext *dc = cls;
 951   struct GAS_Addresses_Handle *handle = dc->handle;
 952   const struct ATS_Address *des = dc->aa;
 953   struct ATS_Address *aa = value;
 954
 955   GNUNET_assert (0 == memcmp (&aa->peer, &des->peer,
 956                               sizeof (struct GNUNET_PeerIdentity)));
 957
 958
 959   if (des->session_id == 0)
 960   {
 961     /* Session == 0, remove full address  */
 962     if ((0 == strcmp (des->plugin, aa->plugin)) &&
 963         (aa->addr_len == des->addr_len) &&
 964         (0 == memcmp (des->addr, aa->addr, aa->addr_len)))
 965     {
 966
 967       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 968                   "Deleting full address for peer `%s' session %u %p\n",
 969                   GNUNET_i2s (&aa->peer), aa->session_id, aa);
 970
 971       /* Notify solver about deletion */
 972       handle->s_del (handle->solver, handle->addresses, aa, GNUNET_NO);
 973       destroy_address (handle, aa);
 974       dc->result = GNUNET_NO;
 975       return GNUNET_OK; /* Continue iteration */
 976     }
 977   }
 978   else
 979   {
 980     /* Session != 0, just remove session */
 981     if (aa->session_id != des->session_id)
 982       return GNUNET_OK; /* irrelevant */
 983
 984     if ((aa->session_id != 0) &&
 985         (0 != strcmp (des->plugin, aa->plugin)))
 986     {
 987         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 988                     "Different plugins during removal: `%s' vs `%s' \n",
 989                     des->plugin, aa->plugin);
 990         GNUNET_break (0);
 991         return GNUNET_OK;
 992     }
 993
 994     if (aa->addr_len == 0)
 995     {
 996         /* Inbound connection died, delete full address */
 997         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 998                     "Deleting inbound address for peer `%s': `%s' session %u\n",
 999                     GNUNET_i2s (&aa->peer), aa->plugin, aa->session_id);
1000
1001         /* Notify solver about deletion */
1002         handle->s_del (handle->solver, handle->addresses, aa, GNUNET_NO);
1003         destroy_address (handle, aa);
1004         dc->result = GNUNET_NO;
1005         return GNUNET_OK; /* Continue iteration */
1006     }
1007     else
1008     {
1009         /* Session died */
1010         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1011                     "Deleting session for peer `%s': `%s' %u\n",
1012                     GNUNET_i2s (&aa->peer), aa->plugin, aa->session_id);
1013         /* Notify solver to delete session */
1014         handle->s_del (handle->solver, handle->addresses, aa, GNUNET_YES);
1015         aa->session_id = 0;
1016         return GNUNET_OK;
1017     }
1018   }
1019   return GNUNET_OK;
1020 }
1021
1022
1023 /**
1024  * Remove an address or just a session for a peer.
1025  *
1026  * @param handle the address handle to use
1027  * @param peer peer
1028  * @param plugin_name transport plugin name
1029  * @param plugin_addr plugin address
1030  * @param plugin_addr_len length of the plugin address
1031  * @param session_id session id, can be 0
1032  */
1033 void
1034 GAS_addresses_destroy (struct GAS_Addresses_Handle *handle,
1035                        const struct GNUNET_PeerIdentity *peer,
1036                        const char *plugin_name,
1037                        const void *plugin_addr,
1038                        size_t plugin_addr_len,
1039                        uint32_t session_id)
1040 {
1041   struct ATS_Address *ea;
1042   struct DestroyContext dc;
1043
1044   if (GNUNET_NO == handle->running)
1045     return;
1046
1047   /* Get existing address */
1048   ea = lookup_address (handle, peer, plugin_name, plugin_addr, plugin_addr_len,
1049                        session_id, NULL, 0);
1050
1051   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1052               "Received `%s' for peer `%s' address %p session %u\n",
1053               "ADDRESS DESTROY",
1054               GNUNET_i2s (peer), ea, session_id);
1055
1056   if (ea == NULL)
1057   {
1058     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Tried to destroy unknown address for peer `%s' `%s' session id %u\n",
1059                 GNUNET_i2s (peer), plugin_name, session_id);
1060     return;
1061   }
1062
1063   GNUNET_break (0 < strlen (plugin_name));
1064   dc.handle = handle;
1065   dc.aa = create_address (peer, plugin_name, plugin_addr, plugin_addr_len, session_id);
1066
1067   GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses, &peer->hashPubKey,
1068                                               &destroy_by_session_id, &dc);
1069   free_address (dc.aa);
1070 }
1071
1072
1073 /**
1074  * Notification about active use of an address.
1075  * in_use == GNUNET_YES:
1076  *      This address is used to maintain an active connection with a peer.
1077  * in_use == GNUNET_NO:
1078  *      This address is no longer used to maintain an active connection with a peer.
1079  *
1080  * Note: can only be called with in_use == GNUNET_NO if called with GNUNET_YES
1081  * before
1082  *
1083  * @param handle the address handle to use
1084  * @param peer peer
1085  * @param plugin_name transport plugin name
1086  * @param plugin_addr plugin address
1087  * @param plugin_addr_len length of the plugin address
1088  * @param session_id session id, can be 0
1089  * @param in_use GNUNET_YES if GNUNET_NO
1090  * @return GNUNET_SYSERR on failure (address unknown ...)
1091  */
1092 int
1093 GAS_addresses_in_use (struct GAS_Addresses_Handle *handle,
1094                       const struct GNUNET_PeerIdentity *peer,
1095                       const char *plugin_name,
1096                       const void *plugin_addr,
1097                       size_t plugin_addr_len,
1098                       uint32_t session_id,
1099                       int in_use)
1100 {
1101   struct ATS_Address *ea;
1102
1103   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1104                 "Received `%s' for peer `%s'\n",
1105                 "ADDRESS IN USE",
1106                 GNUNET_i2s (peer));
1107
1108   if (GNUNET_NO == handle->running)
1109     return GNUNET_SYSERR;
1110
1111   ea = lookup_address (handle, peer, plugin_name,
1112                         plugin_addr, plugin_addr_len,
1113                         session_id, NULL, 0);
1114   if (NULL == ea)
1115   {
1116     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1117                 "Trying to set unknown address `%s', %s %u %s \n",
1118                 GNUNET_i2s (peer),
1119                 plugin_name, session_id,
1120                 (GNUNET_NO == in_use) ? "NO" : "YES");
1121     GNUNET_break (0);
1122     return GNUNET_SYSERR;
1123   }
1124   if (ea->used == in_use)
1125   {
1126     GNUNET_break (0);
1127     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1128                 "Address in use called multiple times for peer `%s': %s -> %s \n",
1129                 GNUNET_i2s (peer),
1130                 (GNUNET_NO == ea->used) ? "NO" : "YES",
1131                 (GNUNET_NO == in_use) ? "NO" : "YES");
1132     return GNUNET_SYSERR;
1133   }
1134
1135   /* Tell solver about update */
1136   handle->s_update (handle->solver, handle->addresses, ea, session_id, in_use, NULL, 0);
1137   ea->used = in_use;
1138
1139   return GNUNET_OK;
1140 }
1141
1142
1143 /**
1144  * Cancel address suggestions for a peer
1145  *
1146  * @param handle the address handle
1147  * @param peer the peer id
1148  */
1149 void
1150 GAS_addresses_request_address_cancel (struct GAS_Addresses_Handle *handle,
1151                                       const struct GNUNET_PeerIdentity *peer)
1152 {
1153   struct GAS_Addresses_Suggestion_Requests *cur = handle->r_head;
1154
1155   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1156               "Received request: `%s' for peer %s\n", "request_address_cancel", GNUNET_i2s (peer));
1157
1158   while (NULL != cur)
1159   {
1160       if (0 == memcmp (peer, &cur->id, sizeof (cur->id)))
1161         break; /* found */
1162       cur = cur->next;
1163   }
1164
1165   if (NULL == cur)
1166   {
1167       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1168                   "No address requests pending for peer `%s', cannot remove!\n", GNUNET_i2s (peer));
1169       return;
1170   }
1171   handle->s_get_stop (handle->solver, handle->addresses, peer);
1172   GAS_addresses_handle_backoff_reset (handle, peer);
1173   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1174               "Removed request pending for peer `%s\n", GNUNET_i2s (peer));
1175   GNUNET_CONTAINER_DLL_remove (handle->r_head, handle->r_tail, cur);
1176   GNUNET_free (cur);
1177 }
1178
1179
1180 /**
1181  * Request address suggestions for a peer
1182  *
1183  * @param handle the address handle
1184  * @param peer the peer id
1185  */
1186 void
1187 GAS_addresses_request_address (struct GAS_Addresses_Handle *handle,
1188                                const struct GNUNET_PeerIdentity *peer)
1189 {
1190   struct GAS_Addresses_Suggestion_Requests *cur = handle->r_head;
1191   struct ATS_Address *aa;
1192
1193   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1194               "Received `%s' for peer `%s'\n",
1195               "REQUEST ADDRESS",
1196               GNUNET_i2s (peer));
1197
1198   if (GNUNET_NO == handle->running)
1199     return;
1200   while (NULL != cur)
1201   {
1202       if (0 == memcmp (peer, &cur->id, sizeof (cur->id)))
1203         break; /* already suggesting */
1204       cur = cur->next;
1205   }
1206   if (NULL == cur)
1207   {
1208       cur = GNUNET_malloc (sizeof (struct GAS_Addresses_Suggestion_Requests));
1209       cur->id = (*peer);
1210       GNUNET_CONTAINER_DLL_insert (handle->r_head, handle->r_tail, cur);
1211   }
1212
1213   /* Get prefered address from solver */
1214   aa = (struct ATS_Address *) handle->s_get (handle->solver, handle->addresses, peer);
1215   if (NULL == aa)
1216   {
1217     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1218                 "Cannot suggest address for peer `%s'\n", GNUNET_i2s (peer));
1219     return;
1220   }
1221
1222   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1223               "Suggesting address %p for peer `%s'\n", aa, GNUNET_i2s (peer));
1224
1225   GAS_scheduling_transmit_address_suggestion (peer,
1226                                               aa->plugin,
1227                                               aa->addr, aa->addr_len,
1228                                               aa->session_id,
1229                                               aa->atsi, aa->atsi_count,
1230                                               aa->assigned_bw_out,
1231                                               aa->assigned_bw_in);
1232
1233   aa->block_interval = GNUNET_TIME_relative_add (aa->block_interval, ATS_BLOCKING_DELTA);
1234   aa->blocked_until = GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get(), aa->block_interval);
1235
1236   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1237        "Address %p ready for suggestion, block interval now %llu \n",
1238        aa, aa->block_interval);
1239 }
1240
1241
1242 /**
1243  * Iterator to reset address blocking
1244  *
1245  * @param cls not used
1246  * @param key the peer
1247  * @param value the address to reset
1248  * @return GNUNET_OK to continue
1249  */
1250 static int
1251 reset_address_it (void *cls, const struct GNUNET_HashCode *key, void *value)
1252 {
1253   struct ATS_Address *aa = value;
1254
1255   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1256               "Resetting interval for peer `%s' address %p from %llu to 0\n",
1257               GNUNET_i2s (&aa->peer), aa, aa->block_interval);
1258
1259   aa->blocked_until = GNUNET_TIME_UNIT_ZERO_ABS;
1260   aa->block_interval = GNUNET_TIME_UNIT_ZERO;
1261   return GNUNET_OK;
1262 }
1263
1264
1265 /**
1266  * Reset suggestion backoff for a peer
1267  *
1268  * Suggesting addresses is blocked for ATS_BLOCKING_DELTA. Blocking can be
1269  * reset using this function
1270  *
1271  * @param handle the address handle
1272  * @param peer the peer id
1273  */
1274 void
1275 GAS_addresses_handle_backoff_reset (struct GAS_Addresses_Handle *handle,
1276                                     const struct GNUNET_PeerIdentity *peer)
1277 {
1278   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1279               "Received `%s' for peer `%s'\n",
1280               "RESET BACKOFF",
1281               GNUNET_i2s (peer));
1282
1283   GNUNET_break (GNUNET_SYSERR != GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses,
1284                                               &peer->hashPubKey,
1285                                               &reset_address_it,
1286                                               NULL));
1287 }
1288
1289
1290 /**
1291  * Change the preference for a peer
1292  *
1293  * @param handle the address handle
1294  * @param client the client sending this request
1295  * @param peer the peer id
1296  * @param kind the preference kind to change
1297  * @param score the new preference score
1298  */
1299 void
1300 GAS_addresses_change_preference (struct GAS_Addresses_Handle *handle,
1301                                  void *client,
1302                                  const struct GNUNET_PeerIdentity *peer,
1303                                  enum GNUNET_ATS_PreferenceKind kind,
1304                                  float score)
1305 {
1306   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1307               "Received `%s' for peer `%s' for client %p\n",
1308               "CHANGE PREFERENCE",
1309               GNUNET_i2s (peer), client);
1310
1311   if (GNUNET_NO == handle->running)
1312     return;
1313
1314   if (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (handle->addresses,
1315                                                           &peer->hashPubKey))
1316   {
1317       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1318                   "Received `%s' for unknown peer `%s' from client %p\n",
1319                   "CHANGE PREFERENCE",
1320                   GNUNET_i2s (peer), client);
1321       return;
1322   }
1323
1324   /* Tell solver about update */
1325   handle->s_pref (handle->solver, client, peer, kind, score);
1326 }
1327
1328
1329 /**
1330  * Load quotas for networks from configuration
1331  *
1332  * @param cfg configuration handle
1333  * @param out_dest where to write outbound quotas
1334  * @param in_dest where to write inbound quotas
1335  * @param dest_length length of inbound and outbound arrays
1336  * @return number of networks loaded
1337  */
1338 static unsigned int
1339 load_quotas (const struct GNUNET_CONFIGURATION_Handle *cfg, unsigned long long *out_dest, unsigned long long *in_dest, int dest_length)
1340 {
1341   char *network_str[GNUNET_ATS_NetworkTypeCount] = GNUNET_ATS_NetworkTypeString;
1342   char * entry_in = NULL;
1343   char * entry_out = NULL;
1344   char * quota_out_str;
1345   char * quota_in_str;
1346   int c;
1347   int res;
1348
1349   for (c = 0; (c < GNUNET_ATS_NetworkTypeCount) && (c < dest_length); c++)
1350   {
1351     in_dest[c] = 0;
1352     out_dest[c] = 0;
1353     GNUNET_asprintf (&entry_out, "%s_QUOTA_OUT", network_str[c]);
1354     GNUNET_asprintf (&entry_in, "%s_QUOTA_IN", network_str[c]);
1355
1356     /* quota out */
1357     if (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string(cfg, "ats", entry_out, &quota_out_str))
1358     {
1359       res = GNUNET_NO;
1360       if (0 == strcmp(quota_out_str, GNUNET_ATS_MaxBandwidthString))
1361       {
1362         out_dest[c] = GNUNET_ATS_MaxBandwidth;
1363         res = GNUNET_YES;
1364       }
1365       if ((GNUNET_NO == res) && (GNUNET_OK == GNUNET_STRINGS_fancy_size_to_bytes (quota_out_str, &out_dest[c])))
1366         res = GNUNET_YES;
1367       if ((GNUNET_NO == res) && (GNUNET_OK == GNUNET_CONFIGURATION_get_value_number (cfg, "ats", entry_out,  &out_dest[c])))
1368          res = GNUNET_YES;
1369
1370       if (GNUNET_NO == res)
1371       {
1372           GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Could not load quota for network `%s':  `%s', assigning default bandwidth %llu\n"),
1373               network_str[c], quota_out_str, GNUNET_ATS_DefaultBandwidth);
1374           out_dest[c] = GNUNET_ATS_DefaultBandwidth;
1375       }
1376       else
1377       {
1378           GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Outbound quota configure for network `%s' is %llu\n"),
1379               network_str[c], out_dest[c]);
1380       }
1381       GNUNET_free (quota_out_str);
1382     }
1383     else
1384     {
1385       GNUNET_log (GNUNET_ERROR_TYPE_WARNING, _("No outbound quota configured for network `%s', assigning default bandwidth %llu\n"),
1386           network_str[c], GNUNET_ATS_DefaultBandwidth);
1387       out_dest[c] = GNUNET_ATS_DefaultBandwidth;
1388     }
1389
1390     /* quota in */
1391     if (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string(cfg, "ats", entry_in, &quota_in_str))
1392     {
1393       res = GNUNET_NO;
1394       if (0 == strcmp(quota_in_str, GNUNET_ATS_MaxBandwidthString))
1395       {
1396         in_dest[c] = GNUNET_ATS_MaxBandwidth;
1397         res = GNUNET_YES;
1398       }
1399       if ((GNUNET_NO == res) && (GNUNET_OK == GNUNET_STRINGS_fancy_size_to_bytes (quota_in_str, &in_dest[c])))
1400         res = GNUNET_YES;
1401       if ((GNUNET_NO == res) && (GNUNET_OK == GNUNET_CONFIGURATION_get_value_number (cfg, "ats", entry_in,  &in_dest[c])))
1402          res = GNUNET_YES;
1403
1404       if (GNUNET_NO == res)
1405       {
1406           GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Could not load quota for network `%s':  `%s', assigning default bandwidth %llu\n"),
1407               network_str[c], quota_in_str, GNUNET_ATS_DefaultBandwidth);
1408           in_dest[c] = GNUNET_ATS_DefaultBandwidth;
1409       }
1410       else
1411       {
1412           GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Inbound quota configured for network `%s' is %llu\n"),
1413               network_str[c], in_dest[c]);
1414       }
1415       GNUNET_free (quota_in_str);
1416     }
1417     else
1418     {
1419       GNUNET_log (GNUNET_ERROR_TYPE_WARNING, _("No outbound quota configure for network `%s', assigning default bandwidth %llu\n"),
1420           network_str[c], GNUNET_ATS_DefaultBandwidth);
1421       out_dest[c] = GNUNET_ATS_DefaultBandwidth;
1422     }
1423     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Loaded quota for network `%s' (in/out): %llu %llu\n", network_str[c], in_dest[c], out_dest[c]);
1424     GNUNET_free (entry_out);
1425     GNUNET_free (entry_in);
1426   }
1427   return GNUNET_ATS_NetworkTypeCount;
1428 }
1429
1430
1431 /**
1432  * Callback for solver to notify about assignment changes
1433  *
1434  * @param cls the GAS_Addresses_Handle
1435  * @param address the address with changes
1436  */
1437 static void
1438 bandwidth_changed_cb (void *cls, struct ATS_Address *address)
1439 {
1440   struct GAS_Addresses_Handle *handle = cls;
1441   struct GAS_Addresses_Suggestion_Requests *cur;
1442
1443   GNUNET_assert (handle != NULL);
1444   GNUNET_assert (address != NULL);
1445
1446   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Bandwidth assignment changed for peer %s \n", GNUNET_i2s(&address->peer));
1447
1448   /* Notify performance clients about changes to address */
1449   GAS_performance_notify_all_clients (&address->peer,
1450       address->plugin,
1451       address->addr, address->addr_len,
1452       address->session_id,
1453       address->atsi, address->atsi_count,
1454       address->assigned_bw_out,
1455       address->assigned_bw_in);
1456   cur = handle->r_head;
1457   while (NULL != cur)
1458   {
1459       if (0 == memcmp (&address->peer, &cur->id, sizeof (cur->id)))
1460         break; /* we have an address request pending*/
1461       cur = cur->next;
1462   }
1463   if (NULL == cur)
1464   {
1465       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1466                   "Nobody is interested in peer `%s' :(\n",GNUNET_i2s (&address->peer));
1467       return;
1468   }
1469
1470   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1471               "Sending bandwidth update for peer `%s'\n",GNUNET_i2s (&address->peer));
1472
1473   /* *Notify scheduling clients about suggestion */
1474   GAS_scheduling_transmit_address_suggestion (&address->peer,
1475                                               address->plugin,
1476                                               address->addr, address->addr_len,
1477                                               address->session_id,
1478                                               address->atsi, address->atsi_count,
1479                                               address->assigned_bw_out,
1480                                               address->assigned_bw_in);
1481 }
1482
1483
1484 /**
1485  * Initialize address subsystem. The addresses subsystem manages the addresses
1486  * known and current performance information. It has a solver component
1487  * responsible for the resource allocation. It tells the solver about changes
1488  * and receives updates when the solver changes the ressource allocation.
1489  *
1490  * @param cfg configuration to use
1491  * @param stats the statistics handle to use
1492  * @return an address handle
1493  */
1494 struct GAS_Addresses_Handle *
1495 GAS_addresses_init (const struct GNUNET_CONFIGURATION_Handle *cfg,
1496                     const struct GNUNET_STATISTICS_Handle *stats)
1497 {
1498   struct GAS_Addresses_Handle *ah;
1499   int quotas[GNUNET_ATS_NetworkTypeCount] = GNUNET_ATS_NetworkType;
1500   unsigned long long  quotas_in[GNUNET_ATS_NetworkTypeCount];
1501   unsigned long long  quotas_out[GNUNET_ATS_NetworkTypeCount];
1502   int quota_count;
1503   char *mode_str;
1504   int c;
1505
1506   ah = GNUNET_malloc (sizeof (struct GAS_Addresses_Handle));
1507   ah->running = GNUNET_NO;
1508
1509   ah->stat = (struct GNUNET_STATISTICS_Handle *) stats;
1510   /* Initialize the addresses database */
1511   ah->addresses = GNUNET_CONTAINER_multihashmap_create (128, GNUNET_NO);
1512   GNUNET_assert (NULL != ah->addresses);
1513
1514   /* Figure out configured solution method */
1515   if (GNUNET_SYSERR == GNUNET_CONFIGURATION_get_value_string (cfg, "ats", "MODE", &mode_str))
1516   {
1517       GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "No ressource assignment method configured, using simplistic approch\n");
1518       ah->ats_mode = MODE_SIMPLISTIC;
1519   }
1520   else
1521   {
1522       for (c = 0; c < strlen (mode_str); c++)
1523         mode_str[c] = toupper (mode_str[c]);
1524       if (0 == strcmp (mode_str, "SIMPLISTIC"))
1525       {
1526           ah->ats_mode = MODE_SIMPLISTIC;
1527       }
1528       else if (0 == strcmp (mode_str, "MLP"))
1529       {
1530           ah->ats_mode = MODE_MLP;
1531 #if !HAVE_LIBGLPK
1532           GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Assignment method `%s' configured, but GLPK is not availabe, please install \n", mode_str);
1533           ah->ats_mode = MODE_SIMPLISTIC;
1534 #endif
1535       }
1536       else
1537       {
1538           GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Invalid ressource assignment method `%s' configured, using simplistic approch\n", mode_str);
1539           ah->ats_mode = MODE_SIMPLISTIC;
1540       }
1541       GNUNET_free (mode_str);
1542   }
1543   /* Start configured solution method */
1544   switch (ah->ats_mode)
1545   {
1546     case MODE_MLP:
1547       /* Init the MLP solver with default values */
1548 #if HAVE_LIBGLPK
1549       ah->ats_mode = MODE_MLP;
1550       ah->s_init = &GAS_mlp_init;
1551       ah->s_add = &GAS_mlp_address_add;
1552       ah->s_update = &GAS_mlp_address_update;
1553       ah->s_get = &GAS_mlp_get_preferred_address;
1554       ah->s_get_stop = &GAS_mlp_stop_get_preferred_address;
1555       ah->s_pref = &GAS_mlp_address_change_preference;
1556       ah->s_del =  &GAS_mlp_address_delete;
1557       ah->s_done = &GAS_mlp_done;
1558 #else
1559       GNUNET_free (ah);
1560       return NULL;
1561 #endif
1562       break;
1563     case MODE_SIMPLISTIC:
1564       /* Init the simplistic solver with default values */
1565       ah->ats_mode = MODE_SIMPLISTIC;
1566       ah->s_init = &GAS_simplistic_init;
1567       ah->s_add = &GAS_simplistic_address_add;
1568       ah->s_update = &GAS_simplistic_address_update;
1569       ah->s_get = &GAS_simplistic_get_preferred_address;
1570       ah->s_get_stop = &GAS_simplistic_stop_get_preferred_address;
1571       ah->s_pref = &GAS_simplistic_address_change_preference;
1572       ah->s_del  = &GAS_simplistic_address_delete;
1573       ah->s_done = &GAS_simplistic_done;
1574       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ATS started in %s mode\n", "SIMPLISTIC");
1575       break;
1576     default:
1577       return NULL;
1578       break;
1579   }
1580
1581   GNUNET_assert (NULL != ah->s_init);
1582   GNUNET_assert (NULL != ah->s_add);
1583   GNUNET_assert (NULL != ah->s_update);
1584   GNUNET_assert (NULL != ah->s_get);
1585   GNUNET_assert (NULL != ah->s_get_stop);
1586   GNUNET_assert (NULL != ah->s_pref);
1587   GNUNET_assert (NULL != ah->s_del);
1588   GNUNET_assert (NULL != ah->s_done);
1589
1590   quota_count = load_quotas(cfg, quotas_in, quotas_out, GNUNET_ATS_NetworkTypeCount);
1591
1592   ah->solver = ah->s_init (cfg, stats, quotas, quotas_in, quotas_out, quota_count, &bandwidth_changed_cb, ah);
1593   if (NULL == ah->solver)
1594   {
1595     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to initialize solver!\n");
1596     GNUNET_free (ah);
1597     return NULL;
1598   }
1599
1600   /* up and running */
1601   ah->running = GNUNET_YES;
1602   return ah;
1603 }
1604
1605
1606 /**
1607  * Free memory of address.
1608  *
1609  * @param cls NULL
1610  * @param key peer identity (unused)
1611  * @param value the 'struct ATS_Address' to free
1612  * @return GNUNET_OK (continue to iterate)
1613  */
1614 static int
1615 free_address_it (void *cls, const struct GNUNET_HashCode * key, void *value)
1616 {
1617   struct GAS_Addresses_Handle *handle = cls;
1618   struct ATS_Address *aa = value;
1619   handle->s_del (handle->solver, handle->addresses, aa, GNUNET_NO);
1620   destroy_address (handle, aa);
1621   return GNUNET_OK;
1622 }
1623
1624
1625 /**
1626  * Remove all addresses
1627  *
1628  * @param handle the address handle to use
1629  */
1630 void
1631 GAS_addresses_destroy_all (struct GAS_Addresses_Handle *handle)
1632 {
1633   if (GNUNET_NO == handle->running)
1634     return;
1635
1636   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1637               "Received `%s'\n",
1638               "DESTROY ALL");
1639
1640   if (handle->addresses != NULL)
1641     GNUNET_CONTAINER_multihashmap_iterate (handle->addresses, &free_address_it, handle);
1642 }
1643
1644
1645 /**
1646  * Shutdown address subsystem.
1647  *
1648  * @param handle the address handle to shutdown
1649  */
1650 void
1651 GAS_addresses_done (struct GAS_Addresses_Handle *handle)
1652 {
1653   struct GAS_Addresses_Suggestion_Requests *cur;
1654
1655   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1656               "Shutting down addresses\n");
1657   GNUNET_assert (NULL != handle);
1658   GAS_addresses_destroy_all (handle);
1659   handle->running = GNUNET_NO;
1660   GNUNET_CONTAINER_multihashmap_destroy (handle->addresses);
1661   handle->addresses = NULL;
1662   while (NULL != (cur = handle->r_head))
1663   {
1664       GNUNET_CONTAINER_DLL_remove (handle->r_head, handle->r_tail, cur);
1665       GNUNET_free (cur);
1666   }
1667   handle->s_done (handle->solver);
1668   GNUNET_free (handle);
1669   /* Stop configured solution method */
1670
1671 }
1672
1673
1674 struct PeerIteratorContext
1675 {
1676   GNUNET_ATS_Peer_Iterator it;
1677   void *it_cls;
1678   struct GNUNET_CONTAINER_MultiHashMap *peers_returned;
1679 };
1680
1681
1682 /**
1683  * Iterator to iterate over all peers
1684  *
1685  * @param cls a PeerIteratorContext
1686  * @param key the peer id
1687  * @param value the ATS_address
1688  * @return GNUNET_OK to continue
1689  */
1690 static int
1691 peer_it (void *cls,
1692          const struct GNUNET_HashCode * key,
1693          void *value)
1694 {
1695   struct PeerIteratorContext *ip_ctx = cls;
1696   struct GNUNET_PeerIdentity tmp;
1697
1698   if (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains(ip_ctx->peers_returned, key))
1699   {
1700       GNUNET_CONTAINER_multihashmap_put(ip_ctx->peers_returned, key, NULL, GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1701       tmp.hashPubKey = (*key);
1702       ip_ctx->it (ip_ctx->it_cls, &tmp);
1703   }
1704
1705   return GNUNET_OK;
1706 }
1707
1708
1709 /**
1710  * Return information all peers currently known to ATS
1711  *
1712  * @param handle the address handle to use
1713  * @param p_it the iterator to call for every peer
1714  * @param p_it_cls the closure for the iterator
1715  */
1716 void
1717 GAS_addresses_iterate_peers (struct GAS_Addresses_Handle *handle,
1718                                                                                                                  GNUNET_ATS_Peer_Iterator p_it,
1719                                                                                                                  void *p_it_cls)
1720 {
1721   struct PeerIteratorContext ip_ctx;
1722   unsigned int size;
1723
1724   if (NULL == p_it)
1725       return;
1726   GNUNET_assert (NULL != handle->addresses);
1727
1728   size = GNUNET_CONTAINER_multihashmap_size(handle->addresses);
1729   if (0 != size)
1730   {
1731     ip_ctx.it = p_it;
1732     ip_ctx.it_cls = p_it_cls;
1733     ip_ctx.peers_returned = GNUNET_CONTAINER_multihashmap_create (size, GNUNET_NO);
1734     GNUNET_CONTAINER_multihashmap_iterate (handle->addresses, &peer_it, &ip_ctx);
1735     GNUNET_CONTAINER_multihashmap_destroy (ip_ctx.peers_returned);
1736   }
1737   p_it (p_it_cls, NULL);
1738 }
1739
1740
1741 struct PeerInfoIteratorContext
1742 {
1743   GNUNET_ATS_PeerInfo_Iterator it;
1744   void *it_cls;
1745 };
1746
1747
1748 /**
1749  * Iterator to iterate over a peer's addresses
1750  *
1751  * @param cls a PeerInfoIteratorContext
1752  * @param key the peer id
1753  * @param value the ATS_address
1754  * @return GNUNET_OK to continue
1755  */
1756 static int
1757 peerinfo_it (void *cls,
1758              const struct GNUNET_HashCode * key,
1759              void *value)
1760 {
1761   struct PeerInfoIteratorContext *pi_ctx = cls;
1762   struct ATS_Address *addr = (struct ATS_Address *)  value;
1763
1764   if (NULL != pi_ctx->it)
1765   {
1766     pi_ctx->it (pi_ctx->it_cls,
1767                 &addr->peer,
1768                 addr->plugin,
1769                 addr->addr, addr->addr_len,
1770                 addr->active,
1771                 addr->atsi, addr->atsi_count,
1772                 addr->assigned_bw_out,
1773                 addr->assigned_bw_in);
1774   }
1775   return GNUNET_YES;
1776 }
1777
1778
1779 /**
1780  * Return information all peers currently known to ATS
1781  *
1782  * @param handle the address handle to use
1783  * @param peer the respective peer
1784  * @param pi_it the iterator to call for every peer
1785  * @param pi_it_cls the closure for the iterator
1786  */
1787 void
1788 GAS_addresses_get_peer_info (struct GAS_Addresses_Handle *handle,
1789                              const struct GNUNET_PeerIdentity *peer,
1790                              GNUNET_ATS_PeerInfo_Iterator pi_it,
1791                              void *pi_it_cls)
1792 {
1793   struct PeerInfoIteratorContext pi_ctx;
1794   struct GNUNET_BANDWIDTH_Value32NBO zero_bw;
1795   GNUNET_assert (NULL != peer);
1796   GNUNET_assert (NULL != handle->addresses);
1797   if (NULL == pi_it)
1798     return; /* does not make sense without callback */
1799
1800   zero_bw = GNUNET_BANDWIDTH_value_init (0);
1801   pi_ctx.it = pi_it;
1802   pi_ctx.it_cls = pi_it_cls;
1803
1804   GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses, &peer->hashPubKey, &peerinfo_it, &pi_ctx);
1805
1806   if (NULL != pi_it)
1807     pi_it (pi_it_cls, NULL, NULL, NULL, 0, GNUNET_NO, NULL, 0, zero_bw, zero_bw);
1808
1809 }
1810
1811
1812 /* end of gnunet-service-ats_addresses.c */