Spawn sample tag tasks

CERT-Polska · Jan 12, 2024 · a25b5a2 · a25b5a2
1 parent 77e3229
commit a25b5a2
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/karton/config_extractor/config_extractor.py b/karton/config_extractor/config_extractor.py
@@ -146,11 +146,12 @@ def report_config(
             return
 
         dhash = config_dhash(legacy_config)
+        family = config["family"]
         task = Task(
             {
                 "type": "config",
                 "kind": "static",
-                "family": config["family"],
+                "family": family,
                 "quality": task.headers.get("quality", "high"),
             },
             payload={
@@ -163,6 +164,24 @@ def report_config(
             },
         )
         self.send_task(task)
+        if parent:
+            self.send_sample_tag_task(sample, [family])
+            self.send_sample_tag_task(parent, [f"ripped: {family}"])
+        else:
+            self.send_sample_tag_task(sample, [f"ripped: {family}", family])
+
+    def send_sample_tag_task(self, sample: Resource, tags: List[str]) -> None:
+        task = Task(
+            {
+                "type": "sample",
+                "stage": "analyzed",
+            },
+            payload={
+                "sample": sample,
+                "tags": tags,
+            }
+        )
+        self.send_task(task)
 
     # analyze a standard, non-dump sample
     def analyze_sample(self, task: Task, sample: ResourceBase) -> None: