enhance memory pattern detection heuristics

This commit is contained in:
Adir Shitrit
2025-11-08 11:06:07 +02:00
parent e6cf909e05
commit ee21be20dc

View File

@@ -71,14 +71,23 @@ impl DetectionEngine {
confidence += 0.4;
}
// Update baseline
// Check for thread count anomalies
if let Some(baseline) = self.baseline.get(&process.pid) {
if process.thread_count > baseline.thread_count {
let diff = process.thread_count - baseline.thread_count;
indicators.push(format!("{} new threads created", diff));
confidence += 0.2;
}
// Detect significant RWX increase (possible injection)
if rwx_count > baseline.rwx_regions + 1 {
indicators.push("Rapid RWX region allocation".to_string());
confidence += 0.5;
}
}
// Check for unusual memory patterns
self.check_memory_patterns(memory_regions, &mut indicators, &mut confidence);
self.baseline.insert(
process.pid,
@@ -103,6 +112,43 @@ impl DetectionEngine {
confidence,
}
}
/// Check for suspicious memory patterns
fn check_memory_patterns(
&self,
regions: &[MemoryRegion],
indicators: &mut Vec<String>,
confidence: &mut f32,
) {
// Look for small executable allocations (typical shellcode size)
let small_exec = regions
.iter()
.filter(|r| {
r.size < 0x10000 // 64KB
&& (r.protection == MemoryProtection::ReadExecute
|| r.protection == MemoryProtection::ReadWriteExecute)
})
.count();
if small_exec >= 3 {
indicators.push("Multiple small executable allocations".to_string());
*confidence += 0.3;
}
// Check for memory gaps that might indicate hollowing
let mut sorted_regions: Vec<_> = regions.iter().collect();
sorted_regions.sort_by_key(|r| r.base_address);
for window in sorted_regions.windows(2) {
let gap = window[1].base_address - (window[0].base_address + window[0].size);
if gap > 0x100000 && gap < 0x1000000 {
// 1MB to 16MB gap is suspicious
indicators.push("Suspicious memory gaps detected".to_string());
*confidence += 0.2;
break;
}
}
}
}
impl Default for DetectionEngine {