1
1
mirror of https://github.com/henrydcase/pqc.git synced 2024-11-22 23:48:58 +00:00

Add environment to sanitizers on ARM

This commit is contained in:
Thom Wiggers 2019-02-27 12:58:00 +01:00
parent 59792522ae
commit 1180de5d30
No known key found for this signature in database
GPG Key ID: 001BB0A7CE26E363
2 changed files with 26 additions and 6 deletions

View File

@ -1,11 +1,17 @@
import os
import subprocess import subprocess
def run_subprocess(command, working_dir='.', expected_returncode=0): def run_subprocess(command, working_dir='.', env=None, expected_returncode=0):
""" """
Helper function to run a shell command and report success/failure Helper function to run a shell command and report success/failure
depending on the exit status of the shell command. depending on the exit status of the shell command.
""" """
if env is not None:
env_ = os.environ.copy()
env_.update(env)
env = env_
# Note we need to capture stdout/stderr from the subprocess, # Note we need to capture stdout/stderr from the subprocess,
# then print it, which nose/unittest will then capture and # then print it, which nose/unittest will then capture and
# buffer appropriately # buffer appropriately
@ -13,7 +19,8 @@ def run_subprocess(command, working_dir='.', expected_returncode=0):
command, command,
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.STDOUT, stderr=subprocess.STDOUT,
cwd=working_dir cwd=working_dir,
env=env,
) )
print(working_dir + " > " + " ".join(command)) print(working_dir + " > " + " ".join(command))
print(result.stdout.decode('utf-8')) print(result.stdout.decode('utf-8'))
@ -21,7 +28,7 @@ def run_subprocess(command, working_dir='.', expected_returncode=0):
return result.stdout.decode('utf-8') return result.stdout.decode('utf-8')
def make(*args, working_dir='.', **kwargs): def make(*args, working_dir='.', env=None, **kwargs):
""" """
Runs a make target in the specified working directory Runs a make target in the specified working directory
@ -36,4 +43,5 @@ def make(*args, working_dir='.', **kwargs):
*['{}={}'.format(k, v) for k, v in kwargs.items()], *['{}={}'.format(k, v) for k, v in kwargs.items()],
], ],
working_dir=working_dir, working_dir=working_dir,
env=env,
) )

View File

@ -40,21 +40,33 @@ def check_functest(scheme_name, implementation_name):
def check_functest_sanitizers(scheme_name, implementation_name): def check_functest_sanitizers(scheme_name, implementation_name):
if platform.machine() not in ['i386', 'x86_64']: env = None
if platform.machine() == 'powerpc':
raise unittest.SkipTest() raise unittest.SkipTest()
elif platform.machine() in ['arm32', 'aarch64']:
env = {'ASAN_OPTIONS': 'detect_leaks=0'}
else:
print("Supported platform: {}".format(platform.machine))
implementation = pqclean.Implementation.by_name( implementation = pqclean.Implementation.by_name(
scheme_name, implementation_name) scheme_name, implementation_name)
helpers.make('clean-scheme', 'functest', helpers.make('clean-scheme', 'functest',
TYPE=implementation.scheme.type, TYPE=implementation.scheme.type,
SCHEME=scheme_name, SCHEME=scheme_name,
IMPLEMENTATION=implementation_name, IMPLEMENTATION=implementation_name,
EXTRAFLAGS='-fsanitize=address,undefined',
working_dir=os.path.join('..', 'test'), working_dir=os.path.join('..', 'test'),
EXTRAFLAGS='-fsanitize=address,undefined') env=env)
helpers.run_subprocess( helpers.run_subprocess(
['./functest_{}_{}'.format(scheme_name, implementation_name)], ['./functest_{}_{}'.format(scheme_name, implementation_name)],
os.path.join('..', 'bin'), os.path.join('..', 'bin'),
env=env,
) )
return check_functest(scheme_name, implementation_name) # Remove files with ASAN library compiled in
helpers.make('clean-scheme',
TYPE=implementation.scheme.type,
SCHEME=scheme_name,
IMPLEMENTATION=implementation_name,
working_dir=os.path.join('..', 'test'))
if __name__ == '__main__': if __name__ == '__main__':