22 #include "llvm/ADT/Triple.h"
23 #include "llvm/Support/MemoryBuffer.h"
24 using namespace clang;
25 using namespace arcmt;
34 while (I != List.end()) {
37 std::find(IDs.begin(), IDs.end(), I->getID()) != IDs.end()) &&
39 (diagLoc == range.
getEnd() ||
47 I = List.erase(eraseS, I);
62 ListTy::const_iterator
I = List.begin();
63 while (I != List.end()) {
66 std::find(IDs.begin(), IDs.end(), I->getID()) != IDs.end()) &&
68 (diagLoc == range.
getEnd() ||
80 for (ListTy::const_iterator
I = List.begin(),
E = List.end();
I !=
E; ++
I)
85 for (ListTy::const_iterator
I = List.begin(),
E = List.end();
I !=
E; ++
I)
98 bool HasBegunSourceFile;
103 : Diags(diags), DiagClient(client), CapturedDiags(capturedDiags),
104 HasBegunSourceFile(
false) { }
111 if (!HasBegunSourceFile) {
112 DiagClient.BeginSourceFile(Opts, PP);
113 HasBegunSourceFile =
true;
117 void FinishCapture() {
121 if (HasBegunSourceFile) {
122 DiagClient.EndSourceFile();
123 HasBegunSourceFile =
false;
127 ~CaptureDiagnosticConsumer()
override {
128 assert(!HasBegunSourceFile &&
"FinishCapture not called!");
141 Diags.setLastDiagnosticIgnored();
154 return triple.getOSMajorVersion() >= 5;
156 if (triple.isWatchOS())
159 if (triple.getOS() == llvm::Triple::Darwin)
160 return triple.getOSMajorVersion() >= 11;
162 if (triple.getOS() == llvm::Triple::MacOSX) {
163 unsigned Major, Minor, Micro;
164 triple.getOSVersion(Major, Minor, Micro);
165 return Major > 10 || (Major == 10 && Minor >= 7);
174 std::unique_ptr<CompilerInvocation> CInvok;
187 if (!OriginalFile.empty())
192 CInvok->getPreprocessorOpts().ImplicitPTHInclude.clear();
195 CInvok->getPreprocessorOpts().addMacroDef(define);
196 CInvok->getLangOpts()->ObjCAutoRefCount =
true;
198 CInvok->getDiagnosticOpts().ErrorLimit = 0;
199 CInvok->getDiagnosticOpts().PedanticErrors = 0;
202 std::vector<std::string> WarnOpts;
204 I = CInvok->getDiagnosticOpts().Warnings.begin(),
205 E = CInvok->getDiagnosticOpts().Warnings.end();
I !=
E; ++
I) {
206 if (!StringRef(*I).startswith(
"error"))
207 WarnOpts.push_back(*
I);
209 WarnOpts.push_back(
"error=arc-unsafe-retained-assign");
210 CInvok->getDiagnosticOpts().Warnings = std::move(WarnOpts);
212 CInvok->getLangOpts()->ObjCWeakRuntime =
HasARCRuntime(origCI);
213 CInvok->getLangOpts()->ObjCWeak = CInvok->getLangOpts()->ObjCWeakRuntime;
215 return CInvok.release();
239 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
241 StringRef plistOut) {
251 assert(!transforms.empty());
253 std::unique_ptr<CompilerInvocation> CInvok;
256 CInvok->getFrontendOpts().Inputs.clear();
257 CInvok->getFrontendOpts().Inputs.push_back(Input);
268 CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags);
269 Diags->setClient(&errRec,
false);
272 CInvok.release(), PCHContainerOps, Diags));
274 errRec.FinishCapture();
279 Diags->setClient(DiagClient,
false);
283 if (Diags->hasFatalErrorOccurred()) {
288 errRec.FinishCapture();
292 if (emitPremigrationARCErrors)
294 Unit->getPreprocessor());
295 if (!plistOut.empty()) {
298 I = capturedDiags.
begin(),
E = capturedDiags.
end();
I !=
E; ++
I)
299 arcDiags.push_back(*
I);
313 std::vector<SourceLocation> ARCMTMacroLocs;
315 TransformActions testAct(*Diags, capturedDiags, Ctx, Unit->getPreprocessor());
316 MigrationPass pass(Ctx, OrigGCMode, Unit->getSema(), testAct, capturedDiags,
319 if (!NoNSAllocReallocError)
323 for (
unsigned i=0, e = transforms.size(); i != e; ++i)
326 capturedDiags.reportDiagnostics(*Diags);
329 errRec.FinishCapture();
331 return capturedDiags.hasErrors() || testAct.hasReportedErrors();
340 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
342 bool emitPremigrationARCErrors, StringRef plistOut) {
351 DiagClient, emitPremigrationARCErrors,
364 assert(!transforms.empty());
366 for (
unsigned i=0, e = transforms.size(); i != e; ++i) {
368 if (err)
return true;
376 if (outputDir.empty()) {
386 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
389 StringRef(),
false, StringRef());
394 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
396 bool emitPremigrationARCErrors, StringRef plistOut) {
397 assert(!outputDir.empty() &&
"Expected output directory path");
398 return applyTransforms(origCI, Input, PCHContainerOps, DiagClient, outputDir,
399 emitPremigrationARCErrors, plistOut);
406 assert(!outputDir.empty());
433 class ARCMTMacroTrackerPPCallbacks :
public PPCallbacks {
434 std::vector<SourceLocation> &ARCMTMacroLocs;
437 ARCMTMacroTrackerPPCallbacks(std::vector<SourceLocation> &ARCMTMacroLocs)
438 : ARCMTMacroLocs(ARCMTMacroLocs) { }
443 ARCMTMacroLocs.push_back(MacroNameTok.
getLocation());
448 std::vector<SourceLocation> &ARCMTMacroLocs;
451 ARCMTMacroTrackerAction(std::vector<SourceLocation> &ARCMTMacroLocs)
452 : ARCMTMacroLocs(ARCMTMacroLocs) { }
455 StringRef InFile)
override {
457 llvm::make_unique<ARCMTMacroTrackerPPCallbacks>(ARCMTMacroLocs));
458 return llvm::make_unique<ASTConsumer>();
469 : rewriter(rewriter), Listener(listener) {
471 Listener->start(ctx);
473 ~RewritesApplicator()
override {
479 bool err = rewriter.
InsertText(loc, text,
true,
481 if (!err && Listener)
482 Listener->insert(loc, text);
491 bool err = rewriter.
RemoveText(range, removeOpts);
492 if (!err && Listener)
493 Listener->remove(range);
509 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
511 : OrigCI(CI), PCHContainerOps(PCHContainerOps), DiagClient(diagClient),
513 if (!outputDir.empty()) {
524 std::unique_ptr<CompilerInvocation> CInvok;
527 CInvok->getDiagnosticOpts().IgnoreWarnings =
true;
532 std::vector<SourceLocation> ARCMTMacroLocs;
541 CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags);
542 Diags->setClient(&errRec,
false);
544 std::unique_ptr<ARCMTMacroTrackerAction> ASTAction;
545 ASTAction.reset(
new ARCMTMacroTrackerAction(ARCMTMacroLocs));
548 CInvok.release(), PCHContainerOps, Diags, ASTAction.get()));
550 errRec.FinishCapture();
553 Unit->setOwnsRemappedFileBuffers(
false);
558 Diags->setClient(DiagClient,
false);
562 if (Diags->hasFatalErrorOccurred()) {
567 errRec.FinishCapture();
581 Unit->getSema(), TA, capturedDiags, ARCMTMacroLocs);
586 RewritesApplicator applicator(rewriter, Ctx, listener);
587 TA.applyRewrites(applicator);
591 errRec.FinishCapture();
602 std::string newFname = file->
getName();
603 newFname +=
"-trans";
605 llvm::raw_svector_ostream vecOS(newText);
607 std::unique_ptr<llvm::MemoryBuffer> memBuf(
608 llvm::MemoryBuffer::getMemBufferCopy(
609 StringRef(newText.data(), newText.size()), newFname));
611 Unit->getFileManager().FixupRelativePath(filePath);
612 Remapper.
remap(filePath.str(), std::move(memBuf));
SourceManager & getSourceManager() const
SourceLocation getEnd() const
void(* TransformFn)(MigrationPass &pass)
unsigned NoFinalizeRemoval
void reportDiagnostics(DiagnosticsEngine &diags) const
bool IncreaseIndentation(CharSourceRange range, SourceLocation parentIndent)
Increase indentation for the lines between the given source range.
Implements support for file system lookup, file system caching, and directory search management...
TargetOptions & getTargetOpts()
ListTy::const_iterator iterator
static StringRef getARCMTMacroName()
Represents a diagnostic in a form that can be retained until its corresponding source manager is dest...
bool applyTransformations(CompilerInvocation &origCI, const FrontendInputFile &Input, std::shared_ptr< PCHContainerOperations > PCHContainerOps, DiagnosticConsumer *DiagClient)
Works similar to checkForManualIssues but instead of checking, it applies automatic modifications to ...
std::vector< std::string > Includes
A description of the current definition of a macro.
StringRef getOriginalSourceFile()
Retrieve the name of the original source file name for the primary module file.
DiagnosticBuilder Report(SourceLocation Loc, unsigned DiagID)
Issue the message to the client.
PreprocessorOptions - This class is used for passing the various options used in preprocessor initial...
bool RemoveText(SourceLocation Start, unsigned Length, RewriteOptions opts=RewriteOptions())
RemoveText - Remove the specified text region.
virtual void EndSourceFile()
Callback to inform the diagnostic client that processing of a source file has ended.
Abstract interface, implemented by clients of the front-end, which formats and prints fully processed...
RewriteBuffer - As code is rewritten, SourceBuffer's from the original input with modifications get a...
This interface provides a way to observe the actions of the preprocessor as it does its thing...
void setNoFinalizeRemoval(bool val)
virtual ~RewriteListener()
Anchor for VTable.
Holds long-lived AST nodes (such as types and decls) that can be referred to throughout the semantic ...
const LangOptions & getLangOpts() const
Token - This structure provides full information about a lexed token.
Keeps track of the various options that can be enabled, which controls the dialect of C or C++ that i...
bool clearDiagnostic(ArrayRef< unsigned > IDs, SourceRange range)
const LangOptions & getLangOpts() const
const SourceLocation & getLocation() const
void remap(StringRef filePath, std::unique_ptr< llvm::MemoryBuffer > memBuf)
void BeginSourceFile(const LangOptions &LO, const Preprocessor *PP) override
Callback to inform the diagnostic client that processing of a source file is beginning.
Concrete class used by the front-end to report problems and issues.
static void emitPremigrationErrors(const CapturedDiagList &arcDiags, DiagnosticOptions *diagOpts, Preprocessor &PP)
Present this diagnostic as an error.
bool hasDiagnostic(ArrayRef< unsigned > IDs, SourceRange range) const
This abstract interface provides operations for unwrapping containers for serialized ASTs (precompile...
detail::InMemoryDirectory::const_iterator I
Preprocessor & getPreprocessor() const
Return the current preprocessor.
buffer_iterator buffer_end()
FrontendOptions & getFrontendOpts()
MigratorOptions & getMigratorOpts()
const FileEntry * getFileEntryForID(FileID FID) const
Returns the FileEntry record for the provided FileID.
static ASTUnit * LoadFromCompilerInvocationAction(CompilerInvocation *CI, std::shared_ptr< PCHContainerOperations > PCHContainerOps, IntrusiveRefCntPtr< DiagnosticsEngine > Diags, ASTFrontendAction *Action=nullptr, ASTUnit *Unit=nullptr, bool Persistent=true, StringRef ResourceFilesPath=StringRef(), bool OnlyLocalDecls=false, bool CaptureDiagnostics=false, unsigned PrecompilePreambleAfterNParses=0, bool CacheCodeCompletionResults=false, bool IncludeBriefCommentsInCodeCompletion=false, bool UserFilesAreVolatile=false, std::unique_ptr< ASTUnit > *ErrAST=nullptr)
Create an ASTUnit from a source file, via a CompilerInvocation object, by invoking the optionally pro...
StringRef getName() const
Return the actual identifier string.
Represents a character-granular source range.
MacroArgs - An instance of this class captures information about the formal arguments specified to a ...
Defines the clang::Preprocessor interface.
void writeARCDiagsToPlist(const std::string &outPath, ArrayRef< StoredDiagnostic > diags, SourceManager &SM, const LangOptions &LangOpts)
SourceLocation getLocation() const
Return a source location identifier for the specified offset in the current file. ...
static bool applyTransforms(CompilerInvocation &origCI, const FrontendInputFile &Input, std::shared_ptr< PCHContainerOperations > PCHContainerOps, DiagnosticConsumer *DiagClient, StringRef outputDir, bool emitPremigrationARCErrors, StringRef plistOut)
bool checkForManualIssues(CompilerInvocation &CI, const FrontendInputFile &Input, std::shared_ptr< PCHContainerOperations > PCHContainerOps, DiagnosticConsumer *DiagClient, bool emitPremigrationARCErrors=false, StringRef plistOut=StringRef())
Creates an AST with the provided CompilerInvocation but with these changes: -if a PCH/PTH is set...
static bool HasARCRuntime(CompilerInvocation &origCI)
void EndSourceFile() override
Callback to inform the diagnostic client that processing of a source file has ended.
void applyMappings(PreprocessorOptions &PPOpts) const
MigrationProcess(const CompilerInvocation &CI, std::shared_ptr< PCHContainerOperations > PCHContainerOps, DiagnosticConsumer *diagClient, StringRef outputDir=StringRef())
unsigned NoNSAllocReallocError
static CompilerInvocation * createInvocationForMigration(CompilerInvocation &origCI, const PCHContainerReader &PCHContainerRdr)
bool overwriteOriginal(DiagnosticsEngine &Diag, StringRef outputDir=StringRef())
const char * getName() const
CompilerInstance - Helper class for managing a single instance of the Clang compiler.
Encodes a location in the source.
const TemplateArgument * iterator
std::string ImplicitPCHInclude
The implicit PCH included at the start of the translation unit, or empty.
bool flushToDisk(StringRef outputDir, DiagnosticsEngine &Diag)
bool isValid() const
Return true if this is a valid SourceLocation object.
Options for controlling the compiler diagnostics engine.
std::vector< FrontendInputFile > Inputs
The input files and their types.
Cached information about one file (either on disk or in the virtual file system). ...
bool InsertText(SourceLocation Loc, StringRef Str, bool InsertAfter=true, bool indentNewLines=false)
InsertText - Insert the specified string at the specified location in the original buffer...
Abstract base class to use for AST consumer-based frontend actions.
bool RemoveLineIfEmpty
If true and removing some text leaves a blank line also remove the empty line (false by default)...
SourceLocation getBegin() const
bool migrateWithTemporaryFiles(CompilerInvocation &origCI, const FrontendInputFile &Input, std::shared_ptr< PCHContainerOperations > PCHContainerOps, DiagnosticConsumer *DiagClient, StringRef outputDir, bool emitPremigrationARCErrors, StringRef plistOut)
Applies automatic modifications and produces temporary files and metadata into the outputDir path...
An opaque identifier used by SourceManager which refers to a source file (MemoryBuffer) along with it...
A diagnostic client that ignores all diagnostics.
std::vector< TransformFn > getAllTransformations(LangOptions::GCMode OrigGCMode, bool NoFinalizeRemoval)
Used for handling and querying diagnostic IDs.
Helper class for holding the data necessary to invoke the compiler.
DiagnosticOptions & getDiagnosticOpts() const
detail::InMemoryDirectory::const_iterator E
raw_ostream & write(raw_ostream &Stream) const
Write to Stream the result of applying all changes to the original buffer.
bool IncludeInsertsAtBeginOfRange
Given a source range, true to include previous inserts at the beginning of the range as part of the r...
std::map< FileID, RewriteBuffer >::iterator buffer_iterator
buffer_iterator buffer_begin()
SourceManager & getSourceManager()
bool initFromDisk(StringRef outputDir, DiagnosticsEngine &Diag, bool ignoreIfFilesChanged)
Rewriter - This is the main interface to the rewrite buffers.
static bool isARCDiagnostic(unsigned DiagID)
Return true if a given diagnostic falls into an ARC diagnostic category.
bool isBeforeInTranslationUnitThan(SourceLocation Loc) const
Determines the order of 2 source locations in the translation unit.
Level
The level of the diagnostic, after it has been through mapping.
FileSystemOptions & getFileSystemOpts()
std::vector< std::pair< std::string, std::string > > RemappedFiles
The set of file remappings, which take existing files on the system (the first part of each pair) and...
A little helper class (which is basically a smart pointer that forwards info from DiagnosticsEngine) ...
FileRemapper & getRemapper()
A SourceLocation and its associated SourceManager.
Defines the clang::FrontendAction interface and various convenience abstract classes (clang::ASTFront...
unsigned getNumErrors() const
bool applyTransform(TransformFn trans, RewriteListener *listener=nullptr)
A trivial tuple used to represent a source range.
LangOptions * getLangOpts()
virtual void BeginSourceFile(const LangOptions &LangOpts, const Preprocessor *PP=nullptr)
Callback to inform the diagnostic client that processing of a source file is beginning.
std::string Triple
If given, the name of the target triple to compile for.
bool getFileRemappings(std::vector< std::pair< std::string, std::string > > &remap, StringRef outputDir, DiagnosticConsumer *DiagClient)
Get the set of file remappings from the outputDir path that migrateWithTemporaryFiles produced...
void addPPCallbacks(std::unique_ptr< PPCallbacks > C)
Engages in a tight little dance with the lexer to efficiently preprocess tokens.
bool IncludeInsertsAtEndOfRange
Given a source range, true to include previous inserts at the end of the range as part of the range i...
IdentifierInfo * getIdentifierInfo() const