bittorrent: use standard tx iterator code

Will make bindgen life easier

As the fn prototype should not have istate: &mut u64
but a *mut AppLayerGetTxIterState which is an union with a u64
pull/14639/head
Philippe Antoine 6 months ago committed by Victor Julien
parent 1a546936c7
commit 076fcd61cd

@ -123,25 +123,6 @@ impl BitTorrentDHTState {
return status;
}
fn tx_iterator(
&mut self, min_tx_id: u64, state: &mut u64,
) -> Option<(&BitTorrentDHTTransaction, 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.
@ -235,20 +216,19 @@ unsafe extern "C" fn tx_get_alstate_progress(
return 0;
}
unsafe extern "C" fn 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, BitTorrentDHTState);
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();
}
impl State<BitTorrentDHTTransaction> for BitTorrentDHTState {
fn get_transaction_count(&self) -> usize {
self.transactions.len()
}
fn get_transaction_by_index(&self, index: usize) -> Option<&BitTorrentDHTTransaction> {
self.transactions.get(index)
}
}
impl Transaction for BitTorrentDHTTransaction {
fn id(&self) -> u64 {
self.tx_id
}
}
@ -280,7 +260,9 @@ pub unsafe extern "C" fn SCRegisterBittorrentDhtUdpParser() {
localstorage_new: None,
localstorage_free: None,
get_tx_files: None,
get_tx_iterator: Some(state_get_tx_iterator),
get_tx_iterator: Some(
applayer::state_get_tx_iterator::<BitTorrentDHTState, BitTorrentDHTTransaction>,
),
get_tx_data,
get_state_data,
apply_tx_config: None,

Loading…
Cancel
Save