regex match on process_analysis rather than os.basepath

socfortress · Jun 28, 2024 · ac779c2 · ac779c2
1 parent 031ea37
commit ac779c2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/backend/app/threat_intel/schema/socfortress.py b/backend/app/threat_intel/schema/socfortress.py
@@ -1,4 +1,4 @@
-import os
+import re
 from typing import List
 from typing import Optional
 
@@ -58,7 +58,8 @@ class SocfortressProcessNameAnalysisRequest(BaseModel):
 
     @validator("process_name", pre=True)
     def extract_filename(cls, v):
-        return os.path.basename(v)
+        match = re.search(r"[^\\]+$", v)
+        return match.group() if match else v
 
 
 class Path(BaseModel):