1d99bd55aSTed Kremenek //=== CastSizeChecker.cpp ---------------------------------------*- C++ -*-===//
2d99bd55aSTed Kremenek //
3d99bd55aSTed Kremenek //                     The LLVM Compiler Infrastructure
4d99bd55aSTed Kremenek //
5d99bd55aSTed Kremenek // This file is distributed under the University of Illinois Open Source
6d99bd55aSTed Kremenek // License. See LICENSE.TXT for details.
7d99bd55aSTed Kremenek //
8d99bd55aSTed Kremenek //===----------------------------------------------------------------------===//
9d99bd55aSTed Kremenek //
10d99bd55aSTed Kremenek // CastSizeChecker checks when casting a malloc'ed symbolic region to type T,
11d99bd55aSTed Kremenek // whether the size of the symbolic region is a multiple of the size of T.
12d99bd55aSTed Kremenek //
13d99bd55aSTed Kremenek //===----------------------------------------------------------------------===//
14d99bd55aSTed Kremenek #include "clang/AST/CharUnits.h"
15d99bd55aSTed Kremenek #include "clang/StaticAnalyzer/BugReporter/BugType.h"
16d99bd55aSTed Kremenek #include "clang/StaticAnalyzer/PathSensitive/CheckerVisitor.h"
17d99bd55aSTed Kremenek #include "ExprEngineInternalChecks.h"
18d99bd55aSTed Kremenek 
19d99bd55aSTed Kremenek using namespace clang;
20d99bd55aSTed Kremenek using namespace ento;
21d99bd55aSTed Kremenek 
22d99bd55aSTed Kremenek namespace {
23d99bd55aSTed Kremenek class CastSizeChecker : public CheckerVisitor<CastSizeChecker> {
24d99bd55aSTed Kremenek   BuiltinBug *BT;
25d99bd55aSTed Kremenek public:
26d99bd55aSTed Kremenek   CastSizeChecker() : BT(0) {}
27d99bd55aSTed Kremenek   static void *getTag();
28d99bd55aSTed Kremenek   void PreVisitCastExpr(CheckerContext &C, const CastExpr *B);
29d99bd55aSTed Kremenek };
30d99bd55aSTed Kremenek }
31d99bd55aSTed Kremenek 
32d99bd55aSTed Kremenek void *CastSizeChecker::getTag() {
33d99bd55aSTed Kremenek   static int x;
34d99bd55aSTed Kremenek   return &x;
35d99bd55aSTed Kremenek }
36d99bd55aSTed Kremenek 
37d99bd55aSTed Kremenek void CastSizeChecker::PreVisitCastExpr(CheckerContext &C, const CastExpr *CE) {
38d99bd55aSTed Kremenek   const Expr *E = CE->getSubExpr();
39d99bd55aSTed Kremenek   ASTContext &Ctx = C.getASTContext();
40d99bd55aSTed Kremenek   QualType ToTy = Ctx.getCanonicalType(CE->getType());
41*424cec97SJohn McCall   const PointerType *ToPTy = dyn_cast<PointerType>(ToTy.getTypePtr());
42d99bd55aSTed Kremenek 
43d99bd55aSTed Kremenek   if (!ToPTy)
44d99bd55aSTed Kremenek     return;
45d99bd55aSTed Kremenek 
46d99bd55aSTed Kremenek   QualType ToPointeeTy = ToPTy->getPointeeType();
47d99bd55aSTed Kremenek 
48d99bd55aSTed Kremenek   // Only perform the check if 'ToPointeeTy' is a complete type.
49d99bd55aSTed Kremenek   if (ToPointeeTy->isIncompleteType())
50d99bd55aSTed Kremenek     return;
51d99bd55aSTed Kremenek 
52d99bd55aSTed Kremenek   const GRState *state = C.getState();
53d99bd55aSTed Kremenek   const MemRegion *R = state->getSVal(E).getAsRegion();
54d99bd55aSTed Kremenek   if (R == 0)
55d99bd55aSTed Kremenek     return;
56d99bd55aSTed Kremenek 
57d99bd55aSTed Kremenek   const SymbolicRegion *SR = dyn_cast<SymbolicRegion>(R);
58d99bd55aSTed Kremenek   if (SR == 0)
59d99bd55aSTed Kremenek     return;
60d99bd55aSTed Kremenek 
61d99bd55aSTed Kremenek   SValBuilder &svalBuilder = C.getSValBuilder();
62d99bd55aSTed Kremenek   SVal extent = SR->getExtent(svalBuilder);
63d99bd55aSTed Kremenek   const llvm::APSInt *extentInt = svalBuilder.getKnownValue(state, extent);
64d99bd55aSTed Kremenek   if (!extentInt)
65d99bd55aSTed Kremenek     return;
66d99bd55aSTed Kremenek 
67d99bd55aSTed Kremenek   CharUnits regionSize = CharUnits::fromQuantity(extentInt->getSExtValue());
68d99bd55aSTed Kremenek   CharUnits typeSize = C.getASTContext().getTypeSizeInChars(ToPointeeTy);
69d99bd55aSTed Kremenek 
70d99bd55aSTed Kremenek   // Ignore void, and a few other un-sizeable types.
71d99bd55aSTed Kremenek   if (typeSize.isZero())
72d99bd55aSTed Kremenek     return;
73d99bd55aSTed Kremenek 
74d99bd55aSTed Kremenek   if (regionSize % typeSize != 0) {
75d99bd55aSTed Kremenek     if (ExplodedNode *errorNode = C.generateSink()) {
76d99bd55aSTed Kremenek       if (!BT)
77d99bd55aSTed Kremenek         BT = new BuiltinBug("Cast region with wrong size.",
78d99bd55aSTed Kremenek                             "Cast a region whose size is not a multiple of the"
79d99bd55aSTed Kremenek                             " destination type size.");
80d99bd55aSTed Kremenek       RangedBugReport *R = new RangedBugReport(*BT, BT->getDescription(),
81d99bd55aSTed Kremenek                                                errorNode);
82d99bd55aSTed Kremenek       R->addRange(CE->getSourceRange());
83d99bd55aSTed Kremenek       C.EmitReport(R);
84d99bd55aSTed Kremenek     }
85d99bd55aSTed Kremenek   }
86d99bd55aSTed Kremenek }
87d99bd55aSTed Kremenek 
88d99bd55aSTed Kremenek 
89d99bd55aSTed Kremenek void ento::RegisterCastSizeChecker(ExprEngine &Eng) {
90d99bd55aSTed Kremenek   Eng.registerCheck(new CastSizeChecker());
91d99bd55aSTed Kremenek }
92