| import functools |
| import hashlib |
| import unittest |
|
|
| try: |
| import _hashlib |
| except ImportError: |
| _hashlib = None |
|
|
|
|
| def requires_hashdigest(digestname, openssl=None, usedforsecurity=True): |
| """Decorator raising SkipTest if a hashing algorithm is not available |
| |
| The hashing algorithm could be missing or blocked by a strict crypto |
| policy. |
| |
| If 'openssl' is True, then the decorator checks that OpenSSL provides |
| the algorithm. Otherwise the check falls back to built-in |
| implementations. The usedforsecurity flag is passed to the constructor. |
| |
| ValueError: [digital envelope routines: EVP_DigestInit_ex] disabled for FIPS |
| ValueError: unsupported hash type md4 |
| """ |
| def decorator(func_or_class): |
| if isinstance(func_or_class, type): |
| setUpClass = func_or_class.__dict__.get('setUpClass') |
| if setUpClass is None: |
| def setUpClass(cls): |
| super(func_or_class, cls).setUpClass() |
| setUpClass.__qualname__ = func_or_class.__qualname__ + '.setUpClass' |
| setUpClass.__module__ = func_or_class.__module__ |
| else: |
| setUpClass = setUpClass.__func__ |
| setUpClass = classmethod(decorator(setUpClass)) |
| func_or_class.setUpClass = setUpClass |
| return func_or_class |
|
|
| @functools.wraps(func_or_class) |
| def wrapper(*args, **kwargs): |
| try: |
| if openssl and _hashlib is not None: |
| _hashlib.new(digestname, usedforsecurity=usedforsecurity) |
| else: |
| hashlib.new(digestname, usedforsecurity=usedforsecurity) |
| except ValueError: |
| raise unittest.SkipTest( |
| f"hash digest '{digestname}' is not available." |
| ) |
| return func_or_class(*args, **kwargs) |
| return wrapper |
| return decorator |
|
|
