rfb: also set unimplemented auth types

(cherry picked from commit 1521b77)
OISF · Jul 14, 2023 · 74cbbd7 · 74cbbd7
1 parent b1d2d78
commit 74cbbd7
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/rust/src/rfb/rfb.rs b/rust/src/rfb/rfb.rs
@@ -228,6 +228,15 @@ impl RFBState {
                             current = rem;
 
                             let chosen_security_type = request.security_type;
+
+                            if let Some(current_transaction) = self.get_current_tx() {
+                                current_transaction.ts_security_type_selection = Some(request);
+                                current_transaction.chosen_security_type =
+                                    Some(chosen_security_type as u32);
+                            } else {
+                                debug_validate_fail!("no transaction set at security type stage");
+                            }
+
                             match chosen_security_type {
                                 2 => self.state = parser::RFBGlobalState::TCVncChallenge,
                                 1 => self.state = parser::RFBGlobalState::TSClientInit,
@@ -244,13 +253,6 @@ impl RFBState {
                                     return AppLayerResult::ok();
                                 }
                             }
-
-                            if let Some(current_transaction) = self.get_current_tx() {
-                                current_transaction.ts_security_type_selection = Some(request);
-                                current_transaction.chosen_security_type = Some(chosen_security_type as u32);
-                            } else {
-                                debug_validate_fail!("no transaction set at security type stage");
-                            }
                         }
                         Err(nom::Err::Incomplete(_)) => {
                             return AppLayerResult::incomplete(