Cpt.Captain/mpv_send
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

only try to use uris starting with http{,s}

Browse Source
This commit is contained in:
CptCaptain 2021-11-02 22:29:35 +01:00
parent 004887799e
commit a309d19e61
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main.rs
View File

@ -26,9 +26,6 @@ struct Video {
link: Link, link: Link,
} }
// Always use a limit to prevent DoS attacks.
const LIMIT: u64 = 1024;
impl fmt::Display for Link { impl fmt::Display for Link {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.uri) write!(f, "{}", self.uri)
@ -53,7 +50,11 @@ impl<'v> FromFormValue<'v> for Link {
Ok(abs_uri) => Some(abs_uri), Ok(abs_uri) => Some(abs_uri),
Err(_) => continue, Err(_) => continue,
}; };
break; if abs_uri.as_ref().unwrap().scheme() == "http"
|| abs_uri.as_ref().unwrap().scheme() == "https"
{
break;
}
} }
// Check for success // Check for success
@ -71,6 +72,7 @@ impl<'v> FromFormValue<'v> for Link {
#[post("/new", data = "<video>")] #[post("/new", data = "<video>")]
fn new(video: Form<Video>) -> Redirect { fn new(video: Form<Video>) -> Redirect {
dbg!(&video.link);
Command::new("/bin/env") Command::new("/bin/env")
.args(&[ .args(&[
"DISPLAY=:0", "DISPLAY=:0",