ruạṛ
a �+�bE� � @ sB d dl mZ d dlZd dlZddlmZ d ZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZdZd ZdZe�ded> eB �Ze�de d> eB �Ze�de d> eB �Ze�d�Z e�d�Z e�d�Z e�d�Z e�d�Z dZ!dZ"dd� Z#d d � Z$d d � Z%dLd!d"�Z&d#d$� Z'd%d&� Z(d'd(� Z)d)d*� Z*d+d,� Z+d-d.� Z,d/d0� Z-d1d2� Z.d3d4� Z/e0e$e1e'ej2e'e3e(e4e(e5e,e6e+e7e.e8d�e/i Z9d5d6� Z:G d7d8� d8e;�Z<ej=j>dk�r�d9d:� Z?nd;d:� Z?e�d�Z@e�d<�ZAe�d=�ZBe�d>�ZCd ZDdZEdZFdZGdZHdZIdMd?d@�ZJdNdBdC�ZKG dDdE� dEe0�ZLG dFdG� dGeM�ZNG dHdI� dIeM�ZOdJdK� ZPdS )O� )�absolute_importN� )�pycompat� � � � � � � � � � � � � i z>Bz>BBz>BHz>BLz>BQ� �� c C s� |dk rt �| d> |B �S |dk r6t�| d> dB |�S |dk rRt�| d> dB |�S |dk rnt�| d> dB |�S t�| d> dB |�S d S ) z6Obtain a value encoding the major type and its length.� r � � r l r r N)�ENCODED_LENGTH_1�pack�ENCODED_LENGTH_2�ENCODED_LENGTH_3�ENCODED_LENGTH_4�ENCODED_LENGTH_5)� majortype�length� r �>/usr/lib64/python3.9/site-packages/mercurial/utils/cborutil.py� encodelengthE s r! c c s t tt| ��V | V d S �N)r! �MAJOR_TYPE_BYTESTRING�len��vr r r �streamencodebytestringS s r' c c s0 t V | D ]}ttt|��V |V q tV dS )z�Convert an iterator of chunks to an indefinite bytestring. Given an input that is iterable and each element in the iterator is representable as bytes, emit an indefinite length bytestring. N)�BEGIN_INDEFINITE_BYTESTRINGr! r# r$ �BREAK)�it�chunkr r r � streamencodebytestringfromiterX s r, r c c sZ t V d}t| �}| ||| � }|t|�7 }ttt|��V |V ||krqPqtV dS )z�Given a large source buffer, emit as an indefinite length bytestring. This is a generator of chunks constituting the encoded CBOR data. r N)r( r$ r! r# r) )�sourceZ chunksize�i�lr+ r r r � streamencodeindefinitebytestringg s r0 c c sF | dks| dk rt d��| dkr.tt| �V nttt| �d �V d S )Nl l���� s big integers not supportedr r )� ValueErrorr! �MAJOR_TYPE_UINT�MAJOR_TYPE_NEGINT�absr% r r r �streamencodeint~ s r5 c c s2 t tt| ��V | D ]}t|�D ] }|V q qdS )z)Encode a known size iterable to an array.N)r! �MAJOR_TYPE_ARRAYr$ � streamencode)r/ r. r+ r r r �streamencodearray� s r8 c c s. t V | D ]}t|�D ] }|V qq tV dS )z:Encode an iterator of items to an indefinite length array.N)�BEGIN_INDEFINITE_ARRAYr7 r) )r* r. r+ r r r �streamencodearrayfromiter� s r: c C s t | �j| fS r" )�type�__name__r% r r r �_mixedtypesortkey� s r= c c s, t tt�V tt| td��D ] }|V q d S )N��key)r! �MAJOR_TYPE_SEMANTIC�SEMANTIC_TAG_FINITE_SETr8 �sortedr= )�sr+ r r r �streamencodeset� s rD c c s\ t tt| ��V tt�| �dd� d�D ]0\}}t|�D ] }|V q6t|�D ] }|V qJq&dS )z_Encode dictionary to a generator. Does not supporting indefinite length dictionaries. c S s t | d �S �Nr )r= )�xr r r �<lambda>� � z!streamencodemap.<locals>.<lambda>r> N)r! �MAJOR_TYPE_MAPr$ rB r Z iteritemsr7 )�dr? �valuer+ r r r �streamencodemap� s � rL c c sF t V | D ]0\}}t|�D ] }|V qt|�D ] }|V q.q tV dS )zFGiven an iterable of (key, value), encode to an indefinite length map.N)�BEGIN_INDEFINITE_MAPr7 r) )r* r? rK r+ r r r �streamencodemapfromiter� s rN c c s | rdndV d S )N� �� �r )�br r r �streamencodebool� s rR c c s dV d S )N� �r r% r r r �streamencodenone� s rT c C sN t �| j�}|s2t D ] }t| |�s$qt | } q2q|sFtdt| � ��|| �S )a1 Encode a value in a streaming manner. Given an input object, encode it to CBOR recursively. Returns a generator of CBOR encoded bytes. There is no guarantee that each emitted chunk fully decodes to a value or sub-value. Encoding is deterministic - unordered collections are sorted. s do not know how to encode %s)�STREAM_ENCODERS�get� __class__� isinstancer1 r; )r& �fn�tyr r r r7 � s r7 c @ s e Zd ZdZdS )�CBORDecodeErrorz"Represents an error decoding CBOR.N)r< � __module__� __qualname__�__doc__r r r r r[ � s r[ c C s | | S r" r �rQ r. r r r �_elementtointeger� s r` c C s t | | �S r" )�ordr_ r r r r` s s >Hs >Ls >Qc C s t | |�}|d7 }|d? }|t@ }|tkr`t|| |�\}}}|rPd||d tfS dd|tfS �n�|tkr�t|| |�\}}}|r�d| d |d tfS dd|tfS �nt|tk�rRt|| |dd�\}}}|s�dd|tfS |du�rB|| | t| �k�r | || || | � }d||| d tfS t| �| | | } dd| tfS n dddtfS �n�|t k�rht d���n�|t k�r�t|| |�\}}}|�r�d||d t fS dd|tfS �nn|t k�r�t|| |�\}}}|�r�d||d tfS dd|tfS �n,|tk�r�t|| |�\}} }|�sdd|tfS | tk�r�|| t| �k�rBdddtfS t| || �\}}} } |�sjdd| tfS | t k�r|t d ��d||| d tfS t d | ��nz|tk�r|tk�r�dddtfS |tk�r�dddtfS |tk�r�dddtfS |tk�r dddtfS t d | ��n d�sJ �dS ) a� Decode a new CBOR value from a buffer at offset. This function attempts to decode up to one complete CBOR value from ``b`` starting at offset ``offset``. The beginning of a collection (such as an array, map, set, or indefinite length bytestring) counts as a single value. For these special cases, a state flag will indicate that a special value was seen. When called, the function either returns a decoded value or gives a hint as to how many more bytes are needed to do so. By calling the function repeatedly given a stream of bytes, the caller can build up the original values. Returns a tuple with the following elements: * Bool indicating whether a complete value was decoded. * A decoded value if first value is True otherwise None * Integer number of bytes. If positive, the number of bytes read. If negative, the number of bytes we need to read to decode this value or the next chunk in this value. * One of the ``SPECIAL_*`` constants indicating special treatment for this value. ``SPECIAL_NONE`` means this is a fully decoded simple value (such as an integer or bool). r r TFN)�allowindefinites string major type not supported���s, expected array after finite set semantic tags semantic tag %d not alloweds special type %d not allowed)r` � SUBTYPE_MASKr2 � decodeuint� SPECIAL_NONEr3 r# r$ �#SPECIAL_START_INDEFINITE_BYTESTRING�MAJOR_TYPE_STRINGr[ r6 �SPECIAL_START_ARRAYrI �SPECIAL_START_MAPr@ rA � decodeitem�SPECIAL_START_SET�MAJOR_TYPE_SPECIAL� SUBTYPE_FALSE� SUBTYPE_TRUE� SUBTYPE_NULL�SUBTYPE_INDEFINITE�SPECIAL_INDEFINITE_BREAK) rQ �offset�initialr �subtype�completerK � readcount�size�wantedZtagvalueZ readcount2�specialr r r rk s� � � � rk Fc C s� | dk rd| dfS | dkr,|r"dS t d��n| dkr@t d| ��| dkrNt}n2| d kr\t}n$| d krjt}n| d krxt}nt d ��t|�| |jkr�d|�||�d |jfS d dt|�| |j fS dS )a Decode an unsigned integer. ``subtype`` is the lower 5 bits from the initial byte CBOR item "header." ``b`` is a buffer containing bytes. ``offset`` points to the index of the first byte after the byte that ``subtype`` was derived from. ``allowindefinite`` allows the special indefinite length value indicator. Returns a 3-tuple of (successful, value, count). The first element is a bool indicating if decoding completed. The 2nd is the decoded integer value or None if not fully decoded or the subtype is 31 and ``allowindefinite`` is True. The 3rd value is the count of bytes. If positive, it is the number of additional bytes decoded. If negative, it is the number of additional bytes needed to decode this value. r Tr r )TNr s' indefinite length uint not allowed here� s' unsupported subtype on integer type: %dr r r s# bounds condition checking violationFN)r[ �STRUCT_BIG_UBYTE�STRUCT_BIG_USHORT�STRUCT_BIG_ULONG�STRUCT_BIG_ULONGLONGr$ rx � unpack_from)ru rQ rs rb rC r r r re � s, �re c @ s e Zd ZdZddd�ZdS )�bytestringchunka Represents a chunk/segment in an indefinite length bytestring. This behaves like a ``bytes`` but in addition has the ``isfirst`` and ``islast`` attributes indicating whether this chunk is the first or last in an indefinite length bytestring. Fc C s t �| |�}||_||_|S r" )�bytes�__new__ZisfirstZislast)�clsr& �first�last�selfr r r r� � s zbytestringchunk.__new__N)FF)r< r\ r] r^ r� r r r r r� � s r� c @ sR e Zd ZdZdZdZdZdZdZdZ dZ d d � Z e d d � �Z dd d�Zdd� ZdS )� sansiodecodera� A CBOR decoder that doesn't perform its own I/O. To use, construct an instance and feed it segments containing CBOR-encoded bytes via ``decode()``. The return value from ``decode()`` indicates whether a fully-decoded value is available, how many bytes were consumed, and offers a hint as to how many bytes should be fed in next time to decode the next value. The decoder assumes it will decode N discrete CBOR values, not just a single value. i.e. if the bytestream contains uints packed one after the other, the decoder will decode them all, rather than just the initial one. When ``decode()`` indicates a value is available, call ``getavailable()`` to return all fully decoded values. ``decode()`` can partially decode input. It is up to the caller to keep track of what data was consumed and to pass unconsumed data in on the next invocation. The decoder decodes atomically at the *item* level. See ``decodeitem()``. If an *item* cannot be fully decoded, the decoder won't record it as partially consumed. Instead, the caller will be instructed to pass in the initial bytes of this item on the next invocation. This does result in some redundant parsing. But the overhead should be minimal. This decoder only supports a subset of CBOR as required by Mercurial. It lacks support for: * Indefinite length arrays * Indefinite length maps * Use of indefinite length bytestrings as keys or values within arrays, maps, or sets. * Nested arrays, maps, or sets within sets * Any semantic tag that isn't a mathematical finite set * Floating point numbers * Undefined special value CBOR types are decoded to Python types as follows: uint -> int negint -> int bytestring -> bytes map -> dict array -> list True -> bool False -> bool null -> None indefinite length bytestring chunk -> [bytestringchunk] The only non-obvious mapping here is an indefinite length bytestring to the ``bytestringchunk`` type. This is to facilitate streaming indefinite length bytestrings out of the decoder and to differentiate a regular bytestring from an indefinite length bytestring. r r r r r r r c C s$ d| _ | j| _g | _d | _g | _d S rE )�decodedbytecount� _STATE_NONE�_state�_collectionstack�_currentmapkey�_decodedvalues�r� r r r �__init__4 s zsansiodecoder.__init__c C s | j | jkS )z2Whether the decoder has partially decoded a value.)r� r� r� r r r � inprogressG s zsansiodecoder.inprogressc C sr |st | j�ddfS |}|t|�k �r^t||�\}}}}|dkrN| j|7 _|st|dk s^J �t | j�|| | fS ||7 }| j| jk�r4|tkr�| j�|� n�|t kr�| j �|g d�� | j | _nl|t kr�| j �|i d�� | j | _nH|tk�r| j �|t� d�� | j| _n |tk�r$| j| _n td| ���n�| j| j k�r�|tk�rx| j d }|d �|� |d d8 < �q�|t k�r�| j d } g } | d �| � | d d8 < | j �|| d�� n�|t k�r| j d } i } | d �| � | d d8 < | j �|| d�� | j | _nt|tk�rh| j d } t� } | d �| � | d d8 < | j �|| d�� | j| _n |tk�r|td��n td | ���nF| j| j k�r�|tk�r�|| _| j| _n:|tk�r�td ��n&|t t tfv �r�td ��n td | ���n�| j| jk�r`|tk�r>| j d } || d | j< | d d8 < | j | _�n|t k�r�| j d } g } | | d | j< | d d8 < | j �|| d�� | j | _n�|t k�r�| j d } i } | | d | j< | d d8 < | j �|| d�� | j | _nt|tk�r6| j d } t� } | | d | j< | d d8 < | j �|| d�� | j| _n |tk�rJtd ��n td| ��d| _�nr| j| jk�r�|tk�r�| j d } | d �|� | d d8 < n:|tk�r�td��n&|t t tfv �r�td��n td| ��n�| j| jk�rX|tk�r| j�t|dd�� | j| _nB|tk�rJ| j�tdddd�� | j �r@J �| j| _n td| ��nz| j| jk�r�|tk�r�| j�t|�� n@|tk�r�| j�tddd�� | j �r�J �| j| _n td| ��ntd| j ��| j| j | j | jfv�r�q| j d } | d �rq| j �� | j �r@t| j t | j t| jit | j d d � | _n| j�| d � | j| _�q�qt | j�|| dfS )a� Attempt to decode bytes from an input buffer. ``b`` is a collection of bytes and ``offset`` is the byte offset within that buffer from which to begin reading data. ``b`` must support ``len()`` and accessing bytes slices via ``__slice__``. Typically ``bytes`` instances are used. Returns a tuple with the following fields: * Bool indicating whether values are available for retrieval. * Integer indicating the number of bytes that were fully consumed, starting from ``offset``. * Integer indicating the number of bytes that are desired for the next call in order to decode an item. r )� remaining� vs unhandled special state: %drc r� r� r s9 indefinite length bytestrings not allowed as array valuess5 unhandled special item when expecting array value: %ds5 indefinite length bytestrings not allowed as map keyss% collections not supported as map keyss1 unhandled special item when expecting map key: %ds7 indefinite length bytestrings not allowed as map valuess3 unhandled special item when expecting map value: %dNs7 indefinite length bytestrings not allowed as set valuess% collections not allowed as set valuess3 unhandled special item when expecting set value: %dT)r� rH )r� r� s< unexpected special value when expecting bytestring chunk: %d)r� s unhandled decoder state: %d) �boolr� r$ rk r� r� r� rf �appendri r� �_STATE_WANT_ARRAY_VALUErj �_STATE_WANT_MAP_KEYrl �set�_STATE_WANT_SET_VALUErg �"_STATE_WANT_BYTESTRING_CHUNK_FIRSTr[ r� �_STATE_WANT_MAP_VALUE�addr� �'_STATE_WANT_BYTESTRING_CHUNK_SUBSEQUENTrr �pop�list�dictr; ) r� rQ rs Z initialoffsetrv rK rw rz �cZlastcZnewvaluer r r �decodeL s� ��� �� �� � �� � �� ��� ����� �� �� �� ��� ����� � � �� ���� ���zsansiodecoder.decodec C s t | j�}g | _|S )z�Returns an iterator over fully decoded values. Once values are retrieved, they won't be available on the next call. )r� r� )r� r/ r r r � getavailable� s zsansiodecoder.getavailableN)r )r< r\ r] r^ r� r� r� r� r� r� r� r� �propertyr� r� r� r r r r r� � s 8 r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �bufferingdecoderah A CBOR decoder that buffers undecoded input. This is a glorified wrapper around ``sansiodecoder`` that adds a buffering layer. All input that isn't consumed by ``sansiodecoder`` will be buffered and concatenated with any new input that arrives later. TODO consider adding limits as to the maximum amount of data that can be buffered. c C s t � | _g | _d| _d S rE )r� �_decoder�_chunks�_wantedr� r r r r� � s zbufferingdecoder.__init__c C s� t |t�rt|�}| jr�t|�| jk rP| j�|� | jt|�8 _dd| jfS t|�}| j�|� d�| j�}g | _t|�| }nd}| j� |�\}}}|| _|t|�k r�| j�||d� � ||| |fS )a0 Attempt to decode bytes to CBOR values. Returns a tuple with the following fields: * Bool indicating whether new values are available for retrieval. * Integer number of bytes decoded from the new input. * Integer number of bytes wanted to decode the next value. Fr rH N) rX � bytearrayr� r� r$ r� r� �joinr� r� )r� rQ ZnewlenZoldlenZ availablerw ry r r r r� � s$ zbufferingdecoder.decodec C s | j �� S r" )r� r� r� r r r r� # s z bufferingdecoder.getavailableN)r< r\ r] r^ r� r� r� r r r r r� � s 3r� c C sH | sg S t � }|�| �\}}}|t| �kr2td��|jr@td��|�� S )av Decode all CBOR items present in an iterable of bytes. In addition to regular decode errors, raises CBORDecodeError if the entirety of the passed buffer does not fully decode to complete CBOR values. This includes failure to decode any value, incomplete collection types, incomplete indefinite length items, and extra data at the end of the buffer. s input data not fully consumeds input data not complete)r� r� r$ r[ r� r� )rQ �decoderZ havevaluesrw Z wantbytesr r r � decodeall' s r� )r )r )r F)QZ __future__r �struct�sys� r r2 r3 r# rh r6 rI r@ rm rd rn ro rp ZSUBTYPE_HALF_FLOATZSUBTYPE_SINGLE_FLOATZSUBTYPE_DOUBLE_FLOATrq rA r r( r9 rM �Structr r r r r r) Z BREAK_INTr! r' r, r0 r5 r8 r: r= rD rL rN rR rT r� �intZlongr� �tupler� r� r� r; rU r7 � Exceptionr[ � version_info�majorr` r| r} r~ r rf rg ri rj rl rr rk re r� �objectr� r� r� r r r r �<module> s� � � � � 3 pG
cải xoăn