1d99bd55aSTed Kremenek //== ReturnUndefChecker.cpp -------------------------------------*- C++ -*--==//
2d99bd55aSTed Kremenek //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6d99bd55aSTed Kremenek //
7d99bd55aSTed Kremenek //===----------------------------------------------------------------------===//
8d99bd55aSTed Kremenek //
9d99bd55aSTed Kremenek // This file defines ReturnUndefChecker, which is a path-sensitive
10d99bd55aSTed Kremenek // check which looks for undefined or garbage values being returned to the
11d99bd55aSTed Kremenek // caller.
12d99bd55aSTed Kremenek //
13d99bd55aSTed Kremenek //===----------------------------------------------------------------------===//
14d99bd55aSTed Kremenek 
1576a21502SKristof Umann #include "clang/StaticAnalyzer/Checkers/BuiltinCheckerRegistration.h"
163a02247dSChandler Carruth #include "clang/StaticAnalyzer/Core/BugReporter/BugType.h"
176a5674ffSArgyrios Kyrtzidis #include "clang/StaticAnalyzer/Core/Checker.h"
184dc7fb37SArgyrios Kyrtzidis #include "clang/StaticAnalyzer/Core/CheckerManager.h"
195d2964e7SAnna Zaks #include "clang/StaticAnalyzer/Core/PathSensitive/CallEvent.h"
204dc7fb37SArgyrios Kyrtzidis #include "clang/StaticAnalyzer/Core/PathSensitive/CheckerContext.h"
21d99bd55aSTed Kremenek 
22d99bd55aSTed Kremenek using namespace clang;
23d99bd55aSTed Kremenek using namespace ento;
24d99bd55aSTed Kremenek 
25d99bd55aSTed Kremenek namespace {
26b41977f8SJordan Rose class ReturnUndefChecker : public Checker< check::PreStmt<ReturnStmt> > {
27b8984329SAhmed Charles   mutable std::unique_ptr<BuiltinBug> BT_Undef;
28b8984329SAhmed Charles   mutable std::unique_ptr<BuiltinBug> BT_NullReference;
29b41977f8SJordan Rose 
30b41977f8SJordan Rose   void emitUndef(CheckerContext &C, const Expr *RetE) const;
31b41977f8SJordan Rose   void checkReference(CheckerContext &C, const Expr *RetE,
32b41977f8SJordan Rose                       DefinedOrUnknownSVal RetVal) const;
33d99bd55aSTed Kremenek public:
344dc7fb37SArgyrios Kyrtzidis   void checkPreStmt(const ReturnStmt *RS, CheckerContext &C) const;
35d99bd55aSTed Kremenek };
36ab9db510SAlexander Kornienko }
37d99bd55aSTed Kremenek 
checkPreStmt(const ReturnStmt * RS,CheckerContext & C) const384dc7fb37SArgyrios Kyrtzidis void ReturnUndefChecker::checkPreStmt(const ReturnStmt *RS,
394dc7fb37SArgyrios Kyrtzidis                                       CheckerContext &C) const {
40d99bd55aSTed Kremenek   const Expr *RetE = RS->getRetValue();
41d99bd55aSTed Kremenek   if (!RetE)
42d99bd55aSTed Kremenek     return;
43b41977f8SJordan Rose   SVal RetVal = C.getSVal(RetE);
44d99bd55aSTed Kremenek 
45b41977f8SJordan Rose   const StackFrameContext *SFC = C.getStackFrame();
46b41977f8SJordan Rose   QualType RT = CallEvent::getDeclaredResultType(SFC->getDecl());
47d99bd55aSTed Kremenek 
48b41977f8SJordan Rose   if (RetVal.isUndef()) {
49b41977f8SJordan Rose     // "return;" is modeled to evaluate to an UndefinedVal. Allow UndefinedVal
50b41977f8SJordan Rose     // to be returned in functions returning void to support this pattern:
515d2964e7SAnna Zaks     //   void foo() {
525d2964e7SAnna Zaks     //     return;
535d2964e7SAnna Zaks     //   }
545d2964e7SAnna Zaks     //   void test() {
555d2964e7SAnna Zaks     //     return foo();
565d2964e7SAnna Zaks     //   }
57add14263SJordan Rose     if (!RT.isNull() && RT->isVoidType())
58add14263SJordan Rose       return;
59add14263SJordan Rose 
60add14263SJordan Rose     // Not all blocks have explicitly-specified return types; if the return type
61add14263SJordan Rose     // is not available, but the return value expression has 'void' type, assume
62add14263SJordan Rose     // Sema already checked it.
63add14263SJordan Rose     if (RT.isNull() && isa<BlockDecl>(SFC->getDecl()) &&
64add14263SJordan Rose         RetE->getType()->isVoidType())
65add14263SJordan Rose       return;
66add14263SJordan Rose 
67b41977f8SJordan Rose     emitUndef(C, RetE);
68b41977f8SJordan Rose     return;
69b41977f8SJordan Rose   }
70b41977f8SJordan Rose 
71b41977f8SJordan Rose   if (RT.isNull())
725d2964e7SAnna Zaks     return;
735d2964e7SAnna Zaks 
74b41977f8SJordan Rose   if (RT->isReferenceType()) {
75b41977f8SJordan Rose     checkReference(C, RetE, RetVal.castAs<DefinedOrUnknownSVal>());
76b41977f8SJordan Rose     return;
77b41977f8SJordan Rose   }
78b41977f8SJordan Rose }
79d99bd55aSTed Kremenek 
emitBug(CheckerContext & C,BuiltinBug & BT,const Expr * RetE,const Expr * TrackingE=nullptr)80b41977f8SJordan Rose static void emitBug(CheckerContext &C, BuiltinBug &BT, const Expr *RetE,
810dbb783cSCraig Topper                     const Expr *TrackingE = nullptr) {
82e39bd407SDevin Coughlin   ExplodedNode *N = C.generateErrorNode();
83d99bd55aSTed Kremenek   if (!N)
84d99bd55aSTed Kremenek     return;
85d99bd55aSTed Kremenek 
862f169e7cSArtem Dergachev   auto Report =
872f169e7cSArtem Dergachev       std::make_unique<PathSensitiveBugReport>(BT, BT.getDescription(), N);
88d99bd55aSTed Kremenek 
89b41977f8SJordan Rose   Report->addRange(RetE->getSourceRange());
90b2cf0063SGeorge Karpenkov   bugreporter::trackExpressionValue(N, TrackingE ? TrackingE : RetE, *Report);
91d99bd55aSTed Kremenek 
928d3a7a56SAaron Ballman   C.emitReport(std::move(Report));
93b41977f8SJordan Rose }
94d99bd55aSTed Kremenek 
emitUndef(CheckerContext & C,const Expr * RetE) const95b41977f8SJordan Rose void ReturnUndefChecker::emitUndef(CheckerContext &C, const Expr *RetE) const {
96b41977f8SJordan Rose   if (!BT_Undef)
974aca9b1cSAlexander Kornienko     BT_Undef.reset(
984aca9b1cSAlexander Kornienko         new BuiltinBug(this, "Garbage return value",
994aca9b1cSAlexander Kornienko                        "Undefined or garbage value returned to caller"));
100b41977f8SJordan Rose   emitBug(C, *BT_Undef, RetE);
101b41977f8SJordan Rose }
102b41977f8SJordan Rose 
checkReference(CheckerContext & C,const Expr * RetE,DefinedOrUnknownSVal RetVal) const103b41977f8SJordan Rose void ReturnUndefChecker::checkReference(CheckerContext &C, const Expr *RetE,
104b41977f8SJordan Rose                                         DefinedOrUnknownSVal RetVal) const {
105b41977f8SJordan Rose   ProgramStateRef StNonNull, StNull;
106867ea1d4SBenjamin Kramer   std::tie(StNonNull, StNull) = C.getState()->assume(RetVal);
107b41977f8SJordan Rose 
108b41977f8SJordan Rose   if (StNonNull) {
109b41977f8SJordan Rose     // Going forward, assume the location is non-null.
110b41977f8SJordan Rose     C.addTransition(StNonNull);
111b41977f8SJordan Rose     return;
112b41977f8SJordan Rose   }
113b41977f8SJordan Rose 
114b41977f8SJordan Rose   // The return value is known to be null. Emit a bug report.
115b41977f8SJordan Rose   if (!BT_NullReference)
1164aca9b1cSAlexander Kornienko     BT_NullReference.reset(new BuiltinBug(this, "Returning null reference"));
117b41977f8SJordan Rose 
118b41977f8SJordan Rose   emitBug(C, *BT_NullReference, RetE, bugreporter::getDerefExpr(RetE));
119d99bd55aSTed Kremenek }
1204dc7fb37SArgyrios Kyrtzidis 
registerReturnUndefChecker(CheckerManager & mgr)1214dc7fb37SArgyrios Kyrtzidis void ento::registerReturnUndefChecker(CheckerManager &mgr) {
1224dc7fb37SArgyrios Kyrtzidis   mgr.registerChecker<ReturnUndefChecker>();
1234dc7fb37SArgyrios Kyrtzidis }
124058a7a45SKristof Umann 
shouldRegisterReturnUndefChecker(const CheckerManager & mgr)125*bda3dd0dSKirstóf Umann bool ento::shouldRegisterReturnUndefChecker(const CheckerManager &mgr) {
126058a7a45SKristof Umann   return true;
127058a7a45SKristof Umann }
128