from __future__ import absolute_import, print_function
"""
M2Crypto utility routines.
NOTHING IN THIS MODULE IS GUARANTEED TO BE STABLE, USED ONLY FOR
INTERNAL PURPOSES OF M2CRYPTO.
Copyright (c) 1999-2004 Ng Pheng Siong. All rights reserved.
Portions created by Open Source Applications Foundation (OSAF) are
Copyright (C) 2004 OSAF. All Rights Reserved.
"""
import binascii
import logging
import struct
import sys
import struct
import unittest
from M2Crypto import m2
from typing import Any, Optional, TextIO, Tuple, Union # noqa
# see https://github.com/python/typeshed/issues/222
AddrType = Union[Tuple[str, int], str]
log = logging.getLogger('util')
[docs]
class UtilError(Exception):
pass
m2.util_init(UtilError)
[docs]
def is_32bit():
# type: () -> bool
return (struct.calcsize("P") * 8) == 32
[docs]
def expectedFailureIf(condition):
"""The test is marked as an expectedFailure if the condition is satisfied."""
def wrapper(func):
if condition:
return unittest.expectedFailure(func)
else:
return func
return wrapper
[docs]
def pkcs5_pad(data, blklen=8):
# type: (str, int) -> str
pad = (8 - (len(data) % 8))
return data + chr(pad) * pad
[docs]
def pkcs7_pad(data, blklen):
# type: (str, int) -> str
if blklen > 255:
raise ValueError('illegal block size')
pad = (blklen - (len(data) % blklen))
return data + chr(pad) * pad
[docs]
def bin_to_hex(b):
# type: (bytes) -> str
return binascii.b2a_base64(b)[:-1].decode()
[docs]
def octx_to_num(x):
# type: (bytes) -> int
return int(binascii.hexlify(x), 16)
[docs]
def genparam_callback(p, n, out=sys.stdout):
# type: (int, Any, TextIO) -> None
ch = ['.', '+', '*', '\n']
out.write(ch[p])
out.flush()
[docs]
def quiet_genparam_callback(p, n, out):
# type: (Any, Any, Any) -> None
pass
[docs]
def passphrase_callback(v, prompt1='Enter passphrase:',
prompt2='Verify passphrase:'):
# type: (bool, str, str) -> Optional[str]
from getpass import getpass
while 1:
try:
p1 = getpass(prompt1)
if v:
p2 = getpass(prompt2)
if p1 == p2:
break
else:
break
except KeyboardInterrupt:
return None
return p1
[docs]
def no_passphrase_callback(*args):
# type: (*Any) -> str
return ''