mirror of
https://github.com/henrydcase/pqc.git
synced 2024-11-27 01:41:40 +00:00
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
"""
|
|
Runs the test files under valgrind to detect memory problems
|
|
"""
|
|
|
|
import os
|
|
import platform
|
|
import unittest
|
|
|
|
import pqclean
|
|
import helpers
|
|
|
|
|
|
def test_functest():
|
|
for scheme in pqclean.Scheme.all_schemes():
|
|
for implementation in scheme.implementations:
|
|
if helpers.permit_test('valgrind', implementation):
|
|
yield check_valgrind, implementation
|
|
|
|
|
|
def check_valgrind(implementation: pqclean.Implementation):
|
|
if (platform.machine() not in ('i386', 'x86_64') or
|
|
platform.system() != 'Linux'):
|
|
raise unittest.SkipTest()
|
|
if os.environ.get('CIRCLECI') == 'true' and os.environ.get('ARCH') == 'i386' and (implementation.scheme.name == 'frodokem1344aes' or implementation.scheme.name == 'frodokem1344shake'):
|
|
raise unittest.SkipTest('Skipping too-slow valgrind test')
|
|
|
|
helpers.make(TYPE=implementation.scheme.type,
|
|
SCHEME=implementation.scheme.name,
|
|
IMPLEMENTATION=implementation.name,
|
|
working_dir=os.path.join('..', 'test'))
|
|
functest_name = './functest_{}_{}'.format(implementation.scheme.name,
|
|
implementation.name)
|
|
helpers.run_subprocess(['valgrind', functest_name],
|
|
os.path.join('..', 'bin'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
import nose2
|
|
nose2.main()
|
|
except ImportError:
|
|
import nose
|
|
nose.runmodule()
|