]> git.ipfire.org Git - thirdparty/suricata.git/commitdiff
rust/smb: simplify bits parser annotations
authorPierre Chifflier <chifflier@wzdftpd.net>
Tue, 18 Jan 2022 21:21:07 +0000 (22:21 +0100)
committerPierre Chifflier <chifflier@wzdftpd.net>
Wed, 19 Jan 2022 08:13:11 +0000 (09:13 +0100)
rust/src/smb/dcerpc_records.rs
rust/src/smb/ntlmssp_records.rs
rust/src/smb/smb2_records.rs

index 4cfd40c567f59b606d20023f6c086addee6a094d..98b072986a7f89b657df2ce3da16a0182d986346 100644 (file)
  * 02110-1301, USA.
  */
 
+use crate::common::nom7::bits;
 use crate::smb::error::SmbError;
-use nom7::bits::{bits, streaming::take as take_bits};
+use nom7::bits::streaming::take as take_bits;
 use nom7::bytes::streaming::take;
 use nom7::combinator::{cond, rest};
-use nom7::error::Error;
 use nom7::multi::count;
 use nom7::number::Endianness;
 use nom7::number::streaming::{be_u16, le_u8, le_u16, le_u32, u16, u32};
@@ -204,7 +204,7 @@ pub struct DceRpcRecord<'a> {
 }
 
 fn parse_dcerpc_flags1(i:&[u8]) -> IResult<&[u8],(u8,u8,u8)> {
-    bits::<_, _, Error<(&[u8], usize)>, _, _>(tuple((
+    bits(tuple((
         take_bits(6u8),
         take_bits(1u8),   // last (1)
         take_bits(1u8),
@@ -212,7 +212,7 @@ fn parse_dcerpc_flags1(i:&[u8]) -> IResult<&[u8],(u8,u8,u8)> {
 }
 
 fn parse_dcerpc_flags2(i:&[u8]) -> IResult<&[u8],(u32,u32,u32)> {
-    bits::<_, _, Error<(&[u8], usize)>, _, _>(tuple((
+    bits(tuple((
        take_bits(3u32),
        take_bits(1u32),     // endianess
        take_bits(28u32),
index 71500812c6daa771da6a19bd15e1cd10d8f56094..35ca3196993e2351eb73d108c18a10c7d9f2a2ae 100644 (file)
  * 02110-1301, USA.
  */
 
-use crate::common::nom7::take_until_and_consume;
+use crate::common::nom7::{bits, take_until_and_consume};
 use std::fmt;
-use nom7::bits::{bits, streaming::take as take_bits};
+use nom7::bits::streaming::take as take_bits;
 use nom7::bytes::streaming::take;
 use nom7::combinator::{cond, rest};
-use nom7::error::Error;
 use nom7::number::streaming::{le_u8, le_u16, le_u32};
 use nom7::sequence::tuple;
 use nom7::IResult;
@@ -65,7 +64,7 @@ pub struct NTLMSSPAuthRecord<'a> {
 }
 
 fn parse_ntlm_auth_nego_flags(i:&[u8]) -> IResult<&[u8],(u8,u8,u32)> {
-    bits::<_, _, Error<(&[u8], usize)>, _, _>(tuple((
+    bits(tuple((
         take_bits(6u8),
         take_bits(1u8),
         take_bits(25u32),
index 5e5f1da44862e87e7a2e85b1077329ac5ed51530..050111537be1b563b41e302d0b87784a519d2166 100644 (file)
  * 02110-1301, USA.
  */
 
+use crate::common::nom7::bits;
 use crate::smb::smb::*;
 use crate::smb::nbss_records::NBSS_MSGTYPE_SESSION_MESSAGE;
-use nom7::bits::{bits, streaming::take as take_bits};
+use nom7::bits::streaming::take as take_bits;
 use nom7::bytes::streaming::{tag, take};
 use nom7::combinator::{cond, map_parser, rest};
-use nom7::error::{make_error, Error, ErrorKind};
+use nom7::error::{make_error, ErrorKind};
 use nom7::multi::count;
 use nom7::number::streaming::{le_u8, le_u16, le_u32, le_u64};
 use nom7::sequence::tuple;
@@ -71,7 +72,7 @@ impl<'a> Smb2Record<'a> {
 }
 
 fn parse_smb2_request_flags(i:&[u8]) -> IResult<&[u8],(u8,u8,u8,u32,u8,u8,u8,u8)> {
-    bits::<_, _, Error<(&[u8], usize)>, _, _>(tuple((
+    bits(tuple((
         take_bits(2u8),      // reserved / unused
         take_bits(1u8),      // replay op
         take_bits(1u8),      // dfs op