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

fix: change set-upstream-host as a default option

Browse source
This commit is contained in:
Jun Kurihara 2024-01-10 14:44:51 +09:00
commit 3fcea9a28c
No known key found for this signature in database
GPG key ID: 6D3FEE70E498C15B
5 changed files with 23 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

9
rpxy-lib/src/message_handler/handler_manipulate_messages.rs
View file

@ -99,14 +99,14 @@ where
headers.insert(header::TE, HeaderValue::from_bytes("trailers".as_bytes()).unwrap());
}
// add "host" header of original server_name if not exist (default)
// by default, add "host" header of original server_name if not exist
if req.headers().get(header::HOST).is_none() {
let org_host = req.uri().host().ok_or_else(|| anyhow!("Invalid request"))?.to_owned();
req
.headers_mut()
.insert(header::HOST, HeaderValue::from_str(&org_host)?);
};
let original_host_header = req.headers().get(header::HOST).unwrap().clone();
println!("{:?}", req.headers().get(header::HOST));
/////////////////////////////////////////////
// Fix unique upstream destination since there could be multiple ones.
@ -133,10 +133,8 @@ where
// apply upstream-specific headers given in upstream_option
let headers = req.headers_mut();
// by default, host header is overwritten with upstream hostname
override_host_header(headers, &upstream_chosen.uri)?;
// apply upstream options to header
apply_upstream_options_to_header(headers, &original_host_header, upstream_candidates)?;
apply_upstream_options_to_header(headers, &upstream_chosen.uri, upstream_candidates)?;
// update uri in request
ensure!(
@ -182,7 +180,6 @@ where
update_request_line(req, upstream_chosen, upstream_candidates)?;
}
Ok(context)
}
}