[IndirectBranch]Filter out EntryBlock

HikariObfuscator · Jan 18, 2018 · 4f9f244 · 4f9f244 · Naville · Jan 18, 2018
1 parent 7b1b01d
commit 4f9f244
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/Transforms/Obfuscation/IndirectBranch.cpp b/lib/Transforms/Obfuscation/IndirectBranch.cpp
@@ -39,8 +39,11 @@ struct IndirectBranch : public FunctionPass {
     unsigned long long i = 0;
     for (auto F = M.begin(); F != M.end(); F++) {
       for (auto BB = F->begin(); BB != F->end(); BB++) {
-        indexmap[&*BB] = i++;
-        BBs.push_back(BlockAddress::get(&*BB));
+        BasicBlock *BBPtr=&*BB;
+        if(BBPtr!=&(BBPtr->getParent()->getEntryBlock())){
+          indexmap[BBPtr] = i++;
+          BBs.push_back(BlockAddress::get(BBPtr));
+        }
       }
     }
     ArrayType *AT =
@@ -82,7 +85,7 @@ struct IndirectBranch : public FunctionPass {
       }
       GlobalVariable *LoadFrom = NULL;
 
-      if (BI->isConditional()) {
+      if (BI->isConditional() || indexmap.find(BI->getSuccessor(0))==indexmap.end()) {
         // Create a new GV
         Constant *BlockAddressArray =
             ConstantArray::get(AT, ArrayRef<Constant *>(BlockAddresses));