axum_csrf 0.11.0

Library to Provide a CSRF (Cross-Site Request Forgery) protection layer.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use crate::{cookies::*, CsrfConfig, CsrfToken};
use http::Request;
use std::task::{Context, Poll};
use tower_service::Service;

#[derive(Clone)]
pub struct AxumCsrfService<S> {
    pub(crate) config: CsrfConfig,
    pub(crate) inner: S,
}

impl<ResBody, S> Service<Request<ResBody>> for AxumCsrfService<S>
where
    S: Service<Request<ResBody>>,
{
    type Response = S::Response;
    type Error = S::Error;
    type Future = S::Future;

    #[inline]
    fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
        self.inner.poll_ready(cx)
    }

    fn call(&mut self, mut req: Request<ResBody>) -> Self::Future {
        let config = self.config.clone();
        let token = get_token(&config, req.headers_mut());

        req.extensions_mut().insert(CsrfToken { token, config });
        self.inner.call(req)
    }
}