B `@sddlZddlZddlmZddlmZddlmZdZddddd Z Gd d d e Z Gd d d e Z Gddde Z GdddejdZeZGdddejdZeZGdddejdZeZdS)N)utils) _get_backend) serializationi DHParameters)returncCst|}|||S)N)rZgenerate_dh_parameters) generatorkey_sizebackendr [/home/dcms/DCMS/lib/python3.7/site-packages/cryptography/hazmat/primitives/asymmetric/dh.pygenerate_parameterssr c@s^eZdZdeeejedddZddZddZdd d Z e d Z e d Z e d ZdS)DHParameterNumbersN)pgqcCsvt|trt|tstd|dk r6t|ts6td|dkrFtd|tkr`tdt||_||_||_ 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__cCs2t|tstS|j|jko0|j|jko0|j|jkS)N)rr NotImplementedrrr)rotherr r r __eq__)s zDHParameterNumbers.__eq__cCs ||k S)Nr )rrr r r __ne__1szDHParameterNumbers.__ne__cCst|}||S)N)rZload_dh_parameter_numbers)rr r r r parameters4szDHParameterNumbers.parametersrrr)N)N)__name__ __module__ __qualname__rtypingOptionalrr r!r"rread_only_propertyrrrr r r r r s   r c@sNeZdZedddZddZddZdd d d d Ze d Z e dZ dS)DHPublicNumbers)parameter_numberscCs4t|tstdt|ts$td||_||_dS)Nzy must be an integer.z5parameters must be an instance of DHParameterNumbers.)rrrr _y_parameter_numbers)ryr*r r r r>s  zDHPublicNumbers.__init__cCs&t|tstS|j|jko$|j|jkS)N)rr)rr+r,)rrr r r r Js  zDHPublicNumbers.__eq__cCs ||k S)Nr )rrr r r r!SszDHPublicNumbers.__ne__N DHPublicKey)rcCst|}||S)N)rZload_dh_public_numbers)rr r r r public_keyVszDHPublicNumbers.public_keyr+r,)N) r#r$r%r rr r!r/rr(r-r*r r r r r)=s    r)c@sNeZdZedddZddZddZdd d d d Ze d Z e dZ dS)DHPrivateNumbers)public_numberscCs4t|tstdt|ts$td||_||_dS)Nzx must be an integer.z6public_numbers must be an instance of DHPublicNumbers.)rrrr)_x_public_numbers)rxr1r r r r_s  zDHPrivateNumbers.__init__cCs&t|tstS|j|jko$|j|jkS)N)rr0rr2r3)rrr r r r ks  zDHPrivateNumbers.__eq__cCs ||k S)Nr )rrr r r r!tszDHPrivateNumbers.__ne__N DHPrivateKey)rcCst|}||S)N)rZload_dh_private_numbers)rr r r r private_keywszDHPrivateNumbers.private_keyr3r2)N) r#r$r%r)rr r!r6rr(r1r4r r r r r0^s    r0c@sLeZdZejddddZejddeddd Zejedd d Z d S) rr5)rcCsdS)z7 Generates and returns a DHPrivateKey. Nr )rr r r generate_private_keysz!DHParameters.generate_private_keyzserialization.Encodingzserialization.ParameterFormat)encodingrrcCsdS)z= Returns the parameters serialized as bytes. Nr )rr8rr r r parameter_bytesszDHParameters.parameter_bytescCsdS)z/ Returns a DHParameterNumbers. Nr )rr r r r*szDHParameters.parameter_numbersN) r#r$r%abcabstractmethodr7bytesr9r r*r r r r rs) metaclassc@s`eZdZejedddZejedddZ eje dddZ ejdd e d d d Z d S)r.)rcCsdS)z6 The bit length of the prime modulus. Nr )rr r r rszDHPublicKey.key_sizecCsdS)zJ The DHParameters object associated with this public key. Nr )rr r r r"szDHPublicKey.parameterscCsdS)z, Returns a DHPublicNumbers. Nr )rr r r r1szDHPublicKey.public_numberszserialization.Encodingzserialization.PublicFormat)r8rrcCsdS)z6 Returns the key serialized as bytes. Nr )rr8rr r r public_bytesszDHPublicKey.public_bytesN)r#r$r%r:abstractpropertyrrr;rr"r)r1r<r>r r r r r.sr.c@seZdZejedddZejedddZ eje dddZ ejee dd d Z ejedd d Zejd dde dddZdS)r5)rcCsdS)z6 The bit length of the prime modulus. Nr )rr r r rszDHPrivateKey.key_sizecCsdS)zC The DHPublicKey associated with this private key. Nr )rr r r r/szDHPrivateKey.public_keycCsdS)zK The DHParameters object associated with this private key. Nr )rr r r r"szDHPrivateKey.parameters)peer_public_keyrcCsdS)zn Given peer's DHPublicKey, carry out the key exchange and return shared key as bytes. Nr )rr@r r r exchangeszDHPrivateKey.exchangecCsdS)z- Returns a DHPrivateNumbers. Nr )rr r r private_numbersszDHPrivateKey.private_numberszserialization.Encodingzserialization.PrivateFormatz(serialization.KeySerializationEncryption)r8rencryption_algorithmrcCsdS)z6 Returns the key serialized as bytes. Nr )rr8rrCr r r private_bytess zDHPrivateKey.private_bytesN)r#r$r%r:r?rrr;r.r/rr"r<rAr0rBrDr r r r r5sr5)N)r:r&Z cryptographyrZcryptography.hazmat.backendsrZcryptography.hazmat.primitivesrrr objectr r)r0ABCMetarZDHParametersWithSerializationr.ZDHPublicKeyWithSerializationr5ZDHPrivateKeyWithSerializationr r r r s   '!!,