diff options
-rw-r--r-- | include/llvm/CompilerDriver/BuiltinOptions.h | 1 | ||||
-rw-r--r-- | lib/CompilerDriver/BuiltinOptions.cpp | 2 | ||||
-rw-r--r-- | lib/CompilerDriver/Main.cpp | 9 |
3 files changed, 12 insertions, 0 deletions
diff --git a/include/llvm/CompilerDriver/BuiltinOptions.h b/include/llvm/CompilerDriver/BuiltinOptions.h index 492dffd307..fe44c30a7a 100644 --- a/include/llvm/CompilerDriver/BuiltinOptions.h +++ b/include/llvm/CompilerDriver/BuiltinOptions.h @@ -22,6 +22,7 @@ namespace SaveTempsEnum { enum Values { Cwd, Obj, Unset }; } extern llvm::cl::list<std::string> InputFilenames; extern llvm::cl::opt<std::string> OutputFilename; +extern llvm::cl::opt<std::string> TempDirname; extern llvm::cl::list<std::string> Languages; extern llvm::cl::opt<bool> DryRun; extern llvm::cl::opt<bool> VerboseMode; diff --git a/lib/CompilerDriver/BuiltinOptions.cpp b/lib/CompilerDriver/BuiltinOptions.cpp index a3364e8a72..c4660e17a5 100644 --- a/lib/CompilerDriver/BuiltinOptions.cpp +++ b/lib/CompilerDriver/BuiltinOptions.cpp @@ -25,6 +25,8 @@ cl::list<std::string> InputFilenames(cl::Positional, cl::desc("<input file>"), cl::ZeroOrMore); cl::opt<std::string> OutputFilename("o", cl::desc("Output file name"), cl::value_desc("file"), cl::Prefix); +cl::opt<std::string> TempDirname("t", cl::desc("Temp dir name"), + cl::value_desc("dir"), cl::Prefix); cl::list<std::string> Languages("x", cl::desc("Specify the language of the following input files"), cl::ZeroOrMore); diff --git a/lib/CompilerDriver/Main.cpp b/lib/CompilerDriver/Main.cpp index 7d1a3d8fbc..57d2937b33 100644 --- a/lib/CompilerDriver/Main.cpp +++ b/lib/CompilerDriver/Main.cpp @@ -31,6 +31,15 @@ namespace { sys::Path getTempDir() { sys::Path tempDir; + if (! TempDirname.empty() { + tempDir = TempDirname; + if (!tempDir.exists()) { + std::string ErrMsg; + if (tempDir.createDirectoryOnDisk(true, &ErrMsg)) + throw std::runtime_error(ErrMsg); + } + } + // GCC 4.5-style -save-temps handling. if (SaveTemps == SaveTempsEnum::Unset) { tempDir = sys::Path::GetTemporaryDirectory(); |