forked from OSchip/llvm-project
125 lines
4.2 KiB
C++
125 lines
4.2 KiB
C++
//===- unittests/StaticAnalyzer/ParamRegionTest.cpp -----------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Reusables.h"
|
|
|
|
#include "clang/Tooling/Tooling.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace clang {
|
|
namespace ento {
|
|
namespace {
|
|
|
|
class ParamRegionTestConsumer : public ExprEngineConsumer {
|
|
void checkForSameParamRegions(MemRegionManager &MRMgr,
|
|
const StackFrameContext *SFC,
|
|
const ParmVarDecl *PVD) {
|
|
ASSERT_TRUE(llvm::all_of(PVD->redecls(), [&](const clang::VarDecl *D2) {
|
|
return MRMgr.getVarRegion(PVD, SFC) ==
|
|
MRMgr.getVarRegion(cast<ParmVarDecl>(D2), SFC);
|
|
}));
|
|
}
|
|
|
|
void performTest(const Decl *D) {
|
|
StoreManager &StMgr = Eng.getStoreManager();
|
|
MemRegionManager &MRMgr = StMgr.getRegionManager();
|
|
const StackFrameContext *SFC =
|
|
Eng.getAnalysisDeclContextManager().getStackFrame(D);
|
|
|
|
if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
|
|
for (const auto *P : FD->parameters()) {
|
|
if (SFC->inTopFrame())
|
|
assert(isa<NonParamVarRegion>(MRMgr.getVarRegion(P, SFC)));
|
|
else
|
|
assert(isa<ParamVarRegion>(MRMgr.getVarRegion(P, SFC)));
|
|
checkForSameParamRegions(MRMgr, SFC, P);
|
|
}
|
|
} else if (const auto *CD = dyn_cast<CXXConstructorDecl>(D)) {
|
|
for (const auto *P : CD->parameters()) {
|
|
if (SFC->inTopFrame())
|
|
assert(isa<NonParamVarRegion>(MRMgr.getVarRegion(P, SFC)));
|
|
else
|
|
assert(isa<ParamVarRegion>(MRMgr.getVarRegion(P, SFC)));
|
|
checkForSameParamRegions(MRMgr, SFC, P);
|
|
}
|
|
} else if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
|
|
for (const auto *P : MD->parameters()) {
|
|
if (SFC->inTopFrame())
|
|
assert(isa<NonParamVarRegion>(MRMgr.getVarRegion(P, SFC)));
|
|
else
|
|
assert(isa<ParamVarRegion>(MRMgr.getVarRegion(P, SFC)));
|
|
checkForSameParamRegions(MRMgr, SFC, P);
|
|
}
|
|
}
|
|
}
|
|
|
|
public:
|
|
ParamRegionTestConsumer(CompilerInstance &C) : ExprEngineConsumer(C) {}
|
|
|
|
bool HandleTopLevelDecl(DeclGroupRef DG) override {
|
|
for (const auto *D : DG) {
|
|
performTest(D);
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
|
|
class ParamRegionTestAction : public ASTFrontendAction {
|
|
public:
|
|
std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &Compiler,
|
|
StringRef File) override {
|
|
return std::make_unique<ParamRegionTestConsumer>(Compiler);
|
|
}
|
|
};
|
|
|
|
TEST(ParamRegion, ParamRegionTest) {
|
|
EXPECT_TRUE(
|
|
tooling::runToolOnCode(std::make_unique<ParamRegionTestAction>(),
|
|
R"(void foo(int n);
|
|
void baz(int p);
|
|
|
|
void foo(int n) {
|
|
auto lambda = [n](int m) {
|
|
return n + m;
|
|
};
|
|
|
|
int k = lambda(2);
|
|
}
|
|
|
|
void bar(int l) {
|
|
foo(l);
|
|
}
|
|
|
|
struct S {
|
|
int n;
|
|
S(int nn): n(nn) {}
|
|
};
|
|
|
|
void baz(int p) {
|
|
S s(p);
|
|
}
|
|
|
|
void bar(int l);
|
|
void baz(int p);)"));
|
|
EXPECT_TRUE(
|
|
tooling::runToolOnCode(std::make_unique<ParamRegionTestAction>(),
|
|
R"(@interface O
|
|
+ alloc;
|
|
- initWithInt:(int)q;
|
|
@end
|
|
|
|
void qix(int r) {
|
|
O *o = [[O alloc] initWithInt:r];
|
|
})",
|
|
"input.m"));
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace ento
|
|
} // namespace clang
|