U ڲg @sdZddlmZmZmZdeeeedddZdgZ e dkre ddd l Z e d D]4Ze \ZZerlqed dkrTerTe d eqTe d d S)zFunctions for PKCS#1 version 2 encryption and signing This module implements certain functionality from PKCS#1 version 2. Main documentation is RFC 2437: https://tools.ietf.org/html/rfc2437 )commonpkcs1 transformSHA-1)seedlengthhasherreturnc sztjj}WnDtk rV}z&tdjdttjd|W5d}~XYnX|d|krvt dj|ddfd d t t ||d D}|d|S) a MGF1 is a Mask Generation Function based on a hash function. A mask generation function takes an octet string of variable length and a desired output length as input, and outputs an octet string of the desired length. The plaintext-awareness of RSAES-OAEP relies on the random nature of the output of the mask generation function, which in turn relies on the random nature of the underlying hash. :param bytes seed: seed from which mask is generated, an octet string :param int length: intended length in octets of the mask, at most 2^32(hLen) :param str hasher: hash function (hLen denotes the length in octets of the hash function output) :return: mask, an octet string of length `length` :rtype: bytes :raise OverflowError: when `length` is too large for the specified `hasher` :raise ValueError: when specified `hasher` is invalid z=Invalid `hasher` specified. Please select one of: {hash_list}z, )Z hash_listNlzmDesired length should be at most 2**32 times the hasher's output length ({hash_length} for {hasher} function)) hash_lengthrc3s*|]"}tjtj|dddVqdS))Z fill_size) method_nameN)rZ compute_hashrZ int2bytes).0counterrr0/tmp/pip-unpacked-wheel-1z1vrokf/rsa/pkcs1_v2.py Hs zmgf1..) rZ HASH_METHODS digest_sizeKeyError ValueErrorformatjoinsortedkeys OverflowErrorrangerZceil_div)rrrr exoutputrrrmgf1s*   r __main__z'Running doctests 1000x or until failureNidz%i timesz Doctests done)r)__doc__rsarrrbytesintstrr __all____name__printdoctestrcounttestmodZfailurestestsrrrrs9