Allow applying attributes to subset of allowed subjects.

Differential Revision: https://reviews.llvm.org/D100136
This commit is contained in:
Artem Belevich 2021-04-08 13:12:10 -07:00
parent f4d682d6ce
commit 38cf112a6b
3 changed files with 36 additions and 6 deletions

View File

@ -875,12 +875,33 @@ void Sema::ActOnPragmaAttributeAttribute(
}
Rules.clear();
} else {
for (const auto &Rule : StrictSubjectMatchRuleSet) {
if (Rules.erase(Rule.first)) {
// Each rule in Rules must be a strict subset of the attribute's
// SubjectMatch rules. I.e. we're allowed to use
// `apply_to=variables(is_global)` on an attrubute with SubjectList<[Var]>,
// but should not allow `apply_to=variables` on an attribute which has
// `SubjectList<[GlobalVar]>`.
for (const auto &StrictRule : StrictSubjectMatchRuleSet) {
// First, check for exact match.
if (Rules.erase(StrictRule.first)) {
// Add the rule to the set of attribute receivers only if it's supported
// in the current language mode.
if (Rule.second)
SubjectMatchRules.push_back(Rule.first);
if (StrictRule.second)
SubjectMatchRules.push_back(StrictRule.first);
}
}
// Check remaining rules for subset matches.
auto RulesToCheck = Rules;
for (const auto &Rule : RulesToCheck) {
attr::SubjectMatchRule MatchRule = attr::SubjectMatchRule(Rule.first);
if (auto ParentRule = getParentAttrMatcherRule(MatchRule)) {
if (llvm::any_of(StrictSubjectMatchRuleSet,
[ParentRule](const auto &StrictRule) {
return StrictRule.first == *ParentRule &&
StrictRule.second; // IsEnabled
})) {
SubjectMatchRules.push_back(MatchRule);
Rules.erase(MatchRule);
}
}
}
}

View File

@ -195,3 +195,12 @@ _Pragma("clang attribute pop");
#pragma clang attribute pop
#pragma clang attribute push([[clang::uninitialized]], apply_to = any(variable(is_parameter), variable(unless(is_parameter)))) // expected-error {{attribute 'uninitialized' can't be applied to 'variable(is_parameter)', and 'variable(unless(is_parameter))'}}
#pragma clang attribute pop
// We're allowed to apply attributes to subsets of allowed subjects.
#pragma clang attribute push([[clang::no_destroy]], apply_to = variable)
#pragma clang attribute pop
#pragma clang attribute push([[clang::no_destroy]], apply_to = variable(is_thread_local))
#pragma clang attribute pop
#pragma clang attribute push([[clang::no_destroy]], apply_to = variable(is_global))
#pragma clang attribute pop
#pragma clang attribute push([[clang::no_destroy]], apply_to = any(variable(is_parameter), variable(unless(is_parameter))))
#pragma clang attribute pop

View File

@ -55,9 +55,9 @@
// expected-error@-1 {{attribute 'abi_tag' can't be applied to 'enum'}}
#pragma clang attribute pop
#pragma clang attribute push (__attribute__((abi_tag("a"))), apply_to = any(enum_constant, function, record(unless(is_union)), variable, variable(is_parameter)))
#pragma clang attribute push (__attribute__((abi_tag("a"))), apply_to = any(enum_constant, function, record(unless(is_union)), variable, variable(is_parameter), enum))
// FIXME: comma in this diagnostic is wrong.
// expected-error@-2 {{attribute 'abi_tag' can't be applied to 'enum_constant', and 'variable(is_parameter)'}}
// expected-error@-2 {{attribute 'abi_tag' can't be applied to 'enum_constant', and 'enum'}}
#pragma clang attribute pop
#pragma clang attribute push (__attribute__((abi_tag("a"))), apply_to = any(function, record(unless(is_union)), enum))