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

use soft_ascii_string::SoftAsciiString;
use idna;

use ::error::{EncodingError, EncodingErrorKind};


/// uses puny code on given domain to return a ascii representation
///
/// # Implementation Detail
/// this function uses `idna::domain_to_ascii`, see the
/// `idna` crates documentation fore more details on how
/// exactly all edgecase are handled
///
/// # Note
/// that this function does not validate the domain, e.g.
/// if you puny code the domain `"this seems\0so;wrong"` it
/// will return `Ok("this seems\0so;wrong")`
///
pub fn puny_code_domain<R: AsRef<str>>(domain: R)
    -> Result<SoftAsciiString, EncodingError>
{
    _puny_code_domain(domain.as_ref())
}

fn _puny_code_domain(domain: &str)
    -> Result<SoftAsciiString, EncodingError>
{
    match idna::domain_to_ascii(domain) {
        Ok(asciified) => {
            //SAFE: well we converted it to ascii, so it's ascii
            Ok(SoftAsciiString::from_unchecked(asciified))
        },
        Err(_non_informative_err) => {
            Err(EncodingErrorKind::NotEncodable { encoding: "punycode" }.into())
        }
    }
}


#[cfg(test)]
mod test {
    use idna;
    use super::puny_code_domain;

    #[test]
    fn idna_does_not_validate() {
        let domain = "this seems\0so;wrong";
        assert_eq!(
            domain.to_owned(),
            assert_ok!( idna::domain_to_ascii(domain) )
        );
    }

    #[test]
    fn nop_puny_code() {
        let domain = "is_ascii.notadomain";

        let encoded = assert_ok!( puny_code_domain( domain ) );
        assert_eq!(
            &*encoded,
            "is_ascii.notadomain"
        );
    }
    #[test]
    fn puny_code_ascii_mail() {
        let domain = "nöt_ascii.ü";
        let encoded = assert_ok!( puny_code_domain(domain) );
        assert_eq!(
            &*encoded,
            "xn--nt_ascii-n4a.xn--tda"
        );
    }
}