58 lines
1.9 KiB
C++
58 lines
1.9 KiB
C++
//
|
|
// Created by erki on 02.03.21.
|
|
//
|
|
|
|
#include "OperationAstMatcher.hpp"
|
|
|
|
#include "OperationFinder.hpp"
|
|
|
|
using namespace clang;
|
|
using namespace clang::ast_matchers;
|
|
|
|
namespace
|
|
{
|
|
|
|
StatementMatcher AssignmentMatcher =
|
|
binaryOperation(isAssignmentOperator(),
|
|
hasLHS(expr().bind("lhs")),
|
|
hasRHS(expr().bind("rhs"))).bind("assignment");
|
|
|
|
StatementMatcher ArithmeticMatcher =
|
|
binaryOperation(hasAnyOperatorName("+", "-", "/", "*", "<", ">",
|
|
"<=", ">=", "==", "<<", ">>", "%"),
|
|
hasLHS(expr().bind("lhs")),
|
|
hasRHS(expr().bind("lhs"))).bind("arithmetic");
|
|
|
|
StatementMatcher UnaryArithmeticMatcher =
|
|
unaryOperator(hasAnyOperatorName("!", "++", "--"),
|
|
hasUnaryOperand(expr().bind("lhs"))).bind("unary_arithmetic");
|
|
|
|
}
|
|
|
|
OperationASTMatcher::OperationASTMatcher(OperationFinder* finder)
|
|
: _op_finder(finder)
|
|
{ }
|
|
|
|
void OperationASTMatcher::addToFinder(clang::ast_matchers::MatchFinder& finder)
|
|
{
|
|
finder.addMatcher(traverse(TK_IgnoreUnlessSpelledInSource, ArithmeticMatcher), this);
|
|
finder.addMatcher(traverse(TK_IgnoreUnlessSpelledInSource, AssignmentMatcher), this);
|
|
finder.addMatcher(traverse(TK_IgnoreUnlessSpelledInSource, UnaryArithmeticMatcher), this);
|
|
}
|
|
|
|
void OperationASTMatcher::run(const clang::ast_matchers::MatchFinder::MatchResult& result)
|
|
{
|
|
if (const auto* op = result.Nodes.getNodeAs<clang::BinaryOperator>("assignment"))
|
|
{
|
|
_op_finder->processArithmetic(op, *result.SourceManager);
|
|
}
|
|
else if (const auto* op = result.Nodes.getNodeAs<clang::BinaryOperator>("arithmetic"))
|
|
{
|
|
_op_finder->processArithmetic(op, *result.SourceManager);
|
|
}
|
|
else if (const auto* op = result.Nodes.getNodeAs<clang::UnaryOperator>("unary_arithmetic"))
|
|
{
|
|
_op_finder->processUnaryArithmetic(op, *result.SourceManager);
|
|
}
|
|
}
|