U g@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.Zctap2CTAP2z ctap1/u2fU2FN)__name__ __module__ __qualname____doc__rstr__annotations__rrrS/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/virtual_authenticator.pyrs  rc@sBeZdZUdZdZeed<dZeed<dZeed<dZ eed <d S) Transportz7Transport method to communicate with the authenticator.ZbleBLEZusbUSBZnfcNFCZinternalINTERNALN) rr r r rr r rrrrrrrr s    rc @sbeZdZeZeZejejddddfeeeeeeddddZ e j ee j eeffddd Z dS) VirtualAuthenticatorOptionsFTN)protocol transporthas_resident_keyhas_user_verificationis_user_consentingis_user_verifiedreturncCs(||_||_||_||_||_||_dS)zMConstructor. Initialize VirtualAuthenticatorOptions object. Nrrrrrr)selfrrrrrrrrr__init__/s z$VirtualAuthenticatorOptions.__init__rcCs|j|j|j|j|j|jdS)N)rrZhasResidentKeyZhasUserVerificationZisUserConsentingZisUserVerifiedrrrrrto_dictDsz#VirtualAuthenticatorOptions.to_dict)rr r rrrrr boolrtypingDictUnionr"rrrrr)s$ rc@s eZdZeeeejeeedddZ e edddZ e edddZ e edd d Z e ejedd d Ze edd dZe edddZeeeeeddddZeeeejeeeddddZejeejfdddZeejeejfddddZedddZdS) Credential) credential_idis_resident_credentialrp_id user_handle private_key sign_countcCs(||_||_||_||_||_||_dS)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): intital value for a signature counter. N)_id_is_resident_credential_rp_id _user_handle _private_key _sign_count)rr(r)r*r+r,r-rrrrPs zCredential.__init__r cCst|jSN)rr.decoder!rrridksz Credential.idcCs|jSr4)r/r!rrrr)osz!Credential.is_resident_credentialcCs|jSr4)r0r!rrrr*sszCredential.rp_idcCs|jrt|jSdSr4)r1rr5r!rrrr+wszCredential.user_handlecCst|jSr4)rr2r5r!rrrr,}szCredential.private_keycCs|jSr4)r3r!rrrr-szCredential.sign_count)r6r*r,r-rcCs||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)clsr6r*r,r-rrrcreate_non_resident_credentials z)Credential.create_non_resident_credential)r6r*r+r,r-rcCs||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)r7r6r*r+r,r-rrrcreate_resident_credentialsz%Credential.create_resident_credentialcCs.|j|j|j|j|jd}|jr*|j|d<|S)N) credentialIdisResidentCredentialrpId privateKey signCount userHandle)r6r/r*r,r-r+)rZcredential_datarrrr"s zCredential.to_dict)datarcCs|t|dd}t|d}|dd}t|dd}t|d}|ddrft|ddnd}|||||||S)Nr:z==r;r<r=r>r?)rr#getint)r7r@r.r)r*r,r-r+rrr 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=))r6r)r*r+r,r-r!rrr__str__szCredential.__str__N)rr r bytesr#r r$OptionalrBrpropertyr6r)r*r+r,r- classmethodr8r9r%Anyr"rCrErrrrr'OsB  r'cstfdd}|S)zKA decorator to ensure that the client used is a chromium based browser.cs*|jddkstd|f||S)NZ browserName)ZfirefoxZsafariz4This only currently works in Chromium based browsers)ZcapslowerAssertionErrorrargskwargsfuncrrwrappersz0required_chromium_based_browser..wrapper) functoolswrapsrQrRrrPrrequired_chromium_based_browsersrVcsttfdd}|S)zSA decorator to ensure that the function is called with a virtual authenticator.cs|jstd|f||S)Nz9This function requires a virtual authenticator to be set.)Zvirtual_authenticator_id ValueErrorrMrPrrrRsz/required_virtual_authenticator..wrapper)rSrTrVrUrrPrrequired_virtual_authenticatorsrX)rSr$base64rrenumrr rrrr'rVrXrrrrs    &u