1 /* Copyright (c) 2012-2020. The SimGrid Team.
2 * All rights reserved. */
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. */
7 #include "s4u-dht-kademlia.hpp"
11 #include "simgrid/s4u.hpp"
13 XBT_LOG_NEW_DEFAULT_CATEGORY(kademlia, "Messages specific for this example");
15 /** @brief Node function
17 * @param the ID of the person I know in the system (or not)
18 * @param Time before I leave the system because I'm bored
20 static void node(std::vector<std::string> args)
22 bool join_success = true;
24 xbt_assert(args.size() == 3 || args.size() == 4, "Wrong number of arguments");
25 /* Node initialization */
26 auto node_id = static_cast<unsigned int>(std::stoul(args[1], 0, 0));
27 kademlia::Node node(node_id);
29 if (args.size() == 4) {
30 XBT_INFO("Hi, I'm going to join the network with id %u", node.getId());
31 auto known_id = static_cast<unsigned int>(std::stoul(args[2], 0, 0));
32 join_success = node.join(known_id);
33 deadline = std::stod(args[3]);
35 deadline = std::stod(args[2]);
36 XBT_INFO("Hi, I'm going to create the network with id %u", node.getId());
37 node.routingTableUpdate(node.getId());
41 XBT_VERB("Ok, I'm joining the network with id %u", node.getId());
42 // We start the main loop
43 double next_lookup_time = simgrid::s4u::Engine::get_clock() + RANDOM_LOOKUP_INTERVAL;
45 XBT_VERB("Main loop start");
47 simgrid::s4u::Mailbox* mailbox = simgrid::s4u::Mailbox::by_name(std::to_string(node.getId()));
49 while (simgrid::s4u::Engine::get_clock() < deadline) {
50 if (node.receive_comm == nullptr)
51 node.receive_comm = mailbox->get_async<kademlia::Message>(&node.received_msg);
53 if (node.receive_comm->test()) {
54 // There has been a message, we need to handle it !
55 if (node.received_msg) {
56 node.handleFindNode(node.received_msg);
57 delete node.received_msg;
58 node.receive_comm = nullptr;
60 simgrid::s4u::this_actor::sleep_for(1);
62 /* We search for a pseudo random node */
63 if (simgrid::s4u::Engine::get_clock() >= next_lookup_time) {
65 next_lookup_time += RANDOM_LOOKUP_INTERVAL;
67 // Didn't get a message: sleep for a while...
68 simgrid::s4u::this_actor::sleep_for(1);
73 XBT_INFO("I couldn't join the network :(");
75 XBT_DEBUG("I'm leaving the network");
76 node.displaySuccessRate();
79 /** @brief Main function */
80 int main(int argc, char* argv[])
82 simgrid::s4u::Engine e(&argc, argv);
84 /* Check the arguments */
86 "Usage: %s platform_file deployment_file\n\tExample: %s cluster_backbone.xml dht-kademlia_d.xml\n",
89 e.load_platform(argv[1]);
90 e.register_function("node", node);
91 e.load_deployment(argv[2]);
95 XBT_INFO("Simulated time: %g", simgrid::s4u::Engine::get_clock());