summaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
Diffstat (limited to 'python')
-rw-r--r--python/types.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/python/types.py b/python/types.py
index fae9d611..03ee3733 100644
--- a/python/types.py
+++ b/python/types.py
@@ -1199,6 +1199,7 @@ class StructureBuilder(TypeBuilder):
class EnumerationMember:
name: str
value: Optional[int] = None
+ isDefault: Optional[bool] = None
def __repr__(self):
value = f"{self.value:#x}" if self.value is not None else "auto()"
@@ -1258,7 +1259,7 @@ class EnumerationBuilder(TypeBuilder):
try:
for i in range(count.value):
result.append(
- EnumerationMember(members[i].name, members[i].value if not members[i].isDefault else None)
+ EnumerationMember(members[i].name, members[i].value if not members[i].isDefault else None, members[i].isDefault)
)
return result
finally:
@@ -2099,7 +2100,7 @@ class EnumerationType(IntegerType):
assert members is not None, "core.BNGetEnumerationMembers returned None"
result = []
for i in range(0, count.value):
- result.append(EnumerationMember(members[i].name, members[i].value))
+ result.append(EnumerationMember(members[i].name, members[i].value, members[i].isDefault))
core.BNFreeEnumerationMemberList(members, count.value)
return result