1 //===- CIndexInclusionStack.cpp - Clang-C Source Indexing Library ---------===//
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 // This file defines a callback mechanism for clients to get the inclusion
11 // stack from a translation unit.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "CIndexer.h"
16 #include "CXSourceLocation.h"
17 #include "CXTranslationUnit.h"
18 #include "clang/AST/DeclVisitor.h"
19 #include "clang/Frontend/ASTUnit.h"
20 #include "llvm/ADT/SmallString.h"
21 #include "llvm/Support/raw_ostream.h"
22 using namespace clang;
23
24 extern "C" {
clang_getInclusions(CXTranslationUnit TU,CXInclusionVisitor CB,CXClientData clientData)25 void clang_getInclusions(CXTranslationUnit TU, CXInclusionVisitor CB,
26 CXClientData clientData) {
27 if (cxtu::isNotUsableTU(TU)) {
28 LOG_BAD_TU(TU);
29 return;
30 }
31
32 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
33 SourceManager &SM = CXXUnit->getSourceManager();
34 ASTContext &Ctx = CXXUnit->getASTContext();
35
36 SmallVector<CXSourceLocation, 10> InclusionStack;
37 unsigned n = SM.local_sloc_entry_size();
38
39 // In the case where all the SLocEntries are in an external source, traverse
40 // those SLocEntries as well. This is the case where we are looking
41 // at the inclusion stack of an AST/PCH file.
42 const SrcMgr::SLocEntry &(SourceManager::*Getter)(unsigned, bool*) const;
43 if (n == 1) {
44 Getter = &SourceManager::getLoadedSLocEntry;
45 n = SM.loaded_sloc_entry_size();
46 } else
47 Getter = &SourceManager::getLocalSLocEntry;
48
49 for (unsigned i = 0 ; i < n ; ++i) {
50 bool Invalid = false;
51 const SrcMgr::SLocEntry &SL = (SM.*Getter)(i, &Invalid);
52
53 if (!SL.isFile() || Invalid)
54 continue;
55
56 const SrcMgr::FileInfo &FI = SL.getFile();
57 if (!FI.getContentCache()->OrigEntry)
58 continue;
59
60 // Build the inclusion stack.
61 SourceLocation L = FI.getIncludeLoc();
62 InclusionStack.clear();
63 while (L.isValid()) {
64 PresumedLoc PLoc = SM.getPresumedLoc(L);
65 InclusionStack.push_back(cxloc::translateSourceLocation(Ctx, L));
66 L = PLoc.isValid()? PLoc.getIncludeLoc() : SourceLocation();
67 }
68
69 // Callback to the client.
70 // FIXME: We should have a function to construct CXFiles.
71 CB(static_cast<CXFile>(
72 const_cast<FileEntry *>(FI.getContentCache()->OrigEntry)),
73 InclusionStack.data(), InclusionStack.size(), clientData);
74 }
75 }
76 } // end extern C
77