File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/ext/tornado/__pycache__/iostream.cpython-310.pyc
Back
o ;j� � @ s d Z ddlmZmZmZ ddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZ ddlmZmZ ddlmZmZmZmZmZ ddlmZ dd lmZ zdd l m Z W n e yi dZ Y nw zddl Z W n e y{ dZ Y nw ej ej!fZ"e#ed �r�e"ej$f7 Z"ej%ej&ej'ej(fZ)e#ed �r�e)ej*ej+ej,f7 Z)e j-d kr�e)ej.f7 Z)ej/fZ0e#ed�r�e0ej1f7 Z0e j-�2d�Z3G dd� de4�Z5G dd� de6�Z7G dd� de6�Z8G dd� de9�Z:G dd� de:�Z;G dd� de;�Z<G d d � d e:�Z=d d � Z>dS ) aB Utility classes to write to and read from non-blocking files and sockets. Contents: * `BaseIOStream`: Generic interface for reading and writing. * `IOStream`: Implementation of BaseIOStream using non-blocking sockets. * `SSLIOStream`: SSL-aware version of IOStream. * `PipeIOStream`: Pipe-based IOStream implementation. � )�absolute_import�division�print_functionN)�TracebackFuture)�ioloop)�gen_log�app_log)�ssl_wrap_socket�ssl_match_hostname�SSLCertificateError�_client_ssl_defaults�_server_ssl_defaults)� stack_context)�errno_from_exception)�_set_nonblocking�WSAEWOULDBLOCK� WSAECONNRESET�darwin�WSAEINPROGRESS�winc s" e Zd ZdZd� fdd� Z� ZS )�StreamClosedErrora� Exception raised by `IOStream` methods when the stream is closed. Note that the close callback is scheduled to run *after* other callbacks on the stream (to allow for buffered data to be processed), so you may see this error before you see the close callback. The ``real_error`` attribute contains the underlying error that caused the stream to close (if any). .. versionchanged:: 4.3 Added the ``real_error`` attribute. Nc s t t| ��d� || _d S )NzStream is closed)�superr �__init__� real_error)�selfr �� __class__� �M/opt/saltstack/salt/lib/python3.10/site-packages/salt/ext/tornado/iostream.pyr f s zStreamClosedError.__init__�N)�__name__� __module__� __qualname__�__doc__r � __classcell__r r r r r Y s r c @ � e Zd ZdZdS )�UnsatisfiableReadErrorz�Exception raised when a read cannot be satisfied. Raised by ``read_until`` and ``read_until_regex`` with a ``max_bytes`` argument. N�r r! r" r# r r r r r& k s r& c @ r% )�StreamBufferFullErrorzDException raised by `IOStream` methods when the buffer is full. Nr' r r r r r( t s r( c @ s^ e Zd ZdZ dRdd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dSdd�Z dSdd�Z dTdd�Z dSdd�Z dUdd�Zdd� ZdVd d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dJdK� Z'dLdM� Z(dNdO� Z)dPdQ� Z*dS )W� BaseIOStreama] A utility class to write to and read from a non-blocking file or socket. We support a non-blocking ``write()`` and a family of ``read_*()`` methods. All of the methods take an optional ``callback`` argument and return a `.Future` only if no callback is given. When the operation completes, the callback will be run or the `.Future` will resolve with the data read (or ``None`` for ``write()``). All outstanding ``Futures`` will resolve with a `StreamClosedError` when the stream is closed; users of the callback interface will be notified via `.BaseIOStream.set_close_callback` instead. When a stream is closed due to an error, the IOStream's ``error`` attribute contains the exception object. Subclasses must implement `fileno`, `close_fd`, `write_to_fd`, `read_from_fd`, and optionally `get_fd_error`. Nc C s� |pt j�� | _|p d| _t|pd| jd �| _|| _d| _t � | _ d| _ d| _ t � | _ d| _d| _d| _d| _d| _g | _d| _d| _d| _d| _d| _d| _d| _d| _d| _ d| _ t � � | _ d| _!d| _"d| _#d| _$d| _%d| _&d| _'d| _(dS )a� `BaseIOStream` constructor. :arg io_loop: The `.IOLoop` to use; defaults to `.IOLoop.current`. Deprecated since Tornado 4.1. :arg max_buffer_size: Maximum amount of incoming data to buffer; defaults to 100MB. :arg read_chunk_size: Amount of data to read at one time from the underlying transport; defaults to 64KB. :arg max_write_buffer_size: Amount of outgoing data to buffer; defaults to unlimited. .. versionchanged:: 4.0 Add the ``max_write_buffer_size`` parameter. Changed default ``read_chunk_size`` to 64KB. i @i � Nr F))r �IOLoop�current�io_loop�max_buffer_size�min�read_chunk_size�max_write_buffer_size�error� bytearray� _read_buffer�_read_buffer_pos�_read_buffer_size� _write_buffer�_write_buffer_pos�_write_buffer_size�_write_buffer_frozen�_total_write_index�_total_write_done_index� _pending_writes_while_frozen�_read_delimiter� _read_regex�_read_max_bytes� _read_bytes� _read_partial�_read_until_close�_read_callback� _read_future�_streaming_callback�_write_callback� collections�deque�_write_futures�_close_callback�_connect_callback�_connect_future�_ssl_connect_future� _connecting�_state�_pending_callbacks�_closed)r r- r. r0 r1 r r r r � sH � zBaseIOStream.__init__c C � t � �)z,Returns the file descriptor for this stream.��NotImplementedError�r r r r �fileno� � zBaseIOStream.filenoc C rS )z�Closes the file underlying this stream. ``close_fd`` is called by `BaseIOStream` and should not be called elsewhere; other users should call `close` instead. rT rV r r r �close_fd� s zBaseIOStream.close_fdc C rS )ziAttempts to write ``data`` to the underlying file. Returns the number of bytes written. rT �r �datar r r � write_to_fd� s zBaseIOStream.write_to_fdc C rS )a- Attempts to read from the underlying file. Returns ``None`` if there was nothing to read (the socket returned `~errno.EWOULDBLOCK` or equivalent), otherwise returns the data. When possible, should return no more than ``self.read_chunk_size`` bytes at a time. rT rV r r r � read_from_fd� s zBaseIOStream.read_from_fdc C � dS )a> Returns information about any error on the underlying file. This method is called after the `.IOLoop` has signaled an error on the file descriptor, and should return an Exception (such as `socket.error` with additional information, or None if no such information is available. Nr rV r r r � get_fd_error� s zBaseIOStream.get_fd_errorc C s� | � |�}t�|�| _|| _z| �� W |S ty6 } zt�d| � | j dd� |W Y d}~S d}~w |durD|� dd� � � )a� Asynchronously read until we have matched the given regex. The result includes the data that matches the regex and anything that came before it. If a callback is given, it will be run with the data as an argument; if not, this method returns a `.Future`. If ``max_bytes`` is not None, the connection will be closed if more than ``max_bytes`` bytes have been read and the regex is not satisfied. .. versionchanged:: 4.0 Added the ``max_bytes`` argument. The ``callback`` argument is now optional and a `.Future` will be returned if it is omitted. �*Unsatisfiable read, closing connection: %sT��exc_infoNc S � | � � S r �� exception��fr r r �<lambda> � z/BaseIOStream.read_until_regex.<locals>.<lambda>) �_set_read_callback�re�compiler? r@ �_try_inline_readr& r �info�close�add_done_callback)r Zregex�callback� max_bytes�future�er r r �read_until_regex� s � �z BaseIOStream.read_until_regexc C s� | � |�}|| _|| _z| �� W |S ty3 } zt�d| � | jdd� |W Y d}~S d}~w |durA|�dd� � � )ai Asynchronously read until we have found the given delimiter. The result includes all the data read including the delimiter. If a callback is given, it will be run with the data as an argument; if not, this method returns a `.Future`. If ``max_bytes`` is not None, the connection will be closed if more than ``max_bytes`` bytes have been read and the delimiter is not found. .. versionchanged:: 4.0 Added the ``max_bytes`` argument. The ``callback`` argument is now optional and a `.Future` will be returned if it is omitted. r` Tra Nc S rc r rd rf r r r rh * ri z)BaseIOStream.read_until.<locals>.<lambda>) rj r> r@ rm r& r rn ro rp )r � delimiterrq rr rs rt r r r � read_until s � �zBaseIOStream.read_untilFc C s` | � |�}t|tj�s J �|| _|| _t�|�| _z| � � W |S |dur/|� dd� � � )a� Asynchronously read a number of bytes. If a ``streaming_callback`` is given, it will be called with chunks of data as they become available, and the final result will be empty. Otherwise, the result is all the data that was read. If a callback is given, it will be run with the data as an argument; if not, this method returns a `.Future`. If ``partial`` is true, the callback is run as soon as we have any bytes to return (but never more than ``num_bytes``) .. versionchanged:: 4.0 Added the ``partial`` argument. The callback argument is now optional and a `.Future` will be returned if it is omitted. Nc S rc r rd rf r r r rh H ri z)BaseIOStream.read_bytes.<locals>.<lambda>) rj � isinstance�numbersZIntegralrA rB r �wraprF rm rp )r � num_bytesrq �streaming_callback�partialrs r r r � read_bytes. s �zBaseIOStream.read_bytesc C s| | � |�}t�|�| _| �� r$| jdur| �| jd� | �| jd� |S d| _z| �� W |S |dur=|� dd� � � )a{ Asynchronously reads all data from the socket until it is closed. If a ``streaming_callback`` is given, it will be called with chunks of data as they become available, and the final result will be empty. Otherwise, the result is all the data that was read. If a callback is given, it will be run with the data as an argument; if not, this method returns a `.Future`. Note that if a ``streaming_callback`` is used, data will be read from the socket as quickly as it becomes available; there is no way to apply backpressure or cancel the reads. If flow control or cancellation are desired, use a loop with `read_bytes(partial=True) <.read_bytes>` instead. .. versionchanged:: 4.0 The callback argument is now optional and a `.Future` will be returned if it is omitted. NTFc S rc r rd rf r r r rh l ri z/BaseIOStream.read_until_close.<locals>.<lambda>) rj r rz rF �closed�_run_read_callbackr6 rC rm rp )r rq r| rs r r r �read_until_closeL s �z BaseIOStream.read_until_closec C s� | � � |r<| jdur| jt|� | jkrtd��| jr#| j�|� n| j|7 _| jt|�7 _| j t|�7 _ |durIt � |�| _ d}nt � }|�dd� � | j�| j |f� | jsq| �� | jrm| �| jj� | �� |S )a� Asynchronously write the given data to this stream. If ``callback`` is given, we call it when all of the buffered write data has been successfully written to the stream. If there was previously buffered write data and an old write callback, that callback is simply overwritten with this new callback. If no ``callback`` is given, this method returns a `.Future` that resolves (with a result of ``None``) when the write has been completed. The ``data`` argument may be of type `bytes` or `memoryview`. .. versionchanged:: 4.0 Now returns a `.Future` if no callback is given. .. versionchanged:: 4.5 Added support for `memoryview` arguments. Nz!Reached maximum write buffer sizec S rc r rd rf r r r rh � ri z$BaseIOStream.write.<locals>.<lambda>)� _check_closedr1 r9 �lenr( r: r= �appendr7 r; r rz rG r rp rJ rO � _handle_write� _add_io_stater- �WRITE�_maybe_add_error_listener)r r[ rq rs r r r �writep s, zBaseIOStream.writec C s t �|�| _| �� dS )z�Call the given callback when the stream is closed. This is not necessary for applications that use the `.Future` interface; all outstanding ``Futures`` will resolve with a `StreamClosedError` when the stream is closed. N)r rz rK r� �r rq r r r �set_close_callback� s z BaseIOStream.set_close_callbackc C s� | � � sK|rt|t�st�� }t|�r|d | _| jr4| jdur*| j r*| � | j d� d| _| � | j d� | j durD| j � | �� � d| _ | �� d| _| �� dS )z�Close this stream. If ``exc_info`` is true, set the ``error`` attribute to the current exception from `sys.exc_info` (or if ``exc_info`` is a tuple, use that instead of `sys.exc_info`). � NTF)r rx �tuple�sysrb �anyr2 rC rF r6 r� rP r- �remove_handlerrW rY rR �_maybe_run_close_callback)r rb r r r ro � s& � zBaseIOStream.closec C s� | � � rp| jdkrrg }| jd ur|�| j� d | _|dd� | jD �7 }| j�� | jd ur6|�| j� d | _| jd urD|�| j� d | _|D ] }|�t | j d�� qF| j d urb| j }d | _ | � |� d | _ | _d | _d| _d S d S d S )Nr c S s g | ]\}}|�qS r r )�.0�_rs r r r � <listcomp>� s z:BaseIOStream._maybe_run_close_callback.<locals>.<listcomp>�r )r rQ rE r� rJ �clearrM rN � set_exceptionr r2 rK � _run_callbackrD rG r7 r9 )r Zfuturesrs �cbr r r r� � s. �z&BaseIOStream._maybe_run_close_callbackc C s | j dup | jduS )z9Returns true if we are currently reading from the stream.N)rD rE rV r r r �reading� s zBaseIOStream.readingc C s | j dkS )z7Returns true if we are currently writing to the stream.r )r9 rV r r r �writing� s zBaseIOStream.writingc C � | j S )z+Returns true if the stream has been closed.)rR rV r r r r � rX zBaseIOStream.closedc C r^ )a� Sets the no-delay flag for this stream. By default, data written to TCP streams may be held for a time to make the most efficient use of bandwidth (according to Nagle's algorithm). The no-delay flag requests that data be written as soon as possible, even if doing so would consume additional bandwidth. This flag is currently defined only for TCP-based ``IOStreams``. .. versionadded:: 3.1 Nr )r �valuer r r � set_nodelay� s zBaseIOStream.set_nodelayc C s� | � � r t�d|� d S z�| jr| �� | � � rW d S || jj@ r%| �� | � � r,W d S || jj@ r6| � � | � � r=W d S || jj @ rR| � � | _ | j� | j� W d S | jj }| �� r`|| jjO }| �� rj|| jjO }|| jj kr{| jdkr{|| jjO }|| jkr�| jd us�J d��|| _| j�| �� | j� W d S W d S ty� } zt�d| � | jdd� W Y d }~d S d }~w ty� tj ddd� | jdd� � w )Nz Got events for closed stream %sr z4shouldn't happen: _handle_events without self._stater` Tra �'Uncaught exception, closing connection.)r r �warningrO �_handle_connectr- �READ� _handle_readr� r� �ERRORr_ r2 � add_callbackro r� r� r6 rP �update_handlerrW r& rn � Exception)r �fd�events�statert r r r �_handle_events� sZ ��� � �zBaseIOStream._handle_eventsc sX � ��fdd�}t �� � � jd7 _�j�|� W d � d S 1 s%w Y d S )Nc sZ � j d8 _ z z �� � W W ��� S ty' tjddd� �jdd� � w ��� w )Nr� r� Tra )rQ r� r� r r2 ro r ��argsrq r r r �wrapper0 s �� � z+BaseIOStream._run_callback.<locals>.wrapperr� )r � NullContextrQ r- r� )r rq r� r� r r� r r� / s "�zBaseIOStream._run_callbackc C s� zi| j d ur | j }n| jd ur| j}n | �� rd }nd}d}| jd7 _| �� s^| �� dkr0n.| �� |d ur>| j|kr>n | j|krZ| �� }|d urU|W | jd8 _S | jd }| �� r)| �� W | jd8 _S | jd8 _w )Nr r� r* ) rA r@ r� rQ r �_read_to_buffer�_run_streaming_callbackr6 �_find_read_pos)r Z target_bytesZ next_find_pos�posr r r �_read_to_buffer_loopR s4 �� z!BaseIOStream._read_to_buffer_loopc C s~ z| � � }W n' ty � ty- } zt�d| � | jdd� W Y d }~d S d }~ww |d ur9| �|� d S | �� d S )Nzerror on read: %sTra )r� r& r� r r� ro �_read_from_bufferr� )r r� rt r r r r� � s �� zBaseIOStream._handle_readc C sL | j d u s J d��| jd u sJ d��|d ur t�|�| _ | jS t� | _| jS )NzAlready reading)rD rE r rz r r� r r r rj � s �z BaseIOStream._set_read_callbackc C s� |r| j }n"| j}d | _| _ | jd ur(|d u sJ �| j}d | _|�| �|�� |d ur@| jd u s5|s5J �| �|| �|�� d S | �� d S r )rF rD rE � set_result�_consumer� r� )r �sizeZ streamingrq rs r r r r� � s z BaseIOStream._run_read_callbackc C s� | � � | �� }|dur| �|� dS | �� z| �� }W n ty) | �� � w |dur5| �|� dS | �� r?| �� dS | �t j j � dS )a Attempt to complete the current read operation from buffered data. If the read can be completed without blocking, schedules the read callback on the next IOLoop iteration; otherwise starts listening for reads on the socket. N) r� r� r� r� r� r� r� r r� r r+ r� �r r� r r r rm � s$ � z BaseIOStream._try_inline_readc C s� z| � � }W n6 tjttfy= } z&t|�tjkr W Y d}~q | �|�r2| j dd� W Y d}~dS | j dd� � d}~ww |du rEdS | j |7 _ | j t |�7 _ | j | j krht�d� | � � td��t |�S )a Reads from the socket and appends the result to the read buffer. Returns the number of bytes read. Returns 0 if there is nothing to read (i.e. the read returns EWOULDBLOCK or equivalent). On error closes the socket and raises an exception. TNra r z Reached maximum read buffer size)r] �socketr2 �IOError�OSErrorr �errnoZEINTR� _is_connresetro r4 r6 r� r. r r( �r �chunkrt r r r r� � s. �� z BaseIOStream._read_to_bufferc C sR | j d ur%| jr'| j}| jd ur t| j|�}| j|8 _| �|d� d S d S d S )NT)rF r6 rA r/ r� )r Zbytes_to_consumer r r r� � s �z$BaseIOStream._run_streaming_callbackc C s( d | _ | _| _d| _| �|d� dS )z�Attempts to complete the currently-pending read from the buffer. The argument is either a position in the read buffer or None, as returned by _find_read_pos. NF)rA r> r? rB r� r� r r r r� s z BaseIOStream._read_from_bufferc C s | j dur | j| j ks| jr | jdkr t| j | j�}|S | jdurR| jrP| j�| j| j�}|dkrH|| j8 }t| j�}| � | j|| � || S | � | j| j� dS | j dur| jr| j � | j| j�}|durw|� � | j }| � | j |� |S | � | j | j� dS )z�Attempts to find a position in the read buffer that satisfies the currently-pending read. Returns a position in the buffer if the current read can be satisfied, or None if it cannot. Nr ���) rA r6 rB r/ r> r4 �findr5 r� �_check_max_bytesr? �search�end)r r{ �locZ delimiter_len�mr r r r� sD � � �� � �zBaseIOStream._find_read_posc C s. | j d ur|| j krtd|| j f ��d S d S )Nz&delimiter %r not found within %d bytes)r@ r& )r rv r� r r r r� 8 s ���z BaseIOStream._check_max_bytesc C s || _ d S r )r: �r r� r r r �_freeze_write_buffer? s z!BaseIOStream._freeze_write_bufferc C sH d| _ | jd�| j�7 _| jttt| j��7 _g | jd d �< d S )NF� )r: r7 �joinr= r9 �sum�mapr� rV r r r �_unfreeze_write_bufferB s z#BaseIOStream._unfreeze_write_bufferc C r^ )zv Called when a non-blocking write() failed writing anything. Can be overridden in subclasses. Nr r� r r r �_got_empty_writeH � z BaseIOStream._got_empty_writec C s� | j r�| j dks J �zY| j}| jr| j}ntrd}n| j }| �t| j�||| � �}|dkr6| �|� W nn| j|7 _| j |8 _ | j| j krT| jd | j�= d| _| jr[| �� | j |7 _ W n= t j t t fy� } z-|jd tv r�| �|� W Y d }~n#| �|�s�t�d| �� |� | jdd� W Y d }~d S d }~ww | j s| jr�| jd \}}|| j kr�n | j�� |�d � | js�| j s�| jr�| j}d | _| �|� d S d S d S )Nr i zWrite error on %s: %sTra )r9 r8 r: �_WINDOWSr\ � memoryviewr7 r� r� r< r� r2 r� r� r� �_ERRNO_WOULDBLOCKr� r r� rW ro rJ �popleftr� rG r� )r �startr� r{ rt �indexrs rq r r r r� N sd � � ���- ��zBaseIOStream._handle_writec C sx |dkrdS || j ks J �t| j�| j| j| � �� }| j|7 _| j |8 _ | j| j kr:| jd | j�= d| _|S )Nr r� )r6 r� r4 r5 �tobytes)r r� �br r r r� � s �� zBaseIOStream._consumec C s | � � r t| jd��d S )Nr� )r r r2 rV r r r r� � s �zBaseIOStream._check_closedc C sl | j dkrd S | jd u s| jtjjkr0| �� r | �� d S | jdkr2| jd ur4| � tjj � d S d S d S d S �Nr ) rQ rP r r+ r� r r� r6 rK r� r� rV r r r r� � s �z&BaseIOStream._maybe_add_error_listenerc C s� | � � rdS | jdu r5tjj|B | _t�� � | j�| � � | j | j� W d � dS 1 s.w Y dS | j|@ sL| j|B | _| j� | � � | j� dS dS )aJ Adds `state` (IOLoop.{READ,WRITE} flags) to our event handler. Implementation notes: Reads and writes have a fast path and a slow path. The fast path reads synchronously from socket buffers, while the slow path uses `_add_io_state` to schedule an IOLoop callback. Note that in both cases, the callback is run asynchronously with `_run_callback`. To detect closed connections, we must have called `_add_io_state` at some point, but we want to delay this as much as possible so we don't have to set an `IOLoop.ERROR` listener that will be overwritten by the next slow-path operation. As long as there are callbacks scheduled for fast-path ops, those callbacks may do more reads. If a sequence of fast-path ops do not end in a slow-path op, (e.g. for an @asynchronous long-poll request), we must add the error handler. This is done in `_run_callback` and `write` (since the write callback is optional so we can have a fast-path write with no `_run_callback`) N) r rP r r+ r� r r� r- Z add_handlerrW r� r� )r r� r r r r� � s �"� �zBaseIOStream._add_io_statec C s t |tjtf�o t|�tv S )zbReturn true if exc is ECONNRESET or equivalent. May be overridden in subclasses. )rx r� r2 r� r �_ERRNO_CONNRESET)r �excr r r r� � s �zBaseIOStream._is_connreset)NNNN�NN)NNFr )F)+r r! r" r# r rW rY r\ r] r_ ru rw r~ r� r� r� ro r� r� r� r r� r� r� r� r� rj r� rm r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r) y sZ �; " � $- 5#< !! +; !r) c sl e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z ddd�Z ddd�Z dd� Z dd� Z � ZS )�IOStreama* Socket-based `IOStream` implementation. This class supports the read and write methods from `BaseIOStream` plus a `connect` method. The ``socket`` parameter may either be connected or unconnected. For server operations the socket is the result of calling `socket.accept <socket.socket.accept>`. For client operations the socket is created with `socket.socket`, and may either be connected before passing it to the `IOStream` or connected with `IOStream.connect`. A very simple (and broken) HTTP client using this class: .. testcode:: import tornado.ioloop import tornado.iostream import socket def send_request(): stream.write(b"GET / HTTP/1.0\r\nHost: friendfeed.com\r\n\r\n") stream.read_until(b"\r\n\r\n", on_headers) def on_headers(data): headers = {} for line in data.split(b"\r\n"): parts = line.split(b":") if len(parts) == 2: headers[parts[0].strip()] = parts[1].strip() stream.read_bytes(int(headers[b"Content-Length"]), on_body) def on_body(data): print(data) stream.close() tornado.ioloop.IOLoop.current().stop() if __name__ == '__main__': s = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) stream = tornado.iostream.IOStream(s) stream.connect(("friendfeed.com", 80), send_request) tornado.ioloop.IOLoop.current().start() .. testoutput:: :hide: c s, || _ | j �d� tt| �j|i |�� d S )NF)r� � setblockingr r� r )r r� r� �kwargsr r r r s zIOStream.__init__c C r� r )r� rV r r r rW � zIOStream.filenoc C s | j �� d | _ d S r )r� ro rV r r r rY s zIOStream.close_fdc C s$ | j �t jt j�}t �|t�|��S r )r� � getsockopt� SOL_SOCKET�SO_ERRORr2 �os�strerror)r r� r r r r_ s �zIOStream.get_fd_errorc C s` z | j �| j�}W n t jy% } z|jd tv r W Y d }~d S � d }~ww |s.| �� d S |S r� )r� �recvr0 r2 r� r� ro r� r r r r] s ��zIOStream.read_from_fdc C s z| j �|�W ~S ~w r )r� �sendrZ r r r r\ ( s zIOStream.write_to_fdNc C s� d| _ |durt�|�| _d}nt� }| _z| j�|� W n; tjyY } z.t |�t vrOt |�t vrO|du rAt � d| j�� |� | jdd� |W Y d}~S W Y d}~nd}~ww | �| jj� |S )a0 Connects the socket to a remote address without blocking. May only be called if the socket passed to the constructor was not previously connected. The address parameter is in the same format as for `socket.connect <socket.socket.connect>` for the type of socket passed to the IOStream constructor, e.g. an ``(ip, port)`` tuple. Hostnames are accepted here, but will be resolved synchronously and block the IOLoop. If you have a hostname instead of an IP address, the `.TCPClient` class is recommended instead of calling this method directly. `.TCPClient` will do asynchronous DNS resolution and handle both IPv4 and IPv6. If ``callback`` is specified, it will be called with no arguments when the connection is completed; if not this method returns a `.Future` (whose result after a successful connection will be the stream itself). In SSL mode, the ``server_hostname`` parameter will be used for certificate validation (unless disabled in the ``ssl_options``) and SNI (if supported; requires Python 2.7.9+). Note that it is safe to call `IOStream.write <BaseIOStream.write>` while the connection is pending, in which case the data will be written as soon as the connection is ready. Calling `IOStream` read methods before the socket is connected works on some platforms but is non-portable. .. versionchanged:: 4.0 If no callback is given, returns a `.Future`. .. versionchanged:: 4.2 SSL certificates are validated by default; pass ``ssl_options=dict(cert_reqs=ssl.CERT_NONE)`` or a suitably-configured `ssl.SSLContext` to the `SSLIOStream` constructor to disable. TN�Connect error on fd %s: %sra )rO r rz rL r rM r� �connectr2 r �_ERRNO_INPROGRESSr� r r� rW ro r� r- r� )r �addressrq �server_hostnamers rt r r r r� 0 s( ' � ��zIOStream.connectc s� | j s | js | js | js | js | js | js | js | js | j r"t d��|du r-|r+t }nt }| j }| j�|� d| _ t||||dd�}| j�d| _t� � t||| jd��� ��fdd�}��|� � �fdd ��_| j�_| j�_� S ) a� Convert this `IOStream` to an `SSLIOStream`. This enables protocols that begin in clear-text mode and switch to SSL after some initial negotiation (such as the ``STARTTLS`` extension to SMTP and IMAP). This method cannot be used if there are outstanding reads or writes on the stream, or if there is any data in the IOStream's buffer (data in the operating system's socket buffer is allowed). This means it must generally be used immediately after reading or writing the last clear-text data. It can also be used immediately after connecting, before any reads or writes. The ``ssl_options`` argument may be either an `ssl.SSLContext` object or a dictionary of keyword arguments for the `ssl.wrap_socket` function. The ``server_hostname`` argument will be used for certificate validation unless disabled in the ``ssl_options``. This method returns a `.Future` whose result is the new `SSLIOStream`. After this method has been called, any other operation on the original stream is undefined. If a close callback is defined on this stream, it will be transferred to the new stream. .. versionadded:: 4.0 .. versionchanged:: 4.2 SSL certificates are validated by default; pass ``ssl_options=dict(cert_reqs=ssl.CERT_NONE)`` or a suitably-configured `ssl.SSLContext` to disable. z+IOStream is not idle; cannot convert to SSLNF)r� � server_side�do_handshake_on_connect)� ssl_optionsr- c s0 � � � s � ��jp t� � �d ur�� d S d S r )Zdoner� r2 r r �rs Zorig_close_callback� ssl_streamr r �close_callback� s �z*IOStream.start_tls.<locals>.close_callbackc s � � ��S r )r� r )rs r� r r rh � s z$IOStream.start_tls.<locals>.<lambda>)rD rE rG rJ rL rM rQ rR r4 r7 � ValueErrorr r r� r- r� r rK r � SSLIOStreamr� �_ssl_connect_callbackr. r0 )r r� r� r� r� r� r r� r � start_tlsq sT #�������� �� zIOStream.start_tlsc C s� | j �t jt j�}|dkr/t �|t�|��| _| jd u r)t� d| j � � t j | � | � � d S | jd ur?| j}d | _| �|� | jd urO| j}d | _|�| � d| _d S )Nr r� F)r� r� r� r� r2 r� r� rM r r� rW r� � errorcodero rL r� r� rO )r �errrq rs r r r r� � s$ � zIOStream._handle_connectc C s� | j d urH| j jt jt jfv rJz| j �t jt j|rdnd� W d S t jyG } z|jtj kr5| � |�s<� W Y d }~d S W Y d }~d S d }~ww d S d S )Nr� r ) r� �family�AF_INET�AF_INET6� setsockopt� IPPROTO_TCP� TCP_NODELAYr2 r� �EINVALr� )r r� rt r r r r� � s � ����zIOStream.set_nodelayr� )r r! r" r# r rW rY r_ r] r\ r� r� r� r� r$ r r r r r� � s / AUr� c s� e Zd ZdZ� fdd�Z� fdd�Z� fdd�Zdd � Zd d � Zd d � Z dd� Z � fdd�Z � fdd�Z d!� fdd� Z � fdd�Zd"dd�Zdd � Zd d � Z� fd d �Z� ZS )#r� an A utility class to write to and read from a non-blocking SSL socket. If the socket passed to the constructor is already connected, it should be wrapped with:: ssl.wrap_socket(sock, do_handshake_on_connect=False, **kwargs) before constructing the `SSLIOStream`. Unconnected sockets will be wrapped when `IOStream.connect` is finished. c sz |� dt�| _tt| �j|i |�� d| _d| _d| _d| _ d| _ z| j � � W n t j y3 Y dS w | �| jj� dS )z�The ``ssl_options`` keyword argument may either be an `ssl.SSLContext` object or a dictionary of keywords arguments for `ssl.wrap_socket` r� TFN)�popr � _ssl_optionsr r� r �_ssl_accepting�_handshake_reading�_handshake_writingr� �_server_hostnamer� � getpeernamer2 r� r- r� )r r� r� r r r r � s �zSSLIOStream.__init__c � | j p tt| ��� S r )r r r� r� rV r r r r� � zSSLIOStream.readingc r r )r r r� r� rV r r r r� r zSSLIOStream.writingc C s | � |� d S r )r� r� r r r r� s z SSLIOStream._got_empty_writec C s� z d| _ d| _| j�� W n� tjy� } zn|jd tjkr(d| _ W Y d }~d S |jd tjkr:d| _W Y d }~d S |jd tj tj fv rQ| j dd�W Y d }~S |jd tj kr�z| j� � }W n tyk d}Y nw t�d| j�� ||� | j dd�W Y d }~S � d }~w tjy� } z | �|�s�|jd tjtjfv r�| j dd�W Y d }~S � d }~w ty� | j dd� Y S w d| _| �| j�� �s�| � � d S | �� d S )NFr Tra z(not connected)zSSL Error on %s %s: %s) r r r� � do_handshake�ssl�SSLErrorr� �SSL_ERROR_WANT_READ�SSL_ERROR_WANT_WRITE� SSL_ERROR_EOF�SSL_ERROR_ZERO_RETURNro � SSL_ERROR_SSLr r� r r� rW r2 r� r� �EBADF�ENOTCONN�AttributeErrorr � _verify_cert� getpeercert�_run_ssl_connect_callback)r r� Zpeerr r r �_do_ssl_handshake sR � � �� � � z SSLIOStream._do_ssl_handshakec C sH | j d ur| j }d | _ | �|� | jd ur"| j}d | _|�| � d S d S r )r� r� rN r� �r rq rs r r r r M s �z%SSLIOStream._run_ssl_connect_callbackc C s� t | jt�r| j�dtj�}n t | jtj�r| jj}|tjtjtj fv s&J �|tjks0| j du r2dS | j � � }|du rG|tjkrGt �d� dS z t|| j � W dS tyj } zt �d| � W Y d}~dS d}~ww )a Returns True if peercert is valid according to the configured validation mode and hostname. The ssl handshake already tested the certificate for a valid CA signature; the only thing that remains is to check the hostname. � cert_reqsNTzNo SSL certificate givenFzInvalid SSL certificate: %s)rx r �dict�getr � CERT_NONE� SSLContext� verify_mode� CERT_REQUIRED� CERT_OPTIONALr r� r r r� r r )r Zpeercertr �certrt r r r r W s& ���zSSLIOStream._verify_certc �$ | j r | �� d S tt| ��� d S r )r r r r� r� rV r r r r� r � zSSLIOStream._handle_readc r" r )r r r r� r� rV r r r r� x r# zSSLIOStream._handle_writeNc s( || _ tt| �j|dd� d� | �|�S )Nc S s d S r r r r r r rh � r� z%SSLIOStream.connect.<locals>.<lambda>)rq )r r r� r� �wait_for_handshake)r r� rq r� r r r r� ~ s zSSLIOStream.connectc sZ t t| ��� | �� r d S | j�| j� | j}d | _t| j| j | j dd�| _| � |� d S )NF)r� r� ) r r� r� r r- r� r� rP r r r r� )r Z old_stater r r r� � s �zSSLIOStream._handle_connectc C sT | j dus | jdurtd��|durt�|�| _ d}nt� }| _| js(| �� |S )a� Wait for the initial SSL handshake to complete. If a ``callback`` is given, it will be called with no arguments once the handshake is complete; otherwise this method returns a `.Future` which will resolve to the stream itself after the handshake is complete. Once the handshake is complete, information such as the peer's certificate and NPN/ALPN selections may be accessed on ``self.socket``. This method is intended for use on server-side streams or after using `IOStream.start_tls`; it should not be used with `IOStream.connect` (which already waits for the handshake to complete). It may only be called once per stream. .. versionadded:: 4.2 NzAlready waiting)r� rN � RuntimeErrorr rz r r r r r r r r$ � s z SSLIOStream.wait_for_handshakec C sX z)z | j �|�W W ~S tjy) } z|jd tjkr$W Y d }~W ~dS � d }~ww ~w r� )r� r� r r r� r )r r[ rt r r r r\ � s � ��� zSSLIOStream.write_to_fdc C s� | j rd S z | j�| j�}W n8 tjy+ } z|jd tjkr&W Y d }~d S � d }~w tjyF } z|jd t v rAW Y d }~d S � d }~ww |sO| � � d S |S r� ) r r� �readr0 r r r� r r2 r� ro r� r r r r] � s& ���zSSLIOStream.read_from_fdc s0 t |tj�r|jd tjkrdS tt| ��|�S )Nr T)rx r r r� r r r� r� )r rt r r r r� � s zSSLIOStream._is_connresetr� r )r r! r" r# r r� r� r� r r r r� r� r� r� r$ r\ r] r� r$ r r r r r� � s" . r� c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Z� Z S ) � PipeIOStreama Pipe-based `IOStream` implementation. The constructor takes an integer file descriptor (such as one returned by `os.pipe`) rather than an open file object. Pipes are generally one-way, so a `PipeIOStream` can be used for reading or writing but not both. c s( || _ t|� tt| �j|i |�� d S r )r� r r r' r )r r� r� r� r r r r � s zPipeIOStream.__init__c C r� r )r� rV r r r rW � r� zPipeIOStream.filenoc C s t �| j� d S r )r� ro r� rV r r r rY s zPipeIOStream.close_fdc C s z t �| j|�W ~S ~w r )r� r� r� rZ r r r r\ s zPipeIOStream.write_to_fdc C s� z t �| j| j�}W n0 ttfy: } z"t|�tv r!W Y d }~d S t|�tj kr5| j dd� W Y d }~d S � d }~ww |sC| � � d S |S )NTra ) r� r& r� r0 r� r� r r� r� r ro r� r r r r] s �� zPipeIOStream.read_from_fd) r r! r" r# r rW rY r\ r] r$ r r r r r' � s r' c C s dd l } | �� S r� )�doctestZ DocTestSuite)r( r r r �doctests s r) )?r# � __future__r r r rH r� ry r� r� r� rk Zsalt.ext.tornado.concurrentr Zsalt.ext.tornador Zsalt.ext.tornado.logr r Zsalt.ext.tornado.netutilr r r r r r Zsalt.ext.tornado.utilr Z salt.ext.tornado.platform.posixr � ImportErrorr � EWOULDBLOCK�EAGAINr� �hasattrr Z ECONNRESETZ ECONNABORTEDZEPIPEZ ETIMEDOUTr� r ZWSAECONNABORTEDZ WSAETIMEDOUT�platformZ EPROTOTYPEZ EINPROGRESSr� r � startswithr� r� r r� r& r( �objectr) r� r� r' r) r r r r �<module> sr � � � g .
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings