MMGen
b5f1ebc640
- binascii.hexlify(b'foo') -> b'foo'.hex()
- binascii.unhexlify('aabb') -> bytes.fromhex('aabb')
- replace HexBytes class with HexStr
This change has led to a ≈10% speedup in the full test-release.sh run
123 lines
3.5 KiB
Python
Executable file
123 lines
3.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
#
|
|
# mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
|
|
# Copyright (C)2013-2019 The MMGen Project <mmgen@tuta.io>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
common.py: Shared routines and data for the MMGen test suites
|
|
"""
|
|
|
|
sample_text = 'The Times 03/Jan/2009 Chancellor on brink of second bailout for banks'
|
|
|
|
ref_kafile_pass = 'kafile password'
|
|
ref_kafile_hash_preset = '1'
|
|
|
|
class TestSuiteException(Exception): pass
|
|
class TestSuiteFatalException(Exception): pass
|
|
|
|
import os
|
|
from mmgen.common import *
|
|
|
|
def getrandnum(n): return int(os.urandom(n).hex(),16)
|
|
def getrandhex(n): return os.urandom(n).hex()
|
|
def getrandnum_range(nbytes,rn_max):
|
|
while True:
|
|
rn = int(os.urandom(nbytes).hex(),16)
|
|
if rn < rn_max: return rn
|
|
|
|
def getrandstr(num_chars,no_space=False):
|
|
n,m = 95,32
|
|
if no_space: n,m = 94,33
|
|
return ''.join([chr(i%n+m) for i in list(os.urandom(num_chars))])
|
|
|
|
# Windows uses non-UTF8 encodings in filesystem, so use raw bytes here
|
|
def cleandir(d,do_msg=False):
|
|
d_enc = d.encode()
|
|
|
|
try: files = os.listdir(d_enc)
|
|
except: return
|
|
|
|
from shutil import rmtree
|
|
if do_msg: gmsg("Cleaning directory '{}'".format(d))
|
|
for f in files:
|
|
try:
|
|
os.unlink(os.path.join(d_enc,f))
|
|
except:
|
|
rmtree(os.path.join(d_enc,f))
|
|
|
|
def mk_tmpdir(d):
|
|
try: os.mkdir(d,0o755)
|
|
except OSError as e:
|
|
if e.errno != 17: raise
|
|
else:
|
|
vmsg("Created directory '{}'".format(d))
|
|
|
|
# def mk_tmpdir_path(path,cfg):
|
|
# try:
|
|
# name = os.path.split(cfg['tmpdir'])[-1]
|
|
# src = os.path.join(path,name)
|
|
# try:
|
|
# os.unlink(cfg['tmpdir'])
|
|
# except OSError as e:
|
|
# if e.errno != 2: raise
|
|
# finally:
|
|
# os.mkdir(src)
|
|
# os.symlink(src,cfg['tmpdir'])
|
|
# except OSError as e:
|
|
# if e.errno != 17: raise
|
|
# else: msg("Created directory '{}'".format(cfg['tmpdir']))
|
|
|
|
def get_tmpfile(cfg,fn):
|
|
return os.path.join(cfg['tmpdir'],fn)
|
|
|
|
def write_to_file(fn,data,binary=False):
|
|
write_data_to_file( fn,
|
|
data,
|
|
silent = True,
|
|
binary = binary,
|
|
ignore_opt_outdir = True )
|
|
|
|
def write_to_tmpfile(cfg,fn,data,binary=False):
|
|
write_to_file( os.path.join(cfg['tmpdir'],fn), data=data, binary=binary )
|
|
|
|
def read_from_file(fn,binary=False):
|
|
from mmgen.util import get_data_from_file
|
|
return get_data_from_file(fn,silent=True,binary=binary)
|
|
|
|
def read_from_tmpfile(cfg,fn,binary=False):
|
|
return read_from_file(os.path.join(cfg['tmpdir'],fn),binary=binary)
|
|
|
|
def joinpath(*args,**kwargs):
|
|
return os.path.join(*args,**kwargs)
|
|
|
|
def ok():
|
|
if opt.profile: return
|
|
if opt.verbose or opt.exact_output:
|
|
gmsg('OK')
|
|
else: msg(' OK')
|
|
|
|
def cmp_or_die(s,t,desc=None):
|
|
if s != t:
|
|
m = 'ERROR: recoded data:\n{!r}\ndiffers from original data:\n{!r}'
|
|
if desc: m = 'For {}:\n{}'.format(desc,m)
|
|
raise TestSuiteFatalException(m.format(t,s))
|
|
|
|
def init_coverage():
|
|
coverdir = os.path.join('test','trace')
|
|
acc_file = os.path.join('test','trace.acc')
|
|
try: os.mkdir(coverdir,0o755)
|
|
except: pass
|
|
return coverdir,acc_file
|