修复 -fla（控制流平坦化）报错

CYRUS-STUDIO · Dec 21, 2024 · fb76c7a · fb76c7a
1 parent 1082301
commit fb76c7a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/llvm/lib/Passes/CMakeLists.txt b/llvm/lib/Passes/CMakeLists.txt
@@ -22,6 +22,8 @@ add_llvm_component_library(LLVMPasses
   ADDITIONAL_HEADER_DIRS
   ${LLVM_MAIN_INCLUDE_DIR}/llvm
   ${LLVM_MAIN_INCLUDE_DIR}/llvm/Passes
+  ${LLVM_MAIN_INCLUDE_DIR}/llvm/lib/Passes/Obfuscation
+  ${LLVM_MAIN_INCLUDE_DIR}/llvm/lib/Passes/Obfuscation/compat
 
   DEPENDS
   intrinsics_gen

diff --git a/llvm/lib/Passes/Obfuscation/Flattening.cpp b/llvm/lib/Passes/Obfuscation/Flattening.cpp
@@ -94,9 +94,7 @@ bool FlatteningPass::flatten(Function &F) {
         // 非条件跳转
         else if(BB->getTerminator()->getNumSuccessors() == 1){
             BasicBlock *sucBB = BB->getTerminator()->getSuccessor(0);
-            if (bEntryBB_isConditional) {
-                entryBB.getTerminator()->eraseFromParent();
-            }
+            BB->getTerminator()->eraseFromParent();
             ConstantInt *numCase = swInst->findCaseDest(sucBB);
             new StoreInst(numCase, swVarPtr, BB);
             BranchInst::Create(returnBB, BB);