|
|
|
|
#include <ServerContext.hpp>
|
|
|
|
|
#include <SessionContext.hpp>
|
|
|
|
|
#include <vector>
|
|
|
|
|
|
|
|
|
|
void TestSessionSettings::set_int(int type, int value) {
|
|
|
|
|
if (type == download_rate_limit) {
|
|
|
|
|
download_rate_limit_ = value;
|
|
|
|
|
}
|
|
|
|
|
if (type == upload_rate_limit) {
|
|
|
|
|
upload_rate_limit_ = value;
|
|
|
|
|
}
|
|
|
|
|
if (type == enable_dht) {
|
|
|
|
|
enable_dht_ = value;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
void TestSessionSettings::set_bool(int type, int value) {
|
|
|
|
|
set_int(type, value);
|
|
|
|
|
}
|
|
|
|
|
int TestSessionSettings::get_int(int type) const {
|
|
|
|
|
if (type == download_rate_limit) {
|
|
|
|
|
return download_rate_limit_;
|
|
|
|
|
}
|
|
|
|
|
if (type == upload_rate_limit) {
|
|
|
|
|
return upload_rate_limit_;
|
|
|
|
|
}
|
|
|
|
|
if (type == enable_dht) {
|
|
|
|
|
return enable_dht_;
|
|
|
|
|
}
|
|
|
|
|
return 0;
|
|
|
|
|
};
|
|
|
|
|
bool TestSessionSettings::get_bool(int type) { return get_int(type); }
|
|
|
|
|
std::string TestSessionSettings::get_str(int type) {
|
|
|
|
|
if (type == listen_interfaces) {
|
|
|
|
|
return listen_interfaces_;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
void TestSessionSettings::set_str(int type, std::string value) {
|
|
|
|
|
if (type == listen_interfaces) {
|
|
|
|
|
listen_interfaces_ = value;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
bool TestSession::is_valid() {
|
|
|
|
|
return valid;
|
|
|
|
|
}
|
|
|
|
|
TestSessionSettings TestSession::get_settings() {
|
|
|
|
|
return settings_;
|
|
|
|
|
}
|
|
|
|
|
bool TestSession::is_paused() const {
|
|
|
|
|
return paused;
|
|
|
|
|
}
|
|
|
|
|
void TestSession::pause() {
|
|
|
|
|
paused = true;
|
|
|
|
|
}
|
|
|
|
|
void TestSession::resume() {
|
|
|
|
|
paused = false;
|
|
|
|
|
}
|
|
|
|
|
int TestSession::listen_port() {
|
|
|
|
|
return std::stoi(settings_.listen_interfaces_.substr(8, settings_.listen_interfaces_.size() - 8));
|
|
|
|
|
}
|
|
|
|
|
bool TestSession::is_dht_running() const {
|
|
|
|
|
return settings_.enable_dht_;
|
|
|
|
|
}
|
|
|
|
|
std::vector<TestTorrent> &TestSession::get_torrents() {
|
|
|
|
|
return torrents;
|
|
|
|
|
}
|
|
|
|
|
void TestSession::apply_settings(TestSessionSettings settings) {
|
|
|
|
|
settings_ = settings;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
void TestSession::async_add_torrent(const libtorrent::add_torrent_params &settings) {
|
|
|
|
|
torrents.emplace_back(settings);
|
|
|
|
|
};
|
|
|
|
|
|