refactor: server host filtering

core/Cargo.toml

@@ -34,21 +34,23 @@ parking_lot = { version = "0.12", optional = true }
 tokio = { version = "1.16", optional = true }
 wasm-bindgen-futures = { version = "0.4.19", optional = true }
 futures-timer = { version = "3", optional = true }
-globset = { version = "0.4", optional = true }
+route-recognizer = { version = "0.3.1", optional = true }
+http = { version = "0.2.9", optional = true }
 
 [features]
 default = []
 http-helpers = ["hyper", "futures-util"]
 server = [
 	"futures-util/alloc",
-	"globset",
+	"route-recognizer",
 	"rustc-hash/std",
 	"parking_lot",
 	"rand",
 	"tokio/rt",
 	"tokio/sync",
 	"tokio/macros",
 	"tokio/time",
+	"http",
 ]
 client = ["futures-util/sink", "tokio/sync"]
 async-client = [

core/src/server/host_filtering.rs

@@ -0,0 +1,243 @@
+// Copyright 2019-2021 Parity Technologies (UK) Ltd.
+//
+// Permission is hereby granted, free of charge, to any
+// person obtaining a copy of this software and associated
+// documentation files (the "Software"), to deal in the
+// Software without restriction, including without
+// limitation the rights to use, copy, modify, merge,
+// publish, distribute, sublicense, and/or sell copies of
+// the Software, and to permit persons to whom the Software
+// is furnished to do so, subject to the following
+// conditions:
+//
+// The above copyright notice and this permission notice
+// shall be included in all copies or substantial portions
+// of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
+// ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
+// TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
+// PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
+// SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
+// IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+// DEALINGS IN THE SOFTWARE.
+
+//! HTTP Host Header validation.
+
+use crate::Error;
+use http::uri::{InvalidUri, Uri};
+use route_recognizer::Router;
+use std::str::FromStr;
+
+/// Port pattern
+#[derive(Clone, Copy, Hash, PartialEq, Eq, Debug)]
+pub enum Port {
+	/// No port specified (default port)
+	Default,
+	/// Port specified as a wildcard pattern (*).
+	Any,
+	/// Fixed numeric port
+	Fixed(u16),
+}
+
+impl From<u16> for Port {
+	fn from(port: u16) -> Port {
+		Port::Fixed(port)
+	}
+}
+
+/// Represent the http URI scheme that is returned by the HTTP host header
+///
+/// <http-URI = "http:" "//" authority path-abempty [ "?" query ][ "#" fragment ]>
+///
+/// Further information can be found: https://www.rfc-editor.org/rfc/rfc7230#section-2.7.1
+#[derive(Clone, Hash, PartialEq, Eq, Debug)]
+pub struct Authority {
+	hostname: String,
+	port: Port,
+}
+
+impl FromStr for Authority {
+	type Err = String;
+
+	fn from_str(s: &str) -> Result<Self, Self::Err> {
+		let uri: Uri = s.parse().map_err(|e: InvalidUri| e.to_string())?;
+		let authority = uri.authority().ok_or_else(|| "HTTP Host must contain authority".to_owned())?;
+		let hostname = authority.host();
+		let maybe_port = &authority.as_str()[hostname.len()..];
+
+		// After the host segment, the authority may contain a port such as `fooo:33`, `foo:*` or `foo`
+		let port = match maybe_port.split_once(':') {
+			Some((_, "*")) => Port::Any,
+			Some((_, p)) => {
+				let port_u16: u16 = p.parse().map_err(|e: std::num::ParseIntError| e.to_string())?;
+
+				// Omit default port to allow both requests with and without the default port.
+				match default_port(uri.scheme_str()) {
+					Some(p) if p == port_u16 => Port::Default,
+					_ => port_u16.into(),
+				}
+			}
+			None => Port::Default,
+		};
+
+		Ok(Self { hostname: hostname.to_string(), port })
+	}
+}
+
+/// Represent the URL patterns that is whitelisted.
+#[derive(Default, Debug, Clone)]
+pub struct UrlPattern(Router<Port>);
+
+impl<T> From<T> for UrlPattern
+where
+	T: IntoIterator<Item = Authority>,
+{
+	fn from(value: T) -> Self {
+		let mut router = Router::new();
+
+		for auth in value.into_iter() {
+			router.add(&auth.hostname, auth.port);
+		}
+
+		Self(router)
+	}
+}
+
+impl UrlPattern {
+	fn recognize(&self, other: &Authority) -> bool {
+		if let Ok(p) = self.0.recognize(&other.hostname) {
+			let p = p.handler();
+
+			match (p, &other.port) {
+				(Port::Any, _) => true,
+				(Port::Default, Port::Default) => true,
+				(Port::Fixed(p1), Port::Fixed(p2)) if p1 == p2 => true,
+				_ => false,
+			}
+		} else {
+			false
+		}
+	}
+}
+
+/// Policy for validating the `HTTP host header`.
+#[derive(Debug, Clone)]
+pub enum AllowHosts {
+	/// Allow all hosts (no filter).
+	Any,
+	/// Allow only specified hosts.
+	Only(UrlPattern),
+}
+
+impl AllowHosts {
+	/// Verify a host.
+	pub fn verify(&self, value: &str) -> Result<(), Error> {
+		let auth = Authority::from_str(value)
+			.map_err(|_| Error::HttpHeaderRejected("host", format!("Invalid authority: {value}")))?;
+
+		if let AllowHosts::Only(url_pat) = self {
+			if !url_pat.recognize(&auth) {
+				return Err(Error::HttpHeaderRejected("host", value.into()));
+			}
+		}
+
+		Ok(())
+	}
+}
+
+fn default_port(scheme: Option<&str>) -> Option<u16> {
+	match scheme {
+		Some("http") | Some("ws") => Some(80),
+		Some("https") | Some("wss") => Some(443),
+		Some("ftp") => Some(21),
+		_ => None,
+	}
+}
+
+#[cfg(test)]
+mod tests {
+	use super::{AllowHosts, Authority, Port};
+	use std::str::FromStr;
+
+	fn authority(host: &str, port: Port) -> Authority {
+		Authority { hostname: host.to_owned(), port }
+	}
+
+	#[test]
+	fn should_parse_valid_authority() {
+		assert_eq!(Authority::from_str("http://parity.io").unwrap(), authority("parity.io", Port::Default));
+		assert_eq!(Authority::from_str("https://parity.io:8443").unwrap(), authority("parity.io", Port::Fixed(8443)));
+		assert_eq!(Authority::from_str("chrome-extension://124.0.0.1").unwrap(), authority("124.0.0.1", Port::Default));
+		assert_eq!(Authority::from_str("http://*.domain:*/somepath").unwrap(), authority("*.domain", Port::Any));
+		assert_eq!(Authority::from_str("parity.io").unwrap(), authority("parity.io", Port::Default));
+		assert_eq!(
+			Authority::from_str("http://[2001:db8:85a3:8d3:1319:8a2e:370:7348]:9933/").unwrap(),
+			authority("[2001:db8:85a3:8d3:1319:8a2e:370:7348]", Port::Fixed(9933))
+		);
+		assert_eq!(
+			Authority::from_str("http://[2001:db8:85a3:8d3:1319:8a2e:370:7348]/").unwrap(),
+			authority("[2001:db8:85a3:8d3:1319:8a2e:370:7348]", Port::Default)
+		);
+		assert_eq!(
+			Authority::from_str("https://user:password@example.com/tmp/foo").unwrap(),
+			authority("example.com", Port::Default)
+		);
+	}
+
+	#[test]
+	fn should_not_parse_invalid_authority() {
+		assert!(Authority::from_str("/foo/bar").is_err());
+		assert!(Authority::from_str("user:password").is_err());
+		assert!(Authority::from_str("parity.io/somepath").is_err());
+		assert!(Authority::from_str("127.0.0.1:8545/somepath").is_err());
+	}
+
+	#[test]
+	fn should_allow_when_validation_is_disabled() {
+		assert!((AllowHosts::Any).verify("any").is_ok());
+	}
+
+	#[test]
+	fn should_reject_if_header_not_on_the_list() {
+		assert!((AllowHosts::Only(vec![].into())).verify("parity.io").is_err());
+	}
+
+	#[test]
+	fn should_accept_if_on_the_list() {
+		assert!(AllowHosts::Only(vec![Authority::from_str("parity.io").unwrap()].into()).verify("parity.io").is_ok());
+	}
+
+	#[test]
+	fn should_accept_if_on_the_list_with_port() {
+		assert!((AllowHosts::Only(vec![Authority::from_str("parity.io:443").unwrap()].into()))
+			.verify("parity.io:443")
+			.is_ok());
+		assert!(AllowHosts::Only(vec![Authority::from_str("parity.io").unwrap()].into())
+			.verify("parity.io:443")
+			.is_err());
+	}
+
+	#[test]
+	fn should_support_wildcards() {
+		assert!((AllowHosts::Only(vec![Authority::from_str("*.web3.site:*").unwrap()].into()))
+			.verify("parity.web3.site:8180")
+			.is_ok());
+		assert!((AllowHosts::Only(vec![Authority::from_str("*.web3.site:*").unwrap()].into()))
+			.verify("parity.web3.site")
+			.is_ok());
+	}
+
+	#[test]
+	fn should_accept_with_and_without_default_port() {
+		assert!(AllowHosts::Only(vec![Authority::from_str("https://parity.io:443").unwrap()].into())
+			.verify("https://parity.io")
+			.is_ok());
+
+		assert!(AllowHosts::Only(vec![Authority::from_str("https://parity.io").unwrap()].into())
+			.verify("https://parity.io:443")
+			.is_ok());
+	}
+}