ARCMT.cpp revision 249423 
1270866Simp//===--- ARCMT.cpp - Migration to ARC mode --------------------------------===//
2270866Simp//
3270866Simp//                     The LLVM Compiler Infrastructure
4270866Simp//
5270866Simp// This file is distributed under the University of Illinois Open Source
6270866Simp// License. See LICENSE.TXT for details.
7270866Simp//
8270866Simp//===----------------------------------------------------------------------===//
9270866Simp
10270866Simp#include "Internals.h"
11270866Simp#include "clang/AST/ASTConsumer.h"
12270866Simp#include "clang/Basic/DiagnosticCategories.h"
13270866Simp#include "clang/Frontend/ASTUnit.h"
14270866Simp#include "clang/Frontend/CompilerInstance.h"
15270866Simp#include "clang/Frontend/FrontendAction.h"
16270866Simp#include "clang/Frontend/TextDiagnosticPrinter.h"
17270866Simp#include "clang/Frontend/Utils.h"
18270866Simp#include "clang/Lex/Preprocessor.h"
19270866Simp#include "clang/Rewrite/Core/Rewriter.h"
20270866Simp#include "clang/Sema/SemaDiagnostic.h"
21270866Simp#include "clang/Serialization/ASTReader.h"
22270866Simp#include "llvm/ADT/Triple.h"
23270866Simp#include "llvm/Support/MemoryBuffer.h"
24using namespace clang;
25using namespace arcmt;
26
27bool 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
57bool 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
79void 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
84bool 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
92namespace {
93
94class CaptureDiagnosticConsumer : public DiagnosticConsumer {
95  DiagnosticsEngine &Diags;
96  DiagnosticConsumer &DiagClient;
97  CapturedDiagList &CapturedDiags;
98  bool HasBegunSourceFile;
99public:
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  DiagnosticConsumer *clone(DiagnosticsEngine &Diags) const {
145    // Just drop any diagnostics that come from cloned consumers; they'll
146    // have different source managers anyway.
147    return new IgnoringDiagConsumer();
148  }
149};
150
151} // end anonymous namespace
152
153static bool HasARCRuntime(CompilerInvocation &origCI) {
154  // This duplicates some functionality from Darwin::AddDeploymentTarget
155  // but this function is well defined, so keep it decoupled from the driver
156  // and avoid unrelated complications.
157  llvm::Triple triple(origCI.getTargetOpts().Triple);
158
159  if (triple.getOS() == llvm::Triple::IOS)
160    return triple.getOSMajorVersion() >= 5;
161
162  if (triple.getOS() == llvm::Triple::Darwin)
163    return triple.getOSMajorVersion() >= 11;
164
165  if (triple.getOS() == llvm::Triple::MacOSX) {
166    unsigned Major, Minor, Micro;
167    triple.getOSVersion(Major, Minor, Micro);
168    return Major > 10 || (Major == 10 && Minor >= 7);
169  }
170
171  return false;
172}
173
174static CompilerInvocation *
175createInvocationForMigration(CompilerInvocation &origCI) {
176  OwningPtr<CompilerInvocation> CInvok;
177  CInvok.reset(new CompilerInvocation(origCI));
178  PreprocessorOptions &PPOpts = CInvok->getPreprocessorOpts();
179  if (!PPOpts.ImplicitPCHInclude.empty()) {
180    // We can't use a PCH because it was likely built in non-ARC mode and we
181    // want to parse in ARC. Include the original header.
182    FileManager FileMgr(origCI.getFileSystemOpts());
183    IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
184    IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
185        new DiagnosticsEngine(DiagID, &origCI.getDiagnosticOpts(),
186                              new IgnoringDiagConsumer()));
187    std::string OriginalFile =
188        ASTReader::getOriginalSourceFile(PPOpts.ImplicitPCHInclude,
189                                         FileMgr, *Diags);
190    if (!OriginalFile.empty())
191      PPOpts.Includes.insert(PPOpts.Includes.begin(), OriginalFile);
192    PPOpts.ImplicitPCHInclude.clear();
193  }
194  // FIXME: Get the original header of a PTH as well.
195  CInvok->getPreprocessorOpts().ImplicitPTHInclude.clear();
196  std::string define = getARCMTMacroName();
197  define += '=';
198  CInvok->getPreprocessorOpts().addMacroDef(define);
199  CInvok->getLangOpts()->ObjCAutoRefCount = true;
200  CInvok->getLangOpts()->setGC(LangOptions::NonGC);
201  CInvok->getDiagnosticOpts().ErrorLimit = 0;
202  CInvok->getDiagnosticOpts().PedanticErrors = 0;
203
204  // Ignore -Werror flags when migrating.
205  std::vector<std::string> WarnOpts;
206  for (std::vector<std::string>::iterator
207         I = CInvok->getDiagnosticOpts().Warnings.begin(),
208         E = CInvok->getDiagnosticOpts().Warnings.end(); I != E; ++I) {
209    if (!StringRef(*I).startswith("error"))
210      WarnOpts.push_back(*I);
211  }
212  WarnOpts.push_back("error=arc-unsafe-retained-assign");
213  CInvok->getDiagnosticOpts().Warnings = llvm_move(WarnOpts);
214
215  CInvok->getLangOpts()->ObjCARCWeak = HasARCRuntime(origCI);
216
217  return CInvok.take();
218}
219
220static void emitPremigrationErrors(const CapturedDiagList &arcDiags,
221                                   DiagnosticOptions *diagOpts,
222                                   Preprocessor &PP) {
223  TextDiagnosticPrinter printer(llvm::errs(), diagOpts);
224  IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
225  IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
226      new DiagnosticsEngine(DiagID, diagOpts, &printer,
227                            /*ShouldOwnClient=*/false));
228  Diags->setSourceManager(&PP.getSourceManager());
229
230  printer.BeginSourceFile(PP.getLangOpts(), &PP);
231  arcDiags.reportDiagnostics(*Diags);
232  printer.EndSourceFile();
233}
234
235//===----------------------------------------------------------------------===//
236// checkForManualIssues.
237//===----------------------------------------------------------------------===//
238
239bool arcmt::checkForManualIssues(CompilerInvocation &origCI,
240                                 const FrontendInputFile &Input,
241                                 DiagnosticConsumer *DiagClient,
242                                 bool emitPremigrationARCErrors,
243                                 StringRef plistOut) {
244  if (!origCI.getLangOpts()->ObjC1)
245    return false;
246
247  LangOptions::GCMode OrigGCMode = origCI.getLangOpts()->getGC();
248  bool NoNSAllocReallocError = origCI.getMigratorOpts().NoNSAllocReallocError;
249  bool NoFinalizeRemoval = origCI.getMigratorOpts().NoFinalizeRemoval;
250
251  std::vector<TransformFn> transforms = arcmt::getAllTransformations(OrigGCMode,
252                                                                     NoFinalizeRemoval);
253  assert(!transforms.empty());
254
255  OwningPtr<CompilerInvocation> CInvok;
256  CInvok.reset(createInvocationForMigration(origCI));
257  CInvok->getFrontendOpts().Inputs.clear();
258  CInvok->getFrontendOpts().Inputs.push_back(Input);
259
260  CapturedDiagList capturedDiags;
261
262  assert(DiagClient);
263  IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
264  IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
265      new DiagnosticsEngine(DiagID, &origCI.getDiagnosticOpts(),
266                            DiagClient, /*ShouldOwnClient=*/false));
267
268  // Filter of all diagnostics.
269  CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags);
270  Diags->setClient(&errRec, /*ShouldOwnClient=*/false);
271
272  OwningPtr<ASTUnit> Unit(
273      ASTUnit::LoadFromCompilerInvocationAction(CInvok.take(), Diags));
274  if (!Unit) {
275    errRec.FinishCapture();
276    return true;
277  }
278
279  // Don't filter diagnostics anymore.
280  Diags->setClient(DiagClient, /*ShouldOwnClient=*/false);
281
282  ASTContext &Ctx = Unit->getASTContext();
283
284  if (Diags->hasFatalErrorOccurred()) {
285    Diags->Reset();
286    DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor());
287    capturedDiags.reportDiagnostics(*Diags);
288    DiagClient->EndSourceFile();
289    errRec.FinishCapture();
290    return true;
291  }
292
293  if (emitPremigrationARCErrors)
294    emitPremigrationErrors(capturedDiags, &origCI.getDiagnosticOpts(),
295                           Unit->getPreprocessor());
296  if (!plistOut.empty()) {
297    SmallVector<StoredDiagnostic, 8> arcDiags;
298    for (CapturedDiagList::iterator
299           I = capturedDiags.begin(), E = capturedDiags.end(); I != E; ++I)
300      arcDiags.push_back(*I);
301    writeARCDiagsToPlist(plistOut, arcDiags,
302                         Ctx.getSourceManager(), Ctx.getLangOpts());
303  }
304
305  // After parsing of source files ended, we want to reuse the
306  // diagnostics objects to emit further diagnostics.
307  // We call BeginSourceFile because DiagnosticConsumer requires that
308  // diagnostics with source range information are emitted only in between
309  // BeginSourceFile() and EndSourceFile().
310  DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor());
311
312  // No macros will be added since we are just checking and we won't modify
313  // source code.
314  std::vector<SourceLocation> ARCMTMacroLocs;
315
316  TransformActions testAct(*Diags, capturedDiags, Ctx, Unit->getPreprocessor());
317  MigrationPass pass(Ctx, OrigGCMode, Unit->getSema(), testAct, capturedDiags,
318                     ARCMTMacroLocs);
319  pass.setNSAllocReallocError(NoNSAllocReallocError);
320  pass.setNoFinalizeRemoval(NoFinalizeRemoval);
321
322  for (unsigned i=0, e = transforms.size(); i != e; ++i)
323    transforms[i](pass);
324
325  capturedDiags.reportDiagnostics(*Diags);
326
327  DiagClient->EndSourceFile();
328  errRec.FinishCapture();
329
330  // If we are migrating code that gets the '-fobjc-arc' flag, make sure
331  // to remove it so that we don't get errors from normal compilation.
332  origCI.getLangOpts()->ObjCAutoRefCount = false;
333
334  return capturedDiags.hasErrors() || testAct.hasReportedErrors();
335}
336
337//===----------------------------------------------------------------------===//
338// applyTransformations.
339//===----------------------------------------------------------------------===//
340
341static bool applyTransforms(CompilerInvocation &origCI,
342                            const FrontendInputFile &Input,
343                            DiagnosticConsumer *DiagClient,
344                            StringRef outputDir,
345                            bool emitPremigrationARCErrors,
346                            StringRef plistOut) {
347  if (!origCI.getLangOpts()->ObjC1)
348    return false;
349
350  LangOptions::GCMode OrigGCMode = origCI.getLangOpts()->getGC();
351
352  // Make sure checking is successful first.
353  CompilerInvocation CInvokForCheck(origCI);
354  if (arcmt::checkForManualIssues(CInvokForCheck, Input, DiagClient,
355                                  emitPremigrationARCErrors, plistOut))
356    return true;
357
358  CompilerInvocation CInvok(origCI);
359  CInvok.getFrontendOpts().Inputs.clear();
360  CInvok.getFrontendOpts().Inputs.push_back(Input);
361
362  MigrationProcess migration(CInvok, DiagClient, outputDir);
363  bool NoFinalizeRemoval = origCI.getMigratorOpts().NoFinalizeRemoval;
364
365  std::vector<TransformFn> transforms = arcmt::getAllTransformations(OrigGCMode,
366                                                                     NoFinalizeRemoval);
367  assert(!transforms.empty());
368
369  for (unsigned i=0, e = transforms.size(); i != e; ++i) {
370    bool err = migration.applyTransform(transforms[i]);
371    if (err) return true;
372  }
373
374  IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
375  IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
376      new DiagnosticsEngine(DiagID, &origCI.getDiagnosticOpts(),
377                            DiagClient, /*ShouldOwnClient=*/false));
378
379  if (outputDir.empty()) {
380    origCI.getLangOpts()->ObjCAutoRefCount = true;
381    return migration.getRemapper().overwriteOriginal(*Diags);
382  } else {
383    // If we are migrating code that gets the '-fobjc-arc' flag, make sure
384    // to remove it so that we don't get errors from normal compilation.
385    origCI.getLangOpts()->ObjCAutoRefCount = false;
386    return migration.getRemapper().flushToDisk(outputDir, *Diags);
387  }
388}
389
390bool arcmt::applyTransformations(CompilerInvocation &origCI,
391                                 const FrontendInputFile &Input,
392                                 DiagnosticConsumer *DiagClient) {
393  return applyTransforms(origCI, Input, DiagClient,
394                         StringRef(), false, StringRef());
395}
396
397bool arcmt::migrateWithTemporaryFiles(CompilerInvocation &origCI,
398                                      const FrontendInputFile &Input,
399                                      DiagnosticConsumer *DiagClient,
400                                      StringRef outputDir,
401                                      bool emitPremigrationARCErrors,
402                                      StringRef plistOut) {
403  assert(!outputDir.empty() && "Expected output directory path");
404  return applyTransforms(origCI, Input, DiagClient,
405                         outputDir, emitPremigrationARCErrors, plistOut);
406}
407
408bool arcmt::getFileRemappings(std::vector<std::pair<std::string,std::string> > &
409                                  remap,
410                              StringRef outputDir,
411                              DiagnosticConsumer *DiagClient) {
412  assert(!outputDir.empty());
413
414  IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
415  IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
416      new DiagnosticsEngine(DiagID, new DiagnosticOptions,
417                            DiagClient, /*ShouldOwnClient=*/false));
418
419  FileRemapper remapper;
420  bool err = remapper.initFromDisk(outputDir, *Diags,
421                                   /*ignoreIfFilesChanged=*/true);
422  if (err)
423    return true;
424
425  PreprocessorOptions PPOpts;
426  remapper.applyMappings(PPOpts);
427  remap = PPOpts.RemappedFiles;
428
429  return false;
430}
431
432bool arcmt::getFileRemappingsFromFileList(
433                        std::vector<std::pair<std::string,std::string> > &remap,
434                        ArrayRef<StringRef> remapFiles,
435                        DiagnosticConsumer *DiagClient) {
436  bool hasErrorOccurred = false;
437  llvm::StringMap<bool> Uniquer;
438
439  IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
440  IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
441      new DiagnosticsEngine(DiagID, new DiagnosticOptions,
442                            DiagClient, /*ShouldOwnClient=*/false));
443
444  for (ArrayRef<StringRef>::iterator
445         I = remapFiles.begin(), E = remapFiles.end(); I != E; ++I) {
446    StringRef file = *I;
447
448    FileRemapper remapper;
449    bool err = remapper.initFromFile(file, *Diags,
450                                     /*ignoreIfFilesChanged=*/true);
451    hasErrorOccurred = hasErrorOccurred || err;
452    if (err)
453      continue;
454
455    PreprocessorOptions PPOpts;
456    remapper.applyMappings(PPOpts);
457    for (PreprocessorOptions::remapped_file_iterator
458           RI = PPOpts.remapped_file_begin(), RE = PPOpts.remapped_file_end();
459           RI != RE; ++RI) {
460      bool &inserted = Uniquer[RI->first];
461      if (inserted)
462        continue;
463      inserted = true;
464      remap.push_back(*RI);
465    }
466  }
467
468  return hasErrorOccurred;
469}
470
471//===----------------------------------------------------------------------===//
472// CollectTransformActions.
473//===----------------------------------------------------------------------===//
474
475namespace {
476
477class ARCMTMacroTrackerPPCallbacks : public PPCallbacks {
478  std::vector<SourceLocation> &ARCMTMacroLocs;
479
480public:
481  ARCMTMacroTrackerPPCallbacks(std::vector<SourceLocation> &ARCMTMacroLocs)
482    : ARCMTMacroLocs(ARCMTMacroLocs) { }
483
484  virtual void MacroExpands(const Token &MacroNameTok, const MacroDirective *MD,
485                            SourceRange Range) {
486    if (MacroNameTok.getIdentifierInfo()->getName() == getARCMTMacroName())
487      ARCMTMacroLocs.push_back(MacroNameTok.getLocation());
488  }
489};
490
491class ARCMTMacroTrackerAction : public ASTFrontendAction {
492  std::vector<SourceLocation> &ARCMTMacroLocs;
493
494public:
495  ARCMTMacroTrackerAction(std::vector<SourceLocation> &ARCMTMacroLocs)
496    : ARCMTMacroLocs(ARCMTMacroLocs) { }
497
498  virtual ASTConsumer *CreateASTConsumer(CompilerInstance &CI,
499                                         StringRef InFile) {
500    CI.getPreprocessor().addPPCallbacks(
501                              new ARCMTMacroTrackerPPCallbacks(ARCMTMacroLocs));
502    return new ASTConsumer();
503  }
504};
505
506class RewritesApplicator : public TransformActions::RewriteReceiver {
507  Rewriter &rewriter;
508  MigrationProcess::RewriteListener *Listener;
509
510public:
511  RewritesApplicator(Rewriter &rewriter, ASTContext &ctx,
512                     MigrationProcess::RewriteListener *listener)
513    : rewriter(rewriter), Listener(listener) {
514    if (Listener)
515      Listener->start(ctx);
516  }
517  ~RewritesApplicator() {
518    if (Listener)
519      Listener->finish();
520  }
521
522  virtual void insert(SourceLocation loc, StringRef text) {
523    bool err = rewriter.InsertText(loc, text, /*InsertAfter=*/true,
524                                   /*indentNewLines=*/true);
525    if (!err && Listener)
526      Listener->insert(loc, text);
527  }
528
529  virtual void remove(CharSourceRange range) {
530    Rewriter::RewriteOptions removeOpts;
531    removeOpts.IncludeInsertsAtBeginOfRange = false;
532    removeOpts.IncludeInsertsAtEndOfRange = false;
533    removeOpts.RemoveLineIfEmpty = true;
534
535    bool err = rewriter.RemoveText(range, removeOpts);
536    if (!err && Listener)
537      Listener->remove(range);
538  }
539
540  virtual void increaseIndentation(CharSourceRange range,
541                                    SourceLocation parentIndent) {
542    rewriter.IncreaseIndentation(range, parentIndent);
543  }
544};
545
546} // end anonymous namespace.
547
548/// \brief Anchor for VTable.
549MigrationProcess::RewriteListener::~RewriteListener() { }
550
551MigrationProcess::MigrationProcess(const CompilerInvocation &CI,
552                                   DiagnosticConsumer *diagClient,
553                                   StringRef outputDir)
554  : OrigCI(CI), DiagClient(diagClient) {
555  if (!outputDir.empty()) {
556    IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
557    IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
558      new DiagnosticsEngine(DiagID, &CI.getDiagnosticOpts(),
559                            DiagClient, /*ShouldOwnClient=*/false));
560    Remapper.initFromDisk(outputDir, *Diags, /*ignoreIfFilesChanges=*/true);
561  }
562}
563
564bool MigrationProcess::applyTransform(TransformFn trans,
565                                      RewriteListener *listener) {
566  OwningPtr<CompilerInvocation> CInvok;
567  CInvok.reset(createInvocationForMigration(OrigCI));
568  CInvok->getDiagnosticOpts().IgnoreWarnings = true;
569
570  Remapper.applyMappings(CInvok->getPreprocessorOpts());
571
572  CapturedDiagList capturedDiags;
573  std::vector<SourceLocation> ARCMTMacroLocs;
574
575  assert(DiagClient);
576  IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
577  IntrusiveRefCntPtr<DiagnosticsEngine> Diags(
578      new DiagnosticsEngine(DiagID, new DiagnosticOptions,
579                            DiagClient, /*ShouldOwnClient=*/false));
580
581  // Filter of all diagnostics.
582  CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags);
583  Diags->setClient(&errRec, /*ShouldOwnClient=*/false);
584
585  OwningPtr<ARCMTMacroTrackerAction> ASTAction;
586  ASTAction.reset(new ARCMTMacroTrackerAction(ARCMTMacroLocs));
587
588  OwningPtr<ASTUnit> Unit(
589      ASTUnit::LoadFromCompilerInvocationAction(CInvok.take(), Diags,
590                                                ASTAction.get()));
591  if (!Unit) {
592    errRec.FinishCapture();
593    return true;
594  }
595  Unit->setOwnsRemappedFileBuffers(false); // FileRemapper manages that.
596
597  // Don't filter diagnostics anymore.
598  Diags->setClient(DiagClient, /*ShouldOwnClient=*/false);
599
600  ASTContext &Ctx = Unit->getASTContext();
601
602  if (Diags->hasFatalErrorOccurred()) {
603    Diags->Reset();
604    DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor());
605    capturedDiags.reportDiagnostics(*Diags);
606    DiagClient->EndSourceFile();
607    errRec.FinishCapture();
608    return true;
609  }
610
611  // After parsing of source files ended, we want to reuse the
612  // diagnostics objects to emit further diagnostics.
613  // We call BeginSourceFile because DiagnosticConsumer requires that
614  // diagnostics with source range information are emitted only in between
615  // BeginSourceFile() and EndSourceFile().
616  DiagClient->BeginSourceFile(Ctx.getLangOpts(), &Unit->getPreprocessor());
617
618  Rewriter rewriter(Ctx.getSourceManager(), Ctx.getLangOpts());
619  TransformActions TA(*Diags, capturedDiags, Ctx, Unit->getPreprocessor());
620  MigrationPass pass(Ctx, OrigCI.getLangOpts()->getGC(),
621                     Unit->getSema(), TA, capturedDiags, ARCMTMacroLocs);
622
623  trans(pass);
624
625  {
626    RewritesApplicator applicator(rewriter, Ctx, listener);
627    TA.applyRewrites(applicator);
628  }
629
630  DiagClient->EndSourceFile();
631  errRec.FinishCapture();
632
633  if (DiagClient->getNumErrors())
634    return true;
635
636  for (Rewriter::buffer_iterator
637        I = rewriter.buffer_begin(), E = rewriter.buffer_end(); I != E; ++I) {
638    FileID FID = I->first;
639    RewriteBuffer &buf = I->second;
640    const FileEntry *file = Ctx.getSourceManager().getFileEntryForID(FID);
641    assert(file);
642    std::string newFname = file->getName();
643    newFname += "-trans";
644    SmallString<512> newText;
645    llvm::raw_svector_ostream vecOS(newText);
646    buf.write(vecOS);
647    vecOS.flush();
648    llvm::MemoryBuffer *memBuf = llvm::MemoryBuffer::getMemBufferCopy(
649                   StringRef(newText.data(), newText.size()), newFname);
650    SmallString<64> filePath(file->getName());
651    Unit->getFileManager().FixupRelativePath(filePath);
652    Remapper.remap(filePath.str(), memBuf);
653  }
654
655  return false;
656}
657