Add event-driven memory syscall telemetry
This commit is contained in:
parent
893409b549
commit
a51478fa22
18 changed files with 589 additions and 32 deletions
|
|
@ -1,5 +1,5 @@
|
|||
# SPDX-License-Identifier: GPL-3.0-or-later
|
||||
"""Runs the eBPF exec source on a thread and routes events through the engine."""
|
||||
"""Runs optional eBPF event sources on threads and routes them through rules."""
|
||||
from __future__ import annotations
|
||||
|
||||
import threading
|
||||
|
|
@ -7,9 +7,11 @@ from collections.abc import Callable
|
|||
|
||||
from ..alert import Alert
|
||||
from ..config import Config
|
||||
from . import bcc_source
|
||||
from . import bcc_source, bcc_syscall_source
|
||||
from .exec_event import ExecEvent
|
||||
from .rules import ExecRuleEngine
|
||||
from .syscall_event import SyscallEvent
|
||||
from .syscall_rules import SyscallRuleEngine
|
||||
|
||||
|
||||
class ExecMonitor:
|
||||
|
|
@ -53,3 +55,46 @@ class ExecMonitor:
|
|||
def stop(self) -> None:
|
||||
self._stop.set()
|
||||
self._source.stop()
|
||||
|
||||
|
||||
class SyscallMonitor:
|
||||
"""Owns the syscall event source + rule engine; emits Alerts via callback."""
|
||||
|
||||
def __init__(self, cfg: Config, on_alert: Callable[[Alert], None]) -> None:
|
||||
self.cfg = cfg
|
||||
self._on_alert = on_alert
|
||||
self.engine = SyscallRuleEngine()
|
||||
self._source = bcc_syscall_source.BccSyscallSource(self._on_event)
|
||||
self._thread: threading.Thread | None = None
|
||||
self._stop = threading.Event()
|
||||
|
||||
def start(self) -> tuple[bool, str]:
|
||||
ok, reason = bcc_syscall_source.available()
|
||||
if not ok:
|
||||
return False, reason
|
||||
try:
|
||||
self._source.start()
|
||||
except Exception as exc:
|
||||
return False, f"probe load failed: {exc!r}"
|
||||
self._thread = threading.Thread(target=self._run, daemon=True)
|
||||
self._thread.start()
|
||||
reason = "ok"
|
||||
if self._source.attach_errors:
|
||||
reason = "ok (some probes unavailable: " + "; ".join(
|
||||
self._source.attach_errors[:4]) + ")"
|
||||
return True, reason
|
||||
|
||||
def _run(self) -> None:
|
||||
while not self._stop.is_set():
|
||||
try:
|
||||
self._source.poll(timeout_ms=200)
|
||||
except Exception:
|
||||
break
|
||||
|
||||
def _on_event(self, ev: SyscallEvent) -> None:
|
||||
for alert in self.engine.match(ev):
|
||||
self._on_alert(alert)
|
||||
|
||||
def stop(self) -> None:
|
||||
self._stop.set()
|
||||
self._source.stop()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue