From 8f9f78c2d0865fde1d82d257cd34220d3be5d972 Mon Sep 17 00:00:00 2001
From: Shivani Bhardwaj <shivanib134@gmail.com>
Date: Thu, 12 Aug 2021 18:52:13 +0530
Subject: [PATCH] sip: use Direction enum

---
 rust/src/sip/detect.rs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/rust/src/sip/detect.rs b/rust/src/sip/detect.rs
index 7d987e6ed8..7098ea54a3 100644
--- a/rust/src/sip/detect.rs
+++ b/rust/src/sip/detect.rs
@@ -17,7 +17,7 @@
 
 // written by Giuseppe Longo <giuseppe@glongo.it>
 
-use crate::core::{STREAM_TOCLIENT, STREAM_TOSERVER};
+use crate::core::Direction;
 use crate::sip::sip::SIPTransaction;
 use std::ptr;
 
@@ -70,8 +70,8 @@ pub unsafe extern "C" fn rs_sip_tx_get_protocol(
     buffer_len: *mut u32,
     direction: u8,
 ) -> u8 {
-    match direction {
-        STREAM_TOSERVER => {
+    match direction.into() {
+        Direction::ToServer => {
             if let Some(ref r) = tx.request {
                 let v = &r.version;
                 if v.len() > 0 {
@@ -81,7 +81,7 @@ pub unsafe extern "C" fn rs_sip_tx_get_protocol(
                 }
             }
         }
-        STREAM_TOCLIENT => {
+        Direction::ToClient => {
             if let Some(ref r) = tx.response {
                 let v = &r.version;
                 if v.len() > 0 {
@@ -91,7 +91,6 @@ pub unsafe extern "C" fn rs_sip_tx_get_protocol(
                 }
             }
         }
-        _ => {}
     }
 
     *buffer = ptr::null();