Diagnostic.cpp revision ba6f816d633e3b88c38c6896c2d78d19489650f2 
1//===--- Diagnostic.cpp - C Language Family Diagnostic Handling -----------===//
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//
10//  This file implements the Diagnostic-related interfaces.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/AST/ASTDiagnostic.h"
15#include "clang/Analysis/AnalysisDiagnostic.h"
16#include "clang/Basic/Diagnostic.h"
17#include "clang/Basic/FileManager.h"
18#include "clang/Basic/IdentifierTable.h"
19#include "clang/Basic/PartialDiagnostic.h"
20#include "clang/Basic/SourceLocation.h"
21#include "clang/Basic/SourceManager.h"
22#include "clang/Driver/DriverDiagnostic.h"
23#include "clang/Frontend/FrontendDiagnostic.h"
24#include "clang/Lex/LexDiagnostic.h"
25#include "clang/Parse/ParseDiagnostic.h"
26#include "clang/Sema/SemaDiagnostic.h"
27#include "llvm/ADT/SmallVector.h"
28#include "llvm/ADT/StringExtras.h"
29#include "llvm/Support/MemoryBuffer.h"
30#include "llvm/Support/raw_ostream.h"
31
32#include <vector>
33#include <map>
34#include <cstring>
35using namespace clang;
36
37//===----------------------------------------------------------------------===//
38// Builtin Diagnostic information
39//===----------------------------------------------------------------------===//
40
41namespace {
42
43// Diagnostic classes.
44enum {
45  CLASS_NOTE       = 0x01,
46  CLASS_WARNING    = 0x02,
47  CLASS_EXTENSION  = 0x03,
48  CLASS_ERROR      = 0x04
49};
50
51struct StaticDiagInfoRec {
52  unsigned short DiagID;
53  unsigned Mapping : 3;
54  unsigned Class : 3;
55  bool SFINAE : 1;
56  unsigned Category : 5;
57
58  const char *Description;
59  const char *OptionGroup;
60
61  bool operator<(const StaticDiagInfoRec &RHS) const {
62    return DiagID < RHS.DiagID;
63  }
64};
65
66}
67
68static const StaticDiagInfoRec StaticDiagInfo[] = {
69#define DIAG(ENUM,CLASS,DEFAULT_MAPPING,DESC,GROUP,SFINAE, CATEGORY)    \
70  { diag::ENUM, DEFAULT_MAPPING, CLASS, SFINAE, CATEGORY, DESC, GROUP },
71#include "clang/Basic/DiagnosticCommonKinds.inc"
72#include "clang/Basic/DiagnosticDriverKinds.inc"
73#include "clang/Basic/DiagnosticFrontendKinds.inc"
74#include "clang/Basic/DiagnosticLexKinds.inc"
75#include "clang/Basic/DiagnosticParseKinds.inc"
76#include "clang/Basic/DiagnosticASTKinds.inc"
77#include "clang/Basic/DiagnosticSemaKinds.inc"
78#include "clang/Basic/DiagnosticAnalysisKinds.inc"
79  { 0, 0, 0, 0, 0, 0, 0}
80};
81#undef DIAG
82
83/// GetDiagInfo - Return the StaticDiagInfoRec entry for the specified DiagID,
84/// or null if the ID is invalid.
85static const StaticDiagInfoRec *GetDiagInfo(unsigned DiagID) {
86  unsigned NumDiagEntries = sizeof(StaticDiagInfo)/sizeof(StaticDiagInfo[0])-1;
87
88  // If assertions are enabled, verify that the StaticDiagInfo array is sorted.
89#ifndef NDEBUG
90  static bool IsFirst = true;
91  if (IsFirst) {
92    for (unsigned i = 1; i != NumDiagEntries; ++i) {
93      assert(StaticDiagInfo[i-1].DiagID != StaticDiagInfo[i].DiagID &&
94             "Diag ID conflict, the enums at the start of clang::diag (in "
95             "Diagnostic.h) probably need to be increased");
96
97      assert(StaticDiagInfo[i-1] < StaticDiagInfo[i] &&
98             "Improperly sorted diag info");
99    }
100    IsFirst = false;
101  }
102#endif
103
104  // Search the diagnostic table with a binary search.
105  StaticDiagInfoRec Find = { DiagID, 0, 0, 0, 0, 0, 0 };
106
107  const StaticDiagInfoRec *Found =
108    std::lower_bound(StaticDiagInfo, StaticDiagInfo + NumDiagEntries, Find);
109  if (Found == StaticDiagInfo + NumDiagEntries ||
110      Found->DiagID != DiagID)
111    return 0;
112
113  return Found;
114}
115
116static unsigned GetDefaultDiagMapping(unsigned DiagID) {
117  if (const StaticDiagInfoRec *Info = GetDiagInfo(DiagID))
118    return Info->Mapping;
119  return diag::MAP_FATAL;
120}
121
122/// getWarningOptionForDiag - Return the lowest-level warning option that
123/// enables the specified diagnostic.  If there is no -Wfoo flag that controls
124/// the diagnostic, this returns null.
125const char *Diagnostic::getWarningOptionForDiag(unsigned DiagID) {
126  if (const StaticDiagInfoRec *Info = GetDiagInfo(DiagID))
127    return Info->OptionGroup;
128  return 0;
129}
130
131/// getWarningOptionForDiag - Return the category number that a specified
132/// DiagID belongs to, or 0 if no category.
133unsigned Diagnostic::getCategoryNumberForDiag(unsigned DiagID) {
134  if (const StaticDiagInfoRec *Info = GetDiagInfo(DiagID))
135    return Info->Category;
136  return 0;
137}
138
139/// getCategoryNameFromID - Given a category ID, return the name of the
140/// category, an empty string if CategoryID is zero, or null if CategoryID is
141/// invalid.
142const char *Diagnostic::getCategoryNameFromID(unsigned CategoryID) {
143  // Second the table of options, sorted by name for fast binary lookup.
144  static const char *CategoryNameTable[] = {
145#define GET_CATEGORY_TABLE
146#define CATEGORY(X) X,
147#include "clang/Basic/DiagnosticGroups.inc"
148#undef GET_CATEGORY_TABLE
149    "<<END>>"
150  };
151  static const size_t CategoryNameTableSize =
152    sizeof(CategoryNameTable) / sizeof(CategoryNameTable[0])-1;
153
154  if (CategoryID >= CategoryNameTableSize) return 0;
155  return CategoryNameTable[CategoryID];
156}
157
158
159
160Diagnostic::SFINAEResponse
161Diagnostic::getDiagnosticSFINAEResponse(unsigned DiagID) {
162  if (const StaticDiagInfoRec *Info = GetDiagInfo(DiagID)) {
163    if (!Info->SFINAE)
164      return SFINAE_Report;
165
166    if (Info->Class == CLASS_ERROR)
167      return SFINAE_SubstitutionFailure;
168
169    // Suppress notes, warnings, and extensions;
170    return SFINAE_Suppress;
171  }
172
173  return SFINAE_Report;
174}
175
176/// getDiagClass - Return the class field of the diagnostic.
177///
178static unsigned getBuiltinDiagClass(unsigned DiagID) {
179  if (const StaticDiagInfoRec *Info = GetDiagInfo(DiagID))
180    return Info->Class;
181  return ~0U;
182}
183
184//===----------------------------------------------------------------------===//
185// Custom Diagnostic information
186//===----------------------------------------------------------------------===//
187
188namespace clang {
189  namespace diag {
190    class CustomDiagInfo {
191      typedef std::pair<Diagnostic::Level, std::string> DiagDesc;
192      std::vector<DiagDesc> DiagInfo;
193      std::map<DiagDesc, unsigned> DiagIDs;
194    public:
195
196      /// getDescription - Return the description of the specified custom
197      /// diagnostic.
198      const char *getDescription(unsigned DiagID) const {
199        assert(this && DiagID-DIAG_UPPER_LIMIT < DiagInfo.size() &&
200               "Invalid diagnosic ID");
201        return DiagInfo[DiagID-DIAG_UPPER_LIMIT].second.c_str();
202      }
203
204      /// getLevel - Return the level of the specified custom diagnostic.
205      Diagnostic::Level getLevel(unsigned DiagID) const {
206        assert(this && DiagID-DIAG_UPPER_LIMIT < DiagInfo.size() &&
207               "Invalid diagnosic ID");
208        return DiagInfo[DiagID-DIAG_UPPER_LIMIT].first;
209      }
210
211      unsigned getOrCreateDiagID(Diagnostic::Level L, llvm::StringRef Message,
212                                 Diagnostic &Diags) {
213        DiagDesc D(L, Message);
214        // Check to see if it already exists.
215        std::map<DiagDesc, unsigned>::iterator I = DiagIDs.lower_bound(D);
216        if (I != DiagIDs.end() && I->first == D)
217          return I->second;
218
219        // If not, assign a new ID.
220        unsigned ID = DiagInfo.size()+DIAG_UPPER_LIMIT;
221        DiagIDs.insert(std::make_pair(D, ID));
222        DiagInfo.push_back(D);
223        return ID;
224      }
225    };
226
227  } // end diag namespace
228} // end clang namespace
229
230
231//===----------------------------------------------------------------------===//
232// Common Diagnostic implementation
233//===----------------------------------------------------------------------===//
234
235static void DummyArgToStringFn(Diagnostic::ArgumentKind AK, intptr_t QT,
236                               const char *Modifier, unsigned ML,
237                               const char *Argument, unsigned ArgLen,
238                               const Diagnostic::ArgumentValue *PrevArgs,
239                               unsigned NumPrevArgs,
240                               llvm::SmallVectorImpl<char> &Output,
241                               void *Cookie) {
242  const char *Str = "<can't format argument>";
243  Output.append(Str, Str+strlen(Str));
244}
245
246
247Diagnostic::Diagnostic(DiagnosticClient *client) : Client(client) {
248  ArgToStringFn = DummyArgToStringFn;
249  ArgToStringCookie = 0;
250
251  AllExtensionsSilenced = 0;
252  IgnoreAllWarnings = false;
253  WarningsAsErrors = false;
254  ErrorsAsFatal = false;
255  SuppressSystemWarnings = false;
256  SuppressAllDiagnostics = false;
257  ShowOverloads = Ovl_All;
258  ExtBehavior = Ext_Ignore;
259
260  ErrorLimit = 0;
261  TemplateBacktraceLimit = 0;
262  CustomDiagInfo = 0;
263
264  // Set all mappings to 'unset'.
265  DiagMappingsStack.clear();
266  DiagMappingsStack.push_back(DiagMappings());
267
268  Reset();
269}
270
271Diagnostic::~Diagnostic() {
272  delete CustomDiagInfo;
273}
274
275
276void Diagnostic::pushMappings() {
277  // Avoids undefined behavior when the stack has to resize.
278  DiagMappingsStack.reserve(DiagMappingsStack.size() + 1);
279  DiagMappingsStack.push_back(DiagMappingsStack.back());
280}
281
282bool Diagnostic::popMappings() {
283  if (DiagMappingsStack.size() == 1)
284    return false;
285
286  DiagMappingsStack.pop_back();
287  return true;
288}
289
290/// getCustomDiagID - Return an ID for a diagnostic with the specified message
291/// and level.  If this is the first request for this diagnosic, it is
292/// registered and created, otherwise the existing ID is returned.
293unsigned Diagnostic::getCustomDiagID(Level L, llvm::StringRef Message) {
294  if (CustomDiagInfo == 0)
295    CustomDiagInfo = new diag::CustomDiagInfo();
296  return CustomDiagInfo->getOrCreateDiagID(L, Message, *this);
297}
298
299
300/// isBuiltinWarningOrExtension - Return true if the unmapped diagnostic
301/// level of the specified diagnostic ID is a Warning or Extension.
302/// This only works on builtin diagnostics, not custom ones, and is not legal to
303/// call on NOTEs.
304bool Diagnostic::isBuiltinWarningOrExtension(unsigned DiagID) {
305  return DiagID < diag::DIAG_UPPER_LIMIT &&
306         getBuiltinDiagClass(DiagID) != CLASS_ERROR;
307}
308
309/// \brief Determine whether the given built-in diagnostic ID is a
310/// Note.
311bool Diagnostic::isBuiltinNote(unsigned DiagID) {
312  return DiagID < diag::DIAG_UPPER_LIMIT &&
313    getBuiltinDiagClass(DiagID) == CLASS_NOTE;
314}
315
316/// isBuiltinExtensionDiag - Determine whether the given built-in diagnostic
317/// ID is for an extension of some sort.  This also returns EnabledByDefault,
318/// which is set to indicate whether the diagnostic is ignored by default (in
319/// which case -pedantic enables it) or treated as a warning/error by default.
320///
321bool Diagnostic::isBuiltinExtensionDiag(unsigned DiagID,
322                                        bool &EnabledByDefault) {
323  if (DiagID >= diag::DIAG_UPPER_LIMIT ||
324      getBuiltinDiagClass(DiagID) != CLASS_EXTENSION)
325    return false;
326
327  EnabledByDefault = GetDefaultDiagMapping(DiagID) != diag::MAP_IGNORE;
328  return true;
329}
330
331void Diagnostic::Reset() {
332  ErrorOccurred = false;
333  FatalErrorOccurred = false;
334
335  NumWarnings = 0;
336  NumErrors = 0;
337  NumErrorsSuppressed = 0;
338  CurDiagID = ~0U;
339  LastDiagLevel = Ignored;
340  DelayedDiagID = 0;
341}
342
343/// getDescription - Given a diagnostic ID, return a description of the
344/// issue.
345const char *Diagnostic::getDescription(unsigned DiagID) const {
346  if (const StaticDiagInfoRec *Info = GetDiagInfo(DiagID))
347    return Info->Description;
348  return CustomDiagInfo->getDescription(DiagID);
349}
350
351void Diagnostic::SetDelayedDiagnostic(unsigned DiagID, llvm::StringRef Arg1,
352                                      llvm::StringRef Arg2) {
353  if (DelayedDiagID)
354    return;
355
356  DelayedDiagID = DiagID;
357  DelayedDiagArg1 = Arg1.str();
358  DelayedDiagArg2 = Arg2.str();
359}
360
361void Diagnostic::ReportDelayed() {
362  Report(DelayedDiagID) << DelayedDiagArg1 << DelayedDiagArg2;
363  DelayedDiagID = 0;
364  DelayedDiagArg1.clear();
365  DelayedDiagArg2.clear();
366}
367
368/// getDiagnosticLevel - Based on the way the client configured the Diagnostic
369/// object, classify the specified diagnostic ID into a Level, consumable by
370/// the DiagnosticClient.
371Diagnostic::Level Diagnostic::getDiagnosticLevel(unsigned DiagID) const {
372  // Handle custom diagnostics, which cannot be mapped.
373  if (DiagID >= diag::DIAG_UPPER_LIMIT)
374    return CustomDiagInfo->getLevel(DiagID);
375
376  unsigned DiagClass = getBuiltinDiagClass(DiagID);
377  assert(DiagClass != CLASS_NOTE && "Cannot get diagnostic level of a note!");
378  return getDiagnosticLevel(DiagID, DiagClass);
379}
380
381/// getDiagnosticLevel - Based on the way the client configured the Diagnostic
382/// object, classify the specified diagnostic ID into a Level, consumable by
383/// the DiagnosticClient.
384Diagnostic::Level
385Diagnostic::getDiagnosticLevel(unsigned DiagID, unsigned DiagClass) const {
386  // Specific non-error diagnostics may be mapped to various levels from ignored
387  // to error.  Errors can only be mapped to fatal.
388  Diagnostic::Level Result = Diagnostic::Fatal;
389
390  // Get the mapping information, if unset, compute it lazily.
391  unsigned MappingInfo = getDiagnosticMappingInfo((diag::kind)DiagID);
392  if (MappingInfo == 0) {
393    MappingInfo = GetDefaultDiagMapping(DiagID);
394    setDiagnosticMappingInternal(DiagID, MappingInfo, false);
395  }
396
397  switch (MappingInfo & 7) {
398  default: assert(0 && "Unknown mapping!");
399  case diag::MAP_IGNORE:
400    // Ignore this, unless this is an extension diagnostic and we're mapping
401    // them onto warnings or errors.
402    if (!isBuiltinExtensionDiag(DiagID) ||  // Not an extension
403        ExtBehavior == Ext_Ignore ||        // Extensions ignored anyway
404        (MappingInfo & 8) != 0)             // User explicitly mapped it.
405      return Diagnostic::Ignored;
406    Result = Diagnostic::Warning;
407    if (ExtBehavior == Ext_Error) Result = Diagnostic::Error;
408    if (Result == Diagnostic::Error && ErrorsAsFatal)
409      Result = Diagnostic::Fatal;
410    break;
411  case diag::MAP_ERROR:
412    Result = Diagnostic::Error;
413    if (ErrorsAsFatal)
414      Result = Diagnostic::Fatal;
415    break;
416  case diag::MAP_FATAL:
417    Result = Diagnostic::Fatal;
418    break;
419  case diag::MAP_WARNING:
420    // If warnings are globally mapped to ignore or error, do it.
421    if (IgnoreAllWarnings)
422      return Diagnostic::Ignored;
423
424    Result = Diagnostic::Warning;
425
426    // If this is an extension diagnostic and we're in -pedantic-error mode, and
427    // if the user didn't explicitly map it, upgrade to an error.
428    if (ExtBehavior == Ext_Error &&
429        (MappingInfo & 8) == 0 &&
430        isBuiltinExtensionDiag(DiagID))
431      Result = Diagnostic::Error;
432
433    if (WarningsAsErrors)
434      Result = Diagnostic::Error;
435    if (Result == Diagnostic::Error && ErrorsAsFatal)
436      Result = Diagnostic::Fatal;
437    break;
438
439  case diag::MAP_WARNING_NO_WERROR:
440    // Diagnostics specified with -Wno-error=foo should be set to warnings, but
441    // not be adjusted by -Werror or -pedantic-errors.
442    Result = Diagnostic::Warning;
443
444    // If warnings are globally mapped to ignore or error, do it.
445    if (IgnoreAllWarnings)
446      return Diagnostic::Ignored;
447
448    break;
449
450  case diag::MAP_ERROR_NO_WFATAL:
451    // Diagnostics specified as -Wno-fatal-error=foo should be errors, but
452    // unaffected by -Wfatal-errors.
453    Result = Diagnostic::Error;
454    break;
455  }
456
457  // Okay, we're about to return this as a "diagnostic to emit" one last check:
458  // if this is any sort of extension warning, and if we're in an __extension__
459  // block, silence it.
460  if (AllExtensionsSilenced && isBuiltinExtensionDiag(DiagID))
461    return Diagnostic::Ignored;
462
463  return Result;
464}
465
466struct WarningOption {
467  const char  *Name;
468  const short *Members;
469  const short *SubGroups;
470};
471
472#define GET_DIAG_ARRAYS
473#include "clang/Basic/DiagnosticGroups.inc"
474#undef GET_DIAG_ARRAYS
475
476// Second the table of options, sorted by name for fast binary lookup.
477static const WarningOption OptionTable[] = {
478#define GET_DIAG_TABLE
479#include "clang/Basic/DiagnosticGroups.inc"
480#undef GET_DIAG_TABLE
481};
482static const size_t OptionTableSize =
483sizeof(OptionTable) / sizeof(OptionTable[0]);
484
485static bool WarningOptionCompare(const WarningOption &LHS,
486                                 const WarningOption &RHS) {
487  return strcmp(LHS.Name, RHS.Name) < 0;
488}
489
490static void MapGroupMembers(const WarningOption *Group, diag::Mapping Mapping,
491                            Diagnostic &Diags) {
492  // Option exists, poke all the members of its diagnostic set.
493  if (const short *Member = Group->Members) {
494    for (; *Member != -1; ++Member)
495      Diags.setDiagnosticMapping(*Member, Mapping);
496  }
497
498  // Enable/disable all subgroups along with this one.
499  if (const short *SubGroups = Group->SubGroups) {
500    for (; *SubGroups != (short)-1; ++SubGroups)
501      MapGroupMembers(&OptionTable[(short)*SubGroups], Mapping, Diags);
502  }
503}
504
505/// setDiagnosticGroupMapping - Change an entire diagnostic group (e.g.
506/// "unknown-pragmas" to have the specified mapping.  This returns true and
507/// ignores the request if "Group" was unknown, false otherwise.
508bool Diagnostic::setDiagnosticGroupMapping(const char *Group,
509                                           diag::Mapping Map) {
510
511  WarningOption Key = { Group, 0, 0 };
512  const WarningOption *Found =
513  std::lower_bound(OptionTable, OptionTable + OptionTableSize, Key,
514                   WarningOptionCompare);
515  if (Found == OptionTable + OptionTableSize ||
516      strcmp(Found->Name, Group) != 0)
517    return true;  // Option not found.
518
519  MapGroupMembers(Found, Map, *this);
520  return false;
521}
522
523
524/// ProcessDiag - This is the method used to report a diagnostic that is
525/// finally fully formed.
526bool Diagnostic::ProcessDiag() {
527  DiagnosticInfo Info(this);
528
529  if (SuppressAllDiagnostics)
530    return false;
531
532  // Figure out the diagnostic level of this message.
533  Diagnostic::Level DiagLevel;
534  unsigned DiagID = Info.getID();
535
536  // ShouldEmitInSystemHeader - True if this diagnostic should be produced even
537  // in a system header.
538  bool ShouldEmitInSystemHeader;
539
540  if (DiagID >= diag::DIAG_UPPER_LIMIT) {
541    // Handle custom diagnostics, which cannot be mapped.
542    DiagLevel = CustomDiagInfo->getLevel(DiagID);
543
544    // Custom diagnostics always are emitted in system headers.
545    ShouldEmitInSystemHeader = true;
546  } else {
547    // Get the class of the diagnostic.  If this is a NOTE, map it onto whatever
548    // the diagnostic level was for the previous diagnostic so that it is
549    // filtered the same as the previous diagnostic.
550    unsigned DiagClass = getBuiltinDiagClass(DiagID);
551    if (DiagClass == CLASS_NOTE) {
552      DiagLevel = Diagnostic::Note;
553      ShouldEmitInSystemHeader = false;  // extra consideration is needed
554    } else {
555      // If this is not an error and we are in a system header, we ignore it.
556      // Check the original Diag ID here, because we also want to ignore
557      // extensions and warnings in -Werror and -pedantic-errors modes, which
558      // *map* warnings/extensions to errors.
559      ShouldEmitInSystemHeader = DiagClass == CLASS_ERROR;
560
561      DiagLevel = getDiagnosticLevel(DiagID, DiagClass);
562    }
563  }
564
565  if (DiagLevel != Diagnostic::Note) {
566    // Record that a fatal error occurred only when we see a second
567    // non-note diagnostic. This allows notes to be attached to the
568    // fatal error, but suppresses any diagnostics that follow those
569    // notes.
570    if (LastDiagLevel == Diagnostic::Fatal)
571      FatalErrorOccurred = true;
572
573    LastDiagLevel = DiagLevel;
574  }
575
576  // If a fatal error has already been emitted, silence all subsequent
577  // diagnostics.
578  if (FatalErrorOccurred) {
579    if (DiagLevel >= Diagnostic::Error) {
580      ++NumErrors;
581      ++NumErrorsSuppressed;
582    }
583
584    return false;
585  }
586
587  // If the client doesn't care about this message, don't issue it.  If this is
588  // a note and the last real diagnostic was ignored, ignore it too.
589  if (DiagLevel == Diagnostic::Ignored ||
590      (DiagLevel == Diagnostic::Note && LastDiagLevel == Diagnostic::Ignored))
591    return false;
592
593  // If this diagnostic is in a system header and is not a clang error, suppress
594  // it.
595  if (SuppressSystemWarnings && !ShouldEmitInSystemHeader &&
596      Info.getLocation().isValid() &&
597      Info.getLocation().getInstantiationLoc().isInSystemHeader() &&
598      (DiagLevel != Diagnostic::Note || LastDiagLevel == Diagnostic::Ignored)) {
599    LastDiagLevel = Diagnostic::Ignored;
600    return false;
601  }
602
603  if (DiagLevel >= Diagnostic::Error) {
604    ErrorOccurred = true;
605    ++NumErrors;
606
607    // If we've emitted a lot of errors, emit a fatal error after it to stop a
608    // flood of bogus errors.
609    if (ErrorLimit && NumErrors >= ErrorLimit &&
610        DiagLevel == Diagnostic::Error)
611      SetDelayedDiagnostic(diag::fatal_too_many_errors);
612  }
613
614  // Finally, report it.
615  Client->HandleDiagnostic(DiagLevel, Info);
616  if (Client->IncludeInDiagnosticCounts()) {
617    if (DiagLevel == Diagnostic::Warning)
618      ++NumWarnings;
619  }
620
621  CurDiagID = ~0U;
622
623  return true;
624}
625
626bool DiagnosticBuilder::Emit() {
627  // If DiagObj is null, then its soul was stolen by the copy ctor
628  // or the user called Emit().
629  if (DiagObj == 0) return false;
630
631  // When emitting diagnostics, we set the final argument count into
632  // the Diagnostic object.
633  DiagObj->NumDiagArgs = NumArgs;
634  DiagObj->NumDiagRanges = NumRanges;
635  DiagObj->NumFixItHints = NumFixItHints;
636
637  // Process the diagnostic, sending the accumulated information to the
638  // DiagnosticClient.
639  bool Emitted = DiagObj->ProcessDiag();
640
641  // Clear out the current diagnostic object.
642  unsigned DiagID = DiagObj->CurDiagID;
643  DiagObj->Clear();
644
645  // If there was a delayed diagnostic, emit it now.
646  if (DiagObj->DelayedDiagID && DiagObj->DelayedDiagID != DiagID)
647    DiagObj->ReportDelayed();
648
649  // This diagnostic is dead.
650  DiagObj = 0;
651
652  return Emitted;
653}
654
655
656DiagnosticClient::~DiagnosticClient() {}
657
658
659/// ModifierIs - Return true if the specified modifier matches specified string.
660template <std::size_t StrLen>
661static bool ModifierIs(const char *Modifier, unsigned ModifierLen,
662                       const char (&Str)[StrLen]) {
663  return StrLen-1 == ModifierLen && !memcmp(Modifier, Str, StrLen-1);
664}
665
666/// ScanForward - Scans forward, looking for the given character, skipping
667/// nested clauses and escaped characters.
668static const char *ScanFormat(const char *I, const char *E, char Target) {
669  unsigned Depth = 0;
670
671  for ( ; I != E; ++I) {
672    if (Depth == 0 && *I == Target) return I;
673    if (Depth != 0 && *I == '}') Depth--;
674
675    if (*I == '%') {
676      I++;
677      if (I == E) break;
678
679      // Escaped characters get implicitly skipped here.
680
681      // Format specifier.
682      if (!isdigit(*I) && !ispunct(*I)) {
683        for (I++; I != E && !isdigit(*I) && *I != '{'; I++) ;
684        if (I == E) break;
685        if (*I == '{')
686          Depth++;
687      }
688    }
689  }
690  return E;
691}
692
693/// HandleSelectModifier - Handle the integer 'select' modifier.  This is used
694/// like this:  %select{foo|bar|baz}2.  This means that the integer argument
695/// "%2" has a value from 0-2.  If the value is 0, the diagnostic prints 'foo'.
696/// If the value is 1, it prints 'bar'.  If it has the value 2, it prints 'baz'.
697/// This is very useful for certain classes of variant diagnostics.
698static void HandleSelectModifier(const DiagnosticInfo &DInfo, unsigned ValNo,
699                                 const char *Argument, unsigned ArgumentLen,
700                                 llvm::SmallVectorImpl<char> &OutStr) {
701  const char *ArgumentEnd = Argument+ArgumentLen;
702
703  // Skip over 'ValNo' |'s.
704  while (ValNo) {
705    const char *NextVal = ScanFormat(Argument, ArgumentEnd, '|');
706    assert(NextVal != ArgumentEnd && "Value for integer select modifier was"
707           " larger than the number of options in the diagnostic string!");
708    Argument = NextVal+1;  // Skip this string.
709    --ValNo;
710  }
711
712  // Get the end of the value.  This is either the } or the |.
713  const char *EndPtr = ScanFormat(Argument, ArgumentEnd, '|');
714
715  // Recursively format the result of the select clause into the output string.
716  DInfo.FormatDiagnostic(Argument, EndPtr, OutStr);
717}
718
719/// HandleIntegerSModifier - Handle the integer 's' modifier.  This adds the
720/// letter 's' to the string if the value is not 1.  This is used in cases like
721/// this:  "you idiot, you have %4 parameter%s4!".
722static void HandleIntegerSModifier(unsigned ValNo,
723                                   llvm::SmallVectorImpl<char> &OutStr) {
724  if (ValNo != 1)
725    OutStr.push_back('s');
726}
727
728/// HandleOrdinalModifier - Handle the integer 'ord' modifier.  This
729/// prints the ordinal form of the given integer, with 1 corresponding
730/// to the first ordinal.  Currently this is hard-coded to use the
731/// English form.
732static void HandleOrdinalModifier(unsigned ValNo,
733                                  llvm::SmallVectorImpl<char> &OutStr) {
734  assert(ValNo != 0 && "ValNo must be strictly positive!");
735
736  llvm::raw_svector_ostream Out(OutStr);
737
738  // We could use text forms for the first N ordinals, but the numeric
739  // forms are actually nicer in diagnostics because they stand out.
740  Out << ValNo;
741
742  // It is critically important that we do this perfectly for
743  // user-written sequences with over 100 elements.
744  switch (ValNo % 100) {
745  case 11:
746  case 12:
747  case 13:
748    Out << "th"; return;
749  default:
750    switch (ValNo % 10) {
751    case 1: Out << "st"; return;
752    case 2: Out << "nd"; return;
753    case 3: Out << "rd"; return;
754    default: Out << "th"; return;
755    }
756  }
757}
758
759
760/// PluralNumber - Parse an unsigned integer and advance Start.
761static unsigned PluralNumber(const char *&Start, const char *End) {
762  // Programming 101: Parse a decimal number :-)
763  unsigned Val = 0;
764  while (Start != End && *Start >= '0' && *Start <= '9') {
765    Val *= 10;
766    Val += *Start - '0';
767    ++Start;
768  }
769  return Val;
770}
771
772/// TestPluralRange - Test if Val is in the parsed range. Modifies Start.
773static bool TestPluralRange(unsigned Val, const char *&Start, const char *End) {
774  if (*Start != '[') {
775    unsigned Ref = PluralNumber(Start, End);
776    return Ref == Val;
777  }
778
779  ++Start;
780  unsigned Low = PluralNumber(Start, End);
781  assert(*Start == ',' && "Bad plural expression syntax: expected ,");
782  ++Start;
783  unsigned High = PluralNumber(Start, End);
784  assert(*Start == ']' && "Bad plural expression syntax: expected )");
785  ++Start;
786  return Low <= Val && Val <= High;
787}
788
789/// EvalPluralExpr - Actual expression evaluator for HandlePluralModifier.
790static bool EvalPluralExpr(unsigned ValNo, const char *Start, const char *End) {
791  // Empty condition?
792  if (*Start == ':')
793    return true;
794
795  while (1) {
796    char C = *Start;
797    if (C == '%') {
798      // Modulo expression
799      ++Start;
800      unsigned Arg = PluralNumber(Start, End);
801      assert(*Start == '=' && "Bad plural expression syntax: expected =");
802      ++Start;
803      unsigned ValMod = ValNo % Arg;
804      if (TestPluralRange(ValMod, Start, End))
805        return true;
806    } else {
807      assert((C == '[' || (C >= '0' && C <= '9')) &&
808             "Bad plural expression syntax: unexpected character");
809      // Range expression
810      if (TestPluralRange(ValNo, Start, End))
811        return true;
812    }
813
814    // Scan for next or-expr part.
815    Start = std::find(Start, End, ',');
816    if (Start == End)
817      break;
818    ++Start;
819  }
820  return false;
821}
822
823/// HandlePluralModifier - Handle the integer 'plural' modifier. This is used
824/// for complex plural forms, or in languages where all plurals are complex.
825/// The syntax is: %plural{cond1:form1|cond2:form2|:form3}, where condn are
826/// conditions that are tested in order, the form corresponding to the first
827/// that applies being emitted. The empty condition is always true, making the
828/// last form a default case.
829/// Conditions are simple boolean expressions, where n is the number argument.
830/// Here are the rules.
831/// condition  := expression | empty
832/// empty      :=                             -> always true
833/// expression := numeric [',' expression]    -> logical or
834/// numeric    := range                       -> true if n in range
835///             | '%' number '=' range        -> true if n % number in range
836/// range      := number
837///             | '[' number ',' number ']'   -> ranges are inclusive both ends
838///
839/// Here are some examples from the GNU gettext manual written in this form:
840/// English:
841/// {1:form0|:form1}
842/// Latvian:
843/// {0:form2|%100=11,%10=0,%10=[2,9]:form1|:form0}
844/// Gaeilge:
845/// {1:form0|2:form1|:form2}
846/// Romanian:
847/// {1:form0|0,%100=[1,19]:form1|:form2}
848/// Lithuanian:
849/// {%10=0,%100=[10,19]:form2|%10=1:form0|:form1}
850/// Russian (requires repeated form):
851/// {%100=[11,14]:form2|%10=1:form0|%10=[2,4]:form1|:form2}
852/// Slovak
853/// {1:form0|[2,4]:form1|:form2}
854/// Polish (requires repeated form):
855/// {1:form0|%100=[10,20]:form2|%10=[2,4]:form1|:form2}
856static void HandlePluralModifier(unsigned ValNo,
857                                 const char *Argument, unsigned ArgumentLen,
858                                 llvm::SmallVectorImpl<char> &OutStr) {
859  const char *ArgumentEnd = Argument + ArgumentLen;
860  while (1) {
861    assert(Argument < ArgumentEnd && "Plural expression didn't match.");
862    const char *ExprEnd = Argument;
863    while (*ExprEnd != ':') {
864      assert(ExprEnd != ArgumentEnd && "Plural missing expression end");
865      ++ExprEnd;
866    }
867    if (EvalPluralExpr(ValNo, Argument, ExprEnd)) {
868      Argument = ExprEnd + 1;
869      ExprEnd = ScanFormat(Argument, ArgumentEnd, '|');
870      OutStr.append(Argument, ExprEnd);
871      return;
872    }
873    Argument = ScanFormat(Argument, ArgumentEnd - 1, '|') + 1;
874  }
875}
876
877
878/// FormatDiagnostic - Format this diagnostic into a string, substituting the
879/// formal arguments into the %0 slots.  The result is appended onto the Str
880/// array.
881void DiagnosticInfo::
882FormatDiagnostic(llvm::SmallVectorImpl<char> &OutStr) const {
883  const char *DiagStr = getDiags()->getDescription(getID());
884  const char *DiagEnd = DiagStr+strlen(DiagStr);
885
886  FormatDiagnostic(DiagStr, DiagEnd, OutStr);
887}
888
889void DiagnosticInfo::
890FormatDiagnostic(const char *DiagStr, const char *DiagEnd,
891                 llvm::SmallVectorImpl<char> &OutStr) const {
892
893  /// FormattedArgs - Keep track of all of the arguments formatted by
894  /// ConvertArgToString and pass them into subsequent calls to
895  /// ConvertArgToString, allowing the implementation to avoid redundancies in
896  /// obvious cases.
897  llvm::SmallVector<Diagnostic::ArgumentValue, 8> FormattedArgs;
898
899  while (DiagStr != DiagEnd) {
900    if (DiagStr[0] != '%') {
901      // Append non-%0 substrings to Str if we have one.
902      const char *StrEnd = std::find(DiagStr, DiagEnd, '%');
903      OutStr.append(DiagStr, StrEnd);
904      DiagStr = StrEnd;
905      continue;
906    } else if (ispunct(DiagStr[1])) {
907      OutStr.push_back(DiagStr[1]);  // %% -> %.
908      DiagStr += 2;
909      continue;
910    }
911
912    // Skip the %.
913    ++DiagStr;
914
915    // This must be a placeholder for a diagnostic argument.  The format for a
916    // placeholder is one of "%0", "%modifier0", or "%modifier{arguments}0".
917    // The digit is a number from 0-9 indicating which argument this comes from.
918    // The modifier is a string of digits from the set [-a-z]+, arguments is a
919    // brace enclosed string.
920    const char *Modifier = 0, *Argument = 0;
921    unsigned ModifierLen = 0, ArgumentLen = 0;
922
923    // Check to see if we have a modifier.  If so eat it.
924    if (!isdigit(DiagStr[0])) {
925      Modifier = DiagStr;
926      while (DiagStr[0] == '-' ||
927             (DiagStr[0] >= 'a' && DiagStr[0] <= 'z'))
928        ++DiagStr;
929      ModifierLen = DiagStr-Modifier;
930
931      // If we have an argument, get it next.
932      if (DiagStr[0] == '{') {
933        ++DiagStr; // Skip {.
934        Argument = DiagStr;
935
936        DiagStr = ScanFormat(DiagStr, DiagEnd, '}');
937        assert(DiagStr != DiagEnd && "Mismatched {}'s in diagnostic string!");
938        ArgumentLen = DiagStr-Argument;
939        ++DiagStr;  // Skip }.
940      }
941    }
942
943    assert(isdigit(*DiagStr) && "Invalid format for argument in diagnostic");
944    unsigned ArgNo = *DiagStr++ - '0';
945
946    Diagnostic::ArgumentKind Kind = getArgKind(ArgNo);
947
948    switch (Kind) {
949    // ---- STRINGS ----
950    case Diagnostic::ak_std_string: {
951      const std::string &S = getArgStdStr(ArgNo);
952      assert(ModifierLen == 0 && "No modifiers for strings yet");
953      OutStr.append(S.begin(), S.end());
954      break;
955    }
956    case Diagnostic::ak_c_string: {
957      const char *S = getArgCStr(ArgNo);
958      assert(ModifierLen == 0 && "No modifiers for strings yet");
959
960      // Don't crash if get passed a null pointer by accident.
961      if (!S)
962        S = "(null)";
963
964      OutStr.append(S, S + strlen(S));
965      break;
966    }
967    // ---- INTEGERS ----
968    case Diagnostic::ak_sint: {
969      int Val = getArgSInt(ArgNo);
970
971      if (ModifierIs(Modifier, ModifierLen, "select")) {
972        HandleSelectModifier(*this, (unsigned)Val, Argument, ArgumentLen, OutStr);
973      } else if (ModifierIs(Modifier, ModifierLen, "s")) {
974        HandleIntegerSModifier(Val, OutStr);
975      } else if (ModifierIs(Modifier, ModifierLen, "plural")) {
976        HandlePluralModifier((unsigned)Val, Argument, ArgumentLen, OutStr);
977      } else if (ModifierIs(Modifier, ModifierLen, "ordinal")) {
978        HandleOrdinalModifier((unsigned)Val, OutStr);
979      } else {
980        assert(ModifierLen == 0 && "Unknown integer modifier");
981        llvm::raw_svector_ostream(OutStr) << Val;
982      }
983      break;
984    }
985    case Diagnostic::ak_uint: {
986      unsigned Val = getArgUInt(ArgNo);
987
988      if (ModifierIs(Modifier, ModifierLen, "select")) {
989        HandleSelectModifier(*this, Val, Argument, ArgumentLen, OutStr);
990      } else if (ModifierIs(Modifier, ModifierLen, "s")) {
991        HandleIntegerSModifier(Val, OutStr);
992      } else if (ModifierIs(Modifier, ModifierLen, "plural")) {
993        HandlePluralModifier((unsigned)Val, Argument, ArgumentLen, OutStr);
994      } else if (ModifierIs(Modifier, ModifierLen, "ordinal")) {
995        HandleOrdinalModifier(Val, OutStr);
996      } else {
997        assert(ModifierLen == 0 && "Unknown integer modifier");
998        llvm::raw_svector_ostream(OutStr) << Val;
999      }
1000      break;
1001    }
1002    // ---- NAMES and TYPES ----
1003    case Diagnostic::ak_identifierinfo: {
1004      const IdentifierInfo *II = getArgIdentifier(ArgNo);
1005      assert(ModifierLen == 0 && "No modifiers for strings yet");
1006
1007      // Don't crash if get passed a null pointer by accident.
1008      if (!II) {
1009        const char *S = "(null)";
1010        OutStr.append(S, S + strlen(S));
1011        continue;
1012      }
1013
1014      llvm::raw_svector_ostream(OutStr) << '\'' << II->getName() << '\'';
1015      break;
1016    }
1017    case Diagnostic::ak_qualtype:
1018    case Diagnostic::ak_declarationname:
1019    case Diagnostic::ak_nameddecl:
1020    case Diagnostic::ak_nestednamespec:
1021    case Diagnostic::ak_declcontext:
1022      getDiags()->ConvertArgToString(Kind, getRawArg(ArgNo),
1023                                     Modifier, ModifierLen,
1024                                     Argument, ArgumentLen,
1025                                     FormattedArgs.data(), FormattedArgs.size(),
1026                                     OutStr);
1027      break;
1028    }
1029
1030    // Remember this argument info for subsequent formatting operations.  Turn
1031    // std::strings into a null terminated string to make it be the same case as
1032    // all the other ones.
1033    if (Kind != Diagnostic::ak_std_string)
1034      FormattedArgs.push_back(std::make_pair(Kind, getRawArg(ArgNo)));
1035    else
1036      FormattedArgs.push_back(std::make_pair(Diagnostic::ak_c_string,
1037                                        (intptr_t)getArgStdStr(ArgNo).c_str()));
1038
1039  }
1040}
1041
1042StoredDiagnostic::StoredDiagnostic() { }
1043
1044StoredDiagnostic::StoredDiagnostic(Diagnostic::Level Level,
1045                                   llvm::StringRef Message)
1046  : Level(Level), Loc(), Message(Message) { }
1047
1048StoredDiagnostic::StoredDiagnostic(Diagnostic::Level Level,
1049                                   const DiagnosticInfo &Info)
1050  : Level(Level), Loc(Info.getLocation()) {
1051  llvm::SmallString<64> Message;
1052  Info.FormatDiagnostic(Message);
1053  this->Message.assign(Message.begin(), Message.end());
1054
1055  Ranges.reserve(Info.getNumRanges());
1056  for (unsigned I = 0, N = Info.getNumRanges(); I != N; ++I)
1057    Ranges.push_back(Info.getRange(I));
1058
1059  FixIts.reserve(Info.getNumFixItHints());
1060  for (unsigned I = 0, N = Info.getNumFixItHints(); I != N; ++I)
1061    FixIts.push_back(Info.getFixItHint(I));
1062}
1063
1064StoredDiagnostic::~StoredDiagnostic() { }
1065
1066static void WriteUnsigned(llvm::raw_ostream &OS, unsigned Value) {
1067  OS.write((const char *)&Value, sizeof(unsigned));
1068}
1069
1070static void WriteString(llvm::raw_ostream &OS, llvm::StringRef String) {
1071  WriteUnsigned(OS, String.size());
1072  OS.write(String.data(), String.size());
1073}
1074
1075static void WriteSourceLocation(llvm::raw_ostream &OS,
1076                                SourceManager *SM,
1077                                SourceLocation Location) {
1078  if (!SM || Location.isInvalid()) {
1079    // If we don't have a source manager or this location is invalid,
1080    // just write an invalid location.
1081    WriteUnsigned(OS, 0);
1082    WriteUnsigned(OS, 0);
1083    WriteUnsigned(OS, 0);
1084    return;
1085  }
1086
1087  Location = SM->getInstantiationLoc(Location);
1088  std::pair<FileID, unsigned> Decomposed = SM->getDecomposedLoc(Location);
1089
1090  const FileEntry *FE = SM->getFileEntryForID(Decomposed.first);
1091  if (FE)
1092    WriteString(OS, FE->getName());
1093  else {
1094    // Fallback to using the buffer name when there is no entry.
1095    WriteString(OS, SM->getBuffer(Decomposed.first)->getBufferIdentifier());
1096  }
1097
1098  WriteUnsigned(OS, SM->getLineNumber(Decomposed.first, Decomposed.second));
1099  WriteUnsigned(OS, SM->getColumnNumber(Decomposed.first, Decomposed.second));
1100}
1101
1102void StoredDiagnostic::Serialize(llvm::raw_ostream &OS) const {
1103  SourceManager *SM = 0;
1104  if (getLocation().isValid())
1105    SM = &const_cast<SourceManager &>(getLocation().getManager());
1106
1107  // Write a short header to help identify diagnostics.
1108  OS << (char)0x06 << (char)0x07;
1109
1110  // Write the diagnostic level and location.
1111  WriteUnsigned(OS, (unsigned)Level);
1112  WriteSourceLocation(OS, SM, getLocation());
1113
1114  // Write the diagnostic message.
1115  llvm::SmallString<64> Message;
1116  WriteString(OS, getMessage());
1117
1118  // Count the number of ranges that don't point into macros, since
1119  // only simple file ranges serialize well.
1120  unsigned NumNonMacroRanges = 0;
1121  for (range_iterator R = range_begin(), REnd = range_end(); R != REnd; ++R) {
1122    if (R->getBegin().isMacroID() || R->getEnd().isMacroID())
1123      continue;
1124
1125    ++NumNonMacroRanges;
1126  }
1127
1128  // Write the ranges.
1129  WriteUnsigned(OS, NumNonMacroRanges);
1130  if (NumNonMacroRanges) {
1131    for (range_iterator R = range_begin(), REnd = range_end(); R != REnd; ++R) {
1132      if (R->getBegin().isMacroID() || R->getEnd().isMacroID())
1133        continue;
1134
1135      WriteSourceLocation(OS, SM, R->getBegin());
1136      WriteSourceLocation(OS, SM, R->getEnd());
1137      WriteUnsigned(OS, R->isTokenRange());
1138    }
1139  }
1140
1141  // Determine if all of the fix-its involve rewrites with simple file
1142  // locations (not in macro instantiations). If so, we can write
1143  // fix-it information.
1144  unsigned NumFixIts = 0;
1145  for (fixit_iterator F = fixit_begin(), FEnd = fixit_end(); F != FEnd; ++F) {
1146    if (F->RemoveRange.isValid() &&
1147        (F->RemoveRange.getBegin().isMacroID() ||
1148         F->RemoveRange.getEnd().isMacroID())) {
1149      NumFixIts = 0;
1150      break;
1151    }
1152
1153    if (F->InsertionLoc.isValid() && F->InsertionLoc.isMacroID()) {
1154      NumFixIts = 0;
1155      break;
1156    }
1157
1158    ++NumFixIts;
1159  }
1160
1161  // Write the fix-its.
1162  WriteUnsigned(OS, NumFixIts);
1163  for (fixit_iterator F = fixit_begin(), FEnd = fixit_end(); F != FEnd; ++F) {
1164    WriteSourceLocation(OS, SM, F->RemoveRange.getBegin());
1165    WriteSourceLocation(OS, SM, F->RemoveRange.getEnd());
1166    WriteUnsigned(OS, F->RemoveRange.isTokenRange());
1167    WriteSourceLocation(OS, SM, F->InsertionLoc);
1168    WriteString(OS, F->CodeToInsert);
1169  }
1170}
1171
1172static bool ReadUnsigned(const char *&Memory, const char *MemoryEnd,
1173                         unsigned &Value) {
1174  if (Memory + sizeof(unsigned) > MemoryEnd)
1175    return true;
1176
1177  memmove(&Value, Memory, sizeof(unsigned));
1178  Memory += sizeof(unsigned);
1179  return false;
1180}
1181
1182static bool ReadSourceLocation(FileManager &FM, SourceManager &SM,
1183                               const char *&Memory, const char *MemoryEnd,
1184                               SourceLocation &Location) {
1185  // Read the filename.
1186  unsigned FileNameLen = 0;
1187  if (ReadUnsigned(Memory, MemoryEnd, FileNameLen) ||
1188      Memory + FileNameLen > MemoryEnd)
1189    return true;
1190
1191  llvm::StringRef FileName(Memory, FileNameLen);
1192  Memory += FileNameLen;
1193
1194  // Read the line, column.
1195  unsigned Line = 0, Column = 0;
1196  if (ReadUnsigned(Memory, MemoryEnd, Line) ||
1197      ReadUnsigned(Memory, MemoryEnd, Column))
1198    return true;
1199
1200  if (FileName.empty()) {
1201    Location = SourceLocation();
1202    return false;
1203  }
1204
1205  const FileEntry *File = FM.getFile(FileName);
1206  if (!File)
1207    return true;
1208
1209  // Make sure that this file has an entry in the source manager.
1210  if (!SM.hasFileInfo(File))
1211    SM.createFileID(File, SourceLocation(), SrcMgr::C_User);
1212
1213  Location = SM.getLocation(File, Line, Column);
1214  return false;
1215}
1216
1217StoredDiagnostic
1218StoredDiagnostic::Deserialize(FileManager &FM, SourceManager &SM,
1219                              const char *&Memory, const char *MemoryEnd) {
1220  while (true) {
1221    if (Memory == MemoryEnd)
1222      return StoredDiagnostic();
1223
1224    if (*Memory != 0x06) {
1225      ++Memory;
1226      continue;
1227    }
1228
1229    ++Memory;
1230    if (Memory == MemoryEnd)
1231      return StoredDiagnostic();
1232
1233    if (*Memory != 0x07) {
1234      ++Memory;
1235      continue;
1236    }
1237
1238    // We found the header. We're done.
1239    ++Memory;
1240    break;
1241  }
1242
1243  // Read the severity level.
1244  unsigned Level = 0;
1245  if (ReadUnsigned(Memory, MemoryEnd, Level) || Level > Diagnostic::Fatal)
1246    return StoredDiagnostic();
1247
1248  // Read the source location.
1249  SourceLocation Location;
1250  if (ReadSourceLocation(FM, SM, Memory, MemoryEnd, Location))
1251    return StoredDiagnostic();
1252
1253  // Read the diagnostic text.
1254  if (Memory == MemoryEnd)
1255    return StoredDiagnostic();
1256
1257  unsigned MessageLen = 0;
1258  if (ReadUnsigned(Memory, MemoryEnd, MessageLen) ||
1259      Memory + MessageLen > MemoryEnd)
1260    return StoredDiagnostic();
1261
1262  llvm::StringRef Message(Memory, MessageLen);
1263  Memory += MessageLen;
1264
1265
1266  // At this point, we have enough information to form a diagnostic. Do so.
1267  StoredDiagnostic Diag;
1268  Diag.Level = (Diagnostic::Level)Level;
1269  Diag.Loc = FullSourceLoc(Location, SM);
1270  Diag.Message = Message;
1271  if (Memory == MemoryEnd)
1272    return Diag;
1273
1274  // Read the source ranges.
1275  unsigned NumSourceRanges = 0;
1276  if (ReadUnsigned(Memory, MemoryEnd, NumSourceRanges))
1277    return Diag;
1278  for (unsigned I = 0; I != NumSourceRanges; ++I) {
1279    SourceLocation Begin, End;
1280    unsigned IsTokenRange;
1281    if (ReadSourceLocation(FM, SM, Memory, MemoryEnd, Begin) ||
1282        ReadSourceLocation(FM, SM, Memory, MemoryEnd, End) ||
1283        ReadUnsigned(Memory, MemoryEnd, IsTokenRange))
1284      return Diag;
1285
1286    Diag.Ranges.push_back(CharSourceRange(SourceRange(Begin, End),
1287                                          IsTokenRange));
1288  }
1289
1290  // Read the fix-it hints.
1291  unsigned NumFixIts = 0;
1292  if (ReadUnsigned(Memory, MemoryEnd, NumFixIts))
1293    return Diag;
1294  for (unsigned I = 0; I != NumFixIts; ++I) {
1295    SourceLocation RemoveBegin, RemoveEnd, InsertionLoc;
1296    unsigned InsertLen = 0, RemoveIsTokenRange;
1297    if (ReadSourceLocation(FM, SM, Memory, MemoryEnd, RemoveBegin) ||
1298        ReadSourceLocation(FM, SM, Memory, MemoryEnd, RemoveEnd) ||
1299        ReadUnsigned(Memory, MemoryEnd, RemoveIsTokenRange) ||
1300        ReadSourceLocation(FM, SM, Memory, MemoryEnd, InsertionLoc) ||
1301        ReadUnsigned(Memory, MemoryEnd, InsertLen) ||
1302        Memory + InsertLen > MemoryEnd) {
1303      Diag.FixIts.clear();
1304      return Diag;
1305    }
1306
1307    FixItHint Hint;
1308    Hint.RemoveRange = CharSourceRange(SourceRange(RemoveBegin, RemoveEnd),
1309                                       RemoveIsTokenRange);
1310    Hint.InsertionLoc = InsertionLoc;
1311    Hint.CodeToInsert.assign(Memory, Memory + InsertLen);
1312    Memory += InsertLen;
1313    Diag.FixIts.push_back(Hint);
1314  }
1315
1316  return Diag;
1317}
1318
1319/// IncludeInDiagnosticCounts - This method (whose default implementation
1320///  returns true) indicates whether the diagnostics handled by this
1321///  DiagnosticClient should be included in the number of diagnostics
1322///  reported by Diagnostic.
1323bool DiagnosticClient::IncludeInDiagnosticCounts() const { return true; }
1324
1325PartialDiagnostic::StorageAllocator::StorageAllocator() {
1326  for (unsigned I = 0; I != NumCached; ++I)
1327    FreeList[I] = Cached + I;
1328  NumFreeListEntries = NumCached;
1329}
1330
1331PartialDiagnostic::StorageAllocator::~StorageAllocator() {
1332  assert(NumFreeListEntries == NumCached && "A partial is on the lamb");
1333}
1334