o h@sddlZddlZddlZddlmZddlmZmZddlmZm Z ddl m Z e ej iZe rHddlZeedrCdeej<deej<ndeej<eeZdZeZd ZGd d d ZdS) N)SEEK_END)OptionalUnion)ConnectionError TimeoutError) SSL_AVAILABLESSLWantReadErrorzConnection closed by server.s c @seZdZdejdedefddZdefddZd ed fd e ed e ee fd e e de fddZ d ede fddZd edefddZdefddZdefddZdedd fddZdddZdddZd S) SocketBuffersocketsocket_read_sizesocket_timeoutcCs ||_||_||_t|_dSN)_sockr r ioBytesIO_buffer)selfr r r rG/var/www/html/env/lib/python3.10/site-packages/redis/_parsers/socket.py__init__szSocketBuffer.__init__returncCs,|j}|jdt}|j|||S)z3 Remaining unread length of buffer r)rtellseekr)rposendrrr unread_bytes&s  zSocketBuffer.unread_bytesNTlengthtimeoutraise_on_timeoutc Cs|j}|j}d}|tu}|j}|} |dt|r ||zz? |j|} t | t r8t | dkr8t t || t | } || 7}|durN||krNq"WW|| |r_||jdSdStjy|rntdYW|| |r||jdSdSty} z-t| jd} |s| j| krWYd} ~ W|| |r||jdSdSt d| jd} ~ ww|| |r||jww)NrTzTimeout reading from socketFz!Error while reading from socket: )rr SENTINELrrrr settimeoutrecv isinstancebyteslenrSERVER_CLOSED_CONNECTION_ERRORwriter r rrNONBLOCKING_EXCEPTIONS#NONBLOCKING_EXCEPTION_ERROR_NUMBERSget __class__errnoargs)rrrrsockr markercustom_timeoutbuf current_posdata data_lengthexallowedrrr_read_from_socket/s^         zSocketBuffer._read_from_socketcCst|p |j|ddS)NF)rr)boolrr8)rrrrrcan_read^szSocketBuffer.can_readcCsJ|d}|j|}|t|}|r||||j|7}|ddS)Nr)rreadr&r8)rrr4missingrrrr<cs    zSocketBuffer.readcCsB|j}|}|ts|||7}|tr |ddS)Nr;)rreadlineendswithSYM_CRLFr8)rr2r4rrrr>ns   zSocketBuffer.readlinecCs |jS)z+ Get current read position )rrrrrrget_posxs zSocketBuffer.get_posrcCs|j|dS)zO Rewind the buffer to a specific position, to re-start reading N)rr)rrrrrrewind~szSocketBuffer.rewindcCsX|}|dkr dS|dkr|j}|| d|d|<|j||jddS)zH After a successful read, purge the read part of buffer rN)rr getbuffertruncater)runreadviewrrrpurges  zSocketBuffer.purgecCs2z|jWn tyYnwd|_d|_dSr)rclose ExceptionrrArrrrIs  zSocketBuffer.close)rN)__name__ __module__ __qualname__r intfloatrrr!rrobjectr9r8r:r%r<r>rBrCrHrIrrrrr s8    /   r )r-rr rtypingrr exceptionsrrutilsrBlockingIOError EWOULDBLOCKr*sslhasattrr SSLWantWriteErrorSSLErrortuplekeysr)r'rPr!r@r rrrrs&