1 //===- Tokens.cpp - collect tokens from preprocessing ---------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 #include "clang/Tooling/Syntax/Tokens.h" 9 10 #include "clang/Basic/Diagnostic.h" 11 #include "clang/Basic/IdentifierTable.h" 12 #include "clang/Basic/LLVM.h" 13 #include "clang/Basic/LangOptions.h" 14 #include "clang/Basic/SourceLocation.h" 15 #include "clang/Basic/SourceManager.h" 16 #include "clang/Basic/TokenKinds.h" 17 #include "clang/Lex/PPCallbacks.h" 18 #include "clang/Lex/Preprocessor.h" 19 #include "clang/Lex/Token.h" 20 #include "llvm/ADT/ArrayRef.h" 21 #include "llvm/ADT/None.h" 22 #include "llvm/ADT/Optional.h" 23 #include "llvm/ADT/STLExtras.h" 24 #include "llvm/Support/Debug.h" 25 #include "llvm/Support/ErrorHandling.h" 26 #include "llvm/Support/FormatVariadic.h" 27 #include "llvm/Support/raw_ostream.h" 28 #include <algorithm> 29 #include <cassert> 30 #include <iterator> 31 #include <string> 32 #include <utility> 33 #include <vector> 34 35 using namespace clang; 36 using namespace clang::syntax; 37 38 syntax::Token::Token(SourceLocation Location, unsigned Length, 39 tok::TokenKind Kind) 40 : Location(Location), Length(Length), Kind(Kind) { 41 assert(Location.isValid()); 42 } 43 44 syntax::Token::Token(const clang::Token &T) 45 : Token(T.getLocation(), T.getLength(), T.getKind()) { 46 assert(!T.isAnnotation()); 47 } 48 49 llvm::StringRef syntax::Token::text(const SourceManager &SM) const { 50 bool Invalid = false; 51 const char *Start = SM.getCharacterData(location(), &Invalid); 52 assert(!Invalid); 53 return llvm::StringRef(Start, length()); 54 } 55 56 FileRange syntax::Token::range(const SourceManager &SM) const { 57 assert(location().isFileID() && "must be a spelled token"); 58 FileID File; 59 unsigned StartOffset; 60 std::tie(File, StartOffset) = SM.getDecomposedLoc(location()); 61 return FileRange(File, StartOffset, StartOffset + length()); 62 } 63 64 FileRange syntax::Token::range(const SourceManager &SM, 65 const syntax::Token &First, 66 const syntax::Token &Last) { 67 auto F = First.range(SM); 68 auto L = Last.range(SM); 69 assert(F.file() == L.file() && "tokens from different files"); 70 assert(F.endOffset() <= L.beginOffset() && "wrong order of tokens"); 71 return FileRange(F.file(), F.beginOffset(), L.endOffset()); 72 } 73 74 llvm::raw_ostream &syntax::operator<<(llvm::raw_ostream &OS, const Token &T) { 75 return OS << T.str(); 76 } 77 78 FileRange::FileRange(FileID File, unsigned BeginOffset, unsigned EndOffset) 79 : File(File), Begin(BeginOffset), End(EndOffset) { 80 assert(File.isValid()); 81 assert(BeginOffset <= EndOffset); 82 } 83 84 FileRange::FileRange(const SourceManager &SM, SourceLocation BeginLoc, 85 unsigned Length) { 86 assert(BeginLoc.isValid()); 87 assert(BeginLoc.isFileID()); 88 89 std::tie(File, Begin) = SM.getDecomposedLoc(BeginLoc); 90 End = Begin + Length; 91 } 92 FileRange::FileRange(const SourceManager &SM, SourceLocation BeginLoc, 93 SourceLocation EndLoc) { 94 assert(BeginLoc.isValid()); 95 assert(BeginLoc.isFileID()); 96 assert(EndLoc.isValid()); 97 assert(EndLoc.isFileID()); 98 assert(SM.getFileID(BeginLoc) == SM.getFileID(EndLoc)); 99 assert(SM.getFileOffset(BeginLoc) <= SM.getFileOffset(EndLoc)); 100 101 std::tie(File, Begin) = SM.getDecomposedLoc(BeginLoc); 102 End = SM.getFileOffset(EndLoc); 103 } 104 105 llvm::raw_ostream &syntax::operator<<(llvm::raw_ostream &OS, 106 const FileRange &R) { 107 return OS << llvm::formatv("FileRange(file = {0}, offsets = {1}-{2})", 108 R.file().getHashValue(), R.beginOffset(), 109 R.endOffset()); 110 } 111 112 llvm::StringRef FileRange::text(const SourceManager &SM) const { 113 bool Invalid = false; 114 StringRef Text = SM.getBufferData(File, &Invalid); 115 if (Invalid) 116 return ""; 117 assert(Begin <= Text.size()); 118 assert(End <= Text.size()); 119 return Text.substr(Begin, length()); 120 } 121 122 llvm::ArrayRef<syntax::Token> TokenBuffer::expandedTokens(SourceRange R) const { 123 if (R.isInvalid()) 124 return {}; 125 const Token *Begin = 126 llvm::partition_point(expandedTokens(), [&](const syntax::Token &T) { 127 return SourceMgr->isBeforeInTranslationUnit(T.location(), R.getBegin()); 128 }); 129 const Token *End = 130 llvm::partition_point(expandedTokens(), [&](const syntax::Token &T) { 131 return !SourceMgr->isBeforeInTranslationUnit(R.getEnd(), T.location()); 132 }); 133 if (Begin > End) 134 return {}; 135 return {Begin, End}; 136 } 137 138 std::pair<const syntax::Token *, const TokenBuffer::Mapping *> 139 TokenBuffer::spelledForExpandedToken(const syntax::Token *Expanded) const { 140 assert(Expanded); 141 assert(ExpandedTokens.data() <= Expanded && 142 Expanded < ExpandedTokens.data() + ExpandedTokens.size()); 143 144 auto FileIt = Files.find( 145 SourceMgr->getFileID(SourceMgr->getExpansionLoc(Expanded->location()))); 146 assert(FileIt != Files.end() && "no file for an expanded token"); 147 148 const MarkedFile &File = FileIt->second; 149 150 unsigned ExpandedIndex = Expanded - ExpandedTokens.data(); 151 // Find the first mapping that produced tokens after \p Expanded. 152 auto It = llvm::partition_point(File.Mappings, [&](const Mapping &M) { 153 return M.BeginExpanded <= ExpandedIndex; 154 }); 155 // Our token could only be produced by the previous mapping. 156 if (It == File.Mappings.begin()) { 157 // No previous mapping, no need to modify offsets. 158 return {&File.SpelledTokens[ExpandedIndex - File.BeginExpanded], nullptr}; 159 } 160 --It; // 'It' now points to last mapping that started before our token. 161 162 // Check if the token is part of the mapping. 163 if (ExpandedIndex < It->EndExpanded) 164 return {&File.SpelledTokens[It->BeginSpelled], /*Mapping*/ &*It}; 165 166 // Not part of the mapping, use the index from previous mapping to compute the 167 // corresponding spelled token. 168 return { 169 &File.SpelledTokens[It->EndSpelled + (ExpandedIndex - It->EndExpanded)], 170 /*Mapping*/ nullptr}; 171 } 172 173 llvm::ArrayRef<syntax::Token> TokenBuffer::spelledTokens(FileID FID) const { 174 auto It = Files.find(FID); 175 assert(It != Files.end()); 176 return It->second.SpelledTokens; 177 } 178 179 std::string TokenBuffer::Mapping::str() const { 180 return llvm::formatv("spelled tokens: [{0},{1}), expanded tokens: [{2},{3})", 181 BeginSpelled, EndSpelled, BeginExpanded, EndExpanded); 182 } 183 184 llvm::Optional<llvm::ArrayRef<syntax::Token>> 185 TokenBuffer::spelledForExpanded(llvm::ArrayRef<syntax::Token> Expanded) const { 186 // Mapping an empty range is ambiguous in case of empty mappings at either end 187 // of the range, bail out in that case. 188 if (Expanded.empty()) 189 return llvm::None; 190 191 // FIXME: also allow changes uniquely mapping to macro arguments. 192 193 const syntax::Token *BeginSpelled; 194 const Mapping *BeginMapping; 195 std::tie(BeginSpelled, BeginMapping) = 196 spelledForExpandedToken(&Expanded.front()); 197 198 const syntax::Token *LastSpelled; 199 const Mapping *LastMapping; 200 std::tie(LastSpelled, LastMapping) = 201 spelledForExpandedToken(&Expanded.back()); 202 203 FileID FID = SourceMgr->getFileID(BeginSpelled->location()); 204 // FIXME: Handle multi-file changes by trying to map onto a common root. 205 if (FID != SourceMgr->getFileID(LastSpelled->location())) 206 return llvm::None; 207 208 const MarkedFile &File = Files.find(FID)->second; 209 210 // Do not allow changes that cross macro expansion boundaries. 211 unsigned BeginExpanded = Expanded.begin() - ExpandedTokens.data(); 212 unsigned EndExpanded = Expanded.end() - ExpandedTokens.data(); 213 if (BeginMapping && BeginMapping->BeginExpanded < BeginExpanded) 214 return llvm::None; 215 if (LastMapping && EndExpanded < LastMapping->EndExpanded) 216 return llvm::None; 217 // All is good, return the result. 218 return llvm::makeArrayRef( 219 BeginMapping ? File.SpelledTokens.data() + BeginMapping->BeginSpelled 220 : BeginSpelled, 221 LastMapping ? File.SpelledTokens.data() + LastMapping->EndSpelled 222 : LastSpelled + 1); 223 } 224 225 llvm::Optional<TokenBuffer::Expansion> 226 TokenBuffer::expansionStartingAt(const syntax::Token *Spelled) const { 227 assert(Spelled); 228 assert(Spelled->location().isFileID() && "not a spelled token"); 229 auto FileIt = Files.find(SourceMgr->getFileID(Spelled->location())); 230 assert(FileIt != Files.end() && "file not tracked by token buffer"); 231 232 auto &File = FileIt->second; 233 assert(File.SpelledTokens.data() <= Spelled && 234 Spelled < (File.SpelledTokens.data() + File.SpelledTokens.size())); 235 236 unsigned SpelledIndex = Spelled - File.SpelledTokens.data(); 237 auto M = llvm::partition_point(File.Mappings, [&](const Mapping &M) { 238 return M.BeginSpelled < SpelledIndex; 239 }); 240 if (M == File.Mappings.end() || M->BeginSpelled != SpelledIndex) 241 return llvm::None; 242 243 Expansion E; 244 E.Spelled = llvm::makeArrayRef(File.SpelledTokens.data() + M->BeginSpelled, 245 File.SpelledTokens.data() + M->EndSpelled); 246 E.Expanded = llvm::makeArrayRef(ExpandedTokens.data() + M->BeginExpanded, 247 ExpandedTokens.data() + M->EndExpanded); 248 return E; 249 } 250 251 std::vector<const syntax::Token *> 252 TokenBuffer::macroExpansions(FileID FID) const { 253 auto FileIt = Files.find(FID); 254 assert(FileIt != Files.end() && "file not tracked by token buffer"); 255 auto &File = FileIt->second; 256 std::vector<const syntax::Token *> Expansions; 257 auto &Spelled = File.SpelledTokens; 258 for (auto Mapping : File.Mappings) { 259 const syntax::Token *Token = &Spelled[Mapping.BeginSpelled]; 260 if (Token->kind() == tok::TokenKind::identifier) 261 Expansions.push_back(Token); 262 } 263 return Expansions; 264 } 265 266 std::vector<syntax::Token> syntax::tokenize(FileID FID, const SourceManager &SM, 267 const LangOptions &LO) { 268 std::vector<syntax::Token> Tokens; 269 IdentifierTable Identifiers(LO); 270 auto AddToken = [&](clang::Token T) { 271 // Fill the proper token kind for keywords, etc. 272 if (T.getKind() == tok::raw_identifier && !T.needsCleaning() && 273 !T.hasUCN()) { // FIXME: support needsCleaning and hasUCN cases. 274 clang::IdentifierInfo &II = Identifiers.get(T.getRawIdentifier()); 275 T.setIdentifierInfo(&II); 276 T.setKind(II.getTokenID()); 277 } 278 Tokens.push_back(syntax::Token(T)); 279 }; 280 281 Lexer L(FID, SM.getBuffer(FID), SM, LO); 282 283 clang::Token T; 284 while (!L.LexFromRawLexer(T)) 285 AddToken(T); 286 // 'eof' is only the last token if the input is null-terminated. Never store 287 // it, for consistency. 288 if (T.getKind() != tok::eof) 289 AddToken(T); 290 return Tokens; 291 } 292 293 /// Records information reqired to construct mappings for the token buffer that 294 /// we are collecting. 295 class TokenCollector::CollectPPExpansions : public PPCallbacks { 296 public: 297 CollectPPExpansions(TokenCollector &C) : Collector(&C) {} 298 299 /// Disabled instance will stop reporting anything to TokenCollector. 300 /// This ensures that uses of the preprocessor after TokenCollector::consume() 301 /// is called do not access the (possibly invalid) collector instance. 302 void disable() { Collector = nullptr; } 303 304 void MacroExpands(const clang::Token &MacroNameTok, const MacroDefinition &MD, 305 SourceRange Range, const MacroArgs *Args) override { 306 if (!Collector) 307 return; 308 // Only record top-level expansions, not those where: 309 // - the macro use is inside a macro body, 310 // - the macro appears in an argument to another macro. 311 if (!MacroNameTok.getLocation().isFileID() || 312 (LastExpansionEnd.isValid() && 313 Collector->PP.getSourceManager().isBeforeInTranslationUnit( 314 Range.getBegin(), LastExpansionEnd))) 315 return; 316 Collector->Expansions[Range.getBegin().getRawEncoding()] = Range.getEnd(); 317 LastExpansionEnd = Range.getEnd(); 318 } 319 // FIXME: handle directives like #pragma, #include, etc. 320 private: 321 TokenCollector *Collector; 322 /// Used to detect recursive macro expansions. 323 SourceLocation LastExpansionEnd; 324 }; 325 326 /// Fills in the TokenBuffer by tracing the run of a preprocessor. The 327 /// implementation tracks the tokens, macro expansions and directives coming 328 /// from the preprocessor and: 329 /// - for each token, figures out if it is a part of an expanded token stream, 330 /// spelled token stream or both. Stores the tokens appropriately. 331 /// - records mappings from the spelled to expanded token ranges, e.g. for macro 332 /// expansions. 333 /// FIXME: also properly record: 334 /// - #include directives, 335 /// - #pragma, #line and other PP directives, 336 /// - skipped pp regions, 337 /// - ... 338 339 TokenCollector::TokenCollector(Preprocessor &PP) : PP(PP) { 340 // Collect the expanded token stream during preprocessing. 341 PP.setTokenWatcher([this](const clang::Token &T) { 342 if (T.isAnnotation()) 343 return; 344 DEBUG_WITH_TYPE("collect-tokens", llvm::dbgs() 345 << "Token: " 346 << syntax::Token(T).dumpForTests( 347 this->PP.getSourceManager()) 348 << "\n" 349 350 ); 351 Expanded.push_back(syntax::Token(T)); 352 }); 353 // And locations of macro calls, to properly recover boundaries of those in 354 // case of empty expansions. 355 auto CB = std::make_unique<CollectPPExpansions>(*this); 356 this->Collector = CB.get(); 357 PP.addPPCallbacks(std::move(CB)); 358 } 359 360 /// Builds mappings and spelled tokens in the TokenBuffer based on the expanded 361 /// token stream. 362 class TokenCollector::Builder { 363 public: 364 Builder(std::vector<syntax::Token> Expanded, PPExpansions CollectedExpansions, 365 const SourceManager &SM, const LangOptions &LangOpts) 366 : Result(SM), CollectedExpansions(std::move(CollectedExpansions)), SM(SM), 367 LangOpts(LangOpts) { 368 Result.ExpandedTokens = std::move(Expanded); 369 } 370 371 TokenBuffer build() && { 372 buildSpelledTokens(); 373 374 // Walk over expanded tokens and spelled tokens in parallel, building the 375 // mappings between those using source locations. 376 // To correctly recover empty macro expansions, we also take locations 377 // reported to PPCallbacks::MacroExpands into account as we do not have any 378 // expanded tokens with source locations to guide us. 379 380 // The 'eof' token is special, it is not part of spelled token stream. We 381 // handle it separately at the end. 382 assert(!Result.ExpandedTokens.empty()); 383 assert(Result.ExpandedTokens.back().kind() == tok::eof); 384 for (unsigned I = 0; I < Result.ExpandedTokens.size() - 1; ++I) { 385 // (!) I might be updated by the following call. 386 processExpandedToken(I); 387 } 388 389 // 'eof' not handled in the loop, do it here. 390 assert(SM.getMainFileID() == 391 SM.getFileID(Result.ExpandedTokens.back().location())); 392 fillGapUntil(Result.Files[SM.getMainFileID()], 393 Result.ExpandedTokens.back().location(), 394 Result.ExpandedTokens.size() - 1); 395 Result.Files[SM.getMainFileID()].EndExpanded = Result.ExpandedTokens.size(); 396 397 // Some files might have unaccounted spelled tokens at the end, add an empty 398 // mapping for those as they did not have expanded counterparts. 399 fillGapsAtEndOfFiles(); 400 401 return std::move(Result); 402 } 403 404 private: 405 /// Process the next token in an expanded stream and move corresponding 406 /// spelled tokens, record any mapping if needed. 407 /// (!) \p I will be updated if this had to skip tokens, e.g. for macros. 408 void processExpandedToken(unsigned &I) { 409 auto L = Result.ExpandedTokens[I].location(); 410 if (L.isMacroID()) { 411 processMacroExpansion(SM.getExpansionRange(L), I); 412 return; 413 } 414 if (L.isFileID()) { 415 auto FID = SM.getFileID(L); 416 TokenBuffer::MarkedFile &File = Result.Files[FID]; 417 418 fillGapUntil(File, L, I); 419 420 // Skip the token. 421 assert(File.SpelledTokens[NextSpelled[FID]].location() == L && 422 "no corresponding token in the spelled stream"); 423 ++NextSpelled[FID]; 424 return; 425 } 426 } 427 428 /// Skipped expanded and spelled tokens of a macro expansion that covers \p 429 /// SpelledRange. Add a corresponding mapping. 430 /// (!) \p I will be the index of the last token in an expansion after this 431 /// function returns. 432 void processMacroExpansion(CharSourceRange SpelledRange, unsigned &I) { 433 auto FID = SM.getFileID(SpelledRange.getBegin()); 434 assert(FID == SM.getFileID(SpelledRange.getEnd())); 435 TokenBuffer::MarkedFile &File = Result.Files[FID]; 436 437 fillGapUntil(File, SpelledRange.getBegin(), I); 438 439 // Skip all expanded tokens from the same macro expansion. 440 unsigned BeginExpanded = I; 441 for (; I + 1 < Result.ExpandedTokens.size(); ++I) { 442 auto NextL = Result.ExpandedTokens[I + 1].location(); 443 if (!NextL.isMacroID() || 444 SM.getExpansionLoc(NextL) != SpelledRange.getBegin()) 445 break; 446 } 447 unsigned EndExpanded = I + 1; 448 consumeMapping(File, SM.getFileOffset(SpelledRange.getEnd()), BeginExpanded, 449 EndExpanded, NextSpelled[FID]); 450 } 451 452 /// Initializes TokenBuffer::Files and fills spelled tokens and expanded 453 /// ranges for each of the files. 454 void buildSpelledTokens() { 455 for (unsigned I = 0; I < Result.ExpandedTokens.size(); ++I) { 456 auto FID = 457 SM.getFileID(SM.getExpansionLoc(Result.ExpandedTokens[I].location())); 458 auto It = Result.Files.try_emplace(FID); 459 TokenBuffer::MarkedFile &File = It.first->second; 460 461 File.EndExpanded = I + 1; 462 if (!It.second) 463 continue; // we have seen this file before. 464 465 // This is the first time we see this file. 466 File.BeginExpanded = I; 467 File.SpelledTokens = tokenize(FID, SM, LangOpts); 468 } 469 } 470 471 void consumeEmptyMapping(TokenBuffer::MarkedFile &File, unsigned EndOffset, 472 unsigned ExpandedIndex, unsigned &SpelledIndex) { 473 consumeMapping(File, EndOffset, ExpandedIndex, ExpandedIndex, SpelledIndex); 474 } 475 476 /// Consumes spelled tokens that form a macro expansion and adds a entry to 477 /// the resulting token buffer. 478 /// (!) SpelledIndex is updated in-place. 479 void consumeMapping(TokenBuffer::MarkedFile &File, unsigned EndOffset, 480 unsigned BeginExpanded, unsigned EndExpanded, 481 unsigned &SpelledIndex) { 482 // We need to record this mapping before continuing. 483 unsigned MappingBegin = SpelledIndex; 484 ++SpelledIndex; 485 486 bool HitMapping = 487 tryConsumeSpelledUntil(File, EndOffset + 1, SpelledIndex).hasValue(); 488 (void)HitMapping; 489 assert(!HitMapping && "recursive macro expansion?"); 490 491 TokenBuffer::Mapping M; 492 M.BeginExpanded = BeginExpanded; 493 M.EndExpanded = EndExpanded; 494 M.BeginSpelled = MappingBegin; 495 M.EndSpelled = SpelledIndex; 496 497 File.Mappings.push_back(M); 498 } 499 500 /// Consumes spelled tokens until location \p L is reached and adds a mapping 501 /// covering the consumed tokens. The mapping will point to an empty expanded 502 /// range at position \p ExpandedIndex. 503 void fillGapUntil(TokenBuffer::MarkedFile &File, SourceLocation L, 504 unsigned ExpandedIndex) { 505 assert(L.isFileID()); 506 FileID FID; 507 unsigned Offset; 508 std::tie(FID, Offset) = SM.getDecomposedLoc(L); 509 510 unsigned &SpelledIndex = NextSpelled[FID]; 511 unsigned MappingBegin = SpelledIndex; 512 while (true) { 513 auto EndLoc = tryConsumeSpelledUntil(File, Offset, SpelledIndex); 514 if (SpelledIndex != MappingBegin) { 515 TokenBuffer::Mapping M; 516 M.BeginSpelled = MappingBegin; 517 M.EndSpelled = SpelledIndex; 518 M.BeginExpanded = M.EndExpanded = ExpandedIndex; 519 File.Mappings.push_back(M); 520 } 521 if (!EndLoc) 522 break; 523 consumeEmptyMapping(File, SM.getFileOffset(*EndLoc), ExpandedIndex, 524 SpelledIndex); 525 526 MappingBegin = SpelledIndex; 527 } 528 }; 529 530 /// Consumes spelled tokens until it reaches Offset or a mapping boundary, 531 /// i.e. a name of a macro expansion or the start '#' token of a PP directive. 532 /// (!) NextSpelled is updated in place. 533 /// 534 /// returns None if \p Offset was reached, otherwise returns the end location 535 /// of a mapping that starts at \p NextSpelled. 536 llvm::Optional<SourceLocation> 537 tryConsumeSpelledUntil(TokenBuffer::MarkedFile &File, unsigned Offset, 538 unsigned &NextSpelled) { 539 for (; NextSpelled < File.SpelledTokens.size(); ++NextSpelled) { 540 auto L = File.SpelledTokens[NextSpelled].location(); 541 if (Offset <= SM.getFileOffset(L)) 542 return llvm::None; // reached the offset we are looking for. 543 auto Mapping = CollectedExpansions.find(L.getRawEncoding()); 544 if (Mapping != CollectedExpansions.end()) 545 return Mapping->second; // found a mapping before the offset. 546 } 547 return llvm::None; // no more tokens, we "reached" the offset. 548 } 549 550 /// Adds empty mappings for unconsumed spelled tokens at the end of each file. 551 void fillGapsAtEndOfFiles() { 552 for (auto &F : Result.Files) { 553 if (F.second.SpelledTokens.empty()) 554 continue; 555 fillGapUntil(F.second, F.second.SpelledTokens.back().endLocation(), 556 F.second.EndExpanded); 557 } 558 } 559 560 TokenBuffer Result; 561 /// For each file, a position of the next spelled token we will consume. 562 llvm::DenseMap<FileID, unsigned> NextSpelled; 563 PPExpansions CollectedExpansions; 564 const SourceManager &SM; 565 const LangOptions &LangOpts; 566 }; 567 568 TokenBuffer TokenCollector::consume() && { 569 PP.setTokenWatcher(nullptr); 570 Collector->disable(); 571 return Builder(std::move(Expanded), std::move(Expansions), 572 PP.getSourceManager(), PP.getLangOpts()) 573 .build(); 574 } 575 576 std::string syntax::Token::str() const { 577 return llvm::formatv("Token({0}, length = {1})", tok::getTokenName(kind()), 578 length()); 579 } 580 581 std::string syntax::Token::dumpForTests(const SourceManager &SM) const { 582 return llvm::formatv("{0} {1}", tok::getTokenName(kind()), text(SM)); 583 } 584 585 std::string TokenBuffer::dumpForTests() const { 586 auto PrintToken = [this](const syntax::Token &T) -> std::string { 587 if (T.kind() == tok::eof) 588 return "<eof>"; 589 return T.text(*SourceMgr); 590 }; 591 592 auto DumpTokens = [this, &PrintToken](llvm::raw_ostream &OS, 593 llvm::ArrayRef<syntax::Token> Tokens) { 594 if (Tokens.empty()) { 595 OS << "<empty>"; 596 return; 597 } 598 OS << Tokens[0].text(*SourceMgr); 599 for (unsigned I = 1; I < Tokens.size(); ++I) { 600 if (Tokens[I].kind() == tok::eof) 601 continue; 602 OS << " " << PrintToken(Tokens[I]); 603 } 604 }; 605 606 std::string Dump; 607 llvm::raw_string_ostream OS(Dump); 608 609 OS << "expanded tokens:\n" 610 << " "; 611 // (!) we do not show '<eof>'. 612 DumpTokens(OS, llvm::makeArrayRef(ExpandedTokens).drop_back()); 613 OS << "\n"; 614 615 std::vector<FileID> Keys; 616 for (auto F : Files) 617 Keys.push_back(F.first); 618 llvm::sort(Keys); 619 620 for (FileID ID : Keys) { 621 const MarkedFile &File = Files.find(ID)->second; 622 auto *Entry = SourceMgr->getFileEntryForID(ID); 623 if (!Entry) 624 continue; // Skip builtin files. 625 OS << llvm::formatv("file '{0}'\n", Entry->getName()) 626 << " spelled tokens:\n" 627 << " "; 628 DumpTokens(OS, File.SpelledTokens); 629 OS << "\n"; 630 631 if (File.Mappings.empty()) { 632 OS << " no mappings.\n"; 633 continue; 634 } 635 OS << " mappings:\n"; 636 for (auto &M : File.Mappings) { 637 OS << llvm::formatv( 638 " ['{0}'_{1}, '{2}'_{3}) => ['{4}'_{5}, '{6}'_{7})\n", 639 PrintToken(File.SpelledTokens[M.BeginSpelled]), M.BeginSpelled, 640 M.EndSpelled == File.SpelledTokens.size() 641 ? "<eof>" 642 : PrintToken(File.SpelledTokens[M.EndSpelled]), 643 M.EndSpelled, PrintToken(ExpandedTokens[M.BeginExpanded]), 644 M.BeginExpanded, PrintToken(ExpandedTokens[M.EndExpanded]), 645 M.EndExpanded); 646 } 647 } 648 return OS.str(); 649 } 650