o l>j@stUdZddlmZddlZddlmZddlmZmZm Z m Z m Z e e BeBeBe eBe eefBZded<eeBeBZded<d?d@ddZdAddZdBddZddZdCddZdDddZdEddZe efe efeefeefe efe efgZd ed!<dFd$d%ZdFd&d'ZdGd)d*ZdHd,d-Z dId/d0Z!dJd2d3Z"dKd5d6Z#eee e!e"e#ed7Z$dLd8d9Z%dMd;d<Z&dNd=d>Z'dS)Oz Minimal CBOR implementation supporting a subset of functionality and types required for FIDO 2 CTAP. Use the :func:`encode`, :func:`decode` and :func:`decode_from` functions to encode and decode objects to/from CBOR. ) annotationsN) UnionType)AnyCallableMappingSequence TypeAliasrCborTypeBufferdataintmtreturnbytescCs|dkr d}d|}|d>}|dkrd||Bf}}n3|dkr)d|d B|f}}n%|d kr7d |d B|f}}n|d krEd|dB|f}}n d|dB|f}}tj|g|RS)Nrz>Bz>BBiz>BHlz>BIz>BQ)structpack)r r fmtargsr?/home/thesage/.local/lib/python3.10/site-packages/fido2/cbor.py _dump_int1srboolcCs |rdSdS)Nrr rrr _dump_boolF r$Sequence[CborType]cCs$tt|ddddd|DS)Nr cSsg|]}t|qSrencode).0xrrr Ksz_dump_list..)rlenjoinr#rrr _dump_listJs$r1cCs|d}|dt||fS)Nr)r/)entrykeyrrr _sort_keysNsr4Mapping[CborType, CborType]cCsBdd|D}|jtdtt|ddddd|DS)NcSs g|] \}}t|t|fqSrr*r,kvrrrr.Ts z_dump_dict..)r3rr(r)cSsg|]\}}||qSrrr6rrrr.Vs)itemssortr4rr/r0)r r9rrr _dump_dictSs $r;cCstt|dd|S)Nr()rr/r#rrr _dump_bytesYsr=strcCs|d}tt|dd|S)Nutf8r()r+rr/)r data_bytesrrr _dump_text]s rBz9Sequence[tuple[type | UnionType, Callable[[Any], bytes]]] _SERIALIZERSaituple[int, bytes]cCs|dkr||fS|dkr|d|ddfS|dkr(td|d|ddfS|dkr:td|d|d dfS|d krLtd |d|d dfStd )Nrrrrz>Hr<rz>Ir'rz>QzInvalid additional information)r unpack_from ValueErrorrDr rrr _load_intlsrJcCst||\}}d||fS)NrrJ)rDr valrestrrr _load_nintzs rNtuple[bool, bytes]cCs |dk|fS)NrrIrrr _load_boolr%rQtuple[bytes, bytes]cCs&t||\}}|d|||dfSNrK)rDr lnrrr _load_bytessrUtuple[str, bytes]cCst||\}}|d|fS)Nr?)rUdecode)rDr encrMrrr _load_textsrY tuple[Sequence[CborType], bytes]cCs>t||\}}g}t|D] }t|\}}||q ||fSrS)rJrange decode_fromappend)rDr rTvaluesirLrrr _load_arrays    r`)tuple[Mapping[CborType, CborType], bytes]cCsHt||\}}i}t|D]}t|\}}t|\}}|||<q ||fSrS)rJr[r\)rDr rTr^r_r7r8rrr _load_maps    rb)rrr<r@r'rcCs2tD]\}}t||r||Sqtd|)z#Encodes data to a CBOR byte string.zUnsupported value: )rC isinstancerH)r r7r8rrrr+s   r+tuple[Any, bytes]cCs&|d}t|d?|d@|ddS)zDecodes a CBOR-encoded value from the start of a byte string. Additional data after a valid CBOR object is returned as well. :return: The decoded object, and any remaining data.rrrN)_DESERIALIZERS)r fbrrrr\sr\cCs t|\}}|dkrtd|S)zuDecodes data from a CBOR-encoded byte string. Also validates that no extra data follows the encoded object. r)zExtraneous data)r\rH)r valuerMrrrrWs rW)r)r r r r rr)r r rr)r r&rr)r r5rr)r rrr)r r>rr)rDr r rrrE)rDr r rrrO)rDr r rrrR)rDr r rrrV)rDr r rrrZ)rDr r rrra)r r rr)r rrre)rr )(__doc__ __future__rrtypesrtypingrrrrrr r r>rr __annotations__ bytearray memoryviewr rr$r1r4r;r=rBrCrJrNrQrUrYr`rbrgr+r\rWrrrrsN  ,