CasperSecurity
U
yi � @ 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/lib/python3.8/xdrlib.py�__init__ s zError.__init__c C s
t | j�S r )�reprr
�r r r r
�__repr__ s zError.__repr__c C s
t | j�S r )�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
sF z � | |�W S t jk
r@ } zt|jd �d �W 5 d }~X Y nX d S �Nr )�struct�errorr �args)r �value�e��functionr r
�result& s z&raise_conversion_error.<locals>.resultr )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 | � � d S r ��resetr r r r
r 2 s zPacker.__init__c C s t � | _d S r )r � _Packer__bufr r r r
r# 5 s z Packer.resetc C s
| j �� S r )r$ �getvaluer r r r
�
get_buffer8 s zPacker.get_bufferc C s | j �t�d|�� d S )N�>L�r$ �writer Zpack�r �xr r r
� pack_uint= s zPacker.pack_uintc C s | j �t�d|�� d S )N�>lr( r* r r r
�pack_intA s zPacker.pack_intc C s" |r| j �d� n | j �d� d S )Ns s )r$ r) r* r r r
� pack_boolG s zPacker.pack_boolc
C s� z| � |d? d@ � W n8 ttjfk
rN } zt|jd �d �W 5 d }~X Y nX z| � |d@ � W n8 ttjfk
r� } zt|jd �d �W 5 d }~X Y 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 �t�d|�� d S )N�>fr( r* r r r
�
pack_floatW s zPacker.pack_floatc C s | j �t�d|�� d S )N�>dr( r* r r r
� pack_double[ s zPacker.pack_doublec C sP |dk rt d��|d |� }|d d d }||t|� d }| j�|� d S )Nr � fstring size must be nonnegative� � � )�
ValueError�lenr$ r) )r �n�s�datar r r
� pack_fstring_ s zPacker.pack_fstringc C s" t |�}| �|� | �||� d S r )r< r, r@ )r r> r= r r r
� pack_stringi s
zPacker.pack_stringc C s* |D ]}| � d� ||� q| � d� d S )N� r )r, )r �list� pack_item�itemr r r
� pack_listq s
zPacker.pack_listc C s* t |�|krtd��|D ] }||� qd S )Nzwrong array size)r< r; )r r= rC rD rE r r r
� pack_farrayw s zPacker.pack_farrayc C s$ t |�}| �|� | �|||� d S r )r< r, rG )r rC rD r= r r r
�
pack_array} s
zPacker.pack_arrayN)r r r r r r# r& Zget_bufr! r, r. Z pack_enumr/ r2 Z
pack_hyperr4 r6 r@ Z pack_fopaquerA Z pack_opaqueZ
pack_bytesrF rG rH r r r r
r / s2
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 | � |� d S r r" �r r? r r r
r � s zUnpacker.__init__c C s || _ d| _d S r )�_Unpacker__buf�_Unpacker__posrI r r r
r# � s zUnpacker.resetc C s | j S r �rK r r r r
� get_position� s zUnpacker.get_positionc C s
|| _ d S r rL )r Zpositionr r r
� set_position� s zUnpacker.set_positionc C s | j S r )rJ r r r r
r& � s zUnpacker.get_bufferc C s | j t| j�k rtd��d S )Nzunextracted data remains)rK r< rJ r r r r r
�done� s z
Unpacker.donec C sB | j }|d | _ }| j||� }t|�dk r2t�t�d|�d S )Nr9 r' r �rK rJ r< �EOFErrorr Zunpack�r �i�jr? r r r
� unpack_uint� s zUnpacker.unpack_uintc C sB | j }|d | _ }| j||� }t|�dk r2t�t�d|�d S )Nr9 r- r rP rR r r r
�
unpack_int� s zUnpacker.unpack_intc C s t | �� �S r )�boolrV r r r r
� unpack_bool� s zUnpacker.unpack_boolc C s | � � }| � � }t|�d> |B S )Nr0 )rU �int)r �hi�lor r r
�
unpack_uhyper� s zUnpacker.unpack_uhyperc C s | � � }|dkr|d }|S )Nl l )r\ r* r r r
� unpack_hyper� s zUnpacker.unpack_hyperc C sB | j }|d | _ }| j||� }t|�dk r2t�t�d|�d S )Nr9 r3 r rP rR r r r
� unpack_float� s zUnpacker.unpack_floatc C sB | j }|d | _ }| j||� }t|�dk r2t�t�d|�d S )N� r5 r rP rR 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 r7 r8 r9 )r; rK r< rJ rQ )r r= rS rT r r r
�unpack_fstring� s zUnpacker.unpack_fstringc C s | � � }| �|�S r )rU ra )r r= r r r
�
unpack_string� s zUnpacker.unpack_stringc C sB g }| � � }|dkrq>|dkr,td|f ��|� }|�|� q|S )Nr rB z0 or 1 expected, got %r)rU r �append)r � unpack_itemrC r+ rE r r r
� unpack_list� s zUnpacker.unpack_listc C s" g }t |�D ]}|�|� � q |S r )�rangerc )r r= rd rC rS r r r
�
unpack_farray� s zUnpacker.unpack_farrayc C s | � � }| �||�S r )rU rg )r rd r= r r r
� unpack_array� s zUnpacker.unpack_arrayN)r r r r r r# rM rN r& rO rU rV Z unpack_enumrX r\ r] r^ r` ra Zunpack_fopaquerb Z
unpack_opaqueZ unpack_bytesre rg rh r r r r
r � s.
)
r r �ior � functoolsr �__all__� Exceptionr r r! r r r r r r
�<module> s U