ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-17 04:18:06 +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

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

@ -62,12 +62,14 @@ namespace {
#if LLVM_MAJOR >= 11 /* use new pass manager */ #if LLVM_MAJOR >= 11 /* use new pass manager */
class CmpLogInstructions : public PassInfoMixin<CmpLogInstructions> { class CmpLogInstructions : public PassInfoMixin<CmpLogInstructions> {
public: public:
CmpLogInstructions() { CmpLogInstructions() {
initInstrumentList(); initInstrumentList();
} }
#else #else
class CmpLogInstructions : public ModulePass { class CmpLogInstructions : public ModulePass {
@ -78,6 +80,7 @@ class CmpLogInstructions : public ModulePass {
initInstrumentList(); initInstrumentList();
} }
#endif #endif
#if LLVM_MAJOR >= 11 /* use new pass manager */ #if LLVM_MAJOR >= 11 /* use new pass manager */
@ -85,16 +88,17 @@ class CmpLogInstructions : public ModulePass {
#else #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 { StringRef getPassName() const override {
#else #else
const char *getPassName() const override { const char *getPassName() const override {
#endif #endif
return "cmplog instructions"; return "cmplog instructions";
} }
#endif #endif
private: private:
@ -637,8 +641,10 @@ 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, PreservedAnalyses CmpLogInstructions::run(Module & M,
ModuleAnalysisManager &MAM) { ModuleAnalysisManager &MAM) {
#else #else
bool CmpLogInstructions::runOnModule(Module &M) { bool CmpLogInstructions::runOnModule(Module &M) {
#endif #endif
if (getenv("AFL_QUIET") == NULL) if (getenv("AFL_QUIET") == NULL)
@ -671,9 +677,10 @@ static RegisterStandardPasses RegisterCmpLogInstructionsPass(
static RegisterStandardPasses RegisterCmpLogInstructionsPass0( static RegisterStandardPasses RegisterCmpLogInstructionsPass0(
PassManagerBuilder::EP_EnabledOnOptLevel0, registerCmpLogInstructionsPass); PassManagerBuilder::EP_EnabledOnOptLevel0, registerCmpLogInstructionsPass);
#if LLVM_VERSION_MAJOR >= 11 #if LLVM_VERSION_MAJOR >= 11
static RegisterStandardPasses RegisterCmpLogInstructionsPassLTO( static RegisterStandardPasses RegisterCmpLogInstructionsPassLTO(
PassManagerBuilder::EP_FullLinkTimeOptimizationLast, PassManagerBuilder::EP_FullLinkTimeOptimizationLast,
registerCmpLogInstructionsPass); registerCmpLogInstructionsPass);
#endif
#endif #endif
#endif

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

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

4
src/afl-cc.c
View File

@ -684,8 +684,8 @@ static void edit_params(u32 argc, char **argv, char **envp) {
#if LLVM_MAJOR >= 11 #if LLVM_MAJOR >= 11
cc_params[cc_par_cnt++] = "-fexperimental-new-pass-manager"; cc_params[cc_par_cnt++] = "-fexperimental-new-pass-manager";
cc_params[cc_par_cnt++] = cc_params[cc_par_cnt++] = alloc_printf(
alloc_printf("-fpass-plugin=%s/cmplog-instructions-pass.so", obj_path); "-fpass-plugin=%s/cmplog-instructions-pass.so", obj_path);
#else #else
cc_params[cc_par_cnt++] = "-Xclang"; cc_params[cc_par_cnt++] = "-Xclang";
cc_params[cc_par_cnt++] = "-load"; cc_params[cc_par_cnt++] = "-load";