frameassembly 1.2.2

Declarative Rust DSL and compiler to generate deterministic .pcap/.pcapng packet captures across 81 L2-L7 protocols for Wireshark, IDS & CI testing.
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

//! Advanced Message Queuing Protocol v0-9-1 (AMQP 0-9-1) wire payload builder.
//!
//! This module implements [`build_amqp_payload`], which serializes an [`AmqpSpec`]
//! AST node into an AMQP 0-9-1 wire frame over TCP port 5672 (or 5671 for AMQPS). Depending on the
//! selected flags, it emits the 8-byte `AMQP\x00\x00\x09\x01` protocol header, a Type 8 Heartbeat
//! frame, or a Type 1 Method frame (1-byte type, 16-bit channel, 32-bit size, 16-bit Class ID,
//! 16-bit Method ID, short-string exchange/routing-key arguments, and trailing `0xCE` frame-end
//! marker) for `Connection.Start`, `Channel.Open`, `Basic.Publish`, and `Basic.Deliver`.

use crate::ast::*;
use crate::builders::WireContext;
use alloc::vec::Vec;

/// Builds an AMQP 0-9-1 protocol header or frame payload.
pub fn build_amqp_payload(
    _ctx: &WireContext,
    spec: &AmqpSpec,
    payload_bytes: &[u8],
) -> Vec<u8> {
    let has_method_or_frame = spec.is_connection_start
        || spec.is_channel_open
        || spec.is_basic_publish
        || spec.is_basic_deliver
        || spec.is_heartbeat
        || spec.channel.is_some()
        || spec.exchange.is_some()
        || spec.routing_key.is_some();

    if spec.is_protocol_header || !has_method_or_frame {
        let mut pkt = b"AMQP\x00\x00\x09\x01".to_vec();
        if !payload_bytes.is_empty() {
            pkt.extend_from_slice(payload_bytes);
        }
        return pkt;
    }

    if spec.is_heartbeat {
        let channel = spec.channel.unwrap_or(0);
        let size = payload_bytes.len() as u32;
        let mut pkt = Vec::with_capacity(8 + payload_bytes.len());
        pkt.push(8); // Frame Type 8: Heartbeat
        pkt.extend_from_slice(&channel.to_be_bytes());
        pkt.extend_from_slice(&size.to_be_bytes());
        if !payload_bytes.is_empty() {
            pkt.extend_from_slice(payload_bytes);
        }
        pkt.push(0xCE); // Frame-End octet
        return pkt;
    }

    let (class_id, method_id, default_channel): (u16, u16, u16) = if spec.is_connection_start {
        (10, 10, 0)
    } else if spec.is_channel_open {
        (20, 10, 1)
    } else if spec.is_basic_deliver {
        (60, 60, 1)
    } else if spec.is_basic_publish || spec.exchange.is_some() || spec.routing_key.is_some() {
        (60, 40, 1)
    } else {
        (20, 10, 1)
    };

    let channel = spec.channel.unwrap_or(default_channel);
    let mut body = Vec::new();
    body.extend_from_slice(&class_id.to_be_bytes());
    body.extend_from_slice(&method_id.to_be_bytes());

    if (class_id, method_id) == (10, 10) {
        // Connection.Start: version-major (0), version-minor (9), server-properties (0), mechanisms ("PLAIN"), locales ("en_US")
        body.push(0);
        body.push(9);
        body.extend_from_slice(&0u32.to_be_bytes());
        body.extend_from_slice(&5u32.to_be_bytes());
        body.extend_from_slice(b"PLAIN");
        body.extend_from_slice(&5u32.to_be_bytes());
        body.extend_from_slice(b"en_US");
    } else if (class_id, method_id) == (20, 10) {
        // Channel.Open: reserved-1 shortstr ("")
        body.push(0);
        if spec.exchange.is_some() || spec.routing_key.is_some() {
            let ex = spec.exchange.unwrap_or("").as_bytes();
            let rk = spec.routing_key.unwrap_or("").as_bytes();
            body.push(ex.len() as u8);
            body.extend_from_slice(ex);
            body.push(rk.len() as u8);
            body.extend_from_slice(rk);
        }
    } else if (class_id, method_id) == (60, 40) {
        // Basic.Publish: ticket (0u16), exchange (shortstr), routing-key (shortstr), mandatory/immediate (0u8)
        body.extend_from_slice(&0u16.to_be_bytes());
        let ex = spec.exchange.unwrap_or("").as_bytes();
        let rk = spec.routing_key.unwrap_or("").as_bytes();
        body.push(ex.len() as u8);
        body.extend_from_slice(ex);
        body.push(rk.len() as u8);
        body.extend_from_slice(rk);
        body.push(0);
    } else if (class_id, method_id) == (60, 60) {
        // Basic.Deliver: consumer-tag (shortstr), delivery-tag (u64), redelivered (u8), exchange (shortstr), routing-key (shortstr)
        let ctag = b"ctag";
        body.push(ctag.len() as u8);
        body.extend_from_slice(ctag);
        body.extend_from_slice(&1u64.to_be_bytes());
        body.push(0);
        let ex = spec.exchange.unwrap_or("").as_bytes();
        let rk = spec.routing_key.unwrap_or("").as_bytes();
        body.push(ex.len() as u8);
        body.extend_from_slice(ex);
        body.push(rk.len() as u8);
        body.extend_from_slice(rk);
    }

    if !payload_bytes.is_empty() {
        body.extend_from_slice(payload_bytes);
    }

    let size = body.len() as u32;
    let mut pkt = Vec::with_capacity(8 + body.len());
    pkt.push(1); // Frame Type 1: Method
    pkt.extend_from_slice(&channel.to_be_bytes());
    pkt.extend_from_slice(&size.to_be_bytes());
    pkt.extend_from_slice(&body);
    pkt.push(0xCE); // Frame-End octet
    pkt
}