3
\ � @ sx d Z ddlZddlmZ ddlmZ ddddgZG d d� de�ZG d
d� de�Z d d � Z
G d
d� d�Z G dd� d�Z dS )zRImplements (a subset of) Sun XDR -- eXternal Data Representation.
See: RFC 1014
� N)�BytesIO)�wraps�Error�Packer�Unpacker�ConversionErrorc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r z�Exception class for this module. Use:
except xdrlib.Error as var:
# var has the Error instance for the exception
Public ivars:
msg -- contains the message
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__�__doc__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 )z5 Wrap any raised struct.errors in a ConversionError. c 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 ZdZdd� Zdd� Zdd� ZeZedd � �Z ed
d � �Z
e
Z d d
� Z dd� Z
e
Zedd� �Zedd� �Zdd� ZeZdd� ZeZeZdd� Zdd� Zd d � Zd S ) r z0Pack various data representations into a buffer.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)r0 r$ r4 )r r2 r1 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)r0 r/ )r r1 r7 r8 r9 r
r
r � pack_farrayw s
zPacker.pack_farrayc C s$ t |�}| j|� | j|||� d S )N)r0 r$ r; )r r7 r8 r1 r
r
r �
pack_array} s
zPacker.pack_arrayN)r r r r r r r Zget_bufr r$ r% Z pack_enumr& r) Z
pack_hyperr* r+ r4 Z pack_fopaquer5 Z pack_opaqueZ
pack_bytesr: r; r< r
r
r
r r / s*
c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z e Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � ZeZd d � ZeZeZd d!� Zd"d#� Zd$d%� Zd&S )'r z;Unpacks various data representations from the given buffer.c C s | j |� d S )N)r )r r3 r
r
r r � s zUnpacker.__init__c C s || _ d| _d S )Nr )�_Unpacker__buf�_Unpacker__pos)r r3 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> r0 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= r0 �EOFErrorr �unpack)r �i�jr3 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= r0 rB r rC )r rD rE r3 r
r
r �
unpack_int� s zUnpacker.unpack_intc C s t | j� �S )N)�boolrG )r r
r
r � unpack_bool� s zUnpacker.unpack_boolc C s | j � }| j � }t|�d> |B S )Nr' )rF �int)r �hi�lor
r
r �
unpack_uhyper� s zUnpacker.unpack_uhyperc C s | j � }|dkr|d }|S )Nl l )rM )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= r0 rB r rC )r rD rE r3 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= r0 rB r rC )r rD rE r3 r
r
r �
unpack_double� s zUnpacker.unpack_doublec C sT |dk rt d��| j}||d d d }|t| j�kr r0 r= rB )r r1 rD rE r
r
r �unpack_fstring� s zUnpacker.unpack_fstringc C s | j � }| j|�S )N)rF rR )r r1 r
r
r �
unpack_string� s zUnpacker.unpack_stringc C sF g }x<| j � }|dkrP |dkr.td|f ��|� }|j|� qW |S )Nr r6 z0 or 1 expected, got %r)rF r �append)r � unpack_itemr7 r# r9 r
r
r � unpack_list� s zUnpacker.unpack_listc C s& g }x t |�D ]}|j|� � qW |S )N)�rangerT )r r1 rU r7 rD r
r
r �
unpack_farray� s zUnpacker.unpack_farrayc C s | j � }| j||�S )N)rF rX )r rU r1 r
r
r � unpack_array� s zUnpacker.unpack_arrayN)r r r r r r r? r@ r rA rF rG Z unpack_enumrI rM rN rO rQ rR Zunpack_fopaquerS Z
unpack_opaqueZ unpack_bytesrV rX rY r
r
r
r r � s.
)
r r �ior � functoolsr �__all__� Exceptionr r r r r r
r
r
r � s U