MacOSKeychainAPIChecker.cpp revision 852aa0d2c5d2d1faf2d77b5aa3c0848068a342c5
1//==--- MacOSKeychainAPIChecker.cpp ------------------------------*- C++ -*-==//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9// This checker flags misuses of KeyChainAPI. In particular, the password data
10// allocated/returned by SecKeychainItemCopyContent,
11// SecKeychainFindGenericPassword, SecKeychainFindInternetPassword functions has
12// to be freed using a call to SecKeychainItemFreeContent.
13//===----------------------------------------------------------------------===//
14
15#include "ClangSACheckers.h"
16#include "clang/StaticAnalyzer/Core/Checker.h"
17#include "clang/StaticAnalyzer/Core/CheckerManager.h"
18#include "clang/StaticAnalyzer/Core/BugReporter/BugType.h"
19#include "clang/StaticAnalyzer/Core/PathSensitive/CheckerContext.h"
20#include "clang/StaticAnalyzer/Core/PathSensitive/ProgramState.h"
21#include "clang/StaticAnalyzer/Core/PathSensitive/ProgramStateTrait.h"
22#include "llvm/ADT/SmallString.h"
23
24using namespace clang;
25using namespace ento;
26
27namespace {
28class MacOSKeychainAPIChecker : public Checker<check::PreStmt<CallExpr>,
29                                               check::PreStmt<ReturnStmt>,
30                                               check::PostStmt<CallExpr>,
31                                               check::EndPath,
32                                               check::DeadSymbols> {
33  mutable OwningPtr<BugType> BT;
34
35public:
36  /// AllocationState is a part of the checker specific state together with the
37  /// MemRegion corresponding to the allocated data.
38  struct AllocationState {
39    /// The index of the allocator function.
40    unsigned int AllocatorIdx;
41    SymbolRef Region;
42
43    AllocationState(const Expr *E, unsigned int Idx, SymbolRef R) :
44      AllocatorIdx(Idx),
45      Region(R) {}
46
47    bool operator==(const AllocationState &X) const {
48      return (AllocatorIdx == X.AllocatorIdx &&
49              Region == X.Region);
50    }
51
52    void Profile(llvm::FoldingSetNodeID &ID) const {
53      ID.AddInteger(AllocatorIdx);
54      ID.AddPointer(Region);
55    }
56  };
57
58  void checkPreStmt(const CallExpr *S, CheckerContext &C) const;
59  void checkPreStmt(const ReturnStmt *S, CheckerContext &C) const;
60  void checkPostStmt(const CallExpr *S, CheckerContext &C) const;
61  void checkDeadSymbols(SymbolReaper &SR, CheckerContext &C) const;
62  void checkEndPath(CheckerContext &C) const;
63
64private:
65  typedef std::pair<SymbolRef, const AllocationState*> AllocationPair;
66  typedef llvm::SmallVector<AllocationPair, 2> AllocationPairVec;
67
68  enum APIKind {
69    /// Denotes functions tracked by this checker.
70    ValidAPI = 0,
71    /// The functions commonly/mistakenly used in place of the given API.
72    ErrorAPI = 1,
73    /// The functions which may allocate the data. These are tracked to reduce
74    /// the false alarm rate.
75    PossibleAPI = 2
76  };
77  /// Stores the information about the allocator and deallocator functions -
78  /// these are the functions the checker is tracking.
79  struct ADFunctionInfo {
80    const char* Name;
81    unsigned int Param;
82    unsigned int DeallocatorIdx;
83    APIKind Kind;
84  };
85  static const unsigned InvalidIdx = 100000;
86  static const unsigned FunctionsToTrackSize = 8;
87  static const ADFunctionInfo FunctionsToTrack[FunctionsToTrackSize];
88  /// The value, which represents no error return value for allocator functions.
89  static const unsigned NoErr = 0;
90
91  /// Given the function name, returns the index of the allocator/deallocator
92  /// function.
93  static unsigned getTrackedFunctionIndex(StringRef Name, bool IsAllocator);
94
95  inline void initBugType() const {
96    if (!BT)
97      BT.reset(new BugType("Improper use of SecKeychain API", "Mac OS API"));
98  }
99
100  void generateDeallocatorMismatchReport(const AllocationPair &AP,
101                                         const Expr *ArgExpr,
102                                         CheckerContext &C) const;
103
104  /// Find the allocation site for Sym on the path leading to the node N.
105  const Stmt *getAllocationSite(const ExplodedNode *N, SymbolRef Sym,
106                                CheckerContext &C) const;
107
108  BugReport *generateAllocatedDataNotReleasedReport(const AllocationPair &AP,
109                                                    ExplodedNode *N,
110                                                    CheckerContext &C) const;
111
112  /// Check if RetSym evaluates to an error value in the current state.
113  bool definitelyReturnedError(SymbolRef RetSym,
114                               ProgramStateRef State,
115                               SValBuilder &Builder,
116                               bool noError = false) const;
117
118  /// Check if RetSym evaluates to a NoErr value in the current state.
119  bool definitelyDidnotReturnError(SymbolRef RetSym,
120                                   ProgramStateRef State,
121                                   SValBuilder &Builder) const {
122    return definitelyReturnedError(RetSym, State, Builder, true);
123  }
124
125  /// Mark an AllocationPair interesting for diagnostic reporting.
126  void markInteresting(BugReport *R, const AllocationPair &AP) const {
127    R->markInteresting(AP.first);
128    R->markInteresting(AP.second->Region);
129  }
130
131  /// The bug visitor which allows us to print extra diagnostics along the
132  /// BugReport path. For example, showing the allocation site of the leaked
133  /// region.
134  class SecKeychainBugVisitor
135    : public BugReporterVisitorImpl<SecKeychainBugVisitor> {
136  protected:
137    // The allocated region symbol tracked by the main analysis.
138    SymbolRef Sym;
139
140  public:
141    SecKeychainBugVisitor(SymbolRef S) : Sym(S) {}
142    virtual ~SecKeychainBugVisitor() {}
143
144    void Profile(llvm::FoldingSetNodeID &ID) const {
145      static int X = 0;
146      ID.AddPointer(&X);
147      ID.AddPointer(Sym);
148    }
149
150    PathDiagnosticPiece *VisitNode(const ExplodedNode *N,
151                                   const ExplodedNode *PrevN,
152                                   BugReporterContext &BRC,
153                                   BugReport &BR);
154  };
155};
156}
157
158/// ProgramState traits to store the currently allocated (and not yet freed)
159/// symbols. This is a map from the allocated content symbol to the
160/// corresponding AllocationState.
161typedef llvm::ImmutableMap<SymbolRef,
162                       MacOSKeychainAPIChecker::AllocationState> AllocatedSetTy;
163
164namespace { struct AllocatedData {}; }
165namespace clang { namespace ento {
166template<> struct ProgramStateTrait<AllocatedData>
167    :  public ProgramStatePartialTrait<AllocatedSetTy > {
168  static void *GDMIndex() { static int index = 0; return &index; }
169};
170}}
171
172static bool isEnclosingFunctionParam(const Expr *E) {
173  E = E->IgnoreParenCasts();
174  if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
175    const ValueDecl *VD = DRE->getDecl();
176    if (isa<ImplicitParamDecl>(VD) || isa<ParmVarDecl>(VD))
177      return true;
178  }
179  return false;
180}
181
182const MacOSKeychainAPIChecker::ADFunctionInfo
183  MacOSKeychainAPIChecker::FunctionsToTrack[FunctionsToTrackSize] = {
184    {"SecKeychainItemCopyContent", 4, 3, ValidAPI},                       // 0
185    {"SecKeychainFindGenericPassword", 6, 3, ValidAPI},                   // 1
186    {"SecKeychainFindInternetPassword", 13, 3, ValidAPI},                 // 2
187    {"SecKeychainItemFreeContent", 1, InvalidIdx, ValidAPI},              // 3
188    {"SecKeychainItemCopyAttributesAndData", 5, 5, ValidAPI},             // 4
189    {"SecKeychainItemFreeAttributesAndData", 1, InvalidIdx, ValidAPI},    // 5
190    {"free", 0, InvalidIdx, ErrorAPI},                                    // 6
191    {"CFStringCreateWithBytesNoCopy", 1, InvalidIdx, PossibleAPI},        // 7
192};
193
194unsigned MacOSKeychainAPIChecker::getTrackedFunctionIndex(StringRef Name,
195                                                          bool IsAllocator) {
196  for (unsigned I = 0; I < FunctionsToTrackSize; ++I) {
197    ADFunctionInfo FI = FunctionsToTrack[I];
198    if (FI.Name != Name)
199      continue;
200    // Make sure the function is of the right type (allocator vs deallocator).
201    if (IsAllocator && (FI.DeallocatorIdx == InvalidIdx))
202      return InvalidIdx;
203    if (!IsAllocator && (FI.DeallocatorIdx != InvalidIdx))
204      return InvalidIdx;
205
206    return I;
207  }
208  // The function is not tracked.
209  return InvalidIdx;
210}
211
212static bool isBadDeallocationArgument(const MemRegion *Arg) {
213  if (!Arg)
214    return false;
215  if (isa<AllocaRegion>(Arg) ||
216      isa<BlockDataRegion>(Arg) ||
217      isa<TypedRegion>(Arg)) {
218    return true;
219  }
220  return false;
221}
222
223/// Given the address expression, retrieve the value it's pointing to. Assume
224/// that value is itself an address, and return the corresponding symbol.
225static SymbolRef getAsPointeeSymbol(const Expr *Expr,
226                                    CheckerContext &C) {
227  ProgramStateRef State = C.getState();
228  SVal ArgV = State->getSVal(Expr, C.getLocationContext());
229
230  if (const loc::MemRegionVal *X = dyn_cast<loc::MemRegionVal>(&ArgV)) {
231    StoreManager& SM = C.getStoreManager();
232    SymbolRef sym = SM.getBinding(State->getStore(), *X).getAsLocSymbol();
233    if (sym)
234      return sym;
235  }
236  return 0;
237}
238
239// When checking for error code, we need to consider the following cases:
240// 1) noErr / [0]
241// 2) someErr / [1, inf]
242// 3) unknown
243// If noError, returns true iff (1).
244// If !noError, returns true iff (2).
245bool MacOSKeychainAPIChecker::definitelyReturnedError(SymbolRef RetSym,
246                                                      ProgramStateRef State,
247                                                      SValBuilder &Builder,
248                                                      bool noError) const {
249  DefinedOrUnknownSVal NoErrVal = Builder.makeIntVal(NoErr,
250    Builder.getSymbolManager().getType(RetSym));
251  DefinedOrUnknownSVal NoErr = Builder.evalEQ(State, NoErrVal,
252                                                     nonloc::SymbolVal(RetSym));
253  ProgramStateRef ErrState = State->assume(NoErr, noError);
254  if (ErrState == State) {
255    return true;
256  }
257
258  return false;
259}
260
261// Report deallocator mismatch. Remove the region from tracking - reporting a
262// missing free error after this one is redundant.
263void MacOSKeychainAPIChecker::
264  generateDeallocatorMismatchReport(const AllocationPair &AP,
265                                    const Expr *ArgExpr,
266                                    CheckerContext &C) const {
267  ProgramStateRef State = C.getState();
268  State = State->remove<AllocatedData>(AP.first);
269  ExplodedNode *N = C.addTransition(State);
270
271  if (!N)
272    return;
273  initBugType();
274  SmallString<80> sbuf;
275  llvm::raw_svector_ostream os(sbuf);
276  unsigned int PDeallocIdx =
277               FunctionsToTrack[AP.second->AllocatorIdx].DeallocatorIdx;
278
279  os << "Deallocator doesn't match the allocator: '"
280     << FunctionsToTrack[PDeallocIdx].Name << "' should be used.";
281  BugReport *Report = new BugReport(*BT, os.str(), N);
282  Report->addVisitor(new SecKeychainBugVisitor(AP.first));
283  Report->addRange(ArgExpr->getSourceRange());
284  markInteresting(Report, AP);
285  C.EmitReport(Report);
286}
287
288void MacOSKeychainAPIChecker::checkPreStmt(const CallExpr *CE,
289                                           CheckerContext &C) const {
290  unsigned idx = InvalidIdx;
291  ProgramStateRef State = C.getState();
292
293  StringRef funName = C.getCalleeName(CE);
294  if (funName.empty())
295    return;
296
297  // If it is a call to an allocator function, it could be a double allocation.
298  idx = getTrackedFunctionIndex(funName, true);
299  if (idx != InvalidIdx) {
300    const Expr *ArgExpr = CE->getArg(FunctionsToTrack[idx].Param);
301    if (SymbolRef V = getAsPointeeSymbol(ArgExpr, C))
302      if (const AllocationState *AS = State->get<AllocatedData>(V)) {
303        if (!definitelyReturnedError(AS->Region, State, C.getSValBuilder())) {
304          // Remove the value from the state. The new symbol will be added for
305          // tracking when the second allocator is processed in checkPostStmt().
306          State = State->remove<AllocatedData>(V);
307          ExplodedNode *N = C.addTransition(State);
308          if (!N)
309            return;
310          initBugType();
311          SmallString<128> sbuf;
312          llvm::raw_svector_ostream os(sbuf);
313          unsigned int DIdx = FunctionsToTrack[AS->AllocatorIdx].DeallocatorIdx;
314          os << "Allocated data should be released before another call to "
315              << "the allocator: missing a call to '"
316              << FunctionsToTrack[DIdx].Name
317              << "'.";
318          BugReport *Report = new BugReport(*BT, os.str(), N);
319          Report->addVisitor(new SecKeychainBugVisitor(V));
320          Report->addRange(ArgExpr->getSourceRange());
321          Report->markInteresting(AS->Region);
322          C.EmitReport(Report);
323        }
324      }
325    return;
326  }
327
328  // Is it a call to one of deallocator functions?
329  idx = getTrackedFunctionIndex(funName, false);
330  if (idx == InvalidIdx)
331    return;
332
333  // Check the argument to the deallocator.
334  const Expr *ArgExpr = CE->getArg(FunctionsToTrack[idx].Param);
335  SVal ArgSVal = State->getSVal(ArgExpr, C.getLocationContext());
336
337  // Undef is reported by another checker.
338  if (ArgSVal.isUndef())
339    return;
340
341  SymbolRef ArgSM = ArgSVal.getAsLocSymbol();
342
343  // If the argument is coming from the heap, globals, or unknown, do not
344  // report it.
345  bool RegionArgIsBad = false;
346  if (!ArgSM) {
347    if (!isBadDeallocationArgument(ArgSVal.getAsRegion()))
348      return;
349    RegionArgIsBad = true;
350  }
351
352  // Is the argument to the call being tracked?
353  const AllocationState *AS = State->get<AllocatedData>(ArgSM);
354  if (!AS && FunctionsToTrack[idx].Kind != ValidAPI) {
355    return;
356  }
357  // If trying to free data which has not been allocated yet, report as a bug.
358  // TODO: We might want a more precise diagnostic for double free
359  // (that would involve tracking all the freed symbols in the checker state).
360  if (!AS || RegionArgIsBad) {
361    // It is possible that this is a false positive - the argument might
362    // have entered as an enclosing function parameter.
363    if (isEnclosingFunctionParam(ArgExpr))
364      return;
365
366    ExplodedNode *N = C.addTransition(State);
367    if (!N)
368      return;
369    initBugType();
370    BugReport *Report = new BugReport(*BT,
371        "Trying to free data which has not been allocated.", N);
372    Report->addRange(ArgExpr->getSourceRange());
373    if (AS)
374      Report->markInteresting(AS->Region);
375    C.EmitReport(Report);
376    return;
377  }
378
379  // Process functions which might deallocate.
380  if (FunctionsToTrack[idx].Kind == PossibleAPI) {
381
382    if (funName == "CFStringCreateWithBytesNoCopy") {
383      const Expr *DeallocatorExpr = CE->getArg(5)->IgnoreParenCasts();
384      // NULL ~ default deallocator, so warn.
385      if (DeallocatorExpr->isNullPointerConstant(C.getASTContext(),
386          Expr::NPC_ValueDependentIsNotNull)) {
387        const AllocationPair AP = std::make_pair(ArgSM, AS);
388        generateDeallocatorMismatchReport(AP, ArgExpr, C);
389        return;
390      }
391      // One of the default allocators, so warn.
392      if (const DeclRefExpr *DE = dyn_cast<DeclRefExpr>(DeallocatorExpr)) {
393        StringRef DeallocatorName = DE->getFoundDecl()->getName();
394        if (DeallocatorName == "kCFAllocatorDefault" ||
395            DeallocatorName == "kCFAllocatorSystemDefault" ||
396            DeallocatorName == "kCFAllocatorMalloc") {
397          const AllocationPair AP = std::make_pair(ArgSM, AS);
398          generateDeallocatorMismatchReport(AP, ArgExpr, C);
399          return;
400        }
401        // If kCFAllocatorNull, which does not deallocate, we still have to
402        // find the deallocator. Otherwise, assume that the user had written a
403        // custom deallocator which does the right thing.
404        if (DE->getFoundDecl()->getName() != "kCFAllocatorNull") {
405          State = State->remove<AllocatedData>(ArgSM);
406          C.addTransition(State);
407          return;
408        }
409      }
410    }
411    return;
412  }
413
414  // The call is deallocating a value we previously allocated, so remove it
415  // from the next state.
416  State = State->remove<AllocatedData>(ArgSM);
417
418  // Check if the proper deallocator is used.
419  unsigned int PDeallocIdx = FunctionsToTrack[AS->AllocatorIdx].DeallocatorIdx;
420  if (PDeallocIdx != idx || (FunctionsToTrack[idx].Kind == ErrorAPI)) {
421    const AllocationPair AP = std::make_pair(ArgSM, AS);
422    generateDeallocatorMismatchReport(AP, ArgExpr, C);
423    return;
424  }
425
426  // If the buffer can be null and the return status can be an error,
427  // report a bad call to free.
428  if (State->assume(cast<DefinedSVal>(ArgSVal), false) &&
429      !definitelyDidnotReturnError(AS->Region, State, C.getSValBuilder())) {
430    ExplodedNode *N = C.addTransition(State);
431    if (!N)
432      return;
433    initBugType();
434    BugReport *Report = new BugReport(*BT,
435        "Only call free if a valid (non-NULL) buffer was returned.", N);
436    Report->addVisitor(new SecKeychainBugVisitor(ArgSM));
437    Report->addRange(ArgExpr->getSourceRange());
438    Report->markInteresting(AS->Region);
439    C.EmitReport(Report);
440    return;
441  }
442
443  C.addTransition(State);
444}
445
446void MacOSKeychainAPIChecker::checkPostStmt(const CallExpr *CE,
447                                            CheckerContext &C) const {
448  ProgramStateRef State = C.getState();
449  StringRef funName = C.getCalleeName(CE);
450
451  // If a value has been allocated, add it to the set for tracking.
452  unsigned idx = getTrackedFunctionIndex(funName, true);
453  if (idx == InvalidIdx)
454    return;
455
456  const Expr *ArgExpr = CE->getArg(FunctionsToTrack[idx].Param);
457  // If the argument entered as an enclosing function parameter, skip it to
458  // avoid false positives.
459  if (isEnclosingFunctionParam(ArgExpr) &&
460      C.getLocationContext()->getParent() == 0)
461    return;
462
463  if (SymbolRef V = getAsPointeeSymbol(ArgExpr, C)) {
464    // If the argument points to something that's not a symbolic region, it
465    // can be:
466    //  - unknown (cannot reason about it)
467    //  - undefined (already reported by other checker)
468    //  - constant (null - should not be tracked,
469    //              other constant will generate a compiler warning)
470    //  - goto (should be reported by other checker)
471
472    // The call return value symbol should stay alive for as long as the
473    // allocated value symbol, since our diagnostics depend on the value
474    // returned by the call. Ex: Data should only be freed if noErr was
475    // returned during allocation.)
476    SymbolRef RetStatusSymbol =
477      State->getSVal(CE, C.getLocationContext()).getAsSymbol();
478    C.getSymbolManager().addSymbolDependency(V, RetStatusSymbol);
479
480    // Track the allocated value in the checker state.
481    State = State->set<AllocatedData>(V, AllocationState(ArgExpr, idx,
482                                                         RetStatusSymbol));
483    assert(State);
484    C.addTransition(State);
485  }
486}
487
488void MacOSKeychainAPIChecker::checkPreStmt(const ReturnStmt *S,
489                                           CheckerContext &C) const {
490  const Expr *retExpr = S->getRetValue();
491  if (!retExpr)
492    return;
493
494  // If inside inlined call, skip it.
495  const LocationContext *LC = C.getLocationContext();
496  if (LC->getParent() != 0)
497    return;
498
499  // Check  if the value is escaping through the return.
500  ProgramStateRef state = C.getState();
501  SymbolRef sym = state->getSVal(retExpr, LC).getAsLocSymbol();
502  if (!sym)
503    return;
504  state = state->remove<AllocatedData>(sym);
505
506  // Proceed from the new state.
507  C.addTransition(state);
508}
509
510// TODO: This logic is the same as in Malloc checker.
511const Stmt *
512MacOSKeychainAPIChecker::getAllocationSite(const ExplodedNode *N,
513                                           SymbolRef Sym,
514                                           CheckerContext &C) const {
515  const LocationContext *LeakContext = N->getLocationContext();
516  // Walk the ExplodedGraph backwards and find the first node that referred to
517  // the tracked symbol.
518  const ExplodedNode *AllocNode = N;
519
520  while (N) {
521    if (!N->getState()->get<AllocatedData>(Sym))
522      break;
523    // Allocation node, is the last node in the current context in which the
524    // symbol was tracked.
525    if (N->getLocationContext() == LeakContext)
526      AllocNode = N;
527    N = N->pred_empty() ? NULL : *(N->pred_begin());
528  }
529
530  ProgramPoint P = AllocNode->getLocation();
531  if (CallExitEnd *Exit = dyn_cast<CallExitEnd>(&P))
532    return Exit->getCalleeContext()->getCallSite();
533  if (clang::PostStmt *PS = dyn_cast<clang::PostStmt>(&P))
534    return PS->getStmt();
535  return 0;
536}
537
538BugReport *MacOSKeychainAPIChecker::
539  generateAllocatedDataNotReleasedReport(const AllocationPair &AP,
540                                         ExplodedNode *N,
541                                         CheckerContext &C) const {
542  const ADFunctionInfo &FI = FunctionsToTrack[AP.second->AllocatorIdx];
543  initBugType();
544  SmallString<70> sbuf;
545  llvm::raw_svector_ostream os(sbuf);
546  os << "Allocated data is not released: missing a call to '"
547      << FunctionsToTrack[FI.DeallocatorIdx].Name << "'.";
548
549  // Most bug reports are cached at the location where they occurred.
550  // With leaks, we want to unique them by the location where they were
551  // allocated, and only report a single path.
552  PathDiagnosticLocation LocUsedForUniqueing;
553  if (const Stmt *AllocStmt = getAllocationSite(N, AP.first, C))
554    LocUsedForUniqueing = PathDiagnosticLocation::createBegin(AllocStmt,
555                            C.getSourceManager(), N->getLocationContext());
556
557  BugReport *Report = new BugReport(*BT, os.str(), N, LocUsedForUniqueing);
558  Report->addVisitor(new SecKeychainBugVisitor(AP.first));
559  markInteresting(Report, AP);
560  return Report;
561}
562
563void MacOSKeychainAPIChecker::checkDeadSymbols(SymbolReaper &SR,
564                                               CheckerContext &C) const {
565  ProgramStateRef State = C.getState();
566  AllocatedSetTy ASet = State->get<AllocatedData>();
567  if (ASet.isEmpty())
568    return;
569
570  bool Changed = false;
571  AllocationPairVec Errors;
572  for (AllocatedSetTy::iterator I = ASet.begin(), E = ASet.end(); I != E; ++I) {
573    if (SR.isLive(I->first))
574      continue;
575
576    Changed = true;
577    State = State->remove<AllocatedData>(I->first);
578    // If the allocated symbol is null or if the allocation call might have
579    // returned an error, do not report.
580    if (State->getSymVal(I->first) ||
581        definitelyReturnedError(I->second.Region, State, C.getSValBuilder()))
582      continue;
583    Errors.push_back(std::make_pair(I->first, &I->second));
584  }
585  if (!Changed) {
586    // Generate the new, cleaned up state.
587    C.addTransition(State);
588    return;
589  }
590
591  static SimpleProgramPointTag Tag("MacOSKeychainAPIChecker : DeadSymbolsLeak");
592  ExplodedNode *N = C.addTransition(C.getState(), C.getPredecessor(), &Tag);
593
594  // Generate the error reports.
595  for (AllocationPairVec::iterator I = Errors.begin(), E = Errors.end();
596                                                       I != E; ++I) {
597    C.EmitReport(generateAllocatedDataNotReleasedReport(*I, N, C));
598  }
599
600  // Generate the new, cleaned up state.
601  C.addTransition(State, N);
602}
603
604// TODO: Remove this after we ensure that checkDeadSymbols are always called.
605void MacOSKeychainAPIChecker::checkEndPath(CheckerContext &C) const {
606  ProgramStateRef state = C.getState();
607
608  // If inside inlined call, skip it.
609  if (C.getLocationContext()->getParent() != 0)
610    return;
611
612  AllocatedSetTy AS = state->get<AllocatedData>();
613  if (AS.isEmpty())
614    return;
615
616  // Anything which has been allocated but not freed (nor escaped) will be
617  // found here, so report it.
618  bool Changed = false;
619  AllocationPairVec Errors;
620  for (AllocatedSetTy::iterator I = AS.begin(), E = AS.end(); I != E; ++I ) {
621    Changed = true;
622    state = state->remove<AllocatedData>(I->first);
623    // If the allocated symbol is null or if error code was returned at
624    // allocation, do not report.
625    if (state->getSymVal(I.getKey()) ||
626        definitelyReturnedError(I->second.Region, state,
627                                C.getSValBuilder())) {
628      continue;
629    }
630    Errors.push_back(std::make_pair(I->first, &I->second));
631  }
632
633  // If no change, do not generate a new state.
634  if (!Changed) {
635    C.addTransition(state);
636    return;
637  }
638
639  static SimpleProgramPointTag Tag("MacOSKeychainAPIChecker : EndPathLeak");
640  ExplodedNode *N = C.addTransition(C.getState(), C.getPredecessor(), &Tag);
641
642  // Generate the error reports.
643  for (AllocationPairVec::iterator I = Errors.begin(), E = Errors.end();
644                                                       I != E; ++I) {
645    C.EmitReport(generateAllocatedDataNotReleasedReport(*I, N, C));
646  }
647
648  C.addTransition(state, N);
649}
650
651
652PathDiagnosticPiece *MacOSKeychainAPIChecker::SecKeychainBugVisitor::VisitNode(
653                                                      const ExplodedNode *N,
654                                                      const ExplodedNode *PrevN,
655                                                      BugReporterContext &BRC,
656                                                      BugReport &BR) {
657  const AllocationState *AS = N->getState()->get<AllocatedData>(Sym);
658  if (!AS)
659    return 0;
660  const AllocationState *ASPrev = PrevN->getState()->get<AllocatedData>(Sym);
661  if (ASPrev)
662    return 0;
663
664  // (!ASPrev && AS) ~ We started tracking symbol in node N, it must be the
665  // allocation site.
666  const CallExpr *CE = cast<CallExpr>(cast<StmtPoint>(N->getLocation())
667                                                            .getStmt());
668  const FunctionDecl *funDecl = CE->getDirectCallee();
669  assert(funDecl && "We do not support indirect function calls as of now.");
670  StringRef funName = funDecl->getName();
671
672  // Get the expression of the corresponding argument.
673  unsigned Idx = getTrackedFunctionIndex(funName, true);
674  assert(Idx != InvalidIdx && "This should be a call to an allocator.");
675  const Expr *ArgExpr = CE->getArg(FunctionsToTrack[Idx].Param);
676  PathDiagnosticLocation Pos(ArgExpr, BRC.getSourceManager(),
677                             N->getLocationContext());
678  return new PathDiagnosticEventPiece(Pos, "Data is allocated here.");
679}
680
681void ento::registerMacOSKeychainAPIChecker(CheckerManager &mgr) {
682  mgr.registerChecker<MacOSKeychainAPIChecker>();
683}
684