Skip to content

Commit

Permalink
pgsql: trigger raw stream reassembly at tx completion
Browse files Browse the repository at this point in the history
Once we are tracking tx progress per-direction for PGSQL, we can trigger
the raw stream reassembly for detection purposes once the transactions
are completed.

Task OISF#7000
  • Loading branch information
jufajardini committed Aug 8, 2024
1 parent 9065ada commit 97dc1f0
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions rust/src/pgsql/pgsql.rs
Original file line number Diff line number Diff line change
Expand Up @@ -356,7 +356,6 @@ impl PgsqlState {
);
match PgsqlState::state_based_req_parsing(self.state_progress, start) {
Ok((rem, request)) => {
sc_app_layer_parser_trigger_raw_stream_reassembly(flow, Direction::ToServer as i32);
start = rem;
let temp_state: PgsqlStateProgress;
let tx_completed = if let Some(state) = PgsqlState::request_next_state(&request) {
Expand All @@ -377,6 +376,7 @@ impl PgsqlState {
} else {
tx.tx_req_state = PgsqlTxReqProgress::TxReqDone;
}
sc_app_layer_parser_trigger_raw_stream_reassembly(flow, Direction::ToServer as i32);
}
} else {
// If there isn't a new transaction, we'll consider Suri should move on
Expand Down Expand Up @@ -516,7 +516,6 @@ impl PgsqlState {
while !start.is_empty() {
match PgsqlState::state_based_resp_parsing(self.state_progress, start) {
Ok((rem, response)) => {
sc_app_layer_parser_trigger_raw_stream_reassembly(flow, Direction::ToClient as i32);
start = rem;
SCLogDebug!("Response is {:?}", &response);
let (curr_state, tx_completed) = if let Some(state) = self.response_process_next_state(&response, flow) {
Expand Down

0 comments on commit 97dc1f0

Please sign in to comment.