examples/s4u/app-bittorrent/s4u-tracker.hpp

   1 /* Copyright (c) 2012-2019. 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 #ifndef BITTORRENT_TRACKER_HPP_
   8 #define BITTORRENT_TRACKER_HPP_
   9
  10 #include "s4u-bittorrent.hpp"
  11 #include <set>
  12
  13 class TrackerQuery {
  14   int peer_id; // peer id
  15   simgrid::s4u::Mailbox* return_mailbox;
  16
  17 public:
  18   explicit TrackerQuery(int peer_id, simgrid::s4u::Mailbox* return_mailbox)
  19       : peer_id(peer_id), return_mailbox(return_mailbox){};
  20   ~TrackerQuery() = default;
  21   int getPeerId() { return peer_id; }
  22   simgrid::s4u::Mailbox* getReturnMailbox() { return return_mailbox; }
  23 };
  24
  25 class TrackerAnswer {
  26   // int interval; // how often the peer should contact the tracker (unused for now)
  27   std::set<int>* peers; // the peer list the peer has asked for.
  28 public:
  29   explicit TrackerAnswer(int /*interval*/) /*: interval(interval)*/ { peers = new std::set<int>; }
  30   TrackerAnswer(const TrackerAnswer&)                                       = delete;
  31   TrackerAnswer& operator=(const TrackerAnswer&) = delete;
  32   ~TrackerAnswer() { delete peers; };
  33   void addPeer(int peer) { peers->insert(peer); }
  34   std::set<int>* getPeers() { return peers; }
  35 };
  36
  37 class Tracker {
  38   double deadline;
  39   RngStream stream;
  40   simgrid::s4u::Mailbox* mailbox;
  41   std::set<int> known_peers;
  42
  43 public:
  44   explicit Tracker(std::vector<std::string> args);
  45   void operator()();
  46 };
  47
  48 #endif /* BITTORRENT_TRACKER_HPP */