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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use std::collections::HashMap;

use crate::{
    hist::{Body, Ent, HistoryEvent},
    Scribe,
};

use super::Hist;

#[tokio::test]
async fn test_record() {
    let hist = Hist::default();

    let empty = hist.entry(0);
    assert_eq!(empty, None);

    let req = hyper::Request::new(b"ping".to_vec());
    let res = hyper::Response::new(b"pong".to_vec());

    let hreq = super::Request {
        method: "GET".to_string(),
        path: "/".to_string(),
        query: HashMap::default(),
        version: "HTTP/1.1".to_string(),
        headers: HashMap::default(),
        body: Body::from(b"ping".to_vec()),
    };

    let hres = super::Response {
        status: 200,
        headers: HashMap::default(),
        body: Body::from(b"pong".to_vec()),
    };

    let id = hist.report_request(&req).await;

    let partial = hist.entry(0);
    assert_eq!(
        partial,
        Some(Ent {
            request: &hreq,
            response: None
        })
    );

    hist.report_response(id, &res).await;

    let full = hist.entry(0);

    assert_eq!(
        full,
        Some(Ent {
            request: &hreq,
            response: Some(&hres),
        })
    )
}

#[tokio::test]
async fn test_listen() {
    let hist = Hist::default();

    let req = hyper::Request::new(b"ping".to_vec());
    let res = hyper::Response::new(b"pong".to_vec());

    let mut listener = hist.listen();

    assert!(listener.try_recv().is_err());
    let id = hist.report_request(&req).await;
    assert_eq!(listener.try_recv(), Ok(HistoryEvent::Request { index: 0 }));

    assert!(listener.try_recv().is_err());
    hist.report_response(id, &res).await;
    assert_eq!(listener.try_recv(), Ok(HistoryEvent::Response { index: 0 }));
}

#[test]
fn test_binary_body() {
    let body = Body::from([1, 2, 3, 4].to_vec());

    let mut repr = Vec::new();
    body.hex(&mut repr);

    assert_eq!(repr, vec!["00000000  01 02 03 04".to_string()]);

    let body = Body::from(
        [
            0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
        ]
        .to_vec(),
    );

    let mut repr = Vec::new();
    body.hex(&mut repr);

    assert_eq!(
        repr,
        vec![
            "00000000  00 01 02 03 04 05 06 07  08 09 0A 0B 0C 0D 0E 0F".to_string(),
            "00000010  10 11 12 13 14".to_string()
        ]
    );
}

#[tokio::test]
async fn test_arbitrary_access() {
    let hist = Hist::default();

    let req = hyper::Request::new(b"ping".to_vec());
    let res = hyper::Response::new(b"pong".to_vec());

    let mut listener = hist.listen();

    assert!(listener.try_recv().is_err());
    let id = hist.report_request(&req).await;
    assert_eq!(listener.try_recv(), Ok(HistoryEvent::Request { index: 0 }));

    assert!(listener.try_recv().is_err());
    hist.report_response(id, &res).await;
    assert_eq!(listener.try_recv(), Ok(HistoryEvent::Response { index: 0 }));

    assert!(hist.request(0).is_some());
    assert!(hist.response(0).is_some());

    assert!(hist.request(1).is_none());
    assert!(hist.response(1).is_none());
}