frameassembly 1.2.0

Deterministic network packet crafting DSL and compiler synthesizing nanosecond-accurate PCAPNG captures across 81 L2-L7 protocols (Ethernet, VLAN, MPLS, ARP, IPv4, IPv6, ICMP, BGP, OSPF, WireGuard, IPsec, TCP, UDP, SCTP, DNS, HTTP/2, gRPC, TLS, QUIC, SNMPv3 USM, Modbus, DNP3, S7comm, OPC UA).
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

//! Modbus TCP Layer 7 industrial automation and SCADA register field parser.
//!
//! This module implements the `winnow` parser combinator for `MODBUS` Application Protocol (`MBAP`)
//! frames over TCP port 502 in FrameAssembly `.fasm` specifications. It recognizes the `MODBUS`
//! protocol keyword, MBAP transaction IDs (`TXID`), protocol IDs (`PROTO_ID`), slave unit IDs
//! (`UNITID`), explicit `FUNC` names, shorthand function keywords (`READ_COILS`,
//! `READ_DISCRETE_INPUTS`, `READ_HOLDING_REGISTERS`, `READ_INPUT_REGISTERS`, `WRITE_SINGLE_COIL`,
//! `WRITE_SINGLE_REGISTER`, `WRITE_MULTIPLE_REGISTERS`), starting addresses (`ADDR`), register
//! counts (`COUNT`), register values (`VALUE`), and `IS_RESPONSE` flags.

use winnow::ModalResult as PResult;
use winnow::combinator::{alt, cut_err};
use winnow::prelude::*;

use crate::ast::*;
use crate::lexer::Token;
use crate::parser::{parse_any_str, parse_u32, tag, ParsedField, Stream};

/// Parses a Modbus protocol declaration or register field.
pub(crate) fn parse_modbus_field<'i, 'a>(input: &mut Stream<'i, 'a>) -> PResult<ParsedField<'a>> {
    alt((
        alt((
            tag(Token::Modbus)
                .map(|(_, s)| ParsedField::Protocol(Protocol::Modbus, (s.start, s.len()))),
            tag(Token::ReadCoils)
                .map(|(_, s)| ParsedField::L7StringField("FUNC", "READ_COILS", (s.start, s.len()))),
            tag(Token::ReadDiscreteInputs).map(|(_, s)| {
                ParsedField::L7StringField("FUNC", "READ_DISCRETE_INPUTS", (s.start, s.len()))
            }),
            tag(Token::ReadHoldingRegisters).map(|(_, s)| {
                ParsedField::L7StringField("FUNC", "READ_HOLDING_REGISTERS", (s.start, s.len()))
            }),
            tag(Token::ReadInputRegisters).map(|(_, s)| {
                ParsedField::L7StringField("FUNC", "READ_INPUT_REGISTERS", (s.start, s.len()))
            }),
            tag(Token::WriteSingleCoil).map(|(_, s)| {
                ParsedField::L7StringField("FUNC", "WRITE_SINGLE_COIL", (s.start, s.len()))
            }),
            tag(Token::WriteSingleRegister).map(|(_, s)| {
                ParsedField::L7StringField("FUNC", "WRITE_SINGLE_REGISTER", (s.start, s.len()))
            }),
            tag(Token::WriteMultipleRegisters).map(|(_, s)| {
                ParsedField::L7StringField("FUNC", "WRITE_MULTIPLE_REGISTERS", (s.start, s.len()))
            }),
        )),
        alt((
            (tag(Token::Txid), cut_err(parse_u32))
                .map(|((_, s), v)| ParsedField::L7U32Field("TXID", v, (s.start, s.end - s.start))),
            (tag(Token::ProtoId), cut_err(parse_u32)).map(|((_, s), v)| {
                ParsedField::L7U32Field("PROTO_ID", v, (s.start, s.end - s.start))
            }),
            (tag(Token::UnitId), cut_err(parse_u32)).map(|((_, s), v)| {
                ParsedField::L7U32Field("UNITID", v, (s.start, s.end - s.start))
            }),
            (tag(Token::Func), parse_any_str)
                .map(|((_, s), v)| ParsedField::L7StringField("FUNC", v, (s.start, s.end - s.start))),
            (tag(Token::Addr), cut_err(parse_u32))
                .map(|((_, s), v)| ParsedField::L7U32Field("ADDR", v, (s.start, s.end - s.start))),
            (tag(Token::Count), cut_err(parse_u32))
                .map(|((_, s), v)| ParsedField::L7U32Field("COUNT", v, (s.start, s.end - s.start))),
            (tag(Token::Value), parse_u32)
                .map(|((_, s), v)| ParsedField::L7U32Field("VALUE", v, (s.start, s.end - s.start))),
            tag(Token::IsResponse)
                .map(|(_, s)| ParsedField::L7Flag("IS_RESPONSE", (s.start, s.len()))),
        )),
    ))
    .parse_next(input)
}