diff --git a/clang/lib/AST/ASTImporter.cpp b/clang/lib/AST/ASTImporter.cpp --- a/clang/lib/AST/ASTImporter.cpp +++ b/clang/lib/AST/ASTImporter.cpp @@ -8094,9 +8094,6 @@ else return ToTOrErr.takeError(); } - To->setInherited(From->isInherited()); - To->setPackExpansion(From->isPackExpansion()); - To->setImplicit(From->isImplicit()); ToAttr = To; break; } @@ -8106,7 +8103,6 @@ IdentifierInfo *ToAttrType = Import(From->getType()); To = FormatAttr::Create(ToContext, ToAttrType, From->getFormatIdx(), From->getFirstArg(), ToRange, From->getSyntax()); - To->setInherited(From->isInherited()); ToAttr = To; break; } @@ -8117,8 +8113,15 @@ ToAttr->setRange(ToRange); break; } + assert(ToAttr && "Attribute should be created."); - + if (const auto *InheritableFromAttr = dyn_cast(FromAttr)) + cast(ToAttr)->setInherited( + InheritableFromAttr->isInherited()); + ToAttr->setAttributeSpellingListIndex( + FromAttr->getAttributeSpellingListIndex()); + ToAttr->setPackExpansion(FromAttr->isPackExpansion()); + ToAttr->setImplicit(FromAttr->isImplicit()); return ToAttr; } diff --git a/clang/test/ASTMerge/attr/Inputs/RestrictAttr.cpp b/clang/test/ASTMerge/attr/Inputs/RestrictAttr.cpp new file mode 100644 --- /dev/null +++ b/clang/test/ASTMerge/attr/Inputs/RestrictAttr.cpp @@ -0,0 +1 @@ +void *foo(unsigned, unsigned) __attribute__((__malloc__)); diff --git a/clang/test/ASTMerge/attr/testRestrictAttr.cpp b/clang/test/ASTMerge/attr/testRestrictAttr.cpp new file mode 100644 --- /dev/null +++ b/clang/test/ASTMerge/attr/testRestrictAttr.cpp @@ -0,0 +1,2 @@ +// RUN: %clang -x c++-header -o %t.a.ast %S/Inputs/RestrictAttr.cpp +// RUN: %clang_cc1 -x c++ -ast-merge %t.a.ast /dev/null -ast-dump diff --git a/clang/unittests/AST/ASTImporterTest.cpp b/clang/unittests/AST/ASTImporterTest.cpp --- a/clang/unittests/AST/ASTImporterTest.cpp +++ b/clang/unittests/AST/ASTImporterTest.cpp @@ -5767,11 +5767,35 @@ EXPECT_TRUE(ToA); } +TEST_P(ASTImporterOptionSpecificTestBase, ImportRestrictAttr) { + Decl *FromTU = getTuDecl( + R"( + void *foo(unsigned, unsigned) __attribute__((__malloc__)); + )", + Lang_CXX03, "input.cc"); + auto *FromD = FirstDeclMatcher().match( + FromTU, functionDecl(hasName("foo"))); + ASSERT_TRUE(FromD); + + auto *ToD = Import(FromD, Lang_CXX03); + ASSERT_TRUE(ToD); + ToD->dump(); // Should not crash! + + auto *FromAttr = FromD->getAttr(); + auto *ToAttr = ToD->getAttr(); + EXPECT_EQ(FromAttr->isInherited(), ToAttr->isInherited()); + EXPECT_EQ(FromAttr->isPackExpansion(), ToAttr->isPackExpansion()); + EXPECT_EQ(FromAttr->isImplicit(), ToAttr->isImplicit()); + EXPECT_EQ(FromAttr->getSyntax(), ToAttr->getSyntax()); + EXPECT_EQ(FromAttr->getAttributeSpellingListIndex(), + ToAttr->getAttributeSpellingListIndex()); +} + TEST_P(ASTImporterOptionSpecificTestBase, ImportFormatAttr) { Decl *FromTU = getTuDecl( R"( int foo(const char * fmt, ...) - __attribute__ ((__format__ (__scanf__, 1, 2))); + __attribute__ ((__format__ (__scanf__, 1, 2))); )", Lang_CXX03, "input.cc"); auto *FromD = FirstDeclMatcher().match( @@ -5792,6 +5816,7 @@ ToAttr->getAttributeSpellingListIndex()); EXPECT_EQ(FromAttr->getType()->getName(), ToAttr->getType()->getName()); } + template auto ExtendWithOptions(const T &Values, const std::vector &Args) { auto Copy = Values;