tuxmain/rust-rpxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

todo: switching_protocols in response and transfer-encoding

Browse source
This commit is contained in:
Jun Kurihara 2022-06-24 23:35:47 -04:00
commit c122ca8b4d
No known key found for this signature in database
GPG key ID: 48ADFD173ED22B03
1 changed files with 26 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

38
src/proxy/proxy_handler.rs
View file

@ -1,9 +1,10 @@
// Motivated by https://github.com/felipenoris/hyper-reverse-proxy
// Highly motivated by https://github.com/felipenoris/hyper-reverse-proxy
use super::Proxy;
use crate::{error::*, log::*};
use hyper::{
client::connect::Connect,
header::{HeaderMap, HeaderValue},
http::response,
Body, Request, Response, StatusCode, Uri,
};
use std::net::SocketAddr;
@ -77,21 +78,34 @@ where
};
debug!("Request to be forwarded: {:?}", req_forwarded);
// // Forward request to
// let res_backend = match self.forwarder.request(req_forwarded).await {
// Ok(res) => res,
// Err(e) => {
// error!("Failed to get response from backend: {}", e);
// return http_error(StatusCode::BAD_REQUEST);
// }
// };
// debug!("Response from backend: {:?}", res_backend.status());
// Ok(res_backend)
// Forward request to
let mut res_backend = match self.forwarder.request(req_forwarded).await {
Ok(res) => res,
Err(e) => {
error!("Failed to get response from backend: {}", e);
return http_error(StatusCode::BAD_REQUEST);
}
};
debug!("Response from backend: {:?}", res_backend.status());
http_error(StatusCode::NOT_FOUND)
// TODO: Handle StatusCode::SWITCHING_PROTOCOLS
// Generate response to client
if generate_response_forwarded(&mut res_backend).is_ok() {
Ok(res_backend)
} else {
http_error(StatusCode::BAD_GATEWAY)
}
}
}
fn generate_response_forwarded<B: core::fmt::Debug>(response: &mut Response<B>) -> Result<()> {
let headers = response.headers_mut();
remove_hop_header(headers);
remove_connection_header(headers);
Ok(())
}
fn generate_request_forwarded<B: core::fmt::Debug>(
client_addr: SocketAddr,
mut req: Request<B>,