}
}
+impl Transaction for MQTTTransaction {
+ fn id(&self) -> u64 {
+ self.tx_id
+ }
+}
+
pub struct MQTTState {
tx_id: u64,
pub protocol_version: u8,
max_msg_len: usize,
}
+impl State<MQTTTransaction> for MQTTState {
+ fn get_transactions(&self) -> &[MQTTTransaction] {
+ &self.transactions
+ }
+}
+
impl MQTTState {
pub fn new() -> Self {
Self {
let ev = event as u8;
core::sc_app_layer_decoder_events_set_event_raw(&mut tx.events, ev);
}
-
- fn tx_iterator(
- &mut self,
- min_tx_id: u64,
- state: &mut u64,
- ) -> Option<(&MQTTTransaction, u64, bool)> {
- let mut index = *state as usize;
- let len = self.transactions.len();
-
- while index < len {
- let tx = &self.transactions[index];
- if tx.tx_id < min_tx_id + 1 {
- index += 1;
- continue;
- }
- *state = index as u64;
- return Some((tx, tx.tx_id - 1, (len - index) > 1));
- }
-
- return None;
- }
}
// C exports.
return tx.events;
}
-#[no_mangle]
-pub unsafe extern "C" fn rs_mqtt_state_get_tx_iterator(
- _ipproto: u8,
- _alproto: AppProto,
- state: *mut std::os::raw::c_void,
- min_tx_id: u64,
- _max_tx_id: u64,
- istate: &mut u64,
-) -> applayer::AppLayerGetTxIterTuple {
- let state = cast_pointer!(state, MQTTState);
- match state.tx_iterator(min_tx_id, istate) {
- Some((tx, out_tx_id, has_next)) => {
- let c_tx = tx as *const _ as *mut _;
- let ires = applayer::AppLayerGetTxIterTuple::with_values(c_tx, out_tx_id, has_next);
- return ires;
- }
- None => {
- return applayer::AppLayerGetTxIterTuple::not_found();
- }
- }
-}
-
// Parser name as a C style string.
const PARSER_NAME: &'static [u8] = b"mqtt\0";
localstorage_new: None,
localstorage_free: None,
get_files: None,
- get_tx_iterator: Some(rs_mqtt_state_get_tx_iterator),
+ get_tx_iterator: Some(crate::applayer::state_get_tx_iterator::<MQTTState, MQTTTransaction>),
get_tx_data: rs_mqtt_get_tx_data,
apply_tx_config: None,
flags: APP_LAYER_PARSER_OPT_UNIDIR_TXS,