adds a support for Primus taint propagation

BinaryAnalysisPlatform · Aug 18, 2017 · 20d386c · 20d386c
1 parent d0d7bf8
commit 20d386c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/plugins/bap/plugins/bap_taint.py b/plugins/bap/plugins/bap_taint.py
@@ -27,15 +27,18 @@
     ('taints', 'yellow')
 ]
 
-
 class PropagateTaint(BapIda):
+    ENGINE='primus'
+
     "Propagate taint information using BAP"
     def __init__(self, addr, kind):
         super(PropagateTaint,self).__init__()
+
         self.action = 'taint propagating from {:s}0x{:X}'.format(
             '*' if kind == 'ptr' else '',
             addr)
-        self.passes = ['taint','propagate-taint','map-terms','emit-ida-script']
+        propagate = 'run' if self.ENGINE == 'primus' else 'propagate-taint'
+        self.passes = ['taint', propagate, 'map-terms','emit-ida-script']
         self.script = self.tmpfile('py')
         scheme = self.tmpfile('scm')
         for (pat,color) in patterns:
@@ -50,6 +53,12 @@ def __init__(self, addr, kind):
             '--emit-ida-script-file', self.script.name
         ]
 
+        if self.ENGINE == 'primus':
+            self.args += [
+                '--primus-propagate-taint-run',
+                '--primus-promiscuous-mode',
+                '--primus-greedy-scheduler'
+            ]
 
 
 class BapTaint(idaapi.plugin_t):