o M,hR@sddlZddlmZmZddlmZddlmZmZm Z Gddde eZ Gddde eZ Gd d d Z Gd d d Zd dZddZdS)N)urlsafe_b64decodeurlsafe_b64encode)Enum)AnyOptionalUnionc@seZdZdZdZdZdS)Protocolz/Protocol to communicate with the authenticator.ctap2z ctap1/u2fN)__name__ __module__ __qualname____doc__CTAP2U2Frr{/var/www/www-root/data/www/bot.pdev.uz/venv/lib/python3.10/site-packages/selenium/webdriver/common/virtual_authenticator.pyrsrc@s eZdZdZdZdZdZdZdS) Transportz7Transport method to communicate with the authenticator.bleusbnfcinternalN)r r r r BLEUSBNFCINTERNALrrrrrs rc@sjeZdZeZeZejejddddfdedededededed d fd d Z d e ee eefffd dZ d S)VirtualAuthenticatorOptionsFTprotocol transporthas_resident_keyhas_user_verificationis_user_consentingis_user_verifiedreturnNcC(||_||_||_||_||_||_dS)zMConstructor. Initialize VirtualAuthenticatorOptions object. Nrrrrr r!)selfrrrrr r!rrr__init__.s  z$VirtualAuthenticatorOptions.__init__cCs|j|j|j|j|j|jdS)N)rrhasResidentKeyhasUserVerificationisUserConsentingisUserVerifiedr$r%rrrto_dictCsz#VirtualAuthenticatorOptions.to_dict) r r r rrrrstrboolr&dictrr,rrrrr(s2 "rc @s4eZdZdededeedeededef ddZe d efd d Z e d efd d Z e d efddZ e d eefddZ e d efddZe d efddZededededed df ddZedededeededed df ddZd eeeffddZedeeefd dfddZd efd d!Zd"S)# Credential credential_idis_resident_credentialrp_id user_handle private_key sign_countcCr#)arConstructor. 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%r1r2r3r4r5r6rrrr&Os  zCredential.__init__r"cCt|jSN)rr7decoder+rrridjz Credential.idcC|jSr>)r8r+rrrr2nz!Credential.is_resident_credentialcCrBr>)r9r+rrrr3rrCzCredential.rp_idcCs|jr t|jSdSr>)r:rr?r+rrrr4vszCredential.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): initial 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): initial 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.rr-rPr&propertyr@r2r3r4r5r6 classmethodrErFr/rr,rQrSrrrrr0Ns\    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) functoolswrapsr`rarr_rrequired_chromium_based_browsersrecsttfdd}|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\r_rrrasz/required_virtual_authenticator..wrapper)rbrcrerdrr_rrequired_virtual_authenticatorsrh)rbbase64rrenumrtypingrrrr-rrrr0rerhrrrrs  &u