security_rust/protocol/
ssrf.rs1use regex::Regex;
4use std::sync::LazyLock;
5
6use crate::{AttackCategory, DetectionResult, Detector, Severity};
7
8static PATTERNS: LazyLock<Vec<Regex>> = LazyLock::new(|| {
9 vec![
10 Regex::new(r"(?i)169\.254\.169\.254").unwrap(),
11 Regex::new(r"(?i)10\.\d{1,3}\.\d{1,3}\.\d{1,3}").unwrap(),
12 Regex::new(r"(?i)172\.(1[6-9]|2\d|3[01])\.\d{1,3}\.\d{1,3}").unwrap(),
13 Regex::new(r"(?i)192\.168\.\d{1,3}\.\d{1,3}").unwrap(),
14 Regex::new(r"(?i)127\.\d{1,3}\.\d{1,3}\.\d{1,3}").unwrap(),
15 Regex::new(r"(?i)\[::1\]").unwrap(),
16 Regex::new(r"(?i)0\.0\.0\.0").unwrap(),
17 Regex::new(r"(?i)gopher://").unwrap(),
18 Regex::new(r"(?i)dict://").unwrap(),
19 Regex::new(r"(?i)ftp://[^/]*@").unwrap(),
20 Regex::new(r"(?i)file:///").unwrap(),
21 ]
22});
23
24pub struct SsrfDetector;
25
26impl Detector for SsrfDetector {
27 fn name(&self) -> &'static str {
28 "ssrf"
29 }
30
31 fn detect(&self, input: &str) -> Option<DetectionResult> {
32 for re in PATTERNS.iter() {
33 if let Some(m) = re.find(input) {
34 return Some(DetectionResult {
35 attack_type: "ssrf".into(),
36 category: AttackCategory::Protocol,
37 severity: Severity::Critical,
38 matched_pattern: m.as_str().to_string(),
39 offset: m.start(),
40 message: "SSRF server-side request forgery detected".into(),
41 });
42 }
43 }
44 None
45 }
46}