mirror of
https://github.com/capstone-engine/capstone.git
synced 2024-11-23 05:29:53 +00:00
af1ed2fb3d
Some checks failed
Run Test / ${{ matrix.config.name }} (map[arch:x64 build-system:cmake diet-build:OFF enable-asan:OFF name:ubuntu-22.04 x64 cmake os:ubuntu-22.04]) (push) Failing after 0s
Run Test / ${{ matrix.config.name }} (map[arch:x64 build-system:make diet-build:OFF enable-asan:OFF name:ubuntu-22.04 x64 make os:ubuntu-22.04]) (push) Failing after 0s
RELEASE BUILD - PyPI 📦 Distribution / Build wheels on ${{ matrix.os }} (ubuntu-latest) (push) Failing after 0s
RELEASE BUILD - PyPI 📦 Distribution / Make SDist (push) Failing after 0s
Run Test / ${{ matrix.config.name }} (map[arch:x64 build-system:cmake diet-build:OFF enable-asan:ON name:ubuntu-24.04 x64 ASAN os:ubuntu-24.04]) (push) Has been cancelled
Run Test / ${{ matrix.config.name }} (map[arch:x64 name:windows x64 MSVC 64bit os:windows-latest platform:windows python-arch:x64 python-version:3.9]) (push) Has been cancelled
Auto-Sync / check (push) Has been cancelled
Run clang-tidy / clang-tidy (push) Has been cancelled
RELEASE BUILD - PyPI 📦 Distribution / Build wheels on ${{ matrix.os }} (macos-latest) (push) Has been cancelled
RELEASE BUILD - PyPI 📦 Distribution / Build wheels on ${{ matrix.os }} (windows-latest) (push) Has been cancelled
Python Package CI / build (macOS-14, 3.12) (push) Has been cancelled
Python Package CI / build (macOS-14, 3.8) (push) Has been cancelled
Python Package CI / build (ubuntu-24.04, 3.12) (push) Has been cancelled
Python Package CI / build (ubuntu-24.04, 3.8) (push) Has been cancelled
Python Package CI / build (windows-2022, 3.12) (push) Has been cancelled
Python Package CI / build (windows-2022, 3.8) (push) Has been cancelled
RELEASE BUILD - PyPI 📦 Distribution / publish (push) Has been cancelled
180 lines
7.6 KiB
Python
Executable File
180 lines
7.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Test tool to compare Capstone output with llvm-mc. By Nguyen Anh Quynh, 2014
|
|
import sys
|
|
import os
|
|
from capstone import *
|
|
from pathlib import Path
|
|
import codecs
|
|
|
|
|
|
def test_file(fname):
|
|
print("Test %s" % fname)
|
|
fpath = Path(fname) if isinstance(fname, str) else fname
|
|
if fpath.is_dir():
|
|
if fpath.exists() is False:
|
|
return
|
|
for f in fpath.iterdir():
|
|
test_file(f)
|
|
return
|
|
|
|
with fpath.open() as f:
|
|
lines = f.readlines()
|
|
|
|
if not lines[0].startswith('# '):
|
|
print("ERROR: decoding information is missing")
|
|
return
|
|
|
|
# skip '# ' at the front, then split line to get out hexcode
|
|
# Note: option can be '', or 'None'
|
|
# print lines[0]
|
|
# print lines[0][2:].split(', ')
|
|
(arch, mode, option) = lines[0][2:].split(', ')
|
|
mode = mode.replace(' ', '')
|
|
option = option.strip()
|
|
|
|
archs = {
|
|
"CS_ARCH_ARM": CS_ARCH_ARM,
|
|
"CS_ARCH_AARCH64": CS_ARCH_AARCH64,
|
|
"CS_ARCH_MIPS": CS_ARCH_MIPS,
|
|
"CS_ARCH_PPC": CS_ARCH_PPC,
|
|
"CS_ARCH_SPARC": CS_ARCH_SPARC,
|
|
"CS_ARCH_SYSTEMZ": CS_ARCH_SYSTEMZ,
|
|
"CS_ARCH_X86": CS_ARCH_X86,
|
|
"CS_ARCH_XCORE": CS_ARCH_XCORE,
|
|
"CS_ARCH_RISCV": CS_ARCH_RISCV,
|
|
"CS_ARCH_TRICORE": CS_ARCH_TRICORE,
|
|
"CS_ARCH_ALPHA": CS_ARCH_ALPHA,
|
|
"CS_ARCH_HPPA": CS_ARCH_HPPA,
|
|
}
|
|
|
|
modes = {
|
|
"CS_MODE_16": CS_MODE_16,
|
|
"CS_MODE_32": CS_MODE_32,
|
|
"CS_MODE_64": CS_MODE_64,
|
|
"CS_MODE_MIPS32": CS_MODE_MIPS32,
|
|
"CS_MODE_MIPS64": CS_MODE_MIPS64,
|
|
"0": CS_MODE_ARM,
|
|
"CS_MODE_ARM": CS_MODE_ARM,
|
|
"CS_MODE_THUMB": CS_MODE_THUMB,
|
|
"CS_MODE_ARM+CS_MODE_V8": CS_MODE_ARM + CS_MODE_V8,
|
|
"CS_MODE_THUMB+CS_MODE_V8": CS_MODE_THUMB + CS_MODE_V8,
|
|
"CS_MODE_THUMB+CS_MODE_MCLASS": CS_MODE_THUMB + CS_MODE_MCLASS,
|
|
"CS_MODE_THUMB+CS_MODE_V8+CS_MODE_MCLASS": CS_MODE_THUMB+CS_MODE_V8+CS_MODE_MCLASS,
|
|
"CS_MODE_LITTLE_ENDIAN": CS_MODE_LITTLE_ENDIAN,
|
|
"CS_MODE_BIG_ENDIAN": CS_MODE_BIG_ENDIAN,
|
|
"CS_MODE_64+CS_MODE_LITTLE_ENDIAN": CS_MODE_64 + CS_MODE_LITTLE_ENDIAN,
|
|
"CS_MODE_64+CS_MODE_BIG_ENDIAN": CS_MODE_64 + CS_MODE_BIG_ENDIAN,
|
|
"CS_MODE_MIPS32+CS_MODE_MICRO": CS_MODE_MIPS32 + CS_MODE_MICRO,
|
|
"CS_MODE_MIPS32+CS_MODE_MICRO+CS_MODE_BIG_ENDIAN": CS_MODE_MIPS32 + CS_MODE_MICRO + CS_MODE_BIG_ENDIAN,
|
|
"CS_MODE_MIPS32+CS_MODE_BIG_ENDIAN+CS_MODE_MICRO": CS_MODE_MIPS32 + CS_MODE_MICRO + CS_MODE_BIG_ENDIAN,
|
|
"CS_MODE_BIG_ENDIAN+CS_MODE_V9": CS_MODE_BIG_ENDIAN + CS_MODE_V9,
|
|
"CS_MODE_MIPS32+CS_MODE_BIG_ENDIAN": CS_MODE_MIPS32 + CS_MODE_BIG_ENDIAN,
|
|
"CS_MODE_MIPS32+CS_MODE_LITTLE_ENDIAN": CS_MODE_MIPS32 + CS_MODE_LITTLE_ENDIAN,
|
|
"CS_MODE_MIPS64+CS_MODE_LITTLE_ENDIAN": CS_MODE_MIPS64 + CS_MODE_LITTLE_ENDIAN,
|
|
"CS_MODE_MIPS64+CS_MODE_BIG_ENDIAN": CS_MODE_MIPS64 + CS_MODE_BIG_ENDIAN,
|
|
"CS_MODE_RISCV32": CS_MODE_RISCV32,
|
|
"CS_MODE_RISCV64": CS_MODE_RISCV64,
|
|
"CS_MODE_TRICORE_110": CS_MODE_TRICORE_110,
|
|
"CS_MODE_TRICORE_120": CS_MODE_TRICORE_120,
|
|
"CS_MODE_TRICORE_130": CS_MODE_TRICORE_130,
|
|
"CS_MODE_TRICORE_131": CS_MODE_TRICORE_131,
|
|
"CS_MODE_TRICORE_160": CS_MODE_TRICORE_160,
|
|
"CS_MODE_TRICORE_161": CS_MODE_TRICORE_161,
|
|
"CS_MODE_TRICORE_162": CS_MODE_TRICORE_162,
|
|
"CS_MODE_BIG_ENDIAN+CS_MODE_QPX": CS_MODE_BIG_ENDIAN+CS_MODE_QPX,
|
|
"CS_MODE_HPPA_11": CS_MODE_HPPA_11,
|
|
"CS_MODE_HPPA_20": CS_MODE_HPPA_20,
|
|
"CS_MODE_HPPA_20W": CS_MODE_HPPA_20W,
|
|
}
|
|
|
|
mc_modes = {
|
|
("CS_ARCH_X86", "CS_MODE_32"): 0,
|
|
("CS_ARCH_X86", "CS_MODE_64"): 1,
|
|
("CS_ARCH_ARM", "CS_MODE_ARM"): 2,
|
|
("CS_ARCH_ARM", "CS_MODE_THUMB"): 3,
|
|
("CS_ARCH_ARM", "CS_MODE_ARM+CS_MODE_V8"): 4,
|
|
("CS_ARCH_ARM", "CS_MODE_THUMB+CS_MODE_V8"): 5,
|
|
("CS_ARCH_ARM", "CS_MODE_THUMB+CS_MODE_MCLASS"): 6,
|
|
("CS_ARCH_ARM", "CS_MODE_THUMB+CS_MODE_V8+CS_MODE_MCLASS"): 7,
|
|
("CS_ARCH_AARCH64", "0"): 8,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32+CS_MODE_BIG_ENDIAN"): 9,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32+CS_MODE_MICRO"): 10,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS64"): 11,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32"): 12,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS64+CS_MODE_BIG_ENDIAN"): 13,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32+CS_MODE_MICRO+CS_MODE_BIG_ENDIAN"): 14,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32+CS_MODE_BIG_ENDIAN+CS_MODE_MICRO"): 14,
|
|
("CS_ARCH_PPC", "CS_MODE_BIG_ENDIAN"): 15,
|
|
("CS_ARCH_SPARC", "CS_MODE_BIG_ENDIAN"): 16,
|
|
("CS_ARCH_SPARC", "CS_MODE_BIG_ENDIAN+CS_MODE_V9"): 17,
|
|
("CS_ARCH_SYSTEMZ", "CS_MODE_BIG_ENDIAN"): 18,
|
|
("CS_ARCH_XCORE", "0"): 19,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32R6+CS_MODE_BIG_ENDIAN"): 20,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32R6+CS_MODE_MICRO+CS_MODE_BIG_ENDIAN"): 21,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32R6"): 22,
|
|
("CS_ARCH_MIPS", "CS_MODE_MIPS32R6+CS_MODE_MICRO"): 23,
|
|
("CS_ARCH_M68K", "0"): 24,
|
|
("CS_ARCH_M680X", "CS_MODE_M680X_6809"): 25,
|
|
("CS_ARCH_EVM", "0"): 26,
|
|
("CS_ARCH_BPF", "CS_MODE_LITTLE_ENDIAN+CS_MODE_BPF_CLASSIC"): 30,
|
|
("CS_ARCH_BPF", "CS_MODE_LITTLE_ENDIAN+CS_MODE_BPF_EXTENDED"): 31,
|
|
("CS_ARCH_BPF", "CS_MODE_BIG_ENDIAN+CS_MODE_BPF_CLASSIC"): 32,
|
|
("CS_ARCH_BPF", "CS_MODE_BIG_ENDIAN+CS_MODE_BPF_EXTENDED"): 33,
|
|
("CS_ARCH_RISCV", "CS_MODE_RISCV32"): 45,
|
|
("CS_ARCH_RISCV", "CS_MODE_RISCV64"): 46,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_110"): 47,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_120"): 48,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_130"): 49,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_131"): 50,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_160"): 51,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_161"): 52,
|
|
("CS_ARCH_TRICORE", "CS_MODE_TRICORE_162"): 53,
|
|
("CS_ARCH_PPC", "CS_MODE_BIG_ENDIAN+CS_MODE_QPX"): 54,
|
|
("CS_ARCH_ALPHA", "CS_MODE_LITTLE_ENDIAN"): 55,
|
|
("CS_ARCH_ALPHA", "CS_MODE_BIG_ENDIAN"): 56,
|
|
("CS_ARCH_HPPA", "CS_MODE_HPPA_11+CS_MODE_BIG_ENDIAN"): 57,
|
|
("CS_ARCH_HPPA", "CS_MODE_HPPA_20+CS_MODE_BIG_ENDIAN"): 58,
|
|
}
|
|
|
|
# if not option in ('', 'None'):
|
|
# print archs[arch], modes[mode], options[option]
|
|
|
|
for line in lines[1:]:
|
|
# ignore all the input lines having # in front.
|
|
if line.startswith('#'):
|
|
continue
|
|
if line.startswith('// '):
|
|
line = line[3:]
|
|
# print("Check %s" %line)
|
|
code = line.split(' = ')[0]
|
|
if len(code) < 2:
|
|
continue
|
|
if code.find('//') >= 0:
|
|
continue
|
|
hex_code = code.replace('0x', '').replace(',', '').replace(' ', '').strip()
|
|
try:
|
|
hex_data = bytes.fromhex(hex_code)
|
|
fpath = Path("fuzz/corpus/%s_%s" % (os.path.basename(fname), hex_code))
|
|
if fpath.parent.exists() is False:
|
|
fpath.parent.mkdir(parents=True)
|
|
with fpath.open('wb') as fout:
|
|
if (arch, mode) not in mc_modes:
|
|
print("fail", arch, mode)
|
|
if mode == "None":
|
|
mode = "0"
|
|
fout.write(mc_modes[(arch, mode)].to_bytes(1, 'little'))
|
|
fout.write(hex_data)
|
|
except Exception as e:
|
|
print(f"skipping: {hex_code} with: {e}")
|
|
continue
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) == 1:
|
|
fnames = sys.stdin.readlines()
|
|
for fname in fnames:
|
|
test_file(fname.strip())
|
|
else:
|
|
# print("Usage: ./test_mc.py <input-file.s.cs>")
|
|
test_file(sys.argv[1])
|