prax 0.1.1

a web penetration proxy hosted in neovim
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use futures::Future;

use super::{Req, Res};

/// A trait for modifying in flight requests
pub trait Filter {
    fn modify_request(
        &self,
        hostname: &str,
        req: &mut Req<Vec<u8>>,
    ) -> impl Future<Output = crate::Result<()>> + Send;

    fn modify_response(
        &self,
        hostname: &str,
        req: &mut Res<Vec<u8>>,
    ) -> impl Future<Output = crate::Result<()>> + Send;
}

impl Filter for () {
    async fn modify_request(&self, _: &str, _: &mut super::Req<Vec<u8>>) -> crate::Result<()> {
        Ok(())
    }

    async fn modify_response(&self, _: &str, _: &mut super::Res<Vec<u8>>) -> crate::Result<()> {
        Ok(())
    }
}

#[tokio::test]
async fn test_null_filter() {
    let mut req = Req::builder()
        .method("GET")
        .header("Host", "example.com")
        .body(Vec::new())
        .unwrap();
    let init_req = req.clone();

    let mut res = Res::builder().status(200).body(Vec::new()).unwrap();
    let init_res = res.clone();

    ().modify_request("example.com", &mut req).await.unwrap();
    ().modify_response("example.com", &mut res).await.unwrap();

    assert_eq!(req.uri(), init_req.uri());
    assert_eq!(req.headers(), init_req.headers());
    assert_eq!(req.body(), init_req.body());

    assert_eq!(res.status(), init_res.status());
    assert_eq!(res.headers(), init_res.headers());
    assert_eq!(res.body(), init_res.body());
}