ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-16 20:08:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

code format

Browse Source
This commit is contained in:
vanhauser-thc
2022-03-01 20:13:57 +01:00
parent 5e9286b9ea
commit d325fa5db8
3 changed files with 36 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
instrumentation/cmplog-instructions-pass.cc
View File

@ -60,14 +60,16 @@ using namespace llvm;
namespace {
#if LLVM_MAJOR >= 11 /* use new pass manager */
#if LLVM_MAJOR >= 11 /* use new pass manager */
class CmpLogInstructions : public PassInfoMixin<CmpLogInstructions> {
public:
CmpLogInstructions() {
initInstrumentList();
}
#else
class CmpLogInstructions : public ModulePass {
@ -78,23 +80,25 @@ class CmpLogInstructions : public ModulePass {
initInstrumentList();
}
#endif
#if LLVM_MAJOR >= 11 /* use new pass manager */
#if LLVM_MAJOR >= 11 /* use new pass manager */
PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
#else
bool runOnModule(Module &M) override;
bool runOnModule(Module &M) override;
#if LLVM_VERSION_MAJOR >= 4
#if LLVM_VERSION_MAJOR >= 4
StringRef getPassName() const override {
#else
#else
const char *getPassName() const override {
#endif
#endif
return "cmplog instructions";
}
#endif
private:
@ -104,7 +108,7 @@ class CmpLogInstructions : public ModulePass {
} // namespace
#if LLVM_MAJOR <= 10 /* use old pass manager */
#if LLVM_MAJOR <= 10 /* use old pass manager */
char CmpLogInstructions::ID = 0;
#endif
@ -634,11 +638,13 @@ bool CmpLogInstructions::hookInstrs(Module &M) {
}
#if LLVM_MAJOR >= 11 /* use new pass manager */
#if LLVM_MAJOR >= 11 /* use new pass manager */
PreservedAnalyses CmpLogInstructions::run(Module & M,
ModuleAnalysisManager &MAM) {
ModuleAnalysisManager &MAM) {
#else
bool CmpLogInstructions::runOnModule(Module &M) {
#endif
if (getenv("AFL_QUIET") == NULL)
@ -648,7 +654,7 @@ bool CmpLogInstructions::runOnModule(Module &M) {
hookInstrs(M);
verifyModule(M);
#if LLVM_MAJOR >= 11 /* use new pass manager */
#if LLVM_MAJOR >= 11 /* use new pass manager */
return PreservedAnalyses::all();
#else
return true;
@ -671,9 +677,10 @@ static RegisterStandardPasses RegisterCmpLogInstructionsPass(
static RegisterStandardPasses RegisterCmpLogInstructionsPass0(
PassManagerBuilder::EP_EnabledOnOptLevel0, registerCmpLogInstructionsPass);
#if LLVM_VERSION_MAJOR >= 11
#if LLVM_VERSION_MAJOR >= 11
static RegisterStandardPasses RegisterCmpLogInstructionsPassLTO(
PassManagerBuilder::EP_FullLinkTimeOptimizationLast,
registerCmpLogInstructionsPass);
#endif
#endif
#endif

23
instrumentation/compare-transform-pass.so.cc
View File

@ -81,27 +81,28 @@ class CompareTransform : public ModulePass {
}
#if LLVM_MAJOR < 11
#if LLVM_VERSION_MAJOR >= 4
#if LLVM_VERSION_MAJOR >= 4
StringRef getPassName() const override {
#else
#else
const char *getPassName() const override {
#endif
#endif
#endif
#if LLVM_MAJOR >= 11 /* use new pass manager */
PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
PreservedAnalyses run(Module & M, ModuleAnalysisManager & MAM);
#else
bool runOnModule(Module &M) override;
#endif
private:
bool transformCmps(Module &M, const bool processStrcmp,
const bool processMemcmp, const bool processStrncmp,
const bool processStrcasecmp,
const bool processStrncasecmp);
private:
bool transformCmps(Module & M, const bool processStrcmp,
const bool processMemcmp, const bool processStrncmp,
const bool processStrcasecmp,
const bool processStrncasecmp);
};
};
} // namespace
@ -170,7 +171,7 @@ bool CompareTransform::transformCmps(Module &M, const bool processStrcmp,
#if LLVM_VERSION_MAJOR >= 9
FunctionCallee tolowerFn;
#else
Function * tolowerFn;
Function *tolowerFn;
#endif
{

10
src/afl-cc.c
View File

@ -462,7 +462,7 @@ static void edit_params(u32 argc, char **argv, char **envp) {
} else {
#if LLVM_MAJOR >= 11 /* use new pass manager */
#if LLVM_MAJOR >= 11 /* use new pass manager */
cc_params[cc_par_cnt++] = "-fexperimental-new-pass-manager";
cc_params[cc_par_cnt++] =
alloc_printf("-fpass-plugin=%s/split-switches-pass.so", obj_path);
@ -488,7 +488,7 @@ static void edit_params(u32 argc, char **argv, char **envp) {
} else {
#if LLVM_MAJOR >= 11 /* use new pass manager */
#if LLVM_MAJOR >= 11 /* use new pass manager */
cc_params[cc_par_cnt++] = "-fexperimental-new-pass-manager";
cc_params[cc_par_cnt++] = alloc_printf(
"-fpass-plugin=%s/compare-transform-pass.so", obj_path);
@ -571,7 +571,7 @@ static void edit_params(u32 argc, char **argv, char **envp) {
}
#if LLVM_MAJOR == 13 // TODO: set to 14 when done FIXME
#if LLVM_MAJOR == 13 // TODO: set to 14 when done FIXME
// Use the old pass manager in LLVM 13 which the afl++ passes still use.
cc_params[cc_par_cnt++] = "-flegacy-pass-manager";
#endif
@ -684,8 +684,8 @@ static void edit_params(u32 argc, char **argv, char **envp) {
#if LLVM_MAJOR >= 11
cc_params[cc_par_cnt++] = "-fexperimental-new-pass-manager";
cc_params[cc_par_cnt++] =
alloc_printf("-fpass-plugin=%s/cmplog-instructions-pass.so", obj_path);
cc_params[cc_par_cnt++] = alloc_printf(
"-fpass-plugin=%s/cmplog-instructions-pass.so", obj_path);
#else
cc_params[cc_par_cnt++] = "-Xclang";
cc_params[cc_par_cnt++] = "-load";