83 lines
2.1 KiB
Rust
83 lines
2.1 KiB
Rust
|
use ipnet::IpNet;
|
||
|
|
use lazy_static::lazy_static;
|
||
|
|
use nix::unistd;
|
||
|
|
use regex::Regex;
|
||
|
|
use std::boxed::Box;
|
||
|
|
use std::fs::File;
|
||
|
|
use std::io::*;
|
||
|
|
use std::path::Path;
|
||
|
|
use std::time::Duration;
|
||
|
|
|
||
|
|
lazy_static! {
|
||
|
|
static ref R_FILE_GZIP: Regex = Regex::new(r".*\.gz.*").unwrap();
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn read_lines(filename: &String, offset: u64) -> Option<Box<dyn Read>> {
|
||
|
|
let mut file = match File::open(filename) {
|
||
|
|
Ok(f) => f,
|
||
|
|
Err(err) => {
|
||
|
|
println!("{err}");
|
||
|
|
return None;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
file.seek(SeekFrom::Start(offset)).unwrap();
|
||
|
|
let lines: Box<dyn Read> = Box::new(BufReader::new(file));
|
||
|
|
Some(lines)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn _dedup<T: Ord + PartialOrd>(list: &mut Vec<T>) -> usize {
|
||
|
|
// Begin with sorting entries
|
||
|
|
list.sort();
|
||
|
|
// Then deduplicate
|
||
|
|
list.dedup();
|
||
|
|
// Return the length
|
||
|
|
list.len()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn build_trustnets(cfgtrustnets: &Vec<String>) -> Vec<IpNet> {
|
||
|
|
let mut trustnets: Vec<IpNet> = vec![];
|
||
|
|
for trustnet in cfgtrustnets {
|
||
|
|
match trustnet.parse() {
|
||
|
|
Ok(net) => trustnets.push(net),
|
||
|
|
Err(err) => {
|
||
|
|
println!("error parsing {trustnet}, error: {err}");
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
trustnets
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn sleep(seconds: u64) {
|
||
|
|
std::thread::sleep(Duration::from_secs(seconds));
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn gethostname(show_fqdn: bool) -> String {
|
||
|
|
let mut buf = [0u8; 64];
|
||
|
|
let hostname_cstr = unistd::gethostname(&mut buf).expect("Failed getting hostname");
|
||
|
|
let fqdn = hostname_cstr
|
||
|
|
.to_str()
|
||
|
|
.expect("Hostname wasn't valid UTF-8")
|
||
|
|
.to_string();
|
||
|
|
let hostname: Vec<&str> = fqdn.split(".").collect();
|
||
|
|
if show_fqdn {
|
||
|
|
return fqdn;
|
||
|
|
}
|
||
|
|
hostname[0].to_string()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn _search_subfolders(path: &Path) -> Vec<String> {
|
||
|
|
let dirs = std::fs::read_dir(path).unwrap();
|
||
|
|
let mut folders: Vec<String> = vec![];
|
||
|
|
for dir in dirs {
|
||
|
|
let dirpath = dir.unwrap().path();
|
||
|
|
let path = Path::new(dirpath.as_path());
|
||
|
|
if path.is_dir() {
|
||
|
|
folders.push(dirpath.to_str().unwrap().to_string());
|
||
|
|
for f in _search_subfolders(path) {
|
||
|
|
folders.push(f);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
folders
|
||
|
|
}
|