mtorrent-core 0.5.2

Fundamentals for building asynchronous BitTorrent clients
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

use derive_more::{Debug, Deref};
use serde::de::IntoDeserializer;
use serde::{Deserialize, Serialize, Serializer};
use std::str::FromStr;

/// Parsed tracker address.
#[derive(Debug, PartialEq, Eq, Clone, Hash, PartialOrd, Ord)]
pub enum TrackerUrl {
    Http(Http),
    Udp(Udp),
}

#[derive(Debug, PartialEq, Eq, Clone, Hash, PartialOrd, Ord, Deref)]
#[deref(forward)]
#[debug("{_0}")]
pub struct Http(String);

#[derive(Debug, PartialEq, Eq, Clone, Hash, PartialOrd, Ord, Deref)]
#[deref(forward)]
#[debug("{_0}")]
pub struct Udp(String);

impl Serialize for TrackerUrl {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        match self {
            TrackerUrl::Http(Http(url)) => url.serialize(serializer),
            TrackerUrl::Udp(Udp(addr)) => format!("udp://{addr}").serialize(serializer),
        }
    }
}

impl<'de> Deserialize<'de> for TrackerUrl {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        use serde::de::{Error, Unexpected};

        let text_url = String::deserialize(deserializer)?;
        let parsed_url = url::Url::parse(&text_url).map_err(Error::custom)?;
        match parsed_url.scheme() {
            "http" | "https" => Ok(TrackerUrl::Http(Http(text_url))),
            "udp" => Ok(TrackerUrl::Udp(Udp(format!(
                "{}:{}",
                parsed_url
                    .host_str()
                    .ok_or(Error::invalid_value(Unexpected::Str(&text_url), &"hostname present"))?,
                parsed_url
                    .port()
                    .ok_or(Error::invalid_value(Unexpected::Str(&text_url), &"port present"))?
            )))),
            scheme => Err(Error::invalid_value(Unexpected::Str(scheme), &"supported scheme")),
        }
    }
}

impl FromStr for TrackerUrl {
    type Err = serde::de::value::Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        TrackerUrl::deserialize(s.into_deserializer())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::collections::HashSet;

    #[test]
    fn test_extract_http_and_udp_trackers() {
        let trackers = [
            "udp://open.stealth.si:80/announce",
            "udp://tracker.opentrackr.org:1337/announce",
            "udp://tracker.tiny-vps.com:6969",
            "https://example.com",
            "udp://tracker.internetwarriors.net:1337/",
            "udp://tracker.skyts.net:6969/announce330",
            "http://example.com",
            "udp://[2001:db8:85a3:8d3:1319:8a2e:370:7348]:6969/",
        ];
        let expected_udp_trackers = [
            "open.stealth.si:80",
            "tracker.opentrackr.org:1337",
            "tracker.tiny-vps.com:6969",
            "tracker.internetwarriors.net:1337",
            "tracker.skyts.net:6969",
            "[2001:db8:85a3:8d3:1319:8a2e:370:7348]:6969",
        ];
        let expected_http_trackers = ["https://example.com", "http://example.com"];

        let mut udp_trackers = HashSet::new();
        let mut http_trackers = HashSet::new();
        for tracker in trackers {
            match tracker.parse::<TrackerUrl>().unwrap() {
                TrackerUrl::Http(addr) => http_trackers.insert(addr),
                TrackerUrl::Udp(addr) => udp_trackers.insert(addr),
            };
        }

        assert_eq!(
            http_trackers,
            expected_http_trackers.into_iter().map(ToString::to_string).map(Http).collect()
        );
        assert_eq!(
            udp_trackers,
            expected_udp_trackers.into_iter().map(ToString::to_string).map(Udp).collect()
        );
    }

    #[test]
    fn test_parse_malformed_tracker_urls() {
        let result = "invalid".parse::<TrackerUrl>();
        assert!(result.is_err(), "{result:?}");

        let err = "udp://tracker.tiny-vps.com/announce".parse::<TrackerUrl>().unwrap_err();
        assert!(err.to_string().contains("expected port present"), "{err}");

        let err = "udp:/run/foo.socket".parse::<TrackerUrl>().unwrap_err();
        assert!(err.to_string().contains("expected hostname present"), "{err}");

        let err = "ssh://open.stealth.si:80/announce".parse::<TrackerUrl>().unwrap_err();
        assert!(err.to_string().contains("expected supported scheme"), "{err}");
    }
}