o ҷh@sddlZddlZddlmZddlmZddlmZGdddeeZGdddeeZ Gd d d Z Gd d d Z d dZ ddZ dS)N)urlsafe_b64decode)urlsafe_b64encode)Enumc@s*eZdZUdZdZeed<dZeed<dS)Protocolz/Protocol to communicate with the authenticator.ctap2CTAP2z ctap1/u2fU2FN)__name__ __module__ __qualname____doc__rstr__annotations__rrrb/var/www/html/venv/lib/python3.10/site-packages/selenium/webdriver/common/virtual_authenticator.pyrs  rc@sBeZdZUdZdZeed<dZeed<dZeed<dZ eed <d S) Transportz7Transport method to communicate with the authenticator.bleBLEusbUSBnfcNFCinternalINTERNALN) r r r r rr rrrrrrrrr s    rc@sneZdZeZeZejejddddfdedededededed d fd d Z d e j ee j eefffd dZ d S)VirtualAuthenticatorOptionsFTprotocol transporthas_resident_keyhas_user_verificationis_user_consentingis_user_verifiedreturnNcC(||_||_||_||_||_||_dS)zMConstructor. Initialize VirtualAuthenticatorOptions object. Nrrrrrr )selfrrrrrr rrr__init__/s  z$VirtualAuthenticatorOptions.__init__cCs|j|j|j|j|j|jdS)N)rrhasResidentKeyhasUserVerificationisUserConsentingisUserVerifiedr#r$rrrto_dictDsz#VirtualAuthenticatorOptions.to_dict)r r r rrrrr boolr%typingDictUnionr+rrrrr)s2 &rc @s>eZdZdedededejededef ddZ e d efd d Z e d efd d Z e d efddZ e d ejefddZe d efddZe d efddZededededed df ddZedededejededed df ddZd ejeejffddZedejeejfd dfddZd efd d!Zd"S)# Credential credential_idis_resident_credentialrp_id user_handle private_key sign_countcCr")ahConstructor. 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): intital value for a signature counter. N)_id_is_resident_credential_rp_id _user_handle _private_key _sign_count)r$r1r2r3r4r5r6rrrr%Ps  zCredential.__init__r!cCt|jSN)rr7decoder*rrridkz Credential.idcC|jSr>)r8r*rrrr2oz!Credential.is_resident_credentialcCrBr>)r9r*rrrr3srCzCredential.rp_idcCs|jr t|jSdSr>)r:rr?r*rrrr4wszCredential.user_handlecCr=r>)rr;r?r*rrrr5}rAzCredential.private_keycCrBr>)r<r*rrrr6rCzCredential.sign_countr@cCs||d|d||S)auCreates 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): intital value for a signature counter. :Returns: - Credential: A non-resident credential. FNr)clsr@r3r5r6rrrcreate_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): intital value for a signature counter. :returns: - Credential: A resident credential. Tr)rDr@r3r4r5r6rrrcreate_resident_credentialsz%Credential.create_resident_credentialcCs.|j|j|j|j|jd}|jr|j|d<|S)N) credentialIdisResidentCredentialrpId privateKey signCount userHandle)r@r8r3r5r6r4)r$credential_datarrrr+s zCredential.to_dictdatacCs|t|dd}t|d}|dd}t|dd}t|d}|ddr3t|ddnd}|||||||S)NrGz==rHrIrJrKrL)rr,getint)rDrNr7r2r3r5r6r4rrr 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@r2r3r4r5r6r*rrr__str__szCredential.__str__N)r r r bytesr,r r-OptionalrPr%propertyr@r2r3r4r5r6 classmethodrErFr.Anyr+rQrSrrrrr0Os\     r0cstfdd}|S)zKA decorator 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) functoolswrapsrbrcrrarrequired_chromium_based_browsersrgcsttfdd}|S)zSA decorator 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 ValueErrorr^rarrrcsz/required_virtual_authenticator..wrapper)rdrergrfrrarrequired_virtual_authenticatorsrj)rdr-base64rrenumrr rrrr0rgrjrrrrs    &u