diff --git a/tools/driver/driver.cpp b/tools/driver/driver.cpp index 02efeb0c288580c020fc00aa1630d40f7934d22b..2533401082a7ecb8640c84ef99acfcc0239ac2da 100644 --- a/tools/driver/driver.cpp +++ b/tools/driver/driver.cpp @@ -41,7 +41,6 @@ #include "llvm/Support/Program.h" #include "llvm/Support/Regex.h" #include "llvm/Support/Signals.h" -#include "llvm/Support/StringSaver.h" #include "llvm/Support/TargetRegistry.h" #include "llvm/Support/TargetSelect.h" #include "llvm/Support/Timer.h" @@ -282,6 +281,18 @@ static void ParseProgName(SmallVectorImpl<const char *> &ArgVector, } } +namespace { + class StringSetSaver : public llvm::cl::StringSaver { + public: + StringSetSaver(std::set<std::string> &Storage) : Storage(Storage) {} + const char *SaveString(const char *Str) override { + return GetStableCStr(Storage, Str); + } + private: + std::set<std::string> &Storage; + }; +} + static void SetBackdoorDriverOutputsFromEnvVars(Driver &TheDriver) { // Handle CC_PRINT_OPTIONS and CC_PRINT_OPTIONS_FILE. TheDriver.CCPrintOptions = !!::getenv("CC_PRINT_OPTIONS"); @@ -372,7 +383,7 @@ int main(int argc_, const char **argv_) { } std::set<std::string> SavedStrings; - llvm::StringSaver Saver; + StringSetSaver Saver(SavedStrings); llvm::cl::ExpandResponseFiles(Saver, llvm::cl::TokenizeGNUCommandLine, argv); // Handle -cc1 integrated tools.