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
//! Host ("authority") filter
//!
use crate::filter::{filter_fn_one, Filter, One};
use crate::reject::{self, Rejection};
use futures_util::future;
pub use http::uri::Authority;
use std::str::FromStr;

/// Creates a `Filter` that requires a specific authority (target server's
/// host and port) in the request.
///
/// Authority is specified either in the `Host` header or in the target URI.
///
/// # Example
///
/// ```
/// use warp::Filter;
///
/// let multihost =
///     warp::host::exact("foo.com").map(|| "you've reached foo.com")
///     .or(warp::host::exact("bar.com").map(|| "you've reached bar.com"));
/// ```
pub fn exact(expected: &str) -> impl Filter<Extract = (), Error = Rejection> + Clone {
    let expected = Authority::from_str(expected).expect("invalid host/authority");
    optional()
        .and_then(move |option: Option<Authority>| match option {
            Some(authority) if authority == expected => future::ok(()),
            _ => future::err(reject::not_found()),
        })
        .untuple_one()
}

/// Creates a `Filter` that looks for an authority (target server's host
/// and port) in the request.
///
/// Authority is specified either in the `Host` header or in the target URI.
///
/// If found, extracts the `Authority`, otherwise continues the request,
/// extracting `None`.
///
/// Rejects with `400 Bad Request` if the `Host` header is malformed or if there
/// is a mismatch between the `Host` header and the target URI.
///
/// # Example
///
/// ```
/// use warp::{Filter, host::Authority};
///
/// let host = warp::host::optional()
///     .map(|authority: Option<Authority>| {
///         if let Some(a) = authority {
///             format!("{} is currently not at home", a.host())
///         } else {
///             "please state who you're trying to reach".to_owned()
///         }
///     });
/// ```
pub fn optional() -> impl Filter<Extract = One<Option<Authority>>, Error = Rejection> + Copy {
    filter_fn_one(move |route| {
        // The authority can be sent by clients in various ways:
        //
        //  1) in the "target URI"
        //    a) serialized in the start line (HTTP/1.1 proxy requests)
        //    b) serialized in `:authority` pseudo-header (HTTP/2 generated - "SHOULD")
        //  2) in the `Host` header (HTTP/1.1 origin requests, HTTP/2 converted)
        //
        // Hyper transparently handles 1a/1b, but not 2, so we must look at both.

        let from_uri = route.uri().authority();

        let name = "host";
        let from_header = route.headers()
            .get(name)
            .map(|value|
                // Header present, parse it
                value.to_str().map_err(|_| reject::invalid_header(name))
                    .and_then(|value| Authority::from_str(value).map_err(|_| reject::invalid_header(name)))
            );

        future::ready(match (from_uri, from_header) {
            // no authority in the request (HTTP/1.0 or non-conforming)
            (None, None) => Ok(None),

            // authority specified in either or both matching
            (Some(a), None) => Ok(Some(a.clone())),
            (None, Some(Ok(a))) => Ok(Some(a)),
            (Some(a), Some(Ok(b))) if *a == b => Ok(Some(b)),

            // mismatch
            (Some(_), Some(Ok(_))) => Err(reject::invalid_header(name)),

            // parse error
            (_, Some(Err(r))) => Err(r),
        })
    })
}