o &g@sddlZddlZddlmZmZddlmZddlmZddZ ddZ d d Z d d Z Gd dde ZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZeddZGdd d eZdS)!N)urlsafe_b64decodeurlsafe_b64encode) namedtuple)MutableMappingcCs,t|ts |d}t|}|ddS)Nutf-8=) isinstancebytesencoderdecoderstrip)payloadr rN/root/parts/websockify/install/lib/python3.10/site-packages/jwcrypto/common.pybase64url_encode s  rcCsNt|d}|dkr|d7}n|dkr|d7}n|dkr tdt|dS) Nz==rrzInvalid base64 stringr)len ValueErrorrr )r sizerrrbase64url_decodes   rcCs$t|tr |d}tj|dddS)Nr),:T) separators sort_keys)rr r jsondumpsstringrrr json_encode!s  r cCst|tr |d}t|S)Nr)rr r rloadsrrrr json_decode's   r"c@s eZdZdS) JWExceptionN)__name__ __module__ __qualname__rrrrr#-sr#cseZdZdfdd ZZS)InvalidJWAAlgorithmNcs(d}|r |d|7}tt||dS)NzInvalid JWA Algorithm namez (%s))superr'__init__selfmessagemsg __class__rrr)2s zInvalidJWAAlgorithm.__init__N)r$r%r&r) __classcell__rrr.rr'1sr'c eZdZdZfddZZS)InvalidCEKeyLengthzInvalid CEK Key Length. This exception is raised when a Content Encryption Key does not match the required length. c d||f}tt||dS)Nz&Expected key of length %d bits, got %d)r(r3r)r+expectedobtainedr-r.rrr)@ zInvalidCEKeyLength.__init__r$r%r&__doc__r)r1rrr.rr39r3c"eZdZdZdfdd ZZS)InvalidJWEOperationzInvalid JWS Object. This exception is raised when a requested operation cannot be execute due to unsatisfied conditions. Nc:d}|r|}nd}|r|dt|7}tt||dSNzUnknown Operation Failurez {%s})reprr(r=r)r+r, exceptionr-r.rrr)LzInvalidJWEOperation.__init__NNr9rrr.rr=Er=cr2)InvalidJWEKeyTypezInvalid JWE Key Type. This exception is raised when the provided JWK Key does not match the type required by the specified algorithm. cr4)NzExpected key type %s, got %s)r(rFr)r5r.rrr)^r8zInvalidJWEKeyType.__init__r9rrr.rrFWr;rFcr2)InvalidJWEKeyLengthzInvalid JWE Key Length. This exception is raised when the provided JWK Key does not match the length required by the specified algorithm. cr4)Nz!Expected key of length %d, got %d)r(rGr)r5r.rrr)jr8zInvalidJWEKeyLength.__init__r9rrr.rrGcr;rGcr<)InvalidJWSERegOperationzInvalid JWSE Header Registry Operation. This exception is raised when there is an error in trying to add a JW Signature or Encryption header to the Registry. Ncr>r?)r@r(rHr)rAr.rrr)vrCz InvalidJWSERegOperation.__init__rDr9rrr.rrHorErHcr<) JWKeyNotFoundzThe key needed to complete the operation was not found. This exception is raised when a JWKSet is used to perform some operation and the key required to successfully complete the operation is not found. Ncs"|r|}nd}tt||dS)Nz Key Not Found)r(rIr)r*r.rrr)szJWKeyNotFound.__init__r0r9rrr.rrIsrI Parameterz*description mustprotect supported check_fnc@sFeZdZdddZddZddZdd Zd d Zd d ZddZ dS)JWSEHeaderRegistryNcCs8|rt|trt||_ntdi|_t|dS)NzUnknown input type)rdictcopydeepcopy _registryrHrr))r+ init_registryrrrr)s  zJWSEHeaderRegistry.__init__cCs8||jvr td||j|}|jdurdS||S)Nz No header "%s" found in registryT)rOrHcheck_fn)r+hvalueparamrrr check_headers    zJWSEHeaderRegistry.check_headercCs |j|Sr0)rO __getitem__r+keyrrrrVs zJWSEHeaderRegistry.__getitem__cC |jSr0)rO__iter__r+rrrrZ zJWSEHeaderRegistry.__iter__cCs0|j|js |j|jrtd|j|dS)Nz-Unable to delete protected or supported field)rO mustprotect supportedrH __delitem__rWrrrr_s  zJWSEHeaderRegistry.__delitem__cCsL||jvr|j|}|jrtd|jr|jstd|j|=||j|<dS)Nz+Supported header already exists in registryz,Header specified should bea protected header)rOr^rHr])r+rRjwse_header_paramprrr __setitem__s   zJWSEHeaderRegistry.__setitem__cCrYr0)rO__len__r[rrrrcr\zJWSEHeaderRegistry.__len__r0) r$r%r&r)rUrVrZr_rbrcrrrrrKs    rK)rMrbase64rr collectionsrcollections.abcrrrr r" Exceptionr#r'r3r=rFrGrHrIJWSEHeaderParameterrKrrrrs*