dustin
/
jmap-client
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use IpAddr on forwardedfor()

main
Mauro D 2022-08-27 14:45:17 +00:00
parent 9f50a0f41c
commit 5e9c4f825e
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/client.rs
View File

@ -10,6 +10,7 @@
*/ */
use std::{ use std::{
net::IpAddr,
sync::{ sync::{
atomic::{AtomicBool, Ordering}, atomic::{AtomicBool, Ordering},
Arc, Arc,
@ -106,8 +107,11 @@ impl ClientBuilder {
self self
} }
pub fn forwarded_for(mut self, forwarded_for: impl ToString) -> Self { pub fn forwarded_for(mut self, forwarded_for: IpAddr) -> Self {
self.forwarded_for = Some(forwarded_for.to_string()); self.forwarded_for = Some(match forwarded_for {
IpAddr::V4(addr) => format!("for={}", addr),
IpAddr::V6(addr) => format!("for=\"{}\"", addr),
});
self self
} }
@ -129,7 +133,7 @@ impl ClientBuilder {
if let Some(forwarded_for) = self.forwarded_for { if let Some(forwarded_for) = self.forwarded_for {
headers.insert( headers.insert(
header::FORWARDED, header::FORWARDED,
header::HeaderValue::from_str(&format!("for={}", forwarded_for)).unwrap(), header::HeaderValue::from_str(&forwarded_for).unwrap(),
); );
} }
@ -215,7 +219,7 @@ impl ClientBuilder {
if let Some(forwarded_for) = self.forwarded_for { if let Some(forwarded_for) = self.forwarded_for {
headers.insert( headers.insert(
header::FORWARDED, header::FORWARDED,
header::HeaderValue::from_str(&format!("for={}", forwarded_for)).unwrap(), header::HeaderValue::from_str(&forwarded_for).unwrap(),
); );
} }