ss_rs/acl/
rule_set.rs

1//! A set of rules.
2
3use regex::Regex;
4
5/// A set of rules.
6pub struct RuleSet {
7    rules: Vec<Regex>,
8}
9
10impl RuleSet {
11    /// Creates a empty rule set.
12    pub fn new() -> RuleSet {
13        RuleSet { rules: Vec::new() }
14    }
15
16    /// Inserts a rule into the set.
17    pub fn insert(&mut self, rule: Regex) {
18        self.rules.push(rule);
19    }
20
21    /// Checks the given rule against all rules in the set.
22    pub fn contains(&self, data: &str) -> bool {
23        self.rules.iter().any(|r| r.is_match(data))
24    }
25}
26
27#[cfg(test)]
28mod tests {
29    use super::*;
30
31    #[test]
32    fn test_rule_set() {
33        let rules = [r"(^|\.)030buy\.com$", r"(^|\.)12vpn\.com$", "127.0.0.1"];
34
35        let mut rule_set = RuleSet::new();
36
37        for rule in rules {
38            rule_set.insert(Regex::new(rule).unwrap());
39        }
40
41        assert_eq!(rule_set.contains("030buy.com"), true);
42        assert_eq!(rule_set.contains("12vpn.com"), true);
43        assert_eq!(rule_set.contains(".12vpn.com"), true);
44        assert_eq!(rule_set.contains("34.12vpn.com"), true);
45        assert_eq!(rule_set.contains("127.0.0.1"), true);
46
47        assert_eq!(rule_set.contains("1112vpn.com"), false);
48        assert_eq!(rule_set.contains("12vpn.com "), false);
49        assert_eq!(rule_set.contains("12vpn.comm"), false);
50        assert_eq!(rule_set.contains("2vpn.net.com"), false);
51        assert_eq!(rule_set.contains("2vpn.netccom"), false);
52        assert_eq!(rule_set.contains("127.0.0.0"), false);
53    }
54}
ss_rs/acl/rule_set.rs

ss_rs/acl/
rule_set.rs
