mirror of
https://github.com/mvt-project/mvt.git
synced 2026-09-03 08:30:51 +02:00
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
# Mobile Verification Toolkit (MVT)
|
|
# Copyright (c) 2021-2023 The MVT Authors.
|
|
# Use of this software is governed by the MVT License 1.1 that can be found at
|
|
# https://license.mvt.re/1.1/
|
|
|
|
import logging
|
|
|
|
from mvt.common.indicators import Indicators
|
|
from mvt.common.module import run_module
|
|
from mvt.common.alerts import AlertLevel
|
|
from mvt.ios.modules.mixed.net_datausage import Datausage
|
|
|
|
from ..utils import get_ios_backup_folder
|
|
|
|
|
|
class TestDatausageModule:
|
|
def test_datausage(self):
|
|
m = Datausage(target_path=get_ios_backup_folder())
|
|
run_module(m)
|
|
assert m.results[0]["isodate"][0:19] == "2019-08-27 15:08:09"
|
|
assert len(m.results) == 42
|
|
assert len(m.timeline) == 60
|
|
assert "live_ZBILLCYCLEEND" in m.results[0]["record"]
|
|
assert "process_Z_ENT" in m.results[0]["record"]
|
|
assert (
|
|
len(m.alertstore.alerts) == 1
|
|
) # We now have a detection for missing processes.
|
|
|
|
def test_detection(self, indicator_file):
|
|
m = Datausage(target_path=get_ios_backup_folder())
|
|
ind = Indicators(log=logging.getLogger())
|
|
ind.parse_stix2(indicator_file)
|
|
# Adds a file that exists in the manifest.
|
|
ind.ioc_collections[0]["processes"].append("CumulativeUsageTracker")
|
|
m.indicators = ind
|
|
run_module(m)
|
|
critical_alerts = [
|
|
alert for alert in m.alertstore.alerts if alert.level == AlertLevel.CRITICAL
|
|
]
|
|
assert len(critical_alerts) == 2
|
|
assert all(
|
|
"matched_indicator" not in alert.event for alert in critical_alerts
|
|
)
|
|
serialized_alerts = [
|
|
alert
|
|
for alert in m.alertstore.as_json()
|
|
if alert["matched_indicator"] is not None
|
|
]
|
|
assert len(serialized_alerts) == 2
|
|
assert all(
|
|
"matched_indicator" not in alert["event"] for alert in serialized_alerts
|
|
)
|