53c8ca25cd04108d14d9ef848437d0ce287fcd5e
[oweals/gnunet.git] / 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_ats_plugin.h"
30 #include "gnunet-service-ats.h"
31 #include "gnunet-service-ats_addresses.h"
32 #include "gnunet-service-ats_normalization.h"
33 #include "gnunet-service-ats_performance.h"
34 #include "gnunet-service-ats_scheduling.h"
35 #include "gnunet-service-ats_reservations.h"
36
37
38 /**
39  * NOTE: Do not change this documentation. This documentation is based on
40  * gnunet.org:/vcs/fsnsg/ats-paper.git/tech-doku/ats-tech-guide.tex
41  * use build_txt.sh to generate plaintext output
42  *
43  *   1 ATS addresses : ATS address management
44  *
45  *    This ATS addresses ("addresses") component manages the addresses known to
46  *    ATS service and suggests addresses to transport service when it is
47  *    interested in address suggestion for a peer. ATS addresses also
48  *    instantiates the bandwidth assignment mechanism (solver), notifies it
49  *    about changes to addresses and forwards changes to bandwidth assignments
50  *    to transport, depending if transport is interested in this change.
51  *
52  *     1.1 Input data
53  *
54  *       1.1.1 Addresses
55  *
56  *    Addresses are added by specifying peer ID, plugin, address, address length
57  *    and session, if available. ATS information can be specified if available.
58  *
59  *       1.1.2 Networks
60  *
61  *    ATS specifies a fix set of networks an address can belong to. For each
62  *    network an inbound and outbound quota will be specified. The available
63  *    networks and addtional helper varaibles are defined in
64  *    gnunet_ats_service.h. At the moment 5 networks are defined:
65  *      * GNUNET_ATS_NET_UNSPECIFIED
66  *      * GNUNET_ATS_NET_LOOPBACK
67  *      * GNUNET_ATS_NET_LAN
68  *      * GNUNET_ATS_NET_WAN
69  *      * GNUNET_ATS_NET_WLAN
70  *
71  *    The total number of networks defined is stored in
72  *    GNUNET_ATS_NetworkTypeCount GNUNET_ATS_NetworkType can be used array
73  *    initializer for an int array, while GNUNET_ATS_NetworkType is an
74  *    initializer for a char array containing a string description of all
75  *    networks
76  *
77  *       1.1.3 Quotas
78  *
79  *    An inbound and outbound quota for each of the networks mentioned in 1.1.2
80  *    is loaded from ats configuration during initialization. This quota defines
81  *    to total amount of inbound and outbound traffic allowed for a specific
82  *    network. The configuration values used are in section ats:
83  *      * "NETWORK"_QUOTA_IN = <value>
84  *      * "NETWORK"_QUOTA_IN = <value>
85  *
86  *    You can specify quotas by setting the <value> to a:
87  *      * unrestricted: unlimited
88  *      * number of bytes: e.g. 10240
89  *      * fancy value: e.g. 64 Kib
90  *
91  *    unlimited is defined as GNUNET_ATS_MaxBandwidthString and equivalent to
92  *    the value GNUNET_ATS_MaxBandwidth Important predefined values for quotas
93  *    are:
94  *      * GNUNET_ATS_DefaultBandwidth: 65536
95  *      * GNUNET_ATS_MaxBandwidth: UINT32_MAX
96  *      * GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT: 1024
97  *
98  *    Details of loading quotas and default values will be described on
99  *
100  *       1.1.4 Preference values
101  *
102  *     1.2 Data structures used
103  *
104  *    Addresse uses struct ATS_Address for each address. The structs are stored
105  *    in a linked list and provides a pointer void *solver_information for the
106  *    solver to store address specific information. It provides the int values
107  *    active which is set to GNUNET_YES if the address is select for transport
108  *    use and used, representing that transport service is actively using this
109  *    address. Address information are stored in peer, addr, addr_len, plugin.
110  *
111  *     1.3 Initialization
112  *
113  *    During initialization a hashmap to store addresses is created. The quotas
114  *    for all networks defined for ATS are loaded from configuration. For each
115  *    network first the logic will check if the string
116  *    GNUNET_ATS_MaxBandwidthString is configured, if not it will try to convert
117  *    the configured value as a fancy size and if this fails it will try to use
118  *    it as a value_number. If no configuration value is found it will assign
119  *    GNUNET_ATS_DefaultBandwidth. The most important step is to load the
120  *    configured solver using configuration "[ats]:MODE". Current solvers are
121  *    MODE_PROPORTIONAL, MODE_MLP. Interaction is done using a solver API
122  *
123  *     1.4 Solver API
124  *
125  *    Solver functions:
126  *      * s_init: init the solver with required information
127  *      * s_add: add a new address
128  *      * s_update: update ATS values or session for an address
129  *      * s_get: get prefered address for a peer
130  *      * s_del: delete an address
131  *      * s_pref: change preference value for a peer
132  *      * s_done: shutdown solver
133  *
134  *    Callbacks: addresses provides a bandwidth_changed_cb callback to the
135  *    solver which is called when bandwidth assigned to peer has changed
136  *
137  *     1.5 Shutdown
138  *
139  *    During shutdown all addresses are freed and the solver told to shutdown
140  *
141  *     1.6 Addresses and sessions
142  *
143  *    Addresses consist of the address itself and a numerical session. When a
144  *    new address without a session is added it has no session, so it gets
145  *    session 0 assigned. When an address with a session is added and an address
146  *    object with session 0 is found, this object is updated with the session
147  *    otherwise a new address object with this session assigned is created.
148  *
149  *       1.6.1 Terminology
150  *
151  *    Addresses a1,a2 with session s1, s2 are "exact" if:
152  *    (a1 == a2)&&(s1 == s2)
153  *    Addresses a1,a2 with session s1, s2 are "equivalent" if:
154  *    (a1 == a2)&&((s1 == s2)||(s1 == 0)||(s2 == 0)
155  *
156  *     1.7 Address management
157  *
158  *    Transport service notifies ATS about changes to the addresses known to
159  *    him.
160  *
161  *       1.7.1 Adding an address
162  *
163  *    When transport learns a new address it tells ATS and ATS is telling
164  *    addresses about it using GAS_address_add. If not known to addresses it
165  *    creates a new address object and calls solver's s_add. ATS information are
166  *    deserialized and solver is notified about the session and ATS information
167  *    using s_update.
168  *
169  *       1.7.2 Updating an address
170  *
171  *    Addresses does an lookup up for the existing address with the given
172  *    session. If disassembles included ATS information and notifies the solver
173  *    using s_update about the update.
174  *
175  *       1.7.3 Deleting an address
176  *
177  *    Addresses does an lookup for the exact address and session and if removes
178  *    this address. If session != 0 the session is set to 0 and the address is
179  *    kept. If session == 0, the addresses is removed.
180  *
181  *       1.7.4 Requesting an address suggestion
182  *
183  *    The address client issues a request address message to be notified about
184  *    address suggestions for a specific peer. Addresses asks the solver with
185  *    s_get. If no address is available, it will not send a response, otherwise
186  *    it will respond with the choosen address.
187  *
188  *       1.7.5 Address suggestions
189  *
190  *    Addresses will notify the client automatically on any bandwidth_changed_cb
191  *    by the solver if a address suggestion request is pending. If no address is
192  *    available it will not respond at all If the client is not interested
193  *    anymore, it has to cancel the address suggestion request.
194  *
195  *       1.7.6 Suggestions blocks and reset
196  *
197  *    After suggesting an address it is blocked for ATS_BLOCKING_DELTA sec. to
198  *    prevent the client from being thrashed. If the client requires immediately
199  *    it can reset this block using GAS_addresses_handle_backoff_reset.
200  *
201  *       1.7.7 Marking address in use
202  *
203  *    The client can notify addresses that it successfully uses an address and
204  *    wants this address to be kept by calling GSA_address_in_use. Adresses will
205  *    mark the address as used an notify the solver about the use.
206  *
207  *       1.7.8 Address lifecycle
208  *
209  *      * (add address)
210  *      * (updated address) || (address in use)
211  *      * (delete address)
212  *
213  *     1.8 Bandwidth assignment
214  *
215  *    The addresses are used to perform resource allocation operations. ATS
216  *    addresses takes care of instantiating the solver configured and notifies
217  *    the respective solver about address changes and receives changes to the
218  *    bandwidth assignment from the solver. The current bandwidth assignment is
219  *    sent to transport. The specific solvers will be described in the specific
220  *    section.
221  *
222  *     1.9 Changing peer preferences
223  *
224  *    The bandwidth assigned to a peer can be influenced by setting a preference
225  *    for a peer. The prefernce will be given to to the solver with s_pref which
226  *    has to take care of the preference value
227
228  */
229
230
231 /**
232  * Pending Address suggestion requests
233  */
234 struct GAS_Addresses_Suggestion_Requests
235 {
236   /**
237    * Next in DLL
238    */
239   struct GAS_Addresses_Suggestion_Requests *next;
240
241   /**
242    * Previous in DLL
243    */
244   struct GAS_Addresses_Suggestion_Requests *prev;
245
246   /**
247    * Peer ID
248    */
249   struct GNUNET_PeerIdentity id;
250 };
251
252 /**
253  * Pending Address suggestion requests
254  */
255 struct GAS_Addresses_Preference_Clients
256 {
257   /**
258    * Next in DLL
259    */
260   struct GAS_Addresses_Preference_Clients *next;
261
262   /**
263    * Previous in DLL
264    */
265   struct GAS_Addresses_Preference_Clients *prev;
266
267   /**
268    * Peer ID
269    */
270   void *client;
271 };
272
273 /**
274  * Handle for ATS address component
275  */
276 struct GAS_Addresses_Handle
277 {
278   /**
279    *
280    */
281   struct GNUNET_STATISTICS_Handle *stat;
282
283   /**
284    * A multihashmap to store all addresses
285    */
286   struct GNUNET_CONTAINER_MultiPeerMap *addresses;
287
288   /**
289    * Is ATS addresses running
290    */
291   int running;
292
293   /**
294    * Preferences clients
295    */
296   int pref_clients;
297
298   /**
299    * Configured ATS solver
300    */
301   int ats_mode;
302
303   /**
304    * Solver handle
305    */
306   void *solver;
307
308   /**
309    * Address suggestion requests DLL head
310    */
311   struct GAS_Addresses_Suggestion_Requests *pending_requests_head;
312
313   /**
314    * Address suggestion requests DLL tail
315    */
316   struct GAS_Addresses_Suggestion_Requests *pending_requests_tail;
317
318   /**
319    * Preference requests DLL head
320    */
321   struct GAS_Addresses_Preference_Clients *preference_clients_head;
322
323   /**
324    * Preference requests DLL head
325    */
326   struct GAS_Addresses_Preference_Clients *preference_clients_tail;
327
328   /**
329    * Solver functions
330    */
331   struct GNUNET_ATS_PluginEnvironment env;
332
333   /**
334    * Solver plugin name as string
335    */
336   char *plugin;
337 };
338
339
340 /**
341  * Value we pass for zero bandwidth.
342  */
343 const static struct GNUNET_BANDWIDTH_Value32NBO zero_bw;
344
345
346 /**
347  * Disassemble ATS information and update performance information in address
348  *
349  * Updates existing information and adds new information
350  *
351  * @param dest destination address
352  * @param update source ATS information
353  * @param update_count number of ATS information in @a update
354  * @param delta_dest ats performance information which were updated
355  *                              including previous value
356  * @param delta_count number of ATS information in the @a delta_dest
357  * @return #GNUNET_YES if address was address updated, GNUNET_NO otherwise
358  */
359 static unsigned int
360 disassemble_ats_information (struct ATS_Address *dest,
361                              const struct GNUNET_ATS_Information *update,
362                              uint32_t update_count,
363                              struct GNUNET_ATS_Information **delta_dest,
364                              uint32_t *delta_count)
365 {
366   int c1;
367   int c2;
368   int found;
369   int change;
370   struct GNUNET_ATS_Information add_atsi[update_count];
371   struct GNUNET_ATS_Information delta_atsi[update_count];
372   struct GNUNET_ATS_Information *tmp_atsi;
373   uint32_t add_atsi_count;
374   uint32_t delta_atsi_count;
375
376   change = GNUNET_NO;
377   add_atsi_count = 0;
378   delta_atsi_count = 0;
379
380   if (0 == update_count)
381     return GNUNET_NO;
382
383   if (NULL == dest->atsi)
384   {
385     /* Create performance information */
386     dest->atsi =
387         GNUNET_malloc (update_count * sizeof (struct GNUNET_ATS_Information));
388     dest->atsi_count = update_count;
389     memcpy (dest->atsi, update,
390         update_count * sizeof(struct GNUNET_ATS_Information));
391     (*delta_dest) =
392         GNUNET_malloc (update_count * sizeof (struct GNUNET_ATS_Information));
393     for (c1 = 0; c1 < update_count; c1++)
394     {
395       (*delta_dest)[c1].type = update[c1].type;
396       (*delta_dest)[c1].value = htonl (GNUNET_ATS_VALUE_UNDEFINED);
397     }
398     (*delta_count) = update_count;
399     return GNUNET_YES;
400   }
401
402   for (c1 = 0; c1 < update_count; c1++)
403   {
404     /* Update existing performance information */
405     found = GNUNET_NO;
406     for (c2 = 0; c2 < dest->atsi_count; c2++)
407     {
408       if (update[c1].type == dest->atsi[c2].type)
409       {
410         if (update[c1].value != dest->atsi[c2].value)
411         {
412           /* Save previous value in delta */
413           delta_atsi[delta_atsi_count] = dest->atsi[c2];
414           delta_atsi_count++;
415           /* Set new value */
416           dest->atsi[c2].value = update[c1].value;
417           change = GNUNET_YES;
418         }
419         found = GNUNET_YES;
420         break;
421       }
422     }
423     if (GNUNET_NO == found)
424     {
425       add_atsi[add_atsi_count] = update[c1];
426       add_atsi_count++;
427       delta_atsi[delta_atsi_count].type = update[c1].type;
428       delta_atsi[delta_atsi_count].value = htonl (GNUNET_ATS_VALUE_UNDEFINED);
429       delta_atsi_count++;
430     }
431   }
432
433   if (add_atsi_count > 0)
434   {
435     /* Extend ats performance information */
436
437     tmp_atsi = GNUNET_malloc ((dest->atsi_count + add_atsi_count) *
438         (sizeof (struct GNUNET_ATS_Information)));
439     memcpy (tmp_atsi, dest->atsi,
440         dest->atsi_count * sizeof(struct GNUNET_ATS_Information));
441     memcpy (&tmp_atsi[dest->atsi_count], add_atsi,
442         add_atsi_count * sizeof(struct GNUNET_ATS_Information));
443     GNUNET_free(dest->atsi);
444     dest->atsi = tmp_atsi;
445     dest->atsi_count = dest->atsi_count + add_atsi_count;
446     change = GNUNET_YES;
447   }
448
449   if (delta_atsi_count > 0)
450   {
451     /* Copy delta */
452     (*delta_dest) =
453         GNUNET_malloc (delta_atsi_count * sizeof (struct GNUNET_ATS_Information));
454     memcpy ((*delta_dest), delta_atsi,
455         delta_atsi_count * sizeof(struct GNUNET_ATS_Information));
456     (*delta_count) = delta_atsi_count;
457   }
458
459   return change;
460 }
461
462
463 /**
464  * Free the given address
465  *
466  * @param addr address to destroy
467  */
468 static void
469 free_address (struct ATS_Address *addr)
470 {
471   GNUNET_free (addr->plugin);
472   GNUNET_free_non_null (addr->atsi);
473   GNUNET_free (addr);
474 }
475
476
477 /**
478  * Create a ATS_address with the given information
479  *
480  * @param peer peer
481  * @param plugin_name plugin
482  * @param plugin_addr address
483  * @param plugin_addr_len address length
484  * @param local_address_info additional local info for the address
485  * @param session_id session identifier, can never be 0
486  * @return the ATS_Address
487  */
488 static struct ATS_Address *
489 create_address (const struct GNUNET_PeerIdentity *peer,
490                 const char *plugin_name,
491                 const void *plugin_addr,
492                 size_t plugin_addr_len,
493                 uint32_t local_address_info,
494                 uint32_t session_id)
495 {
496   struct ATS_Address *aa;
497   unsigned int c1;
498   unsigned int c2;
499
500   aa = GNUNET_malloc (sizeof (struct ATS_Address) + plugin_addr_len);
501   aa->peer = *peer;
502   aa->addr_len = plugin_addr_len;
503   aa->addr = &aa[1];
504   memcpy (&aa[1], plugin_addr, plugin_addr_len);
505   aa->plugin = GNUNET_strdup (plugin_name);
506   aa->session_id = session_id;
507   aa->local_address_info = local_address_info;
508
509   for (c1 = 0; c1 < GNUNET_ATS_QualityPropertiesCount; c1++)
510   {
511     aa->atsin[c1].avg_queue_index = 0;
512     for (c2 = 0; c2 < GAS_normalization_queue_length; c2++)
513       aa->atsin[c1].atsi_abs[c2] = GNUNET_ATS_VALUE_UNDEFINED;
514   }
515   return aa;
516 }
517
518
519 /**
520  * Closure for #compare_address_it()
521  */
522 struct CompareAddressContext
523 {
524   const struct ATS_Address *search;
525
526   /* exact_address != NULL if address and session is equal */
527   struct ATS_Address *exact_address;
528   /* exact_address != NULL if address and session is 0 */
529   struct ATS_Address *base_address;
530 };
531
532
533 /**
534  * Comapre addresses.
535  *
536  * @param cls a CompareAddressContext containin the source address
537  * @param key peer id
538  * @param value the address to compare with
539  * @return #GNUNET_YES to continue, #GNUNET_NO if address is founce
540  */
541 static int
542 compare_address_it (void *cls,
543                     const struct GNUNET_PeerIdentity *key,
544                     void *value)
545 {
546   struct CompareAddressContext *cac = cls;
547   struct ATS_Address *aa = value;
548
549   /* Find an matching exact address:
550    *
551    * Compare by:
552    * aa->addr_len == cac->search->addr_len
553    * aa->plugin == cac->search->plugin
554    * aa->addr == cac->search->addr
555    * aa->session == cac->search->session
556    *
557    * return as exact address
558    */
559   if ((aa->addr_len == cac->search->addr_len)
560       && (0 == strcmp (aa->plugin, cac->search->plugin)))
561   {
562     if ((0 == memcmp (aa->addr, cac->search->addr, aa->addr_len))
563         && (aa->session_id == cac->search->session_id))
564       cac->exact_address = aa;
565   }
566
567   /* Find an matching base address:
568    *
569    * Properties:
570    *
571    * aa->session_id == 0
572    *
573    * Compare by:
574    * aa->addr_len == cac->search->addr_len
575    * aa->plugin == cac->search->plugin
576    * aa->addr == cac->search->addr
577    *
578    * return as base address
579    */
580   if ((aa->addr_len == cac->search->addr_len)
581       && (0 == strcmp (aa->plugin, cac->search->plugin)))
582   {
583     if ((0 == memcmp (aa->addr, cac->search->addr, aa->addr_len))
584         && (aa->session_id == 0))
585       cac->base_address = aa;
586   }
587
588   /* Find an matching exact address based on session:
589    *
590    * Properties:
591    *
592    * cac->search->addr_len == 0
593    *
594    * Compare by:
595    * aa->plugin == cac->search->plugin
596    * aa->session_id == cac->search->session_id
597    *
598    * return as exact address
599    */
600   if (0 == cac->search->addr_len)
601   {
602     if ((0 == strcmp (aa->plugin, cac->search->plugin))
603         && (aa->session_id == cac->search->session_id))
604       cac->exact_address = aa;
605   }
606
607   if (cac->exact_address == NULL )
608     return GNUNET_YES; /* Continue iteration to find exact address */
609   else
610     return GNUNET_NO; /* Stop iteration since we have an exact address */
611 }
612
613
614 /**
615  * Find an existing equivalent address record.
616  * Compares by peer identity and network address OR by session ID
617  * (one of the two must match).
618  *
619  * @param handle the address handle
620  * @param peer peer to lookup addresses for
621  * @param addr existing address record
622  * @return existing address record, NULL for none
623  */
624 struct ATS_Address *
625 find_equivalent_address (struct GAS_Addresses_Handle *handle,
626                          const struct GNUNET_PeerIdentity *peer,
627                          const struct ATS_Address *addr)
628 {
629   struct CompareAddressContext cac;
630
631   cac.exact_address = NULL;
632   cac.base_address = NULL;
633   cac.search = addr;
634   GNUNET_CONTAINER_multipeermap_get_multiple (handle->addresses,
635                                               peer,
636                                               &compare_address_it, &cac);
637
638   if (cac.exact_address == NULL)
639     return cac.base_address;
640   return cac.exact_address;
641 }
642
643
644 /**
645  * Closure for #find_address_cb()
646  */
647 struct FindAddressContext
648 {
649   /**
650    * Session Id to look for.
651    */
652   uint32_t session_id;
653
654   /**
655    * Where to store matching address result.
656    */
657   struct ATS_Address *exact_address;
658
659 };
660
661
662 /**
663  * Find session matching given session ID.
664  *
665  * @param cls a `struct FindAddressContext`
666  * @param key peer id
667  * @param value the address to compare with
668  * @return #GNUNET_YES to continue, #GNUNET_NO if address is found
669  */
670 static int
671 find_address_cb (void *cls,
672                  const struct GNUNET_PeerIdentity *key,
673                  void *value)
674 {
675   struct FindAddressContext *fac = cls;
676   struct ATS_Address *aa = value;
677
678   if (aa->session_id == fac->session_id)
679   {
680     fac->exact_address = aa;
681     return GNUNET_NO;
682   }
683   return GNUNET_YES;
684 }
685
686
687 /**
688  * Find the exact address
689  *
690  * @param handle the address handle to use
691  * @param peer peer
692  * @param session_id session id, can never be 0
693  * @return an ATS_address or NULL
694  */
695 static struct ATS_Address *
696 find_exact_address (struct GAS_Addresses_Handle *handle,
697                     const struct GNUNET_PeerIdentity *peer,
698                     uint32_t session_id)
699 {
700   struct FindAddressContext fac;
701
702   fac.exact_address = NULL;
703   fac.session_id = session_id;
704   GNUNET_CONTAINER_multipeermap_get_multiple (handle->addresses,
705                                               peer,
706                                               &find_address_cb, &fac);
707   return fac.exact_address;
708 }
709
710
711 /**
712  * Function allowing the solver to obtain normalized preference
713  * values from solver
714  *
715  * @param cls unused
716  * @param id the peer to return the normalized properties for
717  * @return array of double values with |GNUNET_ATS_PreferenceCount| elements
718  */
719 const double *
720 get_preferences_cb (void *cls,
721                     const struct GNUNET_PeerIdentity *id)
722 {
723   return GAS_normalization_get_preferences_by_peer (id);
724 }
725
726
727 /**
728  * Function allowing the solver to obtain normalized property
729  * values for an address from solver
730  *
731  * @param cls unused
732  * @param address the address
733  * @return array of double values with |GNUNET_ATS_QualityPropertiesCount| elements
734  */
735 const double *
736 get_property_cb (void *cls, const struct ATS_Address *address)
737 {
738   return GAS_normalization_get_properties ((struct ATS_Address *) address);
739 }
740
741
742 /**
743  * Extract an ATS performance info from an address
744  *
745  * @param address the address
746  * @param type the type to extract in HBO
747  * @return the value in HBO or #GNUNET_ATS_VALUE_UNDEFINED in HBO if value does not exist
748  */
749 static int
750 get_performance_info (struct ATS_Address *address, uint32_t type)
751 {
752   int c1;
753   GNUNET_assert(NULL != address);
754
755   if ((NULL == address->atsi) || (0 == address->atsi_count))
756     return GNUNET_ATS_VALUE_UNDEFINED;
757
758   for (c1 = 0; c1 < address->atsi_count; c1++)
759   {
760     if (ntohl (address->atsi[c1].type) == type)
761       return ntohl (address->atsi[c1].value);
762   }
763   return GNUNET_ATS_VALUE_UNDEFINED;
764 }
765
766
767 /**
768  * Add a new address for a peer.
769  *
770  * @param handle the address handle to use
771  * @param peer peer
772  * @param plugin_name transport plugin name
773  * @param plugin_addr plugin address
774  * @param plugin_addr_len length of the plugin address in @a plugin_addr
775  * @param local_address_info the local address for the address
776  * @param session_id session id, can be 0
777  * @param atsi performance information for this address
778  * @param atsi_count number of performance information contained in @a atsi
779  */
780 void
781 GAS_addresses_add (struct GAS_Addresses_Handle *handle,
782                    const struct GNUNET_PeerIdentity *peer,
783                    const char *plugin_name,
784                    const void *plugin_addr,
785                    size_t plugin_addr_len,
786                    uint32_t local_address_info,
787                    uint32_t session_id,
788                    const struct GNUNET_ATS_Information *atsi,
789                    uint32_t atsi_count)
790 {
791   struct ATS_Address *new_address;
792   struct ATS_Address *existing_address;
793   struct GNUNET_ATS_Information *atsi_delta;
794   uint32_t atsi_delta_count;
795   uint32_t addr_net;
796   uint32_t previous_session;
797   int c1;
798
799   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
800              "Received `%s' for peer `%s'\n",
801              "ADDRESS ADD",
802              GNUNET_i2s (peer));
803
804   if (GNUNET_NO == handle->running)
805     return;
806
807   GNUNET_assert(NULL != handle->addresses);
808
809   new_address = create_address (peer, plugin_name,
810                                 plugin_addr, plugin_addr_len,
811                                 local_address_info, session_id);
812   atsi_delta = NULL;
813   disassemble_ats_information (new_address, atsi, atsi_count, &atsi_delta,
814                                &atsi_delta_count);
815   GNUNET_free_non_null (atsi_delta);
816   addr_net = get_performance_info (new_address, GNUNET_ATS_NETWORK_TYPE);
817   if (GNUNET_ATS_VALUE_UNDEFINED == addr_net)
818     addr_net = GNUNET_ATS_NET_UNSPECIFIED;
819
820   /* Get existing address or address with session == 0 */
821   existing_address = find_equivalent_address (handle, peer, new_address);
822   if (existing_address == NULL)
823   {
824     /* Add a new address */
825     new_address->t_added = GNUNET_TIME_absolute_get();
826     new_address->t_last_activity = GNUNET_TIME_absolute_get();
827     GNUNET_assert(GNUNET_OK ==
828                   GNUNET_CONTAINER_multipeermap_put (handle->addresses,
829                                                      peer,
830                                                      new_address,
831                                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
832
833     GNUNET_STATISTICS_set (handle->stat,
834                            "# addresses",
835                            GNUNET_CONTAINER_multipeermap_size (handle->addresses),
836                            GNUNET_NO);
837
838     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
839                 "Adding new address %p for peer `%s', length %u, session id %u, %s\n",
840                 new_address,
841                 GNUNET_i2s (peer),
842                 plugin_addr_len,
843                 session_id,
844                 GNUNET_ATS_print_network_type (addr_net));
845
846     /* Tell solver about new address */
847     handle->env.sf.s_add (handle->solver, new_address, addr_net);
848
849     handle->env.sf.s_bulk_start (handle->solver);
850     GAS_normalization_normalize_property (handle->addresses,
851                                           new_address,
852                                           atsi,
853                                           atsi_count);
854     handle->env.sf.s_bulk_stop (handle->solver);
855
856     /* Notify performance clients about new address */
857     GAS_performance_notify_all_clients (&new_address->peer, new_address->plugin,
858         new_address->addr, new_address->addr_len, new_address->active,
859         new_address->atsi, new_address->atsi_count,
860         GNUNET_BANDWIDTH_value_init (new_address->assigned_bw_out),
861         GNUNET_BANDWIDTH_value_init (new_address->assigned_bw_in));
862     return;
863   }
864
865   /* We have an existing address we can use, clean up new */
866   GNUNET_free(new_address->plugin);
867   GNUNET_free_non_null(new_address->atsi);
868   GNUNET_free(new_address);
869   new_address = NULL;
870
871   if (0 != existing_address->session_id)
872   {
873     /* Should not happen */
874     GNUNET_break(0);
875     return;
876   }
877
878   addr_net = get_performance_info (existing_address, GNUNET_ATS_NETWORK_TYPE);
879   if (GNUNET_ATS_VALUE_UNDEFINED == addr_net)
880     addr_net = GNUNET_ATS_NET_UNSPECIFIED;
881
882   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
883       "Found existing address for peer `%s' %p with new session %u in network %s\n",
884       GNUNET_i2s (peer), existing_address, session_id,
885       GNUNET_ATS_print_network_type (addr_net));
886   /* We have an address without an session, update this address */
887   existing_address->t_added = GNUNET_TIME_absolute_get();
888   existing_address->t_last_activity = GNUNET_TIME_absolute_get();
889   atsi_delta = NULL;
890   atsi_delta_count = 0;
891   if (GNUNET_YES
892       == disassemble_ats_information (existing_address, atsi, atsi_count,
893           &atsi_delta, &atsi_delta_count))
894   {
895     /* Notify performance clients about properties */
896     GAS_performance_notify_all_clients (&existing_address->peer,
897         existing_address->plugin, existing_address->addr,
898         existing_address->addr_len, existing_address->active,
899         existing_address->atsi, existing_address->atsi_count,
900         GNUNET_BANDWIDTH_value_init (existing_address->assigned_bw_out),
901         GNUNET_BANDWIDTH_value_init (existing_address->assigned_bw_in));
902
903     for (c1 = 0; c1 < atsi_delta_count; c1++)
904     {
905       if ((GNUNET_ATS_NETWORK_TYPE == ntohl (atsi_delta[c1].type))
906           && (addr_net != ntohl (atsi_delta[c1].value)))
907       {
908         /* Network type changed */
909         GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
910             "Address for peer `%s' %p changed from network %s to %s\n",
911             GNUNET_i2s (peer), existing_address,
912             GNUNET_ATS_print_network_type (addr_net),
913             GNUNET_ATS_print_network_type (ntohl (atsi_delta[c1].value)));
914         handle->env.sf.s_address_update_network (handle->solver, existing_address,
915             ntohl (atsi_delta[c1].value),
916             get_performance_info (existing_address, GNUNET_ATS_NETWORK_TYPE));
917         addr_net = get_performance_info (existing_address,
918             GNUNET_ATS_NETWORK_TYPE);
919       }
920     }
921     /* Notify solver about update with atsi information and session */
922     handle->env.sf.s_bulk_start (handle->solver);
923     GAS_normalization_normalize_property (handle->addresses, existing_address,
924         atsi, atsi_count);
925     handle->env.sf.s_bulk_stop (handle->solver);
926   }
927   GNUNET_free_non_null(atsi_delta);
928
929   /* Notify solver about new session */
930   if (existing_address->session_id == session_id)
931     return; /* possible, can both be 0 since address is revalidated */
932
933   previous_session = existing_address->session_id;
934   existing_address->session_id = session_id;
935   handle->env.sf.s_address_update_session (handle->solver, existing_address,
936       previous_session, session_id);
937
938   GNUNET_log(GNUNET_ERROR_TYPE_INFO,
939       "Updated existing address for peer `%s' %p length %u with new session %u in network %s\n",
940       GNUNET_i2s (peer), existing_address, existing_address->addr_len,
941       session_id, GNUNET_ATS_print_network_type (addr_net));
942 }
943
944
945 /**
946  * Update an address with new performance information for a peer.
947  *
948  * @param handle the address handle to use
949  * @param peer peer
950  * @param session_id session id, never 0
951  * @param atsi performance information for this address
952  * @param atsi_count number of performance information contained in @a atsi
953  */
954 void
955 GAS_addresses_update (struct GAS_Addresses_Handle *handle,
956                       const struct GNUNET_PeerIdentity *peer,
957                       uint32_t session_id,
958                       const struct GNUNET_ATS_Information *atsi,
959                       uint32_t atsi_count)
960 {
961   struct ATS_Address *aa;
962   struct GNUNET_ATS_Information *atsi_delta;
963   uint32_t atsi_delta_count;
964   uint32_t prev_session;
965   int c1;
966
967   if (GNUNET_NO == handle->running)
968     return;
969   GNUNET_assert (NULL != handle->addresses);
970
971   /* Get existing address */
972   aa = find_exact_address (handle,
973                            peer,
974                            session_id);
975   if (NULL == aa)
976   {
977     GNUNET_break (0);
978     return;
979   }
980   if (NULL == aa->solver_information)
981   {
982     GNUNET_break (0);
983     return;
984   }
985
986   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
987               "Received `%s' for peer `%s' address \n",
988               "ADDRESS UPDATE",
989               GNUNET_i2s (peer),
990               aa);
991
992   /* Update address */
993   aa->t_last_activity = GNUNET_TIME_absolute_get();
994   if (session_id != aa->session_id)
995   {
996     /* Session changed */
997     prev_session = aa->session_id;
998     aa->session_id = session_id;
999     handle->env.sf.s_address_update_session (handle->solver,
1000                                              aa,
1001                                              prev_session,
1002                                              aa->session_id);
1003   }
1004
1005   atsi_delta = NULL;
1006   atsi_delta_count = 0;
1007   if (GNUNET_YES ==
1008       disassemble_ats_information (aa, atsi,
1009                                    atsi_count,
1010                                    &atsi_delta,
1011                                    &atsi_delta_count))
1012   {
1013     /* ATS properties changed */
1014     for (c1 = 0; c1 < atsi_delta_count; c1++)
1015     {
1016       if (GNUNET_ATS_NETWORK_TYPE == ntohl (atsi_delta[c1].type))
1017       {
1018         /* Network type changed */
1019         handle->env.sf.s_address_update_network (handle->solver, aa,
1020             ntohl (atsi_delta[c1].value),
1021             get_performance_info (aa, GNUNET_ATS_NETWORK_TYPE));
1022       }
1023     }
1024
1025     /* Notify performance clients about updated address */
1026     GAS_performance_notify_all_clients (&aa->peer, aa->plugin, aa->addr,
1027         aa->addr_len, aa->active, aa->atsi, aa->atsi_count,
1028         GNUNET_BANDWIDTH_value_init (aa->assigned_bw_out),
1029         GNUNET_BANDWIDTH_value_init (aa->assigned_bw_in));
1030
1031     handle->env.sf.s_bulk_start (handle->solver);
1032     GAS_normalization_normalize_property (handle->addresses,
1033                                           aa,
1034                                           atsi,
1035                                           atsi_count);
1036     handle->env.sf.s_bulk_stop (handle->solver);
1037   }
1038   GNUNET_free_non_null (atsi_delta);
1039 }
1040
1041
1042 /**
1043  * Remove an address or just a session for a peer.
1044  *
1045  * @param handle the address handle to use
1046  * @param peer peer
1047  * @param session_id session id, can never be 0
1048  */
1049 void
1050 GAS_addresses_destroy (struct GAS_Addresses_Handle *handle,
1051                        const struct GNUNET_PeerIdentity *peer,
1052                        uint32_t session_id)
1053 {
1054   struct ATS_Address *ea;
1055
1056   if (GNUNET_NO == handle->running)
1057     return;
1058
1059   /* Get existing address */
1060   ea = find_exact_address (handle,
1061                            peer,
1062                            session_id);
1063   if (NULL == ea)
1064   {
1065     GNUNET_break (0);
1066     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1067                 "Tried to destroy unknown address for peer `%s' session id %u\n",
1068                 GNUNET_i2s (peer),
1069                 session_id);
1070     return;
1071   }
1072
1073   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1074               "Received `%s' for peer `%s' address %p session %u\n",
1075               "ADDRESS DESTROYED",
1076               GNUNET_i2s (peer),
1077               ea,
1078               session_id);
1079   GNUNET_CONTAINER_multipeermap_remove (handle->addresses,
1080                                         peer,
1081                                         ea);
1082
1083   handle->env.sf.s_del (handle->solver, ea, GNUNET_NO);
1084   GAS_performance_notify_all_clients (peer,
1085                                       ea->plugin,
1086                                       ea->addr,
1087                                       ea->addr_len,
1088                                       GNUNET_SYSERR,
1089                                       NULL, 0,
1090                                       zero_bw,
1091                                       zero_bw);
1092   free_address (ea);
1093   GNUNET_STATISTICS_set (handle->stat,
1094                          "# addresses",
1095                          GNUNET_CONTAINER_multipeermap_size (handle->addresses),
1096                          GNUNET_NO);
1097 }
1098
1099
1100 /**
1101  * Notification about active use of an address.
1102  * in_use == #GNUNET_YES:
1103  *      This address is used to maintain an active connection with a peer.
1104  * in_use == #GNUNET_NO:
1105  *      This address is no longer used to maintain an active connection with a peer.
1106  *
1107  * Note: can only be called with in_use == #GNUNET_NO if called with #GNUNET_YES
1108  * before
1109  *
1110  * @param handle the address handle to use
1111  * @param peer peer
1112  * @param session_id session id, can be 0
1113  * @param in_use #GNUNET_YES if #GNUNET_NO FIXME
1114  * @return #GNUNET_SYSERR on failure (address unknown ...)
1115  */
1116 int
1117 GAS_addresses_in_use (struct GAS_Addresses_Handle *handle,
1118                       const struct GNUNET_PeerIdentity *peer,
1119                       uint32_t session_id,
1120                       int in_use)
1121 {
1122   struct ATS_Address *ea;
1123
1124   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1125              "Received `%s' for peer `%s'\n",
1126              "ADDRESS IN USE",
1127              GNUNET_i2s (peer));
1128   if (GNUNET_NO == handle->running)
1129     return GNUNET_SYSERR;
1130   ea = find_exact_address (handle,
1131                            peer,
1132                            session_id);
1133   if (NULL == ea)
1134   {
1135     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1136                 "Trying to set unknown address `%s' `%u' to %s \n",
1137                 GNUNET_i2s (peer),
1138                 session_id,
1139                 (GNUNET_NO == in_use) ? "NO" : "YES");
1140     GNUNET_break (0);
1141     return GNUNET_SYSERR;
1142   }
1143   if (ea->used == in_use)
1144   {
1145     GNUNET_break (0);
1146     GNUNET_log(GNUNET_ERROR_TYPE_WARNING,
1147                "Address in use called multiple times for peer `%s': %s -> %s \n",
1148                GNUNET_i2s (peer),
1149                (GNUNET_NO == ea->used) ? "NO" : "YES",
1150                (GNUNET_NO == in_use) ? "NO" : "YES");
1151     return GNUNET_SYSERR;
1152   }
1153   /* Tell solver about update */
1154   ea->used = in_use;
1155   ea->t_last_activity = GNUNET_TIME_absolute_get();
1156   handle->env.sf.s_address_update_inuse (handle->solver,
1157                                          ea,
1158                                          ea->used);
1159   return GNUNET_OK;
1160 }
1161
1162
1163 /**
1164  * Cancel address suggestions for a peer
1165  *
1166  * @param handle the address handle
1167  * @param peer the peer id
1168  */
1169 void
1170 GAS_addresses_request_address_cancel (struct GAS_Addresses_Handle *handle,
1171                                       const struct GNUNET_PeerIdentity *peer)
1172 {
1173   struct GAS_Addresses_Suggestion_Requests *cur = handle->pending_requests_head;
1174
1175   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Received request: `%s' for peer %s\n",
1176       "request_address_cancel", GNUNET_i2s (peer));
1177
1178   while (NULL != cur)
1179   {
1180     if (0 == memcmp (peer, &cur->id, sizeof(cur->id)))
1181       break; /* found */
1182     cur = cur->next;
1183   }
1184
1185   if (NULL == cur)
1186   {
1187     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1188         "No address requests pending for peer `%s', cannot remove!\n",
1189         GNUNET_i2s (peer));
1190     return;
1191   }
1192   handle->env.sf.s_get_stop (handle->solver, peer);
1193   GAS_addresses_handle_backoff_reset (handle, peer);
1194   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Removed request pending for peer `%s\n",
1195       GNUNET_i2s (peer));
1196   GNUNET_CONTAINER_DLL_remove(handle->pending_requests_head, handle->pending_requests_tail, cur);
1197   GNUNET_free(cur);
1198 }
1199
1200
1201 /**
1202  * Request address suggestions for a peer
1203  *
1204  * @param handle the address handle
1205  * @param peer the peer id
1206  */
1207 void
1208 GAS_addresses_request_address (struct GAS_Addresses_Handle *handle,
1209     const struct GNUNET_PeerIdentity *peer)
1210 {
1211   struct GAS_Addresses_Suggestion_Requests *cur = handle->pending_requests_head;
1212   struct ATS_Address *aa;
1213
1214   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Received `%s' for peer `%s'\n",
1215       "REQUEST ADDRESS", GNUNET_i2s (peer));
1216
1217   if (GNUNET_NO == handle->running)
1218     return;
1219   while (NULL != cur)
1220   {
1221     if (0 == memcmp (peer, &cur->id, sizeof(cur->id)))
1222       break; /* already suggesting */
1223     cur = cur->next;
1224   }
1225   if (NULL == cur)
1226   {
1227     cur = GNUNET_new (struct GAS_Addresses_Suggestion_Requests);
1228     cur->id = (*peer);
1229     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1230         "Adding new address suggestion request for `%s'\n",
1231          GNUNET_i2s (peer));
1232     GNUNET_CONTAINER_DLL_insert(handle->pending_requests_head, handle->pending_requests_tail, cur);
1233   }
1234
1235   /* Get prefered address from solver */
1236   aa = (struct ATS_Address *) handle->env.sf.s_get (handle->solver, peer);
1237   if (NULL == aa)
1238   {
1239     GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Cannot suggest address for peer `%s'\n",
1240         GNUNET_i2s (peer));
1241     return;
1242   }
1243
1244   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Suggesting address %p for peer `%s'\n",
1245       aa, GNUNET_i2s (peer));
1246
1247   GAS_scheduling_transmit_address_suggestion (peer,
1248                                               aa->session_id,
1249                                               GNUNET_BANDWIDTH_value_init (aa->assigned_bw_out),
1250                                               GNUNET_BANDWIDTH_value_init (aa->assigned_bw_in));
1251
1252   aa->block_interval = GNUNET_TIME_relative_add (aa->block_interval,
1253       ATS_BLOCKING_DELTA);
1254   aa->blocked_until = GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get (),
1255       aa->block_interval);
1256
1257   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1258       "Address %p ready for suggestion, block interval now %llu \n", aa,
1259       aa->block_interval);
1260 }
1261
1262 /**
1263  * Iterator to reset address blocking
1264  *
1265  * @param cls not used
1266  * @param key the peer
1267  * @param value the address to reset
1268  * @return #GNUNET_OK to continue
1269  */
1270 static int
1271 reset_address_it (void *cls,
1272                   const struct GNUNET_PeerIdentity *key,
1273                   void *value)
1274 {
1275   struct ATS_Address *aa = value;
1276
1277   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1278              "Resetting interval for peer `%s' address %p from %llu to 0\n",
1279              GNUNET_i2s (&aa->peer),
1280              aa,
1281              aa->block_interval);
1282   aa->blocked_until = GNUNET_TIME_UNIT_ZERO_ABS;
1283   aa->block_interval = GNUNET_TIME_UNIT_ZERO;
1284   return GNUNET_OK;
1285 }
1286
1287
1288 /**
1289  * Reset suggestion backoff for a peer
1290  *
1291  * Suggesting addresses is blocked for ATS_BLOCKING_DELTA. Blocking can be
1292  * reset using this function
1293  *
1294  * @param handle the address handle
1295  * @param peer the peer id
1296  */
1297 void
1298 GAS_addresses_handle_backoff_reset (struct GAS_Addresses_Handle *handle,
1299     const struct GNUNET_PeerIdentity *peer)
1300 {
1301   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Received `%s' for peer `%s'\n",
1302       "RESET BACKOFF", GNUNET_i2s (peer));
1303
1304   GNUNET_break(
1305       GNUNET_SYSERR != GNUNET_CONTAINER_multipeermap_get_multiple (handle->addresses,
1306                                                                    peer,
1307                                                                    &reset_address_it, NULL));
1308 }
1309
1310
1311 /**
1312  * Solver information callback
1313  *
1314  * @param cls the closure
1315  * @param op the operation
1316  * @param stat operation status
1317  * @param add additional information
1318  */
1319
1320 static void
1321 solver_info_cb (void *cls,
1322     enum GAS_Solver_Operation op,
1323     enum GAS_Solver_Status stat,
1324     enum GAS_Solver_Additional_Information add)
1325 {
1326   char *add_info;
1327
1328   switch (add) {
1329     case GAS_INFO_NONE:
1330       add_info = "GAS_INFO_NONE";
1331       break;
1332     case GAS_INFO_FULL:
1333       add_info = "GAS_INFO_MLP_FULL";
1334       break;
1335     case GAS_INFO_UPDATED:
1336       add_info = "GAS_INFO_MLP_UPDATED";
1337       break;
1338     case GAS_INFO_PROP_ALL:
1339       add_info = "GAS_INFO_PROP_ALL";
1340       break;
1341     case GAS_INFO_PROP_SINGLE:
1342       add_info = "GAS_INFO_PROP_SINGLE";
1343       break;
1344     default:
1345       add_info = "INVALID";
1346       break;
1347   }
1348   switch (op)
1349   {
1350     case GAS_OP_SOLVE_START:
1351       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1352           "Solver notifies `%s' with result `%s' `%s'\n", "GAS_OP_SOLVE_START",
1353           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL", add_info);
1354       return;
1355     case GAS_OP_SOLVE_STOP:
1356       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1357           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_STOP",
1358           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL", add_info);
1359       return;
1360
1361     case GAS_OP_SOLVE_SETUP_START:
1362       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1363           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_SETUP_START",
1364           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1365       return;
1366
1367     case GAS_OP_SOLVE_SETUP_STOP:
1368       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1369           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_SETUP_STOP",
1370           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1371       return;
1372
1373     case GAS_OP_SOLVE_MLP_LP_START:
1374       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1375           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_LP_START",
1376           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1377       return;
1378     case GAS_OP_SOLVE_MLP_LP_STOP:
1379       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1380           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_LP_STOP",
1381           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1382       return;
1383
1384     case GAS_OP_SOLVE_MLP_MLP_START:
1385       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1386           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_MLP_START",
1387           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1388       return;
1389     case GAS_OP_SOLVE_MLP_MLP_STOP:
1390       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1391           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_MLP_STOP",
1392           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1393       return;
1394     case GAS_OP_SOLVE_UPDATE_NOTIFICATION_START:
1395       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1396           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_UPDATE_NOTIFICATION_START",
1397           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1398       return;
1399     case GAS_OP_SOLVE_UPDATE_NOTIFICATION_STOP:
1400       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1401           "Solver notifies `%s' with result `%s'\n", "GAS_OP_SOLVE_UPDATE_NOTIFICATION_STOP",
1402           (GAS_STAT_SUCCESS == stat) ? "SUCCESS" : "FAIL");
1403       return;
1404     default:
1405       break;
1406     }
1407 }
1408
1409
1410 /**
1411  * The preference changed for a peer
1412  *
1413  * @param cls the address handle
1414  * @param peer the peer
1415  * @param kind the ATS kind
1416  * @param pref_rel the new relative preference value
1417  */
1418 static void
1419 normalized_preference_changed_cb (void *cls,
1420     const struct GNUNET_PeerIdentity *peer, enum GNUNET_ATS_PreferenceKind kind,
1421     double pref_rel)
1422 {
1423   GNUNET_assert(NULL != cls);
1424   struct GAS_Addresses_Handle *handle = cls;
1425
1426   /* Tell solver about update */
1427   handle->env.sf.s_pref (handle->solver, peer, kind, pref_rel);
1428 }
1429
1430 /**
1431  * The relative value for a property changed
1432  *
1433  * @param cls the address handle
1434  * @param address the peer
1435  * @param type the ATS type
1436  * @param prop_rel the new relative preference value
1437  */
1438 static void
1439 normalized_property_changed_cb (void *cls, struct ATS_Address *address,
1440     uint32_t type, double prop_rel)
1441 {
1442   struct GAS_Addresses_Handle *ah = (struct GAS_Addresses_Handle *) cls;
1443   GNUNET_assert(NULL != ah);
1444
1445   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1446       "Normalized property %s for peer `%s' changed to %.3f \n",
1447       GNUNET_ATS_print_property_type (type), GNUNET_i2s (&address->peer),
1448       prop_rel);
1449
1450   ah->env.sf.s_address_update_property (ah->solver, address, type, 0, prop_rel);
1451 }
1452
1453 static struct GAS_Addresses_Preference_Clients *
1454 find_preference_client (struct GAS_Addresses_Handle *handle, void *client)
1455 {
1456   struct GAS_Addresses_Preference_Clients *cur;
1457
1458   for (cur = handle->preference_clients_head; NULL != cur; cur = cur->next)
1459   {
1460     if (cur->client == client)
1461       return cur;
1462   }
1463   return NULL;
1464 }
1465
1466
1467 /**
1468  * A performance client disconnected
1469  *
1470  * @param handle address handle
1471  * @param client the client
1472  */
1473 void
1474 GAS_addresses_preference_client_disconnect (struct GAS_Addresses_Handle *handle,
1475                                             void *client)
1476 {
1477   struct GAS_Addresses_Preference_Clients * pc;
1478   if (NULL != (pc = find_preference_client (handle, client)))
1479   {
1480     GNUNET_CONTAINER_DLL_remove (handle->preference_clients_head,
1481         handle->preference_clients_tail, pc);
1482     GNUNET_free (pc);
1483     GNUNET_assert (handle->pref_clients > 0);
1484     handle->pref_clients --;
1485     GNUNET_STATISTICS_set (handle->stat, "# active performance clients", handle->pref_clients, GNUNET_NO);
1486   }
1487   GAS_normalization_preference_client_disconnect (client);
1488 }
1489
1490
1491 /**
1492  * Change the preference for a peer
1493  *
1494  * @param handle the address handle
1495  * @param client the client sending this request
1496  * @param peer the peer id
1497  * @param kind the preference kind to change
1498  * @param score_abs the new preference score
1499  */
1500 void
1501 GAS_addresses_preference_change (struct GAS_Addresses_Handle *handle,
1502     void *client, const struct GNUNET_PeerIdentity *peer,
1503     enum GNUNET_ATS_PreferenceKind kind, float score_abs)
1504 {
1505   struct GAS_Addresses_Preference_Clients * pc;
1506   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1507       "Received `%s' for peer `%s' for client %p\n", "CHANGE PREFERENCE",
1508       GNUNET_i2s (peer), client);
1509
1510   if (GNUNET_NO == handle->running)
1511     return;
1512
1513   if (GNUNET_NO ==
1514       GNUNET_CONTAINER_multipeermap_contains (handle->addresses,
1515                                               peer))
1516   {
1517     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1518         "Received `%s' for unknown peer `%s' from client %p\n",
1519         "CHANGE PREFERENCE", GNUNET_i2s (peer), client);
1520     return;
1521   }
1522
1523   if (NULL == find_preference_client (handle, client))
1524   {
1525     pc = GNUNET_new (struct GAS_Addresses_Preference_Clients);
1526     pc->client = client;
1527     GNUNET_CONTAINER_DLL_insert (handle->preference_clients_head,
1528         handle->preference_clients_tail, pc);
1529     handle->pref_clients ++;
1530     GNUNET_STATISTICS_set (handle->stat,
1531                            "# active performance clients",
1532                            handle->pref_clients,
1533                            GNUNET_NO);
1534   }
1535
1536   handle->env.sf.s_bulk_start (handle->solver);
1537   /* Tell normalization about change, normalization will call callback if preference changed */
1538   GAS_normalization_normalize_preference (client, peer, kind, score_abs);
1539   handle->env.sf.s_bulk_stop (handle->solver);
1540 }
1541
1542
1543 /**
1544  * Change the preference for a peer
1545  *
1546  * @param handle the address handle
1547  * @param application the client sending this request
1548  * @param peer the peer id
1549  * @param scope the time interval for this feedback: [now - scope .. now]
1550  * @param kind the preference kind to change
1551  * @param score_abs the new preference score
1552  */
1553 void
1554 GAS_addresses_preference_feedback (struct GAS_Addresses_Handle *handle,
1555                                    void *application,
1556                                    const struct GNUNET_PeerIdentity *peer,
1557                                    const struct GNUNET_TIME_Relative scope,
1558                                    enum GNUNET_ATS_PreferenceKind kind,
1559                                    float score_abs)
1560 {
1561   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1562       "Received `%s' for peer `%s' for client %p\n", "PREFERENCE FEEDBACK",
1563       GNUNET_i2s (peer), application);
1564
1565   if (GNUNET_NO == handle->running)
1566     return;
1567
1568   if (GNUNET_NO ==
1569       GNUNET_CONTAINER_multipeermap_contains (handle->addresses,
1570                                               peer))
1571   {
1572     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1573         "Received `%s' for unknown peer `%s' from client %p\n",
1574         "PREFERENCE FEEDBACK", GNUNET_i2s (peer), application);
1575     return;
1576   }
1577
1578   handle->env.sf.s_feedback (handle->solver, application, peer, scope, kind,
1579       score_abs);
1580 }
1581
1582
1583 /**
1584  * Load quotas for networks from configuration
1585  *
1586  * @param cfg configuration handle
1587  * @param out_dest where to write outbound quotas
1588  * @param in_dest where to write inbound quotas
1589  * @param dest_length length of inbound and outbound arrays
1590  * @return number of networks loaded
1591  */
1592 static unsigned int
1593 load_quotas (const struct GNUNET_CONFIGURATION_Handle *cfg,
1594     unsigned long long *out_dest, unsigned long long *in_dest, int dest_length)
1595 {
1596   char * entry_in = NULL;
1597   char * entry_out = NULL;
1598   char * quota_out_str;
1599   char * quota_in_str;
1600   int c;
1601   int res;
1602
1603   for (c = 0; (c < GNUNET_ATS_NetworkTypeCount) && (c < dest_length); c++)
1604   {
1605     in_dest[c] = 0;
1606     out_dest[c] = 0;
1607     GNUNET_asprintf (&entry_out,
1608                      "%s_QUOTA_OUT",
1609                      GNUNET_ATS_print_network_type (c));
1610     GNUNET_asprintf (&entry_in,
1611                      "%s_QUOTA_IN",
1612                      GNUNET_ATS_print_network_type (c));
1613
1614     /* quota out */
1615     if (GNUNET_OK
1616         == GNUNET_CONFIGURATION_get_value_string (cfg, "ats", entry_out,
1617             &quota_out_str))
1618     {
1619       res = GNUNET_NO;
1620       if (0 == strcmp (quota_out_str, GNUNET_ATS_MaxBandwidthString))
1621       {
1622         out_dest[c] = GNUNET_ATS_MaxBandwidth;
1623         res = GNUNET_YES;
1624       }
1625       if ((GNUNET_NO == res)
1626           && (GNUNET_OK
1627               == GNUNET_STRINGS_fancy_size_to_bytes (quota_out_str,
1628                   &out_dest[c])))
1629         res = GNUNET_YES;
1630       if ((GNUNET_NO == res)
1631           && (GNUNET_OK
1632               == GNUNET_CONFIGURATION_get_value_number (cfg, "ats", entry_out,
1633                   &out_dest[c])))
1634         res = GNUNET_YES;
1635
1636       if (GNUNET_NO == res)
1637       {
1638         GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
1639                    _("Could not load quota for network `%s':  `%s', assigning default bandwidth %llu\n"),
1640                    GNUNET_ATS_print_network_type (c),
1641                    quota_out_str,
1642                    GNUNET_ATS_DefaultBandwidth);
1643         out_dest[c] = GNUNET_ATS_DefaultBandwidth;
1644       }
1645       else
1646       {
1647         GNUNET_log(GNUNET_ERROR_TYPE_INFO,
1648                    _("Outbound quota configure for network `%s' is %llu\n"),
1649                    GNUNET_ATS_print_network_type (c),
1650                    out_dest[c]);
1651       }
1652       GNUNET_free(quota_out_str);
1653     }
1654     else
1655     {
1656       GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
1657                  _("No outbound quota configured for network `%s', assigning default bandwidth %llu\n"),
1658                  GNUNET_ATS_print_network_type (c),
1659                  GNUNET_ATS_DefaultBandwidth);
1660       out_dest[c] = GNUNET_ATS_DefaultBandwidth;
1661     }
1662
1663     /* quota in */
1664     if (GNUNET_OK
1665         == GNUNET_CONFIGURATION_get_value_string (cfg, "ats", entry_in,
1666             &quota_in_str))
1667     {
1668       res = GNUNET_NO;
1669       if (0 == strcmp (quota_in_str, GNUNET_ATS_MaxBandwidthString))
1670       {
1671         in_dest[c] = GNUNET_ATS_MaxBandwidth;
1672         res = GNUNET_YES;
1673       }
1674       if ((GNUNET_NO == res)
1675           && (GNUNET_OK
1676               == GNUNET_STRINGS_fancy_size_to_bytes (quota_in_str, &in_dest[c])))
1677         res = GNUNET_YES;
1678       if ((GNUNET_NO == res)
1679           && (GNUNET_OK
1680               == GNUNET_CONFIGURATION_get_value_number (cfg, "ats", entry_in,
1681                   &in_dest[c])))
1682         res = GNUNET_YES;
1683
1684       if (GNUNET_NO == res)
1685       {
1686         GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
1687                    _("Could not load quota for network `%s':  `%s', assigning default bandwidth %llu\n"),
1688                    GNUNET_ATS_print_network_type (c),
1689                    quota_in_str,
1690                    GNUNET_ATS_DefaultBandwidth);
1691         in_dest[c] = GNUNET_ATS_DefaultBandwidth;
1692       }
1693       else
1694       {
1695         GNUNET_log(GNUNET_ERROR_TYPE_INFO,
1696                    _("Inbound quota configured for network `%s' is %llu\n"),
1697                    GNUNET_ATS_print_network_type (c),
1698                    in_dest[c]);
1699       }
1700       GNUNET_free(quota_in_str);
1701     }
1702     else
1703     {
1704       GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
1705                  _("No outbound quota configure for network `%s', assigning default bandwidth %llu\n"),
1706                  GNUNET_ATS_print_network_type (c),
1707                  GNUNET_ATS_DefaultBandwidth);
1708       in_dest[c] = GNUNET_ATS_DefaultBandwidth;
1709     }
1710     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1711                "Loaded quota for network `%s' (in/out): %llu %llu\n",
1712                GNUNET_ATS_print_network_type (c),
1713                in_dest[c],
1714                out_dest[c]);
1715     GNUNET_free(entry_out);
1716     GNUNET_free(entry_in);
1717   }
1718   return GNUNET_ATS_NetworkTypeCount;
1719 }
1720
1721
1722 /**
1723  * Callback for solver to notify about assignment changes
1724  *
1725  * @param cls the GAS_Addresses_Handle
1726  * @param address the address with changes
1727  */
1728 static void
1729 bandwidth_changed_cb (void *cls, struct ATS_Address *address)
1730 {
1731   struct GAS_Addresses_Handle *handle = cls;
1732   struct GAS_Addresses_Suggestion_Requests *cur;
1733   uint32_t diff_out;
1734   uint32_t diff_in;
1735
1736   GNUNET_assert(handle != NULL);
1737   GNUNET_assert(address != NULL);
1738   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1739               "Bandwidth assignment changed for peer %s \n",
1740               GNUNET_i2s (&address->peer));
1741
1742   /* Notify performance clients about changes to address */
1743   GAS_performance_notify_all_clients (&address->peer, address->plugin,
1744       address->addr, address->addr_len, address->active, address->atsi,
1745       address->atsi_count,
1746       GNUNET_BANDWIDTH_value_init (address->assigned_bw_out),
1747       GNUNET_BANDWIDTH_value_init (address->assigned_bw_in));
1748
1749   cur = handle->pending_requests_head;
1750   while (NULL != cur)
1751   {
1752     if (0 == memcmp (&address->peer, &cur->id, sizeof(cur->id)))
1753       break; /* we have an address request pending*/
1754     cur = cur->next;
1755   }
1756   if (NULL == cur)
1757   {
1758     GNUNET_log(GNUNET_ERROR_TYPE_INFO,
1759                "Nobody is interested in peer `%s' :(\n",
1760                GNUNET_i2s (&address->peer));
1761     return;
1762   }
1763
1764   if ((0 == address->assigned_bw_in) && (0 == address->assigned_bw_out))
1765   {
1766     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1767                "Telling transport to disconnect peer `%s'\n",
1768                 GNUNET_i2s (&address->peer));
1769
1770     /* Notify scheduling clients about suggestion */
1771     GAS_scheduling_transmit_address_suggestion (&address->peer,
1772                                                 address->session_id,
1773                                                 GNUNET_BANDWIDTH_value_init (0),
1774                                                 GNUNET_BANDWIDTH_value_init (0));
1775     return;
1776   }
1777
1778   /* Do bandwidth stability check */
1779   diff_out = abs (address->assigned_bw_out - address->last_notified_bw_out);
1780   diff_in = abs (address->assigned_bw_in - address->last_notified_bw_in);
1781
1782   if ( (diff_out < htonl(GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT.value__)) &&
1783        (diff_in < htonl(GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT.value__)) )
1784     return;
1785
1786   GNUNET_log(GNUNET_ERROR_TYPE_INFO,
1787       "Sending bandwidth update for peer `%s': %u %u\n",
1788       GNUNET_i2s (&address->peer), address->assigned_bw_out,
1789       address->assigned_bw_out);
1790
1791   /* *Notify scheduling clients about suggestion */
1792   GAS_scheduling_transmit_address_suggestion (&address->peer,
1793                                               address->session_id,
1794                                               GNUNET_BANDWIDTH_value_init (address->assigned_bw_out),
1795                                               GNUNET_BANDWIDTH_value_init (address->assigned_bw_in));
1796
1797   address->last_notified_bw_out = address->assigned_bw_out;
1798   address->last_notified_bw_in = address->assigned_bw_in;
1799 }
1800
1801
1802 /**
1803  * Initialize address subsystem. The addresses subsystem manages the addresses
1804  * known and current performance information. It has a solver component
1805  * responsible for the resource allocation. It tells the solver about changes
1806  * and receives updates when the solver changes the resource allocation.
1807  *
1808  * @param cfg configuration to use
1809  * @param stats the statistics handle to use
1810  * @return an address handle
1811  */
1812 struct GAS_Addresses_Handle *
1813 GAS_addresses_init (const struct GNUNET_CONFIGURATION_Handle *cfg,
1814                     const struct GNUNET_STATISTICS_Handle *stats)
1815 {
1816   struct GAS_Addresses_Handle *ah;
1817   unsigned long long quotas_in[GNUNET_ATS_NetworkTypeCount];
1818   unsigned long long quotas_out[GNUNET_ATS_NetworkTypeCount];
1819   char *mode_str;
1820   char *plugin_short;
1821   int c;
1822
1823   ah = GNUNET_new (struct GAS_Addresses_Handle);
1824   ah->running = GNUNET_NO;
1825
1826   ah->stat = (struct GNUNET_STATISTICS_Handle *) stats;
1827   /* Initialize the addresses database */
1828   ah->addresses = GNUNET_CONTAINER_multipeermap_create (128, GNUNET_NO);
1829   ah->pref_clients = 0;
1830   GNUNET_assert(NULL != ah->addresses);
1831
1832   /* Figure out configured solution method */
1833   if (GNUNET_SYSERR ==
1834       GNUNET_CONFIGURATION_get_value_string (cfg, "ats", "MODE", &mode_str))
1835   {
1836     GNUNET_log(GNUNET_ERROR_TYPE_WARNING,
1837                "No resource assignment method configured, using proportional approach\n");
1838     ah->ats_mode = MODE_PROPORTIONAL;
1839   }
1840   else
1841   {
1842     for (c = 0; c < strlen (mode_str); c++)
1843       mode_str[c] = toupper (mode_str[c]);
1844     if (0 == strcmp (mode_str, "PROPORTIONAL"))
1845       ah->ats_mode = MODE_PROPORTIONAL;
1846     else if (0 == strcmp (mode_str, "MLP"))
1847     {
1848       ah->ats_mode = MODE_MLP;
1849 #if !HAVE_LIBGLPK
1850       GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
1851                  "Assignment method `%s' configured, but GLPK is not available, please install \n",
1852                  mode_str);
1853       ah->ats_mode = MODE_PROPORTIONAL;
1854 #endif
1855     }
1856     else if (0 == strcmp (mode_str, "RIL"))
1857       ah->ats_mode = MODE_RIL;
1858     else
1859     {
1860       GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
1861                  "Invalid resource assignment method `%s' configured, using proportional approach\n",
1862                  mode_str);
1863       ah->ats_mode = MODE_PROPORTIONAL;
1864     }
1865     GNUNET_free(mode_str);
1866   }
1867
1868   load_quotas (cfg, quotas_out, quotas_in, GNUNET_ATS_NetworkTypeCount);
1869   ah->env.info_cb = &solver_info_cb;
1870   ah->env.info_cb_cls = ah;
1871   ah->env.bandwidth_changed_cb = &bandwidth_changed_cb;
1872   ah->env.bw_changed_cb_cls = ah;
1873   ah->env.get_preferences = &get_preferences_cb;
1874   ah->env.get_preference_cls = ah;
1875   ah->env.get_property = &get_property_cb;
1876   ah->env.get_property_cls = ah;
1877   ah->env.cfg = cfg;
1878   ah->env.stats = stats;
1879   ah->env.addresses = ah->addresses;
1880
1881   ah->env.network_count = GNUNET_ATS_NetworkTypeCount;
1882   int networks[GNUNET_ATS_NetworkTypeCount] = GNUNET_ATS_NetworkType;
1883   for (c = 0; c < GNUNET_ATS_NetworkTypeCount; c++)
1884   {
1885     ah->env.networks[c] = networks[c];
1886     ah->env.out_quota[c] = quotas_out[c];
1887     ah->env.in_quota[c] = quotas_in[c];
1888   }
1889
1890   switch (ah->ats_mode) {
1891     case MODE_PROPORTIONAL:
1892       plugin_short = "proportional";
1893       break;
1894     case MODE_MLP:
1895       plugin_short = "mlp";
1896       break;
1897     case MODE_RIL:
1898       plugin_short = "ril";
1899       break;
1900     default:
1901       plugin_short = NULL;
1902       break;
1903   }
1904   GNUNET_asprintf (&ah->plugin,
1905                    "libgnunet_plugin_ats_%s",
1906                    plugin_short);
1907   GNUNET_log(GNUNET_ERROR_TYPE_INFO,
1908              _("Initializing solver `%s '`%s'\n"),
1909              plugin_short,
1910              ah->plugin);
1911   if (NULL == (ah->solver = GNUNET_PLUGIN_load (ah->plugin, &ah->env)))
1912   {
1913     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1914                 _("Failed to initialize solver `%s'!\n"),
1915                 ah->plugin);
1916     return NULL;
1917   }
1918
1919   GNUNET_assert (NULL != ah->env.sf.s_add);
1920   GNUNET_assert (NULL != ah->env.sf.s_address_update_inuse);
1921   GNUNET_assert (NULL != ah->env.sf.s_address_update_property);
1922   GNUNET_assert (NULL != ah->env.sf.s_address_update_session);
1923   GNUNET_assert (NULL != ah->env.sf.s_address_update_network);
1924   GNUNET_assert (NULL != ah->env.sf.s_get);
1925   GNUNET_assert (NULL != ah->env.sf.s_get_stop);
1926   GNUNET_assert (NULL != ah->env.sf.s_pref);
1927   GNUNET_assert (NULL != ah->env.sf.s_feedback);
1928   GNUNET_assert (NULL != ah->env.sf.s_del);
1929   GNUNET_assert (NULL != ah->env.sf.s_bulk_start);
1930   GNUNET_assert (NULL != ah->env.sf.s_bulk_stop);
1931
1932
1933   GAS_normalization_start (&normalized_preference_changed_cb, ah,
1934                            &normalized_property_changed_cb, ah);
1935
1936   if (NULL == ah->solver)
1937   {
1938     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1939                 _("Failed to initialize solver!\n"));
1940     GNUNET_free(ah);
1941     return NULL ;
1942   }
1943   /* up and running */
1944   ah->running = GNUNET_YES;
1945
1946   GNUNET_STATISTICS_set (ah->stat, "# addresses",
1947       GNUNET_CONTAINER_multipeermap_size (ah->addresses), GNUNET_NO);
1948
1949   return ah;
1950 }
1951
1952
1953 /**
1954  * Destroy all addresses iterator
1955  *
1956  * @param cls NULL
1957  * @param key peer identity (unused)
1958  * @param value the 'struct ATS_Address' to free
1959  * @return #GNUNET_OK (continue to iterate)
1960  */
1961 static int
1962 destroy_all_address_it (void *cls,
1963                         const struct GNUNET_PeerIdentity *key,
1964                         void *value)
1965 {
1966   struct GAS_Addresses_Handle *handle = cls;
1967   struct ATS_Address *aa = value;
1968
1969   /* Remove */
1970   GNUNET_assert(GNUNET_YES ==
1971                 GNUNET_CONTAINER_multipeermap_remove (handle->addresses, key, value));
1972   /* Notify */
1973   handle->env.sf.s_del (handle->solver, aa, GNUNET_NO);
1974   /* Destroy */
1975   GAS_performance_notify_all_clients (&aa->peer,
1976                                       aa->plugin,
1977                                       aa->addr,
1978                                       aa->addr_len,
1979                                       GNUNET_NO,
1980                                       NULL, 0,
1981                                       zero_bw,
1982                                       zero_bw);
1983   free_address (aa);
1984   return GNUNET_OK;
1985 }
1986
1987
1988 /**
1989  * Remove all addresses
1990  *
1991  * @param handle the address handle to use
1992  */
1993 void
1994 GAS_addresses_destroy_all (struct GAS_Addresses_Handle *handle)
1995 {
1996   if (GNUNET_NO == handle->running)
1997     return;
1998
1999   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
2000               "Destroying all addresses\n");
2001   handle->env.sf.s_bulk_start (handle->solver);
2002   if (NULL != handle->addresses)
2003     GNUNET_CONTAINER_multipeermap_iterate (handle->addresses,
2004                                            &destroy_all_address_it,
2005                                            handle);
2006   handle->env.sf.s_bulk_start (handle->solver);
2007 }
2008
2009
2010 /**
2011  * Shutdown address subsystem.
2012  *
2013  * @param handle the address handle to shutdown
2014  */
2015 void
2016 GAS_addresses_done (struct GAS_Addresses_Handle *handle)
2017 {
2018   struct GAS_Addresses_Suggestion_Requests *cur;
2019   struct GAS_Addresses_Preference_Clients *pcur;
2020
2021   GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Shutting down addresses\n");
2022   GNUNET_assert(NULL != handle);
2023   GAS_addresses_destroy_all (handle);
2024   handle->running = GNUNET_NO;
2025   GNUNET_CONTAINER_multipeermap_destroy (handle->addresses);
2026   handle->addresses = NULL;
2027   while (NULL != (cur = handle->pending_requests_head))
2028   {
2029     GNUNET_CONTAINER_DLL_remove (handle->pending_requests_head,
2030                                  handle->pending_requests_tail,
2031                                  cur);
2032     GNUNET_free(cur);
2033   }
2034
2035   while (NULL != (pcur = handle->preference_clients_head))
2036   {
2037     GNUNET_CONTAINER_DLL_remove (handle->preference_clients_head,
2038                                  handle->preference_clients_tail,
2039                                  pcur);
2040     GNUNET_assert (handle->pref_clients > 0);
2041     handle->pref_clients --;
2042     GNUNET_STATISTICS_set (handle->stat,
2043                            "# active performance clients",
2044                            handle->pref_clients,
2045                            GNUNET_NO);
2046     GNUNET_free (pcur);
2047   }
2048   GNUNET_PLUGIN_unload (handle->plugin,
2049                         handle->solver);
2050   GNUNET_free (handle->plugin);
2051   GNUNET_free(handle);
2052   /* Stop configured solution method */
2053   GAS_normalization_stop ();
2054 }
2055
2056
2057 /**
2058  * Closure for #peerinfo_it().
2059  */
2060 struct PeerInfoIteratorContext
2061 {
2062   /**
2063    * Function to call for each address.
2064    */
2065   GNUNET_ATS_PeerInfo_Iterator it;
2066
2067   /**
2068    * Closure for @e it.
2069    */
2070   void *it_cls;
2071 };
2072
2073
2074 /**
2075  * Iterator to iterate over a peer's addresses
2076  *
2077  * @param cls a `struct PeerInfoIteratorContext`
2078  * @param key the peer id
2079  * @param value the `struct ATS_address`
2080  * @return #GNUNET_OK to continue
2081  */
2082 static int
2083 peerinfo_it (void *cls,
2084              const struct GNUNET_PeerIdentity *key,
2085              void *value)
2086 {
2087   struct PeerInfoIteratorContext *pi_ctx = cls;
2088   struct ATS_Address *addr = value;
2089
2090   pi_ctx->it (pi_ctx->it_cls,
2091               &addr->peer,
2092               addr->plugin,
2093               addr->addr,
2094               addr->addr_len,
2095               addr->active,
2096               addr->atsi, addr->atsi_count,
2097               GNUNET_BANDWIDTH_value_init (addr->assigned_bw_out),
2098               GNUNET_BANDWIDTH_value_init (addr->assigned_bw_in));
2099   return GNUNET_OK;
2100 }
2101
2102
2103 /**
2104  * Return information all peers currently known to ATS
2105  *
2106  * @param handle the address handle to use
2107  * @param peer the respective peer, NULL for 'all' peers
2108  * @param pi_it the iterator to call for every peer
2109  * @param pi_it_cls the closure for @a pi_it
2110  */
2111 void
2112 GAS_addresses_get_peer_info (struct GAS_Addresses_Handle *handle,
2113                              const struct GNUNET_PeerIdentity *peer,
2114                              GNUNET_ATS_PeerInfo_Iterator pi_it,
2115                              void *pi_it_cls)
2116 {
2117   struct PeerInfoIteratorContext pi_ctx;
2118
2119   if (NULL == pi_it)
2120   {
2121     /* does not make sense without callback */
2122     GNUNET_break (0);
2123     return;
2124   }
2125   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2126               "Returning information for %s from a total of %u known addresses\n",
2127               (NULL == peer)
2128               ? "all peers"
2129               : GNUNET_i2s (peer),
2130               (unsigned int) GNUNET_CONTAINER_multipeermap_size (handle->addresses));
2131   pi_ctx.it = pi_it;
2132   pi_ctx.it_cls = pi_it_cls;
2133   if (NULL == peer)
2134     GNUNET_CONTAINER_multipeermap_iterate (handle->addresses,
2135                                            &peerinfo_it,
2136                                            &pi_ctx);
2137   else
2138     GNUNET_CONTAINER_multipeermap_get_multiple (handle->addresses,
2139                                                 peer,
2140                                                 &peerinfo_it, &pi_ctx);
2141   pi_it (pi_it_cls,
2142          NULL, NULL, NULL, 0,
2143          GNUNET_NO,
2144          NULL, 0,
2145          zero_bw,
2146          zero_bw);
2147 }
2148
2149 /* end of gnunet-service-ats_addresses.c */