itsscb/wordl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: prioritize client IP headers

Browse Source
This commit is contained in:
itsscb 2024-08-25 20:39:52 +02:00
parent 3d1dda6503
commit 23eb629fe8
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/main.rs
View File

@ -36,20 +36,20 @@ async fn word() -> String {
async fn log_ip(req: Request<Body>, next: Next) -> Response { async fn log_ip(req: Request<Body>, next: Next) -> Response {
let ip = req let ip = req
.extensions() .headers()
.get::<SocketAddr>() .get("x-forwarded-for")
.map(|addr| addr.ip().to_string()) .and_then(|hv| hv.to_str().ok())
.or_else(|| {
req.headers()
.get("x-forwarded-for")
.and_then(|hv| hv.to_str().ok().map(String::from))
})
.or_else(|| { .or_else(|| {
req.headers() req.headers()
.get("x-real-ip") .get("x-real-ip")
.and_then(|hv| hv.to_str().ok().map(String::from)) .and_then(|hv| hv.to_str().ok())
}) })
.unwrap_or_else(|| "Unknown".to_string()); .or_else(|| {
req.headers()
.get("cf-connecting-ip")
.and_then(|hv| hv.to_str().ok())
})
.unwrap_or("Unknown");
info!(PATH = req.uri().path().to_string(), IP = ip); info!(PATH = req.uri().path().to_string(), IP = ip);