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

fix x-forwarded-for bug

Browse source
This commit is contained in:
Jun Kurihara 2022-07-07 21:47:05 +09:00
commit 98fce0e4df
No known key found for this signature in database
GPG key ID: 48ADFD173ED22B03
3 changed files with 23 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/proxy/proxy_handler.rs
View file

@ -212,6 +212,7 @@ fn generate_request_forwarded<B: core::fmt::Debug>(
remove_hop_header(headers);
// X-Forwarded-For
add_forwarding_header(headers, client_addr)?;
println!("{:?}", headers);
// Add te: trailer if te_trailer
if te_trailers {
@ -285,7 +286,7 @@ fn add_forwarding_header(headers: &mut HeaderMap, client_addr: SocketAddr) -> Re
hyper::header::Entry::Vacant(entry) => {
entry.insert(client_ip.to_string().parse()?);
}
hyper::header::Entry::Occupied(entry) => {
hyper::header::Entry::Occupied(mut entry) => {
let client_ip_str = client_ip.to_string();
let mut addr = String::with_capacity(entry.get().as_bytes().len() + 2 + client_ip_str.len());
@ -293,6 +294,7 @@ fn add_forwarding_header(headers: &mut HeaderMap, client_addr: SocketAddr) -> Re
addr.push(',');
addr.push(' ');
addr.push_str(&client_ip_str);
entry.insert(addr.to_owned().parse()?);
}
}
Ok(())