blob: 7bf8742a8040748b03e52ae0f30480ba010b69b9 [file] [log] [blame]
Ted Kremenek78002122011-10-29 00:12:39 +00001//===--- SerializedDiagnosticPrinter.cpp - Serializer for 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#include <vector>
Ted Kremenek78002122011-10-29 00:12:39 +000011#include "llvm/Support/raw_ostream.h"
12#include "llvm/ADT/StringRef.h"
13#include "llvm/ADT/SmallString.h"
14#include "llvm/ADT/DenseSet.h"
15#include "clang/Basic/SourceManager.h"
16#include "clang/Basic/FileManager.h"
17#include "clang/Basic/Diagnostic.h"
18#include "clang/Basic/Version.h"
Ted Kremenek15322172011-11-10 08:43:12 +000019#include "clang/Lex/Lexer.h"
Ted Kremenek78002122011-10-29 00:12:39 +000020#include "clang/Frontend/SerializedDiagnosticPrinter.h"
Ted Kremenek2a764102011-12-17 05:26:11 +000021#include "clang/Frontend/DiagnosticRenderer.h"
Ted Kremenek78002122011-10-29 00:12:39 +000022
23using namespace clang;
Ted Kremenekfdd0ced2011-11-05 00:09:57 +000024using namespace clang::serialized_diags;
Ted Kremenek78002122011-10-29 00:12:39 +000025
26namespace {
Ted Kremenek78002122011-10-29 00:12:39 +000027
28class AbbreviationMap {
29 llvm::DenseMap<unsigned, unsigned> Abbrevs;
30public:
31 AbbreviationMap() {}
32
33 void set(unsigned recordID, unsigned abbrevID) {
34 assert(Abbrevs.find(recordID) == Abbrevs.end()
35 && "Abbreviation already set.");
36 Abbrevs[recordID] = abbrevID;
37 }
38
39 unsigned get(unsigned recordID) {
40 assert(Abbrevs.find(recordID) != Abbrevs.end() &&
41 "Abbreviation not set.");
42 return Abbrevs[recordID];
43 }
44};
45
46typedef llvm::SmallVector<uint64_t, 64> RecordData;
47typedef llvm::SmallVectorImpl<uint64_t> RecordDataImpl;
Ted Kremenek2a764102011-12-17 05:26:11 +000048
49class SDiagsWriter;
50
Ted Kremenek8be51ea2012-02-14 02:46:00 +000051class SDiagsRenderer : public DiagnosticNoteRenderer {
Ted Kremenek2a764102011-12-17 05:26:11 +000052 SDiagsWriter &Writer;
53 RecordData &Record;
54public:
55 SDiagsRenderer(SDiagsWriter &Writer, RecordData &Record,
56 const SourceManager &SM,
57 const LangOptions &LangOpts,
58 const DiagnosticOptions &DiagOpts)
Ted Kremenek8be51ea2012-02-14 02:46:00 +000059 : DiagnosticNoteRenderer(SM, LangOpts, DiagOpts),
Ted Kremenek2a764102011-12-17 05:26:11 +000060 Writer(Writer), Record(Record){}
61
62 virtual ~SDiagsRenderer() {}
63
64protected:
65 virtual void emitDiagnosticMessage(SourceLocation Loc,
66 PresumedLoc PLoc,
67 DiagnosticsEngine::Level Level,
68 StringRef Message,
69 ArrayRef<CharSourceRange> Ranges,
Ted Kremenek8be51ea2012-02-14 02:46:00 +000070 DiagOrStoredDiag D);
Ted Kremenek2a764102011-12-17 05:26:11 +000071
72 virtual void emitDiagnosticLoc(SourceLocation Loc, PresumedLoc PLoc,
73 DiagnosticsEngine::Level Level,
74 ArrayRef<CharSourceRange> Ranges) {}
75
Ted Kremenek2a764102011-12-17 05:26:11 +000076 void emitNote(SourceLocation Loc, StringRef Message);
77
Ted Kremenek2a764102011-12-17 05:26:11 +000078 virtual void emitCodeContext(SourceLocation Loc,
79 DiagnosticsEngine::Level Level,
80 SmallVectorImpl<CharSourceRange>& Ranges,
81 ArrayRef<FixItHint> Hints);
82
Ted Kremenek8be51ea2012-02-14 02:46:00 +000083 virtual void beginDiagnostic(DiagOrStoredDiag D,
Ted Kremenek2a764102011-12-17 05:26:11 +000084 DiagnosticsEngine::Level Level);
Ted Kremenek8be51ea2012-02-14 02:46:00 +000085 virtual void endDiagnostic(DiagOrStoredDiag D,
Ted Kremenek2a764102011-12-17 05:26:11 +000086 DiagnosticsEngine::Level Level);
87};
Ted Kremenek78002122011-10-29 00:12:39 +000088
89class SDiagsWriter : public DiagnosticConsumer {
Ted Kremenek2a764102011-12-17 05:26:11 +000090 friend class SDiagsRenderer;
Ted Kremenek78002122011-10-29 00:12:39 +000091public:
Ted Kremenek2a764102011-12-17 05:26:11 +000092 explicit SDiagsWriter(llvm::raw_ostream *os, const DiagnosticOptions &diags)
93 : LangOpts(0), DiagOpts(diags),
94 Stream(Buffer), OS(os), inNonNoteDiagnostic(false)
Ted Kremenek78002122011-10-29 00:12:39 +000095 {
96 EmitPreamble();
Devang Patel02ae32a2011-11-15 01:30:40 +000097 }
Ted Kremenek78002122011-10-29 00:12:39 +000098
99 ~SDiagsWriter() {}
100
101 void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
102 const Diagnostic &Info);
103
Ted Kremenek15322172011-11-10 08:43:12 +0000104 void BeginSourceFile(const LangOptions &LO,
105 const Preprocessor *PP) {
106 LangOpts = &LO;
107 }
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000108
109 virtual void finish();
110
Ted Kremenek78002122011-10-29 00:12:39 +0000111 DiagnosticConsumer *clone(DiagnosticsEngine &Diags) const {
112 // It makes no sense to clone this.
113 return 0;
114 }
115
116private:
117 /// \brief Emit the preamble for the serialized diagnostics.
118 void EmitPreamble();
119
120 /// \brief Emit the BLOCKINFO block.
121 void EmitBlockInfoBlock();
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000122
Ted Kremenek0b69aa82011-11-08 20:27:29 +0000123 /// \brief Emit the META data block.
124 void EmitMetaBlock();
125
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000126 /// \brief Emit a record for a CharSourceRange.
Ted Kremenek2a764102011-12-17 05:26:11 +0000127 void EmitCharSourceRange(CharSourceRange R, const SourceManager &SM);
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000128
Ted Kremenek2a764102011-12-17 05:26:11 +0000129 /// \brief Emit the string information for the category.
130 unsigned getEmitCategory(unsigned category = 0);
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000131
132 /// \brief Emit the string information for diagnostic flags.
133 unsigned getEmitDiagnosticFlag(DiagnosticsEngine::Level DiagLevel,
Ted Kremenek2a764102011-12-17 05:26:11 +0000134 unsigned DiagID = 0);
Ted Kremenek0dbadc42011-11-05 00:10:04 +0000135
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000136 /// \brief Emit (lazily) the file string and retrieved the file identifier.
Ted Kremenek2a764102011-12-17 05:26:11 +0000137 unsigned getEmitFile(const char *Filename);
138
139 /// \brief Add SourceLocation information the specified record.
140 void AddLocToRecord(SourceLocation Loc, const SourceManager &SM,
141 PresumedLoc PLoc, RecordDataImpl &Record,
142 unsigned TokSize = 0);
143
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000144 /// \brief Add SourceLocation information the specified record.
Ted Kremenek15322172011-11-10 08:43:12 +0000145 void AddLocToRecord(SourceLocation Loc, RecordDataImpl &Record,
Ted Kremenek2a764102011-12-17 05:26:11 +0000146 const SourceManager &SM,
147 unsigned TokSize = 0) {
148 AddLocToRecord(Loc, SM, SM.getPresumedLoc(Loc), Record, TokSize);
149 }
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000150
Ted Kremenek96dcade2011-11-05 03:34:23 +0000151 /// \brief Add CharSourceRange information the specified record.
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000152 void AddCharSourceRangeToRecord(CharSourceRange R, RecordDataImpl &Record,
Ted Kremenek2a764102011-12-17 05:26:11 +0000153 const SourceManager &SM);
Ted Kremenek96dcade2011-11-05 03:34:23 +0000154
Ted Kremenek78002122011-10-29 00:12:39 +0000155 /// \brief The version of the diagnostics file.
156 enum { Version = 1 };
157
Ted Kremenek15322172011-11-10 08:43:12 +0000158 const LangOptions *LangOpts;
Ted Kremenek2a764102011-12-17 05:26:11 +0000159 const DiagnosticOptions &DiagOpts;
Ted Kremenek15322172011-11-10 08:43:12 +0000160
Ted Kremenek78002122011-10-29 00:12:39 +0000161 /// \brief The byte buffer for the serialized content.
Daniel Dunbar8d6ff022012-02-29 20:31:23 +0000162 SmallString<1024> Buffer;
Ted Kremenek78002122011-10-29 00:12:39 +0000163
164 /// \brief The BitStreamWriter for the serialized diagnostics.
165 llvm::BitstreamWriter Stream;
166
167 /// \brief The name of the diagnostics file.
Dylan Noblesmith6f42b622012-02-05 02:12:40 +0000168 OwningPtr<llvm::raw_ostream> OS;
Ted Kremenek78002122011-10-29 00:12:39 +0000169
Ted Kremenek78002122011-10-29 00:12:39 +0000170 /// \brief The set of constructed record abbreviations.
171 AbbreviationMap Abbrevs;
172
173 /// \brief A utility buffer for constructing record content.
174 RecordData Record;
175
176 /// \brief A text buffer for rendering diagnostic text.
Dylan Noblesmithf7ccbad2012-02-05 02:13:05 +0000177 SmallString<256> diagBuf;
Ted Kremenek78002122011-10-29 00:12:39 +0000178
179 /// \brief The collection of diagnostic categories used.
180 llvm::DenseSet<unsigned> Categories;
181
182 /// \brief The collection of files used.
Ted Kremenek2a764102011-12-17 05:26:11 +0000183 llvm::DenseMap<const char *, unsigned> Files;
Ted Kremenek45d92752011-11-05 00:09:50 +0000184
185 typedef llvm::DenseMap<const void *, std::pair<unsigned, llvm::StringRef> >
186 DiagFlagsTy;
187
188 /// \brief Map for uniquing strings.
189 DiagFlagsTy DiagFlags;
Ted Kremenek78002122011-10-29 00:12:39 +0000190
Ted Kremenek59b61612011-11-05 00:09:47 +0000191 /// \brief Flag indicating whether or not we are in the process of
192 /// emitting a non-note diagnostic.
193 bool inNonNoteDiagnostic;
Ted Kremenek78002122011-10-29 00:12:39 +0000194};
195} // end anonymous namespace
196
197namespace clang {
198namespace serialized_diags {
Ted Kremenek2a764102011-12-17 05:26:11 +0000199DiagnosticConsumer *create(llvm::raw_ostream *OS,
200 const DiagnosticOptions &diags) {
201 return new SDiagsWriter(OS, diags);
Ted Kremenek78002122011-10-29 00:12:39 +0000202}
203} // end namespace serialized_diags
204} // end namespace clang
205
206//===----------------------------------------------------------------------===//
207// Serialization methods.
208//===----------------------------------------------------------------------===//
209
210/// \brief Emits a block ID in the BLOCKINFO block.
211static void EmitBlockID(unsigned ID, const char *Name,
212 llvm::BitstreamWriter &Stream,
213 RecordDataImpl &Record) {
214 Record.clear();
215 Record.push_back(ID);
216 Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_SETBID, Record);
217
218 // Emit the block name if present.
219 if (Name == 0 || Name[0] == 0)
220 return;
221
222 Record.clear();
223
224 while (*Name)
225 Record.push_back(*Name++);
226
227 Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_BLOCKNAME, Record);
228}
229
230/// \brief Emits a record ID in the BLOCKINFO block.
231static void EmitRecordID(unsigned ID, const char *Name,
232 llvm::BitstreamWriter &Stream,
233 RecordDataImpl &Record){
234 Record.clear();
235 Record.push_back(ID);
236
237 while (*Name)
238 Record.push_back(*Name++);
239
240 Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_SETRECORDNAME, Record);
241}
242
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000243void SDiagsWriter::AddLocToRecord(SourceLocation Loc,
Ted Kremenek2a764102011-12-17 05:26:11 +0000244 const SourceManager &SM,
245 PresumedLoc PLoc,
Ted Kremenek15322172011-11-10 08:43:12 +0000246 RecordDataImpl &Record,
247 unsigned TokSize) {
Ted Kremenek2a764102011-12-17 05:26:11 +0000248 if (PLoc.isInvalid()) {
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000249 // Emit a "sentinel" location.
Ted Kremenek15322172011-11-10 08:43:12 +0000250 Record.push_back((unsigned)0); // File.
251 Record.push_back((unsigned)0); // Line.
252 Record.push_back((unsigned)0); // Column.
253 Record.push_back((unsigned)0); // Offset.
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000254 return;
255 }
256
Ted Kremenek2a764102011-12-17 05:26:11 +0000257 Record.push_back(getEmitFile(PLoc.getFilename()));
258 Record.push_back(PLoc.getLine());
259 Record.push_back(PLoc.getColumn()+TokSize);
Benjamin Kramer6eb29d22011-11-10 11:29:20 +0000260 Record.push_back(SM.getFileOffset(Loc));
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000261}
262
Ted Kremenek96dcade2011-11-05 03:34:23 +0000263void SDiagsWriter::AddCharSourceRangeToRecord(CharSourceRange Range,
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000264 RecordDataImpl &Record,
Ted Kremenek2a764102011-12-17 05:26:11 +0000265 const SourceManager &SM) {
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000266 AddLocToRecord(Range.getBegin(), Record, SM);
Ted Kremenek15322172011-11-10 08:43:12 +0000267 unsigned TokSize = 0;
268 if (Range.isTokenRange())
269 TokSize = Lexer::MeasureTokenLength(Range.getEnd(),
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000270 SM, *LangOpts);
Ted Kremenek15322172011-11-10 08:43:12 +0000271
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000272 AddLocToRecord(Range.getEnd(), Record, SM, TokSize);
Ted Kremenek96dcade2011-11-05 03:34:23 +0000273}
274
Ted Kremenek2a764102011-12-17 05:26:11 +0000275unsigned SDiagsWriter::getEmitFile(const char *FileName){
276 if (!FileName)
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000277 return 0;
278
Ted Kremenek2a764102011-12-17 05:26:11 +0000279 unsigned &entry = Files[FileName];
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000280 if (entry)
281 return entry;
282
283 // Lazily generate the record for the file.
284 entry = Files.size();
285 RecordData Record;
286 Record.push_back(RECORD_FILENAME);
287 Record.push_back(entry);
Ted Kremenek2a764102011-12-17 05:26:11 +0000288 Record.push_back(0); // For legacy.
289 Record.push_back(0); // For legacy.
290 StringRef Name(FileName);
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000291 Record.push_back(Name.size());
292 Stream.EmitRecordWithBlob(Abbrevs.get(RECORD_FILENAME), Record, Name);
293
294 return entry;
295}
296
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000297void SDiagsWriter::EmitCharSourceRange(CharSourceRange R,
Ted Kremenek2a764102011-12-17 05:26:11 +0000298 const SourceManager &SM) {
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000299 Record.clear();
300 Record.push_back(RECORD_SOURCE_RANGE);
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000301 AddCharSourceRangeToRecord(R, Record, SM);
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000302 Stream.EmitRecordWithAbbrev(Abbrevs.get(RECORD_SOURCE_RANGE), Record);
303}
304
Ted Kremenek78002122011-10-29 00:12:39 +0000305/// \brief Emits the preamble of the diagnostics file.
306void SDiagsWriter::EmitPreamble() {
Ted Kremenek78002122011-10-29 00:12:39 +0000307 // Emit the file header.
308 Stream.Emit((unsigned)'D', 8);
Ted Kremenek0b69aa82011-11-08 20:27:29 +0000309 Stream.Emit((unsigned)'I', 8);
310 Stream.Emit((unsigned)'A', 8);
311 Stream.Emit((unsigned)'G', 8);
Ted Kremenek069f9c22011-11-05 00:09:53 +0000312
Ted Kremenek78002122011-10-29 00:12:39 +0000313 EmitBlockInfoBlock();
Ted Kremenek0b69aa82011-11-08 20:27:29 +0000314 EmitMetaBlock();
Ted Kremenek78002122011-10-29 00:12:39 +0000315}
316
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000317static void AddSourceLocationAbbrev(llvm::BitCodeAbbrev *Abbrev) {
318 using namespace llvm;
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000319 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 10)); // File ID.
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000320 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // Line.
321 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // Column.
322 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // Offset;
323}
Ted Kremenek96dcade2011-11-05 03:34:23 +0000324
325static void AddRangeLocationAbbrev(llvm::BitCodeAbbrev *Abbrev) {
326 AddSourceLocationAbbrev(Abbrev);
327 AddSourceLocationAbbrev(Abbrev);
328}
329
Ted Kremenek78002122011-10-29 00:12:39 +0000330void SDiagsWriter::EmitBlockInfoBlock() {
331 Stream.EnterBlockInfoBlock(3);
Ted Kremenek0b69aa82011-11-08 20:27:29 +0000332
333 using namespace llvm;
334
335 // ==---------------------------------------------------------------------==//
336 // The subsequent records and Abbrevs are for the "Meta" block.
337 // ==---------------------------------------------------------------------==//
338
339 EmitBlockID(BLOCK_META, "Meta", Stream, Record);
340 EmitRecordID(RECORD_VERSION, "Version", Stream, Record);
341 BitCodeAbbrev *Abbrev = new BitCodeAbbrev();
342 Abbrev->Add(BitCodeAbbrevOp(RECORD_VERSION));
343 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
344 Abbrevs.set(RECORD_VERSION, Stream.EmitBlockInfoAbbrev(BLOCK_META, Abbrev));
345
Ted Kremenek78002122011-10-29 00:12:39 +0000346 // ==---------------------------------------------------------------------==//
347 // The subsequent records and Abbrevs are for the "Diagnostic" block.
348 // ==---------------------------------------------------------------------==//
349
Ted Kremenek45d92752011-11-05 00:09:50 +0000350 EmitBlockID(BLOCK_DIAG, "Diag", Stream, Record);
351 EmitRecordID(RECORD_DIAG, "DiagInfo", Stream, Record);
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000352 EmitRecordID(RECORD_SOURCE_RANGE, "SrcRange", Stream, Record);
Ted Kremenek0dbadc42011-11-05 00:10:04 +0000353 EmitRecordID(RECORD_CATEGORY, "CatName", Stream, Record);
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000354 EmitRecordID(RECORD_DIAG_FLAG, "DiagFlag", Stream, Record);
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000355 EmitRecordID(RECORD_FILENAME, "FileName", Stream, Record);
Ted Kremenek96dcade2011-11-05 03:34:23 +0000356 EmitRecordID(RECORD_FIXIT, "FixIt", Stream, Record);
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000357
Ted Kremenek78002122011-10-29 00:12:39 +0000358 // Emit abbreviation for RECORD_DIAG.
Ted Kremenek0b69aa82011-11-08 20:27:29 +0000359 Abbrev = new BitCodeAbbrev();
Ted Kremenek78002122011-10-29 00:12:39 +0000360 Abbrev->Add(BitCodeAbbrevOp(RECORD_DIAG));
Ted Kremenek45d92752011-11-05 00:09:50 +0000361 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 3)); // Diag level.
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000362 AddSourceLocationAbbrev(Abbrev);
Ted Kremenek45d92752011-11-05 00:09:50 +0000363 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 10)); // Category.
364 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 10)); // Mapped Diag ID.
Ted Kremenek78002122011-10-29 00:12:39 +0000365 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Text size.
366 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // Diagnostc text.
367 Abbrevs.set(RECORD_DIAG, Stream.EmitBlockInfoAbbrev(BLOCK_DIAG, Abbrev));
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000368
Ted Kremenek0dbadc42011-11-05 00:10:04 +0000369 // Emit abbrevation for RECORD_CATEGORY.
370 Abbrev = new BitCodeAbbrev();
371 Abbrev->Add(BitCodeAbbrevOp(RECORD_CATEGORY));
372 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Category ID.
373 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); // Text size.
374 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // Category text.
375 Abbrevs.set(RECORD_CATEGORY, Stream.EmitBlockInfoAbbrev(BLOCK_DIAG, Abbrev));
376
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000377 // Emit abbrevation for RECORD_SOURCE_RANGE.
378 Abbrev = new BitCodeAbbrev();
379 Abbrev->Add(BitCodeAbbrevOp(RECORD_SOURCE_RANGE));
Ted Kremenek96dcade2011-11-05 03:34:23 +0000380 AddRangeLocationAbbrev(Abbrev);
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000381 Abbrevs.set(RECORD_SOURCE_RANGE,
382 Stream.EmitBlockInfoAbbrev(BLOCK_DIAG, Abbrev));
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000383
384 // Emit the abbreviation for RECORD_DIAG_FLAG.
385 Abbrev = new BitCodeAbbrev();
386 Abbrev->Add(BitCodeAbbrevOp(RECORD_DIAG_FLAG));
387 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 10)); // Mapped Diag ID.
388 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Text size.
389 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // Flag name text.
390 Abbrevs.set(RECORD_DIAG_FLAG, Stream.EmitBlockInfoAbbrev(BLOCK_DIAG,
391 Abbrev));
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000392
393 // Emit the abbreviation for RECORD_FILENAME.
Ted Kremenek78002122011-10-29 00:12:39 +0000394 Abbrev = new BitCodeAbbrev();
Ted Kremenek28eac522011-11-05 00:09:43 +0000395 Abbrev->Add(BitCodeAbbrevOp(RECORD_FILENAME));
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000396 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 10)); // Mapped file ID.
397 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // Size.
398 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); // Modifcation time.
Ted Kremenek78002122011-10-29 00:12:39 +0000399 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Text size.
400 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // File name text.
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000401 Abbrevs.set(RECORD_FILENAME, Stream.EmitBlockInfoAbbrev(BLOCK_DIAG,
Ted Kremenek78002122011-10-29 00:12:39 +0000402 Abbrev));
Ted Kremenek96dcade2011-11-05 03:34:23 +0000403
404 // Emit the abbreviation for RECORD_FIXIT.
405 Abbrev = new BitCodeAbbrev();
406 Abbrev->Add(BitCodeAbbrevOp(RECORD_FIXIT));
407 AddRangeLocationAbbrev(Abbrev);
408 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 16)); // Text size.
409 Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); // FixIt text.
410 Abbrevs.set(RECORD_FIXIT, Stream.EmitBlockInfoAbbrev(BLOCK_DIAG,
411 Abbrev));
Ted Kremenek78002122011-10-29 00:12:39 +0000412
413 Stream.ExitBlock();
414}
415
Ted Kremenek0b69aa82011-11-08 20:27:29 +0000416void SDiagsWriter::EmitMetaBlock() {
417 Stream.EnterSubblock(BLOCK_META, 3);
418 Record.clear();
419 Record.push_back(RECORD_VERSION);
420 Record.push_back(Version);
421 Stream.EmitRecordWithAbbrev(Abbrevs.get(RECORD_VERSION), Record);
422 Stream.ExitBlock();
423}
424
Ted Kremenek2a764102011-12-17 05:26:11 +0000425unsigned SDiagsWriter::getEmitCategory(unsigned int category) {
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000426 if (Categories.count(category))
427 return category;
428
429 Categories.insert(category);
Ted Kremenek0dbadc42011-11-05 00:10:04 +0000430
431 // We use a local version of 'Record' so that we can be generating
432 // another record when we lazily generate one for the category entry.
433 RecordData Record;
434 Record.push_back(RECORD_CATEGORY);
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000435 Record.push_back(category);
436 StringRef catName = DiagnosticIDs::getCategoryNameFromID(category);
Ted Kremenek0dbadc42011-11-05 00:10:04 +0000437 Record.push_back(catName.size());
438 Stream.EmitRecordWithBlob(Abbrevs.get(RECORD_CATEGORY), Record, catName);
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000439
440 return category;
441}
442
443unsigned SDiagsWriter::getEmitDiagnosticFlag(DiagnosticsEngine::Level DiagLevel,
Ted Kremenek2a764102011-12-17 05:26:11 +0000444 unsigned DiagID) {
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000445 if (DiagLevel == DiagnosticsEngine::Note)
446 return 0; // No flag for notes.
447
Ted Kremenek2a764102011-12-17 05:26:11 +0000448 StringRef FlagName = DiagnosticIDs::getWarningOptionForDiag(DiagID);
Ted Kremenek3baf63d2011-11-05 00:10:07 +0000449 if (FlagName.empty())
450 return 0;
451
452 // Here we assume that FlagName points to static data whose pointer
453 // value is fixed. This allows us to unique by diagnostic groups.
454 const void *data = FlagName.data();
455 std::pair<unsigned, StringRef> &entry = DiagFlags[data];
456 if (entry.first == 0) {
457 entry.first = DiagFlags.size();
458 entry.second = FlagName;
459
460 // Lazily emit the string in a separate record.
461 RecordData Record;
462 Record.push_back(RECORD_DIAG_FLAG);
463 Record.push_back(entry.first);
464 Record.push_back(FlagName.size());
465 Stream.EmitRecordWithBlob(Abbrevs.get(RECORD_DIAG_FLAG),
466 Record, FlagName);
467 }
468
469 return entry.first;
Ted Kremenek0dbadc42011-11-05 00:10:04 +0000470}
471
Ted Kremenek78002122011-10-29 00:12:39 +0000472void SDiagsWriter::HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
473 const Diagnostic &Info) {
Ted Kremenek59b61612011-11-05 00:09:47 +0000474 if (DiagLevel != DiagnosticsEngine::Note) {
475 if (inNonNoteDiagnostic) {
476 // We have encountered a non-note diagnostic. Finish up the previous
477 // diagnostic block before starting a new one.
478 Stream.ExitBlock();
479 }
480 inNonNoteDiagnostic = true;
481 }
Ted Kremenek96dcade2011-11-05 03:34:23 +0000482
Ted Kremenek2a764102011-12-17 05:26:11 +0000483 // Compute the diagnostic text.
484 diagBuf.clear();
485 Info.FormatDiagnostic(diagBuf);
486
487 SourceManager &SM = Info.getSourceManager();
488 SDiagsRenderer Renderer(*this, Record, SM, *LangOpts, DiagOpts);
489 Renderer.emitDiagnostic(Info.getLocation(), DiagLevel,
490 diagBuf.str(),
491 Info.getRanges(),
492 llvm::makeArrayRef(Info.getFixItHints(),
493 Info.getNumFixItHints()),
494 &Info);
495}
496
497void
498SDiagsRenderer::emitDiagnosticMessage(SourceLocation Loc,
499 PresumedLoc PLoc,
500 DiagnosticsEngine::Level Level,
501 StringRef Message,
502 ArrayRef<clang::CharSourceRange> Ranges,
Ted Kremenek8be51ea2012-02-14 02:46:00 +0000503 DiagOrStoredDiag D) {
Ted Kremenek2a764102011-12-17 05:26:11 +0000504 // Emit the RECORD_DIAG record.
505 Writer.Record.clear();
506 Writer.Record.push_back(RECORD_DIAG);
507 Writer.Record.push_back(Level);
508 Writer.AddLocToRecord(Loc, SM, PLoc, Record);
509
Ted Kremenek8be51ea2012-02-14 02:46:00 +0000510 if (const Diagnostic *Info = D.dyn_cast<const Diagnostic*>()) {
Ted Kremenek2a764102011-12-17 05:26:11 +0000511 // Emit the category string lazily and get the category ID.
512 unsigned DiagID = DiagnosticIDs::getCategoryNumberForDiag(Info->getID());
513 Writer.Record.push_back(Writer.getEmitCategory(DiagID));
514 // Emit the diagnostic flag string lazily and get the mapped ID.
515 Writer.Record.push_back(Writer.getEmitDiagnosticFlag(Level, Info->getID()));
516 }
517 else {
518 Writer.Record.push_back(Writer.getEmitCategory());
519 Writer.Record.push_back(Writer.getEmitDiagnosticFlag(Level));
520 }
521
522 Writer.Record.push_back(Message.size());
523 Writer.Stream.EmitRecordWithBlob(Writer.Abbrevs.get(RECORD_DIAG),
524 Writer.Record, Message);
525}
526
Ted Kremenek8be51ea2012-02-14 02:46:00 +0000527void SDiagsRenderer::beginDiagnostic(DiagOrStoredDiag D,
Ted Kremenek2a764102011-12-17 05:26:11 +0000528 DiagnosticsEngine::Level Level) {
529 Writer.Stream.EnterSubblock(BLOCK_DIAG, 4);
530}
531
Ted Kremenek8be51ea2012-02-14 02:46:00 +0000532void SDiagsRenderer::endDiagnostic(DiagOrStoredDiag D,
Ted Kremenek2a764102011-12-17 05:26:11 +0000533 DiagnosticsEngine::Level Level) {
Ted Kremenek8be51ea2012-02-14 02:46:00 +0000534 if (D && Level != DiagnosticsEngine::Note)
Ted Kremenek2a764102011-12-17 05:26:11 +0000535 return;
536 Writer.Stream.ExitBlock();
537}
538
539void SDiagsRenderer::emitCodeContext(SourceLocation Loc,
540 DiagnosticsEngine::Level Level,
541 SmallVectorImpl<CharSourceRange> &Ranges,
542 ArrayRef<FixItHint> Hints) {
Ted Kremenek96dcade2011-11-05 03:34:23 +0000543 // Emit Source Ranges.
Ted Kremenek2a20b4f2011-11-05 00:10:01 +0000544 for (ArrayRef<CharSourceRange>::iterator it=Ranges.begin(), ei=Ranges.end();
545 it != ei; ++it) {
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000546 if (it->isValid())
Ted Kremenek2a764102011-12-17 05:26:11 +0000547 Writer.EmitCharSourceRange(*it, SM);
Ted Kremenek96dcade2011-11-05 03:34:23 +0000548 }
Ted Kremenek59b61612011-11-05 00:09:47 +0000549
Ted Kremenek2a764102011-12-17 05:26:11 +0000550 // Emit FixIts.
551 for (ArrayRef<FixItHint>::iterator it = Hints.begin(), et = Hints.end();
552 it != et; ++it) {
553 const FixItHint &fix = *it;
554 if (fix.isNull())
555 continue;
556 Writer.Record.clear();
557 Writer.Record.push_back(RECORD_FIXIT);
558 Writer.AddCharSourceRangeToRecord(fix.RemoveRange, Record, SM);
559 Writer.Record.push_back(fix.CodeToInsert.size());
560 Writer.Stream.EmitRecordWithBlob(Writer.Abbrevs.get(RECORD_FIXIT), Record,
561 fix.CodeToInsert);
Ted Kremenek59b61612011-11-05 00:09:47 +0000562 }
Ted Kremenek78002122011-10-29 00:12:39 +0000563}
564
Ted Kremenek2a764102011-12-17 05:26:11 +0000565void SDiagsRenderer::emitNote(SourceLocation Loc, StringRef Message) {
566 Writer.Stream.EnterSubblock(BLOCK_DIAG, 4);
567 RecordData Record;
568 Record.push_back(RECORD_DIAG);
569 Record.push_back(DiagnosticsEngine::Note);
570 Writer.AddLocToRecord(Loc, Record, SM);
571 Record.push_back(Writer.getEmitCategory());
572 Record.push_back(Writer.getEmitDiagnosticFlag(DiagnosticsEngine::Note));
573 Record.push_back(Message.size());
574 Writer.Stream.EmitRecordWithBlob(Writer.Abbrevs.get(RECORD_DIAG),
575 Record, Message);
576 Writer.Stream.ExitBlock();
577}
578
Argyrios Kyrtzidis29f27872011-12-07 05:52:12 +0000579void SDiagsWriter::finish() {
Ted Kremenek59b61612011-11-05 00:09:47 +0000580 if (inNonNoteDiagnostic) {
581 // Finish off any diagnostics we were in the process of emitting.
582 Stream.ExitBlock();
583 inNonNoteDiagnostic = false;
584 }
Ted Kremenek0d34e6e2011-11-05 00:10:11 +0000585
Ted Kremenek78002122011-10-29 00:12:39 +0000586 // Write the generated bitstream to "Out".
587 OS->write((char *)&Buffer.front(), Buffer.size());
588 OS->flush();
589
590 OS.reset(0);
591}
592