o ѷ6i@sddlZddlmZmZddlmZddlmZGdddeeZ GdddeeZ Gd d d Z Gd d d Z d dZ ddZdS)N)urlsafe_b64decodeurlsafe_b64encode)Enum)Anyc@seZdZdZdZdZdS)Protocolz/Protocol to communicate with the authenticator.ctap2z ctap1/u2fN)__name__ __module__ __qualname____doc__CTAP2U2Frrt/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/selenium/webdriver/common/virtual_authenticator.pyrsrc@s eZdZdZdZdZdZdZdS) Transportz7Transport method to communicate with the authenticator.bleusbnfcinternalN)rr r r BLEUSBNFCINTERNALrrrrrs rc@sfeZdZeZeZejejddddfdedededededed d fd d Z d e eeeBffd dZ d S)VirtualAuthenticatorOptionsFTprotocol transporthas_resident_keyhas_user_verificationis_user_consentingis_user_verifiedreturnNcC(||_||_||_||_||_||_dS)zMConstructor. Initialize VirtualAuthenticatorOptions object. Nrrrrrr)selfrrrrrrrrr__init__.s   z$VirtualAuthenticatorOptions.__init__cCs|j|j|j|j|j|jdS)N)rrhasResidentKeyhasUserVerificationisUserConsentingisUserVerifiedr"r#rrrto_dictBsz#VirtualAuthenticatorOptions.to_dict) rr r rrr rstrboolr$dictr*rrrrr(s2 rc @s8eZdZdedededBdedBdedef dd Zed efd d Z ed efd dZ ed edBfddZ ed edBfddZ ed efddZ ed efddZededededed df ddZededededBdeded df ddZd eeeffddZedeeefd dfdd Zd efd!d"ZdS)# Credential credential_idis_resident_credentialrp_idN user_handle private_key sign_countcCr!)afConstructor. A credential stored in a virtual authenticator. https://w3c.github.io/webauthn/#credential-parameters. Args: credential_id (bytes): Unique base64 encoded string. is_resident_credential (bool): Whether the credential is client-side discoverable. rp_id (str): Relying party identifier. user_handle (bytes): userHandle associated to the credential. Must be Base64 encoded string. Can be None. private_key (bytes): Base64 encoded PKCS#8 private key. sign_count (int): initial value for a signature counter. N)_id_is_resident_credential_rp_id _user_handle _private_key _sign_count)r#r/r0r1r2r3r4rrrr$Ns  zCredential.__init__r cCt|jSN)rr5decoder)rrridjz Credential.idcC|jSr<)r6r)rrrr0nz!Credential.is_resident_credentialcCr@r<)r7r)rrrr1rrAzCredential.rp_idcCs|jr t|jSdSr<)r8rr=r)rrrr2vszCredential.user_handlecCr;r<)rr9r=r)rrrr3|r?zCredential.private_keycCr@r<)r:r)rrrr4rAzCredential.sign_countr>cCs||d|d||S)asCreates a non-resident (i.e. stateless) credential. Args: id (bytes): Unique base64 encoded string. rp_id (str): Relying party identifier. private_key (bytes): Base64 encoded PKCS sign_count (int): initial value for a signature counter. Returns: Credential: A non-resident credential. FNr)clsr>r1r3r4rrrcreate_non_resident_credentials z)Credential.create_non_resident_credentialcCs||d||||S)aCreates a resident (i.e. stateful) credential. Args: id (bytes): Unique base64 encoded string. rp_id (str): Relying party identifier. user_handle (bytes): userHandle associated to the credential. Must be Base64 encoded string. private_key (bytes): Base64 encoded PKCS sign_count (int): initial value for a signature counter. Returns: Credential: A resident credential. Tr)rBr>r1r2r3r4rrrcreate_resident_credentialsz%Credential.create_resident_credentialcCs.|j|j|j|j|jd}|jr|j|d<|S)N) credentialIdisResidentCredentialrpId privateKey signCount userHandle)r>r6r1r3r4r2)r#credential_datarrrr*s zCredential.to_dictdatacCs|t|dd}t|d}|dd}t|dd}t|d}|ddr3t|ddnd}|||||||S)NrEz==rFrGrHrIrJ)rr,getint)rBrLr5r0r1r3r4r2rrr from_dicts   "zCredential.from_dictc Cs6d|jd|jd|jd|jd|jd|jd S)NzCredential(id=z, is_resident_credential=z, rp_id=z, user_handle=z, private_key=z , sign_count=))r>r0r1r2r3r4r)rrr__str__szCredential.__str__)rr r bytesr,r+rNr$propertyr>r0r1r2r3r4 classmethodrCrDr-rr*rOrQrrrrr.Ms\    r.cstfdd}|S)zEDecorator to ensure that the client used is a chromium-based browser.cs0|jddvs Jd|g|Ri|S)N browserName)firefoxsafariz4This only currently works in Chromium based browsers)capslowerr#argskwargsfuncrrwrappersz0required_chromium_based_browser..wrapper) functoolswrapsr^r_rr]rrequired_chromium_based_browsersrccsttfdd}|S)zMDecorator to ensure that the function is called with a virtual authenticator.cs$|jstd|g|Ri|S)Nz9This function requires a virtual authenticator to be set.)virtual_authenticator_id ValueErrorrZr]rrr_sz/required_virtual_authenticator..wrapper)r`rarcrbrr]rrequired_virtual_authenticatorsrf)r`base64rrenumrtypingrr+rrrr.rcrfrrrrs   %v