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

fix: add souce header to log

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

20
src/main.rs
View File

@ -8,7 +8,6 @@ use axum::{
}; };
use http::Method; use http::Method;
use rand::seq::SliceRandom; use rand::seq::SliceRandom;
use std::net::SocketAddr;
use tower_http::cors::{Any, CorsLayer}; use tower_http::cors::{Any, CorsLayer};
use tower_http::services::ServeDir; use tower_http::services::ServeDir;
use tracing::info; use tracing::info;
@ -35,23 +34,26 @@ 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 mut head = "x-forwarded-for";
let ip = req let ip = req
.headers() .headers()
.get("x-forwarded-for") .get(head)
.and_then(|hv| hv.to_str().ok()) .and_then(|hv| hv.to_str().ok())
.or_else(|| { .or_else(|| {
req.headers() head = "x-real-ip";
.get("x-real-ip") req.headers().get(head).and_then(|hv| hv.to_str().ok())
.and_then(|hv| hv.to_str().ok())
}) })
.or_else(|| { .or_else(|| {
req.headers() head = "cf-connection-ip";
.get("cf-connecting-ip") req.headers().get(head).and_then(|hv| hv.to_str().ok())
.and_then(|hv| hv.to_str().ok())
}) })
.unwrap_or("Unknown"); .unwrap_or("Unknown");
info!(PATH = req.uri().path().to_string(), IP = ip); if format!("{ip}") == "Unknown" {
head = "none"
}
info!(PATH = req.uri().path().to_string(), IP = ip, HEADER = head);
next.run(req).await next.run(req).await
} }