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..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 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 � s U