fix cff pass crash on llvm-9

Ylarod · May 4, 2021 · 33bd5d4 · 33bd5d4
1 parent c76f673
commit 33bd5d4
Show file tree

Hide file tree

Showing 4 changed files with 568 additions and 0 deletions.
diff --git a/llvm/include/llvm/Transforms/Obfuscation/LegacyLowerSwitch.h b/llvm/include/llvm/Transforms/Obfuscation/LegacyLowerSwitch.h
@@ -0,0 +1,10 @@
+#ifndef _LEGACY_LOWERSWITCH_INCLUDES_
+#define _LEGACY_LOWERSWITCH_INCLUDES_
+
+namespace llvm {
+class FunctionPass;
+FunctionPass *createLegacyLowerSwitchPass();
+}
+
+#endif
+
diff --git a/llvm/lib/Transforms/Obfuscation/CMakeLists.txt b/llvm/lib/Transforms/Obfuscation/CMakeLists.txt
@@ -9,6 +9,7 @@ add_llvm_library(LLVMObfuscation
   IndirectGlobalVariable.cpp
   Flattening.cpp
   StringEncryption.cpp
+  LegacyLowerSwitch.cpp
   DEPENDS
   LLVMLinker
   )

diff --git a/llvm/lib/Transforms/Obfuscation/Flattening.cpp b/llvm/lib/Transforms/Obfuscation/Flattening.cpp
@@ -12,6 +12,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Transforms/Obfuscation/Flattening.h"
+#include "llvm/Transforms/Obfuscation/LegacyLowerSwitch.h"
 #include "llvm/Transforms/Obfuscation/Utils.h"
 #include "llvm/Transforms/Obfuscation/IPObfuscationContext.h"
 #include "llvm/Transforms/Utils.h"
@@ -78,7 +79,11 @@ bool Flattening::flatten(Function *f) {
   // END OF SCRAMBLER
 
   // Lower switch
+#if LLVM_VERSION_MAJOR * 10 + LLVM_VERSION_MINOR >= 90
+  FunctionPass *lower = createLegacyLowerSwitchPass();
+#else
   FunctionPass *lower = createLowerSwitchPass();
+#endif
   lower->runOnFunction(*f);
 
   // Save all original BB