sanitize strings in model

mandiant · Dec 2, 2024 · bbe2223 · bbe2223
1 parent 5b737c8
commit bbe2223
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 13 deletions.
diff --git a/capa/features/extractors/vmray/__init__.py b/capa/features/extractors/vmray/__init__.py
@@ -13,15 +13,7 @@
 from dataclasses import dataclass
 
 from capa.exceptions import UnsupportedFormatError
-from capa.features.extractors.vmray.models import (
-    File,
-    Flog,
-    SummaryV2,
-    StaticData,
-    FunctionCall,
-    xml_to_dict,
-    sanitize_string,
-)
+from capa.features.extractors.vmray.models import File, Flog, SummaryV2, StaticData, FunctionCall, xml_to_dict
 
 logger = logging.getLogger(__name__)
 
@@ -174,8 +166,8 @@ def _compute_monitor_processes(self):
                 ppid,
                 process.monitor_id,
                 process.image_name,
-                sanitize_string(process.filename),
-                sanitize_string(process.cmd_line),
+                process.filename,
+                process.cmd_line,
             )
 
         # not all processes are recorded in SummaryV2.json, get missing data from flog.xml, see #2394

diff --git a/capa/features/extractors/vmray/models.py b/capa/features/extractors/vmray/models.py
@@ -315,9 +315,9 @@ class Process(BaseModel):
     monitor_id: int
     # monitor_reason: str
     os_pid: int
-    filename: str
+    filename: SanitizedString
     image_name: str
-    cmd_line: str
+    cmd_line: SanitizedString
     ref_parent_process: Optional[GenericReference] = None