Fix unsafe use of StringRef I introduced.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@86829 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/tools/driver/driver.cpp b/tools/driver/driver.cpp
index 1947bac99b50161b2461039bb7079bbcf1a1f2b4..fb7c6cc7b03254a89851d776cd1197b5531a2791 100644 (file)
--- a/tools/driver/driver.cpp
+++ b/tools/driver/driver.cpp
@@ -202,8 +202,9 @@ int main(int argc, const char **argv) {
   //
   // Note that we intentionally want to use argv[0] here, to support "clang++"
   // being a symlink.
-  llvm::StringRef ProgName(llvm::sys::Path(argv[0]).getBasename());
-  if (ProgName.endswith("++") || ProgName.rsplit('-').first.endswith("++"))
+  std::string ProgName(llvm::sys::Path(argv[0]).getBasename());
+  if (llvm::StringRef(ProgName).endswith("++") ||
+      llvm::StringRef(ProgName).rsplit('-').first.endswith("++"))
     TheDriver.CCCIsCXX = true;
 
   llvm::OwningPtr<Compilation> C;