| 1 | |
| 2 | |
| 3 | |
| 4 | |
| 5 | |
| 6 | |
| 7 | |
| 8 | |
| 9 | |
| 10 | |
| 11 | |
| 12 | |
| 13 | |
| 14 | #include "clang/Lex/PreprocessingRecord.h" |
| 15 | #include "clang/Basic/IdentifierTable.h" |
| 16 | #include "clang/Basic/LLVM.h" |
| 17 | #include "clang/Basic/SourceLocation.h" |
| 18 | #include "clang/Basic/SourceManager.h" |
| 19 | #include "clang/Basic/TokenKinds.h" |
| 20 | #include "clang/Lex/MacroInfo.h" |
| 21 | #include "clang/Lex/Token.h" |
| 22 | #include "llvm/ADT/DenseMap.h" |
| 23 | #include "llvm/ADT/Optional.h" |
| 24 | #include "llvm/ADT/StringRef.h" |
| 25 | #include "llvm/ADT/iterator_range.h" |
| 26 | #include "llvm/Support/Capacity.h" |
| 27 | #include "llvm/Support/Casting.h" |
| 28 | #include "llvm/Support/ErrorHandling.h" |
| 29 | #include <algorithm> |
| 30 | #include <cassert> |
| 31 | #include <cstddef> |
| 32 | #include <cstring> |
| 33 | #include <iterator> |
| 34 | #include <utility> |
| 35 | #include <vector> |
| 36 | |
| 37 | using namespace clang; |
| 38 | |
| 39 | ExternalPreprocessingRecordSource::~ExternalPreprocessingRecordSource() = |
| 40 | default; |
| 41 | |
| 42 | InclusionDirective::InclusionDirective(PreprocessingRecord &PPRec, |
| 43 | InclusionKind Kind, StringRef FileName, |
| 44 | bool InQuotes, bool ImportedModule, |
| 45 | const FileEntry *File, SourceRange Range) |
| 46 | : PreprocessingDirective(InclusionDirectiveKind, Range), InQuotes(InQuotes), |
| 47 | Kind(Kind), ImportedModule(ImportedModule), File(File) { |
| 48 | char *Memory = (char *)PPRec.Allocate(FileName.size() + 1, alignof(char)); |
| 49 | memcpy(Memory, FileName.data(), FileName.size()); |
| 50 | Memory[FileName.size()] = 0; |
| 51 | this->FileName = StringRef(Memory, FileName.size()); |
| 52 | } |
| 53 | |
| 54 | PreprocessingRecord::PreprocessingRecord(SourceManager &SM) : SourceMgr(SM) {} |
| 55 | |
| 56 | |
| 57 | |
| 58 | llvm::iterator_range<PreprocessingRecord::iterator> |
| 59 | PreprocessingRecord::getPreprocessedEntitiesInRange(SourceRange Range) { |
| 60 | if (Range.isInvalid()) |
| 61 | return llvm::make_range(iterator(), iterator()); |
| 62 | |
| 63 | if (CachedRangeQuery.Range == Range) { |
| 64 | return llvm::make_range(iterator(this, CachedRangeQuery.Result.first), |
| 65 | iterator(this, CachedRangeQuery.Result.second)); |
| 66 | } |
| 67 | |
| 68 | std::pair<int, int> Res = getPreprocessedEntitiesInRangeSlow(Range); |
| 69 | |
| 70 | CachedRangeQuery.Range = Range; |
| 71 | CachedRangeQuery.Result = Res; |
| 72 | |
| 73 | return llvm::make_range(iterator(this, Res.first), |
| 74 | iterator(this, Res.second)); |
| 75 | } |
| 76 | |
| 77 | static bool isPreprocessedEntityIfInFileID(PreprocessedEntity *PPE, FileID FID, |
| 78 | SourceManager &SM) { |
| 79 | assert(FID.isValid()); |
| 80 | if (!PPE) |
| 81 | return false; |
| 82 | |
| 83 | SourceLocation Loc = PPE->getSourceRange().getBegin(); |
| 84 | if (Loc.isInvalid()) |
| 85 | return false; |
| 86 | |
| 87 | return SM.isInFileID(SM.getFileLoc(Loc), FID); |
| 88 | } |
| 89 | |
| 90 | |
| 91 | |
| 92 | |
| 93 | |
| 94 | |
| 95 | |
| 96 | |
| 97 | bool PreprocessingRecord::isEntityInFileID(iterator PPEI, FileID FID) { |
| 98 | if (FID.isInvalid()) |
| 99 | return false; |
| 100 | |
| 101 | int Pos = std::distance(iterator(this, 0), PPEI); |
| 102 | if (Pos < 0) { |
| 103 | if (unsigned(-Pos-1) >= LoadedPreprocessedEntities.size()) { |
| 104 | (0) . __assert_fail ("0 && \"Out-of bounds loaded preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 104, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(0 && "Out-of bounds loaded preprocessed entity"); |
| 105 | return false; |
| 106 | } |
| 107 | (0) . __assert_fail ("ExternalSource && \"No external source to load from\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 107, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(ExternalSource && "No external source to load from"); |
| 108 | unsigned LoadedIndex = LoadedPreprocessedEntities.size()+Pos; |
| 109 | if (PreprocessedEntity *PPE = LoadedPreprocessedEntities[LoadedIndex]) |
| 110 | return isPreprocessedEntityIfInFileID(PPE, FID, SourceMgr); |
| 111 | |
| 112 | |
| 113 | |
| 114 | Optional<bool> IsInFile = |
| 115 | ExternalSource->isPreprocessedEntityInFileID(LoadedIndex, FID); |
| 116 | if (IsInFile.hasValue()) |
| 117 | return IsInFile.getValue(); |
| 118 | |
| 119 | |
| 120 | |
| 121 | return isPreprocessedEntityIfInFileID( |
| 122 | getLoadedPreprocessedEntity(LoadedIndex), |
| 123 | FID, SourceMgr); |
| 124 | } |
| 125 | |
| 126 | if (unsigned(Pos) >= PreprocessedEntities.size()) { |
| 127 | (0) . __assert_fail ("0 && \"Out-of bounds local preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 127, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(0 && "Out-of bounds local preprocessed entity"); |
| 128 | return false; |
| 129 | } |
| 130 | return isPreprocessedEntityIfInFileID(PreprocessedEntities[Pos], |
| 131 | FID, SourceMgr); |
| 132 | } |
| 133 | |
| 134 | |
| 135 | |
| 136 | std::pair<int, int> |
| 137 | PreprocessingRecord::getPreprocessedEntitiesInRangeSlow(SourceRange Range) { |
| 138 | assert(Range.isValid()); |
| 139 | assert(!SourceMgr.isBeforeInTranslationUnit(Range.getEnd(),Range.getBegin())); |
| 140 | |
| 141 | std::pair<unsigned, unsigned> |
| 142 | Local = findLocalPreprocessedEntitiesInRange(Range); |
| 143 | |
| 144 | |
| 145 | if (!ExternalSource || SourceMgr.isLocalSourceLocation(Range.getBegin())) |
| 146 | return std::make_pair(Local.first, Local.second); |
| 147 | |
| 148 | std::pair<unsigned, unsigned> |
| 149 | Loaded = ExternalSource->findPreprocessedEntitiesInRange(Range); |
| 150 | |
| 151 | |
| 152 | if (Loaded.first == Loaded.second) |
| 153 | return std::make_pair(Local.first, Local.second); |
| 154 | |
| 155 | unsigned TotalLoaded = LoadedPreprocessedEntities.size(); |
| 156 | |
| 157 | |
| 158 | if (Local.first == Local.second) |
| 159 | return std::make_pair(int(Loaded.first)-TotalLoaded, |
| 160 | int(Loaded.second)-TotalLoaded); |
| 161 | |
| 162 | |
| 163 | return std::make_pair(int(Loaded.first)-TotalLoaded, Local.second); |
| 164 | } |
| 165 | |
| 166 | std::pair<unsigned, unsigned> |
| 167 | PreprocessingRecord::findLocalPreprocessedEntitiesInRange( |
| 168 | SourceRange Range) const { |
| 169 | if (Range.isInvalid()) |
| 170 | return std::make_pair(0,0); |
| 171 | assert(!SourceMgr.isBeforeInTranslationUnit(Range.getEnd(),Range.getBegin())); |
| 172 | |
| 173 | unsigned Begin = findBeginLocalPreprocessedEntity(Range.getBegin()); |
| 174 | unsigned End = findEndLocalPreprocessedEntity(Range.getEnd()); |
| 175 | return std::make_pair(Begin, End); |
| 176 | } |
| 177 | |
| 178 | namespace { |
| 179 | |
| 180 | template <SourceLocation (SourceRange::*getRangeLoc)() const> |
| 181 | struct PPEntityComp { |
| 182 | const SourceManager &SM; |
| 183 | |
| 184 | explicit PPEntityComp(const SourceManager &SM) : SM(SM) {} |
| 185 | |
| 186 | bool operator()(PreprocessedEntity *L, PreprocessedEntity *R) const { |
| 187 | SourceLocation LHS = getLoc(L); |
| 188 | SourceLocation RHS = getLoc(R); |
| 189 | return SM.isBeforeInTranslationUnit(LHS, RHS); |
| 190 | } |
| 191 | |
| 192 | bool operator()(PreprocessedEntity *L, SourceLocation RHS) const { |
| 193 | SourceLocation LHS = getLoc(L); |
| 194 | return SM.isBeforeInTranslationUnit(LHS, RHS); |
| 195 | } |
| 196 | |
| 197 | bool operator()(SourceLocation LHS, PreprocessedEntity *R) const { |
| 198 | SourceLocation RHS = getLoc(R); |
| 199 | return SM.isBeforeInTranslationUnit(LHS, RHS); |
| 200 | } |
| 201 | |
| 202 | SourceLocation getLoc(PreprocessedEntity *PPE) const { |
| 203 | SourceRange Range = PPE->getSourceRange(); |
| 204 | return (Range.*getRangeLoc)(); |
| 205 | } |
| 206 | }; |
| 207 | |
| 208 | } |
| 209 | |
| 210 | unsigned PreprocessingRecord::findBeginLocalPreprocessedEntity( |
| 211 | SourceLocation Loc) const { |
| 212 | if (SourceMgr.isLoadedSourceLocation(Loc)) |
| 213 | return 0; |
| 214 | |
| 215 | size_t Count = PreprocessedEntities.size(); |
| 216 | size_t Half; |
| 217 | std::vector<PreprocessedEntity *>::const_iterator |
| 218 | First = PreprocessedEntities.begin(); |
| 219 | std::vector<PreprocessedEntity *>::const_iterator I; |
| 220 | |
| 221 | |
| 222 | |
| 223 | |
| 224 | |
| 225 | while (Count > 0) { |
| 226 | Half = Count/2; |
| 227 | I = First; |
| 228 | std::advance(I, Half); |
| 229 | if (SourceMgr.isBeforeInTranslationUnit((*I)->getSourceRange().getEnd(), |
| 230 | Loc)){ |
| 231 | First = I; |
| 232 | ++First; |
| 233 | Count = Count - Half - 1; |
| 234 | } else |
| 235 | Count = Half; |
| 236 | } |
| 237 | |
| 238 | return First - PreprocessedEntities.begin(); |
| 239 | } |
| 240 | |
| 241 | unsigned PreprocessingRecord::findEndLocalPreprocessedEntity( |
| 242 | SourceLocation Loc) const { |
| 243 | if (SourceMgr.isLoadedSourceLocation(Loc)) |
| 244 | return 0; |
| 245 | |
| 246 | std::vector<PreprocessedEntity *>::const_iterator |
| 247 | I = std::upper_bound(PreprocessedEntities.begin(), |
| 248 | PreprocessedEntities.end(), |
| 249 | Loc, |
| 250 | PPEntityComp<&SourceRange::getBegin>(SourceMgr)); |
| 251 | return I - PreprocessedEntities.begin(); |
| 252 | } |
| 253 | |
| 254 | PreprocessingRecord::PPEntityID |
| 255 | PreprocessingRecord::addPreprocessedEntity(PreprocessedEntity *Entity) { |
| 256 | assert(Entity); |
| 257 | SourceLocation BeginLoc = Entity->getSourceRange().getBegin(); |
| 258 | |
| 259 | if (isa<MacroDefinitionRecord>(Entity)) { |
| 260 | (0) . __assert_fail ("(PreprocessedEntities.empty() || !SourceMgr.isBeforeInTranslationUnit( BeginLoc, PreprocessedEntities.back()->getSourceRange().getBegin())) && \"a macro definition was encountered out-of-order\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 264, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert((PreprocessedEntities.empty() || |
| 261 | (0) . __assert_fail ("(PreprocessedEntities.empty() || !SourceMgr.isBeforeInTranslationUnit( BeginLoc, PreprocessedEntities.back()->getSourceRange().getBegin())) && \"a macro definition was encountered out-of-order\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 264, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> !SourceMgr.isBeforeInTranslationUnit( |
| 262 | (0) . __assert_fail ("(PreprocessedEntities.empty() || !SourceMgr.isBeforeInTranslationUnit( BeginLoc, PreprocessedEntities.back()->getSourceRange().getBegin())) && \"a macro definition was encountered out-of-order\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 264, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> BeginLoc, |
| 263 | (0) . __assert_fail ("(PreprocessedEntities.empty() || !SourceMgr.isBeforeInTranslationUnit( BeginLoc, PreprocessedEntities.back()->getSourceRange().getBegin())) && \"a macro definition was encountered out-of-order\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 264, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> PreprocessedEntities.back()->getSourceRange().getBegin())) && |
| 264 | (0) . __assert_fail ("(PreprocessedEntities.empty() || !SourceMgr.isBeforeInTranslationUnit( BeginLoc, PreprocessedEntities.back()->getSourceRange().getBegin())) && \"a macro definition was encountered out-of-order\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 264, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> "a macro definition was encountered out-of-order"); |
| 265 | PreprocessedEntities.push_back(Entity); |
| 266 | return getPPEntityID(PreprocessedEntities.size()-1, ); |
| 267 | } |
| 268 | |
| 269 | |
| 270 | if (PreprocessedEntities.empty() || |
| 271 | !SourceMgr.isBeforeInTranslationUnit(BeginLoc, |
| 272 | PreprocessedEntities.back()->getSourceRange().getBegin())) { |
| 273 | PreprocessedEntities.push_back(Entity); |
| 274 | return getPPEntityID(PreprocessedEntities.size()-1, ); |
| 275 | } |
| 276 | |
| 277 | |
| 278 | |
| 279 | |
| 280 | |
| 281 | |
| 282 | |
| 283 | |
| 284 | |
| 285 | |
| 286 | |
| 287 | |
| 288 | |
| 289 | using pp_iter = std::vector<PreprocessedEntity *>::iterator; |
| 290 | |
| 291 | |
| 292 | |
| 293 | unsigned count = 0; |
| 294 | for (pp_iter RI = PreprocessedEntities.end(), |
| 295 | Begin = PreprocessedEntities.begin(); |
| 296 | RI != Begin && count < 4; --RI, ++count) { |
| 297 | pp_iter I = RI; |
| 298 | --I; |
| 299 | if (!SourceMgr.isBeforeInTranslationUnit(BeginLoc, |
| 300 | (*I)->getSourceRange().getBegin())) { |
| 301 | pp_iter insertI = PreprocessedEntities.insert(RI, Entity); |
| 302 | return getPPEntityID(insertI - PreprocessedEntities.begin(), |
| 303 | ); |
| 304 | } |
| 305 | } |
| 306 | |
| 307 | |
| 308 | pp_iter I = std::upper_bound(PreprocessedEntities.begin(), |
| 309 | PreprocessedEntities.end(), |
| 310 | BeginLoc, |
| 311 | PPEntityComp<&SourceRange::getBegin>(SourceMgr)); |
| 312 | pp_iter insertI = PreprocessedEntities.insert(I, Entity); |
| 313 | return getPPEntityID(insertI - PreprocessedEntities.begin(), |
| 314 | ); |
| 315 | } |
| 316 | |
| 317 | void PreprocessingRecord::SetExternalSource( |
| 318 | ExternalPreprocessingRecordSource &Source) { |
| 319 | (0) . __assert_fail ("!ExternalSource && \"Preprocessing record already has an external source\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 320, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(!ExternalSource && |
| 320 | (0) . __assert_fail ("!ExternalSource && \"Preprocessing record already has an external source\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 320, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> "Preprocessing record already has an external source"); |
| 321 | ExternalSource = &Source; |
| 322 | } |
| 323 | |
| 324 | unsigned PreprocessingRecord::allocateLoadedEntities(unsigned NumEntities) { |
| 325 | unsigned Result = LoadedPreprocessedEntities.size(); |
| 326 | LoadedPreprocessedEntities.resize(LoadedPreprocessedEntities.size() |
| 327 | + NumEntities); |
| 328 | return Result; |
| 329 | } |
| 330 | |
| 331 | unsigned PreprocessingRecord::allocateSkippedRanges(unsigned NumRanges) { |
| 332 | unsigned Result = SkippedRanges.size(); |
| 333 | SkippedRanges.resize(SkippedRanges.size() + NumRanges); |
| 334 | SkippedRangesAllLoaded = false; |
| 335 | return Result; |
| 336 | } |
| 337 | |
| 338 | void PreprocessingRecord::ensureSkippedRangesLoaded() { |
| 339 | if (SkippedRangesAllLoaded || !ExternalSource) |
| 340 | return; |
| 341 | for (unsigned Index = 0; Index != SkippedRanges.size(); ++Index) { |
| 342 | if (SkippedRanges[Index].isInvalid()) |
| 343 | SkippedRanges[Index] = ExternalSource->ReadSkippedRange(Index); |
| 344 | } |
| 345 | SkippedRangesAllLoaded = true; |
| 346 | } |
| 347 | |
| 348 | void PreprocessingRecord::RegisterMacroDefinition(MacroInfo *Macro, |
| 349 | MacroDefinitionRecord *Def) { |
| 350 | MacroDefinitions[Macro] = Def; |
| 351 | } |
| 352 | |
| 353 | |
| 354 | PreprocessedEntity *PreprocessingRecord::getPreprocessedEntity(PPEntityID PPID){ |
| 355 | if (PPID.ID < 0) { |
| 356 | unsigned Index = -PPID.ID - 1; |
| 357 | (0) . __assert_fail ("Index < LoadedPreprocessedEntities.size() && \"Out-of bounds loaded preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 358, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(Index < LoadedPreprocessedEntities.size() && |
| 358 | (0) . __assert_fail ("Index < LoadedPreprocessedEntities.size() && \"Out-of bounds loaded preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 358, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> "Out-of bounds loaded preprocessed entity"); |
| 359 | return getLoadedPreprocessedEntity(Index); |
| 360 | } |
| 361 | |
| 362 | if (PPID.ID == 0) |
| 363 | return nullptr; |
| 364 | unsigned Index = PPID.ID - 1; |
| 365 | (0) . __assert_fail ("Index < PreprocessedEntities.size() && \"Out-of bounds local preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 366, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(Index < PreprocessedEntities.size() && |
| 366 | (0) . __assert_fail ("Index < PreprocessedEntities.size() && \"Out-of bounds local preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 366, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> "Out-of bounds local preprocessed entity"); |
| 367 | return PreprocessedEntities[Index]; |
| 368 | } |
| 369 | |
| 370 | |
| 371 | PreprocessedEntity * |
| 372 | PreprocessingRecord::getLoadedPreprocessedEntity(unsigned Index) { |
| 373 | (0) . __assert_fail ("Index < LoadedPreprocessedEntities.size() && \"Out-of bounds loaded preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 374, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(Index < LoadedPreprocessedEntities.size() && |
| 374 | (0) . __assert_fail ("Index < LoadedPreprocessedEntities.size() && \"Out-of bounds loaded preprocessed entity\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 374, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true"> "Out-of bounds loaded preprocessed entity"); |
| 375 | (0) . __assert_fail ("ExternalSource && \"No external source to load from\"", "/home/seafit/code_projects/clang_source/clang/lib/Lex/PreprocessingRecord.cpp", 375, __PRETTY_FUNCTION__))" file_link="../../../include/assert.h.html#88" macro="true">assert(ExternalSource && "No external source to load from"); |
| 376 | PreprocessedEntity *&Entity = LoadedPreprocessedEntities[Index]; |
| 377 | if (!Entity) { |
| 378 | Entity = ExternalSource->ReadPreprocessedEntity(Index); |
| 379 | if (!Entity) |
| 380 | Entity = new (*this) |
| 381 | PreprocessedEntity(PreprocessedEntity::InvalidKind, SourceRange()); |
| 382 | } |
| 383 | return Entity; |
| 384 | } |
| 385 | |
| 386 | MacroDefinitionRecord * |
| 387 | PreprocessingRecord::findMacroDefinition(const MacroInfo *MI) { |
| 388 | llvm::DenseMap<const MacroInfo *, MacroDefinitionRecord *>::iterator Pos = |
| 389 | MacroDefinitions.find(MI); |
| 390 | if (Pos == MacroDefinitions.end()) |
| 391 | return nullptr; |
| 392 | |
| 393 | return Pos->second; |
| 394 | } |
| 395 | |
| 396 | void PreprocessingRecord::addMacroExpansion(const Token &Id, |
| 397 | const MacroInfo *MI, |
| 398 | SourceRange Range) { |
| 399 | |
| 400 | if (Id.getLocation().isMacroID()) |
| 401 | return; |
| 402 | |
| 403 | if (MI->isBuiltinMacro()) |
| 404 | addPreprocessedEntity(new (*this) |
| 405 | MacroExpansion(Id.getIdentifierInfo(), Range)); |
| 406 | else if (MacroDefinitionRecord *Def = findMacroDefinition(MI)) |
| 407 | addPreprocessedEntity(new (*this) MacroExpansion(Def, Range)); |
| 408 | } |
| 409 | |
| 410 | void PreprocessingRecord::Ifdef(SourceLocation Loc, const Token &MacroNameTok, |
| 411 | const MacroDefinition &MD) { |
| 412 | |
| 413 | if (MD) |
| 414 | addMacroExpansion(MacroNameTok, MD.getMacroInfo(), |
| 415 | MacroNameTok.getLocation()); |
| 416 | } |
| 417 | |
| 418 | void PreprocessingRecord::Ifndef(SourceLocation Loc, const Token &MacroNameTok, |
| 419 | const MacroDefinition &MD) { |
| 420 | |
| 421 | if (MD) |
| 422 | addMacroExpansion(MacroNameTok, MD.getMacroInfo(), |
| 423 | MacroNameTok.getLocation()); |
| 424 | } |
| 425 | |
| 426 | void PreprocessingRecord::Defined(const Token &MacroNameTok, |
| 427 | const MacroDefinition &MD, |
| 428 | SourceRange Range) { |
| 429 | |
| 430 | if (MD) |
| 431 | addMacroExpansion(MacroNameTok, MD.getMacroInfo(), |
| 432 | MacroNameTok.getLocation()); |
| 433 | } |
| 434 | |
| 435 | void PreprocessingRecord::SourceRangeSkipped(SourceRange Range, |
| 436 | SourceLocation EndifLoc) { |
| 437 | assert(Range.isValid()); |
| 438 | SkippedRanges.emplace_back(Range.getBegin(), EndifLoc); |
| 439 | } |
| 440 | |
| 441 | void PreprocessingRecord::MacroExpands(const Token &Id, |
| 442 | const MacroDefinition &MD, |
| 443 | SourceRange Range, |
| 444 | const MacroArgs *Args) { |
| 445 | addMacroExpansion(Id, MD.getMacroInfo(), Range); |
| 446 | } |
| 447 | |
| 448 | void PreprocessingRecord::MacroDefined(const Token &Id, |
| 449 | const MacroDirective *MD) { |
| 450 | const MacroInfo *MI = MD->getMacroInfo(); |
| 451 | SourceRange R(MI->getDefinitionLoc(), MI->getDefinitionEndLoc()); |
| 452 | MacroDefinitionRecord *Def = |
| 453 | new (*this) MacroDefinitionRecord(Id.getIdentifierInfo(), R); |
| 454 | addPreprocessedEntity(Def); |
| 455 | MacroDefinitions[MI] = Def; |
| 456 | } |
| 457 | |
| 458 | void PreprocessingRecord::MacroUndefined(const Token &Id, |
| 459 | const MacroDefinition &MD, |
| 460 | const MacroDirective *Undef) { |
| 461 | MD.forAllDefinitions([&](MacroInfo *MI) { MacroDefinitions.erase(MI); }); |
| 462 | } |
| 463 | |
| 464 | void PreprocessingRecord::InclusionDirective( |
| 465 | SourceLocation HashLoc, |
| 466 | const Token &IncludeTok, |
| 467 | StringRef FileName, |
| 468 | bool IsAngled, |
| 469 | CharSourceRange FilenameRange, |
| 470 | const FileEntry *File, |
| 471 | StringRef SearchPath, |
| 472 | StringRef RelativePath, |
| 473 | const Module *Imported, |
| 474 | SrcMgr::CharacteristicKind FileType) { |
| 475 | InclusionDirective::InclusionKind Kind = InclusionDirective::Include; |
| 476 | |
| 477 | switch (IncludeTok.getIdentifierInfo()->getPPKeywordID()) { |
| 478 | case tok::pp_include: |
| 479 | Kind = InclusionDirective::Include; |
| 480 | break; |
| 481 | |
| 482 | case tok::pp_import: |
| 483 | Kind = InclusionDirective::Import; |
| 484 | break; |
| 485 | |
| 486 | case tok::pp_include_next: |
| 487 | Kind = InclusionDirective::IncludeNext; |
| 488 | break; |
| 489 | |
| 490 | case tok::pp___include_macros: |
| 491 | Kind = InclusionDirective::IncludeMacros; |
| 492 | break; |
| 493 | |
| 494 | default: |
| 495 | llvm_unreachable("Unknown include directive kind"); |
| 496 | } |
| 497 | |
| 498 | SourceLocation EndLoc; |
| 499 | if (!IsAngled) { |
| 500 | EndLoc = FilenameRange.getBegin(); |
| 501 | } else { |
| 502 | EndLoc = FilenameRange.getEnd(); |
| 503 | if (FilenameRange.isCharRange()) |
| 504 | EndLoc = EndLoc.getLocWithOffset(-1); |
| 505 | |
| 506 | } |
| 507 | clang::InclusionDirective *ID = |
| 508 | new (*this) clang::InclusionDirective(*this, Kind, FileName, !IsAngled, |
| 509 | (bool)Imported, File, |
| 510 | SourceRange(HashLoc, EndLoc)); |
| 511 | addPreprocessedEntity(ID); |
| 512 | } |
| 513 | |
| 514 | size_t PreprocessingRecord::getTotalMemory() const { |
| 515 | return BumpAlloc.getTotalMemory() |
| 516 | + llvm::capacity_in_bytes(MacroDefinitions) |
| 517 | + llvm::capacity_in_bytes(PreprocessedEntities) |
| 518 | + llvm::capacity_in_bytes(LoadedPreprocessedEntities) |
| 519 | + llvm::capacity_in_bytes(SkippedRanges); |
| 520 | } |
| 521 | |