diff options
| -rw-r--r-- | mutator.cpp | 601 |
1 files changed, 0 insertions, 601 deletions
diff --git a/mutator.cpp b/mutator.cpp deleted file mode 100644 index 19dd13d..0000000 --- a/mutator.cpp +++ /dev/null @@ -1,601 +0,0 @@ - -/*first line intentionally left blank.*/ -/*code structure inspired by Eli Bendersky's tutorial on Rewriters.*/ -/**********************************************************************************************************************/ -/*FIXME-all classes should use replacements.*/ -/**********************************************************************************************************************/ -/*included modules*/ -/*project headers*/ -#include "mutator_aux.h" -/*standard headers*/ -#include <string> -#include <iostream> -#include <cassert> -/*LLVM headers*/ -#include "clang/AST/AST.h" -#include "clang/AST/ASTConsumer.h" -#include "clang/ASTMatchers/ASTMatchers.h" -#include "clang/ASTMatchers/ASTMatchFinder.h" -#include "clang/Frontend/CompilerInstance.h" -#include "clang/Frontend/FrontendActions.h" -#include "clang/Lex/Lexer.h" -#include "clang/Tooling/CommonOptionsParser.h" -#include "clang/Tooling/Tooling.h" -#include "clang/Rewrite/Core/Rewriter.h" -#include "llvm/Support/raw_ostream.h" -#include "llvm/IR/Function.h" -/**********************************************************************************************************************/ -/*used namespaces*/ -using namespace llvm; -using namespace clang; -using namespace clang::ast_matchers; -using namespace clang::driver; -using namespace clang::tooling; -/**********************************************************************************************************************/ -/*global vars*/ -/*the variable that holds the previously-matched SOC for class StmtTrap.*/ -std::string g_linenoold; - -static llvm::cl::OptionCategory MutatorLVL1Cat("mutator-lvl1 options category"); -/**********************************************************************************************************************/ -/*matcher callback for something.*/ -class FunctionHandler : public MatchFinder::MatchCallback { -public: - FunctionHandler (Rewriter &Rewrite) : Rewrite(Rewrite) {} - - virtual void run(const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::BinaryOperator>("binopeq") != nullptr) - { - /*get the matched node.*/ - const BinaryOperator *BinOp = MR.Nodes.getNodeAs<clang::BinaryOperator>("binopeq"); - - /*get the sourceloation.*/ - SourceLocation BinOpSL = BinOp->getLocStart(); - - BinOpSL = Devi::SourceLocationHasMacro(BinOpSL, Rewrite, "start"); - /*does the sourcelocation include a macro expansion?*/ - - /*replace it.*/ -#if 0 - Rewrite.ReplaceText(BinOpSL, 2U , "XXX"); -#endif - } - else - { - std::cout << "the macther -binopeq- returned nullptr!" << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class StmtTrapIf : public MatchFinder::MatchCallback { -public: - StmtTrapIf (Rewriter &Rewrite) : Rewrite(Rewrite) {} - - virtual void run (const MatchFinder::MatchResult &MR) - { - /*just in case*/ - if (MR.Nodes.getNodeAs<clang::IfStmt>("iftrap") != nullptr) - { - /*getting the matched ifStmt.*/ - const IfStmt *TrapIf = MR.Nodes.getNodeAs<clang::IfStmt>("iftrap"); - - /*getting the condition of the matched ifStmt.*/ - const Expr *IfCond = TrapIf->getCond(); - - /*getting the sourcerange of the condition of the trapped ifstmt.*/ - SourceLocation TrapIfCondSL = IfCond->getLocStart(); - TrapIfCondSL = Devi::SourceLocationHasMacro(TrapIfCondSL, Rewrite, "start"); - SourceLocation TrapIfCondSLEnd = IfCond->getLocEnd(); - TrapIfCondSLEnd = Devi::SourceLocationHasMacro(TrapIfCondSLEnd, Rewrite, "end"); - SourceRange TrapIfCondSR; - TrapIfCondSR.setBegin(TrapIfCondSL); - TrapIfCondSR.setEnd(TrapIfCondSLEnd); - - /*replacing the condition with the utility trap function.*/ -#if 0 - Rewrite.ReplaceText(TrapIfCondSR, "C_Trap_P()"); -#endif - } - else - { - std::cout << "the matcher -iftrap- returned nullptr!" << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class StmtTrap : public MatchFinder::MatchCallback { -public: - StmtTrap (Rewriter &Rewrite) : Rewrite(Rewrite) {} - - /*if there are more than 2 statements in the traditional sense in one line, the behavior is undefined.*/ - /*for now this class only finds an SOC. later to be used for something useful.*/ - virtual void run (const MatchFinder::MatchResult &MR) - { - /*out of paranoia*/ - if (MR.Nodes.getNodeAs<clang::Stmt>("stmttrap") != nullptr) - { - const Stmt *StmtTrapMR = MR.Nodes.getNodeAs<clang::Stmt>("stmttrap"); - - SourceLocation STSL = StmtTrapMR->getLocStart(); - STSL = Devi::SourceLocationHasMacro(STSL, Rewrite, "start"); - SourceLocation STSLE = StmtTrapMR->getLocEnd(); - STSLE = Devi::SourceLocationHasMacro(STSLE, Rewrite, "end"); - - /*just getting the SOC of the matched statement out of its sourcelocation.*/ - /*this only works since we are guaranteed the same and known matching pattern by MatchFinder.*/ - size_t startloc = 0U; - size_t endloc = 0U; - std::string lineno; - startloc = STSL.printToString(*MR.SourceManager).find(":", 0U); - endloc = STSLE.printToString(*MR.SourceManager).find(":", startloc + 1U); - lineno = STSL.printToString(*MR.SourceManager).substr(startloc, endloc - startloc); - - /*just prints out the sourcelocations for diagnostics.*/ -#if 0 - std::cout << STSL.printToString(*MR.SourceManager) << std::endl; - std::cout << STSLE.printToString(*MR.SourceManager) << std::endl; - std::cout << startloc << "---" << endloc << "---" << lineno << std::endl; -#endif - - /*have we matched a new SOC? if yes:*/ - if (lineno != g_linenoold) - { - SourceRange SR; - SR.setBegin(StmtTrapMR->getLocStart()); - SR.setEnd(StmtTrapMR->getLocEnd()); - -#if 0 - Rewrite.InsertText(StmtTrapMR->getLocStart(), "XXX", "true", "true"); -#endif - } - else - { - /*intentionally left blank.*/ - } - - /*set the string representing the old SOC line number to the one matched now.*/ - g_linenoold = lineno; - } - else - { - std::cout << "the matcher -stmttrap- returned nullptr!" << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class StmtRet : public MatchFinder::MatchCallback -{ -public: - StmtRet (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - virtual void run (const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::ReturnStmt>("stmtret") != nullptr) - { - /*getting the returnstmt.*/ - const ReturnStmt *RetStmtMR = MR.Nodes.getNodeAs<clang::ReturnStmt>("stmtret"); - - /*getting the sourcerange of the matched returnstmt.*/ - SourceLocation RSMR = RetStmtMR->getLocStart(); - RSMR = Devi::SourceLocationHasMacro(RSMR, Rewrite, "start"); - SourceLocation RSMRE = RetStmtMR->getLocEnd(); - RSMRE = Devi::SourceLocationHasMacro(RSMRE, Rewrite, "end"); - SourceRange RSMRSR; - RSMRSR.setBegin(RSMR); - RSMRSR.setEnd(RSMRE); - -#if 0 - Rewrite.ReplaceText(RSMRSR, "C_Trap_P()"); -#endif - } - else - { - std::cout << "matcher -stmtret- returned nullptr." << std::endl; - } - } - -private: - Rewriter &Rewrite; - -}; -/**********************************************************************************************************************/ -class ForFixer : public MatchFinder::MatchCallback -{ -public: - ForFixer (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - /*adds curly braces for forstmts that don't have it.*/ - virtual void run(const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::ForStmt>("mrfor") != nullptr) - { - const ForStmt *MRFor = MR.Nodes.getNodeAs<clang::ForStmt>("mrfor"); - - Rewriter::RewriteOptions opts; - - SourceLocation MRForSL = MRFor->getBody()->getLocStart(); - MRForSL = Devi::SourceLocationHasMacro(MRForSL, Rewrite, "start"); - SourceLocation MRForSLE = MRFor->getBody()->getLocEnd(); - MRForSLE = Devi::SourceLocationHasMacro(MRForSLE, Rewrite, "end"); - - SourceRange SR; - SR.setBegin(MRForSL); - SR.setEnd(MRForSLE); - - int RangeSize = Rewrite.getRangeSize(SR, opts); - - Rewrite.InsertText(MRForSL, "{\n", true, false); - /*we're getting the endloc with an offset of 2 to accomodate unary operators like '++'.*/ - /*line-terminating semicolons are not included in the matches.*/ - Rewrite.InsertTextAfterToken(MRForSL.getLocWithOffset(RangeSize), "\n}"); - } - else - { - std::cout << "matcher -mrfor- returned nullptr." << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class WhileFixer : public MatchFinder::MatchCallback -{ -public: - WhileFixer (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - /*adds curly braces for whilestmts that don't have it.*/ - virtual void run (const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::WhileStmt>("mrwhile") != nullptr) - { - const WhileStmt *MRWhile = MR.Nodes.getNodeAs<clang::WhileStmt>("mrwhile"); - -#if 0 - std::cout << MRWhile->getBody()->getLocStart().printToString(*MR.SourceManager) << std::endl; - std::cout << MRWhile->getBody()->getLocEnd().printToString(*MR.SourceManager) << std::endl; -#endif - - SourceLocation WFSL = MRWhile->getBody()->getLocStart(); - WFSL = Devi::SourceLocationHasMacro(WFSL, Rewrite, "start"); - SourceLocation WFSLE = MRWhile->getBody()->getLocEnd(); - WFSLE = Devi::SourceLocationHasMacro(WFSLE, Rewrite, "end"); - - /*we're getting the endloc with an offset of 2 to accomodate unary operators like '++'.*/ - /*line-terminating semicolons are not included in the matches.*/ - Rewrite.InsertText(WFSL, "{\n", true, true); - Rewrite.InsertTextAfterToken(WFSLE.getLocWithOffset(2U), "\n}"); - } - else - { -#if 1 - std::cout << "matcher -mrwhile- returned nullptr." << std::endl; -#endif - } - } - - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class IfElseFixer : public MatchFinder::MatchCallback -{ -public: - IfElseFixer (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - virtual void run(const MatchFinder::MatchResult &MR) - { - /*underdev*/ - if (MR.Nodes.getNodeAs<clang::IfStmt>("mrifelse") != nullptr) - { - const IfStmt *ElseIf = MR.Nodes.getNodeAs<clang::IfStmt>("mrifelse"); - //const IfStmt *LastIf = MR.Nodes.getNodeAs<clang::IfStmt>("mrifelse"); - - SourceLocation IFESL = ElseIf->getElse()->getLocStart(); - IFESL = Devi::SourceLocationHasMacro(IFESL, Rewrite, "start"); - SourceLocation IFESLE = ElseIf->getElse()->getLocEnd(); - IFESLE = Devi::SourceLocationHasMacro(IFESLE, Rewrite, "end"); - SourceRange SR; - SR.setBegin(IFESL); - SR.setEnd(IFESLE); - - clang::Rewriter::RewriteOptions opts; - - int RangeSize = Rewrite.getRangeSize(SR, opts); - - //std::cout << IFESLE.printToString(*MR.SourceManager) << "\n" << std::endl; - -#if 1 - Rewrite.InsertText(IFESL, "{\n", true, true); - Rewrite.InsertTextAfterToken(IFESL.getLocWithOffset(RangeSize), "\n}"); -#endif - } - else - { - std::cout << "matcher -mrifelse- returned nullptr." << std::endl; - } - } - - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class IfFixer : public MatchFinder::MatchCallback -{ -public: - IfFixer (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - /*adds curly braces to ifstmts that dont have it.*/ - virtual void run(const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::IfStmt>("mrif") != nullptr) - { - const IfStmt *MRIf = MR.Nodes.getNodeAs<clang::IfStmt>("mrif"); - - SourceLocation FFSL = MRIf->getThen()->getLocStart(); - FFSL = Devi::SourceLocationHasMacro(FFSL, Rewrite, "start"); - SourceLocation FFSLE = MRIf->getThen()->getLocEnd(); - FFSLE = Devi::SourceLocationHasMacro(FFSLE, Rewrite, "end"); - - SourceRange SR; - SR.setBegin(FFSL); - SR.setEnd(FFSLE); - Rewriter::RewriteOptions opts; - int RangeSize = Rewrite.getRangeSize(SR, opts); - - /*we're getting the endloc with an offset of 2 to accomodate unary operators like '++'.*/ - /*line-terminating semicolons are not included in the matches.*/ -#if 1 - Rewrite.InsertText(FFSL, "{\n", true, true); - Rewrite.InsertTextAfterToken(FFSL.getLocWithOffset(RangeSize), "\n}"); -#endif - } - else - { - std::cout << "matcher -mrif- returned nullptr." << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class SwitchFixer : public MatchFinder::MatchCallback -{ -public: - SwitchFixer (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - virtual void run(const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::CaseStmt>("bubba-hotep") != nullptr) - { - const CaseStmt *CS = MR.Nodes.getNodeAs<clang::CaseStmt>("bubba-hotep"); - - const Stmt *SB = CS->getSubStmt(); - - SourceLocation SBSL = SB->getLocStart(); - SBSL = Devi::SourceLocationHasMacro(SBSL, Rewrite, "start"); - SourceLocation SBSLE = SB->getLocEnd(); - SBSLE = Devi::SourceLocationHasMacro(SBSLE, Rewrite, "end"); - - SourceLocation SL = CS->getLocStart(); - SL = Devi::SourceLocationHasMacro(SL, Rewrite, "start"); - SourceLocation SLE = CS->getLocEnd(); - SLE = Devi::SourceLocationHasMacro(SLE, Rewrite, "end"); - - SourceRange SR; - SR.setBegin(SL); - SR.setEnd(SLE); - Rewriter::RewriteOptions opts; - int RangeSize [[maybe_unused]] = Rewrite.getRangeSize(SR, opts); - -#if 1 - Rewrite.InsertText(SBSL, "{\n", true, true); - Rewrite.InsertTextAfterToken(SL.getLocWithOffset(RangeSize), "\n}"); -#endif - } - else - { - std::cout << "matcher -bubba-hotep- returned nullptr." << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class SwitchDfFixer : public MatchFinder::MatchCallback -{ -public: - SwitchDfFixer (Rewriter &Rewrite) : Rewrite (Rewrite) {} - - virtual void run(const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::DefaultStmt>("mumma-hotep") != nullptr) - { - const DefaultStmt *DS = MR.Nodes.getNodeAs<clang::DefaultStmt>("mumma-hotep"); - - const Stmt *SB = DS->getSubStmt(); - - SourceLocation CSL = SB->getLocStart(); - CSL = Devi::SourceLocationHasMacro(CSL, Rewrite, "start"); - - SourceLocation SL = DS->getLocStart(); - SL = Devi::SourceLocationHasMacro(SL, Rewrite, "start"); - SourceLocation SLE = DS->getLocEnd(); - SLE = Devi::SourceLocationHasMacro(SLE, Rewrite, "end"); - - SourceRange SR; - SR.setBegin(SL); - SR.setEnd(SLE); - Rewriter::RewriteOptions opts; - int RangeSize [[maybe_unused]] = Rewrite.getRangeSize(SR, opts); - -#if 1 - Rewrite.InsertText(CSL, "{\n", true, true); - Rewrite.InsertTextAfterToken(SL.getLocWithOffset(RangeSize), "\n}"); -#endif - } - else - { - std::cout << "matcher -mumma-hotep- returned nullptr." << std::endl; - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class IfConstSwapper : public MatchFinder::MatchCallback -{ -public: - IfConstSwapper (Rewriter &Rewrite) : Rewrite(Rewrite) {} - - virtual void run(const MatchFinder::MatchResult &MR) - { - if (MR.Nodes.getNodeAs<clang::BinaryOperator>("ifconstswapbinop") != nullptr) - { - const BinaryOperator* BO = MR.Nodes.getNodeAs<clang::BinaryOperator>("ifconstswapbinop"); - - const Expr* LHS = BO->getLHS(); - const Expr* RHS = BO->getRHS(); - - ASTContext *const ASTC = MR.Context; - SourceManager *const SM = MR.SourceManager; - - if (RHS->isEvaluatable(*ASTC, Expr::SideEffectsKind::SE_NoSideEffects) && !LHS->isEvaluatable(*ASTC, Expr::SideEffectsKind::SE_NoSideEffects)) - { - SourceLocation SLLHS = LHS->getLocStart(); - SLLHS = Devi::SourceLocationHasMacro(SLLHS, Rewrite, "start"); - SourceLocation SLELHS = LHS->getLocStart(); - SLELHS = Devi::SourceLocationHasMacro(SLELHS, Rewrite, "start"); - SourceRange SRLHS; - SRLHS.setBegin(SLLHS); - SRLHS.setEnd(SLELHS); - - if (!SM->isInMainFile(SLLHS)) - { - return void(); - } - - if (SM->isInSystemHeader(SLLHS)) - { - return void(); - } - - SourceLocation SLRHS = RHS->getLocStart(); - SLRHS = Devi::SourceLocationHasMacro(SLRHS, Rewrite, "start"); - SourceLocation SLERHS = RHS->getLocEnd(); - SLERHS = Devi::SourceLocationHasMacro(SLERHS, Rewrite, "start"); - SourceRange SRRHS; - SRRHS.setBegin(SLRHS); - SRRHS.setEnd(SLERHS); - - const std::string LHSString = Rewrite.getRewrittenText(SRLHS); - const std::string RHSString = Rewrite.getRewrittenText(SRRHS); - - StringRef LHSRef = StringRef(LHSString); - StringRef RHSRef = StringRef(RHSString); - - Rewriter::RewriteOptions opts; - int RangeSizeLHS = Rewrite.getRangeSize(SRLHS, opts); - int RangeSizeRHS = Rewrite.getRangeSize(SRRHS, opts); - - Rewrite.ReplaceText(SRLHS.getBegin(), RangeSizeLHS, RHSRef); - Rewrite.ReplaceText(SRRHS.getBegin(), RangeSizeRHS, LHSRef); - } - } - } - -private: - Rewriter &Rewrite; -}; -/**********************************************************************************************************************/ -class MyASTConsumer : public ASTConsumer { - -public: - MyASTConsumer(Rewriter &R) : HandlerForFunction(R), HandlerForIfTrap(R), HandlerForStmtTrap(R), HandlerForStmtRet(R), HandlerForFixer(R), \ - HandlerForWhile(R), HandlerForIfElse(R), HandlerForIfFixer(R), HandlerForSwitchFixer(R), HandlerForSwitchDf(R), HandlerForIfConstSwap(R) - { - Matcher.addMatcher(binaryOperator(hasOperatorName("==")).bind("binopeq"), &HandlerForFunction); - - Matcher.addMatcher(ifStmt(hasCondition(anything())).bind("iftrap"), &HandlerForIfTrap); - - Matcher.addMatcher(stmt().bind("stmttrap") , &HandlerForStmtTrap); - - Matcher.addMatcher(returnStmt().bind("stmtret"), &HandlerForStmtRet); - - Matcher.addMatcher(forStmt(unless(hasDescendant(compoundStmt()))).bind("mrfor"), &HandlerForFixer); - - Matcher.addMatcher(whileStmt(unless(hasDescendant(compoundStmt()))).bind("mrwhile"), &HandlerForWhile); - - Matcher.addMatcher(ifStmt(allOf(hasElse(unless(ifStmt())), hasElse(unless(compoundStmt())))).bind("mrifelse"), &HandlerForIfElse); - - Matcher.addMatcher(ifStmt(unless(hasDescendant(compoundStmt()))).bind("mrif"), &HandlerForIfFixer); - - Matcher.addMatcher(switchStmt(forEachDescendant(caseStmt(unless(hasDescendant(compoundStmt()))).bind("bubba-hotep"))), &HandlerForSwitchFixer); - - Matcher.addMatcher(switchStmt(hasDescendant(defaultStmt(unless(hasDescendant(compoundStmt()))).bind("mumma-hotep"))), &HandlerForSwitchDf); - - Matcher.addMatcher(ifStmt(hasDescendant(binaryOperator(anyOf(hasOperatorName("=="), hasOperatorName("="))).bind("ifconstswapbinop"))).bind("ifconstswapper"), &HandlerForIfConstSwap); - } - - void HandleTranslationUnit(ASTContext & Context) override { - Matcher.matchAST(Context); - } - -private: - FunctionHandler HandlerForFunction; - StmtTrapIf HandlerForIfTrap; - StmtTrap HandlerForStmtTrap; - StmtRet HandlerForStmtRet; - ForFixer HandlerForFixer; - WhileFixer HandlerForWhile; - IfElseFixer HandlerForIfElse; - IfFixer HandlerForIfFixer; - SwitchFixer HandlerForSwitchFixer; - SwitchDfFixer HandlerForSwitchDf; - IfConstSwapper HandlerForIfConstSwap; - MatchFinder Matcher; -}; -/**********************************************************************************************************************/ -class MyFrontendAction : public ASTFrontendAction -{ -public: - MyFrontendAction() {} - void EndSourceFileAction() override - { - TheRewriter.getEditBuffer(TheRewriter.getSourceMgr().getMainFileID()).write(llvm::outs()); - } - - std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI, StringRef file) override - { - TheRewriter.setSourceMgr(CI.getSourceManager(), CI.getLangOpts()); - return llvm::make_unique<MyASTConsumer>(TheRewriter); - } - -private: - Rewriter TheRewriter; -}; -/**********************************************************************************************************************/ -/*Main*/ -int main(int argc, const char **argv) -{ - CommonOptionsParser op(argc, argv, MutatorLVL1Cat); - ClangTool Tool(op.getCompilations(), op.getSourcePathList()); - - return Tool.run(newFrontendActionFactory<MyFrontendAction>().get()); -} -/*last line intentionally left blank.*/ |