o jT@sddlmZmZmZmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZddlmZdd lmZmZgd Zd d Zd dZddZddZddZddZddZddZddZddZdd Z d!d"Z!d#d$Z"d%d&Z#d'S)()unicode_literalsdivisionabsolute_importprint_function)pretty_message)newnull)CoreFoundation CFHelpershandle_cf_error)Security) rand_bytes) type_namebyte_cls) aes_cbc_no_padding_decryptaes_cbc_no_padding_encryptaes_cbc_pkcs7_decryptaes_cbc_pkcs7_encryptdes_cbc_pkcs5_decryptdes_cbc_pkcs5_encryptrc2_cbc_pkcs5_decryptrc2_cbc_pkcs5_encrypt rc4_decrypt rc4_encrypttripledes_cbc_pkcs5_decrypttripledes_cbc_pkcs5_encryptcCst|dvrttdt||std}nt|dkr%ttdt|t|ddkr6ttdt||ttj|||tjfS)a Encrypts plaintext using AES in CBC mode with a 128, 192 or 256 bit key and no padding. This means the ciphertext must be an exact multiple of 16 bytes long. :param key: The encryption key - a byte string either 16, 24 or 32 bytes long :param data: The plaintext - a byte string :param iv: The initialization vector - either a byte string 16-bytes long or None to generate an IV :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A tuple of two byte strings (iv, ciphertext)  c key must be either 16, 24 or 32 bytes (128, 192 or 256 bits) long - is %s r: iv must be 16 bytes long - is %s rzJ data must be a multiple of 16 bytes long - is %s )len ValueErrorrr_encryptrkSecAttrKeyTypeAESkSecPaddingNoneKeykeydataivr-/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/oscrypto/_mac/symmetric.pyrs$   rcCPt|dvrttdt|t|dkrttdt|ttj|||tjS)a[ Decrypts AES ciphertext in CBC mode using a 128, 192 or 256 bit key and no padding. :param key: The encryption key - a byte string either 16, 24 or 32 bytes long :param data: The ciphertext - a byte string :param iv: The initialization vector - a byte string 16-bytes long :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext rr"rr#)r$r%r_decryptrr'r(r)r-r-r.rRs  rcCsbt|dvrttdt||std}nt|dkr%ttdt||ttj|||tjfS)a Encrypts plaintext using AES in CBC mode with a 128, 192 or 256 bit key and PKCS#7 padding. :param key: The encryption key - a byte string either 16, 24 or 32 bytes long :param data: The plaintext - a byte string :param iv: The initialization vector - either a byte string 16-bytes long or None to generate an IV :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A tuple of two byte strings (iv, ciphertext) rr"rr#)r$r%rrr&rr'kSecPaddingPKCS7Keyr)r-r-r.r|s   rcCr/)aG Decrypts AES ciphertext in CBC mode using a 128, 192 or 256 bit key :param key: The encryption key - a byte string either 16, 24 or 32 bytes long :param data: The ciphertext - a byte string :param iv: The initialization vector - a byte string 16-bytes long :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext rzo key must be either 16, 24 or 32 bytes (128, 192 or 256 bits) long - is %s rr#)r$r%rr0rr'r1r)r-r-r.rs  rcC<t|dks t|dkrttdt|ttj||ddS)a Encrypts plaintext using RC4 with a 40-128 bit key :param key: The encryption key - a byte string 5-16 bytes long :param data: The plaintext - a byte string :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the ciphertext rQ key must be 5 to 16 bytes (40 to 128 bits) long - is %s N)r$r%rr&rkSecAttrKeyTypeRC4r*r+r-r-r.r rcCr2)a Decrypts RC4 ciphertext using a 40-128 bit key :param key: The encryption key - a byte string 5-16 bytes long :param data: The ciphertext - a byte string :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext r3rr4N)r$r%rr0rr5r6r-r-r.rr7rcCsnt|dks t|dkrttdt||std}nt|dkr+ttdt||ttj|||tjfS)ab Encrypts plaintext using RC2 with a 64 bit key :param key: The encryption key - a byte string 8 bytes long :param data: The plaintext - a byte string :param iv: The 8-byte initialization vector to use - a byte string - set as None to generate an appropriate one :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A tuple of two byte strings (iv, ciphertext) r3rr49 iv must be 8 bytes long - is %s )r$r%rrr&rkSecAttrKeyTypeRC2kSecPaddingPKCS5Keyr)r-r-r.rs  rcCs\t|dks t|dkrttdt|t|dkr$ttdt|ttj|||tjS)a" Decrypts RC2 ciphertext using a 64 bit key :param key: The encryption key - a byte string 8 bytes long :param data: The ciphertext - a byte string :param iv: The initialization vector used for encryption - a byte string :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext r3rr4r8r9)r$r%rr0rr:r;r)r-r-r.r;s rcCst|dkrt|dkrttdt||std}nt|dkr+ttdt|t|dkr9||dd}|ttj|||tjfS)a Encrypts plaintext using 3DES in either 2 or 3 key mode :param key: The encryption key - a byte string 16 or 24 bytes long (2 or 3 key mode) :param data: The plaintext - a byte string :param iv: The 8-byte initialization vector to use - a byte string - set as None to generate an appropriate one :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A tuple of two byte strings (iv, ciphertext) rr zT key must be 16 bytes (2 key) or 24 bytes (3 key) long - %s r8z6 iv must be 8 bytes long - %s r)r$r%rrr&rkSecAttrKeyType3DESr;r)r-r-r.rds   rcCsxt|dkrt|dkrttdt|t|dkr$ttdt|t|dkr2||dd}ttj|||tjS)aC Decrypts 3DES ciphertext in either 2 or 3 key mode :param key: The encryption key - a byte string 16 or 24 bytes long (2 or 3 key mode) :param data: The ciphertext - a byte string :param iv: The initialization vector used for encryption - a byte string :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext rr zW key must be 16 bytes (2 key) or 24 bytes (3 key) long - is %s r8r9r)r$r%rr0rr<r;r)r-r-r.rs  rcCsbt|dkrttdt||std}nt|dkr%ttdt||ttj|||tjfS)a Encrypts plaintext using DES with a 56 bit key :param key: The encryption key - a byte string 8 bytes long (includes error correction bits) :param data: The plaintext - a byte string :param iv: The 8-byte initialization vector to use - a byte string - set as None to generate an appropriate one :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A tuple of two byte strings (iv, ciphertext) r8T key must be 8 bytes (56 bits + 8 parity bits) long - is %s r9)r$r%rrr&rkSecAttrKeyTypeDESr;r)r-r-r.rs   rcCsPt|dkrttdt|t|dkrttdt|ttj|||tjS)aC Decrypts DES ciphertext using a 56 bit key :param key: The encryption key - a byte string 8 bytes long (includes error correction bits) :param data: The ciphertext - a byte string :param iv: The initialization vector used for encryption - a byte string :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext r8r=r9)r$r%rr0rr>r;r)r-r-r.rs  rc C<t|tsttdt|t|tsttdt||tjkr/t|ts/ttdt||tjkr:|s:tdd}d}d}d}d} d} zt tj |fg}t |}t |}t t d} t||| } t| t| | } t| |tjkrt| tjt| t| t| tj|| t| t |}t| tj|| t| t| tj|| t| t| | } t| t| W|rt ||rt ||rt ||rt || rt | | rt | SS|rt ||rt ||rt ||r t || rt | | rt | ww)a Encrypts plaintext :param cipher: A kSecAttrKeyType* value that specifies the cipher to use :param key: The encryption key - a byte string 5-16 bytes long :param data: The plaintext - a byte string :param iv: The initialization vector - a byte string - unused for RC4 :param padding: The padding mode to use, specified as a kSecPadding*Key value - unused for RC4 :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the ciphertext ; key must be a byte string, not %s < data must be a byte string, not %s : iv must be a byte string, not %s padding must be specifiedN CFErrorRef *) isinstancer TypeErrorrrrr5r%r cf_dictionary_from_pairskSecAttrKeyTypecf_data_from_bytesrr SecKeyCreateFromDatar SecEncryptTransformCreateSecTransformSetAttributekSecModeCBCKeyr kSecPaddingKey kSecIVKeykSecTransformInputAttributeNameSecTransformExecutecf_data_to_bytes CFRelease) cipherr*r+r,paddingcf_dictcf_keycf_datacf_ivsec_key sec_transform error_pointer ciphertextr-r-r.r&                      r&c Cr?)a Decrypts AES/RC4/RC2/3DES/DES ciphertext :param cipher: A kSecAttrKeyType* value that specifies the cipher to use :param key: The encryption key - a byte string 5-16 bytes long :param data: The ciphertext - a byte string :param iv: The initialization vector - a byte string - unused for RC4 :param padding: The padding mode to use, specified as a kSecPadding*Key value - unused for RC4 :raises: ValueError - when any of the parameters contain an invalid value TypeError - when any of the parameters are of the wrong type OSError - when an error is returned by the OS crypto library :return: A byte string of the plaintext r@rArBrCNrD)rErrFrrrr5r%r rGrHrIrr rJr SecDecryptTransformCreaterLrMr rNrOrPrQrRrS) rTr*r+r,rUrVrWrXrYrZr[r\ plaintextr-r-r.r0r^r0N)$ __future__rrrr_errorsr_ffirr _core_foundationr r r _securityrutilr_typesrr__all__rrrrrrrrrrrrr&r0r-r-r-r.s,   6*-*,)0-,) q