TeamPiped-mirror/piped-proxy
1
0
Fork 0
mirror of https://github.com/TeamPiped/piped-proxy.git synced 2024-08-14 23:50:45 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #65 from Bnyro/code-cleanup

Browse source 
cleanup query handling and domain validation
This commit is contained in:
Kavin 2023-08-04 10:35:59 +01:00 committed by GitHub
commit 53329c43e8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/main.rs
View file

@ -52,6 +52,7 @@ static CLIENT: Lazy<Client> = Lazy::new(|| {
} }
}); });
const ANDROID_USER_AGENT: &str = "com.google.android.youtube/1537338816 (Linux; U; Android 13; en_US; ; Build/TQ2A.230505.002; Cronet/113.0.5672.24)";
const ALLOWED_DOMAINS: [&str; 8] = [ const ALLOWED_DOMAINS: [&str; 8] = [
"youtube.com", "youtube.com",
"googlevideo.com", "googlevideo.com",
@ -110,22 +111,10 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
return Err("No host provided".into()); return Err("No host provided".into());
} }
let rewrite = { let rewrite = query.get("rewrite") != Some("false");
if let Some(rewrite) = query.get("rewrite") {
rewrite == "true"
} else {
true
}
};
#[cfg(feature = "avif")] #[cfg(feature = "avif")]
let avif = { let avif = query.get("avif") == Some("true");
if let Some(avif) = query.get("avif") {
avif == "true"
} else {
false
}
};
let host = res.unwrap(); let host = res.unwrap();
let domain = RE_DOMAIN.captures(host.as_str()); let domain = RE_DOMAIN.captures(host.as_str());
@ -136,16 +125,7 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
let domain = domain.unwrap().get(1).unwrap().as_str(); let domain = domain.unwrap().get(1).unwrap().as_str();
let mut allowed = false; if !ALLOWED_DOMAINS.contains(&domain) {
for allowed_domain in ALLOWED_DOMAINS.iter() {
if &domain == allowed_domain {
allowed = true;
break;
}
}
if !allowed {
return Err("Domain not allowed".into()); return Err("Domain not allowed".into());
} }
@ -187,7 +167,7 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
} }
if is_android { if is_android {
request_headers.insert("User-Agent", "com.google.android.youtube/1537338816 (Linux; U; Android 13; en_US; ; Build/TQ2A.230505.002; Cronet/113.0.5672.24)".parse().unwrap()); request_headers.insert("User-Agent", ANDROID_USER_AGENT.parse().unwrap());
} }
let resp = CLIENT.execute(request).await; let resp = CLIENT.execute(request).await;