2 This file is part of GNUnet.
3 (C) 2011 Christian Grothoff (and other contributing authors)
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.
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.
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.
22 * @file ats/gnunet-service-ats_addresses.c
23 * @brief ats service address management
24 * @author Matthias Wachs
25 * @author Christian Grothoff
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"
35 #include "gnunet-service-ats_addresses_mlp.h"
37 #include "gnunet-service-ats_addresses_simplistic.h"
40 * ATS address management
44 * - If you add a new address without a session, a new address will be added
45 * - If you add this address again now with a session a, the existing address
46 * will be updated with this session
47 * - If you add this address again now with a session b, a new address object
48 * with this session will be added
50 * Destroying addresses:
52 * - If you destroy an address without a session, the address itself and all
53 * address instances with an session will be removed
54 * - If you destroy an address with a session, the session for this address
58 * Addresses without a session will be updated with a new session and if the
59 * the session is destroyed the session is removed and address itself still
60 * exists for suggestion
66 * Available ressource assignment modes
73 * Assign each peer an equal amount of bandwidth (bw)
75 * bw_per_peer = bw_total / #active addresses
82 * Solve ressource assignment as an optimization problem
83 * Uses an mixed integer programming solver
89 * Pending Address suggestion requests
91 struct GAS_Addresses_Suggestion_Requests
96 struct GAS_Addresses_Suggestion_Requests *next;
101 struct GAS_Addresses_Suggestion_Requests *prev;
106 struct GNUNET_PeerIdentity id;
110 * Handle for ATS address component
112 struct GAS_Addresses_Handle
117 struct GNUNET_STATISTICS_Handle *stat;
120 * A multihashmap to store all addresses
122 struct GNUNET_CONTAINER_MultiHashMap *addresses;
125 * Configure WAN quota in
127 unsigned long long wan_quota_in;
130 * Configure WAN quota out
132 unsigned long long wan_quota_out;
135 * Is ATS addresses running
140 * Configured ATS solver
150 * Address suggestion requests DLL head
152 struct GAS_Addresses_Suggestion_Requests *r_head;
155 * Address suggestion requests DLL tail
157 struct GAS_Addresses_Suggestion_Requests *r_tail;
159 /* Solver functions */
164 GAS_solver_init s_init;
167 * Add an address to the solver
169 GAS_solver_address_add s_add;
172 * Update address in solver
174 GAS_solver_address_update s_update;
177 * Get address from solver
179 GAS_solver_get_preferred_address s_get;
182 * Delete address in solver
184 GAS_solver_address_delete s_del;
187 * Change preference for quality in solver
189 GAS_solver_address_change_preference s_pref;
194 GAS_solver_done s_done;
199 assemble_ats_information (const struct ATS_Address *aa, struct GNUNET_ATS_Information **dest)
201 unsigned int ats_count = GNUNET_ATS_PropertyCount - 1;
202 struct GNUNET_ATS_Information *ats = GNUNET_malloc (ats_count * sizeof (struct GNUNET_ATS_Information));
205 ats[0].type = ntohl(GNUNET_ATS_UTILIZATION_UP);
206 ats[0].value = aa->atsp_utilization_out.value__;
207 ats[1].type = ntohl(GNUNET_ATS_UTILIZATION_DOWN);
208 ats[1].value = aa->atsp_utilization_in.value__;
209 ats[2].type = ntohl(GNUNET_ATS_NETWORK_TYPE);
210 ats[2].value = ntohl(aa->atsp_network_type);
211 ats[3].type = ntohl(GNUNET_ATS_QUALITY_NET_DELAY);
212 ats[3].value = ntohl(aa->atsp_latency.rel_value);
213 ats[4].type = ntohl(GNUNET_ATS_QUALITY_NET_DISTANCE);
214 ats[4].value = ntohl(aa->atsp_distance);
215 ats[5].type = ntohl(GNUNET_ATS_COST_WAN);
216 ats[5].value = ntohl (aa->atsp_cost_wan);
217 ats[6].type = ntohl(GNUNET_ATS_COST_LAN);
218 ats[6].value = ntohl (aa->atsp_cost_lan);
219 ats[7].type = ntohl(GNUNET_ATS_COST_WLAN);
220 ats[7].value = ntohl (aa->atsp_cost_wlan);
225 disassemble_ats_information (const struct GNUNET_ATS_Information *src,
227 struct ATS_Address *dest)
231 for (i = 0; i < ats_count; i++)
232 switch (ntohl (src[i].type))
234 case GNUNET_ATS_UTILIZATION_UP:
235 dest->atsp_utilization_out.value__ = src[i].value;
238 case GNUNET_ATS_UTILIZATION_DOWN:
239 dest->atsp_utilization_in.value__ = src[i].value;
242 case GNUNET_ATS_QUALITY_NET_DELAY:
243 dest->atsp_latency.rel_value = ntohl (src[i].value);
246 case GNUNET_ATS_QUALITY_NET_DISTANCE:
247 dest->atsp_distance = ntohl (src[i].value);
250 case GNUNET_ATS_COST_WAN:
251 dest->atsp_cost_wan = ntohl (src[i].value);
254 case GNUNET_ATS_COST_LAN:
255 dest->atsp_cost_lan = ntohl (src[i].value);
258 case GNUNET_ATS_COST_WLAN:
259 dest->atsp_cost_wlan = ntohl (src[i].value);
262 case GNUNET_ATS_NETWORK_TYPE:
263 dest->atsp_network_type = ntohl (src[i].value);
266 case GNUNET_ATS_ARRAY_TERMINATOR:
269 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
270 "Received unsupported ATS type %u\n", ntohl (src[i].type));
278 * Free the given address
279 * @param addr address to destroy
282 free_address (struct ATS_Address *addr)
284 GNUNET_free (addr->plugin);
289 * Create a ATS_address with the given information
291 * @param plugin_name plugin
292 * @param plugin_addr address
293 * @param plugin_addr_len address length
294 * @param session_id session
295 * @return the ATS_Address
297 static struct ATS_Address *
298 create_address (const struct GNUNET_PeerIdentity *peer,
299 const char *plugin_name,
300 const void *plugin_addr, size_t plugin_addr_len,
303 struct ATS_Address *aa = NULL;
305 aa = GNUNET_malloc (sizeof (struct ATS_Address) + plugin_addr_len);
307 aa->addr_len = plugin_addr_len;
309 memcpy (&aa[1], plugin_addr, plugin_addr_len);
310 aa->plugin = GNUNET_strdup (plugin_name);
311 aa->session_id = session_id;
312 aa->active = GNUNET_NO;
313 aa->solver_information = NULL;
314 aa->assigned_bw_in = GNUNET_BANDWIDTH_value_init(0);
315 aa->assigned_bw_out = GNUNET_BANDWIDTH_value_init(0);
321 * Destroy the given address.
323 * @param addr address to destroy
324 * @return GNUNET_YES if bandwidth allocations should be recalcualted
327 destroy_address (struct GAS_Addresses_Handle *handle, struct ATS_Address *addr)
332 GNUNET_assert (GNUNET_YES ==
333 GNUNET_CONTAINER_multihashmap_remove (handle->addresses,
334 &addr->peer.hashPubKey,
341 struct CompareAddressContext
343 const struct ATS_Address *search;
345 /* exact_address != NULL if address and session is equal */
346 struct ATS_Address *exact_address;
347 /* exact_address != NULL if address and session is 0 */
348 struct ATS_Address *base_address;
353 compare_address_it (void *cls, const struct GNUNET_HashCode * key, void *value)
355 struct CompareAddressContext *cac = cls;
356 struct ATS_Address *aa = value;
358 /* Find an matching exact address:
361 * aa->addr_len == cac->search->addr_len
362 * aa->plugin == cac->search->plugin
363 * aa->addr == cac->search->addr
364 * aa->session == cac->search->session
366 * return as exact address
368 if ((aa->addr_len == cac->search->addr_len) && (0 == strcmp (aa->plugin, cac->search->plugin)))
370 if ((0 == memcmp (aa->addr, cac->search->addr, aa->addr_len)) && (aa->session_id == cac->search->session_id))
371 cac->exact_address = aa;
374 /* Find an matching base address:
378 * aa->session_id == 0
381 * aa->addr_len == cac->search->addr_len
382 * aa->plugin == cac->search->plugin
383 * aa->addr == cac->search->addr
385 * return as base address
387 if ((aa->addr_len == cac->search->addr_len) && (0 == strcmp (aa->plugin, cac->search->plugin)))
389 if ((0 == memcmp (aa->addr, cac->search->addr, aa->addr_len)) && (aa->session_id == 0))
390 cac->base_address = aa;
393 /* Find an matching exact address based on session:
397 * cac->search->addr_len == 0
400 * aa->plugin == cac->search->plugin
401 * aa->session_id == cac->search->session_id
403 * return as exact address
405 if (0 == cac->search->addr_len)
407 if ((0 == strcmp (aa->plugin, cac->search->plugin)) && (aa->session_id == cac->search->session_id))
408 cac->exact_address = aa;
411 if (cac->exact_address == NULL)
412 return GNUNET_YES; /* Continue iteration to find exact address */
414 return GNUNET_NO; /* Stop iteration since we have an exact address */
419 * Find an existing equivalent address record.
420 * Compares by peer identity and network address OR by session ID
421 * (one of the two must match).
423 * @param peer peer to lookup addresses for
424 * @param addr existing address record
425 * @return existing address record, NULL for none
428 find_equivalent_address (struct GAS_Addresses_Handle *handle,
429 const struct GNUNET_PeerIdentity *peer,
430 const struct ATS_Address *addr)
432 struct CompareAddressContext cac;
434 cac.exact_address = NULL;
435 cac.base_address = NULL;
437 GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses, &peer->hashPubKey,
438 &compare_address_it, &cac);
440 if (cac.exact_address == NULL)
441 return cac.base_address;
442 return cac.exact_address;
446 static struct ATS_Address *
447 lookup_address (struct GAS_Addresses_Handle *handle,
448 const struct GNUNET_PeerIdentity *peer,
449 const char *plugin_name,
450 const void *plugin_addr,
451 size_t plugin_addr_len,
453 const struct GNUNET_ATS_Information *atsi,
456 struct ATS_Address *aa;
457 struct ATS_Address *ea;
459 aa = create_address (peer,
461 plugin_addr, plugin_addr_len,
464 /* Get existing address or address with session == 0 */
465 ea = find_equivalent_address (handle, peer, aa);
471 else if (ea->session_id != session_id)
480 GAS_addresses_add (struct GAS_Addresses_Handle *handle,
481 const struct GNUNET_PeerIdentity *peer,
482 const char *plugin_name, const void *plugin_addr,
483 size_t plugin_addr_len, uint32_t session_id,
484 const struct GNUNET_ATS_Information *atsi,
487 struct ATS_Address *aa;
488 struct ATS_Address *ea;
489 unsigned int ats_res;
491 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
492 "Received `%s' for peer `%s'\n",
496 if (GNUNET_NO == handle->running)
499 GNUNET_assert (NULL != handle->addresses);
501 aa = create_address (peer, plugin_name, plugin_addr, plugin_addr_len,
503 if (atsi_count != (ats_res = disassemble_ats_information(atsi, atsi_count, aa)))
505 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
506 "While adding address: had %u ATS elements to add, could only add %u\n",
507 atsi_count, ats_res);
510 /* Get existing address or address with session == 0 */
511 ea = find_equivalent_address (handle, peer, aa);
514 /* We have a new address */
515 GNUNET_assert (GNUNET_OK ==
516 GNUNET_CONTAINER_multihashmap_put (handle->addresses,
517 &peer->hashPubKey, aa,
518 GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
519 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Added new address for peer `%s' session id %u, %p\n",
520 GNUNET_i2s (peer), session_id, aa);
521 /* Tell solver about new address */
522 handle->s_add (handle->solver, handle->addresses, aa);
528 GNUNET_free (aa->plugin);
531 if (ea->session_id != 0)
533 /* This addresswith the same session is already existing */
534 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
535 "Added already existing address for peer `%s' `%s' %p with new session %u\n",
536 GNUNET_i2s (peer), plugin_name, session_id);
541 /* We have an address without an session, update this address */
542 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
543 "Updated existing address for peer `%s' %p with new session %u\n",
544 GNUNET_i2s (peer), ea, session_id);
545 ea->session_id = session_id;
546 if (atsi_count != (ats_res = disassemble_ats_information(atsi, atsi_count, ea)))
548 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
549 "While updating address: had %u ATS elements to add, could only add %u\n",
550 atsi_count, ats_res);
553 handle->s_update (handle->solver, handle->addresses, ea);
558 GAS_addresses_update (struct GAS_Addresses_Handle *handle,
559 const struct GNUNET_PeerIdentity *peer,
560 const char *plugin_name, const void *plugin_addr,
561 size_t plugin_addr_len, uint32_t session_id,
562 const struct GNUNET_ATS_Information *atsi,
565 struct ATS_Address *aa;
568 if (GNUNET_NO == handle->running)
571 GNUNET_assert (NULL != handle->addresses);
573 /* Get existing address */
574 aa = lookup_address (handle, peer, plugin_name, plugin_addr, plugin_addr_len,
575 session_id, atsi, atsi_count);
578 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Tried to update unknown address for peer `%s' `%s' session id %u\n",
579 GNUNET_i2s (peer), plugin_name, session_id);
584 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
585 "Received `%s' for peer `%s' address \n",
587 GNUNET_i2s (peer), aa);
589 if (atsi_count != (ats_res = disassemble_ats_information (atsi, atsi_count, aa)))
591 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
592 "While adding address: had %u ATS elements to add, could only add %u\n",
593 atsi_count, ats_res);
595 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
596 "Updated %u ATS elements for address %p\n",
599 /* Tell solver about update */
600 handle->s_update (handle->solver, handle->addresses, aa);
604 struct DestroyContext
606 struct ATS_Address *aa;
608 struct GAS_Addresses_Handle *handle;
611 * GNUNET_NO : full address
612 * GNUNET_YES : just session
621 * If session != 0, just the session is deleted, the address itself still exists
622 * If session == 0, remove full address
623 * If session == 0 and addrlen == 0, destroy inbound address
627 * @param value the 'struct ATS_Address'
628 * @return GNUNET_OK (continue to iterate)
631 destroy_by_session_id (void *cls, const struct GNUNET_HashCode * key, void *value)
633 struct DestroyContext *dc = cls;
634 struct GAS_Addresses_Handle *handle = dc->handle;
635 const struct ATS_Address *des = dc->aa;
636 struct ATS_Address *aa = value;
638 GNUNET_assert (0 == memcmp (&aa->peer, &des->peer,
639 sizeof (struct GNUNET_PeerIdentity)));
642 if (des->session_id == 0)
644 /* Session == 0, remove full address */
645 if ((0 == strcmp (des->plugin, aa->plugin)) &&
646 (aa->addr_len == des->addr_len) &&
647 (0 == memcmp (des->addr, aa->addr, aa->addr_len)))
650 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
651 "Deleting full address for peer `%s' session %u %p\n",
652 GNUNET_i2s (&aa->peer), aa->session_id, aa);
654 /* Notify solver about deletion */
655 handle->s_del (handle->solver, handle->addresses, aa, GNUNET_NO);
656 destroy_address (handle, aa);
657 dc->result = GNUNET_NO;
658 return GNUNET_OK; /* Continue iteration */
663 /* Session != 0, just remove session */
664 if (aa->session_id != des->session_id)
665 return GNUNET_OK; /* irrelevant */
667 if ((aa->session_id != 0) &&
668 (0 != strcmp (des->plugin, aa->plugin)))
670 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
671 "Different plugins during removal: `%s' vs `%s' \n",
672 des->plugin, aa->plugin);
677 if (aa->addr_len == 0)
679 /* Inbound connection died, delete full address */
680 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
681 "Deleting inbound address for peer `%s': `%s' session %u\n",
682 GNUNET_i2s (&aa->peer), aa->plugin, aa->session_id);
684 /* Notify solver about deletion */
685 handle->s_del (handle->solver, handle->addresses, aa, GNUNET_NO);
686 destroy_address (handle, aa);
687 dc->result = GNUNET_NO;
688 return GNUNET_OK; /* Continue iteration */
693 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
694 "Deleting session for peer `%s': `%s' %u\n",
695 GNUNET_i2s (&aa->peer), aa->plugin, aa->session_id);
696 /* Notify solver to delete session */
697 handle->s_del (handle->solver, handle->addresses, aa, GNUNET_YES);
706 GAS_addresses_destroy (struct GAS_Addresses_Handle *handle,
707 const struct GNUNET_PeerIdentity *peer,
708 const char *plugin_name, const void *plugin_addr,
709 size_t plugin_addr_len, uint32_t session_id)
711 struct ATS_Address *ea;
712 struct DestroyContext dc;
714 if (GNUNET_NO == handle->running)
717 /* Get existing address */
718 ea = lookup_address (handle, peer, plugin_name, plugin_addr, plugin_addr_len,
719 session_id, NULL, 0);
721 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
722 "Received `%s' for peer `%s' address %p session %u\n",
724 GNUNET_i2s (peer), ea, session_id);
728 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Tried to destroy unknown address for peer `%s' `%s' session id %u\n",
729 GNUNET_i2s (peer), plugin_name, session_id);
733 GNUNET_break (0 < strlen (plugin_name));
735 dc.aa = create_address (peer, plugin_name, plugin_addr, plugin_addr_len, session_id);
737 GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses, &peer->hashPubKey,
738 &destroy_by_session_id, &dc);
739 free_address (dc.aa);
744 GAS_addresses_in_use (struct GAS_Addresses_Handle *handle,
745 const struct GNUNET_PeerIdentity *peer,
746 const char *plugin_name, const void *plugin_addr,
747 size_t plugin_addr_len, uint32_t session_id, int in_use)
749 struct ATS_Address *ea;
751 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
752 "Received `%s' for peer `%s'\n",
756 if (GNUNET_NO == handle->running)
757 return GNUNET_SYSERR;
759 ea = lookup_address (handle, peer, plugin_name,
760 plugin_addr, plugin_addr_len,
761 session_id, NULL, 0);
764 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
765 "Trying to set unknown address `%s', %s %u %s \n",
767 plugin_name, session_id,
768 (GNUNET_NO == in_use) ? "NO" : "YES");
770 return GNUNET_SYSERR;
772 if (ea->used == in_use)
775 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
776 "Address in use called multiple times for peer `%s': %s -> %s \n",
778 (GNUNET_NO == ea->used) ? "NO" : "YES",
779 (GNUNET_NO == in_use) ? "NO" : "YES");
780 return GNUNET_SYSERR;
784 /* Tell solver about update */
785 handle->s_update (handle->solver, handle->addresses, ea);
792 * Cancel address suggestions for a peer
794 * @param peer the respective peer
797 GAS_addresses_request_address_cancel (struct GAS_Addresses_Handle *handle,
798 const struct GNUNET_PeerIdentity *peer)
800 struct GAS_Addresses_Suggestion_Requests *cur = handle->r_head;
802 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
803 "Received request: `%s' for peer %s\n", "request_address_cancel", GNUNET_i2s (peer));
807 if (0 == memcmp (peer, &cur->id, sizeof (cur->id)))
814 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
815 "No address requests pending for peer `%s', cannot remove!\n", GNUNET_i2s (peer));
818 GAS_addresses_handle_backoff_reset (handle, peer);
819 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
820 "Removed request pending for peer `%s\n", GNUNET_i2s (peer));
821 GNUNET_CONTAINER_DLL_remove (handle->r_head, handle->r_tail, cur);
827 * Add an address suggestions for a peer
829 * @param peer the respective peer
832 GAS_addresses_request_address (struct GAS_Addresses_Handle *handle,
833 const struct GNUNET_PeerIdentity *peer)
835 struct GAS_Addresses_Suggestion_Requests *cur = handle->r_head;
836 struct ATS_Address *aa;
837 struct GNUNET_ATS_Information *ats;
838 unsigned int ats_count;
840 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
841 "Received `%s' for peer `%s'\n",
845 if (GNUNET_NO == handle->running)
849 if (0 == memcmp (peer, &cur->id, sizeof (cur->id)))
850 break; /* already suggesting */
855 cur = GNUNET_malloc (sizeof (struct GAS_Addresses_Suggestion_Requests));
857 GNUNET_CONTAINER_DLL_insert (handle->r_head, handle->r_tail, cur);
860 /* Get prefered address from solver */
861 aa = (struct ATS_Address *) handle->s_get (handle->solver, handle->addresses, peer);
864 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
865 "Cannot suggest address for peer `%s'\n", GNUNET_i2s (peer));
869 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
870 "Suggesting address %p for peer `%s'\n", aa, GNUNET_i2s (peer));
872 ats_count = assemble_ats_information (aa, &ats);
873 GAS_scheduling_transmit_address_suggestion (peer,
875 aa->addr, aa->addr_len,
881 aa->block_interval = GNUNET_TIME_relative_add (aa->block_interval, ATS_BLOCKING_DELTA);
882 aa->blocked_until = GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get(), aa->block_interval);
884 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
885 "Address %p ready for suggestion, block interval now %llu \n",
886 aa, aa->block_interval);
893 reset_address_it (void *cls, const struct GNUNET_HashCode * key, void *value)
895 struct ATS_Address *aa = value;
897 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
898 "Resetting interval for peer `%s' address %p from %llu to 0\n",
899 GNUNET_i2s (&aa->peer), aa, aa->block_interval);
901 aa->blocked_until = GNUNET_TIME_UNIT_ZERO_ABS;
902 aa->block_interval = GNUNET_TIME_UNIT_ZERO;
908 GAS_addresses_handle_backoff_reset (struct GAS_Addresses_Handle *handle,
909 const struct GNUNET_PeerIdentity *peer)
911 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
912 "Received `%s' for peer `%s'\n",
916 GNUNET_break (GNUNET_SYSERR != GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses,
924 GAS_addresses_change_preference (struct GAS_Addresses_Handle *handle,
925 const struct GNUNET_PeerIdentity *peer,
926 enum GNUNET_ATS_PreferenceKind kind,
929 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
930 "Received `%s' for peer `%s'\n",
934 if (GNUNET_NO == handle->running)
937 /* Tell solver about update */
938 handle->s_pref (handle->solver, peer, kind, score);
942 load_quotas (const struct GNUNET_CONFIGURATION_Handle *cfg, unsigned long long *out_dest, unsigned long long *in_dest, int dest_length)
944 int quotas[GNUNET_ATS_NetworkTypeCount] = GNUNET_ATS_NetworkType;
945 char * entry_in = NULL;
946 char * entry_out = NULL;
947 char * quota_out_str;
951 for (c = 0; (c < GNUNET_ATS_NetworkTypeCount) && (c < dest_length); c++)
956 case GNUNET_ATS_NET_UNSPECIFIED:
957 entry_out = "UNSPECIFIED_QUOTA_OUT";
958 entry_in = "UNSPECIFIED_QUOTA_IN";
960 case GNUNET_ATS_NET_LOOPBACK:
961 entry_out = "LOOPBACK_QUOTA_OUT";
962 entry_in = "LOOPBACK_QUOTA_IN";
964 case GNUNET_ATS_NET_LAN:
965 entry_out = "LAN_QUOTA_OUT";
966 entry_in = "LAN_QUOTA_IN";
968 case GNUNET_ATS_NET_WAN:
969 entry_out = "WAN_QUOTA_OUT";
970 entry_in = "WAN_QUOTA_IN";
972 case GNUNET_ATS_NET_WLAN:
973 entry_out = "WLAN_QUOTA_OUT";
974 entry_in = "WLAN_QUOTA_IN";
980 if ((entry_in == NULL) || (entry_out == NULL))
984 if (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string(cfg, "ats", entry_out, "a_out_str))
986 if (0 == strcmp(quota_out_str, BIG_M_STRING) ||
987 (GNUNET_SYSERR == GNUNET_STRINGS_fancy_size_to_bytes (quota_out_str, &out_dest[c])))
988 out_dest[c] = UINT32_MAX;
990 GNUNET_free (quota_out_str);
991 quota_out_str = NULL;
993 else if (GNUNET_ATS_NET_UNSPECIFIED == quotas[c])
994 out_dest[c] = UINT32_MAX;
996 out_dest[c] = UINT32_MAX;
999 if (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string(cfg, "ats", entry_in, "a_in_str))
1001 if (0 == strcmp(quota_in_str, BIG_M_STRING) ||
1002 (GNUNET_SYSERR == GNUNET_STRINGS_fancy_size_to_bytes (quota_in_str, &in_dest[c])))
1003 in_dest[c] = UINT32_MAX;
1005 GNUNET_free (quota_in_str);
1006 quota_in_str = NULL;
1008 else if (GNUNET_ATS_NET_UNSPECIFIED == quotas[c])
1010 in_dest[c] = UINT32_MAX;
1014 in_dest[c] = UINT32_MAX;
1016 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Loaded quota: %s %u, %s %u\n", entry_in, in_dest[c], entry_out, out_dest[c]);
1019 return GNUNET_ATS_NetworkTypeCount;
1024 bandwidth_changed_cb (void *cls, struct ATS_Address *address)
1026 struct GAS_Addresses_Handle *handle = cls;
1027 struct GAS_Addresses_Suggestion_Requests *cur;
1029 GNUNET_assert (handle != NULL);
1030 GNUNET_assert (address != NULL);
1032 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Bandwidth assignment changed for peer %s \n", GNUNET_i2s(&address->peer));
1033 struct GNUNET_ATS_Information *ats;
1034 unsigned int ats_count;
1036 cur = handle->r_head;
1039 if (0 == memcmp (&address->peer, &cur->id, sizeof (cur->id)))
1040 break; /* we have an address request pending*/
1045 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1046 "Nobody is interested in peer `%s' :(\n",GNUNET_i2s (&address->peer));
1050 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1051 "Sending bandwidth update for peer `%s'\n",GNUNET_i2s (&address->peer));
1053 ats_count = assemble_ats_information (address, &ats);
1054 GAS_scheduling_transmit_address_suggestion (&address->peer,
1056 address->addr, address->addr_len,
1057 address->session_id,
1059 address->assigned_bw_out,
1060 address->assigned_bw_in);
1066 * Initialize address subsystem.
1068 * @param cfg configuration to use
1069 * @param stats the statistics handle to use
1071 struct GAS_Addresses_Handle *
1072 GAS_addresses_init (const struct GNUNET_CONFIGURATION_Handle *cfg,
1073 const struct GNUNET_STATISTICS_Handle *stats)
1075 struct GAS_Addresses_Handle *ah;
1076 int quotas[GNUNET_ATS_NetworkTypeCount] = GNUNET_ATS_NetworkType;
1077 unsigned long long quotas_in[GNUNET_ATS_NetworkTypeCount];
1078 unsigned long long quotas_out[GNUNET_ATS_NetworkTypeCount];
1083 ah = GNUNET_malloc (sizeof (struct GAS_Addresses_Handle));
1084 ah->running = GNUNET_NO;
1086 ah->stat = (struct GNUNET_STATISTICS_Handle *) stats;
1087 /* Initialize the addresses database */
1088 ah->addresses = GNUNET_CONTAINER_multihashmap_create (128, GNUNET_NO);
1089 GNUNET_assert (NULL != ah->addresses);
1091 /* Figure out configured solution method */
1092 if (GNUNET_SYSERR == GNUNET_CONFIGURATION_get_value_string (cfg, "ats", "MODE", &mode_str))
1094 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "No ressource assignment method configured, using simplistic approch\n");
1095 ah->ats_mode = MODE_SIMPLISTIC;
1099 for (c = 0; c < strlen (mode_str); c++)
1100 mode_str[c] = toupper (mode_str[c]);
1101 if (0 == strcmp (mode_str, "SIMPLISTIC"))
1103 ah->ats_mode = MODE_SIMPLISTIC;
1105 else if (0 == strcmp (mode_str, "MLP"))
1107 ah->ats_mode = MODE_MLP;
1109 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Assignment method `%s' configured, but GLPK is not availabe, please install \n", mode_str);
1110 ah->ats_mode = MODE_SIMPLISTIC;
1115 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Invalid ressource assignment method `%s' configured, using simplistic approch\n", mode_str);
1116 ah->ats_mode = MODE_SIMPLISTIC;
1118 GNUNET_free (mode_str);
1120 /* Start configured solution method */
1121 switch (ah->ats_mode)
1124 /* Init the MLP solver with default values */
1126 ah->ats_mode = MODE_MLP;
1127 ah->s_init = &GAS_mlp_init;
1128 ah->s_add = &GAS_mlp_address_add;
1129 ah->s_update = &GAS_mlp_address_update;
1130 ah->s_get = &GAS_mlp_get_preferred_address;
1131 ah->s_pref = &GAS_mlp_address_change_preference;
1132 ah->s_del = &GAS_mlp_address_delete;
1133 ah->s_done = &GAS_mlp_done;
1139 case MODE_SIMPLISTIC:
1140 /* Init the simplistic solver with default values */
1141 ah->ats_mode = MODE_SIMPLISTIC;
1142 ah->s_init = &GAS_simplistic_init;
1143 ah->s_add = &GAS_simplistic_address_add;
1144 ah->s_update = &GAS_simplistic_address_update;
1145 ah->s_get = &GAS_simplistic_get_preferred_address;
1146 ah->s_pref = &GAS_simplistic_address_change_preference;
1147 ah->s_del = &GAS_simplistic_address_delete;
1148 ah->s_done = &GAS_simplistic_done;
1149 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ATS started in %s mode\n", "SIMPLISTIC");
1156 GNUNET_assert (NULL != ah->s_init);
1157 GNUNET_assert (NULL != ah->s_add);
1158 GNUNET_assert (NULL != ah->s_update);
1159 GNUNET_assert (NULL != ah->s_get);
1160 GNUNET_assert (NULL != ah->s_pref);
1161 GNUNET_assert (NULL != ah->s_del);
1162 GNUNET_assert (NULL != ah->s_done);
1164 quota_count = load_quotas(cfg, quotas_in, quotas_out, GNUNET_ATS_NetworkTypeCount);
1166 ah->solver = ah->s_init (cfg, stats, quotas, quotas_in, quotas_out, quota_count, &bandwidth_changed_cb, ah);
1167 if (NULL == ah->solver)
1169 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to initialize solver!\n");
1174 /* up and running */
1175 ah->running = GNUNET_YES;
1181 * Free memory of address.
1184 * @param key peer identity (unused)
1185 * @param value the 'struct ATS_Address' to free
1186 * @return GNUNET_OK (continue to iterate)
1189 free_address_it (void *cls, const struct GNUNET_HashCode * key, void *value)
1191 struct GAS_Addresses_Handle *handle = cls;
1192 struct ATS_Address *aa = value;
1193 handle->s_del (handle->solver, handle->addresses, aa, GNUNET_NO);
1194 destroy_address (handle, aa);
1200 GAS_addresses_destroy_all (struct GAS_Addresses_Handle *handle)
1202 if (GNUNET_NO == handle->running)
1205 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1209 if (handle->addresses != NULL)
1210 GNUNET_CONTAINER_multihashmap_iterate (handle->addresses, &free_address_it, handle);
1215 * Shutdown address subsystem.
1218 GAS_addresses_done (struct GAS_Addresses_Handle *handle)
1220 struct GAS_Addresses_Suggestion_Requests *cur;
1222 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1223 "Shutting down addresses\n");
1224 GNUNET_assert (NULL != handle);
1225 GAS_addresses_destroy_all (handle);
1226 handle->running = GNUNET_NO;
1227 GNUNET_CONTAINER_multihashmap_destroy (handle->addresses);
1228 handle->addresses = NULL;
1229 while (NULL != (cur = handle->r_head))
1231 GNUNET_CONTAINER_DLL_remove (handle->r_head, handle->r_tail, cur);
1234 handle->s_done (handle->solver);
1235 GNUNET_free (handle);
1236 /* Stop configured solution method */
1240 struct PeerIteratorContext
1242 GNUNET_ATS_Peer_Iterator it;
1244 struct GNUNET_CONTAINER_MultiHashMap *peers_returned;
1249 const struct GNUNET_HashCode * key,
1252 struct PeerIteratorContext *ip_ctx = cls;
1253 struct GNUNET_PeerIdentity tmp;
1255 if (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains(ip_ctx->peers_returned, key))
1257 GNUNET_CONTAINER_multihashmap_put(ip_ctx->peers_returned, key, NULL, GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1258 tmp.hashPubKey = (*key);
1259 ip_ctx->it (ip_ctx->it_cls, &tmp);
1266 * Return all peers currently known to ATS
1268 * @param p_it the iterator to call for every peer, callbach with id == NULL
1270 * @param p_it_cls the closure for the iterator
1273 GAS_addresses_iterate_peers (struct GAS_Addresses_Handle *handle, GNUNET_ATS_Peer_Iterator p_it, void *p_it_cls)
1275 struct PeerIteratorContext ip_ctx;
1280 GNUNET_assert (NULL != handle->addresses);
1282 size = GNUNET_CONTAINER_multihashmap_size(handle->addresses);
1286 ip_ctx.it_cls = p_it_cls;
1287 ip_ctx.peers_returned = GNUNET_CONTAINER_multihashmap_create (size, GNUNET_NO);
1288 GNUNET_CONTAINER_multihashmap_iterate (handle->addresses, &peer_it, &ip_ctx);
1289 GNUNET_CONTAINER_multihashmap_destroy (ip_ctx.peers_returned);
1291 p_it (p_it_cls, NULL);
1294 struct PeerInfoIteratorContext
1296 GNUNET_ATS_PeerInfo_Iterator it;
1302 peerinfo_it (void *cls,
1303 const struct GNUNET_HashCode * key,
1306 struct PeerInfoIteratorContext *pi_ctx = cls;
1307 struct ATS_Address *addr = (struct ATS_Address *) value;
1308 struct GNUNET_ATS_Information *ats;
1311 if (NULL != pi_ctx->it)
1313 ats_count = assemble_ats_information (addr, &ats);
1315 pi_ctx->it (pi_ctx->it_cls,
1318 addr->addr, addr->addr_len,
1321 addr->assigned_bw_out,
1322 addr->assigned_bw_in);
1330 * Return all peers currently known to ATS
1332 * @param peer the respective peer
1333 * @param pi_it the iterator to call for every peer
1334 * @param pi_it_cls the closure for the iterator
1337 GAS_addresses_get_peer_info (struct GAS_Addresses_Handle *handle,
1338 const struct GNUNET_PeerIdentity *peer,
1339 GNUNET_ATS_PeerInfo_Iterator pi_it,
1342 struct PeerInfoIteratorContext pi_ctx;
1343 struct GNUNET_BANDWIDTH_Value32NBO zero_bw;
1344 GNUNET_assert (NULL != peer);
1345 GNUNET_assert (NULL != handle->addresses);
1347 return; /* does not make sense without callback */
1349 zero_bw = GNUNET_BANDWIDTH_value_init (0);
1351 pi_ctx.it_cls = pi_it_cls;
1353 GNUNET_CONTAINER_multihashmap_get_multiple (handle->addresses, &peer->hashPubKey, &peerinfo_it, &pi_ctx);
1356 pi_it (pi_it_cls, NULL, NULL, NULL, 0, GNUNET_NO, NULL, 0, zero_bw, zero_bw);
1361 /* end of gnunet-service-ats_addresses.c */