o
    “Hb  ã                   @   st   d Z ddlZddlmZ ddlmZ g d¢ZG dd„ deƒZG dd	„ d	eƒZ	d
d„ Z
G dd„ dƒZG dd„ dƒZdS )zRImplements (a subset of) Sun XDR -- eXternal Data Representation.

See: RFC 1014

é    N)ÚBytesIO©Úwraps)ÚErrorÚPackerÚUnpackerÚConversionErrorc                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	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   ó
   || _ d S ©N)Úmsg)Úselfr   © r   ú/usr/lib/python3.10/xdrlib.pyÚ__init__   ó   
zError.__init__c                 C   ó
   t | jƒS r
   )Úreprr   ©r   r   r   r   Ú__repr__   r   zError.__repr__c                 C   r   r
   )Ústrr   r   r   r   r   Ú__str__   r   zError.__str__N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r      s
    	r   c                   @   s   e Zd ZdS )r   N)r   r   r   r   r   r   r   r       s    r   c                    s   t ˆ ƒ‡ fdd„ƒ}|S )z5 Wrap any raised struct.errors in a ConversionError. c              
      s8   zˆ | |ƒW S  t jy } zt|jd ƒd ‚d }~ww ©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 Zd ZdZdd„ Zdd„ Zdd„ ZeZedd	„ ƒZ	ed
d„ ƒZ
e
Zdd„ Zdd„ ZeZedd„ ƒZedd„ ƒZdd„ ZeZdd„ ZeZeZdd„ Zdd„ Zdd„ ZdS )r   z0Pack various data representations into a buffer.c                 C   s   |   ¡  d S r
   ©Úresetr   r   r   r   r   2   ó   zPacker.__init__c                 C   s   t ƒ | _d S r
   )r   Ú_Packer__bufr   r   r   r   r&   5   r'   zPacker.resetc                 C   s
   | j  ¡ S r
   )r(   Úgetvaluer   r   r   r   Ú
get_buffer8   r   zPacker.get_bufferc                 C   ó   | j  t d|¡¡ d S )Nú>L©r(   Úwriter   Zpack©r   Úxr   r   r   Ú	pack_uint=   ó   zPacker.pack_uintc                 C   r+   )Nú>lr-   r/   r   r   r   Úpack_intA   r2   zPacker.pack_intc                 C   s$   |r
| j  d¡ d S | j  d¡ d S )Ns      s       )r(   r.   r/   r   r   r   Ú	pack_boolG   s   zPacker.pack_boolc              
   C   sŠ   z|   |d? d@ ¡ W n ttjfy" } zt|jd ƒd ‚d }~ww z
|   |d@ ¡ W d S  ttjfyD } zt|jd ƒd ‚d }~ww )Né    l   ÿÿ r   )r1   Ú	TypeErrorr   r   r   r   )r   r0   r    r   r   r   Úpack_uhyperK   s   €ÿ€ÿzPacker.pack_uhyperc                 C   r+   )Nú>fr-   r/   r   r   r   Ú
pack_floatW   r2   zPacker.pack_floatc                 C   r+   )Nú>dr-   r/   r   r   r   Úpack_double[   r2   zPacker.pack_doublec                 C   sP   |dk rt 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   zPacker.pack_fstringc                 C   s"   t |ƒ}|  |¡ |  ||¡ d S r
   )rB   r1   rF   )r   rD   rC   r   r   r   Úpack_stringi   s   
zPacker.pack_stringc                 C   s*   |D ]}|   d¡ ||ƒ q|   d¡ d S )Né   r   )r1   )r   ÚlistÚ	pack_itemÚitemr   r   r   Ú	pack_listq   s   

zPacker.pack_listc                 C   s*   t |ƒ|kr
tdƒ‚|D ]}||ƒ qd S )Nzwrong array size)rB   rA   )r   rC   rI   rJ   rK   r   r   r   Úpack_farrayw   s
   
ÿzPacker.pack_farrayc                 C   s$   t |ƒ}|  |¡ |  |||¡ d S r
   )rB   r1   rM   )r   rI   rJ   rC   r   r   r   Ú
pack_array}   s   
zPacker.pack_arrayN)r   r   r   r   r   r&   r*   Zget_bufr$   r1   r4   Z	pack_enumr5   r8   Z
pack_hyperr:   r<   rF   Zpack_fopaquerG   Zpack_opaqueZ
pack_bytesrL   rM   rN   r   r   r   r   r   /   s4    




r   c                   @   s°   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ ZeZdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZeZdd„ ZeZeZd d!„ Zd"d#„ Zd$d%„ Zd&S )'r   z;Unpacks various data representations from the given buffer.c                 C   s   |   |¡ d S r
   r%   ©r   rE   r   r   r   r   ‡   s   zUnpacker.__init__c                 C   s   || _ d| _d S r   )Ú_Unpacker__bufÚ_Unpacker__posrO   r   r   r   r&   Š   s   
zUnpacker.resetc                 C   ó   | j S r
   ©rQ   r   r   r   r   Úget_positionŽ   ó   zUnpacker.get_positionc                 C   r	   r
   rS   )r   Zpositionr   r   r   Úset_position‘   r   zUnpacker.set_positionc                 C   rR   r
   )rP   r   r   r   r   r*   ”   rU   zUnpacker.get_bufferc                 C   s   | j t| jƒk rtdƒ‚d S )Nzunextracted data remains)rQ   rB   rP   r   r   r   r   r   Údone—   s   ÿzUnpacker.donec                 C   óB   | j }|d  | _ }| j||… }t|ƒdk rt‚t d|¡d S )Nr?   r,   r   ©rQ   rP   rB   ÚEOFErrorr   Úunpack©r   ÚiÚjrE   r   r   r   Úunpack_uint›   ó   zUnpacker.unpack_uintc                 C   rX   )Nr?   r3   r   rY   r\   r   r   r   Ú
unpack_int£   r`   zUnpacker.unpack_intc                 C   s   t |  ¡ ƒS r
   )Úboolra   r   r   r   r   Úunpack_bool­   r'   zUnpacker.unpack_boolc                 C   s    |   ¡ }|   ¡ }t|ƒd> |B S )Nr6   )r_   Úint)r   ÚhiÚlor   r   r   Úunpack_uhyper°   s   zUnpacker.unpack_uhyperc                 C   s   |   ¡ }|dkr|d }|S )Nl            l            )rg   r/   r   r   r   Úunpack_hyperµ   s   zUnpacker.unpack_hyperc                 C   rX   )Nr?   r9   r   rY   r\   r   r   r   Úunpack_float»   r`   zUnpacker.unpack_floatc                 C   rX   )Né   r;   r   rY   r\   r   r   r   Úunpack_doubleÃ   r`   zUnpacker.unpack_doublec                 C   sT   |dk rt dƒ‚| j}||d d d  }|t| jƒkrt‚|| _| j||| … S )Nr   r=   r>   r?   )rA   rQ   rB   rP   rZ   )r   rC   r]   r^   r   r   r   Úunpack_fstringË   s   zUnpacker.unpack_fstringc                 C   s   |   ¡ }|  |¡S r
   )r_   rl   )r   rC   r   r   r   Úunpack_string×   s   
zUnpacker.unpack_stringc                 C   sB   g }	 |   ¡ }|dkr|S |dkrtd|f ƒ‚|ƒ }| |¡ q)NrH   r   z0 or 1 expected, got %r)r_   r   Úappend)r   Úunpack_itemrI   r0   rK   r   r   r   Úunpack_listÞ   s   ü
úzUnpacker.unpack_listc                 C   s"   g }t |ƒD ]}| |ƒ ¡ q|S r
   )Úrangern   )r   rC   ro   rI   r]   r   r   r   Úunpack_farrayé   s   zUnpacker.unpack_farrayc                 C   s   |   ¡ }|  ||¡S r
   )r_   rr   )r   ro   rC   r   r   r   Úunpack_arrayï   s   zUnpacker.unpack_arrayN)r   r   r   r   r   r&   rT   rV   r*   rW   r_   ra   Zunpack_enumrc   rg   rh   ri   rk   rl   Zunpack_fopaquerm   Zunpack_opaqueZunpack_bytesrp   rr   rs   r   r   r   r   r   „   s0    
r   )r   r   Úior   Ú	functoolsr   Ú__all__Ú	Exceptionr   r   r$   r   r   r   r   r   r   Ú<module>   s    U