diff --git a/src/main.rs b/src/main.rs index b6c6dfe..5aabbce 100644 --- a/src/main.rs +++ b/src/main.rs @@ -26,9 +26,6 @@ struct Video { link: Link, } -// Always use a limit to prevent DoS attacks. -const LIMIT: u64 = 1024; - impl fmt::Display for Link { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { write!(f, "{}", self.uri) @@ -53,7 +50,11 @@ impl<'v> FromFormValue<'v> for Link { Ok(abs_uri) => Some(abs_uri), Err(_) => continue, }; - break; + if abs_uri.as_ref().unwrap().scheme() == "http" + || abs_uri.as_ref().unwrap().scheme() == "https" + { + break; + } } // Check for success @@ -71,6 +72,7 @@ impl<'v> FromFormValue<'v> for Link { #[post("/new", data = "