o c @sddlZddlZddlmZdZ ddededejddfd d ZGd d d ZGd ddZ GdddZ Gdddej dZ e Z Gdddej dZeZGdddej dZeZdS)N)_serializationi generatorkey_sizebackendreturn DHParameterscCsddlm}|||SNr)r),cryptography.hazmat.backends.openssl.backendrZgenerate_dh_parameters)rrrosslr Pusr/lib/python3.10/site-packages/cryptography/hazmat/primitives/asymmetric/dh.pygenerate_parameterss  r c @seZdZddededejeddfddZdedefd d Z dd ej dd fd dZ e defddZ e defddZe dejefddZdS)DHParameterNumbersNpgqrcCsvt|tr t|tstd|durt|tstd|dkr#td|tkr0tdt||_||_||_ dS)Nzp and g must be integerszq must be integer or Nonez!DH generator must be 2 or greaterz#p (modulus) must be at least {}-bit) isinstanceint TypeError ValueError bit_length_MIN_MODULUS_SIZEformat_p_g_q)selfrrrr r r __init__s  zDHParameterNumbers.__init__othercCs2t|tstS|j|jko|j|jko|j|jkSN)rrNotImplementedrrrrrr r r __eq__*s "zDHParameterNumbers.__eq__rrcCddlm}||Sr)r rZload_dh_parameter_numbersrrr r r r parameters2  zDHParameterNumbers.parameterscC|jSr )rrr r r r9zDHParameterNumbers.pcCr(r )rr)r r r r=r*zDHParameterNumbers.gcCr(r )rr)r r r rAr*zDHParameterNumbers.qr )__name__ __module__ __qualname__rtypingOptionalrobjectboolr#Anyr&propertyrrrr r r r rs"rc@sneZdZdededdfddZdedefdd Zdd e j dd fd d Z e defddZ e defddZdS)DHPublicNumbersyparameter_numbersrNcC4t|ts tdt|tstd||_||_dS)Nzy must be an integer.z5parameters must be an instance of DHParameterNumbers.)rrrr_y_parameter_numbers)rr5r6r r r rG   zDHPublicNumbers.__init__rcC&t|tstS|j|jko|j|jkSr )rr4r!r8r9r"r r r r#S   zDHPublicNumbers.__eq__r DHPublicKeycCr$r)r rZload_dh_public_numbersr%r r r public_key\r'zDHPublicNumbers.public_keycCr(r )r8r)r r r r5cr*zDHPublicNumbers.ycCr(r )r9r)r r r r6gr*z!DHPublicNumbers.parameter_numbersr )r+r,r-rrrr0r1r#r.r2r>r3r5r6r r r r r4F  r4c@sneZdZdededdfddZdedefdd Zdd e j dd fd d Z e defddZ e defddZdS)DHPrivateNumbersxpublic_numbersrNcCr7)Nzx must be an integer.z6public_numbers must be an instance of DHPublicNumbers.)rrrr4_x_public_numbers)rrArBr r r rmr:zDHPrivateNumbers.__init__rcCr;r )rr@r!rCrDr"r r r r#yr<zDHPrivateNumbers.__eq__r DHPrivateKeycCr$r)r rZload_dh_private_numbersr%r r r private_keyr'zDHPrivateNumbers.private_keycCr(r )rDr)r r r rBr*zDHPrivateNumbers.public_numberscCr(r )rCr)r r r rAr*zDHPrivateNumbers.xr )r+r,r-rr4rr0r1r#r.r2rFr3rBrAr r r r r@lr?r@c@sPeZdZejd ddZejdejdejde fddZ ejde fd d Z d S) rrrEcCdS)z7 Generates and returns a DHPrivateKey. Nr r)r r r generate_private_keyz!DHParameters.generate_private_keyencodingrcCrG)z= Returns the parameters serialized as bytes. Nr rrJrr r r parameter_bytesrIzDHParameters.parameter_bytescCrG)z/ Returns a DHParameterNumbers. Nr r)r r r r6rIzDHParameters.parameter_numbersN)rrE) r+r,r-abcabstractmethodrHrEncodingZParameterFormatbytesrLrr6r r r r rs   ) metaclassc@sheZdZejdefddZejdefddZ ejde fddZ ejde j d e jdefd d Zd S) r=rcCrGz6 The bit length of the prime modulus. Nr r)r r r rrIzDHPublicKey.key_sizecCrG)zJ The DHParameters object associated with this public key. Nr r)r r r r&rIzDHPublicKey.parameterscCrG)z, Returns a DHPublicNumbers. Nr r)r r r rBrIzDHPublicKey.public_numbersrJrcCrGz6 Returns the key serialized as bytes. Nr rKr r r public_bytesrIzDHPublicKey.public_bytesN)r+r,r-rMabstractpropertyrrrNrr&r4rBrrOZ PublicFormatrPrTr r r r r=sr=c @seZdZejdefddZejdefddZ ejde fddZ ejdede fd d Z ejdefd d Zejd ejdejdejde fddZdS)rErcCrGrRr r)r r r rrIzDHPrivateKey.key_sizecCrG)zC The DHPublicKey associated with this private key. Nr r)r r r r>rIzDHPrivateKey.public_keycCrG)zK The DHParameters object associated with this private key. Nr r)r r r r&rIzDHPrivateKey.parameterspeer_public_keycCrG)zn Given peer's DHPublicKey, carry out the key exchange and return shared key as bytes. Nr )rrVr r r exchangerIzDHPrivateKey.exchangecCrG)z- Returns a DHPrivateNumbers. Nr r)r r r private_numbersrIzDHPrivateKey.private_numbersrJrencryption_algorithmcCrGrSr )rrJrrYr r r private_bytesrIzDHPrivateKey.private_bytesN)r+r,r-rMrUrrrNr=r>rr&rPrWr@rXrrOZ PrivateFormatZKeySerializationEncryptionrZr r r r rEs*rEr )rMr.Zcryptography.hazmat.primitivesrrrr2r rr4r@ABCMetarZDHParametersWithSerializationr=ZDHPublicKeyWithSerializationrEZDHPrivateKeyWithSerializationr r r r s.  /&&,