From 640dd7c5827e506c7a3bc6a9ff14f92770659540 Mon Sep 17 00:00:00 2001 From: Jason Ish Date: Tue, 1 Apr 2025 10:55:18 -0600 Subject: [PATCH] rust/conf: rust format --- rust/src/conf.rs | 66 +++++++++++++++++++++--------------------------- 1 file changed, 29 insertions(+), 37 deletions(-) diff --git a/rust/src/conf.rs b/rust/src/conf.rs index 912252ecb6..109c9b22de 100644 --- a/rust/src/conf.rs +++ b/rust/src/conf.rs @@ -17,18 +17,18 @@ //! 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 { 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 "; -- 2.47.2