diff --git a/capa/features/address.py b/capa/features/address.py index d716aa2d..251b498a 100644 --- a/capa/features/address.py +++ b/capa/features/address.py @@ -32,6 +32,9 @@ class AbsoluteVirtualAddress(int, Address): def __repr__(self): return f"absolute(0x{self:x})" + def __hash__(self): + return int.__hash__(self) + class RelativeVirtualAddress(int, Address): """a memory address relative to a base address""" @@ -39,6 +42,9 @@ class RelativeVirtualAddress(int, Address): def __repr__(self): return f"relative(0x{self:x})" + def __hash__(self): + return int.__hash__(self) + class FileOffsetAddress(int, Address): """an address relative to the start of a file""" @@ -50,6 +56,9 @@ class FileOffsetAddress(int, Address): def __repr__(self): return f"file(0x{self:x})" + def __hash__(self): + return int.__hash__(self) + class DNTokenAddress(int, Address): """a .NET token""" @@ -60,6 +69,9 @@ class DNTokenAddress(int, Address): def __repr__(self): return f"token(0x{self:x})" + def __hash__(self): + return int.__hash__(self) + class DNTokenOffsetAddress(Address): """an offset into an object specified by a .NET token""" diff --git a/setup.py b/setup.py index be2bcc6a..1043abc5 100644 --- a/setup.py +++ b/setup.py @@ -76,7 +76,7 @@ setuptools.setup( "pycodestyle==2.9.1", "black==22.10.0", "isort==5.10.1", - "mypy==0.971", + "mypy==0.982", "psutil==5.9.2", "stix2==3.0.1", "requests==2.28.0",