diff --git a/clang/include/clang/Driver/Job.h b/clang/include/clang/Driver/Job.h --- a/clang/include/clang/Driver/Job.h +++ b/clang/include/clang/Driver/Job.h @@ -10,6 +10,7 @@ #define LLVM_CLANG_DRIVER_JOB_H #include "clang/Basic/LLVM.h" +#include "clang/Driver/InputInfo.h" #include "llvm/ADT/ArrayRef.h" #include "llvm/ADT/Optional.h" #include "llvm/ADT/SmallVector.h" @@ -119,8 +120,8 @@ /// argument, which will be the executable). llvm::opt::ArgStringList Arguments; - /// The list of program arguments which are inputs. - llvm::opt::ArgStringList InputFilenames; + /// The list of program inputs. + std::vector InputInfoList; /// The list of program arguments which are outputs. May be empty. std::vector OutputFilenames; @@ -207,9 +208,7 @@ const llvm::opt::ArgStringList &getArguments() const { return Arguments; } - const llvm::opt::ArgStringList &getInputFilenames() const { - return InputFilenames; - } + const std::vector &getInputInfos() const { return InputInfoList; } const std::vector &getOutputFilenames() const { return OutputFilenames; diff --git a/clang/lib/Driver/Job.cpp b/clang/lib/Driver/Job.cpp --- a/clang/lib/Driver/Job.cpp +++ b/clang/lib/Driver/Job.cpp @@ -43,7 +43,7 @@ Executable(Executable), Arguments(Arguments) { for (const auto &II : Inputs) if (II.isFilename()) - InputFilenames.push_back(II.getFilename()); + InputInfoList.push_back(II); for (const auto &II : Outputs) if (II.isFilename()) OutputFilenames.push_back(II.getFilename()); @@ -237,9 +237,10 @@ } } - auto Found = llvm::find_if(InputFilenames, - [&Arg](StringRef IF) { return IF == Arg; }); - if (Found != InputFilenames.end() && + auto Found = llvm::find_if(InputInfoList, [&Arg](const InputInfo &II) { + return II.getFilename() == Arg; + }); + if (Found != InputInfoList.end() && (i == 0 || StringRef(Args[i - 1]) != "-main-file-name")) { // Replace the input file name with the crashinfo's file name. OS << ' '; @@ -302,8 +303,8 @@ void Command::PrintFileNames() const { if (PrintInputFilenames) { - for (const char *Arg : InputFilenames) - llvm::outs() << llvm::sys::path::filename(Arg) << "\n"; + for (const auto &Arg : InputInfoList) + llvm::outs() << llvm::sys::path::filename(Arg.getFilename()) << "\n"; llvm::outs().flush(); } } diff --git a/clang/unittests/Driver/ToolChainTest.cpp b/clang/unittests/Driver/ToolChainTest.cpp --- a/clang/unittests/Driver/ToolChainTest.cpp +++ b/clang/unittests/Driver/ToolChainTest.cpp @@ -321,13 +321,13 @@ const JobList &Jobs = CC->getJobs(); const auto &CmdCompile = Jobs.getJobs().front(); - const auto &InFile = CmdCompile->getInputFilenames().front(); + const auto &InFile = CmdCompile->getInputInfos().front().getFilename(); EXPECT_STREQ(InFile, "foo.cpp"); auto ObjFile = CmdCompile->getOutputFilenames().front(); EXPECT_TRUE(StringRef(ObjFile).endswith(".o")); const auto &CmdLink = Jobs.getJobs().back(); - const auto LinkInFile = CmdLink->getInputFilenames().front(); + const auto LinkInFile = CmdLink->getInputInfos().front().getFilename(); EXPECT_EQ(ObjFile, LinkInFile); auto ExeFile = CmdLink->getOutputFilenames().front(); EXPECT_EQ("a.out", ExeFile);