blob: b099f99888dcbfe2bd8d31a376f7bfd718cf1dfe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
/*
* This file is part of the LibreOffice project.
*
* Based on LLVM/Clang.
*
* This file is distributed under the University of Illinois Open Source
* License. See LICENSE.TXT for details.
*
*/
#include "tutorial3.hxx"
/*
This is a rewriter.
It looks for if statements with a comparison followed by literal return false
and modifies the return statements to 'return maybereturntrue;'
*/
namespace loplugin
{
// Ctor, pass arguments.
Tutorial3::Tutorial3( CompilerInstance& compiler, Rewriter& rewriter )
: RewritePlugin( compiler, rewriter )
{
}
void Tutorial3::run()
{
TraverseDecl( compiler.getASTContext().getTranslationUnitDecl());
}
bool Tutorial3::VisitIfStmt( const IfStmt* ifstmt )
{
if( ignoreLocation( ifstmt ))
return true;
if( const BinaryOperator* oper = dyn_cast< BinaryOperator >( ifstmt->getCond()))
{
if( oper->getOpcode() == BO_EQ )
{
// Modify the sub-statement if it is 'return false'.
modifyReturnFalse( ifstmt->getThen());
// Modify the sub-statement if it is '{ return false; }'.
if( const CompoundStmt* compound = dyn_cast< CompoundStmt >( ifstmt->getThen()))
{
if( compound->size() == 1 ) // one statement
modifyReturnFalse( *compound->body_begin());
}
}
}
return true;
}
void Tutorial3::modifyReturnFalse( const Stmt* stmt )
{
// Is it return statement?
if( const ReturnStmt* returnstmt = dyn_cast< ReturnStmt >( stmt ))
{
// dyn_cast_or_null<> can also be passed NULL, unlike dyn_cast<>
if( const CXXBoolLiteralExpr* boolliteral = dyn_cast_or_null< CXXBoolLiteralExpr >( returnstmt->getRetValue()))
{
if( boolliteral->getValue() == false )
{ // It is, modify the false to true using LO plugin helper function.
replaceText( boolliteral->getSourceRange(), "maybereturntrue" );
}
}
}
}
// Register the plugin action with the LO plugin handling.
static Plugin::Registration< Tutorial3 > X( "tutorial3" );
} // namespace
|
