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
PDBSymbolCompiland(const IPDBSession & PDBSession,std::unique_ptr<IPDBRawSymbol> Symbol)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
dump(PDBSymDumper & Dumper) const31 void PDBSymbolCompiland::dump(PDBSymDumper &Dumper) const {
32 Dumper.dump(*this);
33 }
34
getSourceFileName() const35 std::string PDBSymbolCompiland::getSourceFileName() const {
36 return sys::path::filename(getSourceFileFullPath()).str();
37 }
38
getSourceFileFullPath() const39 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