3 \ � @ st d dl Z d dlmZ d dlmZ ddddgZG dd� de�ZG d d� de�Zd d � Z G d d� d�Z G d d� d�Z dS )� N)�BytesIO)�wraps�Error�Packer�Unpacker�ConversionErrorc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c C s || _ d S )N)�msg)�selfr � r � /usr/lib64/python3.6/xdrlib.py�__init__ s zError.__init__c C s t | j�S )N)�reprr )r r r r �__repr__ s zError.__repr__c C s t | j�S )N)�strr )r r r r �__str__ s z Error.__str__N)�__name__� __module__� __qualname__r r r r r r r r s c @ s e Zd ZdS )r N)r r r r r r r r s c s t � �� fdd��}|S )Nc sD y � | |�S t jk r> } zt|jd �d �W Y d d }~X nX d S )Nr )�struct�errorr �args)r �value�e)�functionr r �result& s z&raise_conversion_error.<locals>.result)r )r r r )r r �raise_conversion_error# s r c @ s� e Zd Zdd� Zdd� Zdd� ZeZedd� �Zed d � �Z e Z d d � Z d d� Z e Z edd� �Zedd� �Zdd� ZeZdd� ZeZeZdd� Zdd� Zdd � Zd S ) r c C s | j � d S )N)�reset)r r r r r 2 s zPacker.__init__c C s t � | _d S )N)r � _Packer__buf)r r r r r 5 s z Packer.resetc C s | j j� S )N)r �getvalue)r r r r � get_buffer8 s zPacker.get_bufferc C s | j jtjd|�� d S )Nz>L)r �writer �pack)r �xr r r � pack_uint= s zPacker.pack_uintc C s | j jtjd|�� d S )Nz>l)r r r r! )r r" r r r �pack_intA s zPacker.pack_intc C s" |r| j jd� n | j jd� d S )Ns s )r r )r r" r r r � pack_boolG s zPacker.pack_boolc C s� y| j |d? d@ � W n8 ttjfk rN } zt|jd �d �W Y d d }~X nX y| j |d@ � W n8 ttjfk r� } zt|jd �d �W Y d d }~X nX d S )N� l �� r )r# � TypeErrorr r r r )r r" r r r r � pack_uhyperK s "zPacker.pack_uhyperc C s | j jtjd|�� d S )Nz>f)r r r r! )r r" r r r � pack_floatW s zPacker.pack_floatc C s | j jtjd|�� d S )Nz>d)r r r r! )r r" r r r � pack_double[ s zPacker.pack_doublec C sP |dk rt d��|d |� }|d d d }||t|� d }| jj|� d S )Nr z fstring size must be nonnegative� � � )� ValueError�lenr r )r �n�s�datar r r � pack_fstring_ s zPacker.pack_fstringc C s" t |�}| j|� | j||� d S )N)r/ r# r3 )r r1 r0 r r r � pack_stringi s zPacker.pack_stringc C s. x |D ]}| j d� ||� qW | j d� d S )N� r )r# )r �list� pack_item�itemr r r � pack_listq s zPacker.pack_listc C s. t |�|krtd��x|D ] }||� qW d S )Nzwrong array size)r/ r. )r r0 r6 r7 r8 r r r � pack_farrayw s zPacker.pack_farrayc C s$ t |�}| j|� | j|||� d S )N)r/ r# r: )r r6 r7 r0 r r r � pack_array} s zPacker.pack_arrayN)r r r r r r Zget_bufr r# r$ Z pack_enumr% r( Z pack_hyperr) r* r3 Z pack_fopaquer4 Z pack_opaqueZ pack_bytesr9 r: r; r r r r r / s( c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z e Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � ZeZd d � ZeZeZd d � Zd!d"� Zd#d$� Zd%S )&r c C s | j |� d S )N)r )r r2 r r r r � s zUnpacker.__init__c C s || _ d| _d S )Nr )�_Unpacker__buf�_Unpacker__pos)r r2 r r r r � s zUnpacker.resetc C s | j S )N)r= )r r r r � get_position� s zUnpacker.get_positionc C s || _ d S )N)r= )r Zpositionr r r � set_position� s zUnpacker.set_positionc C s | j S )N)r< )r r r r r � s zUnpacker.get_bufferc C s | j t| j�k rtd��d S )Nzunextracted data remains)r= r/ r< r )r r r r �done� s z Unpacker.donec C sB | j }|d | _ }| j||� }t|�dk r2t�tjd|�d S )Nr, z>Lr )r= r< r/ �EOFErrorr �unpack)r �i�jr2 r r r � unpack_uint� s zUnpacker.unpack_uintc C sB | j }|d | _ }| j||� }t|�dk r2t�tjd|�d S )Nr, z>lr )r= r< r/ rA r rB )r rC rD r2 r r r � unpack_int� s zUnpacker.unpack_intc C s t | j� �S )N)�boolrF )r r r r � unpack_bool� s zUnpacker.unpack_boolc C s | j � }| j � }t|�d> |B S )Nr& )rE �int)r �hi�lor r r � unpack_uhyper� s zUnpacker.unpack_uhyperc C s | j � }|dkr|d }|S )Nl l )rL )r r" r r r � unpack_hyper� s zUnpacker.unpack_hyperc C sB | j }|d | _ }| j||� }t|�dk r2t�tjd|�d S )Nr, z>fr )r= r< r/ rA r rB )r rC rD r2 r r r � unpack_float� s zUnpacker.unpack_floatc C sB | j }|d | _ }| j||� }t|�dk r2t�tjd|�d S )N� z>dr )r= r< r/ rA r rB )r rC rD r2 r r r � unpack_double� s zUnpacker.unpack_doublec C sT |dk rt d��| j}||d d d }|t| j�kr<t�|| _| j||| � S )Nr z fstring size must be nonnegativer+ r, )r. r= r/ r< rA )r r0 rC rD r r r �unpack_fstring� s zUnpacker.unpack_fstringc C s | j � }| j|�S )N)rE rQ )r r0 r r r � unpack_string� s zUnpacker.unpack_stringc C sF g }x<| j � }|dkrP |dkr.td|f ��|� }|j|� qW |S )Nr r5 z0 or 1 expected, got %r)rE r �append)r � unpack_itemr6 r" r8 r r r � unpack_list� s zUnpacker.unpack_listc C s& g }x t |�D ]}|j|� � qW |S )N)�rangerS )r r0 rT r6 rC r r r � unpack_farray� s zUnpacker.unpack_farrayc C s | j � }| j||�S )N)rE rW )r rT r0 r r r � unpack_array� s zUnpacker.unpack_arrayN)r r r r r r> r? r r@ rE rF Z unpack_enumrH rL rM rN rP rQ Zunpack_fopaquerR Z unpack_opaqueZ unpack_bytesrU rW rX r r r r r � s, ) r �ior � functoolsr �__all__� Exceptionr r r r r r r r r �<module> s U