Add configuration knob

CERT-Polska · Apr 19, 2021 · 79011d8 · 79011d8
1 parent f798942
commit 79011d8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/drakrun/drakrun/config.dist.ini b/drakrun/drakrun/config.dist.ini
@@ -74,6 +74,8 @@ enable_ipt=0
 ; protects against API hammering techniques, default 0 (means: off)
 ;   anti_hammering_threshold=0
 
+; (advanced) Attach DLL profiles to analyses
+; attach_profiles=0
 
 [drakvuf_plugins]
 ; list of enabled DRAKVUF plugins that are used by default,

diff --git a/drakrun/drakrun/main.py b/drakrun/drakrun/main.py
@@ -366,8 +366,9 @@ def send_analysis(self, sample, outdir, metadata, quality):
         if self.test_run:
             task.add_payload('testcase', self.current_task.payload['testcase'])
 
-        self.log.info("Uploading profiles...")
-        task.add_payload("profiles", self.build_profile_payload())
+        if self.config.config.getboolean("drakrun", "attach_profiles", fallback=False):
+            self.log.info("Uploading profiles...")
+            task.add_payload("profiles", self.build_profile_payload())
 
         self.log.info("Uploading artifacts...")
         for resource in self.upload_artifacts(self.analysis_uid, outdir):