]> git.ipfire.org Git - thirdparty/suricata.git/commitdiff
rust/conf: rust format 12907/head
authorJason Ish <jason.ish@oisf.net>
Tue, 1 Apr 2025 16:55:18 +0000 (10:55 -0600)
committerVictor Julien <victor@inliniac.net>
Tue, 1 Apr 2025 19:11:14 +0000 (21:11 +0200)
rust/src/conf.rs

index 912252ecb6d604458ee0cf775b8941f3b5e989be..109c9b22dea0c37ddd5b2c37a996a96ef04a9d56 100644 (file)
 
 //! Module for retrieving configuration details.
 
-use std::os::raw::c_char;
-use std::os::raw::c_int;
-use std::ffi::{CString, CStr};
-use std::ptr;
-use std::str;
 use nom7::{
     character::complete::{multispace0, not_line_ending},
-    sequence::{preceded, tuple},
-    number::complete::double,
     combinator::verify,
+    number::complete::double,
+    sequence::{preceded, tuple},
     IResult,
 };
+use std::ffi::{CStr, CString};
+use std::os::raw::c_char;
+use std::os::raw::c_int;
+use std::ptr;
+use std::str;
 use suricata_sys::sys::SCConfGet;
 use suricata_sys::sys::SCConfGetChildValue;
 use suricata_sys::sys::SCConfGetChildValueBool;
@@ -66,9 +66,7 @@ pub fn conf_get(key: &str) -> Option<&str> {
         return None;
     }
 
-    let value = str::from_utf8(unsafe{
-        CStr::from_ptr(vptr).to_bytes()
-    }).unwrap();
+    let value = str::from_utf8(unsafe { CStr::from_ptr(vptr).to_bytes() }).unwrap();
 
     return Some(value);
 }
@@ -90,9 +88,8 @@ pub struct ConfNode {
 }
 
 impl ConfNode {
-
     pub fn wrap(conf: *const SCConfNode) -> Self {
-        return Self { conf }
+        return Self { conf };
     }
 
     pub fn get_child_value(&self, key: &str) -> Option<&str> {
@@ -100,9 +97,7 @@ impl ConfNode {
 
         unsafe {
             let s = CString::new(key).unwrap();
-            if SCConfGetChildValue(self.conf,
-                                 s.as_ptr(),
-                                 &mut vptr) != 1 {
+            if SCConfGetChildValue(self.conf, s.as_ptr(), &mut vptr) != 1 {
                 return None;
             }
         }
@@ -111,9 +106,7 @@ impl ConfNode {
             return None;
         }
 
-        let value = str::from_utf8(unsafe{
-            CStr::from_ptr(vptr).to_bytes()
-        }).unwrap();
+        let value = str::from_utf8(unsafe { CStr::from_ptr(vptr).to_bytes() }).unwrap();
 
         return Some(value);
     }
@@ -123,9 +116,7 @@ impl ConfNode {
 
         unsafe {
             let s = CString::new(key).unwrap();
-            if SCConfGetChildValueBool(self.conf,
-                                     s.as_ptr(),
-                                     &mut vptr) != 1 {
+            if SCConfGetChildValueBool(self.conf, s.as_ptr(), &mut vptr) != 1 {
                 return false;
             }
         }
@@ -135,13 +126,12 @@ impl ConfNode {
         }
         return false;
     }
-
 }
 
-const BYTE: u64       = 1;
-const KILOBYTE: u64   = 1024;
-const MEGABYTE: u64   = 1_048_576;
-const GIGABYTE: u64   = 1_073_741_824;
+const BYTE: u64 = 1;
+const KILOBYTE: u64 = 1024;
+const MEGABYTE: u64 = 1_048_576;
+const GIGABYTE: u64 = 1_073_741_824;
 
 /// Helper function to retrieve memory unit from a string slice
 ///
@@ -153,11 +143,11 @@ const GIGABYTE: u64   = 1_073_741_824;
 fn get_memunit(unit: &str) -> u64 {
     let unit = &unit.to_lowercase()[..];
     match unit {
-        "b"     => { BYTE }
-        "kb"    => { KILOBYTE }
-        "mb"    => { MEGABYTE }
-        "gb"    => { GIGABYTE }
-        _       => { 0 }
+        "b" => BYTE,
+        "kb" => KILOBYTE,
+        "mb" => MEGABYTE,
+        "gb" => GIGABYTE,
+        _ => 0,
     }
 }
 
@@ -174,8 +164,10 @@ pub fn get_memval(arg: &str) -> Result<u64, &'static str> {
     let arg = arg.trim();
     let val: f64;
     let mut unit: &str;
-    let mut parser = tuple((preceded(multispace0, double),
-                        preceded(multispace0, verify(not_line_ending, |c: &str| c.len() < 3))));
+    let mut parser = tuple((
+        preceded(multispace0, double),
+        preceded(multispace0, verify(not_line_ending, |c: &str| c.len() < 3)),
+    ));
     let r: IResult<&str, (f64, &str)> = parser(arg);
     if let Ok(r) = r {
         val = (r.1).0;
@@ -201,7 +193,7 @@ mod tests {
     #[test]
     fn test_memval_nospace() {
         let s = "10";
-        let res = 10 ;
+        let res = 10;
         assert_eq!(Ok(10), get_memval(s));
 
         let s = "10kb";
@@ -223,7 +215,7 @@ mod tests {
     #[test]
     fn test_memval_space_start() {
         let s = " 10";
-        let res = 10 ;
+        let res = 10;
         assert_eq!(Ok(res), get_memval(s));
 
         let s = " 10Kb";
@@ -242,7 +234,7 @@ mod tests {
     #[test]
     fn test_memval_space_end() {
         let s = " 10                  ";
-        let res = 10 ;
+        let res = 10;
         assert_eq!(Ok(res), get_memval(s));
 
         let s = "10Kb    ";
@@ -261,7 +253,7 @@ mod tests {
     #[test]
     fn test_memval_space_in_bw() {
         let s = " 10                  ";
-        let res = 10 ;
+        let res = 10;
         assert_eq!(Ok(res), get_memval(s));
 
         let s = "10 Kb    ";