contrib/llvm/tools/clang/lib/Frontend/TextDiagnosticBuffer.cpp

   1 //===--- TextDiagnosticBuffer.cpp - Buffer Text Diagnostics ---------------===//
   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 is a concrete diagnostic client, which buffers the diagnostic messages.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "clang/Frontend/TextDiagnosticBuffer.h"
  15 #include "llvm/ADT/SmallString.h"
  16 #include "llvm/Support/ErrorHandling.h"
  17 using namespace clang;
  18
  19 /// HandleDiagnostic - Store the errors, warnings, and notes that are
  20 /// reported.
  21 ///
  22 void TextDiagnosticBuffer::HandleDiagnostic(DiagnosticsEngine::Level Level,
  23                                             const Diagnostic &Info) {
  24   // Default implementation (Warnings/errors count).
  25   DiagnosticConsumer::HandleDiagnostic(Level, Info);
  26
  27   SmallString<100> Buf;
  28   Info.FormatDiagnostic(Buf);
  29   switch (Level) {
  30   default: llvm_unreachable(
  31                          "Diagnostic not handled during diagnostic buffering!");
  32   case DiagnosticsEngine::Note:
  33     Notes.push_back(std::make_pair(Info.getLocation(), Buf.str()));
  34     break;
  35   case DiagnosticsEngine::Warning:
  36     Warnings.push_back(std::make_pair(Info.getLocation(), Buf.str()));
  37     break;
  38   case DiagnosticsEngine::Error:
  39   case DiagnosticsEngine::Fatal:
  40     Errors.push_back(std::make_pair(Info.getLocation(), Buf.str()));
  41     break;
  42   }
  43 }
  44
  45 /// \brief Escape diagnostic texts to avoid problems when they are fed into the
  46 /// diagnostic formatter a second time.
  47 static StringRef escapeDiag(StringRef Str, SmallVectorImpl<char> &Buf) {
  48   size_t Pos = Str.find('%');
  49   if (Pos == StringRef::npos)
  50     return Str;
  51
  52   // We found a '%'. Replace this and all following '%' with '%%'.
  53   Buf.clear();
  54   Buf.append(Str.data(), Str.data() + Pos);
  55   for (size_t I = Pos, E = Str.size(); I != E; ++I) {
  56     if (Str[I] == '%')
  57       Buf.push_back('%');
  58     Buf.push_back(Str[I]);
  59   }
  60
  61   return StringRef(Buf.data(), Buf.size());
  62 }
  63
  64 void TextDiagnosticBuffer::FlushDiagnostics(DiagnosticsEngine &Diags) const {
  65   SmallVector<char, 64> Buf;
  66   // FIXME: Flush the diagnostics in order.
  67   for (const_iterator it = err_begin(), ie = err_end(); it != ie; ++it)
  68     Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Error,
  69                                        escapeDiag(it->second, Buf)));
  70   for (const_iterator it = warn_begin(), ie = warn_end(); it != ie; ++it)
  71     Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Warning,
  72                                        escapeDiag(it->second, Buf)));
  73   for (const_iterator it = note_begin(), ie = note_end(); it != ie; ++it)
  74     Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Note,
  75                                        escapeDiag(it->second, Buf)));
  76 }
  77