Virtualization
/
capstone
mirror of https://gitlab.com/qemu-project/capstone.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

python: cs_disasm*() now use generator rather than a list to return disassembled instructions. suggested by Mario Vilas.

This commit is contained in:
Nguyen Anh Quynh 2013-12-31 22:55:27 +08:00
parent bed2b56c80
commit 591d98cb24
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bindings/python/capstone/capstone.py
View File

@ -231,15 +231,15 @@ def cs_disasm_quick(arch, mode, code, offset, count = 0):
if status != CS_ERR_OK:
raise CsError(status)
insns = []
all_insn = ctypes.POINTER(_cs_insn)()
res = _cs.cs_disasm_ex(csh, code, len(code), offset, count, ctypes.byref(all_insn))
if res > 0:
for i in xrange(res):
insns.append(CsInsn(_dummy_cs(csh, arch), all_insn[i]))
yield CsInsn(_dummy_cs(csh, arch), all_insn[i])
_cs.cs_free(all_insn, res)
else:
yield []
status = _cs.cs_errno(csh)
if status != CS_ERR_OK:
raise CsError(status)
@ -248,7 +248,6 @@ def cs_disasm_quick(arch, mode, code, offset, count = 0):
if status != CS_ERR_OK:
raise CsError(status)
return insns
# Python-style class to disasm code
class CsInsn(object):
@ -389,17 +388,15 @@ class Cs(object):
self._mode = opt
def disasm(self, code, offset, count = 0):
insns = []
all_insn = ctypes.POINTER(_cs_insn)()
res = _cs.cs_disasm_ex(self.csh, code, len(code), offset, count, ctypes.byref(all_insn))
if res > 0:
for i in xrange(res):
insns.append(CsInsn(self, all_insn[i]))
yield CsInsn(self, all_insn[i])
_cs.cs_free(all_insn, res)
else:
yield []
status = _cs.cs_errno(self.csh)
if status != CS_ERR_OK:
raise CsError(status)
return insns