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