robin/warp-real-ip
1
0
Fork 0
mirror of https://github.com/icewind1991/warp-real-ip.git synced 2026-06-03 18:54:06 +02:00
Code Issues Projects Releases Wiki Activity

add support for rfc7239 forwarded

Browse source
This commit is contained in:
Robin Appelman 2021-01-06 14:52:52 +01:00
commit d573c92d26
4 changed files with 73 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/tests.rs
View file

@ -61,3 +61,25 @@ async fn test_nested_allowed() {
.await;
assert_eq!(res.body(), "11.11.11.11");
}
#[tokio::test]
async fn test_trusted_forwarded() {
let remote: IpAddr = [1, 2, 3, 4].into();
let res = warp::test::request()
.remote_addr((remote, 80).into())
.header("forwarded", "for=10.10.10.10")
.reply(&serve(vec![remote]))
.await;
assert_eq!(res.body(), "10.10.10.10");
}
#[tokio::test]
async fn test_trusted_forwarded_no_for() {
let remote: IpAddr = [1, 2, 3, 4].into();
let res = warp::test::request()
.remote_addr((remote, 80).into())
.header("forwarded", "by=11.11.11.11")
.reply(&serve(vec![remote]))
.await;
assert_eq!(res.body(), "1.2.3.4");
}