Check if the last word is 'MASK', not if it ends with 'MASK'

At the moment it makes no difference, but it could if e.g. a new
algorithm was called 'foomask'.

Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
Gilles Peskine 2021-01-25 22:44:36 +01:00
parent 13d60eb4fc
commit 0655b4f1e0

View file

@ -50,7 +50,7 @@ class PSAMacroCollector:
return True return True
if '_CATEGORY_' in name: if '_CATEGORY_' in name:
return True return True
return name.endswith('_FLAG') or name.endswith('MASK') return name.endswith('_FLAG') or name.endswith('_MASK')
# "#define" followed by a macro name with either no parameters # "#define" followed by a macro name with either no parameters
# or a single parameter and a non-empty expansion. # or a single parameter and a non-empty expansion.