tools: ynl: default to treating enums as flags for mask generation

I was a bit too optimistic in commit bf51d27704 ("tools: ynl: fix
get_mask utility routine"), not every mask we use is necessarily
coming from an enum of type "flags". We also allow flipping an
enum into flags on per-attribute basis. That's done by
the 'enum-as-flags' property of an attribute.

Restore this functionality, it's not currently used by any in-tree
family.

Signed-off-by: Jakub Kicinski <kuba@kernel.org>
Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
Jakub Kicinski 2023-03-24 12:03:56 -07:00 committed by David S. Miller
parent a504d246d2
commit 4c6170d1ae
2 changed files with 6 additions and 5 deletions

View File

@ -90,8 +90,8 @@ class SpecEnumEntry(SpecElement):
def raw_value(self):
return self.value
def user_value(self):
if self.enum_set['type'] == 'flags':
def user_value(self, as_flags=None):
if self.enum_set['type'] == 'flags' or as_flags:
return 1 << self.value
else:
return self.value
@ -136,10 +136,10 @@ class SpecEnumSet(SpecElement):
return True
return False
def get_mask(self):
def get_mask(self, as_flags=None):
mask = 0
for e in self.entries.values():
mask += e.user_value()
mask += e.user_value(as_flags)
return mask

View File

@ -254,7 +254,8 @@ class TypeScalar(Type):
def _attr_policy(self, policy):
if 'flags-mask' in self.checks or self.is_bitfield:
if self.is_bitfield:
mask = self.family.consts[self.attr['enum']].get_mask()
enum = self.family.consts[self.attr['enum']]
mask = enum.get_mask(as_flags=True)
else:
flags = self.family.consts[self.checks['flags-mask']]
flag_cnt = len(flags['entries'])