HEX

File: //usr/local/CyberCP/lib64/python3.10/site-packages/paramiko/__pycache__/sftp_file.cpython-310.pyc
o

�h<U�@s�dZddlmZddlmZddlZddlZddlZddlm	Z	m
Z
ddlmZddl
mZddlmZmZmZmZmZmZmZmZmZmZmZdd	lmZGd
d�de�ZdS)z
SFTP file object
�)�hexlify)�dequeN)�DEBUG�io_sleep)�BufferedFile)�u)�	CMD_CLOSE�CMD_READ�CMD_DATA�	SFTPError�	CMD_WRITE�
CMD_STATUS�	CMD_FSTAT�	CMD_ATTRS�CMD_FSETSTAT�CMD_EXTENDED�int64)�SFTPAttributesc@seZdZdZdZdAdd�Zdd�Zd	d
�ZdBdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�ZdCd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�ZdDd-d.�ZdEd0d1�ZdFd3d4�ZdGd5d6�Zd7d8�ZdGd9d:�Zd;d<�Zd=d>�Zd?d@�Z d2S)H�SFTPFilez�
    Proxy object for a file on the remote server, in client mode SFTP.

    Instances of this class may be used as context managers in the same way
    that built-in Python file objects are.
    i��r���cCs^t�|�||_||_t�|||�d|_d|_d|_i|_i|_	t
��|_d|_
t�|_dS)NF)r�__init__�sftp�handle�	_set_mode�	pipelined�_prefetching�_prefetch_done�_prefetch_data�_prefetch_extents�	threading�Lock�_prefetch_lock�_saved_exceptionr�_reqs)�selfrr�mode�bufsize�r(�E/usr/local/CyberCP/lib/python3.10/site-packages/paramiko/sftp_file.pyr=s

zSFTPFile.__init__cCs|jdd�dS)NT��async_��_close�r%r(r(r)�__del__KszSFTPFile.__del__cCs|jdd�dS)z!
        Close the file.
        Fr*Nr,r.r(r(r)�closeNszSFTPFile.closeFc	Cs�|jrdS|j�td�tt|j����|jr|j�	|�t
�|�z|r3|j�t
d�t|j�WdS|j�t|j�WdStyGYdSttjfySYdSw)Nz	close({}))�_closedr�_logr�formatrrrr�_finish_responsesrr0�_async_request�typer�_request�EOFError�IOError�socket�error)r%r+r(r(r)r-Ts
�zSFTPFile._closecs��fdd�t|j���D�}t|�dkrdS|jdd�d�|d\}}||�kr,dS||�|kr6d	S|�||�|||�S)
Ncsg|]
}|d�kr|�qS�rr()�.0�x��offsetr(r)�
<listcomp>qsz7SFTPFile._data_in_prefetch_requests.<locals>.<listcomp>rFcSs|dS�Nrr()r>r(r(r)�<lambda>vsz5SFTPFile._data_in_prefetch_requests.<locals>.<lambda>)�keyrT)�listr�values�len�sort�_data_in_prefetch_requests)r%r@�size�k�
buf_offset�buf_sizer(r?r)rIps
��z#SFTPFile._data_in_prefetch_requestscsR�fdd�|j��D�}t|�dkrdSt|�}�|}|t|j|�kr'dS|S)a
        if a block of data is present in the prefetch buffers, at the given
        offset, return the offset of the relevant prefetch buffer.  otherwise,
        return None.  this guarantees nothing about the number of bytes
        collected in the prefetch buffer so far.
        csg|]}|�kr|�qSr(r()r=�ir?r(r)rA�sz6SFTPFile._data_in_prefetch_buffers.<locals>.<listcomp>rN)r�keysrG�max)r%r@rK�indexrLr(r?r)�_data_in_prefetch_buffers�sz"SFTPFile._data_in_prefetch_bufferscCs�	|�|j�}|durn|js|jrn
|j��|��q|dur&d|_dS|j|}|j|=|j|}|dkrG|d|�|j|<||d�}|t	|�kr_||d�|j|j|<|d|�}|S)z�
        read data out of the prefetch buffer, if possible.  if the data isn't
        in the buffer, return None.  otherwise, behaves like a normal read.
        TNFr)
rR�_realposrr1r�_read_response�_check_exceptionrrrG)r%rJr@�prefetchrLr(r(r)�_read_prefetch�s,
�

zSFTPFile._read_prefetchcCsbt||j�}|jr|�|�}|dur|S|j�t|jt|j	�t
|��\}}|tkr-td��|�
�S)N�
Expected data)�min�MAX_REQUEST_SIZErrWrr7r	rrrS�intr
r�
get_string)r%rJ�data�t�msgr(r(r)�_read�s
�zSFTPFile._readc	Cs�tt|�|j�}|j�td�t|jt|j	�|d|��}|j
�|�|jr2t|j
�dkrQ|jj
��rQt|j
�rQ|j
��}|j�|�\}}|tkrLtd��t|j
�s7|S)N�dzExpected status)rYrGrZrr5r6rrrrSr$�appendr�sock�
recv_ready�popleftrTr
r)r%r]�chunk�sftp_async_request�reqr^r_r(r(r)�_write�s$
�


�zSFTPFile._writecC�|jj�|�dS)a@
        Set a timeout on read/write operations on the underlying socket or
        ssh `.Channel`.

        :param float timeout:
            seconds to wait for a pending read/write operation before raising
            ``socket.timeout``, or ``None`` for no timeout

        .. seealso:: `.Channel.settimeout`
        N)rrc�
settimeout)r%�timeoutr(r(r)rk�szSFTPFile.settimeoutcCs|jj��S)z�
        Returns the timeout in seconds (as a `float`) associated with the
        socket or ssh `.Channel` used for this file.

        .. seealso:: `.Channel.gettimeout`
        )rrc�
gettimeoutr.r(r(r)rm�szSFTPFile.gettimeoutcCrj)z�
        Set blocking or non-blocking mode on the underiying socket or ssh
        `.Channel`.

        :param int blocking:
            0 to set non-blocking mode; non-0 to set blocking mode.

        .. seealso:: `.Channel.setblocking`
        N)rrc�setblocking)r%�blockingr(r(r)rn�s
zSFTPFile.setblockingcCsdS)z�
        Check if the file supports random access.

        :return:
            `True` if the file supports random access. If `False`,
            :meth:`seek` will raise an exception
        Tr(r.r(r(r)�seekable�szSFTPFile.seekablercCsb|��||jkr||_|_n||jkr!|j|7_|j|_n
|��||_|_t�|_dS)zX
        Set the file's current position.

        See `file.seek` for details.
        N)�flush�SEEK_SETrS�_pos�SEEK_CUR�	_get_size�bytes�_rbuffer)r%r@�whencer(r(r)�seeks


z
SFTPFile.seekcCs.|j�t|j�\}}|tkrtd��t�|�S)a
        Retrieve information about this file from the remote system.  This is
        exactly like `.SFTPClient.stat`, except that it operates on an
        already-open file.

        :returns:
            an `.SFTPAttributes` object containing attributes about this file.
        zExpected attributes)rr7rrrrr�	_from_msg)r%r^r_r(r(r)�stats	
z
SFTPFile.statcC�>|j�td�t|j�|��t�}||_|j�t	|j|�dS)z�
        Change the mode (permissions) of this file.  The permissions are
        unix-style and identical to those used by Python's `os.chmod`
        function.

        :param int mode: new permissions
        zchmod({}, {!r})N)
rr2rr3rrr�st_moder7r)r%r&�attrr(r(r)�chmod ��zSFTPFile.chmodcCsH|j�td�t|j�||��t�}|||_|_|j�	t
|j|�dS)aV
        Change the owner (``uid``) and group (``gid``) of this file.  As with
        Python's `os.chown` function, you must pass both arguments, so if you
        only want to change one, use `stat` first to retrieve the current
        owner and group.

        :param int uid: new owner's uid
        :param int gid: new group id
        zchown({}, {!r}, {!r})N)rr2rr3rrr�st_uid�st_gidr7r)r%�uid�gidr~r(r(r)�chown/s
�zSFTPFile.chowncCs\|durt��t��f}|j�td�t|j�|��t�}|\|_|_	|j�
t|j|�dS)ad
        Set the access and modified times of this file.  If
        ``times`` is ``None``, then the file's access and modified times are
        set to the current time.  Otherwise, ``times`` must be a 2-tuple of
        numbers, of the form ``(atime, mtime)``, which is used to set the
        access and modified times, respectively.  This bizarre API is mimicked
        from Python for the sake of consistency -- I apologize.

        :param tuple times:
            ``None`` or a tuple of (access time, modified time) in standard
            internet epoch time (seconds since 01 January 1970 GMT)
        Nzutime({}, {!r}))�timerr2rr3rrr�st_atime�st_mtimer7r)r%�timesr~r(r(r)�utimeAs
�zSFTPFile.utimecCr|)z�
        Change the size of this file.  This usually extends
        or shrinks the size of the file, just like the ``truncate()`` method on
        Python file objects.

        :param size: the new size of the file
        ztruncate({}, {!r})N)
rr2rr3rrr�st_sizer7r)r%rJr~r(r(r)�truncateWr�zSFTPFile.truncatec	CsB|j�td|j|t|�t|�|�\}}|��|��|��}|S)a
        Ask the server for a hash of a section of this file.  This can be used
        to verify a successful upload or download, or for various rsync-like
        operations.

        The file is hashed from ``offset``, for ``length`` bytes.
        If ``length`` is 0, the remainder of the file is hashed.  Thus, if both
        ``offset`` and ``length`` are zero, the entire file is hashed.

        Normally, ``block_size`` will be 0 (the default), and this method will
        return a byte string representing the requested hash (for example, a
        string of length 16 for MD5, or 20 for SHA-1).  If a non-zero
        ``block_size`` is given, each chunk of the file (from ``offset`` to
        ``offset + length``) of ``block_size`` bytes is computed as a separate
        hash.  The hash results are all concatenated and returned as a single
        string.

        For example, ``check('sha1', 0, 1024, 512)`` will return a string of
        length 40.  The first 20 bytes will be the SHA-1 of the first 512 bytes
        of the file, and the last 20 bytes will be the SHA-1 of the next 512
        bytes.

        :param str hash_algorithm:
            the name of the hash algorithm to use (normally ``"sha1"`` or
            ``"md5"``)
        :param offset:
            offset into the file to begin hashing (0 means to start from the
            beginning)
        :param length:
            number of bytes to hash (0 means continue to the end of the file)
        :param int block_size:
            number of bytes to hash per result (must not be less than 256; 0
            means to compute only one hash of the entire segment)
        :return:
            `str` of bytes representing the hash of each block, concatenated
            together

        :raises:
            ``IOError`` -- if the server doesn't support the "check-file"
            extension, or possibly doesn't support the hash algorithm requested

        .. note:: Many (most?) servers don't support this extension yet.

        .. versionadded:: 1.4
        z
check-file)rr7rrr�get_text�
get_remainder)r%�hash_algorithmr@�length�
block_sizer^r_r]r(r(r)�checkfs.�	zSFTPFile.checkTcCs
||_dS)a�
        Turn on/off the pipelining of write operations to this file.  When
        pipelining is on, paramiko won't wait for the server response after
        each write operation.  Instead, they're collected as they come in. At
        the first non-write operation (including `.close`), all remaining
        server responses are collected.  This means that if there was an error
        with one of your later writes, an exception might be thrown from within
        `.close` instead of `.write`.

        By default, files are not pipelined.

        :param bool pipelined:
            ``True`` if pipelining should be turned on for this file; ``False``
            otherwise

        .. versionadded:: 1.5
        N)r)r%rr(r(r)�
set_pipelined�s
zSFTPFile.set_pipelinedNcCsr|dur	|��j}g}|j}||kr)t|j||�}|�||f�||7}||kst|�dkr7|�||�dSdS)aE
        Pre-fetch the remaining contents of this file in anticipation of future
        `.read` calls.  If reading the entire file, pre-fetching can
        dramatically improve the download speed by avoiding roundtrip latency.
        The file's contents are incrementally buffered in a background thread.

        The prefetched data is stored in a buffer until read via the `.read`
        method.  Once data has been read, it's removed from the buffer.  The
        data may be read in a random order (using `.seek`); chunks of the
        buffer that haven't been read will continue to be buffered.

        :param int file_size:
            When this is ``None`` (the default), this method calls `stat` to
            determine the remote file size. In some situations, doing so can
            cause exceptions or hangs (see `#562
            <https://github.com/paramiko/paramiko/pull/562>`_); as a
            workaround, one may call `stat` explicitly and pass its value in
            via this parameter.
        :param int max_concurrent_requests:
            The maximum number of concurrent read requests to prefetch. See
            `.SFTPClient.get` (its ``max_concurrent_prefetch_requests`` param)
            for details.

        .. versionadded:: 1.5.1
        .. versionchanged:: 1.16.0
            The ``file_size`` parameter was added (with no default value).
        .. versionchanged:: 1.16.1
            The ``file_size`` parameter was made optional for backwards
            compatibility.
        .. versionchanged:: 3.3
            Added ``max_concurrent_requests``.
        Nr)r{r�rSrYrZrbrG�_start_prefetch)r%�	file_size�max_concurrent_requests�chunks�nrfr(r(r)rV�s!
��zSFTPFile.prefetchccs��|j�td�t|j�|��g}|D]-\}}|�|�s"|�||�r#q|dkr@t||j	�}|�
||f�||7}||8}|dks'q|�||�|D]}|�|d�|�
|d�VqIdS)a/
        Read a set of blocks from the file by (offset, length).  This is more
        efficient than doing a series of `.seek` and `.read` calls, since the
        prefetch machinery is used to retrieve all the requested blocks at
        once.

        :param chunks:
            a list of ``(offset, length)`` tuples indicating which sections of
            the file to read
        :param int max_concurrent_prefetch_requests:
            The maximum number of concurrent read requests to prefetch. See
            `.SFTPClient.get` (its ``max_concurrent_prefetch_requests`` param)
            for details.
        :return: a list of blocks read, in the same order as in ``chunks``

        .. versionadded:: 1.5.4
        .. versionchanged:: 3.3
            Added ``max_concurrent_prefetch_requests``.
        zreadv({}, {!r})r�N)rr2rr3rrrRrIrYrZrbr�ry�read)r%r�� max_concurrent_prefetch_requests�read_chunksr@rJ�
chunk_sizer>r(r(r)�readv�s0���
����zSFTPFile.readvcCsz|��jWSYdSrB)r{r�r.r(r(r)ruszSFTPFile._get_sizecCs2d|_d|_tj|j||fd�}d|_|��dS)NTF)�target�args)rrr �Thread�_prefetch_thread�daemon�start)r%r�r�r^r(r(r)r�s�zSFTPFile._start_prefetchc	Cs�|D]\\}}|dur5	|j�t|j�}||kr 	Wd�nWd�n1s*wYt�t�q|j�|t|j	t
|�t|��}|j�||f|j|<Wd�n1sYwYqdS�N)r"rGrr��sleeprrr5r	rrr[)r%r�r�r@r��pf_len�numr(r(r)r�$s(
��
�����zSFTPFile._prefetch_threadc
Cs�|tkr$z	|j�|�WdSty#}z
||_WYd}~dSd}~ww|tkr,td��|��}	|j�0||j	vr]|j	|\}}||j
|<|j	|=t|j	�dkrTd|_	Wd�dSWd�n1sgwYq1)NrXTr)
r
r�_convert_status�	Exceptionr#r
rr\r"rrrGr)r%r^r_r��er]r@r�r(r(r)�_async_response7s2���

���zSFTPFile._async_responsecCs|jdur
|j}d|_|�dS)z.if there's a saved exception, raise & clear itN)r#)r%r>r(r(r)rUMs

�zSFTPFile._check_exception)rr)Fr<)rrr)T)NNr�)!�__name__�
__module__�__qualname__�__doc__rZrr/r0r-rIrRrWr`rirkrmrnrpryr{rr�r�r�r�r�rVr�rur�r�r�rUr(r(r(r)r1s>	



	



<

./
r)r��binasciir�collectionsrr:r r��paramiko.commonrr�
paramiko.filer�
paramiko.utilr�
paramiko.sftprr	r
rrr
rrrrr�paramiko.sftp_attrrrr(r(r(r)�<module>s4