|
|
|
|
@ -17,6 +17,7 @@
|
|
|
|
|
|
|
|
|
|
//! Utility module.
|
|
|
|
|
|
|
|
|
|
use std::borrow::Cow;
|
|
|
|
|
use std::ffi::CStr;
|
|
|
|
|
use std::os::raw::c_char;
|
|
|
|
|
|
|
|
|
|
@ -27,6 +28,8 @@ use nom7::combinator::verify;
|
|
|
|
|
use nom7::multi::many1_count;
|
|
|
|
|
use nom7::IResult;
|
|
|
|
|
|
|
|
|
|
use humantime::parse_duration;
|
|
|
|
|
|
|
|
|
|
#[no_mangle]
|
|
|
|
|
pub unsafe extern "C" fn SCCheckUtf8(val: *const c_char) -> bool {
|
|
|
|
|
CStr::from_ptr(val).to_str().is_ok()
|
|
|
|
|
@ -64,10 +67,56 @@ pub unsafe extern "C" fn SCValidateDomain(input: *const u8, in_len: u32) -> u32
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Add 's' suffix if input is only digits, and convert to lowercase if needed.
|
|
|
|
|
fn duration_unit_normalize(input: &str) -> Cow<'_, str> {
|
|
|
|
|
if input.bytes().all(|b| b.is_ascii_digit()) {
|
|
|
|
|
let mut owned = String::with_capacity(input.len() + 1);
|
|
|
|
|
owned.push_str(input);
|
|
|
|
|
owned.push('s');
|
|
|
|
|
return Cow::Owned(owned);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if input.bytes().any(|b| b.is_ascii_uppercase()) {
|
|
|
|
|
Cow::Owned(input.to_ascii_lowercase())
|
|
|
|
|
} else {
|
|
|
|
|
Cow::Borrowed(input)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Reads a C string from `input`, parses it, and writes the result to `*res`.
|
|
|
|
|
/// Returns 0 on success (result written to *res), -1 otherwise.
|
|
|
|
|
#[no_mangle]
|
|
|
|
|
pub unsafe extern "C" fn SCParseTimeDuration(input: *const c_char, res: *mut u64) -> i32 {
|
|
|
|
|
if input.is_null() || res.is_null() {
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let input_str = match CStr::from_ptr(input).to_str() {
|
|
|
|
|
Ok(s) => s,
|
|
|
|
|
Err(_) => return -1,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let trimmed = input_str.trim();
|
|
|
|
|
if trimmed.is_empty() {
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let normalized = duration_unit_normalize(trimmed);
|
|
|
|
|
match parse_duration(normalized.as_ref()) {
|
|
|
|
|
Ok(duration) => {
|
|
|
|
|
*res = duration.as_secs();
|
|
|
|
|
0
|
|
|
|
|
}
|
|
|
|
|
Err(_) => -1,
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
|
mod tests {
|
|
|
|
|
|
|
|
|
|
use super::*;
|
|
|
|
|
use std::ffi::CString;
|
|
|
|
|
use std::ptr::{null, null_mut};
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_parse_domain() {
|
|
|
|
|
@ -84,4 +133,73 @@ mod tests {
|
|
|
|
|
let buf1: &[u8] = "a(x)y.com".as_bytes();
|
|
|
|
|
assert!(parse_domain(buf1).is_err());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_parse_time_valid() {
|
|
|
|
|
unsafe {
|
|
|
|
|
let mut v: u64 = 0;
|
|
|
|
|
|
|
|
|
|
let s = CString::new("10").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 10);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("0").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 0);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("2H").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 7200);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("1 day").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 86400);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("1w").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 604800);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("1 week").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 604800);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("1y").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 31557600);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("1 year").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, 31557600);
|
|
|
|
|
|
|
|
|
|
// max
|
|
|
|
|
let s = CString::new("18446744073709551615").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), 0);
|
|
|
|
|
assert_eq!(v, u64::MAX);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn test_parse_time_duration_invalid() {
|
|
|
|
|
unsafe {
|
|
|
|
|
let mut v: u64 = 0;
|
|
|
|
|
let s = CString::new("10q").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), -1);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("abc").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), -1);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("-300s").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), -1);
|
|
|
|
|
|
|
|
|
|
let s = CString::new("1h -600s").unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), -1);
|
|
|
|
|
|
|
|
|
|
assert_eq!(SCParseTimeDuration(null(), &mut v), -1);
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), null_mut()), -1);
|
|
|
|
|
|
|
|
|
|
let overflow_years = (u64::MAX / 31557600) + 1;
|
|
|
|
|
let s = CString::new(format!("{}y", overflow_years)).unwrap();
|
|
|
|
|
assert_eq!(SCParseTimeDuration(s.as_ptr(), &mut v), -1);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|