mirror of
https://github.com/bitdefender/bddisasm.git
synced 2024-11-29 19:08:06 +00:00
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
|
#
|
||
|
|
# Copyright (c) 2020 Bitdefender
|
||
|
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
#
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import glob
|
||
|
|
|
||
|
|
total_tests = 0
|
||
|
|
failed_tests = 0
|
||
|
|
|
||
|
|
def test_dir(dir):
|
||
|
|
global total_tests
|
||
|
|
global failed_tests
|
||
|
|
|
||
|
|
for f in glob.glob('%s\\*' % dir):
|
||
|
|
if -1 == f.find('.'):
|
||
|
|
if 0 < f.find('_16'):
|
||
|
|
mod = '-b16'
|
||
|
|
elif 0 < f.find('_32'):
|
||
|
|
mod = '-b32'
|
||
|
|
else:
|
||
|
|
mod = '-b64'
|
||
|
|
if 0 < f.find('_r0'):
|
||
|
|
mod += ' -k'
|
||
|
|
|
||
|
|
print(' * Running test case %s...' % f)
|
||
|
|
os.system('disasm -shemu %s -f %s >%s.temp' % (mod, f, f))
|
||
|
|
try:
|
||
|
|
res = open('%s.result' % f).read()
|
||
|
|
except:
|
||
|
|
print(' ! No result file provided for test %s!' % f)
|
||
|
|
|
||
|
|
try:
|
||
|
|
tmp = open('%s.temp' % f).read()
|
||
|
|
except:
|
||
|
|
print(' ! No result produced by test %s!' % f)
|
||
|
|
|
||
|
|
total_tests += 1
|
||
|
|
if res != tmp:
|
||
|
|
print(' **** FAILED! ****')
|
||
|
|
failed_tests += 1
|
||
|
|
else:
|
||
|
|
print(' * Passed.')
|
||
|
|
|
||
|
|
for f in glob.glob('%s\\*_decoded.bin' % dir):
|
||
|
|
os.remove(f)
|
||
|
|
for f in glob.glob('%s\\*.temp' % dir):
|
||
|
|
os.remove(f)
|
||
|
|
|
||
|
|
for dn in glob.glob("*"):
|
||
|
|
if not os.path.isdir(dn):
|
||
|
|
continue
|
||
|
|
print('Testing %s...' % dn)
|
||
|
|
test_dir(dn)
|
||
|
|
print("Ran %d tests, %d failed" % (total_tests, failed_tests))
|