diff options
| author | Ajax <commial@gmail.com> | 2017-04-27 08:12:08 +0200 |
|---|---|---|
| committer | Ajax <commial@gmail.com> | 2017-04-27 08:12:08 +0200 |
| commit | b9b4a515f09d8e979c77503f91e652350e6df24a (patch) | |
| tree | 52fee60d4c2a8c9e8cbd1b575a13aebac4de4e44 | |
| parent | ca1482ebc82e768f6113eac90cfe56e82a720cd7 (diff) | |
| download | miasm-b9b4a515f09d8e979c77503f91e652350e6df24a.tar.gz miasm-b9b4a515f09d8e979c77503f91e652350e6df24a.zip | |
Add .items API to AssignBlock
| -rw-r--r-- | miasm2/ir/ir.py | 3 | ||||
| -rw-r--r-- | test/ir/ir.py | 1 |
2 files changed, 4 insertions, 0 deletions
diff --git a/miasm2/ir/ir.py b/miasm2/ir/ir.py index bc23d471..e5f0c8db 100644 --- a/miasm2/ir/ir.py +++ b/miasm2/ir/ir.py @@ -139,6 +139,9 @@ class AssignBlock(object): for dst, src in self._assigns.iteritems(): yield dst, src + def items(self): + return [(dst, src) for dst, src in self.iteritems()] + def itervalues(self): for src in self._assigns.itervalues(): yield src diff --git a/test/ir/ir.py b/test/ir/ir.py index 5c428a94..05936d75 100644 --- a/test/ir/ir.py +++ b/test/ir/ir.py @@ -37,6 +37,7 @@ assert assignblk1.get_rw() == {id_a: set([id_b])} assert assignblk1.keys() == [id_a] assert dict(assignblk1) == {id_a: id_b} assert assignblk1[id_a] == id_b +assert list(assignblk1.iteritems()) == assignblk1.items() ## Simplify assignblk3 = AssignBlock({id_a: id_b - id_b}) |