gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-12 15:49:46 -08:00
Code Issues Packages Projects Releases Wiki Activity

include an address' parent in comparisons

Browse Source
This commit is contained in:
Yacine Elhamer
2023-08-03 11:21:58 +01:00
parent 3c3205adf1
commit 4277b4bef8
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
capa/features/address.py
View File

@@ -66,6 +66,7 @@ class ProcessAddress(Address):
return (self.ppid, self.pid) == (other.ppid, other.pid)
def __lt__(self, other):
assert isinstance(other, ProcessAddress)
return (self.ppid, self.pid) < (other.ppid, other.pid)
@@ -85,11 +86,11 @@ class ThreadAddress(Address):
def __eq__(self, other):
assert isinstance(other, ThreadAddress)
return self.tid == other.tid
return (self.process, self.tid) == (other.process, other.tid)
def __lt__(self, other):
assert isinstance(other, ThreadAddress)
return self.tid < other.tid
return (self.process, self.tid) < (other.process, other.tid)
class CallAddress(Address):
@@ -108,11 +109,11 @@ class CallAddress(Address):
def __eq__(self, other):
assert isinstance(other, CallAddress)
return self.id == other.id
return (self.thread, self.id) == (other.thread, other.id)
def __lt__(self, other):
assert isinstance(other, CallAddress)
return self.id < other.id
return (self.thread, self.id) < (other.thread, other.id)
class DynamicReturnAddress(Address):
@@ -131,11 +132,11 @@ class DynamicReturnAddress(Address):
def __eq__(self, other):
assert isinstance(other, DynamicReturnAddress)
return self.return_address == other.return_address
return (self.call, self.return_address) == other.call, other.return_address)
def __lt__(self, other):
assert isinstance(other, DynamicReturnAddress)
return self.return_address < other.return_address
return (self.call, self.return_address) < (other.call, other.return_address)
class RelativeVirtualAddress(int, Address):