From e2c846d01fd9ae8f05aa8b2014cc0d736c013cec Mon Sep 17 00:00:00 2001 From: Jason Ish Date: Tue, 26 Nov 2019 16:08:18 -0600 Subject: [PATCH] snmp: add tx detect flags --- rust/src/snmp/snmp.rs | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/rust/src/snmp/snmp.rs b/rust/src/snmp/snmp.rs index ebd4d26641..a3c5307c28 100644 --- a/rust/src/snmp/snmp.rs +++ b/rust/src/snmp/snmp.rs @@ -97,6 +97,7 @@ pub struct SNMPTransaction { events: *mut core::AppLayerDecoderEvents, logged: applayer::LoggerFlags, + detect_flags: applayer::TxDetectFlags, } @@ -274,6 +275,7 @@ impl SNMPTransaction { de_state: None, events: std::ptr::null_mut(), logged: applayer::LoggerFlags::new(), + detect_flags: applayer::TxDetectFlags::default(), } } @@ -572,6 +574,9 @@ pub extern "C" fn rs_snmp_probing_parser(_flow: *const Flow, } } +export_tx_detect_flags_set!(rs_snmp_set_tx_detect_flags, SNMPTransaction); +export_tx_detect_flags_get!(rs_snmp_get_tx_detect_flags, SNMPTransaction); + const PARSER_NAME : &'static [u8] = b"snmp\0"; #[no_mangle] @@ -607,8 +612,8 @@ pub unsafe extern "C" fn rs_register_snmp_parser() { set_tx_mpm_id : None, get_files : None, get_tx_iterator : None, - get_tx_detect_flags: None, - set_tx_detect_flags: None, + get_tx_detect_flags: Some(rs_snmp_get_tx_detect_flags), + set_tx_detect_flags: Some(rs_snmp_set_tx_detect_flags), }; let ip_proto_str = CString::new("udp").unwrap(); if AppLayerProtoDetectConfProtoDetectionEnabled(ip_proto_str.as_ptr(), parser.name) != 0 {