From: Shivani Bhardwaj <shivanib134@gmail.com>
Date: Thu, 12 Aug 2021 12:41:42 +0000 (+0530)
Subject: mqtt: use Direction enum
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=a7ac79bed7afc43c8de4ad1023769f39b79847ba;p=people%2Fms%2Fsuricata.git

mqtt: use Direction enum
---

diff --git a/rust/src/mqtt/mqtt.rs b/rust/src/mqtt/mqtt.rs
index 91564838c..9df1ae3a5 100644
--- a/rust/src/mqtt/mqtt.rs
+++ b/rust/src/mqtt/mqtt.rs
@@ -21,7 +21,7 @@ use super::mqtt_message::*;
 use super::parser::*;
 use crate::applayer::{self, LoggerFlags};
 use crate::applayer::*;
-use crate::core::{self, AppProto, Flow, ALPROTO_FAILED, ALPROTO_UNKNOWN, IPPROTO_TCP};
+use crate::core::{self, *};
 use nom7::Err;
 use std;
 use std::ffi::CString;
@@ -660,11 +660,11 @@ pub unsafe extern "C" fn rs_mqtt_tx_get_alstate_progress(
 ) -> std::os::raw::c_int {
     let tx = cast_pointer!(tx, MQTTTransaction);
     if tx.complete {
-        if direction == core::STREAM_TOSERVER {
+        if direction == Direction::ToServer.into() {
             if tx.toserver {
                 return 1;
             }
-        } else if direction == core::STREAM_TOCLIENT {
+        } else if direction == Direction::ToClient.into() {
             if tx.toclient {
                 return 1;
             }