File: //opt/saltstack/salt/lib/python3.10/site-packages/cheroot/__pycache__/makefile.cpython-310.pyc
o
�N�g�? � @ s� d Z ddlmZmZmZ eZddlZzddlZ W n
e
y% ddl Z Y nw ddl Z ddl m
Z
ddlmZmZ dZG dd � d e j�ZG d
d � d eed e��Ze jsmG d
d� de j�ZG dd� de�Zde jfdd�ZdS e Z ZZdS )zSocket file object.� )�absolute_import�division�print_functionN� )�errors)�
extract_bytes�
memoryviewi @ c @ s e Zd ZdZdd� Zdd� ZdS )�BufferedWriter�-Faux file object attached to a socket object.c C s` | � � t|t�r
td��| j� | j�|� | �� t|�W d � S 1 s)w Y dS )zWrite bytes to buffer.z can't write str to binary streamN) � _checkClosed�
isinstance�str� TypeErrorZ _write_lock�
_write_buf�extend�_flush_unlocked�len)�self�b� r �D/opt/saltstack/salt/lib/python3.10/site-packages/cheroot/makefile.py�write s
$�zBufferedWriter.writec
C sn | � d� | jr5z | j�t| j��}W n tjy) } z |j}W Y d }~nd }~ww | jd |�= | jsd S d S )Nzflush of closed file)r r �rawr �bytes�io�BlockingIOError�characters_written)r �n�er r r r ' s
�� �z BufferedWriter._flush_unlockedN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
G dd� d�Z e j
o6ee�e � �je j�Z[ esJd dd�Zd dd�Zdd� Zd S d dd�Zd dd�Zdd� Zd S ) � MakeFile_PY2r
c O s0 d| _ d| _tjj| g|�R i |�� d| _dS )z Initialize faux file object.r N)�
bytes_read�
bytes_written�socket� _fileobject�__init__� _refcount)r �args�kwargsr r r r( 6 s
zMakeFile_PY2.__init__c C s | j d7 _ d S )Nr )r) �r r r r �_reuse= s zMakeFile_PY2._reusec C s( | j dk r | �� d S | j d8 _ d S )Nr r )r) �closer, r r r �_drop@ s
zMakeFile_PY2._dropc
C s� d}t |�}t|�}||k r@z|| �|||t � �7 }W n tjy9 } z|jd tjvr/� W Y d}~nd}~ww ||k sdS dS )z3Send entire data contents for non-blocking sockets.r N) r r �send�SOCK_WRITE_BLOCKSIZEr& �errorr* r �socket_errors_nonblocking)r �data�
bytes_sentZdata_mvZ payload_sizer r r r r F s
� ����zMakeFile_PY2.writec C s" | j �t|��}| j|7 _|S )z(Send some part of message to the socket.)�_sockr0 r r% )r r4 r5 r r r r0 T s zMakeFile_PY2.sendc C s* | j rd�| j �}g | _ | �|� dS dS )z<Write all data from buffer to socket and reset write buffer.� N)Z_wbuf�joinr )r �bufferr r r �flushZ s
�zMakeFile_PY2.flushc
C sv z| j �|�}| jt|�7 _|W S tjy9 } z|jd tjvo+|jd tj v}|r/� W Y d}~nd}~ww q)z*Receive message of a size from the socket.Tr N)
r6 �recvr$ r r& r2 r* r r3 Zsocket_error_eintr)r �sizer4 r Zwhatr r r r; a s � ����zMakeFile_PY2.recvc @ s e Zd ZdZdd� ZdS )zMakeFile_PY2.FauxSocketz8Faux socket with the minimal interface required by pypy.c C s d S )Nr r, r r r r- s s z MakeFile_PY2.FauxSocket._reuseN)r r r! r" r- r r r r �
FauxSocketp s r= ���c C s* t | j| j�}| j}|�dd� |dk r,t�� | _ | �|�}|s& |� � S |�|� q|�
� }||krM|�d� |� |�}t�� | _| j�|� � � |S t�� | _ || }| �|�}|sc |� � S t |�}||kro|so|S ||kr~|�|� ~ |� � S ||ks�J d||f ��|�|� ||7 }~qS)�$Read data from the socket to buffer.r � Tzrecv(%d) returned %d bytes)
�max� _rbufsize�default_bufsize�_rbuf�seekr �BytesIOr; r �getvalue�tell�readr ) r r< Zrbufsize�bufr4 �buf_len�rv�leftr r r r rI ~ sN
��
�
�
�zMakeFile_PY2.readc C s� | j }|�dd� |�� dkr4|�d� |�|�}|�d�s$t|�|kr3t�� | _ | j �|� � � |S ~|dk r�| j
dkrj|�d� |� � g}t�� | _ d}| j }|dkre|d�}|s\n |� |� |dksUd�
|�S |�dd� t�� | _ | � | j
�}|s� |�� S |�d�}|dkr�|d7 }|�|d|� � | j �||d� � ~ |�� S |�|� qv|�dd� |�� }||kr�|�d� |� |�} t�� | _ | j �|� � � | S t�� | _ | � | j
�}|s� |�� S || }
|�dd|
�}|dk�r |d7 }| j �||d� � |�r|�|d|� � |�� S |d|� S t|�} | |k�r,|�s,|S | |
k�rI|�|d|
� � | j �||
d� � |�� S |�|� || 7 }q�)�$Read line from the socket to buffer.r r@ �
r Nr7 )rD rE rH �readline�endswithr r rF r rI rB r; �appendr8 �findrG ) r r< rJ Zbline�buffersr4 r; �nlrK rL rM r r r r rP � s�
�
�
��
�
�
��zMakeFile_PY2.readlinec C s t | j�� �S �z.Return true if there is buffered data to read.)�boolrD rG r, r r r �has_data s zMakeFile_PY2.has_datac C s |dk r+| j g}d| _ | jdkr| j}n| j} | �|�}|s n|�|� qd�|�S | j }t|�}||krC||d� | _ |d|� S g }|rL|�|� d| _ || }t| j|�}| �|�}|sbn"|�|� t|�}||kr||d� | _ |d|� |d< n||7 }qPd�|�S )r? r r7 r TNr> )rD rB rC r; rR r8 r rA )r r< rT Z recv_sizer4 rK rM r r r r rI ! sJ
�
�
c C s0 | j }|dk r�| jdkr.|dksJ �g }|dkr)| �d�}|s n |�|� |dksd�|�S |�d�}|dkrH|d7 }||d� | _ |d|� S g }|rQ|�|� d| _ | �| j�}|s^n#|�|� |�d�}|dkr�|d7 }||d� | _ |d|� |d< nqUd�|�S |�dd|�}|dkr�|d7 }||d� | _ |d|� S t|�}||kr�||d� | _ |d|� S g }|r�|�|� d| _ | �| j�}|s�nF|�|� || }|�dd|�}|dkr�|d7 }||d� | _ |d|� |d< n t|�}||k�r||d� | _ |d|� |d< n||7 }q�d�|�S )rN r r r7 rO NTr> )rD rB r; rR r8 rS r )r r< r4 rT rU rK rM r r r r rP L s�
�
�
�
c C s
t | j�S rV )rW rD r, r r r rX � s
N)r> )r r r! r" r( r- r/ r r0 r: r; r= �six�PY2r r& r' rD Z string_typesZ_fileobject_uses_str_typerI rP rX r r r r r# 3 s* �
B \
+ Hr# r' c s< e Zd ZdZdejf� fdd� Z� fdd�Zdd� Z� Z S ) � StreamReaderzSocket stream reader.�rc � t � �t�||�|� d| _dS )z Initialize socket stream reader.r N)�superr( r& �SocketIOr$ �r �sock�mode�bufsize�� __class__r r r( � �
zStreamReader.__init__c s( t � j|i |��}| jt|�7 _|S )zCapture bytes read.)r^ rI r$ r )r r* r+ �valrd r r rI � s zStreamReader.readc C s t | j�| jkS rV )r Z _read_bufZ _read_posr, r r r rX � s zStreamReader.has_data)
r r r! r" r �DEFAULT_BUFFER_SIZEr( rI rX �
__classcell__r r rd r r[ � s
r[ c s4 e Zd ZdZdejf� fdd� Z� fdd�Z� ZS )� StreamWriterzSocket stream writer.�wc r] )z Initialize socket stream writer.r N)r^ r( r& r_ r% r` rd r r r( � rf zStreamWriter.__init__c s0 t � j|g|�R i |��}| jt|�7 _|S )zCapture bytes written.)r^ r r% r )r rg r* r+ �resrd r r r � s zStreamWriter.write) r r r! r" r rh r( r ri r r rd r rj � s rj r\ c C s d|v rt nt}|| ||�S )z(File object attached to a socket object.r\ )r[ rj )ra rb rc �clsr r r �MakeFile� s rn ) r" Z
__future__r r r �typeZ
__metaclass__r& Z_pyior � ImportErrorrY r7 r Z_compatr r r1 r �getattr�objectr# rZ �BufferedReaderr[ rj rh rn r r r r �<module> s, � h