micropelago/hyper-reverse-proxy
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: do not panic on upgraded connection copy error

Browse Source
This commit is contained in:
Nicolas Pernoud 2022-11-28 09:08:24 +01:00
parent 5d28386c3f
commit 116e84778c
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/lib.rs
View File

@ -93,7 +93,7 @@ fn get_upgrade_type(headers: &HeaderMap) -> Option<String> {
upgrade_value.to_str().unwrap().to_owned() upgrade_value.to_str().unwrap().to_owned()
); );
return Some(upgrade_value.to_str().unwrap().to_lowercase().to_owned()); return Some(upgrade_value.to_str().unwrap().to_lowercase());
} }
} }
@ -128,7 +128,7 @@ fn forward_uri<B>(forward_url: &str, req: &Request<B>) -> String {
let split_url = forward_url.split('?').collect::<Vec<&str>>(); let split_url = forward_url.split('?').collect::<Vec<&str>>();
let mut base_url: &str = split_url.get(0).unwrap_or(&""); let mut base_url: &str = split_url.first().unwrap_or(&"");
let forward_url_query: &str = split_url.get(1).unwrap_or(&""); let forward_url_query: &str = split_url.get(1).unwrap_or(&"");
let path2 = req.uri().path(); let path2 = req.uri().path();
@ -338,9 +338,10 @@ pub async fn call<'a, T: hyper::client::connect::Connect + Clone + Send + Sync +
let mut request_upgraded = let mut request_upgraded =
request_upgraded.await.expect("failed to upgrade request"); request_upgraded.await.expect("failed to upgrade request");
copy_bidirectional(&mut response_upgraded, &mut request_upgraded) match copy_bidirectional(&mut response_upgraded, &mut request_upgraded).await {
.await Ok(_) => debug!("successfull copy between upgraded connections"),
.expect("coping between upgraded connections failed"); Err(_) => error!("failed copy between upgraded connections (EOF)"),
}
}); });
Ok(response) Ok(response)