1 //===- PDBSymbolCompiland.cpp - compiland details ---------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/DebugInfo/PDB/IPDBSession.h"
11 #include "llvm/DebugInfo/PDB/IPDBSourceFile.h"
12 
13 #include "llvm/DebugInfo/PDB/PDBSymbolCompiland.h"
14 #include "llvm/DebugInfo/PDB/PDBSymbolCompilandDetails.h"
15 #include "llvm/DebugInfo/PDB/PDBSymbolCompilandEnv.h"
16 #include "llvm/DebugInfo/PDB/PDBSymDumper.h"
17 
18 #include "llvm/ADT/StringSwitch.h"
19 #include "llvm/Support/Path.h"
20 #include <utility>
21 
22 using namespace llvm;
23 using namespace llvm::pdb;
24 
25 PDBSymbolCompiland::PDBSymbolCompiland(const IPDBSession &PDBSession,
26                                        std::unique_ptr<IPDBRawSymbol> Symbol)
27     : PDBSymbol(PDBSession, std::move(Symbol)) {
28   assert(RawSymbol->getSymTag() == PDB_SymType::Compiland);
29 }
30 
31 void PDBSymbolCompiland::dump(PDBSymDumper &Dumper) const {
32   Dumper.dump(*this);
33 }
34 
35 std::string PDBSymbolCompiland::getSourceFileName() const {
36   return sys::path::filename(getSourceFileFullPath()).str();
37 }
38 
39 std::string PDBSymbolCompiland::getSourceFileFullPath() const {
40   std::string SourceFileFullPath;
41 
42   // RecordedResult could be the basename, relative path or full path of the
43   // source file. Usually it is retrieved and recorded from the command that
44   // compiles this compiland.
45   //
46   //  cmd FileName          -> RecordedResult = .\\FileName
47   //  cmd (Path)\\FileName  -> RecordedResult = (Path)\\FileName
48   //
49   std::string RecordedResult = RawSymbol->getSourceFileName();
50 
51   if (RecordedResult.empty()) {
52     if (auto Envs = findAllChildren<PDBSymbolCompilandEnv>()) {
53       std::string EnvWorkingDir, EnvSrc;
54 
55       while (auto Env = Envs->getNext()) {
56         std::string Var = Env->getName();
57         if (Var == "cwd") {
58           EnvWorkingDir = Env->getValue();
59           continue;
60         }
61         if (Var == "src") {
62           EnvSrc = Env->getValue();
63           if (sys::path::is_absolute(EnvSrc))
64             return EnvSrc;
65           RecordedResult = EnvSrc;
66           continue;
67         }
68       }
69       if (!EnvWorkingDir.empty() && !EnvSrc.empty()) {
70         auto Len = EnvWorkingDir.length();
71         if (EnvWorkingDir[Len - 1] != '/' && EnvWorkingDir[Len - 1] != '\\') {
72           std::string Path = EnvWorkingDir + "\\" + EnvSrc;
73           std::replace(Path.begin(), Path.end(), '/', '\\');
74           // We will return it as full path if we can't find a better one.
75           if (sys::path::is_absolute(Path))
76             SourceFileFullPath = Path;
77         }
78       }
79     }
80   }
81 
82   if (!RecordedResult.empty()) {
83     if (sys::path::is_absolute(RecordedResult))
84       return RecordedResult;
85 
86     // This searches name that has same basename as the one in RecordedResult.
87     auto OneSrcFile = Session.findOneSourceFile(
88         this, RecordedResult, PDB_NameSearchFlags::NS_CaseInsensitive);
89     if (OneSrcFile)
90       return OneSrcFile->getFileName();
91   }
92 
93   // At this point, we have to walk through all source files of this compiland,
94   // and determine the right source file if any that is used to generate this
95   // compiland based on language indicated in compilanddetails language field.
96   auto Details = findOneChild<PDBSymbolCompilandDetails>();
97   PDB_Lang Lang = Details ? Details->getLanguage() : PDB_Lang::Cpp;
98   auto SrcFiles = Session.getSourceFilesForCompiland(*this);
99   if (SrcFiles) {
100     bool LangC = (Lang == PDB_Lang::Cpp || Lang == PDB_Lang::C);
101     while (auto File = SrcFiles->getNext()) {
102       std::string FileName = File->getFileName();
103       auto file_extension = sys::path::extension(FileName);
104       if (StringSwitch<bool>(file_extension.lower())
105               .Case(".cpp", LangC)
106               .Case(".c", LangC)
107               .Case(".cc", LangC)
108               .Case(".cxx", LangC)
109               .Case(".asm", Lang == PDB_Lang::Masm)
110               .Default(false))
111         return File->getFileName();
112     }
113   }
114 
115   return SourceFileFullPath;
116 }
117