From 42e215b4693dff5dd2c4cad9f6f8a059a29d1740 Mon Sep 17 00:00:00 2001 From: Andrew Lamoureux Date: Wed, 2 Feb 2022 10:58:09 -0500 Subject: Save .isDefault from EnumerationMember issue #2896 --- python/types.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'python/types.py') 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 -- cgit v1.3.1