wrong header

Author: hry-gh

src/admin.rs

@@ -32,7 +32,7 @@ impl<'r> FromRequest<'r> for Admin {
         };
 
         let mut ipv4_address = None;
-        for header in req.headers().get("X-Forwarded-For") {
+        for header in req.headers().get("X-Real-IP") {
             if header.contains('.') {
                 ipv4_address = Some(header.to_string());
                 break;

src/twofactor.rs

@@ -11,7 +11,7 @@ pub async fn twofactor_validate(mut db: Connection<Cmdb>, admin: Admin, cid: Str
     };
 
     match query("UPDATE twofactor SET approved = 1 WHERE cid = ? AND ckey = ? AND ip = ?").bind(cid).bind(&admin.username).bind(ip_addr).execute(&mut **db).await {
-        Ok(res) => if res.rows_affected() > 0 { "Two factor request updated.".to_string() } else { "An error occured.".to_string() },
+        Ok(res) => if res.rows_affected() > 0 { "Two factor request updated.".to_string() } else { format!("An error occured: {}.", &ip_addr) },
         Err(err) => format!("An error occured: {:?}", err).to_string(),
     }