bfc unsupport InvokeInst

fighting300 · Aug 30, 2017 · ae0e5ac · ae0e5ac
1 parent 42f732c
commit ae0e5ac
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/lib/Transforms/Obfuscation/BogusControlFlow.cpp b/lib/Transforms/Obfuscation/BogusControlFlow.cpp
@@ -138,14 +138,26 @@ namespace {
       }
       // If fla annotations
       if(toObfuscate(flag,&F,"bcf")) {
-        bogus(F);
-        doF(*F.getParent());
-        return true;
+        if (isInvoke(&F)) {
+          bogus(F);
+          doF(*F.getParent());
+          return true;
+        }
       }
 
       return false;
     } // end of runOnFunction()
 
+    bool isInvoke(Function *f) {
+      for (Function::iterator i = f->begin(); i != f->end(); ++i) {
+          BasicBlock *bb = &*i;
+          if (isa<InvokeInst>(bb->getTerminator())) {
+              return false;
+          }
+      }
+      return true;
+    }
+
     void bogus(Function &F) {
       // For statistics and debug
       ++NumFunction;