B `ci @sdZddlmZmZmZdeeeedddZdgZ e dkre ddd l Z x>e d D]2Ze \ZZerlPed dkrVerVe d eqVWe 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 sytjj}Wn4tk rFtdjdttjdYnX|d|krft dj|ddfdd 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_listlzmDesired 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).0counter)rr3/tmp/pip-target-ywh1gcq3/lib/python/rsa/pkcs1_v2.py Iszmgf1..N) rZ HASH_METHODS digest_sizeKeyError ValueErrorformatjoinsortedkeys OverflowErrorrangerZceil_div)rrrr outputr)rrrmgf1s    r__main__z'Running doctests 1000x or until failureNidz%i timesz Doctests done)r)__doc__Zrsarrrbytesintstrr__all____name__printdoctestrcounttestmodZfailurestestsrrrrs9