1 /* Copyright (c) 2012-2017. 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-tracker.hpp"
9 #include <xbt/RngStream.h>
11 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_bt_tracker, "Messages specific for the tracker");
13 Tracker::Tracker(std::vector<std::string> args)
16 xbt_assert(args.size() == 2, "Wrong number of arguments for the tracker.");
17 // Retrieving end time
19 deadline = std::stod(args[1]);
20 } catch (std::invalid_argument& ia) {
21 throw std::invalid_argument(std::string("Invalid deadline:") + args[1].c_str());
23 xbt_assert(deadline > 0, "Wrong deadline supplied");
25 stream = simgrid::s4u::this_actor::getHost()->extension<HostBittorrent>()->getStream();
27 mailbox = simgrid::s4u::Mailbox::byName(TRACKER_MAILBOX);
29 XBT_INFO("Tracker launched.");
32 void Tracker::operator()()
34 simgrid::s4u::CommPtr comm = nullptr;
36 while (simgrid::s4u::Engine::getClock() < deadline) {
38 comm = mailbox->get_async(&received);
40 // Retrieve the data sent by the peer.
41 TrackerQuery* tq = static_cast<TrackerQuery*>(received);
43 // Add the peer to our peer list, if not already known.
44 if (known_peers.find(tq->getPeerId()) == known_peers.end()) {
45 known_peers.insert(tq->getPeerId());
48 // Sending back peers to the requesting peer
49 TrackerAnswer* ta = new TrackerAnswer(TRACKER_QUERY_INTERVAL);
50 std::set<int>::iterator next_peer;
51 int nb_known_peers = known_peers.size();
52 int max_tries = std::min(MAXIMUM_PEERS, nb_known_peers);
54 while (tried < max_tries) {
56 next_peer = known_peers.begin();
57 std::advance(next_peer, RngStream_RandInt(stream, 0, nb_known_peers - 1));
58 } while (ta->getPeers()->find(*next_peer) != ta->getPeers()->end());
59 ta->addPeer(*next_peer);
62 tq->getReturnMailbox()->put_init(ta, TRACKER_COMM_SIZE)->detach();
67 simgrid::s4u::this_actor::sleep_for(1);
70 XBT_INFO("Tracker is leaving");