ARCMT.cpp revision 305c613af6cfc40e519c75d9d2c84c6fa9a841c0
1//===--- ARCMT.cpp - Migration to ARC mode --------------------------------===// 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 "Internals.h" 11#include "clang/Frontend/ASTUnit.h" 12#include "clang/Frontend/CompilerInstance.h" 13#include "clang/Frontend/FrontendAction.h" 14#include "clang/Frontend/TextDiagnosticPrinter.h" 15#include "clang/Frontend/Utils.h" 16#include "clang/AST/ASTConsumer.h" 17#include "clang/Rewrite/Core/Rewriter.h" 18#include "clang/Sema/SemaDiagnostic.h" 19#include "clang/Basic/DiagnosticCategories.h" 20#include "clang/Lex/Preprocessor.h" 21#include "llvm/Support/MemoryBuffer.h" 22#include "llvm/ADT/Triple.h" 23using namespace clang; 24using namespace arcmt; 25 26bool CapturedDiagList::clearDiagnostic(ArrayRef<unsigned> IDs, 27 SourceRange range) { 28 if (range.isInvalid()) 29 return false; 30 31 bool cleared = false; 32 ListTy::iterator I = List.begin(); 33 while (I != List.end()) { 34 FullSourceLoc diagLoc = I->getLocation(); 35 if ((IDs.empty() || // empty means clear all diagnostics in the range. 36 std::find(IDs.begin(), IDs.end(), I->getID()) != IDs.end()) && 37 !diagLoc.isBeforeInTranslationUnitThan(range.getBegin()) && 38 (diagLoc == range.getEnd() || 39 diagLoc.isBeforeInTranslationUnitThan(range.getEnd()))) { 40 cleared = true; 41 ListTy::iterator eraseS = I++; 42 while (I != List.end() && I->getLevel() == DiagnosticsEngine::Note) 43 ++I; 44 // Clear the diagnostic and any notes following it. 45 I = List.erase(eraseS, I); 46 continue; 47 } 48 49 ++I; 50 } 51 52 return cleared; 53} 54 55bool CapturedDiagList::hasDiagnostic(ArrayRef<unsigned> IDs, 56 SourceRange range) const { 57 if (range.isInvalid()) 58 return false; 59 60 ListTy::const_iterator I = List.begin(); 61 while (I != List.end()) { 62 FullSourceLoc diagLoc = I->getLocation(); 63 if ((IDs.empty() || // empty means any diagnostic in the range. 64 std::find(IDs.begin(), IDs.end(), I->getID()) != IDs.end()) && 65 !diagLoc.isBeforeInTranslationUnitThan(range.getBegin()) && 66 (diagLoc == range.getEnd() || 67 diagLoc.isBeforeInTranslationUnitThan(range.getEnd()))) { 68 return true; 69 } 70 71 ++I; 72 } 73 74 return false; 75} 76 77void CapturedDiagList::reportDiagnostics(DiagnosticsEngine &Diags) const { 78 for (ListTy::const_iterator I = List.begin(), E = List.end(); I != E; ++I) 79 Diags.Report(*I); 80} 81 82bool CapturedDiagList::hasErrors() const { 83 for (ListTy::const_iterator I = List.begin(), E = List.end(); I != E; ++I) 84 if (I->getLevel() >= DiagnosticsEngine::Error) 85 return true; 86 87 return false; 88} 89 90namespace { 91 92class CaptureDiagnosticConsumer : public DiagnosticConsumer { 93 DiagnosticsEngine &Diags; 94 DiagnosticConsumer &DiagClient; 95 CapturedDiagList &CapturedDiags; 96 bool HasBegunSourceFile; 97public: 98 CaptureDiagnosticConsumer(DiagnosticsEngine &diags, 99 DiagnosticConsumer &client, 100 CapturedDiagList &capturedDiags) 101 : Diags(diags), DiagClient(client), CapturedDiags(capturedDiags), 102 HasBegunSourceFile(false) { } 103 104 virtual void BeginSourceFile(const LangOptions &Opts, 105 const Preprocessor *PP) { 106 // Pass BeginSourceFile message onto DiagClient on first call. 107 // The corresponding EndSourceFile call will be made from an 108 // explicit call to FinishCapture. 109 if (!HasBegunSourceFile) { 110 DiagClient.BeginSourceFile(Opts, PP); 111 HasBegunSourceFile = true; 112 } 113 } 114 115 void FinishCapture() { 116 // Call EndSourceFile on DiagClient on completion of capture to 117 // enable VerifyDiagnosticConsumer to check diagnostics *after* 118 // it has received the diagnostic list. 119 if (HasBegunSourceFile) { 120 DiagClient.EndSourceFile(); 121 HasBegunSourceFile = false; 122 } 123 } 124 125 virtual ~CaptureDiagnosticConsumer() { 126 assert(!HasBegunSourceFile && "FinishCapture not called!"); 127 } 128 129 virtual void HandleDiagnostic(DiagnosticsEngine::Level level, 130 const Diagnostic &Info) { 131 if (DiagnosticIDs::isARCDiagnostic(Info.getID()) || 132 level >= DiagnosticsEngine::Error || level == DiagnosticsEngine::Note) { 133 CapturedDiags.push_back(StoredDiagnostic(level, Info)); 134 return; 135 } 136 137 // Non-ARC warnings are ignored. 138 Diags.setLastDiagnosticIgnored(); 139 } 140 141 DiagnosticConsumer *clone(DiagnosticsEngine &Diags) const { 142 // Just drop any diagnostics that come from cloned consumers; they'll 143 // have different source managers anyway. 144 return new IgnoringDiagConsumer(); 145 } 146}; 147 148} // end anonymous namespace 149 150static inline StringRef SimulatorVersionDefineName() { 151 return "__IPHONE_OS_VERSION_MIN_REQUIRED="; 152} 153 154/// \brief Parse the simulator version define: 155/// __IPHONE_OS_VERSION_MIN_REQUIRED=([0-9])([0-9][0-9])([0-9][0-9]) 156// and return the grouped values as integers, e.g: 157// __IPHONE_OS_VERSION_MIN_REQUIRED=40201 158// will return Major=4, Minor=2, Micro=1. 159static bool GetVersionFromSimulatorDefine(StringRef define, 160 unsigned &Major, unsigned &Minor, 161 unsigned &Micro) { 162 assert(define.startswith(SimulatorVersionDefineName())); 163 StringRef name, version; 164 llvm::tie(name, version) = define.split('='); 165 if (version.empty()) 166 return false; 167 std::string verstr = version.str(); 168 char *end; 169 unsigned num = (unsigned) strtol(verstr.c_str(), &end, 10); 170 if (*end != '\0') 171 return false; 172 Major = num / 10000; 173 num = num % 10000; 174 Minor = num / 100; 175 Micro = num % 100; 176 return true; 177} 178 179static bool HasARCRuntime(CompilerInvocation &origCI) { 180 // This duplicates some functionality from Darwin::AddDeploymentTarget 181 // but this function is well defined, so keep it decoupled from the driver 182 // and avoid unrelated complications. 183 184 for (unsigned i = 0, e = origCI.getPreprocessorOpts().Macros.size(); 185 i != e; ++i) { 186 StringRef define = origCI.getPreprocessorOpts().Macros[i].first; 187 bool isUndef = origCI.getPreprocessorOpts().Macros[i].second; 188 if (isUndef) 189 continue; 190 if (!define.startswith(SimulatorVersionDefineName())) 191 continue; 192 unsigned Major = 0, Minor = 0, Micro = 0; 193 if (GetVersionFromSimulatorDefine(define, Major, Minor, Micro) && 194 Major < 10 && Minor < 100 && Micro < 100) 195 return Major >= 5; 196 } 197 198 llvm::Triple triple(origCI.getTargetOpts().Triple); 199 200 if (triple.getOS() == llvm::Triple::IOS) 201 return triple.getOSMajorVersion() >= 5; 202 203 if (triple.getOS() == llvm::Triple::Darwin) 204 return triple.getOSMajorVersion() >= 11; 205 206 if (triple.getOS() == llvm::Triple::MacOSX) { 207 unsigned Major, Minor, Micro; 208 triple.getOSVersion(Major, Minor, Micro); 209 return Major > 10 || (Major == 10 && Minor >= 7); 210 } 211 212 return false; 213} 214 215static CompilerInvocation * 216createInvocationForMigration(CompilerInvocation &origCI) { 217 OwningPtr<CompilerInvocation> CInvok; 218 CInvok.reset(new CompilerInvocation(origCI)); 219 CInvok->getPreprocessorOpts().ImplicitPCHInclude = std::string(); 220 CInvok->getPreprocessorOpts().ImplicitPTHInclude = std::string(); 221 std::string define = getARCMTMacroName(); 222 define += '='; 223 CInvok->getPreprocessorOpts().addMacroDef(define); 224 CInvok->getLangOpts()->ObjCAutoRefCount = true; 225 CInvok->getLangOpts()->setGC(LangOptions::NonGC); 226 CInvok->getDiagnosticOpts().ErrorLimit = 0; 227 CInvok->getDiagnosticOpts().PedanticErrors = 0; 228 229 // Ignore -Werror flags when migrating. 230 std::vector<std::string> WarnOpts; 231 for (std::vector<std::string>::iterator 232 I = CInvok->getDiagnosticOpts().Warnings.begin(), 233 E = CInvok->getDiagnosticOpts().Warnings.end(); I != E; ++I) { 234 if (!StringRef(*I).startswith("error")) 235 WarnOpts.push_back(*I); 236 } 237 WarnOpts.push_back("error=arc-unsafe-retained-assign"); 238 CInvok->getDiagnosticOpts().Warnings = llvm_move(WarnOpts); 239 240 CInvok->getLangOpts()->ObjCARCWeak = HasARCRuntime(origCI); 241 242 return CInvok.take(); 243} 244 245static void emitPremigrationErrors(const CapturedDiagList &arcDiags, 246 const DiagnosticOptions &diagOpts, 247 Preprocessor &PP) { 248 TextDiagnosticPrinter printer(llvm::errs(), diagOpts); 249 IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 250 IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 251 new DiagnosticsEngine(DiagID, &printer, /*ShouldOwnClient=*/false)); 252 Diags->setSourceManager(&PP.getSourceManager()); 253 254 printer.BeginSourceFile(PP.getLangOpts(), &PP); 255 arcDiags.reportDiagnostics(*Diags); 256 printer.EndSourceFile(); 257} 258 259//===----------------------------------------------------------------------===// 260// checkForManualIssues. 261//===----------------------------------------------------------------------===// 262 263bool arcmt::checkForManualIssues(CompilerInvocation &origCI, 264 const FrontendInputFile &Input, 265 DiagnosticConsumer *DiagClient, 266 bool emitPremigrationARCErrors, 267 StringRef plistOut) { 268 if (!origCI.getLangOpts()->ObjC1) 269 return false; 270 271 LangOptions::GCMode OrigGCMode = origCI.getLangOpts()->getGC(); 272 bool NoNSAllocReallocError = origCI.getMigratorOpts().NoNSAllocReallocError; 273 bool NoFinalizeRemoval = origCI.getMigratorOpts().NoFinalizeRemoval; 274 275 std::vector<TransformFn> transforms = arcmt::getAllTransformations(OrigGCMode, 276 NoFinalizeRemoval); 277 assert(!transforms.empty()); 278 279 OwningPtr<CompilerInvocation> CInvok; 280 CInvok.reset(createInvocationForMigration(origCI)); 281 CInvok->getFrontendOpts().Inputs.clear(); 282 CInvok->getFrontendOpts().Inputs.push_back(Input); 283 284 CapturedDiagList capturedDiags; 285 286 assert(DiagClient); 287 IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 288 IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 289 new DiagnosticsEngine(DiagID, DiagClient, /*ShouldOwnClient=*/false)); 290 291 // Filter of all diagnostics. 292 CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags); 293 Diags->setClient(&errRec, /*ShouldOwnClient=*/false); 294 295 OwningPtr<ASTUnit> Unit( 296 ASTUnit::LoadFromCompilerInvocationAction(CInvok.take(), Diags)); 297 if (!Unit) { 298 errRec.FinishCapture(); 299 return true; 300 } 301 302 // Don't filter diagnostics anymore. 303 Diags->setClient(DiagClient, /*ShouldOwnClient=*/false); 304 305 ASTContext &Ctx = Unit->getASTContext(); 306 307 if (Diags->hasFatalErrorOccurred()) { 308 Diags->Reset(); 309 DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor()); 310 capturedDiags.reportDiagnostics(*Diags); 311 DiagClient->EndSourceFile(); 312 errRec.FinishCapture(); 313 return true; 314 } 315 316 if (emitPremigrationARCErrors) 317 emitPremigrationErrors(capturedDiags, origCI.getDiagnosticOpts(), 318 Unit->getPreprocessor()); 319 if (!plistOut.empty()) { 320 SmallVector<StoredDiagnostic, 8> arcDiags; 321 for (CapturedDiagList::iterator 322 I = capturedDiags.begin(), E = capturedDiags.end(); I != E; ++I) 323 arcDiags.push_back(*I); 324 writeARCDiagsToPlist(plistOut, arcDiags, 325 Ctx.getSourceManager(), Ctx.getLangOpts()); 326 } 327 328 // After parsing of source files ended, we want to reuse the 329 // diagnostics objects to emit further diagnostics. 330 // We call BeginSourceFile because DiagnosticConsumer requires that 331 // diagnostics with source range information are emitted only in between 332 // BeginSourceFile() and EndSourceFile(). 333 DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor()); 334 335 // No macros will be added since we are just checking and we won't modify 336 // source code. 337 std::vector<SourceLocation> ARCMTMacroLocs; 338 339 TransformActions testAct(*Diags, capturedDiags, Ctx, Unit->getPreprocessor()); 340 MigrationPass pass(Ctx, OrigGCMode, Unit->getSema(), testAct, ARCMTMacroLocs); 341 pass.setNSAllocReallocError(NoNSAllocReallocError); 342 pass.setNoFinalizeRemoval(NoFinalizeRemoval); 343 344 for (unsigned i=0, e = transforms.size(); i != e; ++i) 345 transforms[i](pass); 346 347 capturedDiags.reportDiagnostics(*Diags); 348 349 DiagClient->EndSourceFile(); 350 errRec.FinishCapture(); 351 352 // If we are migrating code that gets the '-fobjc-arc' flag, make sure 353 // to remove it so that we don't get errors from normal compilation. 354 origCI.getLangOpts()->ObjCAutoRefCount = false; 355 356 return capturedDiags.hasErrors() || testAct.hasReportedErrors(); 357} 358 359//===----------------------------------------------------------------------===// 360// applyTransformations. 361//===----------------------------------------------------------------------===// 362 363static bool applyTransforms(CompilerInvocation &origCI, 364 const FrontendInputFile &Input, 365 DiagnosticConsumer *DiagClient, 366 StringRef outputDir, 367 bool emitPremigrationARCErrors, 368 StringRef plistOut) { 369 if (!origCI.getLangOpts()->ObjC1) 370 return false; 371 372 LangOptions::GCMode OrigGCMode = origCI.getLangOpts()->getGC(); 373 374 // Make sure checking is successful first. 375 CompilerInvocation CInvokForCheck(origCI); 376 if (arcmt::checkForManualIssues(CInvokForCheck, Input, DiagClient, 377 emitPremigrationARCErrors, plistOut)) 378 return true; 379 380 CompilerInvocation CInvok(origCI); 381 CInvok.getFrontendOpts().Inputs.clear(); 382 CInvok.getFrontendOpts().Inputs.push_back(Input); 383 384 MigrationProcess migration(CInvok, DiagClient, outputDir); 385 bool NoFinalizeRemoval = origCI.getMigratorOpts().NoFinalizeRemoval; 386 387 std::vector<TransformFn> transforms = arcmt::getAllTransformations(OrigGCMode, 388 NoFinalizeRemoval); 389 assert(!transforms.empty()); 390 391 for (unsigned i=0, e = transforms.size(); i != e; ++i) { 392 bool err = migration.applyTransform(transforms[i]); 393 if (err) return true; 394 } 395 396 IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 397 IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 398 new DiagnosticsEngine(DiagID, DiagClient, /*ShouldOwnClient=*/false)); 399 400 if (outputDir.empty()) { 401 origCI.getLangOpts()->ObjCAutoRefCount = true; 402 return migration.getRemapper().overwriteOriginal(*Diags); 403 } else { 404 // If we are migrating code that gets the '-fobjc-arc' flag, make sure 405 // to remove it so that we don't get errors from normal compilation. 406 origCI.getLangOpts()->ObjCAutoRefCount = false; 407 return migration.getRemapper().flushToDisk(outputDir, *Diags); 408 } 409} 410 411bool arcmt::applyTransformations(CompilerInvocation &origCI, 412 const FrontendInputFile &Input, 413 DiagnosticConsumer *DiagClient) { 414 return applyTransforms(origCI, Input, DiagClient, 415 StringRef(), false, StringRef()); 416} 417 418bool arcmt::migrateWithTemporaryFiles(CompilerInvocation &origCI, 419 const FrontendInputFile &Input, 420 DiagnosticConsumer *DiagClient, 421 StringRef outputDir, 422 bool emitPremigrationARCErrors, 423 StringRef plistOut) { 424 assert(!outputDir.empty() && "Expected output directory path"); 425 return applyTransforms(origCI, Input, DiagClient, 426 outputDir, emitPremigrationARCErrors, plistOut); 427} 428 429bool arcmt::getFileRemappings(std::vector<std::pair<std::string,std::string> > & 430 remap, 431 StringRef outputDir, 432 DiagnosticConsumer *DiagClient) { 433 assert(!outputDir.empty()); 434 435 IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 436 IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 437 new DiagnosticsEngine(DiagID, DiagClient, /*ShouldOwnClient=*/false)); 438 439 FileRemapper remapper; 440 bool err = remapper.initFromDisk(outputDir, *Diags, 441 /*ignoreIfFilesChanged=*/true); 442 if (err) 443 return true; 444 445 PreprocessorOptions PPOpts; 446 remapper.applyMappings(PPOpts); 447 remap = PPOpts.RemappedFiles; 448 449 return false; 450} 451 452bool arcmt::getFileRemappingsFromFileList( 453 std::vector<std::pair<std::string,std::string> > &remap, 454 ArrayRef<StringRef> remapFiles, 455 DiagnosticConsumer *DiagClient) { 456 bool hasErrorOccurred = false; 457 llvm::StringMap<bool> Uniquer; 458 459 llvm::IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 460 llvm::IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 461 new DiagnosticsEngine(DiagID, DiagClient, /*ShouldOwnClient=*/false)); 462 463 for (ArrayRef<StringRef>::iterator 464 I = remapFiles.begin(), E = remapFiles.end(); I != E; ++I) { 465 StringRef file = *I; 466 467 FileRemapper remapper; 468 bool err = remapper.initFromFile(file, *Diags, 469 /*ignoreIfFilesChanged=*/true); 470 hasErrorOccurred = hasErrorOccurred || err; 471 if (err) 472 continue; 473 474 PreprocessorOptions PPOpts; 475 remapper.applyMappings(PPOpts); 476 for (PreprocessorOptions::remapped_file_iterator 477 RI = PPOpts.remapped_file_begin(), RE = PPOpts.remapped_file_end(); 478 RI != RE; ++RI) { 479 bool &inserted = Uniquer[RI->first]; 480 if (inserted) 481 continue; 482 inserted = true; 483 remap.push_back(*RI); 484 } 485 } 486 487 return hasErrorOccurred; 488} 489 490//===----------------------------------------------------------------------===// 491// CollectTransformActions. 492//===----------------------------------------------------------------------===// 493 494namespace { 495 496class ARCMTMacroTrackerPPCallbacks : public PPCallbacks { 497 std::vector<SourceLocation> &ARCMTMacroLocs; 498 499public: 500 ARCMTMacroTrackerPPCallbacks(std::vector<SourceLocation> &ARCMTMacroLocs) 501 : ARCMTMacroLocs(ARCMTMacroLocs) { } 502 503 virtual void MacroExpands(const Token &MacroNameTok, const MacroInfo *MI, 504 SourceRange Range) { 505 if (MacroNameTok.getIdentifierInfo()->getName() == getARCMTMacroName()) 506 ARCMTMacroLocs.push_back(MacroNameTok.getLocation()); 507 } 508}; 509 510class ARCMTMacroTrackerAction : public ASTFrontendAction { 511 std::vector<SourceLocation> &ARCMTMacroLocs; 512 513public: 514 ARCMTMacroTrackerAction(std::vector<SourceLocation> &ARCMTMacroLocs) 515 : ARCMTMacroLocs(ARCMTMacroLocs) { } 516 517 virtual ASTConsumer *CreateASTConsumer(CompilerInstance &CI, 518 StringRef InFile) { 519 CI.getPreprocessor().addPPCallbacks( 520 new ARCMTMacroTrackerPPCallbacks(ARCMTMacroLocs)); 521 return new ASTConsumer(); 522 } 523}; 524 525class RewritesApplicator : public TransformActions::RewriteReceiver { 526 Rewriter &rewriter; 527 MigrationProcess::RewriteListener *Listener; 528 529public: 530 RewritesApplicator(Rewriter &rewriter, ASTContext &ctx, 531 MigrationProcess::RewriteListener *listener) 532 : rewriter(rewriter), Listener(listener) { 533 if (Listener) 534 Listener->start(ctx); 535 } 536 ~RewritesApplicator() { 537 if (Listener) 538 Listener->finish(); 539 } 540 541 virtual void insert(SourceLocation loc, StringRef text) { 542 bool err = rewriter.InsertText(loc, text, /*InsertAfter=*/true, 543 /*indentNewLines=*/true); 544 if (!err && Listener) 545 Listener->insert(loc, text); 546 } 547 548 virtual void remove(CharSourceRange range) { 549 Rewriter::RewriteOptions removeOpts; 550 removeOpts.IncludeInsertsAtBeginOfRange = false; 551 removeOpts.IncludeInsertsAtEndOfRange = false; 552 removeOpts.RemoveLineIfEmpty = true; 553 554 bool err = rewriter.RemoveText(range, removeOpts); 555 if (!err && Listener) 556 Listener->remove(range); 557 } 558 559 virtual void increaseIndentation(CharSourceRange range, 560 SourceLocation parentIndent) { 561 rewriter.IncreaseIndentation(range, parentIndent); 562 } 563}; 564 565} // end anonymous namespace. 566 567/// \brief Anchor for VTable. 568MigrationProcess::RewriteListener::~RewriteListener() { } 569 570MigrationProcess::MigrationProcess(const CompilerInvocation &CI, 571 DiagnosticConsumer *diagClient, 572 StringRef outputDir) 573 : OrigCI(CI), DiagClient(diagClient) { 574 if (!outputDir.empty()) { 575 IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 576 IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 577 new DiagnosticsEngine(DiagID, DiagClient, /*ShouldOwnClient=*/false)); 578 Remapper.initFromDisk(outputDir, *Diags, /*ignoreIfFilesChanges=*/true); 579 } 580} 581 582bool MigrationProcess::applyTransform(TransformFn trans, 583 RewriteListener *listener) { 584 OwningPtr<CompilerInvocation> CInvok; 585 CInvok.reset(createInvocationForMigration(OrigCI)); 586 CInvok->getDiagnosticOpts().IgnoreWarnings = true; 587 588 Remapper.applyMappings(CInvok->getPreprocessorOpts()); 589 590 CapturedDiagList capturedDiags; 591 std::vector<SourceLocation> ARCMTMacroLocs; 592 593 assert(DiagClient); 594 IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); 595 IntrusiveRefCntPtr<DiagnosticsEngine> Diags( 596 new DiagnosticsEngine(DiagID, DiagClient, /*ShouldOwnClient=*/false)); 597 598 // Filter of all diagnostics. 599 CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags); 600 Diags->setClient(&errRec, /*ShouldOwnClient=*/false); 601 602 OwningPtr<ARCMTMacroTrackerAction> ASTAction; 603 ASTAction.reset(new ARCMTMacroTrackerAction(ARCMTMacroLocs)); 604 605 OwningPtr<ASTUnit> Unit( 606 ASTUnit::LoadFromCompilerInvocationAction(CInvok.take(), Diags, 607 ASTAction.get())); 608 if (!Unit) { 609 errRec.FinishCapture(); 610 return true; 611 } 612 Unit->setOwnsRemappedFileBuffers(false); // FileRemapper manages that. 613 614 // Don't filter diagnostics anymore. 615 Diags->setClient(DiagClient, /*ShouldOwnClient=*/false); 616 617 ASTContext &Ctx = Unit->getASTContext(); 618 619 if (Diags->hasFatalErrorOccurred()) { 620 Diags->Reset(); 621 DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor()); 622 capturedDiags.reportDiagnostics(*Diags); 623 DiagClient->EndSourceFile(); 624 errRec.FinishCapture(); 625 return true; 626 } 627 628 // After parsing of source files ended, we want to reuse the 629 // diagnostics objects to emit further diagnostics. 630 // We call BeginSourceFile because DiagnosticConsumer requires that 631 // diagnostics with source range information are emitted only in between 632 // BeginSourceFile() and EndSourceFile(). 633 DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor()); 634 635 Rewriter rewriter(Ctx.getSourceManager(), Ctx.getLangOpts()); 636 TransformActions TA(*Diags, capturedDiags, Ctx, Unit->getPreprocessor()); 637 MigrationPass pass(Ctx, OrigCI.getLangOpts()->getGC(), 638 Unit->getSema(), TA, ARCMTMacroLocs); 639 640 trans(pass); 641 642 { 643 RewritesApplicator applicator(rewriter, Ctx, listener); 644 TA.applyRewrites(applicator); 645 } 646 647 DiagClient->EndSourceFile(); 648 errRec.FinishCapture(); 649 650 if (DiagClient->getNumErrors()) 651 return true; 652 653 for (Rewriter::buffer_iterator 654 I = rewriter.buffer_begin(), E = rewriter.buffer_end(); I != E; ++I) { 655 FileID FID = I->first; 656 RewriteBuffer &buf = I->second; 657 const FileEntry *file = Ctx.getSourceManager().getFileEntryForID(FID); 658 assert(file); 659 std::string newFname = file->getName(); 660 newFname += "-trans"; 661 SmallString<512> newText; 662 llvm::raw_svector_ostream vecOS(newText); 663 buf.write(vecOS); 664 vecOS.flush(); 665 llvm::MemoryBuffer *memBuf = llvm::MemoryBuffer::getMemBufferCopy( 666 StringRef(newText.data(), newText.size()), newFname); 667 SmallString<64> filePath(file->getName()); 668 Unit->getFileManager().FixupRelativePath(filePath); 669 Remapper.remap(filePath.str(), memBuf); 670 } 671 672 return false; 673} 674