torrust-tracker 3.0.0

A feature rich BitTorrent tracker.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use std::str::FromStr as _;
use std::time::Duration;

use serde::Serialize;
use torrust_tracker_primitives::info_hash::InfoHash;
use url::Url;

use crate::console::clients::http::Error;
use crate::shared::bit_torrent::tracker::http::client::responses::announce::Announce;
use crate::shared::bit_torrent::tracker::http::client::responses::scrape;
use crate::shared::bit_torrent::tracker::http::client::{requests, Client};

#[derive(Debug, Clone, Serialize)]
pub struct Checks {
    url: Url,
    results: Vec<(Check, Result<(), Error>)>,
}

#[derive(Debug, Clone, Serialize)]
pub enum Check {
    Announce,
    Scrape,
}

pub async fn run(http_trackers: Vec<Url>, timeout: Duration) -> Vec<Result<Checks, Checks>> {
    let mut results = Vec::default();

    tracing::debug!("HTTP trackers ...");

    for ref url in http_trackers {
        let mut base_url = url.clone();
        base_url.set_path("");

        let mut checks = Checks {
            url: url.clone(),
            results: Vec::default(),
        };

        // Announce
        {
            let check = check_http_announce(&base_url, timeout).await.map(|_| ());

            checks.results.push((Check::Announce, check));
        }

        // Scrape
        {
            let check = check_http_scrape(&base_url, timeout).await.map(|_| ());

            checks.results.push((Check::Scrape, check));
        }

        if checks.results.iter().any(|f| f.1.is_err()) {
            results.push(Err(checks));
        } else {
            results.push(Ok(checks));
        }
    }

    results
}

async fn check_http_announce(url: &Url, timeout: Duration) -> Result<Announce, Error> {
    let info_hash_str = "9c38422213e30bff212b30c360d26f9a02136422".to_string(); // # DevSkim: ignore DS173237
    let info_hash = InfoHash::from_str(&info_hash_str).expect("a valid info-hash is required");

    let client = Client::new(url.clone(), timeout).map_err(|err| Error::HttpClientError { err })?;

    let response = client
        .announce(
            &requests::announce::QueryBuilder::with_default_values()
                .with_info_hash(&info_hash)
                .query(),
        )
        .await
        .map_err(|err| Error::HttpClientError { err })?;

    let response = response.bytes().await.map_err(|e| Error::ResponseError { err: e.into() })?;

    let response = serde_bencode::from_bytes::<Announce>(&response).map_err(|e| Error::ParseBencodeError {
        data: response,
        err: e.into(),
    })?;

    Ok(response)
}

async fn check_http_scrape(url: &Url, timeout: Duration) -> Result<scrape::Response, Error> {
    let info_hashes: Vec<String> = vec!["9c38422213e30bff212b30c360d26f9a02136422".to_string()]; // # DevSkim: ignore DS173237
    let query = requests::scrape::Query::try_from(info_hashes).expect("a valid array of info-hashes is required");

    let client = Client::new(url.clone(), timeout).map_err(|err| Error::HttpClientError { err })?;

    let response = client.scrape(&query).await.map_err(|err| Error::HttpClientError { err })?;

    let response = response.bytes().await.map_err(|e| Error::ResponseError { err: e.into() })?;

    let response = scrape::Response::try_from_bencoded(&response).map_err(|e| Error::BencodeParseError {
        data: response,
        err: e.into(),
    })?;

    Ok(response)
}