aboutsummaryrefslogtreecommitdiff
path: root/instrumentation/split-compares-pass.so.cc
diff options
context:
space:
mode:
authorllzmb <46303940+llzmb@users.noreply.github.com>2021-11-29 19:55:55 +0100
committerllzmb <46303940+llzmb@users.noreply.github.com>2021-11-29 19:55:55 +0100
commit8968bee836ecde0fa2427b84b2f3ac85c276958f (patch)
tree3be630bc3a5d1ea65f3bb49863b37ff5b9d688c4 /instrumentation/split-compares-pass.so.cc
parent11b3961e687f188aee806afee93bc95807081ff9 (diff)
parent235bb3235e45a81eb2b524f3a76346d91a163e3d (diff)
downloadafl++-8968bee836ecde0fa2427b84b2f3ac85c276958f.tar.gz
Merge branch 'dev' into docs_cleanup_folder_2
Diffstat (limited to 'instrumentation/split-compares-pass.so.cc')
-rw-r--r--instrumentation/split-compares-pass.so.cc97
1 files changed, 14 insertions, 83 deletions
diff --git a/instrumentation/split-compares-pass.so.cc b/instrumentation/split-compares-pass.so.cc
index 8ea67a21..d1254e40 100644
--- a/instrumentation/split-compares-pass.so.cc
+++ b/instrumentation/split-compares-pass.so.cc
@@ -1,7 +1,6 @@
/*
* Copyright 2016 laf-intel
* extended for floating point by Heiko Eißfeldt
- * adapted to new pass manager by Heiko Eißfeldt
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -29,15 +28,8 @@
#include "llvm/Pass.h"
#include "llvm/Support/raw_ostream.h"
-
-#if LLVM_MAJOR >= 7
-#include "llvm/Passes/PassPlugin.h"
-#include "llvm/Passes/PassBuilder.h"
-#include "llvm/IR/PassManager.h"
-#else
#include "llvm/IR/LegacyPassManager.h"
#include "llvm/Transforms/IPO/PassManagerBuilder.h"
-#endif
#include "llvm/Transforms/Utils/BasicBlockUtils.h"
#include "llvm/IR/Module.h"
@@ -61,26 +53,27 @@ using namespace llvm;
namespace {
-#if LLVM_MAJOR >= 7
-class SplitComparesTransform : public PassInfoMixin<SplitComparesTransform> {
- public:
-// static char ID;
- SplitComparesTransform() : enableFPSplit(0) {
-#else
class SplitComparesTransform : public ModulePass {
+
public:
static char ID;
SplitComparesTransform() : ModulePass(ID), enableFPSplit(0) {
-#endif
initInstrumentList();
+
}
-#if LLVM_MAJOR >= 7
- PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
-#else
bool runOnModule(Module &M) override;
+#if LLVM_VERSION_MAJOR >= 4
+ StringRef getPassName() const override {
+
+#else
+ const char *getPassName() const override {
+
#endif
+ return "AFL_SplitComparesTransform";
+
+ }
private:
int enableFPSplit;
@@ -169,40 +162,7 @@ class SplitComparesTransform : public ModulePass {
} // namespace
-#if LLVM_MAJOR >= 7
-extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK
-llvmGetPassPluginInfo() {
- return {
- LLVM_PLUGIN_API_VERSION, "splitcompares", "v0.1",
- /* lambda to insert our pass into the pass pipeline. */
- [](PassBuilder &PB) {
-#if 1
- using OptimizationLevel = typename PassBuilder::OptimizationLevel;
- PB.registerOptimizerLastEPCallback(
- [](ModulePassManager &MPM, OptimizationLevel OL) {
- MPM.addPass(SplitComparesTransform());
- }
- );
-/* TODO LTO registration */
-#else
- using PipelineElement = typename PassBuilder::PipelineElement;
- PB.registerPipelineParsingCallback(
- [](StringRef Name, ModulePassManager &MPM, ArrayRef<PipelineElement>) {
- if ( Name == "splitcompares" ) {
- MPM.addPass(SplitComparesTransform());
- return true;
- } else {
- return false;
- }
- }
- );
-#endif
- }
- };
-}
-#else
char SplitComparesTransform::ID = 0;
-#endif
/// This function splits FCMP instructions with xGE or xLE predicates into two
/// FCMP instructions with predicate xGT or xLT and EQ
@@ -1356,11 +1316,7 @@ size_t SplitComparesTransform::splitFPCompares(Module &M) {
}
-#if LLVM_MAJOR >= 7
-PreservedAnalyses SplitComparesTransform::run(Module &M, ModuleAnalysisManager &MAM) {
-#else
bool SplitComparesTransform::runOnModule(Module &M) {
-#endif
char *bitw_env = getenv("AFL_LLVM_LAF_SPLIT_COMPARES_BITW");
if (!bitw_env) bitw_env = getenv("LAF_SPLIT_COMPARES_BITW");
@@ -1371,7 +1327,7 @@ bool SplitComparesTransform::runOnModule(Module &M) {
if ((isatty(2) && getenv("AFL_QUIET") == NULL) ||
getenv("AFL_DEBUG") != NULL) {
- errs() << "Split-compare-newpass by laf.intel@gmail.com, extended by "
+ errs() << "Split-compare-pass by laf.intel@gmail.com, extended by "
"heiko@hexco.de (splitting icmp to "
<< target_bitwidth << " bit)\n";
@@ -1383,10 +1339,6 @@ bool SplitComparesTransform::runOnModule(Module &M) {
}
-#if LLVM_MAJOR >= 7
- auto PA = PreservedAnalyses::all();
-#endif
-
if (enableFPSplit) {
count = splitFPCompares(M);
@@ -1419,13 +1371,7 @@ bool SplitComparesTransform::runOnModule(Module &M) {
auto op0 = CI->getOperand(0);
auto op1 = CI->getOperand(1);
- if (!op0 || !op1) {
-#if LLVM_MAJOR >= 7
- return PA;
-#else
- return false;
-#endif
- }
+ if (!op0 || !op1) { return false; }
auto iTy1 = dyn_cast<IntegerType>(op0->getType());
if (iTy1 && isa<IntegerType>(op1->getType())) {
@@ -1474,25 +1420,10 @@ bool SplitComparesTransform::runOnModule(Module &M) {
}
- if ((isatty(2) && getenv("AFL_QUIET") == NULL) ||
- getenv("AFL_DEBUG") != NULL) {
- errs() << count << " comparisons found\n";
- }
-
-#if LLVM_MAJOR >= 7
-/* if (modified) {
- PA.abandon<XX_Manager>();
- }*/
-
- return PA;
-#else
return true;
-#endif
}
-#if LLVM_MAJOR < 7 /* use old pass manager */
-
static void registerSplitComparesPass(const PassManagerBuilder &,
legacy::PassManagerBase &PM) {
@@ -1516,4 +1447,4 @@ static RegisterPass<SplitComparesTransform> X("splitcompares",
"AFL++ split compares",
true /* Only looks at CFG */,
true /* Analysis Pass */);
-#endif
+