rtorrent/src/core/dht_manager.h

#ifndef RTORRENT_CORE_DHT_MANAGER_H
#define RTORRENT_CORE_DHT_MANAGER_H

#include <torrent/object.h>
#include <torrent/utils/scheduler.h>

namespace core {

class DhtManager {
public:
  static constexpr int dht_disable = 0;
  static constexpr int dht_off     = 1;
  static constexpr int dht_auto    = 2;
  static constexpr int dht_on      = 3;

  ~DhtManager();

  void                load_dht_cache();
  void                save_dht_cache();
  torrent::Object     dht_statistics();

  void                start_dht();
  void                stop_dht();
  void                auto_start()                 { if (m_start == dht_auto) start_dht(); }

  void                set_mode_by_user(const std::string& arg);
  void                set_mode_directly(unsigned int mode);

  void                set_auto_if_untouched_and_has_session();

private:
  static constexpr int dht_settings_num = 4;
  static const char*   dht_settings[dht_settings_num];

  void                update();
  bool                log_statistics(bool force);

  unsigned int        m_dhtPrevCycle;
  unsigned int        m_dhtPrevQueriesSent;
  unsigned int        m_dhtPrevRepliesReceived;
  unsigned int        m_dhtPrevQueriesReceived;
  uint64_t            m_dhtPrevBytesUp;
  uint64_t            m_dhtPrevBytesDown;

  torrent::utils::SchedulerEntry m_update_timeout;
  torrent::utils::SchedulerEntry m_stop_timeout;

  bool                m_warned{};
  bool                m_set_by_user{};
  int                 m_start{dht_off};
};

}

#endif