-/* Copyright (c) 2012-2018. The SimGrid Team.
+/* Copyright (c) 2012-2021. The SimGrid Team.
* All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
#include "s4u-tracker.hpp"
#include <algorithm>
-#include <xbt/RngStream.h>
XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_bt_tracker, "Messages specific for the tracker");
// Retrieving end time
try {
deadline = std::stod(args[1]);
- } catch (std::invalid_argument& ia) {
- throw std::invalid_argument(std::string("Invalid deadline:") + args[1].c_str());
+ } catch (const std::invalid_argument&) {
+ throw std::invalid_argument("Invalid deadline:" + args[1]);
}
xbt_assert(deadline > 0, "Wrong deadline supplied");
- stream = simgrid::s4u::this_actor::get_host()->extension<HostBittorrent>()->getStream();
-
mailbox = simgrid::s4u::Mailbox::by_name(TRACKER_MAILBOX);
XBT_INFO("Tracker launched.");
void Tracker::operator()()
{
simgrid::s4u::CommPtr comm = nullptr;
- void* received = nullptr;
+ TrackerQuery* query = nullptr;
while (simgrid::s4u::Engine::get_clock() < deadline) {
if (comm == nullptr)
- comm = mailbox->get_async(&received);
+ comm = mailbox->get_async<TrackerQuery>(&query);
if (comm->test()) {
// Retrieve the data sent by the peer.
- xbt_assert(received != nullptr);
- TrackerQuery* tq = static_cast<TrackerQuery*>(received);
+ xbt_assert(query != nullptr);
// Add the peer to our peer list, if not already known.
- if (known_peers.find(tq->getPeerId()) == known_peers.end()) {
- known_peers.insert(tq->getPeerId());
- }
+ known_peers.emplace(query->getPeerId());
// Sending back peers to the requesting peer
- TrackerAnswer* ta = new TrackerAnswer(TRACKER_QUERY_INTERVAL);
+ auto* answer = new TrackerAnswer(TRACKER_QUERY_INTERVAL);
std::set<int>::iterator next_peer;
- int nb_known_peers = known_peers.size();
+ int nb_known_peers = static_cast<int>(known_peers.size());
int max_tries = std::min(MAXIMUM_PEERS, nb_known_peers);
int tried = 0;
while (tried < max_tries) {
do {
next_peer = known_peers.begin();
- std::advance(next_peer, RngStream_RandInt(stream, 0, nb_known_peers - 1));
- } while (ta->getPeers()->find(*next_peer) != ta->getPeers()->end());
- ta->addPeer(*next_peer);
+ std::advance(next_peer, random.uniform_int(0, nb_known_peers - 1));
+ } while (answer->getPeers().find(*next_peer) != answer->getPeers().end());
+ answer->addPeer(*next_peer);
tried++;
}
- tq->getReturnMailbox()->put_init(ta, TRACKER_COMM_SIZE)->detach();
+ query->getReturnMailbox()->put_init(answer, TRACKER_COMM_SIZE)->detach();
- delete tq;
+ delete query;
comm = nullptr;
} else {
simgrid::s4u::this_actor::sleep_for(1);