Merge commit '3621a6c08002c6b3e5b6f91bb0e20d8372613160' into dev
This commit is contained in:
@@ -1,9 +1,20 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import test_lite
|
||||
import test_iter
|
||||
import test_skipdata
|
||||
import test_customized_mnem
|
||||
import test_compatibility_layer
|
||||
|
||||
test_lite.test_class()
|
||||
test_skipdata.test_class()
|
||||
test_customized_mnem.test()
|
||||
errors = []
|
||||
errors.extend(test_lite.test_class())
|
||||
errors.extend(test_iter.test_class())
|
||||
errors.extend(test_skipdata.test_class())
|
||||
errors.extend(test_customized_mnem.test())
|
||||
errors.extend(test_compatibility_layer.test_compatibility())
|
||||
|
||||
if errors:
|
||||
print("Some errors happened. Please check the output")
|
||||
for error in errors:
|
||||
print(error)
|
||||
exit(1)
|
||||
|
||||
73
external/capstone/bindings/python/tests/test_compatibility_layer.py
vendored
Normal file
73
external/capstone/bindings/python/tests/test_compatibility_layer.py
vendored
Normal file
@@ -0,0 +1,73 @@
|
||||
#!/usr/bin/env python3
|
||||
# Copyright © 2024 Peace-Maker <peacemakerctf@gmail.com>
|
||||
# SPDX-License-Identifier: BSD-3
|
||||
|
||||
import capstone.arm64
|
||||
import capstone.sysz_const
|
||||
from capstone import *
|
||||
import capstone.aarch64
|
||||
import capstone.arm
|
||||
import capstone.systemz
|
||||
from xprint import to_hex
|
||||
|
||||
|
||||
AARCH64_CODE = b"\x21\x7c\x02\x9b\x21\x7c\x00\x53\x00\x40\x21\x4b\xe1\x0b\x40\xb9"
|
||||
SYSZ_CODE = b"\xed\x00\x00\x00\x00\x1a\x5a\x0f\x1f\xff\xc2\x09\x80\x00\x00\x00\x07\xf7\xeb\x2a\xff\xff\x7f\x57\xe3\x01\xff\xff\x7f\x57\xeb\x00\xf0\x00\x00\x24\xb2\x4f\x00\x78"
|
||||
|
||||
|
||||
all_tests = (
|
||||
(CS_ARCH_ARM64, CS_MODE_ARM, AARCH64_CODE, "ARM64"),
|
||||
(CS_ARCH_SYSZ, CS_MODE_BIG_ENDIAN, SYSZ_CODE, "SysZ"),
|
||||
)
|
||||
|
||||
|
||||
# Test arm64 and sysz compatibility layer
|
||||
def test_compatibility():
|
||||
errors = []
|
||||
for arch, mode, code, comment in all_tests:
|
||||
print("*" * 16)
|
||||
print("Platform: %s" % comment)
|
||||
print("Code: %s" % to_hex(code))
|
||||
print("Disasm:")
|
||||
|
||||
try:
|
||||
md = Cs(arch, mode)
|
||||
md.detail = True
|
||||
|
||||
for insn in md.disasm(code, 0x1000):
|
||||
print("0x%x:\t%s\t%s\t(cc: %s)" % (insn.address, insn.mnemonic, insn.op_str, insn.cc))
|
||||
|
||||
print("0x%x:" % (insn.address + insn.size))
|
||||
print()
|
||||
except CsError as e:
|
||||
print("ERROR: %s" % e)
|
||||
errors.append(str(e))
|
||||
|
||||
# Test ARM64_ constants
|
||||
print("arm64.ARM64_CC_AL = %d" % capstone.arm64.ARM64_CC_AL)
|
||||
print("aarch64.AArch64CC_AL = %d" % capstone.aarch64.AArch64CC_AL)
|
||||
assert capstone.arm64.ARM64_CC_AL == capstone.aarch64.AArch64CC_AL
|
||||
print("arm64.ARM64_INS_FDIV = %d" % capstone.arm64.ARM64_INS_FDIV)
|
||||
print("aarch64.AARCH64_INS_FDIV = %d" % capstone.aarch64.AARCH64_INS_FDIV)
|
||||
assert capstone.arm64.ARM64_INS_FDIV == capstone.aarch64.AARCH64_INS_FDIV
|
||||
|
||||
# Test SYSZ_ constants
|
||||
print("systemz.SYSZ_INS_LG = %d" % capstone.systemz.SYSZ_INS_LG)
|
||||
print("systemz.SYSTEMZ_INS_LG = %d" % capstone.systemz.SYSTEMZ_INS_LG)
|
||||
assert capstone.systemz.SYSZ_INS_LG == capstone.systemz.SYSTEMZ_INS_LG
|
||||
|
||||
# Test ARM_CC_ constants
|
||||
print("arm.ARM_CC_MI = %d" % capstone.arm.ARM_CC_MI)
|
||||
print("arm.ARMCC_MI = %d" % capstone.arm.ARMCC_MI)
|
||||
assert capstone.arm.ARM_CC_MI == capstone.arm.ARMCC_MI
|
||||
print("arm.ARM_CC_INVALID = %d" % capstone.arm.ARM_CC_INVALID)
|
||||
print("arm.ARMCC_Invalid = %d" % capstone.arm.ARMCC_Invalid)
|
||||
assert capstone.arm.ARM_CC_INVALID == capstone.arm.ARMCC_Invalid
|
||||
|
||||
return errors
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
if test_compatibility():
|
||||
print("Some errors happened. Please check the output")
|
||||
exit(1)
|
||||
@@ -18,6 +18,7 @@ def print_insn(md, code):
|
||||
|
||||
|
||||
def test():
|
||||
errors = []
|
||||
try:
|
||||
md = Cs(CS_ARCH_X86, CS_MODE_32)
|
||||
|
||||
@@ -33,7 +34,11 @@ def test():
|
||||
print_insn(md, X86_CODE32)
|
||||
except CsError as e:
|
||||
print("ERROR: %s" % e)
|
||||
errors.append(str(e))
|
||||
return errors
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
test()
|
||||
if test():
|
||||
print("Some errors happened. Please check the output")
|
||||
exit(1)
|
||||
|
||||
@@ -56,7 +56,7 @@ all_tests = (
|
||||
(CS_ARCH_PPC, CS_MODE_BIG_ENDIAN + CS_MODE_QPX, PPC_CODE2, "PPC-64 + QPX", CS_OPT_SYNTAX_NOREGNAME),
|
||||
(CS_ARCH_SPARC, CS_MODE_BIG_ENDIAN, SPARC_CODE, "Sparc", None),
|
||||
(CS_ARCH_SPARC, CS_MODE_BIG_ENDIAN + CS_MODE_V9, SPARCV9_CODE, "SparcV9", None),
|
||||
(CS_ARCH_SYSTEMZ, 0, SYSZ_CODE, "SystemZ", None),
|
||||
(CS_ARCH_SYSTEMZ, CS_MODE_BIG_ENDIAN, SYSZ_CODE, "SystemZ", None),
|
||||
(CS_ARCH_XCORE, 0, XCORE_CODE, "XCore", None),
|
||||
(CS_ARCH_M68K, CS_MODE_BIG_ENDIAN | CS_MODE_M68K_040, M68K_CODE, "M68K (68040)", None),
|
||||
(CS_ARCH_M680X, CS_MODE_M680X_6809, M680X_CODE, "M680X_M6809", None),
|
||||
@@ -70,6 +70,7 @@ all_tests = (
|
||||
|
||||
# ## Test class Cs
|
||||
def test_class():
|
||||
errors = []
|
||||
for (arch, mode, code, comment, syntax) in all_tests:
|
||||
print('*' * 16)
|
||||
print("Platform: %s" % comment)
|
||||
@@ -89,6 +90,10 @@ def test_class():
|
||||
print()
|
||||
except CsError as e:
|
||||
print("ERROR: %s" % e)
|
||||
errors.append(str(e))
|
||||
return errors
|
||||
|
||||
if __name__ == '__main__':
|
||||
test_class()
|
||||
if test_class():
|
||||
print("Some errors happened. Please check the output")
|
||||
exit(1)
|
||||
|
||||
@@ -50,7 +50,7 @@ all_tests = (
|
||||
(CS_ARCH_PPC, CS_MODE_BIG_ENDIAN + CS_MODE_QPX, PPC_CODE2, "PPC-64 + QPX", CS_OPT_SYNTAX_NOREGNAME),
|
||||
(CS_ARCH_SPARC, CS_MODE_BIG_ENDIAN, SPARC_CODE, "Sparc", None),
|
||||
(CS_ARCH_SPARC, CS_MODE_BIG_ENDIAN + CS_MODE_V9, SPARCV9_CODE, "SparcV9", None),
|
||||
(CS_ARCH_SYSTEMZ, 0, SYSZ_CODE, "SystemZ", None),
|
||||
(CS_ARCH_SYSTEMZ, CS_MODE_BIG_ENDIAN, SYSZ_CODE, "SystemZ", None),
|
||||
(CS_ARCH_XCORE, 0, XCORE_CODE, "XCore", None),
|
||||
(CS_ARCH_M68K, CS_MODE_BIG_ENDIAN | CS_MODE_M68K_040, M68K_CODE, "M68K (68040)", None),
|
||||
(CS_ARCH_M680X, CS_MODE_M680X_6809, M680X_CODE, "M680X_M6809", None),
|
||||
@@ -71,6 +71,7 @@ def test_cs_disasm_quick():
|
||||
|
||||
# ## Test class Cs
|
||||
def test_class():
|
||||
errors = []
|
||||
for (arch, mode, code, comment, syntax) in all_tests:
|
||||
print('*' * 16)
|
||||
print("Platform: %s" % comment)
|
||||
@@ -90,9 +91,13 @@ def test_class():
|
||||
print()
|
||||
except CsError as e:
|
||||
print("ERROR: %s" % e)
|
||||
errors.append(str(e))
|
||||
return errors
|
||||
|
||||
|
||||
# test_cs_disasm_quick()
|
||||
# print "*" * 40
|
||||
if __name__ == '__main__':
|
||||
test_class()
|
||||
if test_class():
|
||||
print("Some errors happened. Please check the output")
|
||||
exit(1)
|
||||
|
||||
@@ -23,6 +23,7 @@ def testcb(buffer, size, offset, userdata):
|
||||
|
||||
# ## Test class Cs
|
||||
def test_class():
|
||||
errors = []
|
||||
for (arch, mode, code, comment, syntax) in all_tests:
|
||||
print('*' * 16)
|
||||
print("Platform: %s" %comment)
|
||||
@@ -65,7 +66,11 @@ def test_class():
|
||||
print
|
||||
except CsError as e:
|
||||
print("ERROR: %s" % e)
|
||||
errors.append(str(e))
|
||||
return errors
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
test_class()
|
||||
if test_class():
|
||||
print("Some errors happened. Please check the output")
|
||||
exit(1)
|
||||
|
||||
Reference in New Issue
Block a user