1 files changed, 15 insertions, 8 deletions
diff --git a/lib/CodeGen/CGDebugInfo.cpp b/lib/CodeGen/CGDebugInfo.cpp
index 753cdcddd2..7458fa3e6f 100644
--- a/lib/CodeGen/CGDebugInfo.cpp
+++ b/lib/CodeGen/CGDebugInfo.cpp
@@ -188,18 +188,25 @@ void CGDebugInfo::CreateCompileUnit() {
   if (MainFileName.empty())
     MainFileName = "<unknown>";
 
-  llvm::sys::Path AbsFileName(MainFileName);
-  AbsFileName.makeAbsolute();
-
   // The main file name provided via the "-main-file-name" option contains just
   // the file name itself with no path information. This file name may have had
   // a relative path, so we look into the actual file entry for the main
   // file to determine the real absolute path for the file.
-  std::string MainFileDir;
-  if (const FileEntry *MainFile = SM.getFileEntryForID(SM.getMainFileID()))
+  char *FileNamePtr = NULL;
+  if (const FileEntry *MainFile = SM.getFileEntryForID(SM.getMainFileID())) {
+    std::string MainFileDir;
     MainFileDir = MainFile->getDir()->getName();
-  else
-    MainFileDir = AbsFileName.getDirname();
+    llvm::sys::Path AbsFileDirName(MainFileDir);
+    AbsFileDirName.makeAbsolute();
+    AbsFileDirName.appendComponent(MainFileName);
+    FileNamePtr = DebugInfoNames.Allocate<char>(AbsFileDirName.size());
+    memcpy(FileNamePtr, AbsFileDirName.c_str(), AbsFileDirName.size());
+  } else {
+    llvm::sys::Path AbsFileDirName(MainFileName);
+    AbsFileDirName.makeAbsolute();
+    FileNamePtr = DebugInfoNames.Allocate<char>(AbsFileDirName.size());
+    memcpy(FileNamePtr, AbsFileDirName.c_str(), AbsFileDirName.size());
+  }
 
   unsigned LangTag;
   const LangOptions &LO = CGM.getLangOptions();
@@ -229,7 +236,7 @@ void CGDebugInfo::CreateCompileUnit() {
 
   // Create new compile unit.
   TheCU = DebugFactory.CreateCompileUnit(
-    LangTag, AbsFileName.getLast(), MainFileDir, Producer, true,
+    LangTag, FileNamePtr, llvm::StringRef(), Producer, true,
     LO.Optimize, CGM.getCodeGenOpts().DwarfDebugFlags, RuntimeVers);
 }