examples/c/dht-kademlia/routing_table.c

   1 /* Copyright (c) 2012-2020. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "routing_table.h"
   8 #include "node.h"
   9
  10 XBT_LOG_NEW_DEFAULT_CATEGORY(dht_kademlia_routing_table, "Messages specific for this example");
  11
  12 /** @brief Initialization of a node routing table.  */
  13 routing_table_t routing_table_init(unsigned int node_id)
  14 {
  15   routing_table_t table = xbt_new(s_routing_table_t, 1);
  16   table->buckets        = xbt_new(s_bucket_t, IDENTIFIER_SIZE + 1);
  17   for (unsigned int i = 0; i < IDENTIFIER_SIZE + 1; i++) {
  18     table->buckets[i].nodes = xbt_dynar_new(sizeof(unsigned int), NULL);
  19     table->buckets[i].id    = i;
  20   }
  21   table->id = node_id;
  22   return table;
  23 }
  24
  25 /** @brief Frees the routing table */
  26 void routing_table_free(routing_table_t table)
  27 {
  28   unsigned int i;
  29   // Free the buckets.
  30   for (i = 0; i <= IDENTIFIER_SIZE; i++) {
  31     xbt_dynar_free(&table->buckets[i].nodes);
  32   }
  33   xbt_free(table->buckets);
  34   xbt_free(table);
  35 }
  36
  37 /**@brief prints the routing table, to debug stuff. */
  38 void routing_table_print(const_routing_table_t table)
  39 {
  40   unsigned int j;
  41   unsigned int value;
  42   XBT_INFO("Routing table of %08x:", table->id);
  43
  44   for (unsigned int i = 0; i <= IDENTIFIER_SIZE; i++) {
  45     if (!xbt_dynar_is_empty(table->buckets[i].nodes)) {
  46       XBT_INFO("Bucket number %u: ", i);
  47       xbt_dynar_foreach (table->buckets[i].nodes, j, value) {
  48         XBT_INFO("Element %u: %08x", j, value);
  49       }
  50     }
  51   }
  52 }
  53
  54 /** @brief Finds an identifier in a bucket and returns its position or returns -1 if it doesn't exists
  55  * @param bucket the bucket in which we try to find our identifier
  56  * @param id the id
  57  */
  58 unsigned int bucket_find_id(const_bucket_t bucket, unsigned int id)
  59 {
  60   unsigned int i;
  61   unsigned int current_id;
  62   xbt_dynar_foreach (bucket->nodes, i, current_id) {
  63     if (id == current_id) {
  64       return i;
  65     }
  66   }
  67   return -1;
  68 }
  69
  70 /** @brief Finds the corresponding bucket in a routing table for a given identifier
  71  * @param table the routing table
  72  * @param id the identifier
  73  * @return the bucket in which the the identifier would be.
  74  */
  75 bucket_t routing_table_find_bucket(const_routing_table_t table, unsigned int id)
  76 {
  77   unsigned int xor_number = table->id ^ id;
  78   unsigned int prefix     = get_node_prefix(xor_number, IDENTIFIER_SIZE);
  79   xbt_assert(prefix <= IDENTIFIER_SIZE, "Tried to return a  bucket that doesn't exist.");
  80   bucket_t bucket = &table->buckets[prefix];
  81   return bucket;
  82 }