Ignore functions, defines and variables from included headers

Author: kornilova203

bindgen/defines/DefineFinder.cpp

@@ -11,6 +11,10 @@ DefineFinder::DefineFinder(IR &ir, const clang::CompilerInstance &compiler,
 void DefineFinder::MacroDefined(const clang::Token &macroNameTok,
                                 const clang::MacroDirective *md) {
     clang::SourceManager &sm = compiler.getSourceManager();
+    if (!sm.isInMainFile(md->getLocation())) {
+        /* include defines only from the original header */
+        return;
+    }
     if (sm.isWrittenInMainFile(macroNameTok.getLocation()) && md->isDefined() &&
         !md->getMacroInfo()->isFunctionLike()) {
         /* save defines only from the given header.
@@ -100,6 +104,9 @@ void DefineFinder::MacroUndefined(const clang::Token &macroNameTok,
                                   const clang::MacroDefinition &md,
                                   const clang::MacroDirective *undef) {
     clang::SourceManager &sm = compiler.getSourceManager();
+    if (!sm.isInMainFile(undef->getLocation())) {
+        return;
+    }
     if (sm.isWrittenInMainFile(macroNameTok.getLocation()) &&
         md.getMacroInfo() && !md.getMacroInfo()->isFunctionLike()) {
         std::string macroName = macroNameTok.getIdentifierInfo()->getName();

bindgen/visitor/TreeVisitor.cpp

@@ -5,6 +5,10 @@ HeaderManager headerMan;
 std::set<std::string> locations;
 
 bool TreeVisitor::VisitFunctionDecl(clang::FunctionDecl *func) {
+    if (!astContext->getSourceManager().isInMainFile(func->getLocation())) {
+        /* include functions only from the original header */
+        return true;
+    }
     std::string funcName = func->getNameInfo().getName().getAsString();
     std::shared_ptr<Type> retType =
         typeTranslator.translate(func->getReturnType());
@@ -157,6 +161,10 @@ void TreeVisitor::handleStruct(clang::RecordDecl *record, std::string name) {
 }
 
 bool TreeVisitor::VisitVarDecl(clang::VarDecl *varDecl) {
+    if (!astContext->getSourceManager().isInMainFile(varDecl->getLocation())) {
+        /* include variables only from the original header */
+        return true;
+    }
     if (!varDecl->isThisDeclarationADefinition()) {
         std::string variableName = varDecl->getName().str();
         std::shared_ptr<Type> type =