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