rocket 0.2.6

Web framework for nightly with a focus on ease-of-use, expressibility, and speed.
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

#![feature(plugin, custom_derive)]
#![plugin(rocket_codegen)]

extern crate rocket;

use std::net::SocketAddr;

#[get("/")]
fn get_ip(remote: SocketAddr) -> String {
    remote.to_string()
}

#[cfg(feature = "testing")]
mod remote_rewrite_tests {
    use super::*;
    use rocket::testing::MockRequest;
    use rocket::http::Method::*;
    use rocket::http::{Header, Status};

    use std::net::SocketAddr;

    const KNOWN_IP: &'static str = "127.0.0.1:8000";

    fn check_ip(header: Option<Header<'static>>, ip: Option<String>) {
        let address: SocketAddr = KNOWN_IP.parse().unwrap();
        let port = address.port();

        let rocket = rocket::ignite().mount("/", routes![get_ip]);
        let mut req = MockRequest::new(Get, "/").remote(address);
        if let Some(header) = header {
            req.add_header(header);
        }

        let mut response = req.dispatch_with(&rocket);
        assert_eq!(response.status(), Status::Ok);
        let body_str = response.body().and_then(|b| b.into_string());
        match ip {
            Some(ip) => assert_eq!(body_str, Some(format!("{}:{}", ip, port))),
            None => assert_eq!(body_str, Some(KNOWN_IP.into()))
        }
    }

    #[test]
    fn x_real_ip_rewrites() {
        let ip = "8.8.8.8";
        check_ip(Some(Header::new("X-Real-IP", ip)), Some(ip.to_string()));

        let ip = "129.120.111.200";
        check_ip(Some(Header::new("X-Real-IP", ip)), Some(ip.to_string()));
    }

    #[test]
    fn x_real_ip_rewrites_ipv6() {
        let ip = "2001:db8:0:1:1:1:1:1";
        check_ip(Some(Header::new("X-Real-IP", ip)), Some(format!("[{}]", ip)));

        let ip = "2001:db8::2:1";
        check_ip(Some(Header::new("X-Real-IP", ip)), Some(format!("[{}]", ip)));
    }

    #[test]
    fn uncased_header_rewrites() {
        let ip = "8.8.8.8";
        check_ip(Some(Header::new("x-REAL-ip", ip)), Some(ip.to_string()));

        let ip = "1.2.3.4";
        check_ip(Some(Header::new("x-real-ip", ip)), Some(ip.to_string()));
    }

    #[test]
    fn no_header_no_rewrite() {
        check_ip(Some(Header::new("real-ip", "?")), None);
        check_ip(None, None);
    }

    #[test]
    fn bad_header_doesnt_rewrite() {
        let ip = "092348092348";
        check_ip(Some(Header::new("X-Real-IP", ip)), None);

        let ip = "1200:100000:0120129";
        check_ip(Some(Header::new("X-Real-IP", ip)), None);

        let ip = "192.168.1.900";
        check_ip(Some(Header::new("X-Real-IP", ip)), None);
    }
}