pierre/BasicOIDC
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improve some code

Browse Source
This commit is contained in:
Pierre HUBERT 2022-04-03 17:44:19 +02:00
parent 48f4d4c6c5
commit 188b4f836d
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/utils/network_utils.rs
View File

@ -25,16 +25,13 @@ pub fn get_remote_ip(req: &HttpRequest, proxy_ip: Option<&str>) -> String {
if let Some(proxy) = proxy_ip.as_ref() { if let Some(proxy) = proxy_ip.as_ref() {
if match_ip(proxy, &ip) { if match_ip(proxy, &ip) {
if let Some(header) = req.headers().get("X-Forwarded-For") { if let Some(header) = req.headers().get("X-Forwarded-For") {
let header: Vec<String> = header let header = header
.to_str() .to_str()
.unwrap() .unwrap()
.to_string() .split_once(',');
.split(',')
.map(|f| f.to_string())
.collect();
if !header.is_empty() { if let Some((upstream_ip, _)) = header {
ip = header[0].to_string(); ip = upstream_ip.to_string();
} }
} }
} }