File: //usr/lib/python3/dist-packages/hgext/fsmonitor/pywatchman/__pycache__/pybser.cpython-310.pyc
o
�]Lb H � @ s. d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZddl m
Z
dZ dZ dZ
dZd Zd
Zd Zd Zd
ZdZdZdZdZdZe
jrPeefZ eZ dd� Z e Z ne efZ eZ dZ!dZ"dd� Z#dd� Z$G dd � d e%�Z&d.d d �Z'G d d � d e%�Z(G d!d"� d"e%�Z)d#d$� Z*d%d&� Z+d'd(� Z,d/d*d+�Z-d/d,d-�Z.dS )0� )�absolute_import�division�print_functionN� )�compat� � � � � � � � � � �
� � �
c C s t | ��d�S )N�ascii)�str�encode)�i� r �C/usr/lib/python3/dist-packages/hgext/fsmonitor/pywatchman/pybser.py�tobytes< s r s s c C sx d| kr dkr dS d| krdkrdS d| kr#dkr$d S t d
�| kr3t d �kr4d S td
t| � ��)z5Return the smallest size int that can store the valuei����� r i ���i� � i �i���� l���� l ���� � zCannot represent value: )�long� RuntimeErrorr )�xr r r � _int_sizeK s ����r# c C s& | | }t jrt|t�st|f�}|S �N)r �PYTHON3�
isinstance�bytes)�buf�pos�retr r r �_buf_posY s
r+ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � _bser_bufferc C s� || _ t�d�| _| j dkr$t�ttt��d | jdt� tt�| _ d S | j dks+J �t�ttt
��d | jdt
� tt
�| _ d S )Ni r � sr r ) � bser_version�ctypes�create_string_bufferr( �struct� pack_intor �len� EMPTY_HEADER�wpos�EMPTY_HEADER_V2)�self�versionr r r �__init__b s
��z_bser_buffer.__init__c C sN t �| j�| j |k r%t �| jt �| j�d � t �| j�| j |k s d S d S )Nr )r/ �sizeofr( r5 �resize)r7 �sizer r r � ensure_sizet s �z_bser_buffer.ensure_sizec C s� t |�}|d }| �|� |dkr t�d| j| jt|� n4|dkr-t�d| j| jt|� n$|dkr=t�d| j| jt|� n|dkrMt�d| j| jt |� nt
d ��| j|7 _d S )
Nr s =cbr s =chr s =cir s =cqz Cannot represent this long value) r# r= r1 r2 r( r5 � BSER_INT8�
BSER_INT16�
BSER_INT32�
BSER_INT64r! )r7 �valr< �to_writer r r � append_longx s
z_bser_buffer.append_longc C s t |t�r
|�d�}t|�}t|�}d| | }| �|� |dkr5t�dt|� d | j | j
t t ||� nL|dkrMt�dt|� d | j | j
t t
||� n4|dkret�dt|� d | j | j
t t||� n |d kr}t�d
t|� d | j | j
t t||� ntd ��| j
|7 _
d S ) N�utf-8r r � =ccbr- � =cchr � =ccir � =ccqz"Cannot represent this string value)r& �unicoder r3 r# r= r1 r2 r r( r5 �BSER_BYTESTRINGr> r? r@ rA r! )r7 �s�s_lenr< rC r r r �
append_string� s`
� � � �
z_bser_buffer.append_stringc C s t |t�r&d}| �|� |rt}nt}t�d| j| j|� | j|7 _d S |d u rDd}| �|� t�d| j| jt � | j|7 _d S t |t
t f�rR| � |� d S t |t
�r^| �|� d S t |t�r~d}| �|� t�d| j| jt|� | j|7 _d S t |tj��r t |tj��r t|�}t|�}d| }| �|� |dkr�t�d| j| jtt|� n7|dkr�t�d| j| jtt|� n&|dkr�t�d | j| jtt|� n|d
kr�t�d | j| jtt|� ntd ��| j|7 _t j r�|� � }n|� � }|D ]\}}| �|� | � |� q�d S t |tj!��r�t |tj��r�t|�}t|�}d| }| �|� |dk�r<t�d| j| jt"t|� n:|dk�rNt�d| j| jt"t|� n(|dk�r`t�d | j| jt"t|� n|d
k�rrt�d | j| jt"t|� ntd
��| j|7 _|D ]}| � |� �qd S td��)Nr s =c� s =cdr rF rG r rH r rI z#Cannot represent this mapping valuez$Cannot represent this sequence valuez#Cannot represent unknown value type)#r& �boolr= � BSER_TRUE�
BSER_FALSEr1 r2 r( r5 � BSER_NULL�intr rD � STRING_TYPESrN �float� BSER_REAL� collections�Mapping�Sizedr3 r# � BSER_OBJECTr> r? r@ rA r! r r% �items� iteritems�append_recursive�Iterable�
BSER_ARRAY) r7 rB �needed� to_encode�val_lenr< r] �k�vr r r r^ � s�
�
����
��
�
�
�
� �z _bser_buffer.append_recursiveN)�__name__�
__module__� __qualname__r9 r= rD rN r^ r r r r r, a s 3r, c C s� t |d�}|�| � |dkr |jtt� }t�d|jd|� n|jtt� }t�d|jd|� t�d|jd|� |jj d |j� S )N)r8 r � =i� r � )
r, r^ r5 r3 r4 r1 r2 r( r6 �raw)�objr8 � capabilities�bser_buf�obj_lenr r r �dumps7 s
rq c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � _BunserDict��_keys�_valuesc C s || _ || _d S r$ rs )r7 �keys�valuesr r r r9 K s
z_BunserDict.__init__c C s
| � |�S r$ )� __getitem__)r7 �namer r r � __getattr__O �
z_BunserDict.__getattr__c C s^ t |ttf�r | j| S |�d�r|dd � }z
| j| j�|� W S ty. td| ��w )N�st_rj z_BunserDict has no key %s) r& rT r ru �
startswithrt �index�
ValueError�KeyError)r7 �keyr r r rx R s
�z_BunserDict.__getitem__c C s
t | j�S r$ )r3 rt )r7 r r r �__len__^ r{ z_BunserDict.__len__N)rf rg rh � __slots__r9 rz rx r� r r r r rr H s rr c @ sR e Zd Zddd�Zedd� �Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�BunserTNc C s: || _ || _|d u rd | _d S |d u rd| _d S || _d S )N�strict��mutable�value_encoding� value_errors)r7 r� r� r� r r r r9 c s
zBunser.__init__c C s� zt | |�}W n ty td��w |tkr d}d}n)|tkr%d}d}n |tkr.d}d}n|tkr7d}d }ntd
t�|�� d �|f ��t
� || |d �d
}||| fS )Nz+Invalid bser int encoding, pos out of ranger s =brj s =h� ri rO s =qz-Invalid bser int encoding 0x%s at position %sr r r ) r+ �
IndexErrorr r> r? r@ rA �binascii�hexlify�decoder1 � unpack_from)r( r) �int_typera �fmt�int_valr r r � unser_intn s0 ��� zBunser.unser_intc C s@ | � ||d �\}}t�t|�d ||�d }|�d�|| fS )Nr r- r rE )r� r1 r� r r� �r7 r( r) �str_len�str_valr r r �unser_utf8_string� s zBunser.unser_utf8_stringc C sT | � ||d �\}}t�t|�d ||�d }| jd ur$|�| j| j�}||| fS )Nr r- r )r� r1 r� r r� r� r� r� r r r �unser_bytestring� s
zBunser.unser_bytestringc C sV | � ||d �\}}g }t|�D ]}| �||�\}}|�|� q| js't|�}||fS �Nr )r� �range�loads_recursive�appendr� �tuple)r7 r( r) �arr_len�arr�_�arr_itemr r r � unser_array� s zBunser.unser_arrayc
C s� | � ||d �\}}| jri }ng }g }t|�D ]$}| �||�\}}| �||�\} }| jr2| ||< q|�|� |�| � q| jsEt||�}||fS r� )r� r� r� r� r� r� rr )
r7 r( r) rp rm rv �valsr� r� rB r r r � unser_object� s
zBunser.unser_objectc C s� t ||d �}|tkrtd��t| jdd�}|�||d �\}}| �||�\}}g }t|�D ]G}| jr6i } ng }
tt|��D ])} t ||�t krN|d7 }d } n| �
||�\} }| jrb|| }
| | |
< q>|
� | � q>| jspt ||
�} |� | � q.||fS )Nr z Expect ARRAY to follow TEMPLATErE )r� r� )
r+ r` r! r� r� r� r� r� r3 � BSER_SKIPr� r� rr )r7 r( r) �val_type� keys_bunserrv �nitemsr� r� rm r� �keyidx�eler� r r r �unser_template� s0
zBunser.unser_templatec C s t ||�}|tks|tks|tks|tkr| �||�S |tkr0t�d||d �d }||d fS |t kr:d|d fS |t
krDd|d fS |t krNd |d fS |t krX| �
||�S |tkrb| �||�S |tkrl| �||�S |tkrv| �||�S |tkr�| �||�S tdt�|��d� ��) Ns =dr r rO TFzunhandled bser opcode 0x%sr )r+ r> r? r@ rA r� rW r1 r� rQ rR rS rK r� �BSER_UTF8STRINGr� r` r� r[ r� �
BSER_TEMPLATEr� r r� r� r� )r7 r( r) r� rB r r r r� � s<
��zBunser.loads_recursive�TNN) rf rg rh r9 � staticmethodr� r� r� r� r� r� r� r r r r r� b s
"r� c C s� | dd� t dd� krd}d}t�| d�\}}n.| dd� tdd� krCt| �dk r/td��d}t�d| d�d }t�| d�\}}ntd��||||fS )Nr r r r zInvalid BSER header�I� )r4 r� r� r6 r3 r r1 r� )r( r. �bser_capabilities� expected_len�pos2r r r �_pdu_info_helper� s r� c C s& t | �}|d |d |d |d fS )Nr r r rj �r� �r( �infor r r �pdu_info
s r� c C s t | �}|d |d S )Nr rj r� r� r r r �pdu_len s r� Tc C s^ t | �}|d }|d }t| �|| kr td|| t| �f ��t|||d�}|�| |�d S )au Deserialize a BSER-encoded blob.
@param buf: The buffer to deserialize.
@type buf: bytes
@param mutable: Whether to return mutable results.
@type mutable: bool
@param value_encoding: Optional codec to use to decode values. If
unspecified or None, return values as bytestrings.
@type value_encoding: str
@param value_errors: Optional error handler for codec. 'strict' by default.
The other most common argument is 'surrogateescape' on
Python 3. If value_encoding is None, this is ignored.
@type value_errors: str
r rj z!bser data len %d != header len %dr� r )r� r3 r r� r� )r( r� r� r� r� r� r) �bunserr r r �loads s ��r� c C s ddl m} |�| |||�S )Nr )�load)� r� )�fpr� r� r� r� r r r r� < s r� )r r r� )/�
__future__r r r r� rX r/ r1 �sysr� r r` r[ rK r> r? r@ rA rW rQ rR rS r� r� r� r% r r' rU rJ r rT r r4 r6 r# r+ �objectr, rq rr r� r� r� r� r� r� r r r r �<module> sV
W
%