pgsql: trigger raw stream reassembly at tx completion

Task OISF#7000
jufajardini · Jul 17, 2024 · de11dfc · de11dfc
1 parent 038543c
commit de11dfc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rust/src/pgsql/pgsql.rs b/rust/src/pgsql/pgsql.rs
@@ -355,7 +355,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;
                     if let Some(state) = PgsqlState::request_next_state(&request) {
                         self.state_progress = state;
@@ -365,6 +364,7 @@ impl PgsqlState {
                         tx.request = Some(request);
                         if tx_completed {
                             tx.tx_state = PgsqlTransactionState::RequestDone;
+                            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
@@ -485,7 +485,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);
                     if let Some(state) = self.response_process_next_state(&response, flow) {
@@ -512,6 +511,7 @@ impl PgsqlState {
                             tx.responses.push(response);
                             if tx_completed {
                                 tx.tx_state = PgsqlTransactionState::ResponseDone;
+                                sc_app_layer_parser_trigger_raw_stream_reassembly(flow, Direction::ToClient as i32);
                             }
                         }
                     } else {