]> git.ipfire.org Git - thirdparty/suricata.git/commitdiff
rust/mqtt: convert to nom 8
authorJason Ish <jason.ish@oisf.net>
Tue, 28 Oct 2025 06:28:49 +0000 (00:28 -0600)
committerJason Ish <jason.ish@oisf.net>
Thu, 30 Oct 2025 15:15:41 +0000 (09:15 -0600)
Ticket: #8047

rust/src/mqtt/mqtt.rs
rust/src/mqtt/mqtt_property.rs
rust/src/mqtt/parser.rs

index a38ecf7985f89937dce7a73ea563d05cfb1e4461..4dfe05f994292589adf2cb88a3c888c4ebf219a7 100644 (file)
@@ -26,7 +26,7 @@ use crate::core::*;
 use crate::direction::Direction;
 use crate::flow::Flow;
 use crate::frames::*;
-use nom7::Err;
+use nom8::Err;
 use std;
 use std::collections::VecDeque;
 use std::ffi::CString;
index 432db5aafc82475bc40e10f0afc38a85df1c7c98..93bbcbe70ba17b10cfdc000cd3fc83e6219e32bb 100644 (file)
@@ -19,8 +19,8 @@
 
 use crate::jsonbuilder::{JsonBuilder, JsonError};
 use crate::mqtt::parser::*;
-use nom7::number::streaming::*;
-use nom7::IResult;
+use nom8::number::streaming::*;
+use nom8::IResult;
 
 // TODO: It might be useful to also add detection on property presence and
 // content, e.g. mqtt.property: AUTHENTICATION_METHOD.
index 8da541eaf58eaaf146b02fdfc7f350368c1de8fa..fdd4a83dbdb20f29fdc625a940a2c3a62c30cc9c 100644 (file)
 
 // written by Sascha Steinbiss <sascha@steinbiss.name>
 
-use crate::common::nom7::bits;
+use crate::common::nom8::bits;
 use crate::mqtt::mqtt_message::*;
 use crate::mqtt::mqtt_property::*;
-use nom7::bits::streaming::take as take_bits;
-use nom7::bytes::complete::take;
-use nom7::bytes::streaming::take_while_m_n;
-use nom7::combinator::{complete, cond, verify};
-use nom7::multi::{length_data, many0, many1};
-use nom7::number::streaming::*;
-use nom7::sequence::tuple;
-use nom7::{Err, IResult, Needed};
+use nom8::bits::streaming::take as take_bits;
+use nom8::bytes::complete::take;
+use nom8::bytes::streaming::take_while_m_n;
+use nom8::combinator::{complete, cond, verify};
+use nom8::multi::{length_data, many0, many1};
+use nom8::number::streaming::*;
+use nom8::{Err, IResult, Needed, Parser};
 use num_traits::FromPrimitive;
 
 #[derive(Copy, Clone, Debug)]
@@ -62,14 +61,14 @@ fn convert_varint(continued: Vec<u8>, last: u8) -> u32 {
 
 #[inline]
 pub fn parse_mqtt_string(i: &[u8]) -> IResult<&[u8], String> {
-    let (i, content) = length_data(be_u16)(i)?;
+    let (i, content) = length_data(be_u16).parse(i)?;
     Ok((i, String::from_utf8_lossy(content).to_string()))
 }
 
 #[inline]
 pub fn parse_mqtt_variable_integer(i: &[u8]) -> IResult<&[u8], u32> {
-    let (i, continued_part) = take_while_m_n(0, 3, is_continuation_bit_set)(i)?;
-    let (i, non_continued_part) = verify(be_u8, |&val| !is_continuation_bit_set(val))(i)?;
+    let (i, continued_part) = take_while_m_n(0, 3, is_continuation_bit_set).parse(i)?;
+    let (i, non_continued_part) = verify(be_u8, |&val| !is_continuation_bit_set(val)).parse(i)?;
     Ok((
         i,
         convert_varint(continued_part.to_vec(), non_continued_part),
@@ -78,7 +77,7 @@ pub fn parse_mqtt_variable_integer(i: &[u8]) -> IResult<&[u8], u32> {
 
 #[inline]
 pub fn parse_mqtt_binary_data(i: &[u8]) -> IResult<&[u8], Vec<u8>> {
-    let (i, data) = length_data(be_u16)(i)?;
+    let (i, data) = length_data(be_u16).parse(i)?;
     Ok((i, data.to_vec()))
 }
 
@@ -131,12 +130,14 @@ fn parse_properties(input: &[u8], precond: bool) -> IResult<&[u8], Option<Vec<MQ
 
 #[inline]
 fn parse_fixed_header_flags(i: &[u8]) -> IResult<&[u8], (u8, u8, u8, u8)> {
-    bits(tuple((
-        take_bits(4u8),
-        take_bits(1u8),
-        take_bits(2u8),
-        take_bits(1u8),
-    )))(i)
+    bits(|input| {
+        (
+            take_bits(4u8),
+            take_bits(1u8),
+            take_bits(2u8),
+            take_bits(1u8),
+        ).parse(input)
+    })(i)
 }
 
 #[inline]
@@ -176,16 +177,16 @@ pub fn parse_fixed_header(i: &[u8]) -> IResult<&[u8], FixedHeader> {
 #[inline]
 fn parse_connect(i: &[u8]) -> IResult<&[u8], MQTTConnectData> {
     let (i, protocol_string) = parse_mqtt_string(i)?;
-    let (i, protocol_version) = be_u8(i)?;
-    let (i, rawflags) = be_u8(i)?;
-    let (i, keepalive) = be_u16(i)?;
+    let (i, protocol_version) = be_u8.parse(i)?;
+    let (i, rawflags) = be_u8.parse(i)?;
+    let (i, keepalive) = be_u16.parse(i)?;
     let (i, properties) = parse_properties(i, protocol_version == 5)?;
     let (i, client_id) = parse_mqtt_string(i)?;
     let (i, will_properties) = parse_properties(i, protocol_version == 5 && rawflags & 0x4 != 0)?;
-    let (i, will_topic) = cond(rawflags & 0x4 != 0, parse_mqtt_string)(i)?;
-    let (i, will_message) = cond(rawflags & 0x4 != 0, parse_mqtt_binary_data)(i)?;
-    let (i, username) = cond(rawflags & 0x80 != 0, parse_mqtt_string)(i)?;
-    let (i, password) = cond(rawflags & 0x40 != 0, parse_mqtt_binary_data)(i)?;
+    let (i, will_topic) = cond(rawflags & 0x4 != 0, parse_mqtt_string).parse(i)?;
+    let (i, will_message) = cond(rawflags & 0x4 != 0, parse_mqtt_binary_data).parse(i)?;
+    let (i, username) = cond(rawflags & 0x80 != 0, parse_mqtt_string).parse(i)?;
+    let (i, password) = cond(rawflags & 0x40 != 0, parse_mqtt_binary_data).parse(i)?;
     Ok((
         i,
         MQTTConnectData {
@@ -233,7 +234,7 @@ fn parse_publish(
 ) -> impl Fn(&[u8]) -> IResult<&[u8], MQTTPublishData> {
     move |i: &[u8]| {
         let (i, topic) = parse_mqtt_string(i)?;
-        let (i, message_id) = cond(has_id, be_u16)(i)?;
+        let (i, message_id) = cond(has_id, be_u16).parse(i)?;
         let (message, properties) = parse_properties(i, protocol_version == 5)?;
         Ok((
             i,
@@ -332,9 +333,9 @@ fn parse_subscribe_topic(i: &[u8]) -> IResult<&[u8], MQTTSubscribeTopicData> {
 #[inline]
 fn parse_subscribe(protocol_version: u8) -> impl Fn(&[u8]) -> IResult<&[u8], MQTTSubscribeData> {
     move |i: &[u8]| {
-        let (i, message_id) = be_u16(i)?;
+        let (i, message_id) = be_u16.parse(i)?;
         let (i, properties) = parse_properties(i, protocol_version == 5)?;
-        let (i, topics) = many1(complete(parse_subscribe_topic))(i)?;
+        let (i, topics) = many1(complete(parse_subscribe_topic)).parse(i)?;
         Ok((
             i,
             MQTTSubscribeData {
@@ -367,9 +368,9 @@ fn parse_unsubscribe(
     protocol_version: u8,
 ) -> impl Fn(&[u8]) -> IResult<&[u8], MQTTUnsubscribeData> {
     move |i: &[u8]| {
-        let (i, message_id) = be_u16(i)?;
+        let (i, message_id) = be_u16.parse(i)?;
         let (i, properties) = parse_properties(i, protocol_version == 5)?;
-        let (i, topics) = many0(complete(parse_mqtt_string))(i)?;
+        let (i, topics) = many0(complete(parse_mqtt_string)).parse(i)?;
         Ok((
             i,
             MQTTUnsubscribeData {
@@ -384,9 +385,9 @@ fn parse_unsubscribe(
 #[inline]
 fn parse_unsuback(protocol_version: u8) -> impl Fn(&[u8]) -> IResult<&[u8], MQTTUnsubackData> {
     move |i: &[u8]| {
-        let (i, message_id) = be_u16(i)?;
+        let (i, message_id) = be_u16.parse(i)?;
         let (i, properties) = parse_properties(i, protocol_version == 5)?;
-        let (i, reason_codes) = many0(complete(be_u8))(i)?;
+        let (i, reason_codes) = many0(complete(be_u8)).parse(i)?;
         Ok((
             i,
             MQTTUnsubackData {
@@ -670,7 +671,7 @@ pub fn parse_message(
                 header,
                 message_type,
                 protocol_version,
-            ))(rem);
+            )).parse(rem);
         }
         Err(err) => {
             return Err(err);
@@ -681,7 +682,7 @@ pub fn parse_message(
 #[cfg(test)]
 mod tests {
     use super::*;
-    use nom7::error::ErrorKind;
+    use nom8::error::ErrorKind;
 
     fn test_mqtt_parse_variable_fail(buf0: &[u8]) {
         let r0 = parse_mqtt_variable_integer(buf0);