/Users/buildslave/jenkins/workspace/coverage/llvm-project/clang/lib/Frontend/InterfaceStubFunctionsConsumer.cpp

Source (jump to first uncovered line)
//===--- InterfaceStubFunctionsConsumer.cpp -------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "clang/AST/Mangle.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/Basic/TargetInfo.h"
#include "clang/Frontend/CompilerInstance.h"
#include "clang/Frontend/FrontendActions.h"
#include "clang/Sema/TemplateInstCallback.h"
#include "llvm/BinaryFormat/ELF.h"

using namespace clang;

namespace {
class InterfaceStubFunctionsConsumer : public ASTConsumer {
  CompilerInstance &Instance;
  StringRef InFile;
  StringRef Format;
  std::set<std::string> ParsedTemplates;

  enum RootDeclOrigin { TopLevel = 0, FromTU = 1, IsLate = 2 };
  struct MangledSymbol {
    std::string ParentName;
    uint8_t Type;
    uint8_t Binding;
    std::vector<std::string> Names;
    MangledSymbol() = delete;

    MangledSymbol(const std::string &ParentName, uint8_t Type, uint8_t Binding,
                  std::vector<std::string> Names)
        : ParentName(ParentName), Type(Type), Binding(Binding), Names(Names) {}
  };
  using MangledSymbols = std::map<const NamedDecl *, MangledSymbol>;

  bool WriteNamedDecl(const NamedDecl *ND, MangledSymbols &Symbols, int RDO) {
    // Here we filter out anything that's not set to DefaultVisibility.
    // DefaultVisibility is set on a decl when -fvisibility is not specified on
    // the command line (or specified as default) and the decl does not have
    // __attribute__((visibility("hidden"))) set or when the command line
    // argument is set to hidden but the decl explicitly has
    // __attribute__((visibility ("default"))) set. We do this so that the user
    // can have fine grain control of what they want to expose in the stub.
    auto isVisible = [](const NamedDecl *ND) -> bool {
      return ND->getVisibility() == DefaultVisibility;
    };

    auto ignoreDecl = [this, isVisible](const NamedDecl *ND) -> bool {
      if (!isVisible(ND))
        return true;

      if (const VarDecl *VD = dyn_cast<VarDecl>(ND)) {
        if (const auto *Parent = VD->getParentFunctionOrMethod())
          if (isa<BlockDecl>(Parent) || isa<CXXMethodDecl>(Parent)17)
            return true;

        if ((VD->getStorageClass() == StorageClass::SC_Extern) ||
            (39VD->getStorageClass() == StorageClass::SC_Static39 &&
             VD->getParentFunctionOrMethod() == nullptr13))
          return true;
      }

      if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
        if (FD->isInlined() && !isa<CXXMethodDecl>(FD)119 &&
            !Instance.getLangOpts().GNUInline16)
          return true;
        if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD)) {
          if (const auto *RC = dyn_cast<CXXRecordDecl>(MD->getParent()))
            if (isa<ClassTemplateDecl>(RC->getParent()) || !isVisible(RC))
              return true;
          if (MD->isDependentContext() || !MD->hasBody()101)
            return true;
        }
        if (FD->getStorageClass() == StorageClass::SC_Static)
          return true;
      }
      return false;
    };

    auto getParentFunctionDecl = [](const NamedDecl *ND) -> const NamedDecl * {
      if (const VarDecl *VD = dyn_cast<VarDecl>(ND))
        if (const auto *FD =
                dyn_cast_or_null<FunctionDecl>(VD->getParentFunctionOrMethod()))
          return FD;
      return nullptr;
    };

    auto getMangledNames = [](const NamedDecl *ND) -> std::vector<std::string> {
      if (!ND)
        return {""};
      ASTNameGenerator NameGen(ND->getASTContext());
      std::vector<std::string> MangledNames = NameGen.getAllManglings(ND);
      if (isa<CXXConstructorDecl>(ND) || isa<CXXDestructorDecl>(ND)118)
        return MangledNames;
#ifdef EXPENSIVE_CHECKS
      assert(MangledNames.size() <= 1 && "Expected only one name mangling.");
#endif
      return {NameGen.getName(ND)};
    };

    if (!(RDO & FromTU))
      return true;
    if (Symbols.find(ND) != Symbols.end())
      return true;
    // - Currently have not figured out how to produce the names for FieldDecls.
    // - Do not want to produce symbols for function paremeters.
    if (isa<FieldDecl>(ND) || isa<ParmVarDecl>(ND)319)
      return true;

    const NamedDecl *ParentDecl = getParentFunctionDecl(ND);
    if ((ParentDecl && ignoreDecl(ParentDecl)21) || ignoreDecl(ND)295)
      return true;

    if (RDO & IsLate) {
      Instance.getDiagnostics().Report(diag::err_asm_invalid_type_in_input)
          << "Generating Interface Stubs is not supported with "
             "delayed template parsing.";
    } else {
      if (const auto *FD = dyn_cast<FunctionDecl>(ND))
        if (FD->isDependentContext())
          return true;

      const bool IsWeak = (ND->hasAttr<WeakAttr>() ||
                           ND->hasAttr<WeakRefAttr>()109 || ND->isWeakImported()109);

      Symbols.insert(std::make_pair(
          ND,
          MangledSymbol(getMangledNames(ParentDecl).front(),
                        // Type:
                        isa<VarDecl>(ND) ? llvm::ELF::STT_OBJECT36
                                         : llvm::ELF::STT_FUNC76,
                        // Binding:
                        IsWeak ? llvm::ELF::STB_WEAK3 : llvm::ELF::STB_GLOBAL109,
                        getMangledNames(ND))));
    }
    return true;
  }

  void
  HandleDecls(const llvm::iterator_range<DeclContext::decl_iterator> &Decls,
              MangledSymbols &Symbols, int RDO) {
    for (const auto *D : Decls)
      HandleNamedDecl(dyn_cast<NamedDecl>(D), Symbols, RDO);
  }

  void HandleTemplateSpecializations(const FunctionTemplateDecl &FTD,
                                     MangledSymbols &Symbols, int RDO) {
    for (const auto *D : FTD.specializations())
      HandleNamedDecl(dyn_cast<NamedDecl>(D), Symbols, RDO);
  }

  void HandleTemplateSpecializations(const ClassTemplateDecl &CTD,
                                     MangledSymbols &Symbols, int RDO) {
    for (const auto *D : CTD.specializations())
      HandleNamedDecl(dyn_cast<NamedDecl>(D), Symbols, RDO);
  }

  bool HandleNamedDecl(const NamedDecl *ND, MangledSymbols &Symbols, int RDO) {
    if (!ND)
      return false;

    switch (ND->getKind()) {
    default:
      break;
    case Decl::Kind::Namespace:
      HandleDecls(cast<NamespaceDecl>(ND)->decls(), Symbols, RDO);
      return true;
    case Decl::Kind::CXXRecord:
      HandleDecls(cast<CXXRecordDecl>(ND)->decls(), Symbols, RDO);
      return true;
    case Decl::Kind::ClassTemplateSpecialization:
      HandleDecls(cast<ClassTemplateSpecializationDecl>(ND)->decls(), Symbols,
                  RDO);
      return true;
    case Decl::Kind::ClassTemplate:
      HandleTemplateSpecializations(*cast<ClassTemplateDecl>(ND), Symbols, RDO);
      return true;
    case Decl::Kind::FunctionTemplate:
      HandleTemplateSpecializations(*cast<FunctionTemplateDecl>(ND), Symbols,
                                    RDO);
      return true;
    case Decl::Kind::Record:
    case Decl::Kind::Typedef:
    case Decl::Kind::Enum:
    case Decl::Kind::EnumConstant:
    case Decl::Kind::TemplateTypeParm:
    case Decl::Kind::NonTypeTemplateParm:
    case Decl::Kind::CXXConversion:
    case Decl::Kind::UnresolvedUsingValue:
    case Decl::Kind::Using:
    case Decl::Kind::UsingShadow:
    case Decl::Kind::TypeAliasTemplate:
    case Decl::Kind::TypeAlias:
    case Decl::Kind::VarTemplate:
    case Decl::Kind::VarTemplateSpecialization:
    case Decl::Kind::UsingDirective:
    case Decl::Kind::TemplateTemplateParm:
    case Decl::Kind::ClassTemplatePartialSpecialization:
    case Decl::Kind::IndirectField:
    case Decl::Kind::ConstructorUsingShadow:
    case Decl::Kind::CXXDeductionGuide:
    case Decl::Kind::NamespaceAlias:
    case Decl::Kind::UnresolvedUsingTypename:
      return true;
    case Decl::Kind::Var: {
      // Bail on any VarDecl that either has no named symbol.
      if (!ND->getIdentifier())
        return true;
      const auto *VD = cast<VarDecl>(ND);
      // Bail on any VarDecl that is a dependent or templated type.
      if (VD->isTemplated() || VD->getType()->isDependentType()47)
        return true;
      if (WriteNamedDecl(ND, Symbols, RDO))
        return true;
      break;
    }
    case Decl::Kind::ParmVar:
    case Decl::Kind::CXXMethod:
    case Decl::Kind::CXXConstructor:
    case Decl::Kind::CXXDestructor:
    case Decl::Kind::Function:
    case Decl::Kind::Field:
      if (WriteNamedDecl(ND, Symbols, RDO))
        return true;
    }

    // While interface stubs are in the development stage, it's probably best to
    // catch anything that's not a VarDecl or Template/FunctionDecl.
    Instance.getDiagnostics().Report(diag::err_asm_invalid_type_in_input)
        << "Expected a function or function template decl.";
    return false;
  }

public:
  InterfaceStubFunctionsConsumer(CompilerInstance &Instance, StringRef InFile,
                                 StringRef Format)
      : Instance(Instance), InFile(InFile), Format(Format) {}

  void HandleTranslationUnit(ASTContext &context) override {
    struct Visitor : public RecursiveASTVisitor<Visitor> {
      bool VisitNamedDecl(NamedDecl *ND) {
        if (const auto *FD = dyn_cast<FunctionDecl>(ND))
          if (FD->isLateTemplateParsed()) {
            LateParsedDecls.insert(FD);
            return true;
          }

        if (const auto *VD = dyn_cast<ValueDecl>(ND)) {
          ValueDecls.insert(VD);
          return true;
        }

        NamedDecls.insert(ND);
        return true;
      }

      std::set<const NamedDecl *> LateParsedDecls;
      std::set<NamedDecl *> NamedDecls;
      std::set<const ValueDecl *> ValueDecls;
    } v;

    v.TraverseDecl(context.getTranslationUnitDecl());

    MangledSymbols Symbols;
    auto OS = Instance.createDefaultOutputFile(/*Binary=*/false, InFile, "ifs");
    if (!OS)
      return;

    if (Instance.getLangOpts().DelayedTemplateParsing) {
      clang::Sema &S = Instance.getSema();
      for (const auto *FD : v.LateParsedDecls) {
        clang::LateParsedTemplate &LPT =
            *S.LateParsedTemplateMap.find(cast<FunctionDecl>(FD))->second;
        S.LateTemplateParser(S.OpaqueParser, LPT);
        HandleNamedDecl(FD, Symbols, (FromTU | IsLate));
      }
    }

    for (const NamedDecl *ND : v.ValueDecls)
      HandleNamedDecl(ND, Symbols, FromTU);
    for (const NamedDecl *ND : v.NamedDecls)
      HandleNamedDecl(ND, Symbols, FromTU);

    auto writeIfsV1 = [this](const llvm::Triple &T,
                             const MangledSymbols &Symbols,
                             const ASTContext &context, StringRef Format,
                             raw_ostream &OS) -> void {
      OS << "--- !" << Format << "\n";
      OS << "IfsVersion: 3.0\n";
      OS << "Target: " << T.str() << "\n";
      OS << "Symbols:\n";
      for (const auto &E : Symbols) {
        const MangledSymbol &Symbol = E.second;
        for (auto Name : Symbol.Names) {
          OS << "  - { Name: \""
             << (Symbol.ParentName.empty() || Instance.getLangOpts().CPlusPlus16
                     ? ""124
                     : (Symbol.ParentName + ".")10)
             << Name << "\", Type: ";
          switch (Symbol.Type) {
          default:
            llvm_unreachable(
                "clang -emit-interface-stubs: Unexpected symbol type.");
          case llvm::ELF::STT_NOTYPE:
            OS << "NoType";
            break;
          case llvm::ELF::STT_OBJECT: {
            auto VD = cast<ValueDecl>(E.first)->getType();
            OS << "Object, Size: "
               << context.getTypeSizeInChars(VD).getQuantity();
            break;
          }
          case llvm::ELF::STT_FUNC:
            OS << "Func";
            break;
          }
          if (Symbol.Binding == llvm::ELF::STB_WEAK)
            OS << ", Weak: true";
          OS << " }\n";
        }
      }
      OS << "...\n";
      OS.flush();
    };

    assert(Format == "ifs-v1" && "Unexpected IFS Format.");
    writeIfsV1(Instance.getTarget().getTriple(), Symbols, context, Format, *OS);
  }
};
} // namespace

std::unique_ptr<ASTConsumer>
GenerateInterfaceStubsAction::CreateASTConsumer(CompilerInstance &CI,
                                                StringRef InFile) {
  return std::make_unique<InterfaceStubFunctionsConsumer>(CI, InFile, "ifs-v1");
}

Coverage Report

Created: 2022-01-22 13:19

Line	Count	Source (jump to first uncovered line)
1		//===--- InterfaceStubFunctionsConsumer.cpp -------------------------------===//
2		//
3		// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4		// See https://llvm.org/LICENSE.txt for license information.
5		// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6		//
7		//===----------------------------------------------------------------------===//
8
9		#include "clang/AST/Mangle.h"
10		#include "clang/AST/RecursiveASTVisitor.h"
11		#include "clang/Basic/TargetInfo.h"
12		#include "clang/Frontend/CompilerInstance.h"
13		#include "clang/Frontend/FrontendActions.h"
14		#include "clang/Sema/TemplateInstCallback.h"
15		#include "llvm/BinaryFormat/ELF.h"
16
17		using namespace clang;
18
19		namespace {
20		class InterfaceStubFunctionsConsumer : public ASTConsumer {
21		CompilerInstance &Instance;
22		StringRef InFile;
23		StringRef Format;
24		std::set<std::string> ParsedTemplates;
25
26		enum RootDeclOrigin { TopLevel = 0, FromTU = 1, IsLate = 2 };
27		struct MangledSymbol {
28		std::string ParentName;
29		uint8_t Type;
30		uint8_t Binding;
31		std::vector<std::string> Names;
32		MangledSymbol() = delete;
33
34		MangledSymbol(const std::string &ParentName, uint8_t Type, uint8_t Binding,
35		std::vector<std::string> Names)
36	112	: ParentName(ParentName), Type(Type), Binding(Binding), Names(Names) {}
37		};
38		using MangledSymbols = std::map<const NamedDecl *, MangledSymbol>;
39
40	340	bool WriteNamedDecl(const NamedDecl *ND, MangledSymbols &Symbols, int RDO) {
41		// Here we filter out anything that's not set to DefaultVisibility.
42		// DefaultVisibility is set on a decl when -fvisibility is not specified on
43		// the command line (or specified as default) and the decl does not have
44		// __attribute__((visibility("hidden"))) set or when the command line
45		// argument is set to hidden but the decl explicitly has
46		// __attribute__((visibility ("default"))) set. We do this so that the user
47		// can have fine grain control of what they want to expose in the stub.
48	435	auto isVisible = [](const NamedDecl *ND) -> bool {
49	435	return ND->getVisibility() == DefaultVisibility;
50	435	};
51
52	340	auto ignoreDecl = [this, isVisible](const NamedDecl *ND) -> bool {
53	316	if (!isVisible(ND))
54	73	return true;
55
56	243	if (const VarDecl *VD = dyn_cast<VarDecl>(ND)) {
57	43	if (const auto *Parent = VD->getParentFunctionOrMethod())
58	18	if (isa<BlockDecl>(Parent) \|\| isa<CXXMethodDecl>(Parent)17 )
59	2	return true;
60
61	41	if ((VD->getStorageClass() == StorageClass::SC_Extern) \|\|
62	41	(39 VD->getStorageClass() == StorageClass::SC_Static39 &&
63	39	VD->getParentFunctionOrMethod() == nullptr13 ))
64	5	return true;
65	41	}
66
67	236	if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
68	200	if (FD->isInlined() && !isa<CXXMethodDecl>(FD)119 &&
69	200	!Instance.getLangOpts().GNUInline16 )
70	4	return true;
71	196	if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD)) {
72	119	if (const auto *RC = dyn_cast<CXXRecordDecl>(MD->getParent()))
73	119	if (isa<ClassTemplateDecl>(RC->getParent()) \|\| !isVisible(RC))
74	0	return true;
75	119	if (MD->isDependentContext() \|\| !MD->hasBody()101 )
76	94	return true;
77	119	}
78	102	if (FD->getStorageClass() == StorageClass::SC_Static)
79	5	return true;
80	102	}
81	133	return false;
82	236	};
83
84	340	auto getParentFunctionDecl = [](const NamedDecl ND) -> const NamedDecl {
85	299	if (const VarDecl *VD = dyn_cast<VarDecl>(ND))
86	47	if (const auto *FD =
87	47	dyn_cast_or_null<FunctionDecl>(VD->getParentFunctionOrMethod()))
88	21	return FD;
89	278	return nullptr;
90	299	};
91
92	340	auto getMangledNames = [](const NamedDecl *ND) -> std::vector<std::string> {
93	224	if (!ND)
94	96	return {""};
95	128	ASTNameGenerator NameGen(ND->getASTContext());
96	128	std::vector<std::string> MangledNames = NameGen.getAllManglings(ND);
97	128	if (isa<CXXConstructorDecl>(ND) \|\| isa<CXXDestructorDecl>(ND)118 )
98	16	return MangledNames;
99		#ifdef EXPENSIVE_CHECKS
100		assert(MangledNames.size() <= 1 && "Expected only one name mangling.");
101		#endif
102	112	return {NameGen.getName(ND)};
103	128	};
104
105	340	if (!(RDO & FromTU))
106	0	return true;
107	340	if (Symbols.find(ND) != Symbols.end())
108	15	return true;
109		// - Currently have not figured out how to produce the names for FieldDecls.
110		// - Do not want to produce symbols for function paremeters.
111	325	if (isa<FieldDecl>(ND) \|\| isa<ParmVarDecl>(ND)319 )
112	26	return true;
113
114	299	const NamedDecl *ParentDecl = getParentFunctionDecl(ND);
115	299	if ((ParentDecl && ignoreDecl(ParentDecl)21 ) \|\| ignoreDecl(ND)295 )
116	183	return true;
117
118	116	if (RDO & IsLate) {
119	0	Instance.getDiagnostics().Report(diag::err_asm_invalid_type_in_input)
120	0	<< "Generating Interface Stubs is not supported with "
121	0	"delayed template parsing.";
122	116	} else {
123	116	if (const auto *FD = dyn_cast<FunctionDecl>(ND))
124	80	if (FD->isDependentContext())
125	4	return true;
126
127	112	const bool IsWeak = (ND->hasAttr<WeakAttr>() \|\|
128	112	ND->hasAttr<WeakRefAttr>()109 \|\| ND->isWeakImported()109 );
129
130	112	Symbols.insert(std::make_pair(
131	112	ND,
132	112	MangledSymbol(getMangledNames(ParentDecl).front(),
133		// Type:
134	112	isa<VarDecl>(ND) ? llvm::ELF::STT_OBJECT36
135	112	: llvm::ELF::STT_FUNC76 ,
136		// Binding:
137	112	IsWeak ? llvm::ELF::STB_WEAK3 : llvm::ELF::STB_GLOBAL109 ,
138	112	getMangledNames(ND))));
139	112	}
140	112	return true;
141	116	}
142
143		void
144		HandleDecls(const llvm::iterator_range<DeclContext::decl_iterator> &Decls,
145	96	MangledSymbols &Symbols, int RDO) {
146	96	for (const auto *D : Decls)
147	222	HandleNamedDecl(dyn_cast<NamedDecl>(D), Symbols, RDO);
148	96	}
149
150		void HandleTemplateSpecializations(const FunctionTemplateDecl &FTD,
151	5	MangledSymbols &Symbols, int RDO) {
152	5	for (const auto *D : FTD.specializations())
153	6	HandleNamedDecl(dyn_cast<NamedDecl>(D), Symbols, RDO);
154	5	}
155
156		void HandleTemplateSpecializations(const ClassTemplateDecl &CTD,
157	18	MangledSymbols &Symbols, int RDO) {
158	18	for (const auto *D : CTD.specializations())
159	7	HandleNamedDecl(dyn_cast<NamedDecl>(D), Symbols, RDO);
160	18	}
161
162	529	bool HandleNamedDecl(const NamedDecl *ND, MangledSymbols &Symbols, int RDO) {
163	529	if (!ND)
164	6	return false;
165
166	523	switch (ND->getKind()) {
167	0	default:
168	0	break;
169	6	case Decl::Kind::Namespace:
170	6	HandleDecls(cast<NamespaceDecl>(ND)->decls(), Symbols, RDO);
171	6	return true;
172	83	case Decl::Kind::CXXRecord:
173	83	HandleDecls(cast<CXXRecordDecl>(ND)->decls(), Symbols, RDO);
174	83	return true;
175	7	case Decl::Kind::ClassTemplateSpecialization:
176	7	HandleDecls(cast<ClassTemplateSpecializationDecl>(ND)->decls(), Symbols,
177	7	RDO);
178	7	return true;
179	18	case Decl::Kind::ClassTemplate:
180	18	HandleTemplateSpecializations(*cast<ClassTemplateDecl>(ND), Symbols, RDO);
181	18	return true;
182	5	case Decl::Kind::FunctionTemplate:
183	5	HandleTemplateSpecializations(*cast<FunctionTemplateDecl>(ND), Symbols,
184	5	RDO);
185	5	return true;
186	1	case Decl::Kind::Record:
187	3	case Decl::Kind::Typedef:
188	4	case Decl::Kind::Enum:
189	5	case Decl::Kind::EnumConstant:
190	27	case Decl::Kind::TemplateTypeParm:
191	33	case Decl::Kind::NonTypeTemplateParm:
192	35	case Decl::Kind::CXXConversion:
193	37	case Decl::Kind::UnresolvedUsingValue:
194	40	case Decl::Kind::Using:
195	40	case Decl::Kind::UsingShadow:
196	41	case Decl::Kind::TypeAliasTemplate:
197	44	case Decl::Kind::TypeAlias:
198	45	case Decl::Kind::VarTemplate:
199	46	case Decl::Kind::VarTemplateSpecialization:
200	47	case Decl::Kind::UsingDirective:
201	48	case Decl::Kind::TemplateTemplateParm:
202	49	case Decl::Kind::ClassTemplatePartialSpecialization:
203	50	case Decl::Kind::IndirectField:
204	53	case Decl::Kind::ConstructorUsingShadow:
205	54	case Decl::Kind::CXXDeductionGuide:
206	55	case Decl::Kind::NamespaceAlias:
207	57	case Decl::Kind::UnresolvedUsingTypename:
208	57	return true;
209	54	case Decl::Kind::Var: {
210		// Bail on any VarDecl that either has no named symbol.
211	54	if (!ND->getIdentifier())
212	1	return true;
213	53	const auto *VD = cast<VarDecl>(ND);
214		// Bail on any VarDecl that is a dependent or templated type.
215	53	if (VD->isTemplated() \|\| VD->getType()->isDependentType()47 )
216	6	return true;
217	47	if (WriteNamedDecl(ND, Symbols, RDO))
218	47	return true;
219	0	break;
220	47	}
221	20	case Decl::Kind::ParmVar:
222	116	case Decl::Kind::CXXMethod:
223	181	case Decl::Kind::CXXConstructor:
224	214	case Decl::Kind::CXXDestructor:
225	287	case Decl::Kind::Function:
226	293	case Decl::Kind::Field:
227	293	if (WriteNamedDecl(ND, Symbols, RDO))
228	293	return true;
229	523	}
230
231		// While interface stubs are in the development stage, it's probably best to
232		// catch anything that's not a VarDecl or Template/FunctionDecl.
233	0	Instance.getDiagnostics().Report(diag::err_asm_invalid_type_in_input)
234	0	<< "Expected a function or function template decl.";
235	0	return false;
236	523	}
237
238		public:
239		InterfaceStubFunctionsConsumer(CompilerInstance &Instance, StringRef InFile,
240		StringRef Format)
241	69	: Instance(Instance), InFile(InFile), Format(Format) {}
242
243	69	void HandleTranslationUnit(ASTContext &context) override {
244	69	struct Visitor : public RecursiveASTVisitor<Visitor> {
245	294	bool VisitNamedDecl(NamedDecl *ND) {
246	294	if (const auto *FD = dyn_cast<FunctionDecl>(ND))
247	110	if (FD->isLateTemplateParsed()) {
248	0	LateParsedDecls.insert(FD);
249	0	return true;
250	0	}
251
252	294	if (const auto *VD = dyn_cast<ValueDecl>(ND)) {
253	192	ValueDecls.insert(VD);
254	192	return true;
255	192	}
256
257	102	NamedDecls.insert(ND);
258	102	return true;
259	294	}
260
261	69	std::set<const NamedDecl *> LateParsedDecls;
262	69	std::set<NamedDecl *> NamedDecls;
263	69	std::set<const ValueDecl *> ValueDecls;
264	69	} v;
265
266	69	v.TraverseDecl(context.getTranslationUnitDecl());
267
268	69	MangledSymbols Symbols;
269	69	auto OS = Instance.createDefaultOutputFile(/Binary=/false, InFile, "ifs");
270	69	if (!OS)
271	0	return;
272
273	69	if (Instance.getLangOpts().DelayedTemplateParsing) {
274	1	clang::Sema &S = Instance.getSema();
275	1	for (const auto *FD : v.LateParsedDecls) {
276	0	clang::LateParsedTemplate &LPT =
277	0	*S.LateParsedTemplateMap.find(cast<FunctionDecl>(FD))->second;
278	0	S.LateTemplateParser(S.OpaqueParser, LPT);
279	0	HandleNamedDecl(FD, Symbols, (FromTU \| IsLate));
280	0	}
281	1	}
282
283	69	for (const NamedDecl *ND : v.ValueDecls)
284	192	HandleNamedDecl(ND, Symbols, FromTU);
285	69	for (const NamedDecl *ND : v.NamedDecls)
286	102	HandleNamedDecl(ND, Symbols, FromTU);
287
288	69	auto writeIfsV1 = [this](const llvm::Triple &T,
289	69	const MangledSymbols &Symbols,
290	69	const ASTContext &context, StringRef Format,
291	69	raw_ostream &OS) -> void {
292	69	OS << "--- !" << Format << "\n";
293	69	OS << "IfsVersion: 3.0\n";
294	69	OS << "Target: " << T.str() << "\n";
295	69	OS << "Symbols:\n";
296	112	for (const auto &E : Symbols) {
297	112	const MangledSymbol &Symbol = E.second;
298	134	for (auto Name : Symbol.Names) {
299	134	OS << " - { Name: \""
300	134	<< (Symbol.ParentName.empty() \|\| Instance.getLangOpts().CPlusPlus16
301	134	? ""124
302	134	: (Symbol.ParentName + ".")10 )
303	134	<< Name << "\", Type: ";
304	134	switch (Symbol.Type) {
305	0	default:
306	0	llvm_unreachable(
307	0	"clang -emit-interface-stubs: Unexpected symbol type.");
308	0	case llvm::ELF::STT_NOTYPE:
309	0	OS << "NoType";
310	0	break;
311	36	case llvm::ELF::STT_OBJECT: {
312	36	auto VD = cast<ValueDecl>(E.first)->getType();
313	36	OS << "Object, Size: "
314	36	<< context.getTypeSizeInChars(VD).getQuantity();
315	36	break;
316	0	}
317	98	case llvm::ELF::STT_FUNC:
318	98	OS << "Func";
319	98	break;
320	134	}
321	134	if (Symbol.Binding == llvm::ELF::STB_WEAK)
322	3	OS << ", Weak: true";
323	134	OS << " }\n";
324	134	}
325	112	}
326	69	OS << "...\n";
327	69	OS.flush();
328	69	};
329
330	69	assert(Format == "ifs-v1" && "Unexpected IFS Format.");
331	0	writeIfsV1(Instance.getTarget().getTriple(), Symbols, context, Format, *OS);
332	69	}
333		};
334		} // namespace
335
336		std::unique_ptr<ASTConsumer>
337		GenerateInterfaceStubsAction::CreateASTConsumer(CompilerInstance &CI,
338	69	StringRef InFile) {
339	69	return std::make_unique<InterfaceStubFunctionsConsumer>(CI, InFile, "ifs-v1");
340	69	}