22 #include "llvm/ADT/Triple.h"
23 #include "llvm/Support/MemoryBuffer.h"
25 using namespace clang;
26 using namespace arcmt;
35 while (I != List.end()) {
38 std::find(IDs.begin(), IDs.end(), I->getID()) != IDs.end()) &&
40 (diagLoc == range.
getEnd() ||
48 I = List.erase(eraseS, I);
63 ListTy::const_iterator
I = List.begin();
64 while (I != List.end()) {
67 std::find(IDs.begin(), IDs.end(), I->getID()) != IDs.end()) &&
69 (diagLoc == range.
getEnd() ||
81 for (ListTy::const_iterator
I = List.begin(),
E = List.end();
I !=
E; ++
I)
86 for (ListTy::const_iterator
I = List.begin(),
E = List.end();
I !=
E; ++
I)
99 bool HasBegunSourceFile;
104 : Diags(diags), DiagClient(client), CapturedDiags(capturedDiags),
105 HasBegunSourceFile(
false) { }
112 if (!HasBegunSourceFile) {
113 DiagClient.BeginSourceFile(Opts, PP);
114 HasBegunSourceFile =
true;
118 void FinishCapture() {
122 if (HasBegunSourceFile) {
123 DiagClient.EndSourceFile();
124 HasBegunSourceFile =
false;
128 ~CaptureDiagnosticConsumer()
override {
129 assert(!HasBegunSourceFile &&
"FinishCapture not called!");
142 Diags.setLastDiagnosticIgnored();
155 return triple.getOSMajorVersion() >= 5;
157 if (triple.isWatchOS())
160 if (triple.getOS() == llvm::Triple::Darwin)
161 return triple.getOSMajorVersion() >= 11;
163 if (triple.getOS() == llvm::Triple::MacOSX) {
164 unsigned Major, Minor, Micro;
165 triple.getOSVersion(Major, Minor, Micro);
166 return Major > 10 || (Major == 10 && Minor >= 7);
175 std::unique_ptr<CompilerInvocation> CInvok;
188 if (!OriginalFile.empty())
193 CInvok->getPreprocessorOpts().ImplicitPTHInclude.clear();
196 CInvok->getPreprocessorOpts().addMacroDef(define);
197 CInvok->getLangOpts()->ObjCAutoRefCount =
true;
199 CInvok->getDiagnosticOpts().ErrorLimit = 0;
200 CInvok->getDiagnosticOpts().PedanticErrors = 0;
203 std::vector<std::string> WarnOpts;
205 I = CInvok->getDiagnosticOpts().Warnings.begin(),
206 E = CInvok->getDiagnosticOpts().Warnings.end();
I !=
E; ++
I) {
207 if (!StringRef(*I).startswith(
"error"))
208 WarnOpts.push_back(*
I);
210 WarnOpts.push_back(
"error=arc-unsafe-retained-assign");
211 CInvok->getDiagnosticOpts().Warnings = std::move(WarnOpts);
213 CInvok->getLangOpts()->ObjCWeakRuntime =
HasARCRuntime(origCI);
214 CInvok->getLangOpts()->ObjCWeak = CInvok->getLangOpts()->ObjCWeakRuntime;
216 return CInvok.release();
240 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
242 StringRef plistOut) {
252 assert(!transforms.empty());
254 std::unique_ptr<CompilerInvocation> CInvok;
257 CInvok->getFrontendOpts().Inputs.clear();
258 CInvok->getFrontendOpts().Inputs.push_back(Input);
269 CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags);
270 Diags->setClient(&errRec,
false);
273 CInvok.release(), PCHContainerOps, Diags));
275 errRec.FinishCapture();
280 Diags->setClient(DiagClient,
false);
284 if (Diags->hasFatalErrorOccurred()) {
289 errRec.FinishCapture();
293 if (emitPremigrationARCErrors)
295 Unit->getPreprocessor());
296 if (!plistOut.empty()) {
299 I = capturedDiags.
begin(),
E = capturedDiags.
end();
I !=
E; ++
I)
300 arcDiags.push_back(*
I);
314 std::vector<SourceLocation> ARCMTMacroLocs;
316 TransformActions testAct(*Diags, capturedDiags, Ctx, Unit->getPreprocessor());
317 MigrationPass pass(Ctx, OrigGCMode, Unit->getSema(), testAct, capturedDiags,
320 if (!NoNSAllocReallocError)
324 for (
unsigned i=0, e = transforms.size(); i != e; ++i)
327 capturedDiags.reportDiagnostics(*Diags);
330 errRec.FinishCapture();
332 return capturedDiags.hasErrors() || testAct.hasReportedErrors();
341 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
343 bool emitPremigrationARCErrors, StringRef plistOut) {
352 DiagClient, emitPremigrationARCErrors,
365 assert(!transforms.empty());
367 for (
unsigned i=0, e = transforms.size(); i != e; ++i) {
369 if (err)
return true;
377 if (outputDir.empty()) {
387 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
390 StringRef(),
false, StringRef());
395 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
397 bool emitPremigrationARCErrors, StringRef plistOut) {
398 assert(!outputDir.empty() &&
"Expected output directory path");
399 return applyTransforms(origCI, Input, PCHContainerOps, DiagClient, outputDir,
400 emitPremigrationARCErrors, plistOut);
407 assert(!outputDir.empty());
434 class ARCMTMacroTrackerPPCallbacks :
public PPCallbacks {
435 std::vector<SourceLocation> &ARCMTMacroLocs;
438 ARCMTMacroTrackerPPCallbacks(std::vector<SourceLocation> &ARCMTMacroLocs)
439 : ARCMTMacroLocs(ARCMTMacroLocs) { }
444 ARCMTMacroLocs.push_back(MacroNameTok.
getLocation());
449 std::vector<SourceLocation> &ARCMTMacroLocs;
452 ARCMTMacroTrackerAction(std::vector<SourceLocation> &ARCMTMacroLocs)
453 : ARCMTMacroLocs(ARCMTMacroLocs) { }
456 StringRef InFile)
override {
458 llvm::make_unique<ARCMTMacroTrackerPPCallbacks>(ARCMTMacroLocs));
459 return llvm::make_unique<ASTConsumer>();
470 : rewriter(rewriter), Listener(listener) {
472 Listener->start(ctx);
474 ~RewritesApplicator()
override {
480 bool err = rewriter.
InsertText(loc, text,
true,
482 if (!err && Listener)
483 Listener->insert(loc, text);
492 bool err = rewriter.
RemoveText(range, removeOpts);
493 if (!err && Listener)
494 Listener->remove(range);
510 std::shared_ptr<PCHContainerOperations> PCHContainerOps,
512 : OrigCI(CI), PCHContainerOps(std::move(PCHContainerOps)),
514 if (!outputDir.empty()) {
525 std::unique_ptr<CompilerInvocation> CInvok;
528 CInvok->getDiagnosticOpts().IgnoreWarnings =
true;
533 std::vector<SourceLocation> ARCMTMacroLocs;
542 CaptureDiagnosticConsumer errRec(*Diags, *DiagClient, capturedDiags);
543 Diags->setClient(&errRec,
false);
545 std::unique_ptr<ARCMTMacroTrackerAction> ASTAction;
546 ASTAction.reset(
new ARCMTMacroTrackerAction(ARCMTMacroLocs));
549 CInvok.release(), PCHContainerOps, Diags, ASTAction.get()));
551 errRec.FinishCapture();
554 Unit->setOwnsRemappedFileBuffers(
false);
559 Diags->setClient(DiagClient,
false);
563 if (Diags->hasFatalErrorOccurred()) {
568 errRec.FinishCapture();
582 Unit->getSema(), TA, capturedDiags, ARCMTMacroLocs);
587 RewritesApplicator applicator(rewriter, Ctx, listener);
588 TA.applyRewrites(applicator);
592 errRec.FinishCapture();
603 std::string newFname = file->
getName();
604 newFname +=
"-trans";
606 llvm::raw_svector_ostream vecOS(newText);
608 std::unique_ptr<llvm::MemoryBuffer> memBuf(
609 llvm::MemoryBuffer::getMemBufferCopy(
610 StringRef(newText.data(), newText.size()), newFname));
612 Unit->getFileManager().FixupRelativePath(filePath);
613 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.
static ASTUnit * LoadFromCompilerInvocationAction(CompilerInvocation *CI, std::shared_ptr< PCHContainerOperations > PCHContainerOps, IntrusiveRefCntPtr< DiagnosticsEngine > Diags, FrontendAction *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...
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.
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