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

use serde::{Deserialize, Serialize};
use std::collections::HashMap;

mod body;
mod conv;
mod deser;
mod encoding;

#[cfg(test)]
mod test;

pub use body::Body;
pub use encoding::Encoding;
use tokio::sync::broadcast;

use crate::bind::{Req, Res, Scribe};

use crate::store::{Append, Random, Store};

#[derive(Deserialize, Serialize, Debug, Clone, PartialEq)]
pub struct Request {
    pub method: String,
    pub path: String,
    pub query: HashMap<String, String>,
    pub version: String,
    pub headers: HashMap<String, String>,
    pub body: Body,
}

#[derive(Deserialize, Serialize, Debug, Clone, PartialEq)]
pub struct Response {
    pub status: u16,
    pub headers: HashMap<String, String>,
    pub body: Body,
}

#[derive(Debug, PartialEq)]
pub struct Ent<'a> {
    pub request: &'a Request,
    pub response: Option<&'a Response>,
}

#[derive(Debug)]
pub struct Entry {
    pub request: Request,
    pub response: Option<Response>,
}

#[derive(Clone, PartialEq, Eq, Debug)]
pub enum HistoryEvent {
    Request { index: usize },
    Response { index: usize },
}

#[derive(Debug)]
pub struct Hist {
    requests: Store<Request, Append>,
    responses: Store<Response, Random>,

    events: broadcast::Sender<HistoryEvent>,
}

#[derive(Default, Debug)]
pub struct History(Vec<Entry>);

impl Scribe for Hist {
    type Ticket = usize;

    async fn report_request(&self, req: &Req<Vec<u8>>) -> usize {
        let req = Request::from(req);
        let index = self.requests.push(req);

        let _ = self.events.send(HistoryEvent::Request { index });

        index
    }

    async fn report_response(&self, index: Self::Ticket, res: &Res<Vec<u8>>) {
        let res = Response::from(res);

        if self.responses.insert(index, res) {
            let _ = self.events.send(HistoryEvent::Response { index });
        }
    }
}

impl Hist {
    pub fn entry(&self, index: usize) -> Option<Ent> {
        let request = self.requests.get(index)?;

        let response = self.responses.get(index);

        Some(Ent { request, response })
    }

    pub fn request(&self, index: usize) -> Option<&Request> {
        self.requests.get(index)
    }

    pub fn response(&self, index: usize) -> Option<&Response> {
        self.responses.get(index)
    }

    pub fn listen(&self) -> broadcast::Receiver<HistoryEvent> {
        self.events.subscribe()
    }
}

impl Default for Hist {
    fn default() -> Self {
        let requests = Store::<Request, Append>::default();
        let responses = Store::default();

        let events = broadcast::Sender::new(16);

        Hist {
            requests,
            responses,
            events,
        }
    }
}