Diagnostic.cpp revision 33e4e70c8c0a17e0ccb7465d96556b077a68ecb1
15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- Diagnostic.cpp - C Language Family Diagnostic Handling -----------===//
25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//                     The LLVM Compiler Infrastructure
45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source
60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details.
75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
95f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//  This file implements the Diagnostic-related interfaces.
115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
14ec55c941f2846db48bce4ed6dd2ce339e1a48962Ted Kremenek#include "clang/Basic/Diagnostic.h"
1543b628cd47ecdc3caf640d79b3ad7ecef0f2c285Chris Lattner#include "clang/Basic/IdentifierTable.h"
16ec55c941f2846db48bce4ed6dd2ce339e1a48962Ted Kremenek#include "clang/Basic/PartialDiagnostic.h"
17f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner#include "llvm/ADT/SmallVector.h"
1823e47c6b6e8ccdd8daa378ab2a879644425c72d8Daniel Dunbar#include "llvm/Support/raw_ostream.h"
195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerusing namespace clang;
205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
213fdf4b071dc79fae778fb5f376485480756c76a3Chris Lattnerstatic void DummyArgToStringFn(Diagnostic::ArgumentKind AK, intptr_t QT,
223fdf4b071dc79fae778fb5f376485480756c76a3Chris Lattner                               const char *Modifier, unsigned ML,
233fdf4b071dc79fae778fb5f376485480756c76a3Chris Lattner                               const char *Argument, unsigned ArgLen,
24b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner                               const Diagnostic::ArgumentValue *PrevArgs,
25b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner                               unsigned NumPrevArgs,
2692dd386e3f05d176b45a638199d51f536bd9d1c4Chris Lattner                               llvm::SmallVectorImpl<char> &Output,
2792dd386e3f05d176b45a638199d51f536bd9d1c4Chris Lattner                               void *Cookie) {
283fdf4b071dc79fae778fb5f376485480756c76a3Chris Lattner  const char *Str = "<can't format argument>";
2922caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner  Output.append(Str, Str+strlen(Str));
3022caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner}
3122caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner
3222caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner
3333e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios KyrtzidisDiagnostic::Diagnostic(const llvm::IntrusiveRefCntPtr<DiagnosticIDs> &diags,
3433e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis                       DiagnosticClient *client, bool ShouldOwnClient)
3533e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  : Diags(diags), Client(client), OwnsDiagClient(ShouldOwnClient),
3633e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis    SourceMgr(0) {
373fdf4b071dc79fae778fb5f376485480756c76a3Chris Lattner  ArgToStringFn = DummyArgToStringFn;
3892dd386e3f05d176b45a638199d51f536bd9d1c4Chris Lattner  ArgToStringCookie = 0;
391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
40cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  AllExtensionsSilenced = 0;
41cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  IgnoreAllWarnings = false;
42cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  WarningsAsErrors = false;
43cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  ErrorsAsFatal = false;
44cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  SuppressSystemWarnings = false;
45cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  SuppressAllDiagnostics = false;
46cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  ShowOverloads = Ovl_All;
47cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  ExtBehavior = Ext_Ignore;
48cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor
49cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  ErrorLimit = 0;
50cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  TemplateBacktraceLimit = 0;
51cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor
52cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  // Set all mappings to 'unset'.
53cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  DiagMappingsStack.clear();
54cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  DiagMappingsStack.push_back(DiagMappings());
55cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor
56abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  Reset();
57182745ae7892bca0842d9c023370ade5f8d1c6e8Chris Lattner}
58182745ae7892bca0842d9c023370ade5f8d1c6e8Chris Lattner
59182745ae7892bca0842d9c023370ade5f8d1c6e8Chris LattnerDiagnostic::~Diagnostic() {
6033e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  if (OwnsDiagClient)
6133e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis    delete Client;
62182745ae7892bca0842d9c023370ade5f8d1c6e8Chris Lattner}
63182745ae7892bca0842d9c023370ade5f8d1c6e8Chris Lattner
6404ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner
6504ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattnervoid Diagnostic::pushMappings() {
66ca2c3e2cfc630db614298e8d7f2aaca4507e01e1John Thompson  // Avoids undefined behavior when the stack has to resize.
67ca2c3e2cfc630db614298e8d7f2aaca4507e01e1John Thompson  DiagMappingsStack.reserve(DiagMappingsStack.size() + 1);
6804ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner  DiagMappingsStack.push_back(DiagMappingsStack.back());
6904ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner}
7004ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner
7104ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattnerbool Diagnostic::popMappings() {
7204ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner  if (DiagMappingsStack.size() == 1)
7304ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner    return false;
7404ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner
7504ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner  DiagMappingsStack.pop_back();
7604ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner  return true;
7704ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner}
7804ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner
79abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregorvoid Diagnostic::Reset() {
80abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  ErrorOccurred = false;
81abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  FatalErrorOccurred = false;
82abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
83abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  NumWarnings = 0;
84abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  NumErrors = 0;
85abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  NumErrorsSuppressed = 0;
86abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  CurDiagID = ~0U;
8733e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  // Set LastDiagLevel to an "unset" state. If we set it to 'Ignored', notes
8833e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  // using a Diagnostic associated to a translation unit that follow
8933e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  // diagnostics from a Diagnostic associated to anoter t.u. will not be
9033e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  // displayed.
9133e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  LastDiagLevel = (DiagnosticIDs::Level)-1;
92abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  DelayedDiagID = 0;
93abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
9593ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregorvoid Diagnostic::SetDelayedDiagnostic(unsigned DiagID, llvm::StringRef Arg1,
9693ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor                                      llvm::StringRef Arg2) {
9793ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  if (DelayedDiagID)
9893ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor    return;
9993ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
10093ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  DelayedDiagID = DiagID;
1019e2dac9c9b6bc4384c816a447cca6516a03c89f2Douglas Gregor  DelayedDiagArg1 = Arg1.str();
1029e2dac9c9b6bc4384c816a447cca6516a03c89f2Douglas Gregor  DelayedDiagArg2 = Arg2.str();
10393ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor}
10493ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
10593ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregorvoid Diagnostic::ReportDelayed() {
10693ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  Report(DelayedDiagID) << DelayedDiagArg1 << DelayedDiagArg2;
10793ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  DelayedDiagID = 0;
10893ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  DelayedDiagArg1.clear();
10993ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  DelayedDiagArg2.clear();
11093ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor}
11193ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
112b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregorvoid DiagnosticBuilder::FlushCounts() {
113b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregor  DiagObj->NumDiagArgs = NumArgs;
114b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregor  DiagObj->NumDiagRanges = NumRanges;
115b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregor  DiagObj->NumFixItHints = NumFixItHints;
116b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregor}
117b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregor
11893ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregorbool DiagnosticBuilder::Emit() {
11993ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // If DiagObj is null, then its soul was stolen by the copy ctor
12093ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // or the user called Emit().
12193ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  if (DiagObj == 0) return false;
12293ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
12393ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // When emitting diagnostics, we set the final argument count into
12493ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // the Diagnostic object.
125b535041ee33c5eff255832bc5541c8d52aae8254Douglas Gregor  FlushCounts();
12693ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
12793ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // Process the diagnostic, sending the accumulated information to the
12893ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // DiagnosticClient.
12993ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  bool Emitted = DiagObj->ProcessDiag();
13093ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
13193ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // Clear out the current diagnostic object.
1329e2dac9c9b6bc4384c816a447cca6516a03c89f2Douglas Gregor  unsigned DiagID = DiagObj->CurDiagID;
13393ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  DiagObj->Clear();
13493ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
13593ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // If there was a delayed diagnostic, emit it now.
1369e2dac9c9b6bc4384c816a447cca6516a03c89f2Douglas Gregor  if (DiagObj->DelayedDiagID && DiagObj->DelayedDiagID != DiagID)
13793ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor    DiagObj->ReportDelayed();
13893ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
13993ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  // This diagnostic is dead.
14093ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  DiagObj = 0;
14193ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
14293ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor  return Emitted;
14393ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor}
14493ea5cb0edf8e509c5113e70cb05ee247c9bdf6bDouglas Gregor
1457bfaaaecb3113f955db31e8d8a51acffd1bc0c27Nico Weber
1465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerDiagnosticClient::~DiagnosticClient() {}
1477bfaaaecb3113f955db31e8d8a51acffd1bc0c27Nico Weber
148f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner
149af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// ModifierIs - Return true if the specified modifier matches specified string.
150af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattnertemplate <std::size_t StrLen>
151af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattnerstatic bool ModifierIs(const char *Modifier, unsigned ModifierLen,
152af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner                       const char (&Str)[StrLen]) {
153af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  return StrLen-1 == ModifierLen && !memcmp(Modifier, Str, StrLen-1);
154af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner}
155af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner
156909c182f6a0e6169475eef6a71add14555b6162cJohn McCall/// ScanForward - Scans forward, looking for the given character, skipping
157909c182f6a0e6169475eef6a71add14555b6162cJohn McCall/// nested clauses and escaped characters.
158909c182f6a0e6169475eef6a71add14555b6162cJohn McCallstatic const char *ScanFormat(const char *I, const char *E, char Target) {
159909c182f6a0e6169475eef6a71add14555b6162cJohn McCall  unsigned Depth = 0;
160909c182f6a0e6169475eef6a71add14555b6162cJohn McCall
161909c182f6a0e6169475eef6a71add14555b6162cJohn McCall  for ( ; I != E; ++I) {
162909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    if (Depth == 0 && *I == Target) return I;
163909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    if (Depth != 0 && *I == '}') Depth--;
164909c182f6a0e6169475eef6a71add14555b6162cJohn McCall
165909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    if (*I == '%') {
166909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      I++;
167909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      if (I == E) break;
168909c182f6a0e6169475eef6a71add14555b6162cJohn McCall
169909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      // Escaped characters get implicitly skipped here.
170909c182f6a0e6169475eef6a71add14555b6162cJohn McCall
171909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      // Format specifier.
172909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      if (!isdigit(*I) && !ispunct(*I)) {
173909c182f6a0e6169475eef6a71add14555b6162cJohn McCall        for (I++; I != E && !isdigit(*I) && *I != '{'; I++) ;
174909c182f6a0e6169475eef6a71add14555b6162cJohn McCall        if (I == E) break;
175909c182f6a0e6169475eef6a71add14555b6162cJohn McCall        if (*I == '{')
176909c182f6a0e6169475eef6a71add14555b6162cJohn McCall          Depth++;
177909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      }
178909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    }
179909c182f6a0e6169475eef6a71add14555b6162cJohn McCall  }
180909c182f6a0e6169475eef6a71add14555b6162cJohn McCall  return E;
181909c182f6a0e6169475eef6a71add14555b6162cJohn McCall}
182909c182f6a0e6169475eef6a71add14555b6162cJohn McCall
183af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// HandleSelectModifier - Handle the integer 'select' modifier.  This is used
184af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// like this:  %select{foo|bar|baz}2.  This means that the integer argument
185af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// "%2" has a value from 0-2.  If the value is 0, the diagnostic prints 'foo'.
186af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// If the value is 1, it prints 'bar'.  If it has the value 2, it prints 'baz'.
187af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// This is very useful for certain classes of variant diagnostics.
1889f28614bf1a8387000d8df57a713fcf69e198145John McCallstatic void HandleSelectModifier(const DiagnosticInfo &DInfo, unsigned ValNo,
189af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner                                 const char *Argument, unsigned ArgumentLen,
190af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner                                 llvm::SmallVectorImpl<char> &OutStr) {
191af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  const char *ArgumentEnd = Argument+ArgumentLen;
1921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
193af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  // Skip over 'ValNo' |'s.
194af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  while (ValNo) {
195909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    const char *NextVal = ScanFormat(Argument, ArgumentEnd, '|');
196af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    assert(NextVal != ArgumentEnd && "Value for integer select modifier was"
197af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner           " larger than the number of options in the diagnostic string!");
198af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    Argument = NextVal+1;  // Skip this string.
199af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    --ValNo;
200af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  }
2011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
202af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  // Get the end of the value.  This is either the } or the |.
203909c182f6a0e6169475eef6a71add14555b6162cJohn McCall  const char *EndPtr = ScanFormat(Argument, ArgumentEnd, '|');
2049f28614bf1a8387000d8df57a713fcf69e198145John McCall
2059f28614bf1a8387000d8df57a713fcf69e198145John McCall  // Recursively format the result of the select clause into the output string.
2069f28614bf1a8387000d8df57a713fcf69e198145John McCall  DInfo.FormatDiagnostic(Argument, EndPtr, OutStr);
207af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner}
208af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner
209af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// HandleIntegerSModifier - Handle the integer 's' modifier.  This adds the
210af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// letter 's' to the string if the value is not 1.  This is used in cases like
211af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner/// this:  "you idiot, you have %4 parameter%s4!".
212af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattnerstatic void HandleIntegerSModifier(unsigned ValNo,
213af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner                                   llvm::SmallVectorImpl<char> &OutStr) {
214af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner  if (ValNo != 1)
215af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    OutStr.push_back('s');
216af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner}
217af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner
2183be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall/// HandleOrdinalModifier - Handle the integer 'ord' modifier.  This
2193be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall/// prints the ordinal form of the given integer, with 1 corresponding
2203be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall/// to the first ordinal.  Currently this is hard-coded to use the
2213be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall/// English form.
2223be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCallstatic void HandleOrdinalModifier(unsigned ValNo,
2233be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall                                  llvm::SmallVectorImpl<char> &OutStr) {
2243be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  assert(ValNo != 0 && "ValNo must be strictly positive!");
2253be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall
2263be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  llvm::raw_svector_ostream Out(OutStr);
2273be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall
2283be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  // We could use text forms for the first N ordinals, but the numeric
2293be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  // forms are actually nicer in diagnostics because they stand out.
2303be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  Out << ValNo;
2313be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall
2323be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  // It is critically important that we do this perfectly for
2333be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  // user-written sequences with over 100 elements.
2343be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  switch (ValNo % 100) {
2353be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  case 11:
2363be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  case 12:
2373be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  case 13:
2383be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    Out << "th"; return;
2393be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  default:
2403be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    switch (ValNo % 10) {
2413be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    case 1: Out << "st"; return;
2423be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    case 2: Out << "nd"; return;
2433be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    case 3: Out << "rd"; return;
2443be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    default: Out << "th"; return;
2453be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall    }
2463be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall  }
2473be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall}
2483be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall
249af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner
250e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// PluralNumber - Parse an unsigned integer and advance Start.
251d2aa7c90e7646c509f3493fa8548635ccf4a2d0aChris Lattnerstatic unsigned PluralNumber(const char *&Start, const char *End) {
252e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  // Programming 101: Parse a decimal number :-)
253e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  unsigned Val = 0;
254e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  while (Start != End && *Start >= '0' && *Start <= '9') {
255e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    Val *= 10;
256e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    Val += *Start - '0';
257e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    ++Start;
258e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  }
259e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  return Val;
260e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl}
261e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
262e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// TestPluralRange - Test if Val is in the parsed range. Modifies Start.
263d2aa7c90e7646c509f3493fa8548635ccf4a2d0aChris Lattnerstatic bool TestPluralRange(unsigned Val, const char *&Start, const char *End) {
264e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  if (*Start != '[') {
265e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    unsigned Ref = PluralNumber(Start, End);
266e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    return Ref == Val;
267e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  }
268e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
269e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  ++Start;
270e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  unsigned Low = PluralNumber(Start, End);
271e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  assert(*Start == ',' && "Bad plural expression syntax: expected ,");
272e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  ++Start;
273e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  unsigned High = PluralNumber(Start, End);
274e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  assert(*Start == ']' && "Bad plural expression syntax: expected )");
275e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  ++Start;
276e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  return Low <= Val && Val <= High;
277e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl}
278e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
279e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// EvalPluralExpr - Actual expression evaluator for HandlePluralModifier.
280d2aa7c90e7646c509f3493fa8548635ccf4a2d0aChris Lattnerstatic bool EvalPluralExpr(unsigned ValNo, const char *Start, const char *End) {
281e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  // Empty condition?
282e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  if (*Start == ':')
283e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    return true;
284e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
285e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  while (1) {
286e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    char C = *Start;
287e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    if (C == '%') {
288e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      // Modulo expression
289e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      ++Start;
290e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      unsigned Arg = PluralNumber(Start, End);
291e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      assert(*Start == '=' && "Bad plural expression syntax: expected =");
292e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      ++Start;
293e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      unsigned ValMod = ValNo % Arg;
294e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      if (TestPluralRange(ValMod, Start, End))
295e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl        return true;
296e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    } else {
297e20653219732b03294130999415fc3aa92d2336aSebastian Redl      assert((C == '[' || (C >= '0' && C <= '9')) &&
298e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl             "Bad plural expression syntax: unexpected character");
299e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      // Range expression
300e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      if (TestPluralRange(ValNo, Start, End))
301e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl        return true;
302e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    }
303e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
304e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    // Scan for next or-expr part.
305e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    Start = std::find(Start, End, ',');
3061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    if (Start == End)
307e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      break;
308e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    ++Start;
309e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  }
310e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  return false;
311e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl}
312e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
313e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// HandlePluralModifier - Handle the integer 'plural' modifier. This is used
314e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// for complex plural forms, or in languages where all plurals are complex.
315e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// The syntax is: %plural{cond1:form1|cond2:form2|:form3}, where condn are
316e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// conditions that are tested in order, the form corresponding to the first
317e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// that applies being emitted. The empty condition is always true, making the
318e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// last form a default case.
319e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Conditions are simple boolean expressions, where n is the number argument.
320e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Here are the rules.
321e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// condition  := expression | empty
322e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// empty      :=                             -> always true
323e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// expression := numeric [',' expression]    -> logical or
324e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// numeric    := range                       -> true if n in range
325e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl///             | '%' number '=' range        -> true if n % number in range
326e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// range      := number
327e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl///             | '[' number ',' number ']'   -> ranges are inclusive both ends
328e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl///
329e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Here are some examples from the GNU gettext manual written in this form:
330e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// English:
331e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {1:form0|:form1}
332e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Latvian:
333e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {0:form2|%100=11,%10=0,%10=[2,9]:form1|:form0}
334e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Gaeilge:
335e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {1:form0|2:form1|:form2}
336e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Romanian:
337e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {1:form0|0,%100=[1,19]:form1|:form2}
338e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Lithuanian:
339e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {%10=0,%100=[10,19]:form2|%10=1:form0|:form1}
340e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Russian (requires repeated form):
341e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {%100=[11,14]:form2|%10=1:form0|%10=[2,4]:form1|:form2}
342e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Slovak
343e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {1:form0|[2,4]:form1|:form2}
344e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// Polish (requires repeated form):
345e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl/// {1:form0|%100=[10,20]:form2|%10=[2,4]:form1|:form2}
346e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCallstatic void HandlePluralModifier(const DiagnosticInfo &DInfo, unsigned ValNo,
347e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl                                 const char *Argument, unsigned ArgumentLen,
348b54b276a920246c595a0498da281821eb9d22996Chris Lattner                                 llvm::SmallVectorImpl<char> &OutStr) {
349e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  const char *ArgumentEnd = Argument + ArgumentLen;
350e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  while (1) {
351e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    assert(Argument < ArgumentEnd && "Plural expression didn't match.");
352e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    const char *ExprEnd = Argument;
353e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    while (*ExprEnd != ':') {
354e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      assert(ExprEnd != ArgumentEnd && "Plural missing expression end");
355e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      ++ExprEnd;
356e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    }
357e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    if (EvalPluralExpr(ValNo, Argument, ExprEnd)) {
358e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      Argument = ExprEnd + 1;
359909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      ExprEnd = ScanFormat(Argument, ArgumentEnd, '|');
360e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall
361e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall      // Recursively format the result of the plural clause into the
362e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall      // output string.
363e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall      DInfo.FormatDiagnostic(Argument, ExprEnd, OutStr);
364e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      return;
365e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl    }
366909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    Argument = ScanFormat(Argument, ArgumentEnd - 1, '|') + 1;
367e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl  }
368e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl}
369e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
370e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl
371f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner/// FormatDiagnostic - Format this diagnostic into a string, substituting the
372f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner/// formal arguments into the %0 slots.  The result is appended onto the Str
373f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner/// array.
374f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattnervoid DiagnosticInfo::
375f4c839657742b823cea1a95b18422f1ba74d3dddChris LattnerFormatDiagnostic(llvm::SmallVectorImpl<char> &OutStr) const {
37633e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  const char *DiagStr = getDiags()->getDiagnosticIDs()->getDescription(getID());
377f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner  const char *DiagEnd = DiagStr+strlen(DiagStr);
3781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3799f28614bf1a8387000d8df57a713fcf69e198145John McCall  FormatDiagnostic(DiagStr, DiagEnd, OutStr);
3809f28614bf1a8387000d8df57a713fcf69e198145John McCall}
3819f28614bf1a8387000d8df57a713fcf69e198145John McCall
3829f28614bf1a8387000d8df57a713fcf69e198145John McCallvoid DiagnosticInfo::
3839f28614bf1a8387000d8df57a713fcf69e198145John McCallFormatDiagnostic(const char *DiagStr, const char *DiagEnd,
3849f28614bf1a8387000d8df57a713fcf69e198145John McCall                 llvm::SmallVectorImpl<char> &OutStr) const {
3859f28614bf1a8387000d8df57a713fcf69e198145John McCall
386b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner  /// FormattedArgs - Keep track of all of the arguments formatted by
387b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner  /// ConvertArgToString and pass them into subsequent calls to
388b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner  /// ConvertArgToString, allowing the implementation to avoid redundancies in
389b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner  /// obvious cases.
390b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner  llvm::SmallVector<Diagnostic::ArgumentValue, 8> FormattedArgs;
391b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner
392f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner  while (DiagStr != DiagEnd) {
393f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner    if (DiagStr[0] != '%') {
394f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner      // Append non-%0 substrings to Str if we have one.
395f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner      const char *StrEnd = std::find(DiagStr, DiagEnd, '%');
396f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner      OutStr.append(DiagStr, StrEnd);
397f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner      DiagStr = StrEnd;
398af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      continue;
399909c182f6a0e6169475eef6a71add14555b6162cJohn McCall    } else if (ispunct(DiagStr[1])) {
400909c182f6a0e6169475eef6a71add14555b6162cJohn McCall      OutStr.push_back(DiagStr[1]);  // %% -> %.
401f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner      DiagStr += 2;
402af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      continue;
403af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    }
4041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
405af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // Skip the %.
406af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    ++DiagStr;
4071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
408af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // This must be a placeholder for a diagnostic argument.  The format for a
409af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // placeholder is one of "%0", "%modifier0", or "%modifier{arguments}0".
410af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // The digit is a number from 0-9 indicating which argument this comes from.
411af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // The modifier is a string of digits from the set [-a-z]+, arguments is a
412af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // brace enclosed string.
413af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    const char *Modifier = 0, *Argument = 0;
414af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    unsigned ModifierLen = 0, ArgumentLen = 0;
4151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
416af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    // Check to see if we have a modifier.  If so eat it.
417af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    if (!isdigit(DiagStr[0])) {
418af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      Modifier = DiagStr;
419af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      while (DiagStr[0] == '-' ||
420af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner             (DiagStr[0] >= 'a' && DiagStr[0] <= 'z'))
421af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        ++DiagStr;
422af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      ModifierLen = DiagStr-Modifier;
423f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner
424af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      // If we have an argument, get it next.
425af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      if (DiagStr[0] == '{') {
426af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        ++DiagStr; // Skip {.
427af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        Argument = DiagStr;
4281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
429909c182f6a0e6169475eef6a71add14555b6162cJohn McCall        DiagStr = ScanFormat(DiagStr, DiagEnd, '}');
430909c182f6a0e6169475eef6a71add14555b6162cJohn McCall        assert(DiagStr != DiagEnd && "Mismatched {}'s in diagnostic string!");
431af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        ArgumentLen = DiagStr-Argument;
432af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        ++DiagStr;  // Skip }.
433f4c839657742b823cea1a95b18422f1ba74d3dddChris Lattner      }
434af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    }
4351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
436af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    assert(isdigit(*DiagStr) && "Invalid format for argument in diagnostic");
43722caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner    unsigned ArgNo = *DiagStr++ - '0';
438af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner
439b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    Diagnostic::ArgumentKind Kind = getArgKind(ArgNo);
440b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner
441b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    switch (Kind) {
44208631c5fa053867146b5ee8be658c229f6bf127cChris Lattner    // ---- STRINGS ----
4433cbfe2c4159e0a219ae660d50625c013aa4afbd0Chris Lattner    case Diagnostic::ak_std_string: {
44422caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner      const std::string &S = getArgStdStr(ArgNo);
445af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      assert(ModifierLen == 0 && "No modifiers for strings yet");
446af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      OutStr.append(S.begin(), S.end());
447af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      break;
448af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    }
4493cbfe2c4159e0a219ae660d50625c013aa4afbd0Chris Lattner    case Diagnostic::ak_c_string: {
45022caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner      const char *S = getArgCStr(ArgNo);
451af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      assert(ModifierLen == 0 && "No modifiers for strings yet");
452e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar
453e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar      // Don't crash if get passed a null pointer by accident.
454e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar      if (!S)
455e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar        S = "(null)";
4561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
457af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      OutStr.append(S, S + strlen(S));
458af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      break;
459af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    }
46008631c5fa053867146b5ee8be658c229f6bf127cChris Lattner    // ---- INTEGERS ----
4613cbfe2c4159e0a219ae660d50625c013aa4afbd0Chris Lattner    case Diagnostic::ak_sint: {
46222caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner      int Val = getArgSInt(ArgNo);
4631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
464af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      if (ModifierIs(Modifier, ModifierLen, "select")) {
465e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall        HandleSelectModifier(*this, (unsigned)Val, Argument, ArgumentLen,
466e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall                             OutStr);
467af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      } else if (ModifierIs(Modifier, ModifierLen, "s")) {
468af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        HandleIntegerSModifier(Val, OutStr);
469e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      } else if (ModifierIs(Modifier, ModifierLen, "plural")) {
470e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall        HandlePluralModifier(*this, (unsigned)Val, Argument, ArgumentLen,
471e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall                             OutStr);
4723be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall      } else if (ModifierIs(Modifier, ModifierLen, "ordinal")) {
4733be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall        HandleOrdinalModifier((unsigned)Val, OutStr);
474af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      } else {
475af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        assert(ModifierLen == 0 && "Unknown integer modifier");
47623e47c6b6e8ccdd8daa378ab2a879644425c72d8Daniel Dunbar        llvm::raw_svector_ostream(OutStr) << Val;
47730bc96544346bea42921cf6837e66cef80d664b4Chris Lattner      }
478af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      break;
479af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    }
4803cbfe2c4159e0a219ae660d50625c013aa4afbd0Chris Lattner    case Diagnostic::ak_uint: {
48122caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner      unsigned Val = getArgUInt(ArgNo);
4821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
483af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      if (ModifierIs(Modifier, ModifierLen, "select")) {
4849f28614bf1a8387000d8df57a713fcf69e198145John McCall        HandleSelectModifier(*this, Val, Argument, ArgumentLen, OutStr);
485af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      } else if (ModifierIs(Modifier, ModifierLen, "s")) {
486af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        HandleIntegerSModifier(Val, OutStr);
487e4c452c4c7b9124fe94a96f559ff077d59cdf996Sebastian Redl      } else if (ModifierIs(Modifier, ModifierLen, "plural")) {
488e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall        HandlePluralModifier(*this, (unsigned)Val, Argument, ArgumentLen,
489e53a44bcd342e964a3c69bc27734f01e23f5fec8John McCall                             OutStr);
4903be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall      } else if (ModifierIs(Modifier, ModifierLen, "ordinal")) {
4913be16b7d9d0ab075461ed3498b4c01b30b517c0eJohn McCall        HandleOrdinalModifier(Val, OutStr);
492af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner      } else {
493af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner        assert(ModifierLen == 0 && "Unknown integer modifier");
49423e47c6b6e8ccdd8daa378ab2a879644425c72d8Daniel Dunbar        llvm::raw_svector_ostream(OutStr) << Val;
49530bc96544346bea42921cf6837e66cef80d664b4Chris Lattner      }
49622caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner      break;
497af7ae4e8160fc5c23e471f2125b3fe5911e3532aChris Lattner    }
49808631c5fa053867146b5ee8be658c229f6bf127cChris Lattner    // ---- NAMES and TYPES ----
49908631c5fa053867146b5ee8be658c229f6bf127cChris Lattner    case Diagnostic::ak_identifierinfo: {
50008631c5fa053867146b5ee8be658c229f6bf127cChris Lattner      const IdentifierInfo *II = getArgIdentifier(ArgNo);
50108631c5fa053867146b5ee8be658c229f6bf127cChris Lattner      assert(ModifierLen == 0 && "No modifiers for strings yet");
502e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar
503e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar      // Don't crash if get passed a null pointer by accident.
504e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar      if (!II) {
505e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar        const char *S = "(null)";
506e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar        OutStr.append(S, S + strlen(S));
507e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar        continue;
508e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar      }
509e46e354ecb9a04c8d3724ae2b0f95f4424e3f69cDaniel Dunbar
51001eb9b9683535d8a65c704ad2c545903409e2d36Daniel Dunbar      llvm::raw_svector_ostream(OutStr) << '\'' << II->getName() << '\'';
51108631c5fa053867146b5ee8be658c229f6bf127cChris Lattner      break;
51208631c5fa053867146b5ee8be658c229f6bf127cChris Lattner    }
51322caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner    case Diagnostic::ak_qualtype:
514011bb4edf731d529da1cbf71c7c2696aaf5a054fChris Lattner    case Diagnostic::ak_declarationname:
51547b9a1ca55e61e37f5a368740e29de190345acc6Douglas Gregor    case Diagnostic::ak_nameddecl:
516dacd434c49658286c380c7b4c357d76d53cb4aa1Douglas Gregor    case Diagnostic::ak_nestednamespec:
5173f09327b26033d0a9676d52d80cf92c48f581affDouglas Gregor    case Diagnostic::ak_declcontext:
518b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner      getDiags()->ConvertArgToString(Kind, getRawArg(ArgNo),
5193fdf4b071dc79fae778fb5f376485480756c76a3Chris Lattner                                     Modifier, ModifierLen,
520b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner                                     Argument, ArgumentLen,
521b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner                                     FormattedArgs.data(), FormattedArgs.size(),
522b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner                                     OutStr);
52322caddc91d2f6186739c6b20ec58ed38cd68e595Chris Lattner      break;
5247bfaaaecb3113f955db31e8d8a51acffd1bc0c27Nico Weber    }
525b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner
526b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    // Remember this argument info for subsequent formatting operations.  Turn
527b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    // std::strings into a null terminated string to make it be the same case as
528b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    // all the other ones.
529b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    if (Kind != Diagnostic::ak_std_string)
530b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner      FormattedArgs.push_back(std::make_pair(Kind, getRawArg(ArgNo)));
531b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner    else
532b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner      FormattedArgs.push_back(std::make_pair(Diagnostic::ak_c_string,
533b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner                                        (intptr_t)getArgStdStr(ArgNo).c_str()));
534b54d8af9a66cc20a6a9a9219c7eaea8df7ee7fd4Chris Lattner
5357bfaaaecb3113f955db31e8d8a51acffd1bc0c27Nico Weber  }
5367bfaaaecb3113f955db31e8d8a51acffd1bc0c27Nico Weber}
537cabe66811fe43835b8c5a0854552768fc53261e3Ted Kremenek
538a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas GregorStoredDiagnostic::StoredDiagnostic() { }
539a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
540a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas GregorStoredDiagnostic::StoredDiagnostic(Diagnostic::Level Level,
541a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                   llvm::StringRef Message)
5420a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  : Level(Level), Loc(), Message(Message) { }
543a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
544a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas GregorStoredDiagnostic::StoredDiagnostic(Diagnostic::Level Level,
545a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                   const DiagnosticInfo &Info)
54633e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  : Level(Level) {
54733e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  assert((Info.getLocation().isInvalid() || Info.hasSourceManager()) &&
54833e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis       "Valid source location without setting a source manager for diagnostic");
54933e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis  if (Info.getLocation().isValid())
55033e4e70c8c0a17e0ccb7465d96556b077a68ecb1Argyrios Kyrtzidis    Loc = FullSourceLoc(Info.getLocation(), Info.getSourceManager());
551a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  llvm::SmallString<64> Message;
552a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  Info.FormatDiagnostic(Message);
553a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  this->Message.assign(Message.begin(), Message.end());
554a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
555a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  Ranges.reserve(Info.getNumRanges());
556a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  for (unsigned I = 0, N = Info.getNumRanges(); I != N; ++I)
557a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor    Ranges.push_back(Info.getRange(I));
558a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
559849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor  FixIts.reserve(Info.getNumFixItHints());
560849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor  for (unsigned I = 0, N = Info.getNumFixItHints(); I != N; ++I)
561849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor    FixIts.push_back(Info.getFixItHint(I));
562a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor}
563a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
564a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas GregorStoredDiagnostic::~StoredDiagnostic() { }
565a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
566cabe66811fe43835b8c5a0854552768fc53261e3Ted Kremenek/// IncludeInDiagnosticCounts - This method (whose default implementation
567cabe66811fe43835b8c5a0854552768fc53261e3Ted Kremenek///  returns true) indicates whether the diagnostics handled by this
568cabe66811fe43835b8c5a0854552768fc53261e3Ted Kremenek///  DiagnosticClient should be included in the number of diagnostics
569cabe66811fe43835b8c5a0854552768fc53261e3Ted Kremenek///  reported by Diagnostic.
570cabe66811fe43835b8c5a0854552768fc53261e3Ted Kremenekbool DiagnosticClient::IncludeInDiagnosticCounts() const { return true; }
571fe6b2d481d91140923f4541f273b253291884214Douglas Gregor
572fe6b2d481d91140923f4541f273b253291884214Douglas GregorPartialDiagnostic::StorageAllocator::StorageAllocator() {
573fe6b2d481d91140923f4541f273b253291884214Douglas Gregor  for (unsigned I = 0; I != NumCached; ++I)
574fe6b2d481d91140923f4541f273b253291884214Douglas Gregor    FreeList[I] = Cached + I;
575fe6b2d481d91140923f4541f273b253291884214Douglas Gregor  NumFreeListEntries = NumCached;
576fe6b2d481d91140923f4541f273b253291884214Douglas Gregor}
577fe6b2d481d91140923f4541f273b253291884214Douglas Gregor
578fe6b2d481d91140923f4541f273b253291884214Douglas GregorPartialDiagnostic::StorageAllocator::~StorageAllocator() {
579fe6b2d481d91140923f4541f273b253291884214Douglas Gregor  assert(NumFreeListEntries == NumCached && "A partial is on the lamb");
580fe6b2d481d91140923f4541f273b253291884214Douglas Gregor}
581