62 lines
1.6 KiB
Rust
62 lines
1.6 KiB
Rust
use axum::{
|
|
extract::{Request, State},
|
|
http::{HeaderValue, header},
|
|
middleware::Next,
|
|
response::Response,
|
|
};
|
|
|
|
#[derive(Clone)]
|
|
pub struct CsrfToken(pub String);
|
|
|
|
pub fn extract_from_cookie(headers: &axum::http::HeaderMap) -> Option<String> {
|
|
headers
|
|
.get(header::COOKIE)
|
|
.and_then(|v| v.to_str().ok())
|
|
.and_then(|cookies| {
|
|
cookies
|
|
.split(';')
|
|
.find_map(|c| c.trim().strip_prefix("csrf=").map(str::to_string))
|
|
})
|
|
}
|
|
|
|
/// Router state for [`csrf_middleware`]: whether cookies carry the `Secure` attribute.
|
|
#[derive(Clone, Copy)]
|
|
pub struct SecureCookies(pub bool);
|
|
|
|
pub(crate) fn secure_flag(secure_cookies: bool) -> &'static str {
|
|
if secure_cookies { "; Secure" } else { "" }
|
|
}
|
|
|
|
pub async fn csrf_middleware(
|
|
State(SecureCookies(secure_cookies)): State<SecureCookies>,
|
|
mut req: Request,
|
|
next: Next,
|
|
) -> Response {
|
|
let existing = extract_from_cookie(req.headers());
|
|
let (token, needs_set) = match existing {
|
|
Some(t) => (t, false),
|
|
None => (uuid::Uuid::new_v4().to_string(), true),
|
|
};
|
|
|
|
req.extensions_mut().insert(CsrfToken(token.clone()));
|
|
|
|
let mut response = next.run(req).await;
|
|
|
|
if needs_set {
|
|
let cookie = format!(
|
|
"csrf={}; HttpOnly; Path=/; SameSite=Strict{}",
|
|
token,
|
|
secure_flag(secure_cookies)
|
|
);
|
|
if let Ok(val) = HeaderValue::from_str(&cookie) {
|
|
response.headers_mut().append(header::SET_COOKIE, val);
|
|
}
|
|
}
|
|
|
|
response
|
|
}
|
|
|
|
pub fn mismatch(token: &CsrfToken, form_value: &str) -> bool {
|
|
token.0 != form_value || form_value.is_empty()
|
|
}
|