You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.1 KiB
Rust
78 lines
2.1 KiB
Rust
use std::fs::File;
|
|
use std::io::{BufRead, BufReader};
|
|
use std::path::Path;
|
|
use std::sync::Arc;
|
|
|
|
use rustc_hash::FxHashMap;
|
|
|
|
use crate::errors::*;
|
|
|
|
const MAX_ITERATIONS: usize = 5;
|
|
|
|
#[derive(Debug)]
|
|
struct BlackListInner {
|
|
map: FxHashMap<Vec<u8>, ()>,
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct BlackList {
|
|
inner: Arc<BlackListInner>,
|
|
max_iterations: usize,
|
|
}
|
|
|
|
impl BlackList {
|
|
pub fn new(map: FxHashMap<Vec<u8>, ()>, max_iterations: usize) -> Self {
|
|
let inner = Arc::new(BlackListInner { map });
|
|
BlackList {
|
|
inner,
|
|
max_iterations,
|
|
}
|
|
}
|
|
|
|
pub fn load(path: impl AsRef<Path>) -> Result<Self, Error> {
|
|
let mut map = FxHashMap::default();
|
|
let fp = BufReader::new(File::open(path)?);
|
|
for (line_nb, line) in fp.lines().enumerate() {
|
|
let line = line?;
|
|
let mut line = line.trim();
|
|
if line.is_empty() || line.starts_with('#') {
|
|
continue;
|
|
}
|
|
while line.starts_with("*.") {
|
|
line = &line[2..];
|
|
}
|
|
while line.ends_with('.') {
|
|
line = &line[..line.len() - 1];
|
|
}
|
|
let qname = line.as_bytes().to_ascii_lowercase();
|
|
if qname.is_empty() {
|
|
bail!("Unexpected blacklist rule at line {}", line_nb)
|
|
}
|
|
map.insert(qname, ());
|
|
}
|
|
Ok(BlackList::new(map, MAX_ITERATIONS))
|
|
}
|
|
|
|
pub fn find(&self, qname: &[u8]) -> bool {
|
|
let qname = qname.to_ascii_lowercase();
|
|
let mut qname = qname.as_slice();
|
|
let map = &self.inner.map;
|
|
let mut iterations = self.max_iterations;
|
|
while qname.len() >= 4 && iterations > 0 {
|
|
if map.contains_key(qname) {
|
|
return true;
|
|
}
|
|
let mut it = qname.splitn(2, |x| *x == b'.');
|
|
if it.next().is_none() {
|
|
break;
|
|
}
|
|
qname = match it.next() {
|
|
None => break,
|
|
Some(qname) => qname,
|
|
};
|
|
iterations -= 1;
|
|
}
|
|
false
|
|
}
|
|
}
|