"""
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 platform
import struct
import sys
import struct
import unittest
from M2Crypto import m2
from typing import (
Callable,
Optional,
TextIO,
Tuple,
Union,
)
# see https://github.com/python/typeshed/issues/222
AddrType = Union[Tuple[str, int], str]
# COMPATIBILITY: Literal is only Python 3.8+
# ModeStr = Literal['r', 'w', 'rw', 'rb', 'wb', 'rwb']
ModeStr = str
log = logging.getLogger("util")
[docs]
class UtilError(Exception):
pass
m2.util_init(UtilError)
[docs]
def is_32bit() -> bool:
# or alternatively (slightly slower)
# (struct.calcsize("P") * 8) == 32
return not (sys.maxsize > 2**32)
[docs]
def expectedFailureIf(condition: bool) -> Callable:
"""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: str, blklen: int = 8) -> str:
pad = 8 - (len(data) % 8)
return data + chr(pad) * pad
[docs]
def pkcs7_pad(data: str, blklen: 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: bytes) -> str:
return binascii.b2a_base64(b)[:-1].decode()
[docs]
def octx_to_num(x: bytes) -> int:
return int(binascii.hexlify(x), 16)
[docs]
def genparam_callback(p: int, n: int) -> int:
ch = [".", "+", "*", "\n"]
sys.stdout.write(ch[p])
sys.stdout.flush()
return 1
[docs]
def quiet_genparam_callback(p: int, n: int) -> int:
return 1
[docs]
def passphrase_callback(
v: bool,
prompt1: str = "Enter passphrase:",
prompt2: str = "Verify passphrase:",
) -> 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) -> str:
return ""