提交 e4a0a85f 编写于 作者: Y Yinan Xu

tests,crypto: code clean up

上级 baab89e2
import os
files = os.listdir('src')
srcfile = 'src/crypto.S'
fd = open(srcfile, 'w')
for f in files:
record = 0
fs = open('src/'+f, 'r+')
for line in fs.readlines():
if 'RVTEST_CODE_BEGIN' in line:
record = 1
elif 'RVMODEL_HALT' in line:
record = 0
if record :
if 'inst_' in line:
fd.write(f[0:-2].lower().replace('-','_')+'_')
fd.write(line.lstrip())
else:
fd.write(line)
fs.close()
fd.close()
import os
spath = './tmp/'
dpath = './src/'
files = os.listdir(spath)
for f in files:
prefix = f[0:-2].lower().replace('-','_')
entry = prefix + '_entry'
init = prefix + '_init'
code_begin = prefix + '_code_begin'
code_end = prefix + '_code_end'
fs = open(spath+f, 'r+')
fd = open(dpath+f, 'w')
record = 1
for line in fs.readlines():
# if ('RVTEST_SIGBASE' in line): #('RVTEST_CASE' in line) |
# continue
# if 'RVTEST_CODE_END' in line:
# record = 0
# if record:
if '.globl rvtest_entry_point' in line:
fd.write('.globl ' + entry + '\n')
elif 'rvtest_entry_point:' in line:
fd.write(entry+':\n')
elif 'RVTEST_CODE_BEGIN' in line:
fd.write('RVTEST_CODE_BEGIN ' + init + ' ' + code_begin + '\n')
elif 'RVTEST_CODE_END' in line:
fd.write('RVTEST_CODE_END ' + code_end + '\n')
elif '.section .text.init' in line:
fd.write(line)
fd.write('.globl main\n')
fd.write('main:\n')
else:
fd.write(line)
# fd.write('mv a0, x0;\n')
# fd.write('.word 0x0000006b\n')
fs.close()
fd.close()
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
此差异已折叠。
# Generate riscv-crypto assembly code(sm4 instructions is machine code)
# Need to modify the file($AM_HOME/am/arch/isa/riscv64.mk) as follow:
# -CROSS_COMPILE := riscv64-linux-gnu-
# -COMMON_FLAGS := -fno-pic -march=rv64g -mcmodel=medany
# +CROSS_COMPILE := /bigdata/zfw/bk-tools/bin/riscv64-unknown-linux-gnu-
# +COMMON_FLAGS := -fno-pic -march=rv64gc_zba_zbb_zbc_zbs_zbkb_zbkc_zbkx_zknd_zkne_zknh_zkr_zksed_zksh_zkt -mcmodel=medany
#
# Run "python3 randtest.py" to generate random test
# Run "make ARCH=riscv64-noop" to compile
import argparse
import random
......
import random
import os
out = 'sm4random.S'
fout = open(out, 'w')
fout.write('#include "model_test.h"\n')
fout.write('#include "arch_test.h"\n')
fout.write('.section .text.init\n')
fout.write('.global _trm_init\n')
fout.write('_trm_init:\n')
fout.write(' j START_RAMDOM_TEST\n')
fout.write('START_RAMDOM_TEST:\n')
# initial
for r in range(1,32):
rand_data = hex(random.randrange(2 ** 64))
fout.write('li x' + str(r) + ',' + str(rand_data) + '\n')
for i in range(0, 10 ** 4):
mask = (3 << 28) + (3 << 4) + 3
isks = random.randrange(2) << 26
bs = random.randrange(4) << 30
rd = random.randrange(32) << 7
rs1 = random.randrange(32) << 15
rs2 = random.randrange(32) << 20
code = str(hex(mask + bs + rd + rs1 + rs2 + isks))
fout.write('.int ' + code + '\n')
fout.write(' mv a0, x0\n')
fout.write(' .int 0x0000006b\n')
fout.close()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册