From: Daniel Dunbar <daniel@zuster.org>
Date: Wed, 11 Nov 2009 10:10:25 +0000 (+0000)
Subject: Fix unsafe use of StringRef I introduced.
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=e26bd90e5f7c1e44dd2b74d84d2f158dc7b983fb;p=clang

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 1947bac99b..fb7c6cc7b0 100644
--- 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;